Merge pull request #15 from GeraudBourdin/main

Add safe_mode alias to safe_prompt to reflect js library.
partITech · Apr 24, 2024 · 9fa20c3 · 9fa20c3
2 parents b488fec + 2ec123d
commit 9fa20c3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/MistralClient.php b/src/MistralClient.php
@@ -155,6 +155,10 @@ protected function makeChatCompletionRequest(Messages $messages, array $params,
             $return['safe_prompt'] = $params['safe_prompt'];
         }
 
+        if (isset($params['safe_mode']) && is_bool($params['safe_mode'])) {
+            $return['safe_prompt'] = $params['safe_prompt'];
+        }
+
         if (isset($params['tool_choice']) && is_array($params['tool_choice'])) {
             $return['tool_choice'] = $params['tool_choice'];
         }