lex-llm-gemini 0.1.7 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3b4d2bbf521fe7cd5757553aff5a6d61c1a9e9ffbe18c683058710ab19a8a6ff
-  data.tar.gz: 98d0db7b428b68d0543682c43a27d77fde00aee8bb02377195c37224e9429fef
+  metadata.gz: '0788daea481b47eead29eb391fa566a2f02540f1f0f5b4c8aca382b457de8c33'
+  data.tar.gz: 356fd313c188e9d585ca9d7a3f9f7fd8b1ab9d687edba232266ea168c306135f
 SHA512:
-  metadata.gz: c40e3408f31507e4de8622e65676a2eb67849a05964dee02585545647563a4655fe808a3673782b1819bc7828a1f9fd3af900aa57ce09f6119ba28caefe01813
-  data.tar.gz: bacceead0f9edceed7da8efb605fe516f0ea7110c57f47cf8b16bad77805e74d8a1a99d37ffec056fa4affdd2c4e6a30f45c7a5812c03221182038373a45c784
+  metadata.gz: '09edc833e89707be34a90292622615b2fcd391d19cee2baa73d29ed7a6f12b1d520cee837684f9db28349488e7d54f7ff312414b6e3ede280627da4d821d456e'
+  data.tar.gz: 634a5703a52c3d0554a47141265430ea1dbe10b43dfef892923ea4b0b4e7a278a914a76a9c1c109196ea27f09d9c5fb2dabdade8b302aff888185f9731a236b4

data/.rubocop.yml

@@ -14,5 +14,7 @@ Metrics/BlockLength:
     - spec/**/*
 Metrics/MethodLength:
   Enabled: false
+RSpec/ExampleLength:
+  Enabled: false
 RSpec/MultipleExpectations:
   Enabled: false

data/CHANGELOG.md

@@ -1,5 +1,15 @@
 # Changelog
 
+## 0.2.0 - 2026-04-30
+
+- **Breaking**: Adopt the base contract from lex-llm 0.1.9.
+- Replace `default_settings` with a flat settings hash (enabled, default_model, api_key, whitelist/blacklist, tls, instances).
+- Remove local `RegistryPublisher` and `RegistryEventBuilder`; use the parameterized base classes from lex-llm.
+- Remove local `transport/` directory (exchanges and messages); the shared `llm.registry` transport in lex-llm is used instead.
+- Remove the deprecated `Provider.register` call; configuration options are registered directly.
+- Update `parse_list_models_response` to use the new `Model::Info` constructor (context_length, modalities_input/output, metadata for max_output_tokens).
+- Require `lex-llm >= 0.1.9`.
+
 ## 0.1.7 - 2026-04-30
 
 - Audit logging, rescue blocks, and README for full observability.

data/lex-llm-gemini.gemspec

@@ -26,5 +26,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.1.5'
+  spec.add_dependency 'lex-llm', '>= 0.2.0'
 end

data/lib/legion/extensions/llm/gemini/provider.rb

@@ -8,7 +8,7 @@ module Legion
       module Gemini
         # Gemini provider implementation for the Legion::Extensions::Llm base provider contract.
         class Provider < Legion::Extensions::Llm::Provider # rubocop:disable Metrics/ClassLength
-          include Legion::Logging::Helper if defined?(Legion::Logging::Helper)
+          include Legion::Logging::Helper
 
           class << self
             attr_writer :registry_publisher
@@ -19,7 +19,7 @@ module Legion
             def capabilities = Capabilities
 
             def registry_publisher
-              @registry_publisher ||= RegistryPublisher.new
+              @registry_publisher ||= Legion::Extensions::Llm::RegistryPublisher.new(provider_family: :gemini)
             end
           end
 
@@ -283,16 +283,18 @@ module Legion
             Array(response.body['models']).map do |model_data|
               model_id = model_data.fetch('name').delete_prefix('models/')
               methods = Array(model_data['supportedGenerationMethods'])
