lex-llm-vertex 0.2.8 → 0.2.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d7dbca12cb0f99264ee9fd713291622b0ee9ad6f238193811bdcfb2847be3211
-  data.tar.gz: be61eafce74bdaec30c737ec906abafbbacad3f4c125b52199db8de8dd3d3ca6
+  metadata.gz: acd43f914dff394b9635eb7596dd108016764e573fe35a477f267c3ab8465971
+  data.tar.gz: 2ee16d4671f38bd7c5ce4de50ea6abd3b1ad4609c06fcf559d89e224c36baa77
 SHA512:
-  metadata.gz: f557d2fbf831656f3aa0d4a0788ecfedab7889f5d9fdd666300ddad2be3348184e56c8bc0f5d3a469b64d9d12e2fae846bcfbd97100ff6a8b1f755cdaea803c2
-  data.tar.gz: 90199aa13e8d105931040d6131f9c1591cc7bc956dd61e0046589a7eba06d0ab5aae04cd8fdce276a28adafd980e8c3a4a781687bc626e31097a719a8c47220b
+  metadata.gz: 47e8e573e518356cc1de63b8f0120bd96024d22266a7a0e37ff9d3e8ed5528028f0a30253d7a48a59d881a5043be6b7100becd6d45ffe13815b5fad715d9647a
+  data.tar.gz: '09d51a90aa46a6595aa611e998f2de12e67d761563c17bab6919fa77ec25c3e7eabf139ae2644915ce0a435e887b217d9c509dcec6f74faf4b75e484300f76e7'

data/CHANGELOG.md

@@ -1,5 +1,14 @@
 # Changelog
 
+## 0.2.9 - 2026-05-21
+
+- Add `default_transport`/`default_tier` class declarations, remove `configured_transport`/`configured_tier`
+- Remove `DEFAULT_LOCATION`/`DEFAULT_PROJECT`/`DEFAULT_PUBLISHER` constants — now read from settings
+- Add `model_allowed?` filtering in `discover_offerings`
+- Default tier corrected from :frontier to :cloud
+- Identity headers included via base provider
+
+
 ## 0.2.8 - 2026-05-18
 
 - Fix streaming tool calls: `build_chunk` now passes `tool_calls: parse_tool_calls(parts)` to the Chunk constructor. Previously tool calls were omitted from streaming responses entirely.

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

@@ -11,10 +11,6 @@ module Legion
       module Vertex
         # Google Cloud Vertex AI provider implementation for the Legion::Extensions::Llm contract.
         class Provider < Legion::Extensions::Llm::Provider # rubocop:disable Metrics/ClassLength
