From 8c6a2ac57c3a3449d424cb643db97cb2e0b284a3 Mon Sep 17 00:00:00 2001
From: patcher99 <patcher99@dokulabs.com>
Date: Sun, 24 Mar 2024 14:09:27 +0530
Subject: [PATCH] add max tokens to mistral tests

---
 tests/test_mistral.py | 1 +
 1 file changed, 1 insertion(+)

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