github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-15 07:36:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

add type hints to messages

Browse Source
This commit is contained in:
Mason Daugherty 2025-08-05 00:04:30 -04:00
parent 5c9ce7fd2b
commit c7c47f81c4
No known key found for this signature in database
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/core/langchain_core/messages/v1.py
View File

@ -323,8 +323,8 @@ class AIMessageChunk(AIMessage):
"""Get the tool calls made by the AI.""" """Get the tool calls made by the AI."""
if not self._tool_calls: if not self._tool_calls:
parsed_content = _init_tool_calls(self.content) parsed_content = _init_tool_calls(self.content)
tool_calls = [] tool_calls: list[types.ToolCall] = []
invalid_tool_calls = [] invalid_tool_calls: list[types.InvalidToolCall] = []
for block in parsed_content: for block in parsed_content:
if types.is_tool_call_block(block): if types.is_tool_call_block(block):
tool_calls.append(block) tool_calls.append(block)
@ -344,8 +344,8 @@ class AIMessageChunk(AIMessage):
"""Get the invalid tool calls made by the AI.""" """Get the invalid tool calls made by the AI."""
if not self._invalid_tool_calls: if not self._invalid_tool_calls:
parsed_content = _init_tool_calls(self.content) parsed_content = _init_tool_calls(self.content)
tool_calls = [] tool_calls: list[types.ToolCall] = []
invalid_tool_calls = [] invalid_tool_calls: list[types.InvalidToolCall] = []
for block in parsed_content: for block in parsed_content:
if types.is_tool_call_block(block): if types.is_tool_call_block(block):
tool_calls.append(block) tool_calls.append(block)