omniai-google 1.8.2 → 1.8.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a94a269754cf5575f61c30d06d0e37eae9e81a5f39b8d5c57d6bb20466b0c4b7
-  data.tar.gz: 1cc25309962eaa19124d1f524283168b45268f518c5cbba823f288dc10ae6db9
+  metadata.gz: 77d2e6565a552f5ef5cba00a7eeea116b84c14957418e8b2ffda0e1ca30ba071
+  data.tar.gz: 8a09e23914a82a3716d047a7c95871e4a7483dac53d066a8d93a4dd7c60ca84e
 SHA512:
-  metadata.gz: ffadc476993045df0e1970c1733ee127f253ba043098b834275365de5c13dbeb6f6f1a8aef5aa93b903b87d514e30a52ad6059b50a7e11edd4774e9ca27049c1
-  data.tar.gz: cc5613811897cd7e3f2171c8ea8cfc2b841d32247adfda2500543bb39e38362569eac6a1dc9febbc258a0ee62239ac545a7ea718646fa0b06f2d53f775070619
+  metadata.gz: aafa974ceeb3a5a46fc22d4ad8c1350742c86f6468f16e73a0bc689111bd8e4ae9463fb8682d19dd529fcadcce61eeabf8510e5d43231e7df315cbbea774b99f
+  data.tar.gz: bc6d8932b68fd2098f5773fdd8e693dfd9d479c2d2b944a823c6df8cf85c6a38c3f9e0661a8a898164af00572b728063aef03c07a6d2a5a4f64f399de75afccb

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

@@ -14,7 +14,7 @@ module OmniAI
 
           {
             candidates:,
-            usage_metadata:,
+            usageMetadata: usage_metadata,
           }
         end
 
@@ -23,7 +23,7 @@ module OmniAI
         # @return [OmniAI::Chat::Payload]
         def self.deserialize(data, context:)
           choices = data['candidates'].map { |candidate| OmniAI::Chat::Choice.deserialize(candidate, context:) }
-          usage = OmniAI::Chat::Usage.deserialize(data['usage_metadata'], context:) if data['usage_metadata']
+          usage = OmniAI::Chat::Usage.deserialize(data['usageMetadata'], context:) if data['usageMetadata']
 
           OmniAI::Chat::Payload.new(choices:, usage:)
         end

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

@@ -9,18 +9,18 @@ module OmniAI
         # @return [Hash]
         def self.serialize(usage, *)
           {
-            prompt_token_count: usage.input_tokens,
-            candidates_token_count: usage.output_tokens,
-            total_token_count: usage.total_tokens,
+            promptTokenCount: usage.input_tokens,
+            candidatesTokenCount: usage.output_tokens,
+            totalTokenCount: usage.total_tokens,
           }
         end
 
         # @param data [Hash]
         # @return [OmniAI::Chat::Usage]
         def self.deserialize(data, *)
-          input_tokens = data['prompt_token_count']
-          output_tokens = data['candidates_token_count']
-          total_tokens = data['total_token_count']
+          input_tokens = data['promptTokenCount']
+          output_tokens = data['candidatesTokenCount']
+          total_tokens = data['totalTokenCount']
           OmniAI::Chat::Usage.new(input_tokens:, output_tokens:, total_tokens:)
         end
       end

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