lex-bedrock 0.1.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 2143094bd8222adb62e51ac6548e82771a77186e59496993a1960e51c433de8b
+  data.tar.gz: 830eb21aac385a64755e418773263966c05d464229715057ddafe77b8ba5f8af
+SHA512:
+  metadata.gz: 63048c5e14420b13e1fcf5c7def1d23fe4f247e09ea9f7dfe3d333b8ca269950398ad1b7cadc3e96b547f857b604df5e92fd854a211f6b445973309fec613c79
+  data.tar.gz: 035fff2df7b594bd0c65c42bda4fed599632428890023968967f630eab3103b5f7b4ff96897b2ef0c38cc3a546463ba48e6b60c425a82d6897ffd9adde2308fd

data/.github/workflows/ci.yml

@@ -0,0 +1,16 @@
+name: CI
+on:
+  push:
+    branches: [origin]
+  pull_request:
+
+jobs:
+  ci:
+    uses: LegionIO/.github/.github/workflows/ci.yml@main
+
+  release:
+    needs: ci
+    if: github.event_name == 'push' && github.ref == 'refs/heads/origin'
+    uses: LegionIO/.github/.github/workflows/release.yml@main
+    secrets:
+      rubygems-api-key: ${{ secrets.RUBYGEMS_API_KEY }}

data/.gitignore

@@ -0,0 +1,12 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+
+# rspec failure tracking
+.rspec_status
+Gemfile.lock

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/.rubocop.yml

@@ -0,0 +1,51 @@
+AllCops:
+  TargetRubyVersion: 3.4
+  NewCops: enable
+  SuggestExtensions: false
+
+Layout/LineLength:
+  Max: 160
+
+Layout/SpaceAroundEqualsInParameterDefault:
+  EnforcedStyle: space
+
+Layout/HashAlignment:
+  EnforcedHashRocketStyle: table
+  EnforcedColonStyle: table
+
+Metrics/MethodLength:
+  Max: 50
+
+Metrics/ClassLength:
+  Max: 1500
+
+Metrics/ModuleLength:
+  Max: 1500
+
+Metrics/BlockLength:
+  Max: 40
+  Exclude:
+    - 'spec/**/*'
+    - '*.gemspec'
+
+Metrics/AbcSize:
+  Max: 60
+
+Metrics/CyclomaticComplexity:
+  Max: 15
+
+Metrics/PerceivedComplexity:
+  Max: 17
+
+Style/Documentation:
+  Enabled: false
+
+Style/SymbolArray:
+  Enabled: true
+
+Style/FrozenStringLiteralComment:
+  Enabled: true
+  EnforcedStyle: always
+
+Naming/FileName:
+  Enabled: false

data/CHANGELOG.md

@@ -0,0 +1,12 @@
+# Changelog
+
+## [0.1.0] - 2026-03-21
+
+### Added
+- Initial release of lex-bedrock
+- `Helpers::Client` module with `bedrock_client` and `bedrock_runtime_client` factory methods
+- `Runners::Models` with `list` (ListFoundationModels) and `get` (GetFoundationModel)
+- `Runners::Converse` with `create` using the Converse API (messages-style interface)
+- `Runners::Invoke` with `invoke_model` using the raw InvokeModel API
+- Standalone `Client` class including all runner modules with persistent config
+- Support for optional `session_token` for assumed-role credentials

data/Gemfile

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+source 'https://rubygems.org'
+gemspec
+
+gem 'rspec'
+gem 'rubocop'

data/README.md

@@ -0,0 +1,62 @@
+# lex-bedrock
+
+Legion Extension (LEX) that connects LegionIO to AWS Bedrock for foundation model inference.
+
+## Installation
+
+Add to your Gemfile:
+
+```ruby
+gem 'lex-bedrock'
+```
+
+## Configuration
+
+AWS credentials are required:
+
+- `access_key_id` — AWS access key ID
+- `secret_access_key` — AWS secret access key
+- `region` — AWS region (default: `us-east-2`)
+- `session_token` — optional, for assumed roles
+
+## Usage
+
+### Standalone Client
+
+```ruby
+client = Legion::Extensions::Bedrock::Client.new(
+  access_key_id: 'AKIA...',
+  secret_access_key: 'secret',
+  region: 'us-east-2'
+)
+
+# List foundation models
+result = client.list
+# => { models: [...] }
+
+# Converse API
+result = client.create(
+  model_id: 'anthropic.claude-3-5-sonnet-20241022-v2:0',
+  messages: [{ role: 'user', content: [{ text: 'Hello' }] }]
+)
+# => { result: ..., usage: ..., stop_reason: 'end_turn' }
+
+# Raw InvokeModel
+result = client.invoke_model(
+  model_id: 'anthropic.claude-3-5-sonnet-20241022-v2:0',
+  body: { prompt: 'Hello', max_tokens: 256 }
+)
+# => { result: { ... }, content_type: 'application/json' }
+```
+
+### Runner Modules
+
+```ruby
+include Legion::Extensions::Bedrock::Runners::Models
+include Legion::Extensions::Bedrock::Runners::Converse
+include Legion::Extensions::Bedrock::Runners::Invoke
+```
+
+## License
+
+MIT

