Add DeepSeek R1 VLLM model config

Author: ilmarinen

eureka_ml_insights/configs/model_configs.py

@@ -172,6 +172,15 @@
     },
 )
 
+OAI_GPT4O_2024_11_20_AUZRE_CONFIG = ModelConfig(
+    AzureOpenAIModel,
+    {
+        "model_name": "gpt-4o",
+        "url": "https://eurekaevals.openai.azure.com/",
+        "api_version": "2025-01-01-preview",
+    },
+)
+
 # Gemini models
 GEMINI_SECRET_KEY_PARAMS = {
     "key_name": "your_gemini_secret_key_name",
@@ -343,6 +352,16 @@
     },
 )
 
+DEEPSEEK_R1_LOCAL_CONFIG = ModelConfig(
+    LocalVLLMModel,
+    {
+        # this name must match the vllm deployment name/path
+        "model_name": "Deepseek-R1",
+        # specify ports in case the model is already deployed
+        "ports": ["5001"],
+    },
+)
+
 # DeepSeek R1 Endpoints on Azure
 DEEPSEEK_R1_CONFIG = ModelConfig(
     DeepseekR1ServerlessAzureRestEndpointModel,