diff --git a/src/askui/chat/api/app.py b/src/askui/chat/api/app.py index 90061b5c..6d21eb55 100644 --- a/src/askui/chat/api/app.py +++ b/src/askui/chat/api/app.py @@ -40,7 +40,6 @@ async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]: # noqa: ARG001 title="AskUI Chat API", version="0.1.0", lifespan=lifespan, - dependencies=[SetEnvFromHeadersDep], ) @@ -67,7 +66,12 @@ async def combined_lifespan(app: FastAPI) -> AsyncGenerator[None, None]: yield -app = FastAPI(title=app.title, version=app.version, lifespan=combined_lifespan) +app = FastAPI( + title=app.title, + version=app.version, + lifespan=combined_lifespan, + dependencies=[SetEnvFromHeadersDep], +) app.mount("/mcp", mcp_app) app.include_router(v1_router)