github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-16 06:53:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

baichuan[patch]:standardize chat init args (#21298)

Browse Source 
Thank you for contributing to LangChain!

community:baichuan[patch]: standardize init args

updated `baichuan_api_key` so that aliased to `api_key`. Added test that
it continues to set the same underlying attribute. Test checks for
`SecretStr`

updated `temperature` with Pydantic Field, added unit test. 

Related to https://github.com/langchain-ai/langchain/issues/20085
This commit is contained in:
Param Singh
2024-05-06 11:33:57 -07:00
committed by GitHub
parent 62559b20b3
commit fee91d43b7
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -25,9 +25,7 @@ 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", api_key="test-api-key", timeout=40),
ChatBaichuan(
model="Baichuan2-Turbo-192K",
baichuan_api_key="test-api-key",
@@ -35,7 +33,9 @@ def test_initialization() -> None:
),
]:
assert model.model == "Baichuan2-Turbo-192K"
assert isinstance(model.baichuan_api_key, SecretStr)
assert model.request_timeout == 40
assert model.temperature == 0.3
def test__convert_message_to_dict_human() -> None: