voyageai 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7f4d48a99c21b4d0120511d09b89038e3a6082a145e620fb6ecaf427f0247390
-  data.tar.gz: 4d931a99a02500a3173404b0ec7ada0c9f1a5203f074d09ddd6ef7769a884aed
+  metadata.gz: a2fc61d8673a96f88ae7e69d9cfb27f00c741f077fd05793ff921e27c0fd869a
+  data.tar.gz: 4734e6ac573b6251ca1789a37fbf9a9db8294c5e82d63b51613b20bfea793e10
 SHA512:
-  metadata.gz: 4f559800105a83db3479237cfa5c2fcd576d609a15a3d88dfc8f448750ee4fd2189abf6ce11d039f2879958b8ba418e00e214b850ce357487a562a4e13fc1f9f
-  data.tar.gz: 8cfaf28bcc73fc5540a17cf857c2458dc4197e4d452e0cd2dbdd409e6d1024992f1eb22998a19ca5733cb7ae3c2208863f71d66b475c60084cba749c9d9b5dab
+  metadata.gz: d403d37213b6f9666fbafb58fbc82fcbb9752be9e5d29a9e88d34ea6a198f43d860e809e2d890906bf0c298254d9054fcace76d3e5e8e3f70620abd8d27ec791
+  data.tar.gz: f1ad0fcaf76b8a0a0296dc46507c46d6d9b7e2f47a21b6ca9e2d7e8140483c18ece161ff4570db71649c344679b7a92e80287648324a3700c720e00a7b6e2854

data/README.md

@@ -10,6 +10,23 @@ gem install voyageai
 
 ## Usage
 
+### Generating Single Embedding
+
+```ruby
+require 'voyageai'
+
+input = 'A quick brown fox jumps over the lazy dog.'
+
+voyageai = VoyageAI::Client.new(api_key: 'pa-...') # or configure ENV['VOYAGEAI_API_KEY']
+
+embed = voyageai.emed(input)
+embed.model # "..."
+embed.usage # "#<VoyageAI::Usage total_tokens=...>"
+embed.embedding # [0.0, ...]
+```
+
+### Generating Multiple Embeddings
+
 ```ruby
 require 'voyageai'
 
@@ -22,10 +39,8 @@ input = [
 
 voyageai = VoyageAI::Client.new(api_key: 'pa-...') # or configure ENV['VOYAGEAI_API_KEY']
 
-result = voyageai.embed(input)
+embed = voyageai.embed(input)
 embed.model # "..."
 embed.usage # "#<VoyageAI::Usage total_tokens=...>"
-embed.embeddings.each do |embedding|
-  embedding.index # "#<VoyageAI::Embedding index=... embedding=...>
-end
+embed.embeddings # [[0.0, ...], ...]
 ```

data/lib/voyageai/client.rb

@@ -33,7 +33,7 @@ module VoyageAI
     #
     # @return [Embedding]
     def embed(input, model: Model::VOYAGE)
-      payload = { input: input, model: model }
+      payload = { input: arrayify(input), model: model }
       response = HTTP
         .accept(:json)
         .auth("Bearer #{@api_key}")
@@ -43,5 +43,11 @@ module VoyageAI
 
       Embed.parse(data: response.parse)
     end
+
+    private
+
+    def arrayify(input)
+      input.is_a?(Array) ? input : [input]
+    end
   end
 end

data/lib/voyageai/embed.rb

@@ -15,7 +15,7 @@ module VoyageAI
     attr_accessor :usage
 
     # @!attribute [rw] embeddings
-    #   @return [Array<Embedding>]
+    #   @return [Array<Array<Float>>]
     attr_accessor :embeddings
 
     # @param data [Hash]
@@ -23,9 +23,7 @@ module VoyageAI
     def self.parse(data:)
       model = data["model"]
       usage = Usage.parse(data: data["usage"])
-      embeddings = data["data"].map do |embedding_data|
-        Embedding.parse(data: embedding_data)
-      end
+      embeddings = data["data"].map { |embedding_data| embedding_data["embedding"] }
 
       Embed.new(model: model, usage: usage, embeddings: embeddings)
     end
@@ -43,5 +41,12 @@ module VoyageAI
     def inspect
       "#<#{self.class.name} model=#{@model.inspect} embeddings=#{@embeddings.inspect} usage=#{@usage.inspect}>"
     end
+
+    # @param index [Integer] optional
+    #
+    # @return [Array<Float>]
+    def embedding(index: 0)
+      @embeddings[index]
+    end
   end
 end

data/lib/voyageai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module VoyageAI
-  VERSION = "1.0.1"
+  VERSION = "1.1.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: voyageai
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors:
 - Kevin Sylvestre
@@ -52,7 +52,6 @@ files:
 - lib/voyageai.rb
 - lib/voyageai/client.rb
 - lib/voyageai/embed.rb
-- lib/voyageai/embedding.rb
 - lib/voyageai/model.rb
 - lib/voyageai/usage.rb
 - lib/voyageai/version.rb

data/lib/voyageai/embedding.rb

@@ -1,37 +0,0 @@
-# frozen_string_literal: true
-
-module VoyageAI
-  # An embedding returned by the VoyageAI API.
-  #
-  # @example
-  #   VoyageAI::Embedding.new(index: 0, embedding: [0.0, 1.0, 2.0, 3.0])
-  class Embedding
-    # @!attribute [rw] model
-    #   @return [Integer]
-    attr_accessor :index
-
-    # @!attribute [rw] embedding
-    #   @return [Array<Float>]
-    attr_accessor :embedding
-
-    # @param data [Hash]
-    # @return [Embedding]
-    def self.parse(data:)
-      index = data["index"]
-      embedding = data["embedding"]
-      new(index:, embedding:)
-    end
-
-    # @param index [Integer]
-    # @param embedding [Array<Float>]
-    def initialize(index:, embedding:)
-      @index = index
-      @embedding = embedding
-    end
-
-    # @return [String]
-    def inspect
-      "#<#{self.class.name} index=#{index} embedding=#{embedding.inspect}>"
-    end
-  end
-end