lex-llm-gemini 0.3.9 → 0.3.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ff2a80df83dbda542e9400aeacc0451a3d1d2b2952e0bd37baab899e81628ed2
-  data.tar.gz: ed6ebc89bccce9798b540bd977f99dd73cd31b51b104f5047e16ce9d322a41ca
+  metadata.gz: d81052fa4455e1c73b7567864422ae494442f0fce0d9206fb0634bc238c939eb
+  data.tar.gz: 9f56e59676cc93aa0b55e4d8958975ce3e0a2c02308daaaeecb40c46c8535564
 SHA512:
-  metadata.gz: 9aed6d6d7b58f6dcd45d565106589b656b3e04cc99bc9ab74311a53bb3b8495e3bb587bc7e35cf7278e8a06ccd54fdc4b351d681cf08c903687ba6959b1c41ab
-  data.tar.gz: 50dd084cad1e94c3e5a16083b6b5a5f3b21080addf3f81a25bc5335163694f994baf075619878de9f87c978fde06a0b3a266d4fdcd11d1ce61eafdee936af1d2
+  metadata.gz: 0fe88e33de1fa4575867f35ffe717528a3759aa1b0e3c15fe42a9cfe99ff8e00826b0a8069f1d4788091b17448c307cf8e4ad694684c166249b9a57360c8dccc
+  data.tar.gz: 999f873f14e211211232b72ba3f5d5dd6dd31560de9e44081a1e2355c6574baa7a4a169701492a00803fcb9fb1bc6b9d73de37f399cb2439505a7c3c970717a7

data/CHANGELOG.md

@@ -1,5 +1,25 @@
 # Changelog
 
+## 0.3.13 - 2026-06-16
+
+- Dependency updates, code quality improvements.
+
+## 0.3.12 - 2026-06-15
+
+- **CapabilityPolicy integration** — `supportedGenerationMethods` mapped to `:model_metadata`. Settings overrides at provider/instance/model level supported.
+
+## 0.3.11 - 2026-06-13
+
+- **Gemfile cleanup** — Remove local path overrides; dependencies resolve from gemspec via rubygems.
+- **Dependency bump** — Require `lex-llm >= 0.5.0` for canonical types support.
+- **Canonical tool support** — Use `ToolSchema.extract` and add `:tools` capability.
+- **Bug fix** — Handle Array tool_calls in `tool_call_parts`.
+- 22 examples, 0 failures; 13 files, 0 rubocop offenses.
+
+## 0.3.10 - 2026-06-02
+
+- Add per-provider scoped discovery refresh actor
+
 ## 0.3.9 - 2026-05-21
 
 - api_base reads from settings[:endpoint] fallback

data/Gemfile

@@ -2,13 +2,6 @@
 
 source 'https://rubygems.org'
 
-group :test do
-  llm_base_path = ENV.fetch('LEX_LLM_PATH', File.expand_path('../lex-llm', __dir__))
-  transport_path = ENV.fetch('LEGION_TRANSPORT_PATH', File.expand_path('../../legion-transport', __dir__))
-  gem 'legion-transport', path: transport_path if File.directory?(transport_path)
-  gem 'lex-llm', path: llm_base_path if File.directory?(llm_base_path)
-end
-
 gemspec
 
 group :development do

data/lex-llm-gemini.gemspec

@@ -27,5 +27,5 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'legion-logging', '>= 1.3.2'
   spec.add_dependency 'legion-settings', '>= 1.3.14'
   spec.add_dependency 'legion-transport', '>= 1.4.14'
