omniai-google 1.9.3 → 1.9.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 80e08395bc3710a72cc59b7c110cc50b1ba4a1dc64e15c622fcacb39448c1265
-  data.tar.gz: cccb64b57c0430caf449ebb1de2620e597409de0179379ef0cb26588d41418a2
+  metadata.gz: 94943a4eb224e5224148c1109d3a8472d26f40d83684c102cee11f6f46d5ae00
+  data.tar.gz: c234428c542e1b507dcf86a80e624eec640173db0604c7e2f10264de569532b4
 SHA512:
-  metadata.gz: cf8a4e78c06a9808930300c1aa05b45211ab0f733a79bddc22bd5ac3ca2bfff1a00d075ca596122295c82aeb909372d2a9c7a66bc27566386d529428209beda7
-  data.tar.gz: 7324a4d99c3ce3a890129eb5a3091459facbd0acf5818b7ff0624d168d2fee94613656db1fe1c1091f8f001c6c34cc34c66a835b69fb7d90191075bde455de5b
+  metadata.gz: 55eec5fa84a4f4662d2623bd79443bf41df07b84961349f2c8fd974bbfb441ec3b3bdfbc8da9738156215da54bd74f3e2bb636e4878180ef523f379b76e474d3
+  data.tar.gz: 1bc60cd887bc6dfd531556291f06b192622586a59872818491d39ea99464f290f4d231d2935a64fd35b0a75736becfc1ac5c3d52f75b127fd6916fe13f572b23

data/Gemfile

@@ -1,16 +1,18 @@
 # frozen_string_literal: true
 
-source 'https://rubygems.org'
+source "https://rubygems.org"
 
 gemspec
 
-gem 'logger'
-gem 'rake'
-gem 'rspec'
-gem 'rspec_junit_formatter'
-gem 'rubocop'
-gem 'rubocop-rake'
-gem 'rubocop-rspec'
-gem 'simplecov'
-gem 'webmock'
-gem 'yard'
+gem "irb"
+gem "logger"
+gem "rake"
+gem "rspec"
+gem "rspec_junit_formatter"
+gem "rubocop"
+gem "rubocop-basic"
+gem "rubocop-rake"
+gem "rubocop-rspec"
+gem "simplecov"
+gem "webmock"
+gem "yard"

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

@@ -17,7 +17,7 @@ module OmniAI
         # @param context [Context]
         # @return [OmniAI::Chat::Choice]
         def self.deserialize(data, context:)
-          message = OmniAI::Chat::Message.deserialize(data['content'], context:)
+          message = OmniAI::Chat::Message.deserialize(data["content"], context:)
           OmniAI::Chat::Choice.new(message:)
         end
       end

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

@@ -10,8 +10,8 @@ module OmniAI
         # @return [OmniAI::Chat::Text, OmniAI::Chat::ToolCall]
         def self.deserialize(data, context:)
           case
-          when data['text'] then data['text']
-          when data['functionCall'] then OmniAI::Chat::ToolCall.deserialize(data, context:)
+          when data["text"] then data["text"]
+          when data["functionCall"] then OmniAI::Chat::ToolCall.deserialize(data, context:)
           end
         end
       end

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

@@ -17,8 +17,8 @@ module OmniAI
         # @param data [Hash]
         # @return [OmniAI::Chat::Function]
         def self.deserialize(data, *)
-          name = data['name']
-          arguments = data['args']
+          name = data["name"]
+          arguments = data["args"]
           OmniAI::Chat::Function.new(name:, arguments:)
         end
       end

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

@@ -44,7 +44,7 @@ module OmniAI
         #
         # @return [Boolean]
         def self.use_file_data?(uri)
-          uri.host.eql?('generativelanguage.googleapis.com') || uri.scheme.eql?('gs')
+          uri.host.eql?("generativelanguage.googleapis.com") || uri.scheme.eql?("gs")
         end
       end
     end

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

@@ -24,12 +24,12 @@ module OmniAI
         # @param context [OmniAI::Context]
         # @return [OmniAI::Chat::Message]
         def self.deserialize(data, context:)
-          role = data['role']
-          parts = arrayify(data['parts']).map do |part|
+          role = data["role"]
+          parts = arrayify(data["parts"]).map do |part|
             case
-            when part['text'] then OmniAI::Chat::Text.deserialize(part, context:)
-            when part['functionCall'] then OmniAI::Chat::ToolCall.deserialize(part, context:)
-            when part['functionResponse'] then OmniAI::Chat::ToolCallResult.deserialize(part, context:)
+            when part["text"] then OmniAI::Chat::Text.deserialize(part, context:)
+            when part["functionCall"] then OmniAI::Chat::ToolCall.deserialize(part, context:)
+            when part["functionResponse"] then OmniAI::Chat::ToolCallResult.deserialize(part, context:)
             end
           end
 

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

