fix: handle calling req-id setting locally (#405)

fal-ai · Feb 4, 2025 · a390060 · a390060
1 parent 9df38f8
commit a390060
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/projects/fal/src/fal/app.py b/projects/fal/src/fal/app.py
@@ -374,6 +374,11 @@ async def set_global_object_preference(request, call_next):
         @app.middleware("http")
         async def set_request_id(request, call_next):
             # NOTE: Setting request_id is not supported for websocket/realtime endpoints
+            if not os.getenv("IS_ISOLATE_AGENT") or not os.environ.get(
+                "NOMAD_ALLOC_PORT_grpc"
+            ):
+                # If not running in the expected environment, skip setting request_id
+                return await call_next(request)
 
             if self.isolate_channel is None:
                 grpc_port = os.environ.get("NOMAD_ALLOC_PORT_grpc")