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.
langchain/libs/core/tests/unit_tests/messages/test_imports.py

30 lines
623 B
Python

from langchain_core.messages import __all__
EXPECTED_ALL = [
"AIMessage",
"AIMessageChunk",
"AnyMessage",
"BaseMessage",
"BaseMessageChunk",
"ChatMessage",
"ChatMessageChunk",
"FunctionMessage",
"FunctionMessageChunk",
"HumanMessage",
"HumanMessageChunk",
"SystemMessage",
"SystemMessageChunk",
"ToolMessage",
"ToolMessageChunk",
"get_buffer_string",
"message_chunk_to_message",
"messages_from_dict",
"messages_to_dict",
"message_to_dict",
"merge_content",
]
def test_all_imports() -> None:
assert set(__all__) == set(EXPECTED_ALL)