diff --git a/libs/community/langchain_community/chat_models/baidu_qianfan_endpoint.py b/libs/community/langchain_community/chat_models/baidu_qianfan_endpoint.py index 801364c4df..aa80fd7fe0 100644 --- a/libs/community/langchain_community/chat_models/baidu_qianfan_endpoint.py +++ b/libs/community/langchain_community/chat_models/baidu_qianfan_endpoint.py @@ -103,7 +103,7 @@ class QianfanChatEndpoint(BaseChatModel): streaming: Optional[bool] = False """Whether to stream the results or not.""" - request_timeout: Optional[int] = 60 + request_timeout: Optional[int] = Field(60, alias="timeout") """request timeout for chat http requests""" top_p: Optional[float] = 0.8 @@ -125,6 +125,11 @@ class QianfanChatEndpoint(BaseChatModel): endpoint: Optional[str] = None """Endpoint of the Qianfan LLM, required if custom model used.""" + class Config: + """Configuration for this pydantic object.""" + + allow_population_by_field_name = True + @root_validator() def validate_environment(cls, values: Dict) -> Dict: values["qianfan_ak"] = convert_to_secret_str( diff --git a/libs/community/tests/integration_tests/chat_models/test_qianfan_endpoint.py b/libs/community/tests/integration_tests/chat_models/test_qianfan_endpoint.py index afe2811211..407f0cd67e 100644 --- a/libs/community/tests/integration_tests/chat_models/test_qianfan_endpoint.py +++ b/libs/community/tests/integration_tests/chat_models/test_qianfan_endpoint.py @@ -86,6 +86,17 @@ _FUNCTIONS: Any = [ ] +def test_initialization() -> None: + """Test chat model initialization.""" + + for model in [ + QianfanChatEndpoint(model="BLOOMZ-7B", timeout=40), + QianfanChatEndpoint(model="BLOOMZ-7B", request_timeout=40), + ]: + assert model.model == "BLOOMZ-7B" + assert model.request_timeout == 40 + + def test_default_call() -> None: """Test default model(`ERNIE-Bot`) call.""" chat = QianfanChatEndpoint()