omniai-google 3.5.0 → 3.5.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 33134e75816a15602fef0394b0be7527930e7bb8f3b8469e5b0acbe7f93edf23
-  data.tar.gz: 1ce3168ea82a7a943e1c6537839b2cec9286bd8684e80e873aca4133ed326c49
+  metadata.gz: 75cf7d6dde07411b0b22958ecc3ea72600956f7bd3a7d68c0bb034b0230e2b5f
+  data.tar.gz: 32d419138aa379822c21fac7fc94402c3423ff9b7cc127bb382d1be24f3c18a1
 SHA512:
-  metadata.gz: e6f3f3f81fa30d51fc892c09f5d13ff00c438b0f064549447a08a25cc814d52650e32ce2140c74b7fdbbdac80b94d8d210ac74978fe4e931837663fa43ae85cd
-  data.tar.gz: b2a5c9217496e893385f617c6250448ad8edfddb4aa6274deeab3e516044aa30e82ece923d2d64eaf4720218195cddfe7f6f07778985d8073063212c52f762e2
+  metadata.gz: 61b0c47b210b5610c331d31af784c249d854e40a8f1246dd39cf68d7b6be032d224cc52455b63917cad748ffb6b575ba0dc5b3e92b9b9a254d53363a5f6eb4ea
+  data.tar.gz: 505c2c82358dfcbcf83b10a90ebb87b1d76c00e46e7318708c465c8c1eb68e4f8ba0a5bb71c0767c694701e98669d3c9fe6e5fec2ce747a6aea59ee0e35fa7c7

data/README.md

@@ -87,7 +87,7 @@ completion.text # 'The capital of Canada is Ottawa.'
 
 #### Model
 
-`model` takes an optional string (default is `gemini-1.5-pro`):
+`model` takes an optional string (default is `gemini-3-flash-preview`):
 
 ```ruby
 completion = client.chat('How fast is a cheetah?', model: OmniAI::Google::Chat::Model::GEMINI_FLASH)
@@ -120,30 +120,45 @@ client.chat('Be poetic.', stream:)
 
 #### Extended Thinking
 
-Google Gemini 2.0+ models support extended thinking, which shows the model's reasoning process.
+Gemini models support extended thinking, which shows the model's reasoning process.
 
 ```ruby
-# Enable thinking
-response = client.chat("What is 25 * 25?", model: "gemini-2.5-pro-preview-05-06", thinking: true)
+# Enable thinking with model defaults
+response = client.chat("What is 25 * 25?", model: OmniAI::Google::Chat::Model::GEMINI_PRO, thinking: true)
 ```
 
-#### Accessing Thinking Content
+##### Thinking Effort
+
+Control how much the model thinks using the `effort` option (`"low"`, `"medium"`, or `"high"`):
+
+```ruby
+# High effort (more reasoning tokens)
+response = client.chat("Solve this step by step.", model: OmniAI::Google::Chat::Model::GEMINI_PRO, thinking: { effort: "high" })
+
+# Medium effort
+response = client.chat("Explain briefly.", model: OmniAI::Google::Chat::Model::GEMINI_PRO, thinking: { effort: "medium" })
+
+# Low effort (fewer reasoning tokens)
+response = client.chat("What is 1+1?", model: OmniAI::Google::Chat::Model::GEMINI_PRO, thinking: { effort: "low" })
+```
+
+##### Accessing Thinking Content
 
 ```ruby
-response.choices.first.message.contents.each do |content|
-  case content
+response.choices.first.message.content.each do |part|
+  case part
   when OmniAI::Chat::Thinking
-    puts "Thinking: #{content.thinking}"
+    puts "Thinking: #{part.thinking}"
   when OmniAI::Chat::Text
-    puts "Response: #{content.text}"
+    puts "Response: #{part.text}"
   end
 end
 ```
 
-#### Streaming with Thinking
+##### Streaming with Thinking
 
 ```ruby
