grasp_agents 0.5.2__tar.gz → 0.5.3__tar.gz

{grasp_agents-0.5.2 → grasp_agents-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: grasp_agents
-Version: 0.5.2
+Version: 0.5.3
 Summary: Grasp Agents Library
 License-File: LICENSE.md
 Requires-Python: <4,>=3.11.4

{grasp_agents-0.5.2 → grasp_agents-0.5.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "grasp_agents"
-version = "0.5.2"
+version = "0.5.3"
 description = "Grasp Agents Library"
 readme = "README.md"
 requires-python = ">=3.11.4,<4"

{grasp_agents-0.5.2 → grasp_agents-0.5.3}/src/grasp_agents/workflow/looped_workflow.py

@@ -151,36 +151,37 @@ class LoopedWorkflow(
         num_iterations = 0
         exit_packet: Packet[OutT_co] | None = None
 
-        for subproc in self.subprocs:
-            async for event in subproc.run_stream(
-                chat_inputs=chat_inputs,
-                in_packet=packet,
-                in_args=in_args,
-                forgetful=forgetful,
-                call_id=f"{call_id}/{subproc.name}",
-                ctx=ctx,
-            ):
-                if isinstance(event, ProcPacketOutputEvent):
-                    packet = event.data
-                yield event
-
-            if subproc is self._end_proc:
-                num_iterations += 1
-                exit_packet = cast("Packet[OutT_co]", packet)
-                if self._exit_workflow_loop(exit_packet, ctx=ctx):
-                    yield WorkflowResultEvent(
-                        data=exit_packet, proc_name=self.name, call_id=call_id
-                    )
-                    return
-                if num_iterations >= self._max_iterations:
-                    logger.info(
-                        f"Max iterations reached ({self._max_iterations}). "
-                        "Exiting loop."
-                    )
-                    yield WorkflowResultEvent(
-                        data=exit_packet, proc_name=self.name, call_id=call_id
-                    )
-                    return
-
-            chat_inputs = None
-            in_args = None
+        while True:
+            for subproc in self.subprocs:
+                async for event in subproc.run_stream(
+                    chat_inputs=chat_inputs,
+                    in_packet=packet,
+                    in_args=in_args,
+                    forgetful=forgetful,
+                    call_id=f"{call_id}/{subproc.name}",
+                    ctx=ctx,
+                ):
+                    if isinstance(event, ProcPacketOutputEvent):
+                        packet = event.data
+                    yield event
+
+                if subproc is self._end_proc:
+                    num_iterations += 1
+                    exit_packet = cast("Packet[OutT_co]", packet)
+                    if self._exit_workflow_loop(exit_packet, ctx=ctx):
+                        yield WorkflowResultEvent(
+                            data=exit_packet, proc_name=self.name, call_id=call_id
+                        )
+                        return
+                    if num_iterations >= self._max_iterations:
+                        logger.info(
+                            f"Max iterations reached ({self._max_iterations}). "
+                            "Exiting loop."
+                        )
+                        yield WorkflowResultEvent(
+                            data=exit_packet, proc_name=self.name, call_id=call_id
+                        )
+                        return
+
+                chat_inputs = None
+                in_args = None