Merge pull request #14 from GeraudBourdin/main

Add response_format json.
partITech · Apr 24, 2024 · b488fec · b488fec
2 parents 5b86271 + 6183243
commit b488fec
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/MistralClient.php b/src/MistralClient.php
@@ -20,6 +20,8 @@ class MistralClient
     const string TOOL_CHOICE_AUTO = 'auto';
     const string TOOL_CHOICE_NONE = 'none';
 
+    const int RESPONSE_FORMAT_JSON = 0;
+
     const array RETRY_STATUS_CODES = [
         429,
         500 => GenericRetryStrategy::IDEMPOTENT_METHODS,
@@ -178,8 +180,8 @@ protected function makeChatCompletionRequest(Messages $messages, array $params,
         }
 
         if (isset($params['presence_penalty']) && is_numeric(
-                $params['presence_penalty']
-            ) && $params['presence_penalty'] >= -2 && $params['presence_penalty'] <= 2) {
+            $params['presence_penalty']
+        ) && $params['presence_penalty'] >= -2 && $params['presence_penalty'] <= 2) {
             $return['presence_penalty'] = (float)$params['presence_penalty'];
         }
 
@@ -214,6 +216,12 @@ protected function makeChatCompletionRequest(Messages $messages, array $params,
             $return['guided_json'] = json_encode($params['guided_json']);
         }
 
+        if(isset($params['response_format']) && $params['response_format'] === self::RESPONSE_FORMAT_JSON) {
+            $return['response_format'] = [
+                'type' => 'json_object'
+            ];
+        }
+
         return $return;
     }