commandnet 0.6.0__tar.gz → 0.6.2__tar.gz

{commandnet-0.6.0 → commandnet-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: commandnet
-Version: 0.6.0
+Version: 0.6.2
 Summary: A lightweight, Pydantic-powered, distributed event-driven state machine and typed node graph runtime.
 Author: Christopher Vaz
 Author-email: christophervaz160@gmail.com

{commandnet-0.6.0 → commandnet-0.6.2}/commandnet/engine/runtime.py

@@ -106,23 +106,17 @@ class Engine:
             if hasattr(ctx, "is_cancelled"):
                 ctx.is_cancelled = await self.db.is_cancelled(subject_id)
 
-            # 2. RELEASE LOCK before long-running compute
-            await self.db.unlock_subject(subject_id)
-
-            # 3. COMPUTE (no DB lock held)
+            # 2. COMPUTE (Lock is HELD via Redis during compute)
             start_t = asyncio.get_event_loop().time()
-            result = await node_cls().run(ctx, payload)
+            try:
+                result = await node_cls().run(ctx, payload)
+            except asyncio.CancelledError:
+                self.logger.warning(f"Task {subject_id} cancelled during compute. Emitting Interrupt.")
+                result = Interrupt(subject_id=subject_id, hard=True)
+                
             duration = (asyncio.get_event_loop().time() - start_t) * 1000
-
-            # 4. RE-LOCK before writing
-            node_name_check, _ = await self.db.lock_and_load(subject_id)
-
-            # 🔴 IMPORTANT SAFETY CHECK (prevents stale writes)
-            if node_name_check != event.node_name and node_name_check != "AWAITING_CALL":
-                await self.db.unlock_subject(subject_id)
-                return
-
-            await self._apply_target(subject_id, ctx, result, duration)
+            
+            await self._apply_target(subject_id, ctx, result, duration, payload=payload) 
 
         except Exception as e:
             await self.observer.on_error(subject_id, event.node_name, e)
@@ -264,6 +258,22 @@ class Engine:
                     subject_id, join_name, len(target.branches)
                 )
 
+            async def _run_branch(task: ParallelTask):
+                branch_payload = (
+                    task.payload if task.payload is not None else payload
+                )
+
+                sub_ctx = self._get_path(context, task.sub_context_path)
+
+                await self._apply_target(
+                    f"{subject_id}#{task.sub_context_path}",
+                    sub_ctx,
+                    task.action,
+                    payload=branch_payload,
+                )
+
+            branch_tasks = []
+
             for branch in target.branches:
                 # Normalize branch into ParallelTask
                 if isinstance(branch, ParallelTask):
@@ -276,19 +286,10 @@ class Engine:
                     )
                     task = ParallelTask(action=branch, sub_context_path=path)
 
-                # PROPAGATION: fallback to incoming payload if branch has none
-                branch_payload = (
-                    task.payload if task.payload is not None else payload
-                )
+                branch_tasks.append(asyncio.create_task(_run_branch(task)))
 
-                sub_ctx = self._get_path(context, task.sub_context_path)
-
-                await self._apply_target(
-                    f"{subject_id}#{task.sub_context_path}",
-                    sub_ctx,
-                    task.action,
-                    payload=branch_payload,
-                )
+            if branch_tasks:
+                await asyncio.gather(*branch_tasks)
 
             if target.join_node:
                 await self.db.save_state(
@@ -379,7 +380,7 @@ class Engine:
             )
 
             result = await node_inst.on_signal(ctx, signal_id, payload)
-            await self._apply_target(subject_id, ctx, result)
+            await self._apply_target(subject_id, ctx, result, payload=payload)
         finally:
             await self.db.unlock_subject(subject_id)
 

{commandnet-0.6.0 → commandnet-0.6.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "commandnet"
-version = "0.6.0"
+version = "0.6.2"
 description = "A lightweight, Pydantic-powered, distributed event-driven state machine and typed node graph runtime."
 authors = [
     { name = "Christopher Vaz", email = "christophervaz160@gmail.com" }