lex-llm-gemini 0.1.6 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dfe3fa2fee00d45f7908b00a559606ad3d64b7b7de557da8dec385e7e0e07ada
-  data.tar.gz: d03f2f8fde3f41aa6273743b738122ee1ea22c03e4a165308526ddd00919ce3d
+  metadata.gz: '0788daea481b47eead29eb391fa566a2f02540f1f0f5b4c8aca382b457de8c33'
+  data.tar.gz: 356fd313c188e9d585ca9d7a3f9f7fd8b1ab9d687edba232266ea168c306135f
 SHA512:
-  metadata.gz: 9d25e055451f57d145b12389d89bc92d4010236151aae33450396d738d959dfcb6950e5f80f39388a02870a71fb16e881164654b8c204be7165da8b22bb58816
-  data.tar.gz: 35144cf3d1e3ccab244650ce8e7b0d4f929eb94c663db4be9fe5930520f31e789084f8962331653c70c0c2fff6f159bdc1293ebf13519cb8a3c7cbed73325444
+  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,23 @@
 # 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.
+- Add `Legion::Logging::Helper` to every module and class in lib/.
+- Replace all bare rescue blocks and custom `log_publish_failure` with `handle_exception(e, level:, handled:, operation:)`.
+- Add info-level action logging for model listing and registry publishing.
+- Update README to reflect registry event publishing and observability patterns.
+
 ## 0.1.6 - 2026-04-28
 
 - Publish best-effort `llm.registry` discovered-model availability events when transport is already loaded.

data/README.md

@@ -1,6 +1,6 @@
 # lex-llm-gemini
 
-LegionIO LLM provider extension for Gemini.
+LegionIO LLM provider extension for Google Gemini.
 
 This gem lives under `Legion::Extensions::Llm::Gemini` and depends on `lex-llm` for shared provider-neutral routing, fleet, and schema primitives.
 
@@ -10,11 +10,24 @@ Load it with `require 'legion/extensions/llm/gemini'`.
 
 - `Legion::Extensions::Llm::Provider` registration as `:gemini`
 - Gemini-native chat requests through `POST /v1beta/{model=models/*}:generateContent`
-- streaming chat support through `POST /v1beta/{model=models/*}:streamGenerateContent?alt=sse`
-- model discovery through `GET /v1beta/models`
-- embeddings through `POST /v1beta/{model=models/*}:embedContent`
-- normalized chat, streaming, vision, function calling, and embedding capability mapping from `supportedGenerationMethods`
-- shared fleet/default settings via `Legion::Extensions::Llm.provider_settings`
+- Streaming chat support through `POST /v1beta/{model=models/*}:streamGenerateContent?alt=sse`
+- Model discovery through `GET /v1beta/models`
+- Embeddings through `POST /v1beta/{model=models/*}:embedContent`
+- Normalized chat, streaming, vision, function calling, and embedding capability mapping from `supportedGenerationMethods`
+- Best-effort `llm.registry` availability events published to AMQP when transport is loaded
+- Shared fleet/default settings via `Legion::Extensions::Llm.provider_settings`
+
+## File Map
+
+| Path | Purpose |
+|------|---------|
+| `lib/legion/extensions/llm/gemini.rb` | Entry point; provider registration |
+| `lib/legion/extensions/llm/gemini/provider.rb` | Gemini provider (chat, streaming, models, embeddings) |
+| `lib/legion/extensions/llm/gemini/registry_event_builder.rb` | Builds sanitized lex-llm registry event envelopes |
+| `lib/legion/extensions/llm/gemini/registry_publisher.rb` | Best-effort async publisher for model availability events |
+| `lib/legion/extensions/llm/gemini/transport/exchanges/llm_registry.rb` | `llm.registry` AMQP topic exchange |
+| `lib/legion/extensions/llm/gemini/transport/messages/registry_event.rb` | AMQP message wrapper for registry events |
+| `lib/legion/extensions/llm/gemini/version.rb` | `VERSION` constant |
 
 ## Defaults
 
@@ -45,3 +58,24 @@ Legion::Extensions::Llm.configure do |config|
   config.default_embedding_model = "gemini-embedding-001"
 end
 ```
+
+## Observability
+
+Every module and class includes or extends `Legion::Logging::Helper`:
+
+- **Info-level logging** on `list_models` and registry event publishing.
+- **All rescue blocks** call `handle_exception(e, level:, handled:, operation:)` for structured exception telemetry.
+- Registry publishing is best-effort; failures are handled at `:debug` or `:warn` level and never block the caller.
+
+## Development
+
+```bash
+bundle install
+bundle exec rspec       # 0 failures
+bundle exec rubocop -A  # auto-fix
+bundle exec rubocop     # lint check
+```
+
+## License
+
+Apache-2.0

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,6 +8,8 @@ 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
+
           class << self
             attr_writer :registry_publisher
 
@@ -17,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
 
@@ -94,7 +96,9 @@ module Legion
           end
 
           def list_models
+            log.info { 'listing available Gemini models' }
             super.tap do |models|
+              log.info { "discovered #{models.size} Gemini model(s); publishing to registry" }
               self.class.registry_publisher.publish_models_async(models, readiness: readiness(live: false))
             end
           end
@@ -279,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
@@ -298,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.6'
+        VERSION = '0.2.0'
       end
     end
   end

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

@@ -1,41 +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
 
         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.6
+  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,72 +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
-          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
-            :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,96 +0,0 @@
-# frozen_string_literal: true
-
-module Legion
-  module Extensions
-    module Llm
-      module Gemini
-        # Best-effort publisher for Gemini provider availability events.
-        class RegistryPublisher
-          APP_ID = 'lex-llm-gemini'
-
-          def initialize(builder: RegistryEventBuilder.new)
-            @builder = builder
-          end
-
-          def publish_models_async(models, readiness:)
-            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
-              log_publish_failure(e, level: :debug)
-            end
-          rescue StandardError => e
-            log_publish_failure(e, level: :debug)
-            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
-            log_publish_failure(e)
-            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
-            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
-            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
-
-          def log_publish_failure(error, level: :warn)
-            message = "[lex-llm-gemini] llm.registry publish failed: #{error.class}: #{error.message}"
-            logger = ::Legion::Extensions::Llm.logger if defined?(::Legion::Extensions::Llm)
-            if logger.respond_to?(level)
-              logger.public_send(level, message)
-            elsif logger.respond_to?(:debug)
-              logger.debug(message)
-            end
-          rescue StandardError
-            nil
-          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