From 8787b8e89dc50e27cc26e2e100f5ba07a8954b8a Mon Sep 17 00:00:00 2001
From: Matthew Farrellee <matt@cs.wisc.edu>
Date: Mon, 17 Jun 2024 17:16:10 -0400
Subject: [PATCH] add mistralai/mistral-7b-instruct-v0.3, 01-ai/yi-large,
 nvidia/nemotron-4-340b-instruct, and mistralai/codestral-22b-instruct-v0.1 to
 ChatNVIDIA supported chat models

---
 .../langchain_nvidia_ai_endpoints/_statics.py | 24 +++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
index 70288bb7..5703d4f0 100644
--- 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 = {