github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-01 02:43:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: standardize baichuan init args (#20209)

Browse Source 
Related to https://github.com/langchain-ai/langchain/issues/20085

@baskaryan
This commit is contained in:
Guangdong Liu 2024-04-10 00:00:40 +08:00 committed by GitHub
parent cd7abc495a
commit 97d91ec17c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/community/langchain_community/chat_models/baichuan.py
View File

@ -95,7 +95,7 @@ class ChatBaichuan(BaseChatModel):
"""[DEPRECATED, keeping it for for backward compatibility] Baichuan Secret Key""" """[DEPRECATED, keeping it for for backward compatibility] Baichuan Secret Key"""
streaming: bool = False streaming: bool = False
"""Whether to stream the results or not.""" """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""" """request timeout for chat http requests"""
model = "Baichuan2-Turbo-192K" model = "Baichuan2-Turbo-192K"
"""model name of Baichuan, default is `Baichuan2-Turbo-192K`, """model name of Baichuan, default is `Baichuan2-Turbo-192K`,

17
libs/community/tests/unit_tests/chat_models/test_baichuan.py
View File

@ -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: def test__convert_message_to_dict_human() -> None:
message = HumanMessage(content="foo") message = HumanMessage(content="foo")
result = _convert_message_to_dict(message) result = _convert_message_to_dict(message)