-client.chat("What are the prime factors of 1234567?", model: "gemini-2.5-pro-preview-05-06", thinking: true, stream: $stdout)
+client.chat("What are the prime factors of 1234567?", model: OmniAI::Google::Chat::Model::GEMINI_PRO, thinking: true, stream: $stdout)
 ```
 
 [Google API Reference `thinking`](https://ai.google.dev/gemini-api/docs/thinking)

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

@@ -34,8 +34,7 @@ module OmniAI
             end
           end
 
-          tool_call_list = parts.select { |part| part.is_a?(OmniAI::Chat::ToolCall) }
-          content = parts.reject { |part| part.is_a?(OmniAI::Chat::ToolCall) }
+          tool_call_list, content = parts.partition { |part| part.is_a?(OmniAI::Chat::ToolCall) }
 
           OmniAI::Chat::Message.new(content:, role:, tool_call_list:)
         end

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

@@ -44,7 +44,7 @@ module OmniAI
             @data[key] = value unless key.eql?("candidates")
           end
 
-          data["candidates"].each_with_index do |candidate, index|
+          data["candidates"]&.each_with_index do |candidate, index|
             process_candidate!(candidate:, index:, &block)
           end
         end
@@ -53,12 +53,12 @@ module OmniAI
         # @yieldparam delta [OmniAI::Chat::Delta]
         #
         # @param candidate [Hash]
+        # @param index [Integer]
         def process_candidate!(candidate:, index:, &block)
           return unless candidate["content"]
 
-          candidate["content"]["parts"].each do |part|
+          candidate["content"]["parts"]&.each do |part|
             if part["thought"]
-              # Google uses thought: true as a flag, content is in text
               block&.call(OmniAI::Chat::Delta.new(thinking: part["text"]))
             elsif part["text"]
               block&.call(OmniAI::Chat::Delta.new(text: part["text"]))
@@ -74,27 +74,23 @@ module OmniAI
           if @data["candidates"][index].nil?
             @data["candidates"][index] = candidate
           else
-            merge_parts!(parts: candidate["content"]["parts"], candidate: @data["candidates"][index])
+            (candidate["content"]["parts"] || []).each do |part|
+              merge_part!(part:, candidate: @data["candidates"][index])
+            end
           end
         end
 
-        # @param parts [Array<Hash>]
-        # @param candidate [Hash]
-        def merge_parts!(parts:, candidate:)
-          parts.each { |part| merge_part!(part:, candidate:) }
-        end
-
         # @param part [Hash]
-        # @param into [Hash]
+        # @param candidate [Hash]
         def merge_part!(part:, candidate:)
-          last_part = candidate["content"]["parts"][-1]
+          parts = candidate["content"]["parts"] ||= []
+          last_part = parts.last
 
-          if last_part&.key?("text") && part["text"]
+          if (last_part&.key?("text") && part.key?("text")) ||
+              (last_part&.key?("thought") && part.key?("thought"))
             last_part["text"] += part["text"]
-          elsif last_part&.key?("thought") && part["thought"]
-            last_part["thought"] += part["thought"]
           else
-            candidate["content"]["parts"] << part
+            parts << part
           end
         end
       end

data/lib/omniai/google/chat.rb

@@ -104,13 +104,13 @@ module OmniAI
         return unless tools?
 
         [
-          function_declarations: custom_tools.map { |tool| tool.serialize(context:) },
+          { function_declarations: custom_tools.map { |tool| tool.serialize(context:) } },
         ].concat(internal_tools.map { |name| { name => {} } })
       end
 
       # @return [Array<OmniAI::Tool>]
       def custom_tools
-        @tools.filter { |tool| tool.is_a?(OmniAI::Tool) }
+        @tools.grep(OmniAI::Tool)
       end
 
       # @return [Array<Symbol, String>]

data/lib/omniai/google/credentials.rb

@@ -23,7 +23,7 @@ module OmniAI
         case value
         when IO, StringIO then ::Google::Auth::ServiceAccountCredentials.make_creds(json_key_io: value, scope: SCOPE)
         when Hash then parse(JSON.generate(value))
-        when Pathname then parse(File.open(value))
+        when Pathname then File.open(value) { |file| parse(file) }
         when String then parse(StringIO.new(value))
         else value
         end

data/lib/omniai/google/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Google
-    VERSION = "3.5.0"
+    VERSION = "3.5.2"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai-google
 version: !ruby/object:Gem::Version
-  version: 3.5.0
+  version: 3.5.2
 platform: ruby
 authors:
 - Kevin Sylvestre