data/lex-bedrock.gemspec

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require_relative 'lib/legion/extensions/bedrock/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'lex-bedrock'
+  spec.version       = Legion::Extensions::Bedrock::VERSION
+  spec.authors       = ['Esity']
+  spec.email         = ['matthewdiverson@gmail.com']
+
+  spec.summary       = 'LEX Bedrock'
+  spec.description   = 'Connects LegionIO to AWS Bedrock for foundation model inference'
+  spec.homepage      = 'https://github.com/LegionIO/lex-bedrock'
+  spec.license       = 'MIT'
+  spec.required_ruby_version = '>= 3.4'
+
+  spec.metadata['homepage_uri']        = spec.homepage
+  spec.metadata['source_code_uri']     = 'https://github.com/LegionIO/lex-bedrock'
+  spec.metadata['documentation_uri']   = 'https://github.com/LegionIO/lex-bedrock'
+  spec.metadata['changelog_uri']       = 'https://github.com/LegionIO/lex-bedrock'
+  spec.metadata['bug_tracker_uri']     = 'https://github.com/LegionIO/lex-bedrock/issues'
+  spec.metadata['rubygems_mfa_required'] = 'true'
+
+  spec.files = Dir.chdir(File.expand_path(__dir__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.require_paths = ['lib']
+
+  spec.add_dependency 'aws-sdk-bedrock'
+  spec.add_dependency 'aws-sdk-bedrockruntime'
+end

data/lib/legion/extensions/bedrock/client.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require 'legion/extensions/bedrock/helpers/client'
+require 'legion/extensions/bedrock/runners/models'
+require 'legion/extensions/bedrock/runners/converse'
+require 'legion/extensions/bedrock/runners/invoke'
+
+module Legion
+  module Extensions
+    module Bedrock
+      class Client
+        include Legion::Extensions::Bedrock::Runners::Models
+        include Legion::Extensions::Bedrock::Runners::Converse
+        include Legion::Extensions::Bedrock::Runners::Invoke
+
+        attr_reader :config
+
+        def initialize(access_key_id:, secret_access_key:, region: Helpers::Client::DEFAULT_REGION, **opts)
+          @config = { access_key_id: access_key_id, secret_access_key: secret_access_key,
+                      region: region, **opts }
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/bedrock/helpers/client.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require 'aws-sdk-bedrock'
+require 'aws-sdk-bedrockruntime'
+
+module Legion
+  module Extensions
+    module Bedrock
+      module Helpers
+        module Client
+          DEFAULT_REGION = 'us-east-2'
+
+          module_function
+
+          def bedrock_runtime_client(access_key_id:, secret_access_key:, region: DEFAULT_REGION,
+                                     session_token: nil, **)
+            opts = {
+              access_key_id:     access_key_id,
+              secret_access_key: secret_access_key,
+              region:            region
+            }
+            opts[:session_token] = session_token if session_token
+
+            Aws::BedrockRuntime::Client.new(**opts)
+          end
+
+          def bedrock_client(access_key_id:, secret_access_key:, region: DEFAULT_REGION,
+                             session_token: nil, **)
+            opts = {
+              access_key_id:     access_key_id,
+              secret_access_key: secret_access_key,
+              region:            region
+            }
+            opts[:session_token] = session_token if session_token
+
+            Aws::Bedrock::Client.new(**opts)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/bedrock/runners/converse.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require 'legion/extensions/bedrock/helpers/client'
+
+module Legion
+  module Extensions
+    module Bedrock
+      module Runners
+        module Converse
+          def create(model_id:, messages:, access_key_id:, secret_access_key:, # rubocop:disable Metrics/ParameterLists
+                     system: nil, max_tokens: 1024, temperature: nil,
+                     region: Helpers::Client::DEFAULT_REGION, **)
+            client = Helpers::Client.bedrock_runtime_client(
+              access_key_id:     access_key_id,
+              secret_access_key: secret_access_key,
+              region:            region
+            )
+
+            inference_config = { max_tokens: max_tokens }
+            inference_config[:temperature] = temperature if temperature
+
+            request = {
+              model_id:         model_id,
+              messages:         messages,
+              inference_config: inference_config
+            }
+            request[:system] = [{ text: system }] if system
+
+            response = client.converse(**request)
+            {
+              result:      response.output,
+              usage:       response.usage,
+              stop_reason: response.stop_reason
+            }
+          end
+
+          include Legion::Extensions::Helpers::Lex if Legion::Extensions.const_defined?(:Helpers) &&
+                                                      Legion::Extensions::Helpers.const_defined?(:Lex)
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/bedrock/runners/invoke.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require 'json'
+require 'legion/extensions/bedrock/helpers/client'
+
+module Legion
+  module Extensions
+    module Bedrock
+      module Runners
+        module Invoke
+          def invoke_model(model_id:, body:, access_key_id:, secret_access_key:, # rubocop:disable Metrics/ParameterLists
+                           content_type: 'application/json', accept: 'application/json',
+                           region: Helpers::Client::DEFAULT_REGION, **)
+            client = Helpers::Client.bedrock_runtime_client(
+              access_key_id:     access_key_id,
+              secret_access_key: secret_access_key,
+              region:            region
+            )
+
+            response = client.invoke_model(
+              model_id:     model_id,
+              body:         ::JSON.dump(body),
+              content_type: content_type,
+              accept:       accept
+            )
+
+            {
+              result:       ::JSON.parse(response.body.read),
+              content_type: response.content_type
+            }
+          end
+
+          include Legion::Extensions::Helpers::Lex if Legion::Extensions.const_defined?(:Helpers) &&
+                                                      Legion::Extensions::Helpers.const_defined?(:Lex)
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/bedrock/runners/models.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require 'legion/extensions/bedrock/helpers/client'
+
+module Legion
+  module Extensions
+    module Bedrock
+      module Runners
+        module Models
+          def list(access_key_id:, secret_access_key:, region: Helpers::Client::DEFAULT_REGION, **)
+            client = Helpers::Client.bedrock_client(
+              access_key_id:     access_key_id,
+              secret_access_key: secret_access_key,
+              region:            region
+            )
+            response = client.list_foundation_models
+            { models: response.model_summaries }
+          end
+
+          def get(model_id:, access_key_id:, secret_access_key:, region: Helpers::Client::DEFAULT_REGION, **)
+            client = Helpers::Client.bedrock_client(
+              access_key_id:     access_key_id,
+              secret_access_key: secret_access_key,
+              region:            region
+            )
+            response = client.get_foundation_model(model_identifier: model_id)
+            { model: response.model_details }
+          end
+
+          include Legion::Extensions::Helpers::Lex if Legion::Extensions.const_defined?(:Helpers) &&
+                                                      Legion::Extensions::Helpers.const_defined?(:Lex)
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/bedrock/version.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Bedrock
+      VERSION = '0.1.1'
+    end
+  end
+end

data/lib/legion/extensions/bedrock.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require 'legion/extensions/bedrock/version'
+require 'legion/extensions/bedrock/helpers/client'
+require 'legion/extensions/bedrock/runners/models'
+require 'legion/extensions/bedrock/runners/converse'
+require 'legion/extensions/bedrock/runners/invoke'
+
+module Legion
+  module Extensions
+    module Bedrock
+      extend Legion::Extensions::Core if Legion::Extensions.const_defined? :Core
+    end
+  end
+end

metadata

@@ -0,0 +1,89 @@
+--- !ruby/object:Gem::Specification
+name: lex-bedrock
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- Esity
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-bedrock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-bedrockruntime
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Connects LegionIO to AWS Bedrock for foundation model inference
+email:
+- matthewdiverson@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".github/workflows/ci.yml"
+- ".gitignore"
+- ".rspec"
+- ".rubocop.yml"
+- CHANGELOG.md
+- Gemfile
+- README.md
+- lex-bedrock.gemspec
+- lib/legion/extensions/bedrock.rb
+- lib/legion/extensions/bedrock/client.rb
+- lib/legion/extensions/bedrock/helpers/client.rb
+- lib/legion/extensions/bedrock/runners/converse.rb
+- lib/legion/extensions/bedrock/runners/invoke.rb
+- lib/legion/extensions/bedrock/runners/models.rb
+- lib/legion/extensions/bedrock/version.rb
+homepage: https://github.com/LegionIO/lex-bedrock
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/LegionIO/lex-bedrock
+  source_code_uri: https://github.com/LegionIO/lex-bedrock
+  documentation_uri: https://github.com/LegionIO/lex-bedrock
+  changelog_uri: https://github.com/LegionIO/lex-bedrock
+  bug_tracker_uri: https://github.com/LegionIO/lex-bedrock/issues
+  rubygems_mfa_required: 'true'
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '3.4'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: LEX Bedrock
+test_files: []