loom-core 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

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

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

@@ -7,10 +7,10 @@ loom/common/errors.py,sha256=jc7_XS8dUPS8fg5pb7kJaTuIR1e6eKBZyZtbsHW3lgM,1712
 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=7zezfvPaPdyfjJnUd8cjTwoxKGUh0N6F1nHn8dCuUVY,9494
+loom/core/context.py,sha256=IsEKaQV_Zzis-6HHLp79LgIZ5IEQvlkg7F8-ODYnVvM,10924
 loom/core/engine.py,sha256=6EvacrZCfVhJQ7yPurRdCeTzlGRaGk5uI-s7yxu9IUs,5822
 loom/core/handle.py,sha256=beOVmJit-gSrk7II4c2KuSubmr5OIPls82_vJ-pI04U,5532
-loom/core/logger.py,sha256=4ej95IdHOdBh-y7PQoRFz_H22PJYDXvAioBoId8mOhg,1729
+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/worker.py,sha256=jpTtvM1rIToVkG4SwJu1W9Q-eL3bw8b3N8KsxxrRovA,5322
@@ -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.5.dist-info/licenses/LICENSE,sha256=8EpC-clAYRUfJQ92T3iQEIIWYjx2A3Kfk28zOd8lh7I,1095
-loom_core-0.1.5.dist-info/METADATA,sha256=TdEYynBj66M4a173vlAyY5YAteLmBNVKEiqSuWZXx3Q,8897
-loom_core-0.1.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-loom_core-0.1.5.dist-info/entry_points.txt,sha256=Jx5HXHL2y7jvSjkwkH3QqF954cbSxiE6OGwL2coldyE,42
-loom_core-0.1.5.dist-info/top_level.txt,sha256=cAfRgAuCuit-cU9iBrf0bS4ovvmq-URykNd9fmYMojg,5
-loom_core-0.1.5.dist-info/RECORD,,
+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,,