|
|
|
@ -9,7 +9,12 @@ from langchain.chat_models.openai import (
|
|
|
|
|
ChatOpenAI,
|
|
|
|
|
_convert_dict_to_message,
|
|
|
|
|
)
|
|
|
|
|
from langchain.schema.messages import FunctionMessage
|
|
|
|
|
from langchain.schema.messages import (
|
|
|
|
|
AIMessage,
|
|
|
|
|
FunctionMessage,
|
|
|
|
|
HumanMessage,
|
|
|
|
|
SystemMessage,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_function_message_dict_to_function_message() -> None:
|
|
|
|
@ -27,6 +32,27 @@ def test_function_message_dict_to_function_message() -> None:
|
|
|
|
|
assert result.content == content
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test__convert_dict_to_message_human() -> None:
|
|
|
|
|
message = {"role": "user", "content": "foo"}
|
|
|
|
|
result = _convert_dict_to_message(message)
|
|
|
|
|
expected_output = HumanMessage(content="foo")
|
|
|
|
|
assert result == expected_output
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test__convert_dict_to_message_ai() -> None:
|
|
|
|
|
message = {"role": "assistant", "content": "foo"}
|
|
|
|
|
result = _convert_dict_to_message(message)
|
|
|
|
|
expected_output = AIMessage(content="foo")
|
|
|
|
|
assert result == expected_output
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test__convert_dict_to_message_system() -> None:
|
|
|
|
|
message = {"role": "system", "content": "foo"}
|
|
|
|
|
result = _convert_dict_to_message(message)
|
|
|
|
|
expected_output = SystemMessage(content="foo")
|
|
|
|
|
assert result == expected_output
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.fixture
|
|
|
|
|
def mock_completion() -> dict:
|
|
|
|
|
return {
|
|
|
|
|