omniai-google 3.7.0 → 3.7.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 81bb938d26c6804a6942a17e57e1b1393f0e77925467a60659443f28c76d3752
-  data.tar.gz: ce70db23296a73089d2cbc6eaf344ae269af945c112dfc0942a17e31dc116dfe
+  metadata.gz: 74848884db95fef34b6c7306b2814ecf8d50797963e690a93b77e69c93cc7d3a
+  data.tar.gz: 9fbb92e5fe4fe36d9086182391ae383180607b5e97b36a17e458dc200bef73e1
 SHA512:
-  metadata.gz: 0f5101592e2603341f545b3402a0c4b3b3516dd3fee36532d2a5691803e11610124db98d89af4e532bd16ccfff2663a37308d999dee79aea700a3066df559771
-  data.tar.gz: 78109e536527bc9853e84111e664c25c3574a0f7c952fadbaf2a025478be0172c7cd1a203c4bdf88aa185b242f2fb74f9b88f6d88205a0ec8c479bfe2391b6a0
+  metadata.gz: c92c79303ec17e46832115d075aa29760ba8d1f52111286d51d370c49c5266bb2e0ec9956e058a133de69bd0bb28c8ea4fa52efab34edea64f9cdc896397069e
+  data.tar.gz: 5fc61a87b13a8bd3ba73bc6b8c150c7da1d77aef938d7c826f9eac541b78ea288fb0698bbaf9a7609d7a19a85607b3d04d97e82cd013a46f64c319c86329ea60

data/lib/omniai/google/client.rb

@@ -86,8 +86,8 @@ module OmniAI
       # @param input [String, Array<String>, Array<Integer>] required
       # @param model [String] optional
       # @param options [Hash] provider-specific options (e.g. task_type: "RETRIEVAL_DOCUMENT")
-      def embed(input, model: Embed::DEFAULT_MODEL, **options)
-        Embed.process!(input, model:, client: self, **options)
+      def embed(input, model: Embed::DEFAULT_MODEL, **)
+        Embed.process!(input, model:, client: self, **)
       end
 
       # @raise [OmniAI::Error]

data/lib/omniai/google/embed.rb

@@ -44,7 +44,7 @@ module OmniAI
         prompt_tokens = data.dig("usageMetadata", "promptTokenCount")
         total_tokens = data.dig("usageMetadata", "totalTokenCount")
 
-        Usage.new(prompt_tokens: prompt_tokens, total_tokens: total_tokens)
+        Usage.new(prompt_tokens:, total_tokens:)
       end
 
       # @return [Context]
@@ -73,13 +73,14 @@ module OmniAI
       #
       # @return [Symbol] :embed_content, :predict, or :batch_embed_contents
       def endpoint
-        @endpoint ||= if @client.vertex? && @model.start_with?("gemini-embedding-2")
-          :embed_content
-        elsif @client.vertex?
-          :predict
-        else
-          :batch_embed_contents
-        end
+        @endpoint ||=
+          if @client.vertex? && @model.start_with?("gemini-embedding-2")
+            :embed_content
+          elsif @client.vertex?
+            :predict
+          else
+            :batch_embed_contents
+          end
       end
 
       # @return [Context]
@@ -106,7 +107,7 @@ module OmniAI
         raise ArgumentError, "embedContent does not support batch input" if @input.is_a?(Array) && @input.length > 1
 
         text = @input.is_a?(Array) ? @input.first : @input
-        result = { content: { parts: [{ text: text }] } }
+        result = { content: { parts: [{ text: }] } }
         result[:taskType] = @options[:task_type] if @options[:task_type]
         result
       end
@@ -126,11 +127,11 @@ module OmniAI
           requests: inputs.map do |text|
             request = {
               model: "models/#{@model}",
-              content: { parts: [{ text: text }] },
+              content: { parts: [{ text: }] },
             }
             request[:taskType] = @options[:task_type] if @options[:task_type]
             request
-          end
+          end,
         }
       end
 
@@ -139,15 +140,15 @@ module OmniAI
         { key: (@client.api_key unless @client.credentials?) }.compact
       end
 
+      PROCEDURES = {
+        embed_content: "embedContent",
+        predict: "predict",
+        batch_embed_contents: "batchEmbedContents",
+      }.freeze
+
       # @return [String]
       def path
-        procedure = case endpoint
-                    when :embed_content then "embedContent"
-                    when :predict then "predict"
-                    when :batch_embed_contents then "batchEmbedContents"
-                    end
-
-        "/#{@client.path}/models/#{@model}:#{procedure}"
+        "/#{@client.path}/models/#{@model}:#{PROCEDURES[endpoint]}"
       end
     end
   end

data/lib/omniai/google/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Google
-    VERSION = "3.7.0"
+    VERSION = "3.7.1"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai-google
 version: !ruby/object:Gem::Version
-  version: 3.7.0
+  version: 3.7.1
 platform: ruby
 authors:
 - Kevin Sylvestre