POST req
This commit is contained in:
+2
-1
@@ -1,4 +1,5 @@
|
||||
.env
|
||||
projects
|
||||
role_id
|
||||
wrapped_secret_id
|
||||
wrapped_secret_id
|
||||
.idea
|
||||
Generated
-14
@@ -1,14 +0,0 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<module type="PYTHON_MODULE" version="4">
|
||||
<component name="NewModuleRootManager">
|
||||
<content url="file://$MODULE_DIR$">
|
||||
<sourceFolder url="file://$MODULE_DIR$" isTestSource="false" />
|
||||
<sourceFolder url="file://$MODULE_DIR$/agents/issue_triage/tests" isTestSource="true" />
|
||||
<excludeFolder url="file://$MODULE_DIR$/.venv" />
|
||||
<excludeFolder url="file://$MODULE_DIR$/projects/agents" />
|
||||
<excludeFolder url="file://$MODULE_DIR$/venv" />
|
||||
</content>
|
||||
<orderEntry type="jdk" jdkName="Python 3.10 (agents) (2)" jdkType="Python SDK" />
|
||||
<orderEntry type="sourceFolder" forTests="false" />
|
||||
</component>
|
||||
</module>
|
||||
@@ -3,6 +3,7 @@ import os
|
||||
from contextlib import asynccontextmanager
|
||||
|
||||
from fastapi import FastAPI, HTTPException, Depends
|
||||
from pydantic import BaseModel
|
||||
|
||||
from agents.coders.BaseCoder import CoderAgent
|
||||
from agents.issue_triage.IssueTriageAgent import IssueTriageAgent
|
||||
@@ -85,17 +86,20 @@ async def say_hello(name: str):
|
||||
return {"message": f"Hello {name}"}
|
||||
|
||||
|
||||
@app.get("/consume_task")
|
||||
async def decomposing_issue(issue: str, agent: IssueTriageAgent = Depends(get_agent)):
|
||||
class IncomingTask(BaseModel):
|
||||
issue: str
|
||||
|
||||
@app.post("/consume_task")
|
||||
async def decomposing_issue(body: IncomingTask, agent: IssueTriageAgent = Depends(get_agent)):
|
||||
try:
|
||||
logger.info(f"Received an issue: {issue}")
|
||||
plan = await agent.plan_issue(issue)
|
||||
logger.info(f"Received an issue: {body.issue}")
|
||||
plan = await agent.plan_issue(body.issue)
|
||||
logger.info("Task planning successful.")
|
||||
logger.info(plan.render())
|
||||
return {"plan": plan}
|
||||
except IssueNotFound:
|
||||
logger.error(f"Issue {issue} not found.")
|
||||
raise HTTPException(status_code=404, detail=f"Issue {issue} not found")
|
||||
logger.error(f"Issue {body.issue} not found.")
|
||||
raise HTTPException(status_code=404, detail=f"Issue {body.issue} not found")
|
||||
except Exception as e:
|
||||
logger.exception("An unexpected error occurred")
|
||||
return {"error": str(e)}
|
||||
|
||||
Reference in New Issue
Block a user