lex-llm-bedrock 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: eafb04fa29d1bf032323c4040c7cdbe0ea2d3cf1c72328864ec54b5e2f7ce117
-  data.tar.gz: 6da27c531876ff32af29b783548b86bed7a4146f45f0b94da679320dd594310b
+  metadata.gz: 2fdf5749de02e358b9458e391edfc790c954d252a29380e3ca2a6ba4c34e085b
+  data.tar.gz: 4662cd33fd1fa36622850662707ef618d936ab96b74ed8dc73680153b5e475c2
 SHA512:
-  metadata.gz: 7b7a1bc938f269380f0310cfd0e4a2f3561e986ed09db904dbe6a31663451658bac03e56b54c57f3c32efd85c7d83eb64e8d50fdbeac20efe2314f6a9e5d0a38
-  data.tar.gz: db293ebc885b48aa2902ecccfd5cf8e47bf58fd775ec211d31cb9a491dcecb60c5146989175affd2e1c59bd3f2429492eea94f36d202f41955708d72a3ec5780
+  metadata.gz: 5117d3c2035ee76b7d3caa18079724b8246aa617765960b32c4b31ab7118171b55a599e87ffb6b712e77ab4bdf00e4fac29f70d0932bd1ea7436b18ea9c4e318
+  data.tar.gz: a0457d98e4da2489fce6c0ee0458b9567606a7cbe71cbe2e64d7b8d75030245cd5f11cbf601ed326eca903f5b5c0579a208f8ab7b5ac783f18421363768a5ad9

data/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## 0.3.0 - 2026-05-01
+
+- Add auto-discovery via CredentialSources and AutoRegistration from lex-llm 0.3.0
+- Self-register discovered instances into Call::Registry at require-time
+- Require lex-llm >= 0.3.0
+
+
 ## 0.2.0 - 2026-04-30
 
 - Adopt lex-llm 0.1.9 base contract: flat `default_settings`, base `RegistryPublisher`, base `RegistryEventBuilder`.

data/lex-llm-bedrock.gemspec

@@ -28,5 +28,5 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'legion-json', '>= 1.2.1'
   spec.add_dependency 'legion-logging', '>= 1.3.2'
   spec.add_dependency 'legion-settings', '>= 1.3.14'
-  spec.add_dependency 'lex-llm', '>= 0.2.0'
+  spec.add_dependency 'lex-llm', '>= 0.3.0'
 end

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

@@ -4,7 +4,7 @@ module Legion
   module Extensions
     module Llm
       module Bedrock
-        VERSION = '0.2.0'
+        VERSION = '0.3.0'
       end
     end
   end

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

@@ -10,14 +10,18 @@ module Legion
       # Amazon Bedrock provider extension namespace.
       module Bedrock
         extend ::Legion::Extensions::Core if ::Legion::Extensions.const_defined?(:Core, false)
+        extend Legion::Logging::Helper
+        extend Legion::Extensions::Llm::AutoRegistration
 
         PROVIDER_FAMILY = :bedrock
 
