This commit is contained in:
Bagatur 2024-08-29 17:40:25 -07:00
parent cc70f2afe9
commit 3de115e46d

View File

@ -1210,12 +1210,14 @@ def _make_message_chunk_from_anthropic_event(
content_block = event.delta.model_dump() content_block = event.delta.model_dump()
content_block["index"] = event.index content_block["index"] = event.index
content_block["type"] = "tool_use" content_block["type"] = "tool_use"
tool_call_chunk = create_tool_call_chunk(**{ tool_call_chunk = create_tool_call_chunk(
"index": event.index, **{
"id": None, "index": event.index,
"name": None, "id": None,
"args": event.delta.partial_json, "name": None,
}) "args": event.delta.partial_json,
}
)
message_chunk = AIMessageChunk( message_chunk = AIMessageChunk(
content=[content_block], content=[content_block],
tool_call_chunks=[tool_call_chunk], # type: ignore tool_call_chunks=[tool_call_chunk], # type: ignore