mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-14 15:16:21 +00:00
6cd452d985
Description:
- Add system templates and user templates in integration testing
- initialize the response id field value to request_id
- Adjust the default model to hunyuan-pro
- Remove the default values of Temperature and TopP
- Add SystemMessage
all the integration tests have passed.
1、Execute integration tests for the first time
<img width="1359" alt="71ca77a2-e9be-4af6-acdc-4d665002bd9b"
src="https://github.com/user-attachments/assets/9298dc3a-aa26-4bfa-968b-c011a4e699c9">
2、Run the integration test a second time
<img width="1501" alt="image"
src="https://github.com/user-attachments/assets/61335416-4a67-4840-bb89-090ba668e237">
Issue: None
Dependencies: None
Twitter handle: None
---------
Co-authored-by: Chester Curme <chester.curme@gmail.com>
87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
import pytest
|
|
from langchain_core.messages import (
|
|
AIMessage,
|
|
AIMessageChunk,
|
|
ChatMessage,
|
|
FunctionMessage,
|
|
HumanMessage,
|
|
HumanMessageChunk,
|
|
SystemMessage,
|
|
)
|
|
|
|
from langchain_community.chat_models.hunyuan import (
|
|
_convert_delta_to_message_chunk,
|
|
_convert_dict_to_message,
|
|
_convert_message_to_dict,
|
|
)
|
|
|
|
|
|
def test__convert_message_to_dict_human() -> None:
|
|
message = HumanMessage(content="foo")
|
|
result = _convert_message_to_dict(message)
|
|
expected_output = {"Role": "user", "Content": "foo"}
|
|
assert result == expected_output
|
|
|
|
|
|
def test__convert_message_to_dict_ai() -> None:
|
|
message = AIMessage(content="foo")
|
|
result = _convert_message_to_dict(message)
|
|
expected_output = {"Role": "assistant", "Content": "foo"}
|
|
assert result == expected_output
|
|
|
|
|
|
def test__convert_message_to_dict_system() -> None:
|
|
message = SystemMessage(content="foo")
|
|
result = _convert_message_to_dict(message)
|
|
expected_output = {"Role": "system", "Content": "foo"}
|
|
assert result == expected_output
|
|
|
|
|
|
def test__convert_message_to_dict_function() -> None:
|
|
message = FunctionMessage(name="foo", content="bar")
|
|
with pytest.raises(TypeError) as e:
|
|
_convert_message_to_dict(message)
|
|
assert "Got unknown type" in str(e)
|
|
|
|
|
|
def test__convert_dict_to_message_human() -> None:
|
|
message_dict = {"Role": "user", "Content": "foo"}
|
|
result = _convert_dict_to_message(message_dict)
|
|
expected_output = HumanMessage(content="foo")
|
|
assert result == expected_output
|
|
|
|
|
|
def test__convert_dict_to_message_ai() -> None:
|
|
message_dict = {"Role": "assistant", "Content": "foo"}
|
|
result = _convert_dict_to_message(message_dict)
|
|
expected_output = AIMessage(content="foo")
|
|
assert result == expected_output
|
|
|
|
|
|
def test__convert_dict_to_message_system() -> None:
|
|
message_dict = {"Role": "system", "Content": "foo"}
|
|
result = _convert_dict_to_message(message_dict)
|
|
expected_output = SystemMessage(content="foo")
|
|
assert result == expected_output
|
|
|
|
|
|
def test__convert_dict_to_message_other_role() -> None:
|
|
message_dict = {"Role": "other", "Content": "foo"}
|
|
result = _convert_dict_to_message(message_dict)
|
|
expected_output = ChatMessage(role="other", content="foo")
|
|
assert result == expected_output
|
|
|
|
|
|
def test__convert_delta_to_message_assistant() -> None:
|
|
delta = {"Role": "assistant", "Content": "foo"}
|
|
result = _convert_delta_to_message_chunk(delta, AIMessageChunk)
|
|
expected_output = AIMessageChunk(content="foo")
|
|
assert result == expected_output
|
|
|
|
|
|
def test__convert_delta_to_message_human() -> None:
|
|
delta = {"Role": "user", "Content": "foo"}
|
|
result = _convert_delta_to_message_chunk(delta, HumanMessageChunk)
|
|
expected_output = HumanMessageChunk(content="foo")
|
|
assert result == expected_output
|