From c7c47f81c4fee7ae6cfeaf39624100ff717c27d4 Mon Sep 17 00:00:00 2001 From: Mason Daugherty Date: Tue, 5 Aug 2025 00:04:30 -0400 Subject: [PATCH] add type hints to messages --- libs/core/langchain_core/messages/v1.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/libs/core/langchain_core/messages/v1.py b/libs/core/langchain_core/messages/v1.py index f1bc0fa2523..87d4bdff1a3 100644 --- a/libs/core/langchain_core/messages/v1.py +++ b/libs/core/langchain_core/messages/v1.py @@ -323,8 +323,8 @@ class AIMessageChunk(AIMessage): """Get the tool calls made by the AI.""" if not self._tool_calls: parsed_content = _init_tool_calls(self.content) - tool_calls = [] - invalid_tool_calls = [] + tool_calls: list[types.ToolCall] = [] + invalid_tool_calls: list[types.InvalidToolCall] = [] for block in parsed_content: if types.is_tool_call_block(block): tool_calls.append(block) @@ -344,8 +344,8 @@ class AIMessageChunk(AIMessage): """Get the invalid tool calls made by the AI.""" if not self._invalid_tool_calls: parsed_content = _init_tool_calls(self.content) - tool_calls = [] - invalid_tool_calls = [] + tool_calls: list[types.ToolCall] = [] + invalid_tool_calls: list[types.InvalidToolCall] = [] for block in parsed_content: if types.is_tool_call_block(block): tool_calls.append(block)