-  spec.add_dependency 'lex-llm', '>= 0.4.3'
+  spec.add_dependency 'lex-llm', '>= 0.5.0'
 end

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

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+begin
+  require 'legion/extensions/actors/every'
+rescue LoadError => e
+  warn(e.message) if $VERBOSE
+end
+
+return unless defined?(Legion::Extensions::Actors::Every)
+
+module Legion
+  module Extensions
+    module Llm
+      module Gemini
+        module Actor
+          class DiscoveryRefresh < Legion::Extensions::Actors::Every # rubocop:disable Style/Documentation
+            include Legion::Logging::Helper
+
+            REFRESH_INTERVAL = 1800
+
+            def runner_class    = self.class
+            def runner_function = 'manual'
+            def run_now?        = true
+            def use_runner?     = false
+            def check_subtask?  = false
+            def generate_task?  = false
+
+            def time
+              return REFRESH_INTERVAL unless defined?(Legion::Settings)
+
+              Legion::Settings.dig(:extensions, :llm, :gemini, :discovery_interval) || REFRESH_INTERVAL
+            end
+
+            def manual
+              log.debug('[gemini][discovery_refresh] refreshing model list')
+              return unless defined?(Legion::LLM::Discovery)
+
+              Legion::LLM::Discovery.refresh_discovered_models!(provider: :gemini)
+
+              if defined?(Legion::LLM::Router) && Legion::LLM::Router.respond_to?(:populate_auto_rules)
+                Legion::LLM::Router.populate_auto_rules(Legion::LLM::Discovery.discovered_instances)
+              end
+              if defined?(Legion::LLM::Inventory) && Legion::LLM::Inventory.respond_to?(:invalidate_offerings_cache!)
+                Legion::LLM::Inventory.invalidate_offerings_cache!
+              end
+            rescue StandardError => e
+              handle_exception(e, level: :warn, handled: true, operation: 'gemini.actor.discovery_refresh')
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -187,7 +187,9 @@ module Legion
           end
 
           def tool_call_parts(message)
-            message.tool_calls.values.map do |tool_call|
+            # Array is canonical (name-keyed hashes dropped parallel same-name calls)
+            calls = message.tool_calls.is_a?(Hash) ? message.tool_calls.values : Array(message.tool_calls)
+            calls.map do |tool_call|
               { functionCall: { name: tool_call.name, args: tool_call.arguments } }
             end
           end
@@ -204,9 +206,12 @@ module Legion
           def format_tools(tools)
             [{
               functionDeclarations: tools.values.map do |tool|
-                declaration = { name: tool.name, description: tool.description }
-                declaration[:parameters] = tool.params_schema if tool.params_schema
-                declaration
+                schema = Legion::Extensions::Llm::Canonical::ToolSchema.extract(tool)
+                {
+                  name: Legion::Extensions::Llm::Canonical::ToolSchema.tool_name(tool),
+                  description: Legion::Extensions::Llm::Canonical::ToolSchema.tool_description(tool),
+                  parameters: schema
+                }
               end
             }]
           end
@@ -307,6 +312,95 @@ module Legion
             end
           end
 
