loom-core 0.1.7__tar.gz → 0.1.9__tar.gz

{loom_core-0.1.7 → loom_core-0.1.9}/PKG-INFO

@@ -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 → loom_core-0.1.9}/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.7 → loom_core-0.1.9}/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_core-0.1.7 → loom_core-0.1.9}/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 → loom_core-0.1.9}/loom_core.egg-info/PKG-INFO

@@ -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 → loom_core-0.1.9}/pyproject.toml

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