From 97d91ec17ce5d4154b27743c712adf9cca0e1872 Mon Sep 17 00:00:00 2001 From: Guangdong Liu Date: Wed, 10 Apr 2024 00:00:40 +0800 Subject: [PATCH] community[patch]: standardize baichuan init args (#20209) Related to https://github.com/langchain-ai/langchain/issues/20085 @baskaryan --- .../langchain_community/chat_models/baichuan.py | 2 +- .../unit_tests/chat_models/test_baichuan.py | 17 +++++++++++++++++ 2 files changed, 18 insertions(+), 1 deletion(-) diff --git a/libs/community/langchain_community/chat_models/baichuan.py b/libs/community/langchain_community/chat_models/baichuan.py index 1b3d8ac093f..e4d2fd6908c 100644 --- a/libs/community/langchain_community/chat_models/baichuan.py +++ b/libs/community/langchain_community/chat_models/baichuan.py @@ -95,7 +95,7 @@ class ChatBaichuan(BaseChatModel): """[DEPRECATED, keeping it for for backward compatibility] Baichuan Secret Key""" streaming: bool = False """Whether to stream the results or not.""" - request_timeout: int = 60 + request_timeout: int = Field(default=60, alias="timeout") """request timeout for chat http requests""" model = "Baichuan2-Turbo-192K" """model name of Baichuan, default is `Baichuan2-Turbo-192K`, diff --git a/libs/community/tests/unit_tests/chat_models/test_baichuan.py b/libs/community/tests/unit_tests/chat_models/test_baichuan.py index f027cbb9c5d..8a3ba346425 100644 --- a/libs/community/tests/unit_tests/chat_models/test_baichuan.py +++ b/libs/community/tests/unit_tests/chat_models/test_baichuan.py @@ -21,6 +21,23 @@ from langchain_community.chat_models.baichuan import ( ) +def test_initialization() -> None: + """Test chat model initialization.""" + + for model in [ + ChatBaichuan( + model="Baichuan2-Turbo-192K", baichuan_api_key="test-api-key", timeout=40 + ), + ChatBaichuan( + model="Baichuan2-Turbo-192K", + baichuan_api_key="test-api-key", + request_timeout=40, + ), + ]: + assert model.model == "Baichuan2-Turbo-192K" + assert model.request_timeout == 40 + + def test__convert_message_to_dict_human() -> None: message = HumanMessage(content="foo") result = _convert_message_to_dict(message)