mirror of
https://github.com/hwchase17/langchain
synced 2024-10-29 17:07:25 +00:00
10adec5f1b
Already supported in the reverse operation in `_convert_message_to_dict()`, this just provides parity. @hwchase17 @agola11 --------- Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>
26 lines
591 B
Python
26 lines
591 B
Python
"""Test OpenAI Chat API wrapper."""
|
|
|
|
import json
|
|
|
|
from langchain.chat_models.openai import (
|
|
_convert_dict_to_message,
|
|
)
|
|
from langchain.schema import (
|
|
FunctionMessage,
|
|
)
|
|
|
|
|
|
def test_function_message_dict_to_function_message() -> None:
|
|
content = json.dumps({"result": "Example #1"})
|
|
name = "test_function"
|
|
result = _convert_dict_to_message(
|
|
{
|
|
"role": "function",
|
|
"name": name,
|
|
"content": content,
|
|
}
|
|
)
|
|
assert isinstance(result, FunctionMessage)
|
|
assert result.name == name
|
|
assert result.content == content
|