community[patch]:sparkllm standardize init args (#20194)

Related to https://github.com/langchain-ai/langchain/issues/20085
@baskaryan
This commit is contained in:
Guangdong Liu
2024-04-14 07:03:19 +08:00
committed by GitHub
parent 7d7a08e458
commit 4be7ca7b4c
2 changed files with 15 additions and 1 deletions

View File

@@ -3,6 +3,15 @@ from langchain_core.messages import AIMessage, AIMessageChunk, HumanMessage
from langchain_community.chat_models.sparkllm import ChatSparkLLM
def test_initialization() -> None:
"""Test chat model initialization."""
for model in [
ChatSparkLLM(timeout=30),
ChatSparkLLM(request_timeout=30),
]:
assert model.request_timeout == 30
def test_chat_spark_llm() -> None:
chat = ChatSparkLLM()
message = HumanMessage(content="Hello")