From 9083688f152f6c0f2734432098a1963790c08173 Mon Sep 17 00:00:00 2001
From: Matthew Farrellee <matt@cs.wisc.edu>
Date: Fri, 9 Aug 2024 12:55:56 -0400
Subject: [PATCH] add mistralai/mistral-large-2-instruct to set of supported
 chat + tool + structured output models

---
 .../ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
index 2aa64387..e9b280e4 100644
--- a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
+++ b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
@@ -332,6 +332,13 @@ def validate_client(cls, client: str, values: dict) -> str:
         model_type="chat",
         client="ChatNVIDIA",
     ),
+    "mistralai/mistral-large-2-instruct": Model(
+        id="mistralai/mistral-large-2-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+        supports_tools=True,
+        supports_structured_output=True,
+    ),
 }
 
 QA_MODEL_TABLE = {