loom-core 0.1.8__tar.gz → 0.2.0__tar.gz

{loom_core-0.1.8 → loom_core-0.2.0}/PKG-INFO

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

{loom_core-0.1.8 → loom_core-0.2.0}/loom/core/engine.py

@@ -137,5 +137,4 @@ class Engine(Generic[InputT, StateT]):
             return
 
         async with Database[InputT, StateT]() as db:
-            await db.create_event(workflow_id, "WORKFLOW_COMPLETED", {})
-            await db.complete_running_step(workflow_id)
+            await db.complete_workflow(workflow_id)

{loom_core-0.1.8 → loom_core-0.2.0}/loom/database/db.py

@@ -154,6 +154,7 @@ class Database(Generic[InputT, StateT]):
                     "id": row["id"],
                     "name": row["name"],
                     "status": row["status"],
+                    "version": row["version"],
                     "module": row["module"],
                     "input": row["input"],
                     "created_at": row["created_at"],
@@ -345,7 +346,7 @@ class Database(Generic[InputT, StateT]):
         )
 
     async def workflow_failed(
-        self, workflow_id: str, error: str, task_id: str = None, task_kind: str = None
+        self, workflow_id: str, error: str, task_id: str|None = None, task_kind: str|None = None
     ) -> None:
         """Mark a workflow as failed due to an unhandled exception.
 
@@ -410,6 +411,60 @@ class Database(Generic[InputT, StateT]):
             (workflow_id,),
         )
 
+    async def complete_workflow(self, workflow_id: str) -> None:
+        """Mark a workflow as successfully completed.
+
+        Creates a WORKFLOW_COMPLETED event and updates the workflow status.
+        Also completes any running step tasks for the workflow.
+
+        Args:
+            workflow_id: Workflow identifier to mark as completed
+
+        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
+
+        # Create completion event
+        await self.execute(
+            """
+            INSERT INTO events (workflow_id, type, payload)
+            VALUES (?, 'WORKFLOW_COMPLETED', ?)
+            """,
+            (workflow_id, json.dumps({
+                "completed_at": datetime.now(timezone.utc).isoformat()
+            })),
+        )
+
+        # Update workflow status
+        await self.execute(
+            """
+            UPDATE workflows
+            SET status = 'COMPLETED',
+                updated_at = CURRENT_TIMESTAMP
+            WHERE id = ?
+            """,
+            (workflow_id,),
+        )
+
+        # Complete any running step tasks
+        await self.execute(
+            """
+            UPDATE tasks
+            SET status = 'COMPLETED',
+                updated_at = CURRENT_TIMESTAMP
+            WHERE workflow_id = ?
+              AND kind = 'STEP'
+              AND status = 'RUNNING'
+            """,
+            (workflow_id,),
+        )
+
     async def cancel_workflow(
         self, workflow_id: str, reason: str | None = None
     ) -> None:

{loom_core-0.1.8 → loom_core-0.2.0}/loom_core.egg-info/PKG-INFO

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

{loom_core-0.1.8 → loom_core-0.2.0}/pyproject.toml

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