-          DEFAULT_LOCATION = 'us-central1'
-          DEFAULT_PROJECT = 'env://GOOGLE_CLOUD_PROJECT'
-          DEFAULT_PUBLISHER = 'google'
-
           STATIC_MODELS = [
             { model: 'gemini-2.5-flash', alias: 'gemini-flash', publisher: 'google', model_family: :gemini },
             { model: 'gemini-2.5-pro', alias: 'gemini-pro', publisher: 'google', model_family: :gemini },
@@ -39,6 +35,8 @@ module Legion
             attr_writer :registry_publisher
 
             def slug = 'vertex'
+            def default_transport = :http
+            def default_tier = :cloud
 
             def configuration_options
               %i[
@@ -83,16 +81,22 @@ module Legion
             end
           end
 
+          def settings
+            Vertex.default_settings
+          end
+
           def api_base
             config.vertex_api_base || "https://#{location}-aiplatform.googleapis.com/v1"
           end
 
           def headers
-            { 'Authorization' => bearer_token, 'Content-Type' => 'application/json; charset=utf-8' }.compact
+            identity_headers.merge({ 'Authorization' => bearer_token,
+                                     'Content-Type' => 'application/json; charset=utf-8' }.compact)
           end
 
-          def project = config.vertex_project || ENV.fetch('GOOGLE_CLOUD_PROJECT', DEFAULT_PROJECT)
-          def location = config.vertex_location || DEFAULT_LOCATION
+          def project = config.vertex_project || settings[:project] || ENV.fetch('GOOGLE_CLOUD_PROJECT', nil)
+          def location = config.vertex_location || settings[:location] || 'us-central1'
+          def default_publisher = settings[:publisher] || 'google'
           def models_url = publisher_parent
           def completion_url = generate_content_url(model: @model || STATIC_MODELS.first.fetch(:model))
           def stream_url = stream_generate_content_url(model: @model || STATIC_MODELS.first.fetch(:model))
@@ -126,7 +130,13 @@ module Legion
 
             response = connection.get(models_url)
             models = response.body['publisherModels'] || response.body['models'] || []
-            offerings = models.map { |model| offering_from_live_model(model) }
+            offerings = models.filter_map do |model|
+              offering = offering_from_live_model(model)
+              model_id = offering.respond_to?(:model) ? offering.model : (offering[:model] || offering[:id])
+              next unless model_allowed?(model_id.to_s)
+
+              offering
+            end
             log.info { "discovered #{offerings.size} live offering(s) from Vertex" }
             model_infos = offerings.map { |o| model_info_from_offering(o) }
             self.class.registry_publisher.publish_models_async(model_infos, readiness: readiness(live: false))
@@ -310,7 +320,7 @@ module Legion
 
           def offering_from_live_model(model)
             name = model['name'] || model['publisherModelName'] || model['model'] || model['id']
-            publisher = publisher_from_resource(name) || model['publisher'] || DEFAULT_PUBLISHER
+            publisher = publisher_from_resource(name) || model['publisher'] || default_publisher
             id = name.to_s.split('/').last
             offering_for(model: id, publisher:, metadata: model)
           end
@@ -320,8 +330,8 @@ module Legion
             Legion::Extensions::Llm::Routing::ModelOffering.new(
               provider_family: :vertex,
               instance_id: instance_id,
-              transport: configured_transport(:http),
-              tier: configured_tier(:frontier),
+              transport: offering_transport,
+              tier: offering_tier,
               model: model,
               usage_type: usage_type,
               capabilities: default_capabilities(model, api:),
@@ -337,16 +347,8 @@ module Legion
             )
           end
 
-          def configured_transport(default)
-            config.respond_to?(:transport) ? config.transport : default
-          end
-
-          def configured_tier(default)
-            config.respond_to?(:tier) ? config.tier : default
-          end
-
           def publisher_parent
-            "projects/#{project}/locations/#{location}/publishers/#{DEFAULT_PUBLISHER}/models"
+            "projects/#{project}/locations/#{location}/publishers/#{default_publisher}/models"
           end
 
           def publisher_model_path(model)
@@ -651,7 +653,7 @@ module Legion
             id = model_id(model)
             return publisher_from_resource(id) if id.start_with?('projects/')
 
-            PUBLISHERS.fetch(id, DEFAULT_PUBLISHER)
+            PUBLISHERS.fetch(id, default_publisher)
           end
 
           def publisher_from_resource(resource)
@@ -662,7 +664,7 @@ module Legion
           def api_for(model)
             id = model_id(model)
             return API_MODES[id] if API_MODES.key?(id)
-            return :raw_predict if publisher_for(id) != DEFAULT_PUBLISHER && !Capabilities.embeddings?(id)
+            return :raw_predict if publisher_for(id) != default_publisher && !Capabilities.embeddings?(id)
 
             :generate_content
           end

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

@@ -4,7 +4,7 @@ module Legion
   module Extensions
     module Llm
       module Vertex
-        VERSION = '0.2.8'
+        VERSION = '0.2.9'
       end
     end
   end

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

@@ -20,7 +20,10 @@ module Legion
             family: PROVIDER_FAMILY,
             instance: {
               endpoint: nil,
-              tier: :frontier,
+              project: nil,
+              location: 'us-central1',
+              publisher: 'google',
+              tier: :cloud,
               transport: :http,
               credentials: {
                 access_token: nil,
@@ -28,7 +31,7 @@ module Legion
               },
               provider: {
                 project: nil,
-                location: Provider::DEFAULT_LOCATION,
+                location: 'us-central1',
                 model_aliases: {}
               },
               usage: { inference: true, embedding: true, image: false },

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-llm-vertex
 version: !ruby/object:Gem::Version
-  version: 0.2.8
+  version: 0.2.9
 platform: ruby
 authors:
 - LegionIO