From c543d495541f83482c150ee55d0ac5301560ab55 Mon Sep 17 00:00:00 2001
From: Matthew Farrellee <matt@cs.wisc.edu>
Date: Mon, 9 Dec 2024 11:25:42 -0500
Subject: [PATCH] add chat support for nvidia/usdcode-llama-3.1-70b-instruct
 and meta/llama-3.3-70b-instruct

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

diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
index 90ee0e83..916b6d2e 100644
--- 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 = {