add baichuan-inc/baichuan2-13b-chat and thudm/chatglm3-6b to set of s…

…upported chat models
langchain-ai · Aug 19, 2024 · baf73fa · baf73fa
1 parent 8f4deaa
commit baf73fa
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
@@ -354,6 +354,16 @@ def validate_client(cls, client: str, values: dict) -> str:
         model_type="chat",
         client="ChatNVIDIA",
     ),
+    "baichuan-inc/baichuan2-13b-chat": Model(
+        id="baichuan-inc/baichuan2-13b-chat",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
+    "thudm/chatglm3-6b": Model(
+        id="thudm/chatglm3-6b",
+        model_type="chat",
+        client="ChatNVIDIA",
+    ),
 }
 
 QA_MODEL_TABLE = {