Herman/Yiqun
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: changed to swarmBountyId Based

Browse Source
This commit is contained in:
HermanL02
2025-05-05 12:08:44 -03:00
parent fdd23af669
commit 84866d3260
4 changed files with 21 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
worker/orca-agent/src/server/routes/task.py
View File

@ -10,7 +10,7 @@ bp = Blueprint("task", __name__)
executor = ThreadPoolExecutor(max_workers=2)
def post_pr_url(agent_result, task_id, signature, round_number):
def post_pr_url(agent_result, task_id, signature, swarmBountyId):
try:
result = agent_result.result() # Get the result from the future
logger.info(f"Result: {result}")
@ -22,7 +22,7 @@ def post_pr_url(agent_result, task_id, signature, round_number):
json={
"prUrl": result_data.get("data", {}).get("pr_url"),
"signature": signature,
"roundNumber": round_number,
"swarmBountyId": swarmBountyId,
"success": result.get("success", False),
"message": result_data.get("error", ""),
},
@ -38,17 +38,18 @@ def post_pr_url(agent_result, task_id, signature, round_number):
logger.error(f"Traceback: {''.join(traceback.format_tb(e.__traceback__))}")
@bp.post("/worker-task/<round_number>")
def start_task(round_number):
@bp.post("/worker-task")
def start_task():
logger = repo_summary_service.logger
logger.info(f"Task started for round: {round_number}")
# logger.info(f"Task started for bounty: {swarmBountyId}")
data = request.get_json()
task_id = data["task_id"]
podcall_signature = data["podcall_signature"]
repo_url = data["repo_url"]
swarmBountyId = data["swarmBountyId"]
logger.info(f"Task data: {data}")
required_fields = ["task_id", "round_number", "repo_url", "podcall_signature"]
required_fields = ["task_id", "swarmBountyId", "repo_url", "podcall_signature"]
if any(data.get(field) is None for field in required_fields):
return jsonify({"error": "Missing data"}), 401
@ -58,7 +59,7 @@ def start_task(round_number):
if os.getenv("TEST_MODE") == "true":
result = repo_summary_service.handle_task_creation(
task_id=task_id,
round_number=int(round_number),
swarmBountyId=swarmBountyId,
repo_url=repo_url,
db=db, # Pass db instance
)
@ -67,12 +68,12 @@ def start_task(round_number):
agent_result = executor.submit(
repo_summary_service.handle_task_creation,
task_id=task_id,
round_number=round_number,
swarmBountyId=swarmBountyId,
repo_url=repo_url,
db=db, # Pass db instance
)
agent_result.add_done_callback(
lambda future: post_pr_url(future, task_id, podcall_signature, round_number)
lambda future: post_pr_url(future, task_id, podcall_signature, swarmBountyId)
)
return jsonify({"status": "Task is being processed"}), 200
@ -87,14 +88,14 @@ if __name__ == "__main__":
# Test data
test_data = {
"taskId": "fake",
"round_number": "1",
"swarmBountyId": "1",
"repo_url": "https://github.com/koii-network/docs",
}
# Set up test context
with app.test_client() as client:
# Make a POST request to the endpoint
response = client.post("/repo_summary/1", json=test_data)
response = client.post("/worker-task", json=test_data)
# Print the response
print(f"Status Code: {response.status_code}")