From 90599f53d47616a716a3cb86a91df2fe277c6c33 Mon Sep 17 00:00:00 2001 From: Ori Kotek Date: Fri, 23 Feb 2024 18:27:34 +0200 Subject: [PATCH] Change github app startup logic to support gunicorj --- pr_agent/servers/github_app.py | 20 +++++++++----------- 1 file changed, 9 insertions(+), 11 deletions(-) diff --git a/pr_agent/servers/github_app.py b/pr_agent/servers/github_app.py index 2191c4d6..1a2f5d1b 100644 --- a/pr_agent/servers/github_app.py +++ b/pr_agent/servers/github_app.py @@ -280,17 +280,15 @@ async def root(): return {"status": "ok"} -def start(): - if get_settings().github_app.override_deployment_type: - # Override the deployment type to app - get_settings().set("GITHUB.DEPLOYMENT_TYPE", "app") - get_settings().set("CONFIG.PUBLISH_OUTPUT_PROGRESS", False) - middleware = [Middleware(RawContextMiddleware)] - app = FastAPI(middleware=middleware) - app.include_router(router) +if get_settings().github_app.override_deployment_type: + # Override the deployment type to app + get_settings().set("GITHUB.DEPLOYMENT_TYPE", "app") +get_settings().set("CONFIG.PUBLISH_OUTPUT_PROGRESS", False) +middleware = [Middleware(RawContextMiddleware)] +app = FastAPI(middleware=middleware) +app.include_router(router) + +def start(): uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "3000"))) - -if __name__ == '__main__': - start()