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

fix: Handle response metadata in merge_messages_runs (#28453)

Browse Source
This commit is contained in:
William FH
2024-12-02 13:56:23 -08:00
committed by GitHub
parent 60021e54b5
commit ecee41ab72
2 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
libs/core/tests/unit_tests/messages/test_utils.py
View File

@@ -59,6 +59,24 @@ def test_merge_message_runs_str_without_separator(
assert messages == messages_model_copy
def test_merge_message_runs_response_metadata() -> None:
messages = [
AIMessage("foo", id="1", response_metadata={"input_tokens": 1}),
AIMessage("bar", id="2", response_metadata={"input_tokens": 2}),
]
expected = [
AIMessage(
"foo\nbar",
id="1",
response_metadata={"input_tokens": 1},
)
]
actual = merge_message_runs(messages)
assert actual == expected
# Check it's not mutated
assert messages[1].response_metadata == {"input_tokens": 2}
def test_merge_message_runs_content() -> None:
messages = [
AIMessage("foo", id="1"),