omniai-google 1.8.3 → 1.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '05953e1154c7f24ee6d502ea7cb45661c64c0633ccf42ca507a3574374912a71'
-  data.tar.gz: 6e01a56c6c1f41bf2129a9aefe053aae33bb3ea6d63beb6b05fb5922ee7b9baf
+  metadata.gz: 18c29d5c7580f38ae07eb596186fea540ccd1e179dccd906b3fb1473c187546c
+  data.tar.gz: 187dc0cffd36841469b5fd998216f3f3fa2a56b1cce7bd7d51c9ececa12784e9
 SHA512:
-  metadata.gz: 3c1cf6d1e36143ad8cfa154b03cace02a4f02e7514778512b3310133a9716ca3291dd7c20b924c475a3006786bfb1bf4830f7d455787c10b18a99ffc66def9e8
-  data.tar.gz: 196f1458b212e5809fdb10c6b2a5a1c7ca7dfa443ef9eed384ef148afad43d56591486e1cb838f35455cd00ecbc0f847a0f0b8ce328848bbb0d5fc8720700d0b
+  metadata.gz: 0ddd16af090b9692ff68f7d816e7d153759dd65793d7b816ae40e5ce195c0341f4c6959df4a1bfc09a9452df07bd6fe9633ccc7b46a89e2513c3a417c9f12ad0
+  data.tar.gz: 48c1d3a78de86387e9a95716ece83d27cb4c128bacc43285ed3830d210f2769516bfd69e9212f66caaf0e38dfe77ddcabd67d0005b4d21eeed41094de60e1590

data/lib/omniai/google/chat.rb

@@ -25,6 +25,8 @@ module OmniAI
 
       DEFAULT_MODEL = Model::GEMINI_PRO
 
+      JSON_MIME_TYPE = 'application/json'
+
       # @return [Context]
       CONTEXT = Context.build do |context|
         context.serializers[:text] = TextSerializer.method(:serialize)
@@ -98,9 +100,20 @@ module OmniAI
 
       # @return [Hash]
       def generation_config
-        return unless @temperature
+        response_mime_type = (JSON_MIME_TYPE if json_mime_type?)
+
+        return unless @temperature || response_mime_type
+
+        {
+          temperature: @temperature,
+          responseMimeType: response_mime_type,
+        }.compact
+      end
 
-        { temperature: @temperature }.compact
+      # Checks if setting a jsonMimeType is supported
+      # @return [Boolean]
+      def json_mime_type?
+        @client.version == OmniAI::Google::Config::Version::BETA && @format.eql?(:json)
       end
 
       # @return [String]

data/lib/omniai/google/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Google
-    VERSION = '1.8.3'
+    VERSION = '1.9.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai-google
 version: !ruby/object:Gem::Version
-  version: 1.8.3
+  version: 1.9.0
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-10-22 00:00:00.000000000 Z
+date: 2024-10-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -92,7 +92,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.3.0
+      version: 3.2.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="