refactor(workflow_agent): iterate over AsyncGenerator in workflow execution

2025-05-13 18:18:06 -03:00 · 2025-05-13 18:18:06 -03:00 · 86de80a998
commit 86de80a998
parent 2bac2b3824
1 changed files with 3 additions and 1 deletions
--- 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)