diff --git a/tests/local_testing/test_router.py b/tests/local_testing/test_router.py index cd5e8f6b2..20867e766 100644 --- a/tests/local_testing/test_router.py +++ b/tests/local_testing/test_router.py @@ -1450,7 +1450,7 @@ async def test_mistral_on_router(): { "model_name": "gpt-3.5-turbo", "litellm_params": { - "model": "mistral/mistral-medium", + "model": "mistral/mistral-small-latest", }, }, ] diff --git a/tests/local_testing/test_streaming.py b/tests/local_testing/test_streaming.py index 0bc6953f9..757ff4d61 100644 --- a/tests/local_testing/test_streaming.py +++ b/tests/local_testing/test_streaming.py @@ -683,7 +683,7 @@ def test_completion_ollama_hosted_stream(): [ # "claude-3-5-haiku-20241022", # "claude-2", - # "mistral/mistral-medium", + # "mistral/mistral-small-latest", "openrouter/openai/gpt-4o-mini", ], )