omniai-google 3.4.3 → 3.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8ed53a642ea244879948b2831fb9f9dfb301eb7a1cedfff39e26131e6a6bb36e
-  data.tar.gz: abe713af3afe2f678e515ea23fd1cc1cc8b6ab0f328d77936e3a3990d3cd06fa
+  metadata.gz: 82ee690e391c20d752962b27396514b2c8ed7f402e279b44115ca717622fabc3
+  data.tar.gz: b559de433576791ac3676ae2d5829369f1951a4824091a0c01add03f7e4a2b50
 SHA512:
-  metadata.gz: 154befd060662f97e6a3b8fdb9a2ba55879f3226f89a8f4c539572c7064c973599e32892567af9b91c173ef42da3995b0d2a237765892a735c27227b59b4c1ec
-  data.tar.gz: 4d4992ccb4a94e074e2ed185afadebb7dfedb1b44c08ed68485b243876f70b6a6dce4242424538545962bf4992c3176b6d26398a998887b09dd0f5ac45b5cd01
+  metadata.gz: 8ac9197937f962bd597c4a876dfc1b29edcb5802aae8b39e9e2962a118208cf0b17829b3a0dda4849c82091ffb60c12f050643712dc0c0c467cea0880bd8fd0b
+  data.tar.gz: 62600d5597689f129a3412b8718058f7de8c34e9e4e42ad0434b945b52c09cb7f4cd60e4dfe23d688d27026d4694b69e1a16febae76a3aa7e90f9680a2a77cb0

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.rb

@@ -17,11 +17,12 @@ module OmniAI
         GEMINI_1_5_PRO = "gemini-1.5-pro"
         GEMINI_2_5_PRO = "gemini-2.5-pro"
         GEMINI_3_0_PRO = "gemini-3-pro-preview"
+        GEMINI_3_1_PRO = "gemini-3.1-pro-preview"
         GEMINI_1_5_FLASH = "gemini-1.5-flash"
         GEMINI_2_0_FLASH = "gemini-2.0-flash"
         GEMINI_2_5_FLASH = "gemini-2.5-flash"
         GEMINI_3_FLASH = "gemini-3-flash-preview"
-        GEMINI_PRO = GEMINI_3_0_PRO
+        GEMINI_PRO = GEMINI_3_1_PRO
         GEMINI_FLASH = GEMINI_3_FLASH
       end
 
@@ -103,7 +104,7 @@ 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
 
@@ -132,7 +133,7 @@ module OmniAI
           end
 
         data[:temperature] = @temperature if @temperature
-        data[:thinkingConfig] = thinking_config if thinking_config
+        data[:thinkingConfig] = thinking_config if @options[:thinking]
 
         data = data.compact
         data unless data.empty?
@@ -148,17 +149,17 @@ module OmniAI
         stream? ? "streamGenerateContent" : "generateContent"
       end
 
-      # Translates unified thinking option to Google's thinkingConfig format.
-      # Example: `thinking: true` becomes `{ includeThoughts: true }`
       # @return [Hash, nil]
       def thinking_config
         thinking = @options[:thinking]
         return unless thinking
+        return { includeThoughts: true } unless thinking.is_a?(Hash)
 
-        case thinking
-        when true then { includeThoughts: true }
-        when Hash then { includeThoughts: true }.merge(thinking)
-        end
+        config = { includeThoughts: true }
+        return config.merge(thinking) unless thinking.key?(:effort)
+
+        config[:thinkingLevel] = thinking[:effort].upcase if thinking[:effort]
+        config
       end
 
       # @return [Array<Message>]

data/lib/omniai/google/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Google
-    VERSION = "3.4.3"
+    VERSION = "3.5.1"
   end
 end

metadata

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