omniai-mistral 1.0.4 → 1.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a8ee32af7fae89e3e808d4cc3a041b360aa4864b1fd737b5a022001207500a2c
-  data.tar.gz: 499fe0f4993fd18effaca28b1d4b6f878a9f9612c396a9567b4d0ff8a242c097
+  metadata.gz: 382f5eb9d01087dbab76c84890ddd1414e89017714c211ec4163206f56f7557b
+  data.tar.gz: dad60217422c646066cd2c0a60339d52cd4347df31175ec889fdfbd735e19a33
 SHA512:
-  metadata.gz: ee764591f245f35c429c639f5bf9d7859a42a7d6b2c03c159bd911b1be48861ec9774eb1e98643d8d516386132e5324f67e20936ed4e77db53fc5dbdcabf3d6c
-  data.tar.gz: eca77c0fdc184c53c8bd590e1bae612a4f7f058a03446ba804689f455dd514c374bfe59bd9893f979feadace0ee490c10b0e1dd000a8d608269a8ae3fcbd84c7
+  metadata.gz: 79e0f8d805a3d577facbbd6477be06ca38de8da9370987e6ff0bea1bc9175222d12ea188d15769ecb571849ff1c3e06f27ff4ca7ef01d98b82d1651dc0670656
+  data.tar.gz: 773bc4ba186ee0a1ae2d27c8ca1862dfafd5d44b90ed99993f8e420c5eba89422855ba433ea14c2d894a327e8e44194df1ff25f7d4436c7ff335f0932d5701e2

data/Gemfile

@@ -11,4 +11,5 @@ gem 'rspec_junit_formatter'
 gem 'rubocop'
 gem 'rubocop-rake'
 gem 'rubocop-rspec'
+gem 'simplecov'
 gem 'webmock'

data/README.md

@@ -1,5 +1,7 @@
 # OmniAI::Mistral
 
+[![CircleCI](https://circleci.com/gh/ksylvest/omniai-mistral.svg?style=svg)](https://circleci.com/gh/ksylvest/omniai-mistral)
+
 An Mistral implementation of the [OmniAI](https://github.com/ksylvest/omniai) APIs.
 
 ## Installation

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.0.4'
+    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.0.4
+  version: 1.3.0
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-06-15 00:00:00.000000000 Z
+date: 2024-06-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -67,7 +67,8 @@ files:
 - lib/omniai/mistral/config.rb
 - lib/omniai/mistral/version.rb
 homepage: https://github.com/ksylvest/omniai-mistral
-licenses: []
+licenses:
+- MIT
 metadata:
   homepage_uri: https://github.com/ksylvest/omniai-mistral
   changelog_uri: https://github.com/ksylvest/omniai-mistral/releases