diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py index 3d3c0df2e..f57d359d4 100644 --- a/llama_stack/distribution/server/server.py +++ b/llama_stack/distribution/server/server.py @@ -230,29 +230,23 @@ class TracingMiddleware: self.app = app self.impls = impls # FastAPI built-in paths that should bypass custom routing - self.fastapi_paths = ( - "/docs", - "/redoc", - "/openapi.json", - "/favicon.ico", - "/static" - ) + self.fastapi_paths = ("/docs", "/redoc", "/openapi.json", "/favicon.ico", "/static") async def __call__(self, scope, receive, send): if scope.get("type") == "lifespan": return await self.app(scope, receive, send) path = scope.get("path", "") - + # Check if the path is a FastAPI built-in path if path.startswith(self.fastapi_paths): # Pass through to FastAPI's built-in handlers logger.debug(f"Bypassing custom routing for FastAPI built-in path: {path}") return await self.app(scope, receive, send) - + if not hasattr(self, "endpoint_impls"): self.endpoint_impls = initialize_endpoint_impls(self.impls) - + try: _, _, trace_path = find_matching_endpoint(scope.get("method", "GET"), path, self.endpoint_impls) except ValueError: