mirror of
https://github.com/hwchase17/langchain.git
synced 2026-07-13 20:47:15 +00:00
24 lines
591 B
Python
24 lines
591 B
Python
"""Test OpenAI Chat API wrapper."""
|
|
|
|
import json
|
|
|
|
from langchain.chat_models.openai import (
|
|
_convert_dict_to_message,
|
|
)
|
|
from langchain.schema.messages 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
|