github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-03 20:16:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes for opengpts release (#13960)

Browse Source
This commit is contained in:
Nuno Campos
2023-11-28 21:49:43 +00:00
committed by GitHub
parent 947daaf833
commit 970fe23feb
7 changed files with 123 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
libs/core/tests/unit_tests/messages/test_imports.py
View File

@@ -17,6 +17,7 @@ EXPECTED_ALL = [
"ToolMessage",
"ToolMessageChunk",
"get_buffer_string",
"message_chunk_to_message",
"messages_from_dict",
"messages_to_dict",
"message_to_dict",

16
libs/core/tests/unit_tests/test_messages.py
View File

@@ -14,6 +14,7 @@ from langchain_core.messages import (
SystemMessage,
ToolMessage,
get_buffer_string,
message_chunk_to_message,
messages_from_dict,
messages_to_dict,
)
@@ -184,3 +185,18 @@ def test_multiple_msg() -> None:
sys_msg,
]
assert messages_from_dict(messages_to_dict(msgs)) == msgs
def test_message_chunk_to_message() -> None:
assert message_chunk_to_message(
AIMessageChunk(content="I am", additional_kwargs={"foo": "bar"})
) == AIMessage(content="I am", additional_kwargs={"foo": "bar"})
assert message_chunk_to_message(HumanMessageChunk(content="I am")) == HumanMessage(
content="I am"
)
assert message_chunk_to_message(
ChatMessageChunk(role="User", content="I am")
) == ChatMessage(role="User", content="I am")
assert message_chunk_to_message(
FunctionMessageChunk(name="hello", content="I am")
) == FunctionMessage(name="hello", content="I am")