fix test_mistralai_stream_chat

genai-impact · Apr 6, 2024 · c3c417c · c3c417c
1 parent f0fa1d2
commit c3c417c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_mistralai.py b/tests/test_mistralai.py
@@ -29,8 +29,7 @@ async def test_mistralai_async_chat(tracer_init):
 def test_mistralai_stream_chat(tracer_init):
     client = MistralClient()
     stream = client.chat_stream(
-        model="gpt-3.5-turbo",
-        messages=[{"role": "user", "content": "Hello World!"}],
+        messages=[{"role": "user", "content": "Hello World!"}], model="mistral-tiny"
     )
     for chunk in stream:
         assert chunk.impacts.energy >= 0