omniai-google 3.9.0 → 3.10.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5683bc29149bb31082f782ed460483773e2a88537290909d04adc3fb5285aad4
-  data.tar.gz: dd2cdde8860b212f25a4c761e1e48eccb31aeb0cbaa58dc963d48386f2352c6c
+  metadata.gz: 2b8dca01238f0fabfede35ce895cb4ec7ae936d9f3296c7accd88aa0284a6bfc
+  data.tar.gz: aa5f4c994eaac71ce5a39e6c3c0e0765eee2da33aacf2753d82b994b8c19b328
 SHA512:
-  metadata.gz: 1880ce5da42484fa6f28ba61d14f4ee1d11cb4f3bf2d2b7d772b663af9222743775618dfc1f02de7ac1f50ac952ca742374b039722930ae28c6b388d67c0eff8
-  data.tar.gz: 6e8d16e2b1686ae61810ece5e5fb6c07fa69a4b0dcd25063261edc54e8765e228e3e40bd9fdde8b56fffc7b5c467f1a4e9c83708bc48b2bcd47ecfb144b52f31
+  metadata.gz: c5b8f8335542020564f81a03bb69236f3db4c2b0683353e84ca8866d69fc557b52ae353deebd272c480fe3a9f54b190a4c451550c8a1f93f5d4e2638731b48d6
+  data.tar.gz: f5a2f43feda13a60a4fcdf7ae9e4826ffae58f527f550d258a3ba8e520421ff1b601e2f59b6affcb6560a2e59aa58452be1ce74fcbdf48c433736d2e98fdc212

data/lib/omniai/google/chat/choice_serializer.rb

@@ -5,6 +5,20 @@ module OmniAI
     class Chat
       # Overrides choice serialize / deserialize.
       module ChoiceSerializer
+        # Maps Gemini candidate `finishReason` values onto the normalized OmniAI::Chat::FinishReason symbols. The
+        # entire content-policy / safety family maps to `:filter`; unrecognized values (e.g. OTHER,
+        # MALFORMED_FUNCTION_CALL, FINISH_REASON_UNSPECIFIED) fall through to `:other`.
+        FINISH_REASONS = {
+          "STOP" => OmniAI::Chat::FinishReason::STOP,
+          "MAX_TOKENS" => OmniAI::Chat::FinishReason::LENGTH,
+          "SAFETY" => OmniAI::Chat::FinishReason::FILTER,
+          "RECITATION" => OmniAI::Chat::FinishReason::FILTER,
+          "LANGUAGE" => OmniAI::Chat::FinishReason::FILTER,
+          "BLOCKLIST" => OmniAI::Chat::FinishReason::FILTER,
+          "PROHIBITED_CONTENT" => OmniAI::Chat::FinishReason::FILTER,
+          "SPII" => OmniAI::Chat::FinishReason::FILTER,
+          "IMAGE_SAFETY" => OmniAI::Chat::FinishReason::FILTER,
+        }.freeze
         # @param choice [OmniAI::Chat::Choice]
         # @param context [Context]
         # @return [Hash]
@@ -18,7 +32,8 @@ module OmniAI
         # @return [OmniAI::Chat::Choice]
         def self.deserialize(data, context:)
           message = OmniAI::Chat::Message.deserialize(data["content"], context:)
-          OmniAI::Chat::Choice.new(message:)
+          finish_reason = OmniAI::Chat::FinishReason.deserialize(data["finishReason"], table: FINISH_REASONS)
+          OmniAI::Chat::Choice.new(message:, finish_reason:)
         end
       end
     end

data/lib/omniai/google/chat/stream.rb

@@ -55,9 +55,7 @@ module OmniAI
         # @param candidate [Hash]
         # @param index [Integer]
         def process_candidate!(candidate:, index:, &block)
-          return unless candidate["content"]
-
-          candidate["content"]["parts"]&.each do |part|
+          candidate.dig("content", "parts")&.each do |part|
             if part["thought"]
               block&.call(OmniAI::Chat::Delta.new(thinking: part["text"]))
             elsif part["text"]
@@ -73,16 +71,29 @@ module OmniAI
         def merge_candidate!(candidate:, index:)
           if @data["candidates"][index].nil?
             @data["candidates"][index] = candidate
-          else
-            (candidate["content"]["parts"] || []).each do |part|
-              merge_part!(part:, candidate: @data["candidates"][index])
-            end
+            return
+          end
+
+          existing = @data["candidates"][index]
+
+          candidate.dig("content", "parts")&.each do |part|
+            merge_part!(part:, candidate: existing)
+          end
+
+          # Preserve top-level candidate keys (most importantly `finishReason`) that arrive on a later — usually
+          # terminal — chunk after the content has already streamed. Without this, the reason a generation stopped
+          # (e.g. MAX_TOKENS / SAFETY) is silently dropped from the assembled response.
+          candidate.each do |key, value|
+            next if key.eql?("content")
+
+            existing[key] = value
           end
         end
 
         # @param part [Hash]
         # @param candidate [Hash]
         def merge_part!(part:, candidate:)
+          candidate["content"] ||= {}
           parts = candidate["content"]["parts"] ||= []
           last_part = parts.last
 

data/lib/omniai/google/transcribe.rb

@@ -113,7 +113,7 @@ module OmniAI
         # Speech-to-Text API uses different endpoints for regional vs global
         endpoint = speech_endpoint
         speech_connection = HTTP.persistent(endpoint)
-          .timeout(**http_timeout_options)
+          .timeout(http_timeout_options)
           .accept(:json)
 
         # Add authentication if using credentials

data/lib/omniai/google/transcribe_helpers.rb

@@ -45,12 +45,14 @@ module OmniAI
       # Normalizes the client timeout into keyword args for HTTP.rb's `.timeout`. The speech
       # endpoints build their own connections, so (unlike the base client, which passes the
       # value straight through) they must accept both a scalar and a per-operation Hash. A Hash
-      # is passed through untouched; a scalar (or nil) is wrapped per-operation as before.
+      # is passed through untouched; a scalar is wrapped per-operation. When no timeout is set, returns `:null`
+      # (the http "no timeout" sentinel) — http 6 rejects nil per-operation values, http 5 accepts `:null` too.
       #
-      # @return [Hash]
+      # @return [Hash, Symbol]
       def http_timeout_options
         timeout = @client.timeout
         return timeout if timeout.is_a?(Hash)
+        return :null if timeout.nil?
 
         { connect: timeout, write: timeout, read: timeout }
       end
@@ -212,7 +214,7 @@ module OmniAI
       def poll_operation!(operation_name)
         endpoint = speech_endpoint
         connection = HTTP.persistent(endpoint)
-          .timeout(**http_timeout_options)
+          .timeout(http_timeout_options)
           .accept(:json)
 
         # Add authentication if using credentials
@@ -250,7 +252,7 @@ module OmniAI
       def request_batch!
         endpoint = speech_endpoint
         connection = HTTP.persistent(endpoint)
-          .timeout(**http_timeout_options)
+          .timeout(http_timeout_options)
           .accept(:json)
 
         # Add authentication if using credentials

data/lib/omniai/google/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Google
-    VERSION = "3.9.0"
+    VERSION = "3.10.0"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai-google
 version: !ruby/object:Gem::Version
-  version: 3.9.0
+  version: 3.10.0
 platform: ruby
 authors:
 - Kevin Sylvestre
@@ -57,14 +57,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.2'
+        version: '3.7'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.2'
+        version: '3.7'
 - !ruby/object:Gem::Dependency
   name: openssl
   requirement: !ruby/object:Gem::Requirement