add mistralai/mistral-7b-instruct-v0.3, 01-ai/yi-large, nvidia/nemotr…

…on-4-340b-instruct, and mistralai/codestral-22b-instruct-v0.1 to ChatNVIDIA supported chat models
langchain-ai · Jun 17, 2024 · 8787b8e · 8787b8e
1 parent 0f2d90b
commit 8787b8e
Showing 1 changed file with 24 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
@@ -208,6 +208,30 @@ def __hash__(self) -> int:
         client="ChatNVIDIA",
         aliases=["ai-palmyra-med-70b"],
     ),
+    "mistralai/mistral-7b-instruct-v0.3": Model(
+        id="mistralai/mistral-7b-instruct-v0.3",
+        model_type="chat",
+        client="ChatNVIDIA",
+        aliases=["ai-mistral-7b-instruct-v03"],
+    ),
+    "01-ai/yi-large": Model(
+        id="01-ai/yi-large",
+        model_type="chat",
+        client="ChatNVIDIA",
+        aliases=["ai-yi-large"],
+    ),
+    "nvidia/nemotron-4-340b-instruct": Model(
+        id="nvidia/nemotron-4-340b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+        aliases=["qa-nemotron-4-340b-instruct"],
+    ),
+    "mistralai/codestral-22b-instruct-v0.1": Model(
+        id="mistralai/codestral-22b-instruct-v0.1",
+        model_type="chat",
+        client="ChatNVIDIA",
+        aliases=["ai-codestral-22b-instruct-v01"],
+    ),
 }
 
 VLM_MODEL_TABLE = {