This commit is contained in:
Mason Daugherty 2025-07-28 18:18:48 -04:00
parent 492e7af5e2
commit 05e1b95f8e
No known key found for this signature in database

View File

@ -1233,7 +1233,7 @@ def test_tool_call_streaming_different_indices() -> None:
) )
# Merge chunks as happens during streaming # Merge chunks as happens during streaming
merged_chunk = chunk1 + chunk2 merged_chunk: AIMessageChunk = chunk1 + chunk2 # type: ignore[assignment]
# Should result in a single merged tool call chunk # Should result in a single merged tool call chunk
assert len(merged_chunk.tool_call_chunks) == 1 assert len(merged_chunk.tool_call_chunks) == 1
@ -1252,7 +1252,7 @@ def test_tool_call_streaming_different_indices() -> None:
assert tool_call["id"] == "call_123" assert tool_call["id"] == "call_123"
# Test with message_chunk_to_message (core functionality) # Test with message_chunk_to_message (core functionality)
message = message_chunk_to_message(merged_chunk) message: AIMessage = message_chunk_to_message(merged_chunk) # type: ignore[assignment]
assert len(message.tool_calls) == 1 assert len(message.tool_calls) == 1
assert len(message.invalid_tool_calls) == 0 assert len(message.invalid_tool_calls) == 0