omniai-mistral 1.2.0 → 1.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 422f9590834021e3c89c5289fbab907797748ec42b486748f55db20845a9005f
-  data.tar.gz: e26a30e7e9532e7a0589f7c3fd076a621771693a8d2af3814f28ccd707bba12e
+  metadata.gz: 382f5eb9d01087dbab76c84890ddd1414e89017714c211ec4163206f56f7557b
+  data.tar.gz: dad60217422c646066cd2c0a60339d52cd4347df31175ec889fdfbd735e19a33
 SHA512:
-  metadata.gz: 0ccec2ed5709e7993f8e0936959d4186e6e08458add50233ded53a0163eaab0dad4dd5d2183ba68add2c2346d1237676705eebca195fd63222db34396c962ac6
-  data.tar.gz: 58913437e42cb83b129920c12c3ce968cd3191e4d496f701f63cdd31509cdbd11e5810956d9ead04b1dfbb10c89b86cbee89c35d45a58f727da3ef5ea43c26d5
+  metadata.gz: 79e0f8d805a3d577facbbd6477be06ca38de8da9370987e6ff0bea1bc9175222d12ea188d15769ecb571849ff1c3e06f27ff4ca7ef01d98b82d1651dc0670656
+  data.tar.gz: 773bc4ba186ee0a1ae2d27c8ca1862dfafd5d44b90ed99993f8e420c5eba89422855ba433ea14c2d894a327e8e44194df1ff25f7d4436c7ff335f0932d5701e2

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]

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.3.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.3.0
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-06-20 00:00:00.000000000 Z
+date: 2024-06-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser