From 38a74a30ffb48b3cdf3e885463239aca9b1fd2fb Mon Sep 17 00:00:00 2001 From: koid Date: Sun, 5 Nov 2023 15:38:19 +0900 Subject: [PATCH] add middleware when initializing fastapi --- pr_agent/servers/serverless.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pr_agent/servers/serverless.py b/pr_agent/servers/serverless.py index 915963156..b03d91714 100644 --- a/pr_agent/servers/serverless.py +++ b/pr_agent/servers/serverless.py @@ -1,12 +1,15 @@ from fastapi import FastAPI from mangum import Mangum +from starlette.middleware import Middleware +from starlette_context.middleware import RawContextMiddleware from pr_agent.log import setup_logger from pr_agent.servers.github_app import router setup_logger() -app = FastAPI() +middleware = [Middleware(RawContextMiddleware)] +app = FastAPI(middleware=middleware) app.include_router(router) handler = Mangum(app, lifespan="off")