From baf73fae0e561449c5977295646fedc427740554 Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Mon, 19 Aug 2024 12:30:37 -0400 Subject: [PATCH] add baichuan-inc/baichuan2-13b-chat and thudm/chatglm3-6b to set of supported chat models --- .../langchain_nvidia_ai_endpoints/_statics.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py index 52e29679..173ddae1 100644 --- 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 = {