github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 06:39:52 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/core/langchain_core/messages/utils.py
View File

@ -556,6 +556,8 @@ def merge_message_runs(
else:
last_chunk = _msg_to_chunk(last)
curr_chunk = _msg_to_chunk(curr)
if curr_chunk.response_metadata:
curr_chunk.response_metadata.clear()
if (
isinstance(last_chunk.content, str)
and isinstance(curr_chunk.content, str)

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"),