From 7f3a1cc196b309ed05ead72f87f1f7e7d87e1a46 Mon Sep 17 00:00:00 2001
From: Matthew Farrellee <matt@cs.wisc.edu>
Date: Tue, 15 Oct 2024 10:12:45 -0400
Subject: [PATCH] add zyphra/zamba2-7b-instruct to set of supported chat models

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

diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
index 1f5d2604..5a6675a1 100644
--- a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
+++ b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
@@ -454,6 +454,11 @@ def validate_client(self) -> "Model":
         client="ChatNVIDIA",
         supports_structured_output=True,
     ),
+    "zyphra/zamba2-7b-instruct": Model(
+        id="zyphra/zamba2-7b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
 }
 
 QA_MODEL_TABLE = {