From e34e4d3b71c8089728c04940e81790b159281b77 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 14 Dec 2023 18:42:43 +0530
Subject: [PATCH] (test) mistral api - safe_mode, random_seed

---
 litellm/tests/test_completion.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 78bbb94ebc7..acd7d90566f 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -70,6 +70,7 @@ def test_completion_claude():
 
 def test_completion_mistral_api():
     try:
+        litellm.set_verbose=True
         response = completion(
             model="mistral/mistral-tiny", 
             messages=[
@@ -77,7 +78,9 @@ def test_completion_mistral_api():
                     "role": "user",
                     "content": "Hey, how's it going?",
                 }
-            ]
+            ],
+            random_seed=42,
+            safe_mode = True
         )
         # Add any assertions here to check the response
         print(response)