omniai-google 1.9.2 → 1.9.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dd9502894aaf9bafda46ad7c4475d0cd155f7ac50b582c5dd2d334357e082c61
-  data.tar.gz: 6e7ccb59aa0bcb92654cf1e6917da4f058b1687aefda2ec096c8f1e725953b47
+  metadata.gz: 80e08395bc3710a72cc59b7c110cc50b1ba4a1dc64e15c622fcacb39448c1265
+  data.tar.gz: cccb64b57c0430caf449ebb1de2620e597409de0179379ef0cb26588d41418a2
 SHA512:
-  metadata.gz: 3100656557336fc035bdb236ecd500b5e17289598f1153af80863c8aac81ecebcd6722fb2166e184a5883ffc5515233294b09cf63edec0c1b3279f928d6788c0
-  data.tar.gz: cd9f92d8b16721306681c83352b3f305eac681a4e0370b24407ca7019a37d3684521386a5682b17ca38e165f2cff84a34d6d84ea409df0295d186dcf7180a94d
+  metadata.gz: cf8a4e78c06a9808930300c1aa05b45211ab0f733a79bddc22bd5ac3ca2bfff1a00d075ca596122295c82aeb909372d2a9c7a66bc27566386d529428209beda7
+  data.tar.gz: 7324a4d99c3ce3a890129eb5a3091459facbd0acf5818b7ff0624d168d2fee94613656db1fe1c1091f8f001c6c34cc34c66a835b69fb7d90191075bde455de5b

data/README.md

@@ -120,3 +120,12 @@ end
 ```
 
 [Google API Reference `stream`](https://ai.google.dev/gemini-api/docs/api-overview#stream)
+
+### Embed
+
+Text can be converted into a vector embedding for similarity comparison usage via:
+
+```ruby
+response = client.embed('The quick brown fox jumps over a lazy dog.')
+response.embedding # [0.0, ...]
+```

data/lib/omniai/google/client.rb

@@ -68,6 +68,23 @@ module OmniAI
       def upload(io)
         Upload.process!(client: self, io:)
       end
+
+      # @raise [OmniAI::Error]
+      #
+      # @param input [String, Array<String>, Array<Integer>] required
+      # @param model [String] optional
+      def embed(input, model: Embed::DEFAULT_MODEL)
+        Embed.process!(input, model:, client: self)
+      end
+
+      # @return [String]
+      def path
+        if @project_id
+          "/#{@version}/projects/#{@project_id}/locations/#{@location}/publishers/google"
+        else
+          "/#{@version}"
+        end
+      end
     end
   end
 end

data/lib/omniai/google/embed.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module Google
+    # An Google embed implementation.
+    #
+    # Usage:
+    #
+    #   input = "..."
+    #   response = OmniAI::Google::Embed.process!(input, client: client)
+    #   response.embedding [0.0, ...]
+    class Embed < OmniAI::Embed
+      module Model
+        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
+
+      DEFAULT_MODEL = Model::EMBEDDING
+
+      EMBEDDINGS_DESERIALIZER = proc do |data, *|
+        data['embeddings'].map { |embedding| embedding['values'] }
+      end
+
+      # @return [Context]
+      CONTEXT = Context.build do |context|
+        context.deserializers[:embeddings] = EMBEDDINGS_DESERIALIZER
+      end
+
+      protected
+
+      # @param response [HTTP::Response]
+      # @return [Response]
+      def parse!(response:)
+        Response.new(data: response.parse, context: CONTEXT)
+      end
+
+      # @return [Array<Hash<{ text: String }>]
+      def requests
+        arrayify(@input).map do |text|
+          {
+            model: "models/#{@model}",
+            content: { parts: [{ text: text }] },
+          }
+        end
+      end
+
+      # @return [Hash]
+      def payload
+        { requests: }
+      end
+
+      # @return [String]
+      def path
+        "/#{@client.path}/models/#{@model}:batchEmbedContents?key=#{@client.api_key}"
+      end
+
+      def arrayify(input)
+        input.is_a?(Array) ? input : [input]
+      end
+    end
+  end
+end

data/lib/omniai/google/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai-google
 version: !ruby/object:Gem::Version
-  version: 1.9.2
+  version: 1.9.3
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-11-06 00:00:00.000000000 Z
+date: 2024-11-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -76,6 +76,7 @@ files:
 - lib/omniai/google/chat/usage_serializer.rb
 - lib/omniai/google/client.rb
 - lib/omniai/google/config.rb
+- lib/omniai/google/embed.rb
 - lib/omniai/google/upload.rb
 - lib/omniai/google/upload/file.rb
 - lib/omniai/google/version.rb