+              input_mods, output_mods = modalities_for(methods)
 
               Legion::Extensions::Llm::Model::Info.new(
                 id: model_id,
                 name: model_data['displayName'] || model_id,
                 provider: provider,
-                context_window: model_data['inputTokenLimit'],
-                max_output_tokens: model_data['outputTokenLimit'],
+                context_length: model_data['inputTokenLimit'],
                 capabilities: capabilities.critical_capabilities_for(model_data),
-                modalities: modalities_for(methods),
+                modalities_input: input_mods,
+                modalities_output: output_mods,
                 metadata: {
+                  max_output_tokens: model_data['outputTokenLimit'],
                   version: model_data['version'],
                   description: model_data['description'],
                   supported_generation_methods: methods
@@ -302,9 +304,9 @@ module Legion
           end
 
           def modalities_for(methods)
-            return { input: %w[text], output: %w[embeddings] } if methods.include?('embedContent')
+            return [%w[text], %w[embeddings]] if methods.include?('embedContent')
 
-            { input: %w[text image audio video], output: %w[text] }
+            [%w[text image audio video], %w[text]]
           end
 
           def render_embedding_payload(text, model:, dimensions:)

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

@@ -4,7 +4,7 @@ module Legion
   module Extensions
     module Llm
       module Gemini
-        VERSION = '0.1.7'
+        VERSION = '0.2.0'
       end
     end
   end

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

@@ -1,42 +1,39 @@
 # frozen_string_literal: true
 
 require 'legion/extensions/llm'
-require 'legion/extensions/llm/gemini/registry_event_builder'
-require 'legion/extensions/llm/gemini/registry_publisher'
 require 'legion/extensions/llm/gemini/provider'
 require 'legion/extensions/llm/gemini/version'
 
 module Legion
   module Extensions
+    # LLM provider framework namespace (reopened by provider extensions).
     module Llm
       # Gemini provider extension namespace.
       module Gemini
         extend ::Legion::Extensions::Core if ::Legion::Extensions.const_defined?(:Core, false)
-        extend Legion::Logging::Helper if defined?(Legion::Logging::Helper)
+        extend Legion::Logging::Helper
 
         PROVIDER_FAMILY = :gemini
 
         def self.default_settings
-          ::Legion::Extensions::Llm.provider_settings(
-            family: PROVIDER_FAMILY,
-            instance: {
-              endpoint: 'https://generativelanguage.googleapis.com/v1beta',
-              tier: :frontier,
-              transport: :http,
-              credentials: { api_key: 'env://GEMINI_API_KEY' },
-              usage: { inference: true, embedding: true },
-              limits: { concurrency: 4 }
-            }
-          )
+          {
+            enabled: false,
+            default_model: 'gemini-2.0-flash',
+            api_key: nil,
+            model_whitelist: [],
+            model_blacklist: [],
+            model_cache_ttl: 3600,
+            tls: { enabled: false, verify: :peer },
+            instances: {}
+          }
         end
 
         def self.provider_class
           Provider
         end
       end
+
+      Configuration.register_provider_options(Gemini::Provider.configuration_options)
     end
   end
 end
-
-Legion::Extensions::Llm::Provider.register(Legion::Extensions::Llm::Gemini::PROVIDER_FAMILY,
-                                           Legion::Extensions::Llm::Gemini::Provider)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-llm-gemini
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.2.0
 platform: ruby
 authors:
 - LegionIO
@@ -57,14 +57,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.1.5
+        version: 0.2.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.1.5
+        version: 0.2.0
 description: Gemini provider integration for the LegionIO LLM routing framework.
 email:
 - matthewdiverson@gmail.com
@@ -84,10 +84,6 @@ files:
 - lex-llm-gemini.gemspec
 - lib/legion/extensions/llm/gemini.rb
 - lib/legion/extensions/llm/gemini/provider.rb
-- lib/legion/extensions/llm/gemini/registry_event_builder.rb
-- lib/legion/extensions/llm/gemini/registry_publisher.rb
-- lib/legion/extensions/llm/gemini/transport/exchanges/llm_registry.rb
-- lib/legion/extensions/llm/gemini/transport/messages/registry_event.rb
 - lib/legion/extensions/llm/gemini/version.rb
 homepage: https://github.com/LegionIO/lex-llm-gemini
 licenses:

data/lib/legion/extensions/llm/gemini/registry_event_builder.rb

@@ -1,75 +0,0 @@
-# frozen_string_literal: true
-
-module Legion
-  module Extensions
-    module Llm
-      module Gemini
-        # Builds sanitized lex-llm registry envelopes for Gemini provider state.
-        class RegistryEventBuilder
-          include Legion::Logging::Helper if defined?(Legion::Logging::Helper)
-
-          def model_available(model, readiness:)
-            registry_event_class.available(
-              model_offering(model),
-              runtime: runtime_metadata,
-              health: model_health(readiness),
-              metadata: model_metadata(model)
-            )
-          end
-
-          private
-
-          def model_offering(model)
-            {
-              provider_family: :gemini,
-              provider_instance: provider_instance,
-              transport: :http,
-              model: model.id,
-              usage_type: usage_type_for(model),
-              capabilities: Array(model.capabilities).map(&:to_sym),
-              limits: model_limits(model),
-              metadata: { lex: :llm_gemini, model_name: model.name }.compact
-            }
-          end
-
-          def model_health(readiness)
-            ready = readiness.fetch(:ready, true) == true
-            { ready:, status: ready ? :available : :degraded }
-          end
-
-          def model_metadata(model)
-            { extension: :lex_llm_gemini, provider: :gemini, model_type: model.type }
-          end
-
-          def runtime_metadata
-            { node: provider_instance }
-          end
-
-          def model_limits(model)
-            {
-              context_window: model.context_window,
-              max_output_tokens: model.max_output_tokens
-            }.compact
-          end
-
-          def usage_type_for(model)
-            model.type == 'embedding' ? :embedding : :inference
-          end
-
-          def provider_instance
-            configured_node = (::Legion::Settings.dig(:node, :canonical_name) if defined?(::Legion::Settings))
-            value = configured_node.to_s.strip
-            value.empty? ? :gemini : value.to_sym
-          rescue StandardError => e
-            handle_exception(e, level: :debug, handled: true, operation: 'gemini.registry.provider_instance')
-            :gemini
-          end
-
-          def registry_event_class
-            ::Legion::Extensions::Llm::Routing::RegistryEvent
-          end
-        end
-      end
-    end
-  end
-end

data/lib/legion/extensions/llm/gemini/registry_publisher.rb

@@ -1,89 +0,0 @@
-# frozen_string_literal: true
-
-module Legion
-  module Extensions
-    module Llm
-      module Gemini
-        # Best-effort publisher for Gemini provider availability events.
-        class RegistryPublisher
-          include Legion::Logging::Helper if defined?(Legion::Logging::Helper)
-
-          APP_ID = 'lex-llm-gemini'
-
-          def initialize(builder: RegistryEventBuilder.new)
-            @builder = builder
-          end
-
-          def publish_models_async(models, readiness:)
-            log.info { "publishing #{Array(models).size} model event(s) to llm.registry" }
-            schedule do
-              Array(models).each do |model|
-                publish_event(@builder.model_available(model, readiness:))
-              end
-            end
-          end
-
-          private
-
-          def schedule(&)
-            return false unless publishing_available?
-
-            Thread.new do
-              Thread.current.abort_on_exception = false
-              yield
-            rescue StandardError => e
-              handle_exception(e, level: :debug, handled: true, operation: 'gemini.registry.schedule_thread')
-            end
-          rescue StandardError => e
-            handle_exception(e, level: :debug, handled: true, operation: 'gemini.registry.schedule')
-            false
-          end
-
-          def publish_event(event)
-            return false unless publishing_available?
-
-            message_class.new(event:, app_id: APP_ID).publish(spool: false)
-          rescue StandardError => e
-            handle_exception(e, level: :warn, handled: true, operation: 'gemini.registry.publish_event')
-            false
-          end
-
-          def publishing_available?
-            return false unless registry_event_available?
-            return false unless transport_message_available?
-            return true unless defined?(::Legion::Transport::Connection)
-            return true unless ::Legion::Transport::Connection.respond_to?(:session_open?)
-
-            ::Legion::Transport::Connection.session_open?
-          rescue StandardError => e
-            handle_exception(e, level: :debug, handled: true, operation: 'gemini.registry.publishing_available?')
-            false
-          end
-
-          def registry_event_available?
-            defined?(::Legion::Extensions::Llm::Routing::RegistryEvent)
-          end
-
-          def transport_message_available?
-            return true if message_class_defined?
-            return false unless defined?(::Legion::Transport::Message) && defined?(::Legion::Transport::Exchange)
-
-            require 'legion/extensions/llm/gemini/transport/messages/registry_event'
-            message_class_defined?
-          rescue LoadError => e
-            handle_exception(e, level: :debug, handled: true, operation: 'gemini.registry.transport_load')
-            false
-          end
-
-          def message_class_defined?
-            defined?(::Legion::Extensions::Llm::Gemini::Transport::Messages::RegistryEvent)
-          end
-
-          def message_class
-            ::Legion::Extensions::Llm::Gemini::Transport::Messages::RegistryEvent
-          end
-        end
-      end
-    end
-  end
-end

data/lib/legion/extensions/llm/gemini/transport/exchanges/llm_registry.rb

@@ -1,24 +0,0 @@
-# frozen_string_literal: true
-
-module Legion
-  module Extensions
-    module Llm
-      module Gemini
-        module Transport
-          module Exchanges
-            # Topic exchange for Gemini provider availability events.
-            class LlmRegistry < ::Legion::Transport::Exchange
-              def exchange_name
-                'llm.registry'
-              end
-
-              def default_type
-                'topic'
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/legion/extensions/llm/gemini/transport/messages/registry_event.rb

@@ -1,42 +0,0 @@
-# frozen_string_literal: true
-
-require 'legion/extensions/llm/gemini/transport/exchanges/llm_registry'
-
-module Legion
-  module Extensions
-    module Llm
-      module Gemini
-        module Transport
-          module Messages
-            # Publishes lex-llm RegistryEvent envelopes to the llm.registry exchange.
-            class RegistryEvent < ::Legion::Transport::Message
-              def initialize(event:, **options)
-                super(**event.to_h.merge(options))
-              end
-
-              def exchange
-                Transport::Exchanges::LlmRegistry
-              end
-
-              def routing_key
-                @options[:routing_key] || "llm.registry.#{@options.fetch(:event_type)}"
-              end
-
-              def type
-                'llm.registry.event'
-              end
-
-              def app_id
-                @options[:app_id] || RegistryPublisher::APP_ID
-              end
-
-              def persistent # rubocop:disable Naming/PredicateMethod
-                false
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-end