From 90bd3efbb9dd27b74919f2d94e13302f08d96ef3 Mon Sep 17 00:00:00 2001
From: patcher99 <patcher99@dokulabs.com>
Date: Sun, 24 Mar 2024 14:14:34 +0530
Subject: [PATCH] max token as 1

---
 tests/test_anthropic.py | 2 +-
 tests/test_mistral.py   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_anthropic.py b/tests/test_anthropic.py
index e3fdae6..091243e 100644
--- 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
index 0e5b136..50aa3a0 100644
--- 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'