loom-core 0.1.8__py3-none-any.whl → 0.2.0__py3-none-any.whl

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/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.dist-info → loom_core-0.2.0.dist-info}/METADATA

@@ -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.dist-info → loom_core-0.2.0.dist-info}/RECORD

@@ -8,7 +8,7 @@ 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=IsEKaQV_Zzis-6HHLp79LgIZ5IEQvlkg7F8-ODYnVvM,10924
-loom/core/engine.py,sha256=6EvacrZCfVhJQ7yPurRdCeTzlGRaGk5uI-s7yxu9IUs,5822
+loom/core/engine.py,sha256=oHz_4YviHiw1E9XHWkELEHQ5KmAQXCKBebZDOwHXCCQ,5744
 loom/core/handle.py,sha256=beOVmJit-gSrk7II4c2KuSubmr5OIPls82_vJ-pI04U,5532
 loom/core/logger.py,sha256=ZSOZydQ-o8bfhh8VBcSWZNVE-NBvEedN7LDcbAEVJL0,1829
 loom/core/runner.py,sha256=nNA-EcdJt3UEgwzKHyb22xD7aLT8kI01x5UVdLteSI8,1998
@@ -16,7 +16,7 @@ loom/core/state.py,sha256=KynxRQfroYxUgv3Toy0c-yrPsmD5zHn7L3s_y8tkgcU,3142
 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=MSNXWhUzfwvjPXaVWY3-UK1lqw-Qas0Cc1FYcU8qdWw,25700
+loom/database/db.py,sha256=kzylC8D4d7w6c0dmgUfmPtKJTYhNCx35_Y1W4PDpevg,27492
 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.8.dist-info/licenses/LICENSE,sha256=8EpC-clAYRUfJQ92T3iQEIIWYjx2A3Kfk28zOd8lh7I,1095
-loom_core-0.1.8.dist-info/METADATA,sha256=H62DUX-YD22J2WMVYAfDyFaQwPPigntONMi_f8pqgVQ,8897
-loom_core-0.1.8.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-loom_core-0.1.8.dist-info/entry_points.txt,sha256=Jx5HXHL2y7jvSjkwkH3QqF954cbSxiE6OGwL2coldyE,42
-loom_core-0.1.8.dist-info/top_level.txt,sha256=cAfRgAuCuit-cU9iBrf0bS4ovvmq-URykNd9fmYMojg,5
-loom_core-0.1.8.dist-info/RECORD,,
+loom_core-0.2.0.dist-info/licenses/LICENSE,sha256=8EpC-clAYRUfJQ92T3iQEIIWYjx2A3Kfk28zOd8lh7I,1095
+loom_core-0.2.0.dist-info/METADATA,sha256=QqC1meKOvgvflzfvHpkFQDhSbHZoW2UHmtBSKCU4yfo,8897
+loom_core-0.2.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+loom_core-0.2.0.dist-info/entry_points.txt,sha256=Jx5HXHL2y7jvSjkwkH3QqF954cbSxiE6OGwL2coldyE,42
+loom_core-0.2.0.dist-info/top_level.txt,sha256=cAfRgAuCuit-cU9iBrf0bS4ovvmq-URykNd9fmYMojg,5
+loom_core-0.2.0.dist-info/RECORD,,