omniai-mistral 1.6.0 → 1.6.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 253b0ae91739077ea70a562cad40ce7c16ead3a62a7650ae7511f4680e34c96b
-  data.tar.gz: bfed7d64757f41170581aafdc1726571bb399b2a6ffd845dff9062a2ebf21c7e
+  metadata.gz: 06a0ef7934fc23948925d5c35b75898df7148f162564db8a59fffa9106efa8d8
+  data.tar.gz: 22edea50f3d9a71b3ef659fd731078730adec47cc46190208fe30022a37a9d2e
 SHA512:
-  metadata.gz: a4dee74fcf4169cf99adeaa5239f9c6afa33f90b007b9d8c5e8368c332ab8e6a4b291e482d55429633957a81c40b2871b2e57ecdd98e99eb233fecd101d57ad6
-  data.tar.gz: 2d3256886c38c5823cf7fcdf2d6d377c2128cc375624aa7c596dd9bb6d308d780a760917b6c6487c18b46a2dbb82636eb05bb82cd692b63146963ecd4b58956f
+  metadata.gz: 506d07b0b05993c88c540778d3116f608dc3740726e6f0453a72ac444c8b9139c2020476fc3c1627f581a8fcb83e4c11113048b58feda99381f6480b56f1eaed
+  data.tar.gz: 0e8dae928a6d2d64ac459ce48d370bc866516f133fc8e60d14a581deb62ff6761cb3480a294ee8c5afb95a675d9ebc0e4887ec5683f8a80872f9bb8afcfd8447

data/lib/omniai/mistral/chat.rb

@@ -19,6 +19,8 @@ module OmniAI
         CODESTRAL = 'codestral-latest'
       end
 
+      DEFAULT_MODEL = Model::MEDIUM
+
       module Role
         ASSISTANT = 'assistant'
         USER = 'user'

data/lib/omniai/mistral/client.rb

@@ -44,16 +44,19 @@ module OmniAI
 
       # @raise [OmniAI::Error]
       #
-      # @param messages [String, Array, Hash]
+      # @param messages [String] optional
       # @param model [String] optional
       # @param format [Symbol] optional :text or :json
       # @param temperature [Float, nil] optional
       # @param stream [Proc, nil] optional
       # @param tools [Array<OmniAI::Tool>, nil] optional
       #
+      # @yield [prompt] optional
+      # @yieldparam prompt [OmniAI::Chat::Prompt]
+      #
       # @return [OmniAI::Chat::Completion]
-      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)
+      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
     end
   end

data/lib/omniai/mistral/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai-mistral
 version: !ruby/object:Gem::Version
-  version: 1.6.0
+  version: 1.6.2
 platform: ruby
 authors:
 - Kevin Sylvestre