github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-10-24 20:20:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(core): propagate extras when aggregating tool calls in v1 content (#33494)

Browse Source
This commit is contained in:
ccurme
2025-10-15 10:38:16 -04:00
committed by GitHub
parent 53e9f00804
commit 4da5a8081f
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
libs/core/langchain_core/messages/ai.py
View File

@@ -545,6 +545,9 @@ class AIMessageChunk(AIMessage, BaseMessageChunk):
and call_id in id_to_tc and call_id in id_to_tc
): ):
self.content[idx] = cast("dict[str, Any]", id_to_tc[call_id]) self.content[idx] = cast("dict[str, Any]", id_to_tc[call_id])
if "extras" in block:
# mypy does not account for instance check for dict above
self.content[idx]["extras"] = block["extras"] # type: ignore[index]
return self return self

3
libs/core/tests/unit_tests/messages/test_ai.py
View File

@@ -358,6 +358,8 @@ def test_content_blocks() -> None:
# test v1 content # test v1 content
chunk_1.content = cast("str | list[str | dict]", chunk_1.content_blocks) chunk_1.content = cast("str | list[str | dict]", chunk_1.content_blocks)
assert len(chunk_1.content) == 1
chunk_1.content[0]["extras"] = {"baz": "qux"} # type: ignore[index]
chunk_1.response_metadata["output_version"] = "v1" chunk_1.response_metadata["output_version"] = "v1"
chunk_2.content = cast("str | list[str | dict]", chunk_2.content_blocks) chunk_2.content = cast("str | list[str | dict]", chunk_2.content_blocks)
@@ -368,6 +370,7 @@ def test_content_blocks() -> None:
"name": "foo", "name": "foo",
"args": {"foo": "bar"}, "args": {"foo": "bar"},
"id": "abc_123", "id": "abc_123",
"extras": {"baz": "qux"},
} }
] ]