mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-20 18:07:05 +00:00
096b66db4a
Description: The old method will be discontinued; use the official SDK for more model options. Issue: None Dependencies: None Twitter handle: None Co-authored-by: trumanyan <trumanyan@tencent.com>
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
import pytest
|
|
from langchain_core.messages import AIMessage, HumanMessage
|
|
|
|
from langchain_community.chat_models.hunyuan import ChatHunyuan
|
|
|
|
|
|
@pytest.mark.requires("tencentcloud-sdk-python")
|
|
def test_chat_hunyuan() -> None:
|
|
chat = ChatHunyuan()
|
|
message = HumanMessage(content="Hello")
|
|
response = chat.invoke([message])
|
|
assert isinstance(response, AIMessage)
|
|
assert isinstance(response.content, str)
|
|
|
|
|
|
@pytest.mark.requires("tencentcloud-sdk-python")
|
|
def test_chat_hunyuan_with_temperature() -> None:
|
|
chat = ChatHunyuan(temperature=0.6)
|
|
message = HumanMessage(content="Hello")
|
|
response = chat.invoke([message])
|
|
assert isinstance(response, AIMessage)
|
|
assert isinstance(response.content, str)
|
|
|
|
|
|
@pytest.mark.requires("tencentcloud-sdk-python")
|
|
def test_chat_hunyuan_with_model_name() -> None:
|
|
chat = ChatHunyuan(model="hunyuan-standard")
|
|
message = HumanMessage(content="Hello")
|
|
response = chat.invoke([message])
|
|
assert isinstance(response, AIMessage)
|
|
assert isinstance(response.content, str)
|
|
|
|
|
|
@pytest.mark.requires("tencentcloud-sdk-python")
|
|
def test_chat_hunyuan_with_stream() -> None:
|
|
chat = ChatHunyuan(streaming=True)
|
|
message = HumanMessage(content="Hello")
|
|
response = chat.invoke([message])
|
|
assert isinstance(response, AIMessage)
|
|
assert isinstance(response.content, str)
|
|
|
|
|
|
def test_extra_kwargs() -> None:
|
|
chat = ChatHunyuan(temperature=0.88, top_p=0.7)
|
|
assert chat.temperature == 0.88
|
|
assert chat.top_p == 0.7
|