test: cleanup mistral model

2024-11-21 23:44:40 +05:30 · 2024-11-21 23:44:40 +05:30 · 50d2510b60
commit 50d2510b60
parent ddfe687b13
2 changed files with 2 additions and 2 deletions
--- 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",
            },
        },
    ]
--- 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",
    ],
 )