Merge pull request #122 from mattf/main

add chat support for nvidia/usdcode-llama-3.1-70b-instruct and meta/llama-3.3-70b-instruct
langchain-ai · Dec 14, 2024 · 218f822 · 218f822
2 parents 440153a + 6a90668
commit 218f822
Showing 1 changed file with 11 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
@@ -481,6 +481,17 @@ def validate_client(self) -> "Model":
         client="ChatNVIDIA",
         supports_structured_output=True,
     ),
+    "nvidia/usdcode-llama-3.1-70b-instruct": Model(
+        id="nvidia/usdcode-llama-3.1-70b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
+    "meta/llama-3.3-70b-instruct": Model(
+        id="meta/llama-3.3-70b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+        supports_structured_output=True,
+    ),
 }
 
 QA_MODEL_TABLE = {