@@ -22,8 +22,8 @@ module OmniAI
         # @param context [OmniAI::Context]
         # @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['usageMetadata'], context:) if data['usageMetadata']
+          choices = data["candidates"].map { |candidate| OmniAI::Chat::Choice.deserialize(candidate, context:) }
+          usage = OmniAI::Chat::Usage.deserialize(data["usageMetadata"], context:) if data["usageMetadata"]
 
           OmniAI::Chat::Payload.new(choices:, usage:)
         end

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

@@ -14,7 +14,7 @@ module OmniAI
         # @param data [Hash]
         # @return [OmniAI::Chat::Text]
         def self.deserialize(data, *)
-          OmniAI::Chat::Text.new(data['text'])
+          OmniAI::Chat::Text.new(data["text"])
         end
       end
     end

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

@@ -22,8 +22,8 @@ module OmniAI
         # @param data [Hash]
         # @return [ToolCallResult]
         def self.deserialize(data, *)
-          tool_call_id = data['functionResponse']['name']
-          content = data['functionResponse']['response']['content']
+          tool_call_id = data["functionResponse"]["name"]
+          content = data["functionResponse"]["response"]["content"]
           OmniAI::Chat::ToolCallResult.new(content:, tool_call_id:)
         end
       end

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

@@ -16,7 +16,7 @@ module OmniAI
         # @param context [OmniAI::Context]
         # @return [OmniAI::Chat::ToolCall]
         def self.deserialize(data, context:)
-          function = OmniAI::Chat::Function.deserialize(data['functionCall'], context:)
+          function = OmniAI::Chat::Function.deserialize(data["functionCall"], context:)
           OmniAI::Chat::ToolCall.new(id: function.name, function:)
         end
       end

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

@@ -18,9 +18,9 @@ module OmniAI
         # @param data [Hash]
         # @return [OmniAI::Chat::Usage]
         def self.deserialize(data, *)
-          input_tokens = data['promptTokenCount']
-          output_tokens = data['candidatesTokenCount']
-          total_tokens = data['totalTokenCount']
+          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

@@ -13,19 +13,17 @@ module OmniAI
     #   chat.completion([{ role: 'system', content: 'Tell me a joke.' }])
     class Chat < OmniAI::Chat
       module Model
-        GEMINI_1_0_PRO = 'gemini-1.0-pro'
-        GEMINI_1_5_PRO = 'gemini-1.5-pro'
-        GEMINI_1_5_FLASH = 'gemini-1.5-flash'
-        GEMINI_1_0_PRO_LATEST = 'gemini-1.0-pro-latest'
-        GEMINI_1_5_PRO_LATEST = 'gemini-1.5-pro-latest'
-        GEMINI_1_5_FLASH_LATEST = 'gemini-1.5-flash-latest'
+        GEMINI_1_0_PRO = "gemini-1.0-pro"
+        GEMINI_1_5_PRO = "gemini-1.5-pro"
+        GEMINI_1_5_FLASH = "gemini-1.5-flash"
+        GEMINI_2_0_FLASH = "gemini-2.0-flash"
         GEMINI_PRO = GEMINI_1_5_PRO
-        GEMINI_FLASH = GEMINI_1_5_FLASH
+        GEMINI_FLASH = GEMINI_2_0_FLASH
       end
 
       DEFAULT_MODEL = Model::GEMINI_PRO
 
-      JSON_MIME_TYPE = 'application/json'
+      JSON_MIME_TYPE = "application/json"
 
       # @return [Context]
       CONTEXT = Context.build do |context|
@@ -61,7 +59,7 @@ module OmniAI
         context.serializers[:tool] = ToolSerializer.method(:serialize)
       end
 
-      protected
+    protected
 
       # @return [Context]
       def context
@@ -75,7 +73,7 @@ module OmniAI
           .accept(:json)
           .post(path, params: {
             key: @client.api_key,
-            alt: ('sse' if @stream),
+            alt: ("sse" if @stream),
           }.compact, json: payload)
       end
 
@@ -123,7 +121,7 @@ module OmniAI
 
       # @return [String]
       def operation
-        @stream ? 'streamGenerateContent' : 'generateContent'
+        @stream ? "streamGenerateContent" : "generateContent"
       end
 
       # @return [Array<Message>]
@@ -132,7 +130,7 @@ module OmniAI
           ToolCallResult.new(tool_call_id: tool_call.id, content: execute_tool_call(tool_call))
         end
 
-        [Message.new(role: 'function', content:)]
+        [Message.new(role: "function", content:)]
       end
     end
   end

data/lib/omniai/google/config.rb

@@ -5,11 +5,11 @@ module OmniAI
     # Configuration for Google.
     class Config < OmniAI::Config
       module Version
-        STABLE = 'v1'
-        BETA = 'v1beta'
+        STABLE = "v1"
+        BETA = "v1beta"
       end
 
