add ibm/granite-3.0-3b-a800m-instruct and ibm/granite-3.0-8b-instruct…

… to set of supported chat models
langchain-ai · Oct 22, 2024 · 2113dac · 2113dac
1 parent 28a1584
commit 2113dac
Showing 1 changed file with 10 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
@@ -459,6 +459,16 @@ def validate_client(self) -> "Model":
         model_type="chat",
         client="ChatNVIDIA",
     ),
+    "ibm/granite-3.0-8b-instruct": Model(
+        id="ibm/granite-3.0-8b-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
+    "ibm/granite-3.0-3b-a800m-instruct": Model(
+        id="ibm/granite-3.0-3b-a800m-instruct",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
 }
 
 QA_MODEL_TABLE = {