docs: Standardize QianfanLLMEndpoint LLM (#25139)

- **Description:** Standardize QianfanLLMEndpoint LLM,include:
  - docs, the issue #24803 
  - model init arg names, the issue #20085
This commit is contained in:
maang-h
2024-08-07 22:57:27 +08:00
committed by GitHub
parent 28e0958ff4
commit 0ba125c3cd
2 changed files with 109 additions and 16 deletions

View File

@@ -1,8 +1,9 @@
"""Test Baidu Qianfan LLM Endpoint."""
from typing import Generator
from typing import Generator, cast
from langchain_core.outputs import LLMResult
from langchain_core.pydantic_v1 import SecretStr
from langchain_community.llms.baidu_qianfan_endpoint import QianfanLLMEndpoint
@@ -42,3 +43,15 @@ def test_rate_limit() -> None:
output = llm.generate(["write a joke"])
assert isinstance(output, LLMResult)
assert isinstance(output.generations, list)
def test_qianfan_with_param_alias() -> None:
"""Test with qianfan llm parameter alias."""
llm = QianfanLLMEndpoint( # type: ignore[call-arg]
api_key="your-api-key", # type: ignore[arg-type]
secret_key="your-secret-key", # type: ignore[arg-type]
timeout=50,
) # type: ignore[call-arg]
assert cast(SecretStr, llm.qianfan_ak).get_secret_value() == "your-api-key"
assert cast(SecretStr, llm.qianfan_sk).get_secret_value() == "your-secret-key"
assert llm.request_timeout == 50