omniai-mistral 1.6.2 → 1.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 06a0ef7934fc23948925d5c35b75898df7148f162564db8a59fffa9106efa8d8
-  data.tar.gz: 22edea50f3d9a71b3ef659fd731078730adec47cc46190208fe30022a37a9d2e
+  metadata.gz: d6fdad55d7cfc5b9d1845a8ec2d64772e2af5cc3d74d31b3d746732bc7509b15
+  data.tar.gz: d31338fb81948d32cb0937a5d206bd9393a92c4fb43523477a1508941e7f2c09
 SHA512:
-  metadata.gz: 506d07b0b05993c88c540778d3116f608dc3740726e6f0453a72ac444c8b9139c2020476fc3c1627f581a8fcb83e4c11113048b58feda99381f6480b56f1eaed
-  data.tar.gz: 0e8dae928a6d2d64ac459ce48d370bc866516f133fc8e60d14a581deb62ff6761cb3480a294ee8c5afb95a675d9ebc0e4887ec5683f8a80872f9bb8afcfd8447
+  metadata.gz: 2282359b0e6fcd610c1d73b7b9a236371196014458aea74585231de83a3697194879d3917b9f64b8ed5c7fabaab3a9a5f1405b849a5e891e55f7308a8143f2b4
+  data.tar.gz: 5358665adf78efed8c92f9e52f3b2854cd23fe6f82598c2639b25b707252906c798bf46250e6f6ec7d1b2c6d39822439497216063f6e398f22470d16c5c088e9

data/README.md

@@ -103,3 +103,12 @@ JSON.parse(completion.choice.message.content) # { "name": "Ringo" }
 [Mistral API Reference `response_format`](https://docs.mistral.ai/api/)
 
 > When using JSON mode you MUST also instruct the model to produce JSON yourself with a system or a user message.
+
+### 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/mistral/client.rb

@@ -58,6 +58,14 @@ module OmniAI
       def chat(messages = nil, model: Chat::DEFAULT_MODEL, temperature: nil, format: nil, stream: nil, tools: nil, &)
         Chat.process!(messages, model:, temperature:, format:, stream:, tools:, client: self, &)
       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
     end
   end
 end

data/lib/omniai/mistral/embed.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module Mistral
+    # An Mistral embed implementation.
+    #
+    # Usage:
+    #
+    #   input = "..."
+    #   response = OmniAI::Mistral::Embed.process!(input, client: client)
+    #   response.embedding [0.0, ...]
+    class Embed < OmniAI::Embed
+      module Model
+        EMBED = 'mistral-embed'
+      end
+
+      DEFAULT_MODEL = Model::EMBED
+
+      protected
+
+      # @return [Hash]
+      def payload
+        { model: @model, input: arrayify(@input) }
+      end
+
+      # @return [String]
+      def path
+        "/#{OmniAI::Mistral::Client::VERSION}/embeddings"
+      end
+
+      # @param [Object] value
+      # @return [Array]
+      def arrayify(value)
+        value.is_a?(Array) ? value : [value]
+      end
+    end
+  end
+end

data/lib/omniai/mistral/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Mistral
-    VERSION = '1.6.2'
+    VERSION = '1.7.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai-mistral
 version: !ruby/object:Gem::Version
-  version: 1.6.2
+  version: 1.7.0
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-07-18 00:00:00.000000000 Z
+date: 2024-08-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -65,6 +65,7 @@ files:
 - lib/omniai/mistral/chat.rb
 - lib/omniai/mistral/client.rb
 - lib/omniai/mistral/config.rb
+- lib/omniai/mistral/embed.rb
 - lib/omniai/mistral/version.rb
 homepage: https://github.com/ksylvest/omniai-mistral
 licenses: