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

This commit is contained in:
Filipi Fuchter
2025-11-18 11:50:28 -03:00
parent a510b276e6
commit fccc91e923

View File

@@ -805,7 +805,12 @@ 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):
for message in self._function_calls_context_messages:
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():
if (
not isinstance(message, LLMSpecificMessage)
and message["role"] == "tool"