max token as 1

openlit · Mar 24, 2024 · 90bd3ef · 90bd3ef
1 parent 8c6a2ac
commit 90bd3ef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_anthropic.py b/tests/test_anthropic.py
@@ -41,7 +41,7 @@ def test_messages():
     """
     try:
         message = client.messages.create(
-            max_tokens=10,
+            max_tokens=1,
             messages=[
                 {
                     "role": "user",

diff --git a/tests/test_mistral.py b/tests/test_mistral.py
@@ -48,7 +48,7 @@ def test_chat():
     message = client.chat(
         model="mistral-large-latest",
         messages=messages,
-        max_tokens=10,
+        max_tokens=1,
     )
     assert message.object == 'chat.completion'