OpenSorus / server /main.py
halfacupoftea's picture
Add modal infra
071ba00
from fastapi import FastAPI, HTTPException
from agent.core import run_agent
app = FastAPI()
@app.post('/webhook')
async def check_payload(payload: dict):
if "action" in payload:
if payload["action"] == "created":
comment_body = payload["comment"]["body"] if "comment" in payload else ""
if "@opensorus" in comment_body.lower():
print("This issue is assigned to OpenSorus Agent.")
issue_url = payload["issue"]["url"]
print("URL", issue_url)
branch_name = payload["repository"]["default_branch"]
print("Branch Name", branch_name)
result = await run_agent(issue_url, branch_name)
return {"message": result or "This issue is assigned to OpenSorus Agent."}
else:
raise HTTPException(status_code=400, detail="Unknown action.")
else:
raise HTTPException(status_code=400, detail="No valid payload.")
@app.get('/health')
def health_check():
return {"status": "Hello World!, I am alive!"}