auto detect skyvern context and set context if it's not set (#269)

Skyvern-AI · May 7, 2024 · 0862232 · 0862232
1 parent 6473635
commit 0862232
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/skyvern/forge/api_app.py b/skyvern/forge/api_app.py
@@ -80,8 +80,12 @@ async def unexpected_exception(request: Request, exc: Exception) -> JSONResponse
 
     @app.middleware("http")
     async def request_middleware(request: Request, call_next: Callable[[Request], Awaitable[Response]]) -> Response:
-        request_id = str(uuid.uuid4())
-        skyvern_context.set(SkyvernContext(request_id=request_id))
+        curr_ctx = skyvern_context.current()
+        if not curr_ctx:
+            request_id = str(uuid.uuid4())
+            skyvern_context.set(SkyvernContext(request_id=request_id))
+        elif not curr_ctx.request_id:
+            curr_ctx.request_id = str(uuid.uuid4())
 
         try:
             return await call_next(request)