update tests with deepseek models

defog-ai · Jan 21, 2025 · 893f946 · 893f946
1 parent e2dee90
commit 893f946
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/test_utils_multi_llm.py b/tests/test_utils_multi_llm.py
@@ -186,7 +186,9 @@ async def test_simple_chat_async(self):
             "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
             # "o1-mini", --o1-mini seems to be having issues, and o3-mini will be out soon anyway. so leaving out for now
             "o1",
-            "gemini-2.0-flash-exp"
+            "gemini-2.0-flash-exp",
+            "deepseek-chat",
+            "deepseek-reasoner"
         ]
         messages = [
             {"role": "user", "content": "Return a greeting in not more than 2 words\n"}
@@ -212,7 +214,9 @@ async def test_sql_chat_async(self):
             "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
             # "o1-mini", --o1-mini seems to be having issues, and o3-mini will be out soon anyway. so leaving out for now
             "o1",
-            "gemini-2.0-flash-exp"
+            "gemini-2.0-flash-exp",
+            "deepseek-chat",
+            "deepseek-reasoner"
         ]
         for model in models:
             response = await chat_async(
@@ -233,6 +237,8 @@ async def test_sql_chat_structured_async(self):
             "gpt-4o",
             "o1",
             "gemini-2.0-flash-exp",
+            "deepseek-chat",
+            "deepseek-reasoner"
         ]
         for model in models:
             response = await chat_async(