omniai-google 2.3.4 → 2.6.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ce7a07b793b56721b69b9094b640ffa834cf89e4c1eeae7845b3d86d9d41fbac
-  data.tar.gz: 1915096495222231feffd39887ffe253b54348d9339763bc54f6f54beebdbb9a
+  metadata.gz: c901bbfae98e15ee5741f97f8a28189814495b3f2b15a40cba9ffe9e4d5e62d4
+  data.tar.gz: 3c5d7e4544a778bc05dda49e2b8d3e6e7675307d0f0985ff55676e9fbcc6ca98
 SHA512:
-  metadata.gz: e39f1dcd34cb6b7a19d69d3d940e5a8e3737917bfcd80e833f2753b2d6ebc6cbd5b1431199c739beb931937acfee1139ef8e13140e3e0520cd49b72ed703ae22
-  data.tar.gz: 4e87328e756c6c7c7fc61e0bcb52ec466ea550ff5fdde3e93c467f2c5a9fc316227e6fc9d06ea7e91e0ba921221f586690831468b2b83359156b7c9b71f69559
+  metadata.gz: bf21ab638a26791419a04139405064bb974120b0be0a45e77981da82d204bb494c1436a270538a31540a7cc7d4c1b351aa594d773de0851ab4d79797cd767209
+  data.tar.gz: c267e9b001991241574a7fe5895c4e3e56336191d647627c0be202db5b84d3ae568659640edde6728ac58d71067cb5d36f1aa280dfda6ebcab36c788fd321975

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

@@ -10,7 +10,7 @@ module OmniAI
           {
             name: tool.name,
             description: tool.description,
-            parameters: tool.parameters.is_a?(Tool::Parameters) ? tool.parameters.serialize : tool.parameters,
+            parameters: tool.parameters.is_a?(OmniAI::Schema::Object) ? tool.parameters.serialize : tool.parameters,
           }
         end
       end

data/lib/omniai/google/chat.rb

@@ -23,9 +23,13 @@ module OmniAI
         GEMINI_FLASH = GEMINI_2_5_FLASH
       end
 
-      DEFAULT_MODEL = Model::GEMINI_PRO
+      DEFAULT_MODEL = Model::GEMINI_2_5_FLASH
 
-      JSON_MIME_TYPE = "application/json"
+      module ResponseMimeType
+        JSON = "application/json"
+        TEXT = "text/plain"
+        SCHEMA = JSON
+      end
 
       # @return [Context]
       CONTEXT = Context.build do |context|
@@ -100,14 +104,22 @@ module OmniAI
 
       # @return [Hash]
       def generation_config
-        response_mime_type = (JSON_MIME_TYPE if @format.eql?(:json))
-
-        return unless @temperature || response_mime_type
-
-        {
-          temperature: @temperature,
-          responseMimeType: response_mime_type,
-        }.compact
+        data =
+          case @format
+          when :json then { responseMimeType: ResponseMimeType::JSON }
+          when :text then { responseMimeType: ResponseMimeType::TEXT }
+          when OmniAI::Schema::Format
+            {
+              responseMimeType: ResponseMimeType::SCHEMA,
+              responseSchema: @format.schema.serialize,
+            }
+          else {}
+          end
+
+        data[:temperature] = @temperature if @temperature
+
+        data = data.compact
+        data unless data.empty?
       end
 
       # @return [String]

data/lib/omniai/google/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Google
-    VERSION = "2.3.4"
+    VERSION = "2.6.1"
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: omniai-google
 version: !ruby/object:Gem::Version
-  version: 2.3.4
+  version: 2.6.1
 platform: ruby
 authors:
 - Kevin Sylvestre
 bindir: exe
 cert_chain: []
-date: 2025-05-05 00:00:00.000000000 Z
+date: 2025-05-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -43,14 +43,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.3'
+        version: '2.6'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.3'
+        version: '2.6'
 - !ruby/object:Gem::Dependency
   name: zeitwerk
   requirement: !ruby/object:Gem::Requirement