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

fix edge cases when data doesn't exist

Browse Source
This commit is contained in:
Laura Abro
2025-04-24 22:00:54 -03:00
parent 80774287e2
commit 5c41ec9119
5 changed files with 19 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
worker/orca-agent/tests/stages/update_audit.py
View File

@ -21,4 +21,5 @@ def execute(runner, worker, data):
)
response.raise_for_status()
return response.json()
# Return a formatted response regardless of type
return {"success": True, "message": response.text}

9
worker/orca-agent/tests/stages/worker_check.py
View File

@ -6,17 +6,22 @@ import requests
def prepare(runner, worker):
"""Prepare data for worker task"""
# Create fetch-todo payload for stakingSignature and publicSignature
round_state = runner.state["rounds"][str(runner.current_round)]
round_state = runner.state["rounds"].get(str(runner.current_round), {})
if not round_state.get("pr_urls"):
print(f"✓ No PR URLs found for {worker.name} - continuing")
return
return {
"stakingKey": worker.staking_public_key,
"roundNumber": runner.current_round,
"githubUsername": worker.env.get("GITHUB_USERNAME"),
"prUrl": round_state["pr_urls"][worker.name],
"prUrl": round_state.get("pr_urls", {}).get(worker.name),
}
def execute(runner, worker, data):
"""Execute worker task step"""
if not data:
return {"success": True, "message": "No PR URL found"}
url = f"{runner.config.middle_server_url}/summarizer/worker/check-todo"
response = requests.post(
url,

3
worker/orca-agent/tests/stages/worker_fetch.py
View File

@ -45,7 +45,8 @@ def execute(runner, worker, data):
response.raise_for_status()
if result.get("success"):
runner.state["repo_url"] = (
round_state = runner.state["rounds"][runner.current_round]
round_state["repo_url"] = (
f"https://github.com/{result['data']['repo_owner']}/{result['data']['repo_name']}"
)

4
worker/orca-agent/tests/stages/worker_pr.py
View File

@ -3,9 +3,9 @@ from prometheus_test.utils import create_signature
def prepare(runner, worker):
round_state = runner.state["rounds"][str(runner.current_round)]
round_state = runner.state["rounds"].get(str(runner.current_round), {})
if worker.name not in round_state["pr_urls"]:
if worker.name not in round_state.get("pr_urls", {}):
print(f"✓ No PR URL found for {worker.name} - continuing")
return None

11
worker/orca-agent/tests/stages/worker_task.py
View File

@ -5,20 +5,21 @@ import requests
def prepare(runner, worker):
"""Prepare data for worker task"""
round_state = runner.state["rounds"].get(str(runner.current_round), {})
if not round_state.get("repo_url"):
print(f"✓ No repo url found for {worker.name} - continuing")
return
return {
"taskId": runner.config.task_id,
"round_number": str(runner.current_round),
"repo_url": runner.state["repo_url"],
"repo_url": round_state["repo_url"],
}
def execute(runner, worker, data):
"""Execute worker task step"""
if not runner.state["repo_url"]:
print(f"✓ No repo url found for {worker.name} - continuing")
if not data:
return {"success": True, "message": "No repo url found"}
url = f"{worker.url}/worker-task/{runner.current_round}"
response = requests.post(url, json=data)
result = response.json()