loom-core 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

loom/core/context.py

@@ -82,6 +82,19 @@ class WorkflowContext(Generic[InputT, StateT]):
         self.cursor += 1
         return event
 
+    def _skip_step_events(self) -> None:
+        """Skip over STEP_START and STEP_END events during replay.
+        
+        These are internal workflow management events that don't affect
+        the deterministic execution logic.
+        """
+        while True:
+            event = self._peek()
+            if event and event["type"] in ("STEP_START", "STEP_END"):
+                self._consume()
+            else:
+                break
+
     def _match_event(self, expected_type: str) -> Event | None:
         """
         Safe helper to check if the NEXT event matches what we expect.
@@ -102,6 +115,15 @@ class WorkflowContext(Generic[InputT, StateT]):
         """
         return self.cursor < len(self.history)
 
+    @property
+    def is_at_end_of_history(self) -> bool:
+        """Check if we've consumed all events in history.
+
+        Returns:
+            True if we're at the end of event history, False otherwise
+        """
+        return self.cursor >= len(self.history)
+
     def _extract_activity_metadata[FuncReturn](
         self, fn: Callable[..., Awaitable[FuncReturn]], args: tuple[Any, ...]
     ) -> ActivityMetadata:
@@ -136,6 +158,8 @@ class WorkflowContext(Generic[InputT, StateT]):
     ) -> FuncReturn:
         metadata = self._extract_activity_metadata(fn, args)
 
+        # Skip any step events first
+        self._skip_step_events()
         scheduled_event = self._match_event("ACTIVITY_SCHEDULED")
 
         if scheduled_event:
@@ -147,6 +171,8 @@ class WorkflowContext(Generic[InputT, StateT]):
 
             self._consume()
 
+            # Skip step events before checking for completion
+            self._skip_step_events()
             completed_event = self._match_event("ACTIVITY_COMPLETED")
 
             if completed_event:
@@ -155,6 +181,8 @@ class WorkflowContext(Generic[InputT, StateT]):
 
             raise StopReplay
 
+        # Skip step events before checking for unexpected events
+        self._skip_step_events()
         unexpected_event = self._peek()
         if unexpected_event:
             raise NonDeterministicWorkflowError(
@@ -180,11 +208,16 @@ class WorkflowContext(Generic[InputT, StateT]):
             datetime.datetime.now(datetime.timezone.utc) + delta if delta else until  # type: ignore
         )
 
+        # Skip any step events first
+        self._skip_step_events()
+        
         scheduled_event = self._match_event("TIMER_SCHEDULED")
 
         if scheduled_event:
             self._consume()
 
+            # Skip step events before checking for timer fired
+            self._skip_step_events()
             fired_event = self._match_event("TIMER_FIRED")
             if fired_event:
                 self._consume()
@@ -192,6 +225,8 @@ class WorkflowContext(Generic[InputT, StateT]):
 
             raise StopReplay
 
