feat(chat): add mistral-small-3

deedy5 · Feb 20, 2025 · 0dd945f · 0dd945f
1 parent fd2a63b
commit 0dd945f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/duckduckgo_search/duckduckgo_search.py b/duckduckgo_search/duckduckgo_search.py
@@ -38,7 +38,7 @@ class DDGS:
         "llama-3.3-70b": "meta-llama/Llama-3.3-70B-Instruct-Turbo",
         "claude-3-haiku": "claude-3-haiku-20240307",
         "o3-mini": "o3-mini",
-        "mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1",
+        "mistral-small-3": "mistralai/Mistral-Small-24B-Instruct-2501",
     }
 
     def __init__(

diff --git a/tests/test_duckduckgo_search.py b/tests/test_duckduckgo_search.py
@@ -15,7 +15,7 @@ def test_context_manager():
         assert 20 <= len(results) <= 30
 
 
-@pytest.mark.parametrize("model", ["gpt-4o-mini", "llama-3.3-70b", "claude-3-haiku", "o3-mini", "mixtral-8x7b"])
+@pytest.mark.parametrize("model", ["gpt-4o-mini", "llama-3.3-70b", "claude-3-haiku", "o3-mini", "mistral-small-3"])
 def test_chat(model):
     results = DDGS().chat("cat", model=model)
     assert  len(results) >= 1