chore: update llm from github models provider

promplate · Feb 19, 2025 · 42f3f35 · 42f3f35
1 parent 5942d55
commit 42f3f35
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/utils/llm/__init__.py b/src/utils/llm/__init__.py
@@ -36,12 +36,14 @@
     "azure:gpt-4o-mini",
     "azure:o1",
     "azure:o1-mini",
+    "azure:o3-mini",
     "azure:o1-preview",
     "Mistral-Nemo",
     "Mistral-large",
     "Mistral-large-2407",
     "Mistral-large-2411",
     "Mistral-small",
+    "Codestral-2501",
     "Ministral-3B",
     "Meta-Llama-3.1-405B-Instruct",
     "Meta-Llama-3.1-70B-Instruct",
@@ -54,6 +56,7 @@
     "Cohere-command-r-08-2024",
     "AI21-Jamba-1.5-Large",
     "AI21-Jamba-1.5-Mini",
+    "Phi-4",
     "Phi-3.5-MoE-instruct",
     "Phi-3.5-mini-instruct",
     "Phi-3-medium-128k-instruct",
@@ -62,6 +65,7 @@
     "Phi-3-mini-4k-instruct",
     "Phi-3-small-128k-instruct",
     "Phi-3-small-8k-instruct",
+    "DeepSeek-R1",
     "chatglm_turbo",
     "claude-instant-1.2",
     "claude-2.1",

diff --git a/src/utils/llm/azure.py b/src/utils/llm/azure.py
@@ -14,13 +14,16 @@
 
 
 @link_llm("azure:o1")
+@link_llm("azure:o3")
 @link_llm("azure:gpt")
 @link_llm("Ministral")
+@link_llm("Codestral")
 @link_llm("Mistral")
 @link_llm("Meta")
 @link_llm("Cohere")
 @link_llm("AI21")
 @link_llm("Phi")
+@link_llm("DeepSeek")
 class AzureOpenAI(AsyncChatOpenAI):
     @staticmethod
     async def generate(prompt, **kwargs):