-      DEFAULT_HOST = 'https://generativelanguage.googleapis.com'
+      DEFAULT_HOST = "https://generativelanguage.googleapis.com"
       DEFAULT_VERSION = Version::BETA
 
       # @!attribute [rw] version
@@ -22,9 +22,9 @@ module OmniAI
       # @param logger [Logger, nil] optional - defaults to
       # @param timeout [Integer, Hash, nil] optional
       def initialize(
-        api_key: ENV.fetch('GOOGLE_API_KEY', nil),
-        host: ENV.fetch('GOOGLE_HOST', DEFAULT_HOST),
-        version: ENV.fetch('GOOGLE_VERSION', DEFAULT_VERSION),
+        api_key: ENV.fetch("GOOGLE_API_KEY", nil),
+        host: ENV.fetch("GOOGLE_HOST", DEFAULT_HOST),
+        version: ENV.fetch("GOOGLE_VERSION", DEFAULT_VERSION),
         logger: nil,
         timeout: nil
       )

data/lib/omniai/google/embed.rb

@@ -11,8 +11,8 @@ module OmniAI
     #   response.embedding [0.0, ...]
     class Embed < OmniAI::Embed
       module Model
-        TEXT_EMBEDDING_004 = 'text-embedding-004'
-        TEXT_MULTILINGUAL_EMBEDDING_002 = 'text-multilingual-embedding-002'
+        TEXT_EMBEDDING_004 = "text-embedding-004"
+        TEXT_MULTILINGUAL_EMBEDDING_002 = "text-multilingual-embedding-002"
         EMBEDDING = TEXT_EMBEDDING_004
         MULTILINGUAL_EMBEDDING = TEXT_MULTILINGUAL_EMBEDDING_002
       end
@@ -20,7 +20,7 @@ module OmniAI
       DEFAULT_MODEL = Model::EMBEDDING
 
       EMBEDDINGS_DESERIALIZER = proc do |data, *|
-        data['embeddings'].map { |embedding| embedding['values'] }
+        data["embeddings"].map { |embedding| embedding["values"] }
       end
 
       # @return [Context]
@@ -28,7 +28,7 @@ module OmniAI
         context.deserializers[:embeddings] = EMBEDDINGS_DESERIALIZER
       end
 
-      protected
+    protected
 
       # @param response [HTTP::Response]
       # @return [Response]

data/lib/omniai/google/upload/file.rb

@@ -30,10 +30,10 @@ module OmniAI
         def self.parse(client:, data:)
           new(
             client: client,
-            name: data['name'],
-            uri: data['uri'],
-            state: data['state'],
-            mime_type: data['mimeType']
+            name: data["name"],
+            uri: data["uri"],
+            state: data["state"],
+            mime_type: data["mimeType"]
           )
         end
 

data/lib/omniai/google/upload.rb

@@ -8,14 +8,12 @@ module OmniAI
 
       # @param client [Client]
       # @param io [IO]
-      # @param mime_type [String]
       def self.process!(client:, io:)
         new(client:, io:).process!
       end
 
       # @param client [Client]
-      # @param file [File]
-      # @param mime_type [String]
+      # @param io [File]
       def initialize(client:, io:)
         @client = client
         @io = io
@@ -28,16 +26,16 @@ module OmniAI
         response = io! do |io|
           response = @client
             .connection
-            .headers({ 'X-Goog-Upload-Protocol' => 'raw' })
+            .headers({ "X-Goog-Upload-Protocol" => "raw" })
             .post("/upload/#{@client.version}/files?key=#{@client.api_key}", body: HTTP::FormData::File.new(io))
         end
 
         raise OmniAI::HTTPError, response.flush unless response.status.ok?
 
-        File.parse(client: @client, data: response.parse['file'])
+        File.parse(client: @client, data: response.parse["file"])
       end
 
-      protected
+    protected
 
       # @raise [FetchError]
       #

data/lib/omniai/google/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Google
-    VERSION = '1.9.3'
+    VERSION = "1.9.6"
   end
 end

data/lib/omniai/google.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require 'event_stream_parser'
-require 'omniai'
-require 'zeitwerk'
+require "event_stream_parser"
+require "omniai"
+require "zeitwerk"
 
 loader = Zeitwerk::Loader.for_gem
 loader.push_dir(__dir__, namespace: OmniAI)

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: omniai-google
 version: !ruby/object:Gem::Version
-  version: 1.9.3
+  version: 1.9.6
 platform: ruby
 authors:
 - Kevin Sylvestre
-autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-11-13 00:00:00.000000000 Z
+date: 2025-02-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -87,7 +86,6 @@ metadata:
   homepage_uri: https://github.com/ksylvest/omniai-google
   changelog_uri: https://github.com/ksylvest/omniai-google/releases
   rubygems_mfa_required: 'true'
-post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -102,8 +100,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.14
-signing_key: 
+rubygems_version: 3.6.2
 specification_version: 4
 summary: A generalized framework for interacting with Google
 test_files: []