chore: add new models from azure

Author: CNSeniorious000

src/utils/llm/__init__.py

@@ -50,9 +50,14 @@
     "gpt-4-turbo-2024-04-09",
     "azure:gpt-4o",
     "azure:gpt-4o-mini",
+    "azure:gpt-4.1",
+    "azure:gpt-4.1-mini",
+    "azure:gpt-4.1-nano",
     "azure:o1",
+    "azure:o3",
     "azure:o1-mini",
     "azure:o3-mini",
+    "azure:o4-mini",
     "azure:o1-preview",
     "Mistral-Nemo",
     "Mistral-large",
@@ -71,6 +76,7 @@
     "Cohere-command-r-plus-08-2024",
     "Cohere-command-r",
     "Cohere-command-r-08-2024",
+    "Cohere-command-a",
     "AI21-Jamba-1.5-Large",
     "AI21-Jamba-1.5-Mini",
     "Phi-4",
@@ -82,6 +88,7 @@
     "Phi-3-mini-4k-instruct",
     "Phi-3-small-128k-instruct",
     "Phi-3-small-8k-instruct",
+    "MAI-DS-R1",
     "DeepSeek-R1",
     "DeepSeek-V3",
     "chatglm_turbo",

src/utils/llm/azure.py

@@ -13,8 +13,7 @@
 )
 
 
-@link_llm("azure:o1")
-@link_llm("azure:o3")
+@link_llm("azure:o")
 @link_llm("azure:gpt")
 @link_llm("Ministral")
 @link_llm("Codestral")
@@ -24,6 +23,7 @@
 @link_llm("AI21")
 @link_llm("Phi")
 @link_llm("DeepSeek")
+@link_llm("MAI")
 class AzureOpenAI(AsyncChatOpenAI):
     @staticmethod
     async def generate(prompt, **kwargs):