omniai-mistral 1.2.0 → 1.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 422f9590834021e3c89c5289fbab907797748ec42b486748f55db20845a9005f
-  data.tar.gz: e26a30e7e9532e7a0589f7c3fd076a621771693a8d2af3814f28ccd707bba12e
+  metadata.gz: c6f7b36b36c5460c0ca69d83449bcc696168a70b772fcd71dc74604a1c1b892b
+  data.tar.gz: 737c9f1d3fa9cfceddf4a0d7cbc074b6ee0769186d7390e387e7d4193237c0c9
 SHA512:
-  metadata.gz: 0ccec2ed5709e7993f8e0936959d4186e6e08458add50233ded53a0163eaab0dad4dd5d2183ba68add2c2346d1237676705eebca195fd63222db34396c962ac6
-  data.tar.gz: 58913437e42cb83b129920c12c3ce968cd3191e4d496f701f63cdd31509cdbd11e5810956d9ead04b1dfbb10c89b86cbee89c35d45a58f727da3ef5ea43c26d5
+  metadata.gz: 063e6252ade8a03eca3b8b29779676b549f0a89c4b6bc6c80f8e59930bbdd89c8018950d5ae7a84a4bab778416b9e7307aa75c083330271e2c0bea982f553821
+  data.tar.gz: 2d3c40e17fcd7c3ce777fa29d271a6bf3103d9c2a76b1b8d7cfa0d8be748bbba86cf2b3c9316618cde9e0393cb673c1299f08b8198d6f76790b665b5ce9d1291

data/Gemfile

@@ -5,7 +5,6 @@ source 'https://rubygems.org'
 gemspec
 
 gem 'rake'
-
 gem 'rspec'
 gem 'rspec_junit_formatter'
 gem 'rubocop'
@@ -13,3 +12,4 @@ gem 'rubocop-rake'
 gem 'rubocop-rspec'
 gem 'simplecov'
 gem 'webmock'
+gem 'yard'

data/lib/omniai/mistral/chat.rb

@@ -37,6 +37,7 @@ module OmniAI
           stream: @stream.nil? ? nil : !@stream.nil?,
           temperature: @temperature,
           response_format: (JSON_RESPONSE_FORMAT if @format.eql?(:json)),
+          tools: @tools&.map(&:prepare),
         }).compact
       end
 

data/lib/omniai/mistral/client.rb

@@ -24,21 +24,22 @@ module OmniAI
 
       # @param api_key [String] optional - defaults to `OmniAI::Mistral.config.api_key`
       # @param host [String] optional - defaults to `OmniAI::Mistral.config.host`
+      # @param logger [Logger] optional - defaults to `OmniAI::Mistral.config.logger`
+      # @param timeout [Integer] optional - defaults to `OmniAI::Mistral.config.timeout`
       def initialize(
         api_key: OmniAI::Mistral.config.api_key,
+        host: OmniAI::Mistral.config.host,
         logger: OmniAI::Mistral.config.logger,
-        host: OmniAI::Mistral.config.host
+        timeout: OmniAI::Mistral.config.timeout
       )
         raise(ArgumentError, %(ENV['MISTRAL_API_KEY'] must be defined or `api_key` must be passed)) if api_key.nil?
 
-        super(api_key:, logger:)
-
-        @host = host
+        super
       end
 
       # @return [HTTP::Client]
       def connection
-        @connection ||= HTTP.auth("Bearer #{api_key}").persistent(@host)
+        @connection ||= super.auth("Bearer #{api_key}")
       end
 
       # @raise [OmniAI::Error]
@@ -48,10 +49,11 @@ module OmniAI
       # @param format [Symbol] optional :text or :json
       # @param temperature [Float, nil] optional
       # @param stream [Proc, nil] optional
+      # @param tools [Array<OmniAI::Tool>, nil] optional
       #
       # @return [OmniAI::Chat::Completion]
-      def chat(messages, model: Chat::Model::MEDIUM, temperature: nil, format: nil, stream: nil)
-        Chat.process!(messages, model:, temperature:, format:, stream:, client: self)
+      def chat(messages, model: Chat::Model::MEDIUM, temperature: nil, format: nil, stream: nil, tools: nil)
+        Chat.process!(messages, model:, temperature:, format:, stream:, tools:, client: self)
       end
     end
   end

data/lib/omniai/mistral/config.rb

@@ -2,15 +2,21 @@
 
 module OmniAI
   module Mistral
-    # Config for the Mistral `api_key` / `host` / `logger`, `chat_options`.
+    # Configuration for Mistral.
     class Config < OmniAI::Config
-      attr_accessor :chat_options
+      DEFAULT_HOST = 'https://api.mistral.ai'
 
-      def initialize
+      # @param api_key [String, nil] optional - defaults to `ENV['MISTRAL_API_KEY']`
+      # @param host [String, nil] optional - defaults to `ENV['MISTRAL_HOST'] w/ fallback to `DEFAULT_HOST`
+      # @param logger [Logger, nil] optional - defaults to
+      # @param timeout [Integer, Hash, nil] optional
+      def initialize(
+        api_key: ENV.fetch('MISTRAL_API_KEY', nil),
+        host: ENV.fetch('MISTRAL_HOST', DEFAULT_HOST),
+        logger: nil,
+        timeout: nil
+      )
         super
-        @api_key = ENV.fetch('MISTRAL_API_KEY', nil)
-        @host = ENV.fetch('MISTRAL_HOST', 'https://api.mistral.ai')
-        @chat_options = {}
       end
     end
   end

data/lib/omniai/mistral/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Mistral
-    VERSION = '1.2.0'
+    VERSION = '1.5.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai-mistral
 version: !ruby/object:Gem::Version
-  version: 1.2.0
+  version: 1.5.0
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-06-20 00:00:00.000000000 Z
+date: 2024-07-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -88,7 +88,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.3
+rubygems_version: 3.5.14
 signing_key: 
 specification_version: 4
 summary: A generalized framework for interacting with Mistral