+        DEFAULT_REGION = 'us-east-2'
+
         def self.default_settings
           {
             enabled: false,
             default_model: 'us.anthropic.claude-sonnet-4-6',
-            region: 'us-east-2',
+            region: DEFAULT_REGION,
             bearer_token: nil,
             api_key: nil,
             secret_key: nil,
@@ -37,11 +41,102 @@ module Legion
         def self.registry_publisher
           @registry_publisher ||= Legion::Extensions::Llm::RegistryPublisher.new(provider_family: PROVIDER_FAMILY)
         end
+
+        def self.discover_instances
+          candidates = {}
+          discover_env_bearer(candidates)
+          discover_claude_bearer(candidates)
+          discover_env_sigv4(candidates)
+          discover_settings(candidates)
+          discover_broker(candidates)
+          CredentialSources.dedup_credentials(candidates)
+        end
+
+        def self.discover_env_bearer(candidates)
+          bearer = CredentialSources.env('AWS_BEARER_TOKEN_BEDROCK')
+          return unless bearer
+
+          candidates[:env_bearer] = {
+            bearer_token: bearer,
+            bedrock_region: CredentialSources.env('AWS_DEFAULT_REGION') || DEFAULT_REGION,
+            tier: :cloud
+          }
+        end
+
+        def self.discover_claude_bearer(candidates)
+          claude_bearer = CredentialSources.claude_env_value('AWS_BEARER_TOKEN_BEDROCK')
+          claude_bearer ||= claude_env_pattern_match
+          return unless claude_bearer
+
+          candidates[:claude] = {
+            bearer_token: claude_bearer,
+            bedrock_region: CredentialSources.claude_env_value('AWS_DEFAULT_REGION') || DEFAULT_REGION,
+            tier: :cloud
+          }
+        end
+
+        def self.discover_env_sigv4(candidates)
+          akid = CredentialSources.env('AWS_ACCESS_KEY_ID')
+          skey = CredentialSources.env('AWS_SECRET_ACCESS_KEY')
+          return unless akid && skey
+
+          candidates[:env_sigv4] = {
+            api_key: akid, bedrock_access_key_id: akid, bedrock_secret_access_key: skey,
+            bedrock_session_token: CredentialSources.env('AWS_SESSION_TOKEN'),
+            bedrock_region: CredentialSources.env('AWS_DEFAULT_REGION') || DEFAULT_REGION, tier: :cloud
+          }.compact
+        end
+
+        def self.discover_settings(candidates)
+          settings = CredentialSources.setting(:extensions, :llm, :bedrock)
+          candidates[:settings] = settings.merge(tier: :cloud) if settings.is_a?(Hash) && !settings.empty?
+        end
+
+        def self.discover_broker(candidates)
+          return unless defined?(Legion::Identity::Broker)
+
+          broker_creds = broker_aws_credentials
+          candidates[:broker] = broker_creds.merge(tier: :cloud) if broker_creds
+        end
+
+        # Scan Claude config env hash for any key containing all of
+        # AWS, BEARER, TOKEN, and BEDROCK fragments (case-insensitive).
+        def self.claude_env_pattern_match
+          env_hash = CredentialSources.claude_config_value(:env)
+          return nil unless env_hash.is_a?(Hash)
+
+          fragments = %w[AWS BEARER TOKEN BEDROCK]
+          _key, value = env_hash.find do |k, _v|
+            upper = k.to_s.upcase
+            fragments.all? { |frag| upper.include?(frag) }
+          end
+          value
+        end
+
+        # Fetch AWS credentials from the Legion Identity Broker.
+        def self.broker_aws_credentials
+          return nil unless defined?(Legion::Identity::Broker)
+
+          creds = Legion::Identity::Broker.credentials_for(:aws)
+          return nil unless creds.is_a?(Hash)
+
+          akid = creds[:access_key_id] || creds['access_key_id']
+          return nil unless akid
+
+          { api_key: akid, bedrock_access_key_id: akid,
+            bedrock_secret_access_key: creds[:secret_access_key] || creds['secret_access_key'],
+            bedrock_session_token: creds[:session_token] || creds['session_token'],
+            bedrock_region: creds[:region] || creds['region'] || DEFAULT_REGION }.compact
+        end
       end
     end
   end
 end
 
-Legion::Extensions::Llm::Configuration.register_provider_options(
-  Legion::Extensions::Llm::Bedrock::Provider.configuration_options
-)
+if Legion::Extensions::Llm::Configuration.respond_to?(:register_provider_options)
+  Legion::Extensions::Llm::Configuration.register_provider_options(
+    Legion::Extensions::Llm::Bedrock::Provider.configuration_options
+  )
+end
+
+Legion::Extensions::Llm::Bedrock.register_discovered_instances

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-llm-bedrock
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - LegionIO
@@ -85,14 +85,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.0
+        version: 0.3.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.0
+        version: 0.3.0
 description: Amazon Bedrock provider integration for the LegionIO LLM routing framework.
 email:
 - matthewdiverson@gmail.com