From 50d2510b60fd7b825bb1a5d79ba3bb741cfccc1b Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 21 Nov 2024 23:44:40 +0530
Subject: [PATCH] test: cleanup mistral model

---
 tests/local_testing/test_router.py    | 2 +-
 tests/local_testing/test_streaming.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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",
     ],
 )