edda-framework 0.14.0__py3-none-any.whl → 0.15.0__py3-none-any.whl

edda/locking.py

@@ -192,7 +192,6 @@ async def _refresh_lock_periodically(
 
 async def cleanup_stale_locks_periodically(
     storage: StorageProtocol,
-    worker_id: str,
     interval: int = 60,
 ) -> None:
     """
@@ -204,49 +203,37 @@ async def cleanup_stale_locks_periodically(
     Note: This function only cleans up locks without resuming workflows.
     For automatic workflow resumption, use auto_resume_stale_workflows_periodically().
 
-    Uses system-level locking to ensure only one pod executes cleanup at a time.
+    Important: This function should only be run by a single worker (e.g., via leader
+    election). It does not perform its own distributed coordination.
 
     Example:
         >>> asyncio.create_task(
-        ...     cleanup_stale_locks_periodically(storage, worker_id, interval=60)
+        ...     cleanup_stale_locks_periodically(storage, interval=60)
         ... )
 
     Args:
         storage: Storage backend
-        worker_id: Unique identifier for this worker (for global lock coordination)
         interval: Cleanup interval in seconds (default: 60)
     """
     with suppress(asyncio.CancelledError):
         while True:
-            # Add jitter to prevent thundering herd in multi-pod deployments
+            # Add jitter to prevent thundering herd
             jitter = random.uniform(0, interval * 0.3)
             await asyncio.sleep(interval + jitter)
 
-            # Try to acquire global lock for this task
-            lock_acquired = await storage.try_acquire_system_lock(
-                lock_name="cleanup_stale_locks",
-                worker_id=worker_id,
-                timeout_seconds=interval,
-            )
-
-            if not lock_acquired:
-                # Another pod is handling this task
-                continue
-
             try:
                 # Clean up stale locks
                 workflows = await storage.cleanup_stale_locks()
 
                 if len(workflows) > 0:
                     logger.info("Cleaned up %d stale locks", len(workflows))
-            finally:
-                await storage.release_system_lock("cleanup_stale_locks", worker_id)
+            except Exception as e:
+                logger.error("Failed to cleanup stale locks: %s", e, exc_info=True)
 
 
 async def auto_resume_stale_workflows_periodically(
     storage: StorageProtocol,
     replay_engine: Any,
-    worker_id: str,
     interval: int = 60,
 ) -> None:
     """
@@ -255,39 +242,27 @@ async def auto_resume_stale_workflows_periodically(
     This combines lock cleanup with automatic workflow resumption, ensuring
     that workflows interrupted by worker crashes are automatically recovered.
 
-    Uses system-level locking to ensure only one pod executes this task at a time,
-    preventing duplicate workflow execution (CRITICAL for safety).
+    Important: This function should only be run by a single worker (e.g., via leader
+    election). It does not perform its own distributed coordination.
 
     Example:
         >>> asyncio.create_task(
         ...     auto_resume_stale_workflows_periodically(
-        ...         storage, replay_engine, worker_id, interval=60
+        ...         storage, replay_engine, interval=60
         ...     )
         ... )
 
     Args:
         storage: Storage backend
         replay_engine: ReplayEngine instance for resuming workflows
-        worker_id: Unique identifier for this worker (for global lock coordination)
         interval: Cleanup interval in seconds (default: 60)
     """
     with suppress(asyncio.CancelledError):
         while True:
-            # Add jitter to prevent thundering herd in multi-pod deployments
+            # Add jitter to prevent thundering herd
             jitter = random.uniform(0, interval * 0.3)
             await asyncio.sleep(interval + jitter)
 
-            # Try to acquire global lock for this task
-            lock_acquired = await storage.try_acquire_system_lock(
-                lock_name="auto_resume_stale_workflows",
-                worker_id=worker_id,
-                timeout_seconds=interval,
-            )
-
-            if not lock_acquired:
-                # Another pod is handling this task
-                continue
-
             try:
                 # Clean up stale locks and get workflows to resume
                 workflows_to_resume = await storage.cleanup_stale_locks()
@@ -369,8 +344,8 @@ async def auto_resume_stale_workflows_periodically(
                                 e,
                                 exc_info=True,
                             )
-            finally:
-                await storage.release_system_lock("auto_resume_stale_workflows", worker_id)
+            except Exception as e:
+                logger.error("Failed to cleanup stale locks: %s", e, exc_info=True)
 
 
 class LockNotAcquiredError(Exception):

{edda_framework-0.14.0.dist-info → edda_framework-0.15.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: edda-framework
-Version: 0.14.0
+Version: 0.15.0
 Summary: Lightweight Durable Execution Framework
 Project-URL: Homepage, https://github.com/i2y/edda
 Project-URL: Documentation, https://github.com/i2y/edda#readme
@@ -42,6 +42,10 @@ Requires-Dist: starlette>=0.40.0; extra == 'dev'
 Requires-Dist: testcontainers[mysql]>=4.0.0; extra == 'dev'
 Requires-Dist: testcontainers[postgres]>=4.0.0; extra == 'dev'
 Requires-Dist: tsuno>=0.1.3; extra == 'dev'
+Provides-Extra: graph
+Requires-Dist: pydantic-graph>=0.1.0; extra == 'graph'
+Provides-Extra: llamaindex
+Requires-Dist: llama-index-core>=0.12.0; extra == 'llamaindex'
 Provides-Extra: mcp
 Requires-Dist: mcp>=1.22.0; extra == 'mcp'
 Provides-Extra: mirascope
@@ -97,6 +101,8 @@ For detailed documentation, visit [https://i2y.github.io/edda/](https://i2y.gith
 - ⚡ **Instant Notifications**: PostgreSQL LISTEN/NOTIFY for near-instant event delivery (optional)
 - 🤖 **MCP Integration**: Expose durable workflows as AI tools via Model Context Protocol
 - 🧠 **Mirascope Integration**: Durable LLM calls
+- 🦙 **LlamaIndex Integration**: Make LlamaIndex Workflows durable with crash recovery
+- 📊 **pydantic-graph Integration**: Durable graph-based workflows (experimental)
 - 🌍 **ASGI/WSGI Support**: Deploy with your preferred server (uvicorn, gunicorn, uWSGI)
 
 ## Use Cases
@@ -233,6 +239,12 @@ uv add edda-framework --extra viewer
 # With PostgreSQL instant notifications (LISTEN/NOTIFY)
 uv add edda-framework --extra postgres-notify
 
+# With LlamaIndex Workflow integration
+uv add edda-framework --extra llamaindex
+
+# With pydantic-graph integration (experimental)
+uv add edda-framework --extra graph
+
 # All extras (PostgreSQL, MySQL, Viewer UI)
 uv add edda-framework --extra postgresql --extra mysql --extra viewer
 ```

{edda_framework-0.14.0.dist-info → edda_framework-0.15.0.dist-info}/RECORD

@@ -1,18 +1,27 @@
 edda/__init__.py,sha256=hGC6WR2R36M8LWC97F-0Rw4Ln0QUUT_1xC-7acOy_Fk,2237
 edda/activity.py,sha256=nRm9eBrr0lFe4ZRQ2whyZ6mo5xd171ITIVhqytUhOpw,21025
-edda/app.py,sha256=ITTc7x5S4ykCP3KPZXKxuNczXkPtbn04ZQaxcem46Hw,68406
+edda/app.py,sha256=gtBtNsWpib8BHwzC02MlP7LxUHNXqcZVwtFcUhTWGkk,67801
 edda/channels.py,sha256=6JFZkeOs0xDumexr0_bLI_Mb4S245hLJM_Sqp3xPCCA,37676
 edda/compensation.py,sha256=iKLlnTxiF1YSatmYQW84EkPB1yMKUEZBtgjuGnghLtY,11824
 edda/context.py,sha256=Qqm_nUC5NNnOfHAb7taqKqZVIc0GoRWUrjZ4L9_-q70,22128
 edda/exceptions.py,sha256=-ntBLGpVQgPFG5N1o8m_7weejAYkNrUdxTkOP38vsHk,1766
 edda/hooks.py,sha256=HUZ6FTM__DZjwuomDfTDEroQ3mugEPuJHcGm7CTQNvg,8193
-edda/locking.py,sha256=NAFJmw-JaSVsXn4Y4czJyv_s9bWG8cdrzDBWIEag5X8,13661
+edda/locking.py,sha256=ZMdzGO4u3h8m3kDysmkDpAkJNvQQADbILPsmo52EQis,12716
 edda/pydantic_utils.py,sha256=dGVPNrrttDeq1k233PopCtjORYjZitsgASPfPnO6R10,9056
 edda/replay.py,sha256=IQGByw9mlTpRulyUgsHJSPsZUULmM2YqFcm2WeB4jtw,43227
 edda/retry.py,sha256=t4_E1skrhotA1XWHTLbKi-DOgCMasOUnhI9OT-O_eCE,6843
 edda/workflow.py,sha256=hfBZM0JrtK0IkvZSrva0VmYVyvKCdiJ5FWFmIVENfrM,8807
 edda/wsgi.py,sha256=1pGE5fhHpcsYnDR8S3NEFKWUs5P0JK4roTAzX9BsIj0,2391
 edda/integrations/__init__.py,sha256=F_CaTvlDEbldfOpPKq_U9ve1E573tS6XzqXnOtyHcXI,33
+edda/integrations/graph/__init__.py,sha256=MwGgkTDsOH1eaLYzWFxBR2DHumiSAz-UUsqXSNU8aWw,1652
+edda/integrations/graph/context.py,sha256=ZQaesBVDAmF01P1liX2BtxaprQDRUgLKL0e94P8Yrdc,2529
+edda/integrations/graph/exceptions.py,sha256=FwDNUafYHzWlPeObLLTjyOVj8M94HgZ2a3DLR6VcOj4,286
+edda/integrations/graph/graph.py,sha256=RY3BbCUO_rsqQ2wP7ghfstT59stC_KNyJ7LmvljJSFk,12957
+edda/integrations/graph/nodes.py,sha256=JHoJYCEAGBKTYJ-458pMYvVxEu9hunAzzEuNVcEBbvM,4746
+edda/integrations/llamaindex/__init__.py,sha256=YR1ikaf7AanaQztGZfHDZtoMqz3ieTGjBeIQ3sAkzP8,1604
+edda/integrations/llamaindex/events.py,sha256=brs0UVu3kM5N9mWKo91FNBBh3FL9xI2WEzgHmp-j4a8,4751
+edda/integrations/llamaindex/exceptions.py,sha256=41BefzhS1qd2L1U-JHBNIkTQS_Vz480OATqOtJNmDGU,376
+edda/integrations/llamaindex/workflow.py,sha256=MXvPe8d7FLivCLoDlzQA-TgQi82nulUmW0omasq4QEk,10813
 edda/integrations/mcp/__init__.py,sha256=YK-8m0DIdP-RSqewlIX7xnWU7TD3NioCiW2_aZSgnn8,1232
 edda/integrations/mcp/decorators.py,sha256=31SmbDwmHEGvUNa3aaatW91hBkpnS5iN9uy47dID3J4,10037
 edda/integrations/mcp/server.py,sha256=Q5r4AbMn-9gBcy2CZocbgW7O0fn7Qb4e9CBJa1FEmzU,14507
@@ -47,8 +56,8 @@ edda/visualizer/mermaid_generator.py,sha256=XWa2egoOTNDfJEjPcwoxwQmblUqXf7YInWFj
 edda/migrations/mysql/20251217000000_initial_schema.sql,sha256=LpINasESRhadOeqABwDk4JZ0OZ4_zQw_opnhIR4Xe9U,12367
 edda/migrations/postgresql/20251217000000_initial_schema.sql,sha256=hCaGMWeptpzpnsjfNKVsMYuwPRe__fK9E0VZpClAumQ,11732
 edda/migrations/sqlite/20251217000000_initial_schema.sql,sha256=Wq9gCnQ0K9SOt0PY_8f1MG4va8rLVWIIcf2lnRzSK5g,11906
-edda_framework-0.14.0.dist-info/METADATA,sha256=FbHEPrrr0THCfzFccY0_NJyCPuGArsxK6AKVrmTUKjQ,37567
-edda_framework-0.14.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-edda_framework-0.14.0.dist-info/entry_points.txt,sha256=dPH47s6UoJgUZxHoeSMqZsQkLaSE-SGLi-gh88k2WrU,48
-edda_framework-0.14.0.dist-info/licenses/LICENSE,sha256=udxb-V7_cYKTHqW7lNm48rxJ-Zpf0WAY_PyGDK9BPCo,1069
-edda_framework-0.14.0.dist-info/RECORD,,
+edda_framework-0.15.0.dist-info/METADATA,sha256=d7VXuP5MWTx0WvL4KzwmZVl0AUtTJzTxaLWL7Q0Ou8Q,38074
+edda_framework-0.15.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+edda_framework-0.15.0.dist-info/entry_points.txt,sha256=dPH47s6UoJgUZxHoeSMqZsQkLaSE-SGLi-gh88k2WrU,48
+edda_framework-0.15.0.dist-info/licenses/LICENSE,sha256=udxb-V7_cYKTHqW7lNm48rxJ-Zpf0WAY_PyGDK9BPCo,1069
+edda_framework-0.15.0.dist-info/RECORD,,