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

core[minor]: update conversion utils to handle RemoveMessage (#23840)

Browse Source
This commit is contained in:
Vadym Barda
2024-07-03 16:13:31 -04:00
committed by GitHub
parent 4ab78572e7
commit 9bb623381b
2 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -12,6 +12,7 @@ from langchain_core.messages import (
FunctionMessageChunk,
HumanMessage,
HumanMessageChunk,
RemoveMessage,
SystemMessage,
ToolCall,
ToolCallChunk,
@@ -649,6 +650,7 @@ def test_convert_to_messages() -> None:
],
},
{"role": "tool", "tool_call_id": "tool_id", "content": "Hi!"},
{"role": "remove", "id": "message_to_remove", "content": ""},
]
) == [
SystemMessage(content="You are a helpful assistant."),
@@ -668,6 +670,7 @@ def test_convert_to_messages() -> None:
tool_calls=[ToolCall(name="greet", args={"name": "Jane"}, id="tool_id")],
),
ToolMessage(tool_call_id="tool_id", content="Hi!"),
RemoveMessage(id="message_to_remove"),
]
# tuples