+          # -- policy resolution requires many sources
+          def offering_from_model(model, health: {})
+            policy = Legion::Extensions::Llm::CapabilityPolicy.resolve(
+              real: real_capabilities_from(model),
+              provider_catalog: {},
+              probe: {},
+              provider_envelope: {},
+              provider_config: provider_capability_config,
+              instance_config: instance_capability_config,
+              model_config: model_capability_config(model.id)
+            )
+
+            build_model_offering(model, policy, health)
+          end
+
+          def build_model_offering(model, policy, health)
+            Routing::ModelOffering.new(
+              provider_family: slug.to_sym,
+              provider_instance: model.instance || provider_instance_id,
+              transport: offering_transport,
+              tier: offering_tier,
+              model: model.id,
+              canonical_model_alias: model.name,
+              model_family: model.family,
+              usage_type: offering_usage_type(model),
+              capabilities: policy[:capabilities],
+              capability_sources: policy[:sources],
+              limits: offering_limits(model),
+              health:,
+              metadata: offering_metadata(model)
+            )
+          end
+
+          def real_capabilities_from(model)
+            meta = model.respond_to?(:metadata) ? (model.metadata || {}) : {}
+            methods = Array(meta[:supported_generation_methods] || meta['supported_generation_methods'])
+            {
+              streaming: methods.include?('streamGenerateContent'),
+              embeddings: methods.include?('embedContent'),
+              vision: Capabilities.vision?(meta.merge('name' => "models/#{model.id}"))
+            }.compact
+          end
+
+          def provider_capability_config
+            return {} unless defined?(Legion::Extensions::Llm::CredentialSources)
+
+            conf = Legion::Extensions::Llm::CredentialSources.setting(:extensions, :llm, :gemini)
+            conf.is_a?(Hash) ? conf.to_h.except(:instances, 'instances') : {}
+          rescue StandardError => e
+            handle_exception(e, level: :debug, handled: true, operation: 'gemini.provider_capability_config')
+            {}
+          end
+
+          def instance_capability_config
+            cfg = config
+            result = {}
+            %i[capabilities enable_streaming enable_tools enable_thinking enable_vision enable_embeddings
+               thinking_flag tools_flag streaming_flag vision_flag embedding_flag embeddings_flag
+               tool_flag images_flag image_flag].each do |key|
+              next unless cfg.respond_to?(key)
+
+              val = cfg.send(key)
+              result[key] = val unless val.nil?
+            rescue StandardError
+              next
+            end
+            result
+          end
+
+          def model_capability_config(model_id)
+            models_conf = resolve_models_config
+            return {} unless models_conf.respond_to?(:to_h)
+
+            hash = models_conf.to_h
+            hash[model_id.to_s] || hash[model_id.to_sym] || {}
+          rescue StandardError => e
+            handle_exception(e, level: :debug, handled: true, operation: 'gemini.model_capability_config')
+            {}
+          end
+
+          def resolve_models_config
+            return config.models if config.respond_to?(:models)
+            return config[:models] if config.respond_to?(:[])
+
+            nil
+          rescue StandardError
+            nil
+          end
+
           def modalities_for(methods)
             return [%w[text], %w[embeddings]] if methods.include?('embedContent')
 

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

@@ -4,7 +4,7 @@ module Legion
   module Extensions
     module Llm
       module Gemini
-        VERSION = '0.3.9'
+        VERSION = '0.3.13'
       end
     end
   end

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

@@ -3,6 +3,7 @@
 require 'legion/extensions/llm'
 require 'legion/extensions/llm/gemini/provider'
 require 'legion/extensions/llm/gemini/version'
+require_relative 'gemini/actors/discovery_refresh'
 
 module Legion
   module Extensions
@@ -30,10 +31,7 @@ module Legion
               fleet: {
                 enabled: false,
                 respond_to_requests: false,
-                capabilities: %i[chat stream_chat embed],
-                lanes: [],
-                concurrency: 4,
-                queue_suffix: nil
+                capabilities: %i[chat stream_chat embed tools]
               }
             }
           )
@@ -106,8 +104,7 @@ module Legion
                              :add_settings_api_key, :add_settings_instances, :normalize_instance_config,
                              :sanitize_instance_config
 
-        Legion::Extensions::Llm::Configuration.register_provider_options(Provider.configuration_options) if
-          Legion::Extensions::Llm::Configuration.respond_to?(:register_provider_options)
+        Legion::Extensions::Llm::Configuration.register_provider_options(Provider.configuration_options)
       end
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-llm-gemini
 version: !ruby/object:Gem::Version
-  version: 0.3.9
+  version: 0.3.13
 platform: ruby
 authors:
 - LegionIO
@@ -71,14 +71,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.4.3
+        version: 0.5.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.4.3
+        version: 0.5.0
 description: Gemini provider integration for the LegionIO LLM routing framework.
 email:
 - matthewdiverson@gmail.com
@@ -97,6 +97,7 @@ files:
 - README.md
 - lex-llm-gemini.gemspec
 - lib/legion/extensions/llm/gemini.rb
+- lib/legion/extensions/llm/gemini/actors/discovery_refresh.rb
 - lib/legion/extensions/llm/gemini/actors/fleet_worker.rb
 - lib/legion/extensions/llm/gemini/provider.rb
 - lib/legion/extensions/llm/gemini/runners/fleet_worker.rb