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

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

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

{loom_core-0.1.5 → loom_core-0.1.7}/loom/core/context.py

@@ -37,6 +37,7 @@ class WorkflowContext(Generic[InputT, StateT]):
     state: StateProxy[InputT, StateT]
     cursor: int = 0
     logger: WorkflowLogger
+    _original_history_length: int  # Track original history size for replay detection
 
     def __init__(
         self, id: str, input: InputT, history: List[Event], state: StateT
@@ -52,6 +53,7 @@ class WorkflowContext(Generic[InputT, StateT]):
         self.id = id
         self.input = input
         self.history = history
+        self._original_history_length = len(history)  # Remember original size
         self.state = StateProxy(self, state)
         self.logger = WorkflowLogger(self)
 
@@ -82,6 +84,19 @@ class WorkflowContext(Generic[InputT, StateT]):
         self.cursor += 1
         return event
 
+    def _skip_step_events(self) -> None:
+        """Skip over STEP_START and STEP_END events during replay.
+        
+        These are internal workflow management events that don't affect
+        the deterministic execution logic.
+        """
+        while True:
+            event = self._peek()
+            if event and event["type"] in ("STEP_START", "STEP_END"):
+                self._consume()
+            else:
+                break
+
     def _match_event(self, expected_type: str) -> Event | None:
         """
         Safe helper to check if the NEXT event matches what we expect.
@@ -95,14 +110,13 @@ class WorkflowContext(Generic[InputT, StateT]):
 
     @property
     def is_replaying(self) -> bool:
-        """Check if the workflow is currently replaying events.
+        """Check if the workflow is currently replaying old events.
 
         Returns:
-            True if there are remaining events to replay, False otherwise
+            True if replaying historical events, False if executing new logic
         """
-        return self.cursor < len(self.history)
-
-    @property
+        # We're replaying if we haven't consumed all the original events yet
+        return self.cursor < self._original_history_length
     def is_at_end_of_history(self) -> bool:
         """Check if we've consumed all events in history.
 
@@ -145,6 +159,8 @@ class WorkflowContext(Generic[InputT, StateT]):
     ) -> FuncReturn:
         metadata = self._extract_activity_metadata(fn, args)
 
+        # Skip any step events first
+        self._skip_step_events()
         scheduled_event = self._match_event("ACTIVITY_SCHEDULED")
 
         if scheduled_event:
@@ -156,6 +172,8 @@ class WorkflowContext(Generic[InputT, StateT]):
 
             self._consume()
 
+            # Skip step events before checking for completion
+            self._skip_step_events()
             completed_event = self._match_event("ACTIVITY_COMPLETED")
 
             if completed_event:
@@ -164,6 +182,8 @@ class WorkflowContext(Generic[InputT, StateT]):
 
             raise StopReplay
 
+        # Skip step events before checking for unexpected events
+        self._skip_step_events()
         unexpected_event = self._peek()
         if unexpected_event:
             raise NonDeterministicWorkflowError(
@@ -189,11 +209,16 @@ class WorkflowContext(Generic[InputT, StateT]):
             datetime.datetime.now(datetime.timezone.utc) + delta if delta else until  # type: ignore
         )
 
+        # Skip any step events first
+        self._skip_step_events()
+        
         scheduled_event = self._match_event("TIMER_SCHEDULED")
 
         if scheduled_event:
             self._consume()
 
+            # Skip step events before checking for timer fired
+            self._skip_step_events()
             fired_event = self._match_event("TIMER_FIRED")
             if fired_event:
                 self._consume()
@@ -201,6 +226,8 @@ class WorkflowContext(Generic[InputT, StateT]):
 
             raise StopReplay
 
+        # Skip step events before checking for unexpected events
+        self._skip_step_events()
         unexpected_event = self._peek()
         if unexpected_event:
             raise NonDeterministicWorkflowError(
@@ -219,6 +246,8 @@ class WorkflowContext(Generic[InputT, StateT]):
         If the signal is already in history (replay), it returns the data immediately.
         If not, it raises StopReplay to suspend execution until the signal arrives.
         """
+        # Skip any step events first
+        self._skip_step_events()
         # 1. Check if the signal is next in history
         event = self._match_event("SIGNAL_RECEIVED")
 
@@ -235,6 +264,8 @@ class WorkflowContext(Generic[InputT, StateT]):
             self._consume()
             return event["payload"]["data"]
 
+        # Skip step events before checking for unexpected events
+        self._skip_step_events()
         unexpected_event = self._peek()
         if unexpected_event:
             raise NonDeterministicWorkflowError(

{loom_core-0.1.5 → loom_core-0.1.7}/loom/core/logger.py

@@ -40,10 +40,11 @@ class WorkflowLogger:
         self._log("DEBUG", msg)
 
     def _log(self, level: str, msg: str):
+        # Only suppress logging if we're actually replaying historical events
         if self._ctx.is_replaying:
             return
 
-        self._std_logger.info(f"{msg}")
+        self._std_logger.info(f"[{self._ctx.id[:8]}] {msg}")
         asyncio.create_task(self._write_log_to_db(level, msg))
 
     async def _write_log_to_db(self, level: str, msg: str):

{loom_core-0.1.5 → loom_core-0.1.7}/loom_core.egg-info/PKG-INFO

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

{loom_core-0.1.5 → loom_core-0.1.7}/pyproject.toml

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