fix: Retry worker on runtine errors

julep-ai · Sep 20, 2024 · 7aedee0 · 7aedee0
1 parent 6b3ab54
commit 7aedee0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/agents-api/agents_api/worker/__main__.py b/agents-api/agents_api/worker/__main__.py
@@ -7,10 +7,13 @@
 
 import asyncio
 
+from tenacity import retry, retry_if_exception_type, wait_fixed
+
 from ..clients import temporal
 from .worker import create_worker
 
 
+@retry(wait=wait_fixed(20), retry=retry_if_exception_type(RuntimeError))
 async def main():
     """
     Initializes the Temporal client and worker with TLS configuration (if provided),