omniai-google 2.6.2 → 2.6.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cb64970aa23a098c48810b8d2d299665c383d188d7a908c1d331375235083dc5
-  data.tar.gz: 10161c3268426a21701d82ecf7a87f24b4afc901f17e283eeeb999ad717d96c0
+  metadata.gz: 9b4a06d7346381cd281088f6e95cb85cf0142f286ff16a26e6681447ba5a0401
+  data.tar.gz: 5ada4e7eb1e51bbbb01e1981917ccc3331500a847e45586b4a3618b51d26992a
 SHA512:
-  metadata.gz: a261a66b7a8ddfb723a2327dda567a8494bba59420626a6f57dbd4f76838b9f603ef5c619b08af245ecb2a72865d7a9bf905c320d6bd1d4df943bf60141fc649
-  data.tar.gz: 9ff5956e21aeefa6fe4829dfe5f629a94ddb35b49e24fad1e5e5f48b065c58c393265aeb7b1bbfa2a4b9136dcfb74798f904bc960c05f14870b69987a1d4160d
+  metadata.gz: 9d929c63f5008bc8c23261d2cdde88f56aca8b17ce1524443915b4181d590fa6b6778b02719ed03f6342eec94c406e64c5b76b411900884e3dae5349a0eb1edf
+  data.tar.gz: 1f68746ee88d7c17804a125b4d48c49b0309438408e555172889e28b7e408b9823bd63226d091d2c5159046a973a8d8e932f1ae4dcc64a3a62ebe68c979c2834

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

@@ -53,11 +53,7 @@ module OmniAI
         # @yieldparam delta [OmniAI::Chat::Delta]
         #
         # @param candidate [Hash]
-        # @param index [Integer]
         def process_candidate!(candidate:, index:, &block)
-          parts = candidate["content"]["parts"]
-          return unless parts
-
           candidate["content"]["parts"].each do |part|
             block&.call(OmniAI::Chat::Delta.new(text: part["text"])) if part["text"]
           end
@@ -71,19 +67,23 @@ module OmniAI
           if @data["candidates"][index].nil?
             @data["candidates"][index] = candidate
           else
-            merge_parts!(content: @data["candidates"][index]["content"], parts: candidate["content"]["parts"])
+            merge_parts!(parts: candidate["content"]["parts"], candidate: @data["candidates"][index])
           end
         end
 
-        # @param content [Hash]
         # @param parts [Array<Hash>]
-        def merge_parts!(content:, parts:)
-          parts.each_with_index do |part, index|
-            if content["parts"][index].nil?
-              content["parts"][index] = part
-            else
-              content["parts"][index]["text"] += part["text"]
-            end
+        # @param candidate [Hash]
+        def merge_parts!(parts:, candidate:)
+          parts.each { |part| merge_part!(part:, candidate:) }
+        end
+
+        # @param part [Hash]
+        # @param into [Hash]
+        def merge_part!(part:, candidate:)
+          if candidate["content"]["parts"][-1]&.key?("text") && part["text"]
+            candidate["content"]["parts"][-1]["text"] += part["text"]
+          else
+            candidate["content"]["parts"] << part
           end
         end
       end

data/lib/omniai/google/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Google
-    VERSION = "2.6.2"
+    VERSION = "2.6.4"
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: omniai-google
 version: !ruby/object:Gem::Version
-  version: 2.6.2
+  version: 2.6.4
 platform: ruby
 authors:
 - Kevin Sylvestre
 bindir: exe
 cert_chain: []
-date: 2025-05-12 00:00:00.000000000 Z
+date: 2025-05-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser