nas/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-03-20 03:57:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Final refinements: fix falsy value handling and add warning for extra ToolMessages

Browse Source 
Co-authored-by: jxxghp <51039935+jxxghp@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-01-24 00:10:00 +00:00
parent 712995dcf3
commit 54422b5181
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/agent/__init__.py
View File

@@ -219,7 +219,10 @@ class MoviePilotAgent:
# Check if this is an AIMessage with tool_calls
if isinstance(msg, AIMessage) and getattr(msg, 'tool_calls', None):
# Extract tool_call IDs (ToolCall is a TypedDict, so it's a dict at runtime)
tool_call_ids = {tc.get('id') or tc.get('name') for tc in msg.tool_calls}
tool_call_ids = {
tc.get('id') if tc.get('id') is not None else tc.get('name')
for tc in msg.tool_calls
}
# Find subsequent ToolMessages
j = i + 1
@@ -231,6 +234,11 @@ class MoviePilotAgent:
# Check if all tool_calls have corresponding ToolMessages
found_tool_call_ids = {tm.tool_call_id for tm in found_tool_messages}
# Warn if there are extra ToolMessages that don't correspond to any tool_call
extra_tool_messages = found_tool_call_ids - tool_call_ids
if extra_tool_messages:
logger.warning(f"Found extra ToolMessages that don't correspond to any tool_call: {extra_tool_messages}")
if not tool_call_ids.issubset(found_tool_call_ids):
# Missing some tool_call responses, skip this AIMessage and all its ToolMessages
logger.warning("Removing incomplete tool_call AIMessage and its partial ToolMessages: missing tool_call responses")