xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-08 06:40:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Support GitLab webhooks in Lambda functions

Browse Source
This commit is contained in:
yoohya
2025-06-16 01:22:01 +09:00
parent cbea0380ec
commit 99904601ce
4 changed files with 70 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
pr_agent/servers/serverless.py
View File

@ -3,8 +3,6 @@ from mangum import Mangum
from starlette.middleware import Middleware
from starlette_context.middleware import RawContextMiddleware
from pr_agent.servers.github_app import router
try:
from pr_agent.config_loader import apply_secrets_manager_config
apply_secrets_manager_config()
@ -16,12 +14,18 @@ except Exception as e:
# Fail completely silently if log module is not available
pass
middleware = [Middleware(RawContextMiddleware)]
app = FastAPI(middleware=middleware)
app.include_router(router)
def _create_handler(router):
middleware = [Middleware(RawContextMiddleware)]
app = FastAPI(middleware=middleware)
app.include_router(router)
return Mangum(app, lifespan="off")
handler = Mangum(app, lifespan="off")
def serverless(event, context):
def serverless_github(event, context):
from pr_agent.servers.github_app import router
handler = _create_handler(router)
return handler(event, context)
def serverless_gitlab(event, context):
from pr_agent.servers.gitlab_webhook import router
handler = _create_handler(router)
return handler(event, context)