forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
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
|