+        # Skip step events before checking for unexpected events
+        self._skip_step_events()
         unexpected_event = self._peek()
         if unexpected_event:
             raise NonDeterministicWorkflowError(
@@ -210,6 +245,8 @@ class WorkflowContext(Generic[InputT, StateT]):
         If the signal is already in history (replay), it returns the data immediately.
         If not, it raises StopReplay to suspend execution until the signal arrives.
         """
+        # Skip any step events first
+        self._skip_step_events()
         # 1. Check if the signal is next in history
         event = self._match_event("SIGNAL_RECEIVED")
 
@@ -226,6 +263,8 @@ class WorkflowContext(Generic[InputT, StateT]):
             self._consume()
             return event["payload"]["data"]
 
+        # Skip step events before checking for unexpected events
+        self._skip_step_events()
         unexpected_event = self._peek()
         if unexpected_event:
             raise NonDeterministicWorkflowError(

loom/core/engine.py

@@ -93,9 +93,42 @@ class Engine(Generic[InputT, StateT]):
 
         try:
             for step in steps:
+                # Check for and consume STEP_START event during replay
+                step_start_event = ctx._match_event("STEP_START")
+                if step_start_event:
+                    if step_start_event["payload"]["step_name"] != step["name"]:
+                        raise StopReplay  # Step mismatch, something changed
+                    ctx._consume()
+                else:
+                    # Not replaying, emit STEP_START event
+                    if not ctx.is_replaying:
+                        await ctx._append_event(
+                            "STEP_START",
+                            {
+                                "step_name": step["name"],
+                                "step_fn": step["fn"],
+                                "started_at": datetime.now(timezone.utc).isoformat(),
+                            },
+                        )
+
                 step_fn = getattr(workflow, step["fn"])
                 await step_fn(ctx)
 
+                # Check for and consume STEP_END event during replay
+                step_end_event = ctx._match_event("STEP_END")
+                if step_end_event:
+                    ctx._consume()
+                else:
+                    # Not replaying, emit STEP_END event
+                    if not ctx.is_replaying:
+                        await ctx._append_event(
+                            "STEP_END",
+                            {
+                                "step_name": step["name"],
+                                "completed_at": datetime.now(timezone.utc).isoformat(),
+                            },
+                        )
+
         except StopReplay:
             last = ctx.last_emitted_event_type()
             if last in ("STATE_SET", "STATE_UPDATE"):

loom/core/runner.py

@@ -50,4 +50,13 @@ async def run_once() -> bool:
         except Exception as e:
             traceback.print_exc()
             await db.task_failed(task["id"], str(e))
+
+            # Mark workflow as failed when unhandled exception occurs
+            await db.workflow_failed(
+                workflow_id=workflow_id,
+                error=str(e),
+                task_id=task["id"],
+                task_kind=task["kind"],
+            )
+
             return True

loom/database/db.py

@@ -344,6 +344,72 @@ class Database(Generic[InputT, StateT]):
             (workflow_id, json.dumps(signal_payload)),
         )
 
+    async def workflow_failed(
+        self, workflow_id: str, error: str, task_id: str = None, task_kind: str = None
+    ) -> None:
+        """Mark a workflow as failed due to an unhandled exception.
+
+        Creates a WORKFLOW_FAILED event and updates the workflow status.
+        Also cancels any remaining pending tasks for the workflow.
+
+        Args:
+            workflow_id: Workflow identifier to mark as failed
+            error: Error message describing the failure
+            task_id: Optional task ID that caused the failure
+            task_kind: Optional task kind that caused the failure
+
+        Raises:
+            WorkflowNotFoundError: If the workflow doesn't exist
+        """
+        # Get workflow info (this will raise WorkflowNotFoundError if not found)
+        workflow = await self.get_workflow_info(workflow_id)
+
+        # Skip if already in terminal state
+        if workflow["status"] in ("COMPLETED", "FAILED", "CANCELED"):
+            return
+
+        # Prepare failure payload
+        payload = {
+            "error": error,
+            "failed_at": datetime.now(timezone.utc).isoformat(),
+        }
+        if task_id:
+            payload["task_id"] = task_id
+        if task_kind:
+            payload["task_kind"] = task_kind
+
+        # Create failure event
+        await self.execute(
+            """
+            INSERT INTO events (workflow_id, type, payload)
+            VALUES (?, 'WORKFLOW_FAILED', ?)
+            """,
+            (workflow_id, json.dumps(payload)),
+        )
+
+        # Update workflow status
+        await self.execute(
+            """
+            UPDATE workflows
+            SET status = 'FAILED',
+                updated_at = CURRENT_TIMESTAMP
+            WHERE id = ?
+            """,
+            (workflow_id,),
+        )
+
+        # Cancel all pending tasks
+        await self.execute(
+            """
+            UPDATE tasks
+            SET status = 'FAILED',
+                last_error = 'workflow failed'
+            WHERE workflow_id = ?
+              AND status = 'PENDING'
+            """,
+            (workflow_id,),
+        )
+
     async def cancel_workflow(
         self, workflow_id: str, reason: str | None = None
     ) -> None:

{loom_core-0.1.4.dist-info → loom_core-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: loom-core
-Version: 0.1.4
+Version: 0.1.6
 Summary: Durable workflow orchestration engine for Python
 Home-page: https://github.com/satadeep3927/loom
 Author: Satadeep Dasgupta

{loom_core-0.1.4.dist-info → loom_core-0.1.6.dist-info}/RECORD

@@ -7,16 +7,16 @@ loom/common/errors.py,sha256=jc7_XS8dUPS8fg5pb7kJaTuIR1e6eKBZyZtbsHW3lgM,1712
 loom/common/workflow.py,sha256=R4X1vt6duqTJz6kGk6YI866O-CleOiyw0dXdUVo_qAE,2118
 loom/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 loom/core/compiled.py,sha256=Y2NQDMZVLee65ILgJoK2Ny_OcR1T4LJnOQWQcyg0Z7U,1178
-loom/core/context.py,sha256=rVWWSOaAmdVOzKKFh9TwpJVX-4Rm0h7bC76ipEcqVBw,9219
-loom/core/engine.py,sha256=twrZ8bABmQS7aVrj6qObPvx4zxhQTTFm2LY-KyGrhYs,4268
+loom/core/context.py,sha256=jxmaA_4Xdmi5FPxnXi3VWNPtyjMbdCKYi4j-Ju6Hi_g,10676
+loom/core/engine.py,sha256=6EvacrZCfVhJQ7yPurRdCeTzlGRaGk5uI-s7yxu9IUs,5822
 loom/core/handle.py,sha256=beOVmJit-gSrk7II4c2KuSubmr5OIPls82_vJ-pI04U,5532
 loom/core/logger.py,sha256=4ej95IdHOdBh-y7PQoRFz_H22PJYDXvAioBoId8mOhg,1729
-loom/core/runner.py,sha256=pYITY35Q88ItZA4EuSWmNE5sgxL0tExXrm9j5vSaz6o,1718
+loom/core/runner.py,sha256=nNA-EcdJt3UEgwzKHyb22xD7aLT8kI01x5UVdLteSI8,1998
 loom/core/state.py,sha256=oXiZ442PL2kl_JyQXLZOFeBLs_cTF2iEjykKVVUNiAI,2997
 loom/core/worker.py,sha256=jpTtvM1rIToVkG4SwJu1W9Q-eL3bw8b3N8KsxxrRovA,5322
 loom/core/workflow.py,sha256=E_A2rI6iaW3liZmVy3X98oxSeBDXMGiQnRZRq3Cp3BA,5755
 loom/database/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-loom/database/db.py,sha256=BtZTNtZ6cXpfvrtdmW8Xb47zoC2NXvVoWVmUILow2eA,23535
+loom/database/db.py,sha256=MSNXWhUzfwvjPXaVWY3-UK1lqw-Qas0Cc1FYcU8qdWw,25700
 loom/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 loom/decorators/activity.py,sha256=xt7oeYOKJxhKIEcwWsSVA26c_jk4yQB1PHtbcg3tCFI,4824
 loom/decorators/workflow.py,sha256=rBpbnpLfDTTAWdgiERpUPIgNUJpnYJKvBZPmzx-ECcc,1516
@@ -42,9 +42,9 @@ loom/schemas/database.py,sha256=DlMBvacsJ9oceBHkrMiJC7uD3_rCHFe6Kwa2STFcKOw,283
 loom/schemas/events.py,sha256=Gz-R836nVXNSsp4Y54idhKs_WTvzFPmx5KlA8PunP28,2216
 loom/schemas/tasks.py,sha256=DJbLInggIGxI-CfP1kSyK79Bz83e3sZyEKh6C2HE8q4,1341
 loom/schemas/workflow.py,sha256=F1cNEJRuEWLjPhGinEZE2T6vwchDC8RuTHmxEIaH6Ls,671
-loom_core-0.1.4.dist-info/licenses/LICENSE,sha256=8EpC-clAYRUfJQ92T3iQEIIWYjx2A3Kfk28zOd8lh7I,1095
-loom_core-0.1.4.dist-info/METADATA,sha256=pRtqEJf1DiwSbXPfnE-cHfuxZs8t2ol1I4LiXfAnm50,8897
-loom_core-0.1.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-loom_core-0.1.4.dist-info/entry_points.txt,sha256=Jx5HXHL2y7jvSjkwkH3QqF954cbSxiE6OGwL2coldyE,42
-loom_core-0.1.4.dist-info/top_level.txt,sha256=cAfRgAuCuit-cU9iBrf0bS4ovvmq-URykNd9fmYMojg,5
-loom_core-0.1.4.dist-info/RECORD,,
+loom_core-0.1.6.dist-info/licenses/LICENSE,sha256=8EpC-clAYRUfJQ92T3iQEIIWYjx2A3Kfk28zOd8lh7I,1095
+loom_core-0.1.6.dist-info/METADATA,sha256=L3FvzaOTy5AXGihpTZcxGeRfXTl4KKGiaYx3uKnCL2k,8897
+loom_core-0.1.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+loom_core-0.1.6.dist-info/entry_points.txt,sha256=Jx5HXHL2y7jvSjkwkH3QqF954cbSxiE6OGwL2coldyE,42
+loom_core-0.1.6.dist-info/top_level.txt,sha256=cAfRgAuCuit-cU9iBrf0bS4ovvmq-URykNd9fmYMojg,5
+loom_core-0.1.6.dist-info/RECORD,,