github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-12 21:11:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: standardize init args (#20210)

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

@baskaryan
This commit is contained in:
Guangdong Liu
2024-04-17 09:29:57 +08:00
committed by GitHub
parent a7c5e41443
commit 3729bec1a2
2 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
libs/community/tests/integration_tests/chat_models/test_tongyi.py
View File

@@ -1,4 +1,5 @@
"""Test Alibaba Tongyi Chat Model."""
from typing import cast
from langchain_core.callbacks import CallbackManager
from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
@@ -10,6 +11,16 @@ from langchain_community.chat_models.tongyi import ChatTongyi
from tests.unit_tests.callbacks.fake_callback_handler import FakeCallbackHandler
def test_initialization() -> None:
"""Test chat model initialization."""
for model in [
ChatTongyi(model_name="qwen-turbo", api_key="xyz"),
ChatTongyi(model="qwen-turbo", dashscope_api_key="xyz"),
]:
assert model.model_name == "qwen-turbo"
assert cast(SecretStr, model.dashscope_api_key).get_secret_value() == "xyz"
def test_api_key_is_string() -> None:
llm = ChatTongyi(dashscope_api_key="secret-api-key")
assert isinstance(llm.dashscope_api_key, SecretStr)