add microsoft/phi-3-medium-128k-instruct and deepseek-ai/deepseek-cod…

…er-6.7b-instruct to ChatNVIDIA supported models
langchain-ai · Jul 5, 2024 · 391a326 · 391a326
1 parent d684480
commit 391a326
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
@@ -245,6 +245,18 @@ def __hash__(self) -> int:
         client="ChatNVIDIA",
         aliases=["ai-gemma-2-27b-it"],
     ),
+    "microsoft/phi-3-medium-128k-instruct": Model(
+        id="microsoft/phi-3-medium-128k-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+        aliases=["ai-phi-3-medium-128k-instruct"],
+    ),
+    "deepseek-ai/deepseek-coder-6.7b-instruct": Model(
+        id="deepseek-ai/deepseek-coder-6.7b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+        aliases=["ai-deepseek-coder-6_7b-instruct"],
+    ),
 }
 
 QA_MODEL_TABLE = {