add nvidia/nemotron-4-mini-hindi-4b-instruct and nvidia/llama-3.1-nem…

…otron-70b-instruct to set of supported chat models
langchain-ai · Oct 28, 2024 · 5637081 · 5637081
1 parent cd48c4a
commit 5637081
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
@@ -469,6 +469,18 @@ def validate_client(self) -> "Model":
         model_type="chat",
         client="ChatNVIDIA",
     ),
+    "nvidia/nemotron-4-mini-hindi-4b-instruct": Model(
+        id="nvidia/nemotron-4-mini-hindi-4b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+        supports_structured_output=True,
+    ),
+    "nvidia/llama-3.1-nemotron-70b-instruct": Model(
+        id="nvidia/llama-3.1-nemotron-70b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+        supports_structured_output=True,
+    ),
 }
 
 QA_MODEL_TABLE = {