omniai-mistral 1.5.0 → 1.6.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c6f7b36b36c5460c0ca69d83449bcc696168a70b772fcd71dc74604a1c1b892b
-  data.tar.gz: 737c9f1d3fa9cfceddf4a0d7cbc074b6ee0769186d7390e387e7d4193237c0c9
+  metadata.gz: 8ab0ad2db899898af639d14d319f8de978b4056cddc0b4015fd65663d50e343f
+  data.tar.gz: 8a3843dbb174c054b26a98a657b0dba537cd77f4dba1dca1baac8d3080b756e7
 SHA512:
-  metadata.gz: 063e6252ade8a03eca3b8b29779676b549f0a89c4b6bc6c80f8e59930bbdd89c8018950d5ae7a84a4bab778416b9e7307aa75c083330271e2c0bea982f553821
-  data.tar.gz: 2d3c40e17fcd7c3ce777fa29d271a6bf3103d9c2a76b1b8d7cfa0d8be748bbba86cf2b3c9316618cde9e0393cb673c1299f08b8198d6f76790b665b5ce9d1291
+  metadata.gz: 2051bf93aac8437af6805882eb9f9d514ba4af1826b8a09234b1b1e67747ba86ae477600e646dd4d63472d464c6b430ee574d95d664c3b11fce891e3848cb33d
+  data.tar.gz: ca1433a9f56e7b859f90e37649d4c1096d790cfe40a112061b26e5ae4d6d532a15395c7f1b93fd60745db95ecd68ca0aaf449b75d154bc597082788e6cd7f078

data/README.md

@@ -46,21 +46,10 @@ completion.choice.message.content # 'Why did the chicken cross the road? To get
 ```
 
 ```ruby
-completion = client.chat({
-  role: OmniAI::Chat::Role::USER,
-  content: 'Is it wise to jump off a bridge?'
-})
-completion.choice.message.content # 'No.'
-```
-
-```ruby
-completion = client.chat([
-  {
-    role: OmniAI::Chat::Role::SYSTEM,
-    content: 'You are a helpful assistant.'
-  },
-  'What is the capital of Canada?',
-])
+completion = client.chat do |prompt|
+  prompt.system('You are a helpful assistant.')
+  prompt.user('What is the capital of Canada?')
+end
 completion.choice.message.content # 'The capital of Canada is Ottawa.'
 ```
 
@@ -104,10 +93,10 @@ client.chat('Be poetic.', stream:)
 `format` takes an optional symbol (`:json`) and that sets the `response_format` to `json_object`:
 
 ```ruby
-completion = client.chat([
-  { role: OmniAI::Chat::Role::SYSTEM, content: OmniAI::Chat::JSON_PROMPT },
-  { role: OmniAI::Chat::Role::USER, content: 'What is the name of the drummer for the Beatles?' }
-], format: :json)
+completion = client.chat(format: :json) do |prompt|
+  prompt.system(OmniAI::Chat::JSON_PROMPT)
+  prompt.user('What is the name of the drummer for the Beatles?')
+end
 JSON.parse(completion.choice.message.content) # { "name": "Ringo" }
 ```
 

data/lib/omniai/mistral/chat.rb

@@ -32,7 +32,7 @@ module OmniAI
       # @return [Hash]
       def payload
         OmniAI::Mistral.config.chat_options.merge({
-          messages:,
+          messages: @prompt.serialize,
           model: @model,
           stream: @stream.nil? ? nil : !@stream.nil?,
           temperature: @temperature,

data/lib/omniai/mistral/client.rb

@@ -51,9 +51,12 @@ module OmniAI
       # @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, 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/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai-mistral
 version: !ruby/object:Gem::Version
-  version: 1.5.0
+  version: 1.6.1
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-07-11 00:00:00.000000000 Z
+date: 2024-07-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser