Revert "Searching in both _function_calls_context_messages and context messages when updating the result."

This reverts commit fccc91e923.
This commit is contained in:
Filipi Fuchter
2025-11-18 21:38:49 -03:00
parent a60618b0ca
commit 1fcaf3a4bf

View File

@@ -802,12 +802,7 @@ class LLMAssistantAggregator(LLMContextAggregator):
del self._function_calls_in_progress[frame.tool_call_id]
def _update_function_call_result(self, function_name: str, tool_call_id: str, result: Any):
def iter_all():
yield from self._function_calls_context_messages
# In case on long-running function call, the function may already be added to the context
yield from self._context.get_messages()
for message in iter_all():
for message in self._function_calls_context_messages:
if (
not isinstance(message, LLMSpecificMessage)
and message["role"] == "tool"