omniai-google 1.8.3 → 1.8.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '05953e1154c7f24ee6d502ea7cb45661c64c0633ccf42ca507a3574374912a71'
-  data.tar.gz: 6e01a56c6c1f41bf2129a9aefe053aae33bb3ea6d63beb6b05fb5922ee7b9baf
+  metadata.gz: 77d2e6565a552f5ef5cba00a7eeea116b84c14957418e8b2ffda0e1ca30ba071
+  data.tar.gz: 8a09e23914a82a3716d047a7c95871e4a7483dac53d066a8d93a4dd7c60ca84e
 SHA512:
-  metadata.gz: 3c1cf6d1e36143ad8cfa154b03cace02a4f02e7514778512b3310133a9716ca3291dd7c20b924c475a3006786bfb1bf4830f7d455787c10b18a99ffc66def9e8
-  data.tar.gz: 196f1458b212e5809fdb10c6b2a5a1c7ca7dfa443ef9eed384ef148afad43d56591486e1cb838f35455cd00ecbc0f847a0f0b8ce328848bbb0d5fc8720700d0b
+  metadata.gz: aafa974ceeb3a5a46fc22d4ad8c1350742c86f6468f16e73a0bc689111bd8e4ae9463fb8682d19dd529fcadcce61eeabf8510e5d43231e7df315cbbea774b99f
+  data.tar.gz: bc6d8932b68fd2098f5773fdd8e693dfd9d479c2d2b944a823c6df8cf85c6a38c3f9e0661a8a898164af00572b728063aef03c07a6d2a5a4f64f399de75afccb

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.8.4'
   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.8.4
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-10-22 00:00:00.000000000 Z
+date: 2024-10-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser