loom-core 0.1.7__py3-none-any.whl → 0.1.9__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/core/state.py

@@ -32,6 +32,7 @@ class StateProxy(Generic[InputT, StateT]):
         event = self._ctx._peek()
         if event and event["type"] == "STATE_SET" and event["payload"]["key"] == name:
             self._ctx._consume()
+            self._data[name] = value
             return
 
         event = ("STATE_SET", {"key": name, "value": value})
@@ -56,6 +57,8 @@ class StateProxy(Generic[InputT, StateT]):
             payload = event["payload"]
             if set(payload["values"].keys()) == set(updaters.keys()):
                 self._ctx._consume()
+                for key, value in payload["values"].items():
+                    self._data[key] = value
                 return
         new_values = {}
 

loom/database/db.py

@@ -345,7 +345,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 +410,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.7.dist-info → loom_core-0.1.9.dist-info}/METADATA

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

{loom_core-0.1.7.dist-info → loom_core-0.1.9.dist-info}/RECORD

@@ -8,15 +8,15 @@ 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
-loom/core/state.py,sha256=oXiZ442PL2kl_JyQXLZOFeBLs_cTF2iEjykKVVUNiAI,2997
+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=Ml-H3ePOaOpJeVPtxGfTRsAzx6cs0zyRQVBVttzmE1U,27444
 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.7.dist-info/licenses/LICENSE,sha256=8EpC-clAYRUfJQ92T3iQEIIWYjx2A3Kfk28zOd8lh7I,1095
-loom_core-0.1.7.dist-info/METADATA,sha256=jxnYEA02G3sXiR4BQ4FWpXGuRIz7NDEux43KwWjyoMk,8897
-loom_core-0.1.7.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-loom_core-0.1.7.dist-info/entry_points.txt,sha256=Jx5HXHL2y7jvSjkwkH3QqF954cbSxiE6OGwL2coldyE,42
-loom_core-0.1.7.dist-info/top_level.txt,sha256=cAfRgAuCuit-cU9iBrf0bS4ovvmq-URykNd9fmYMojg,5
-loom_core-0.1.7.dist-info/RECORD,,
+loom_core-0.1.9.dist-info/licenses/LICENSE,sha256=8EpC-clAYRUfJQ92T3iQEIIWYjx2A3Kfk28zOd8lh7I,1095
+loom_core-0.1.9.dist-info/METADATA,sha256=5Hjfpo7DUJcht4m0NlV1WjiYdtUBD4iOBbWnzUqSA7w,8897
+loom_core-0.1.9.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+loom_core-0.1.9.dist-info/entry_points.txt,sha256=Jx5HXHL2y7jvSjkwkH3QqF954cbSxiE6OGwL2coldyE,42
+loom_core-0.1.9.dist-info/top_level.txt,sha256=cAfRgAuCuit-cU9iBrf0bS4ovvmq-URykNd9fmYMojg,5
+loom_core-0.1.9.dist-info/RECORD,,