Support nil messages

ksylvest · Jul 18, 2024 · 3123155 · 3123155
1 parent 95f6215
commit 3123155
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 5 deletions.
diff --git a/Gemfile.lock b/Gemfile.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    omniai-mistral (1.6.1)
+    omniai-mistral (1.6.2)
       event_stream_parser
       omniai
       zeitwerk
@@ -50,7 +50,7 @@ GEM
     llhttp-ffi (0.5.0)
       ffi-compiler (~> 1.0)
       rake (~> 13.0)
-    omniai (1.6.1)
+    omniai (1.6.2)
       event_stream_parser
       http
       zeitwerk

diff --git a/lib/omniai/mistral/chat.rb b/lib/omniai/mistral/chat.rb
@@ -19,6 +19,8 @@ module Model
         CODESTRAL = 'codestral-latest'
       end
 
+      DEFAULT_MODEL = Model::MEDIUM
+
       module Role
         ASSISTANT = 'assistant'
         USER = 'user'

diff --git a/lib/omniai/mistral/client.rb b/lib/omniai/mistral/client.rb
@@ -44,7 +44,7 @@ def connection
 
       # @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
@@ -55,7 +55,7 @@ def connection
       # @yieldparam prompt [OmniAI::Chat::Prompt]
       #
       # @return [OmniAI::Chat::Completion]
-      def chat(messages, model: Chat::Model::MEDIUM, temperature: nil, format: nil, stream: nil, tools: nil, &)
+      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

diff --git a/lib/omniai/mistral/version.rb b/lib/omniai/mistral/version.rb
@@ -2,6 +2,6 @@
 
 module OmniAI
   module Mistral
-    VERSION = '1.6.1'
+    VERSION = '1.6.2'
   end
 end