From 86de80a998d3a4e75fd6f7f4ca27f0497cba9515 Mon Sep 17 00:00:00 2001 From: Davidson Gomes Date: Tue, 13 May 2025 18:18:06 -0300 Subject: [PATCH] refactor(workflow_agent): iterate over AsyncGenerator in workflow execution --- src/services/workflow_agent.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/services/workflow_agent.py b/src/services/workflow_agent.py index f9ad44e7..97c99ba4 100644 --- a/src/services/workflow_agent.py +++ b/src/services/workflow_agent.py @@ -788,7 +788,9 @@ class WorkflowAgent(BaseAgent): print("\nšŸš€ Starting workflow execution:") print(f"Initial content: {user_message[:100]}...") - await self._execute_workflow(ctx, graph, initial_state) + # Iterar sobre o AsyncGenerator em vez de usar await + async for event in self._execute_workflow(ctx, graph, initial_state): + yield event except Exception as e: yield await self._handle_workflow_error(e)