loom-core 0.1.4__tar.gz → 0.1.5__tar.gz

{loom_core-0.1.4 → loom_core-0.1.5}/PKG-INFO

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

{loom_core-0.1.4 → loom_core-0.1.5}/loom/core/context.py

@@ -102,6 +102,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:

{loom_core-0.1.4 → loom_core-0.1.5}/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-0.1.4 → loom_core-0.1.5}/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_core-0.1.4 → loom_core-0.1.5}/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 → loom_core-0.1.5}/loom_core.egg-info/PKG-INFO

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

{loom_core-0.1.4 → loom_core-0.1.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "loom-core"
-version = "0.1.4"
+version = "0.1.5"
 description = "Durable workflow orchestration engine for Python"
 readme = "README.md"
 requires-python = ">=3.12"