add worker check
This commit is contained in:
41
worker/orca-agent/tests/stages/worker_check.py
Normal file
41
worker/orca-agent/tests/stages/worker_check.py
Normal file
@ -0,0 +1,41 @@
|
|||||||
|
"""Stage for executing worker tasks."""
|
||||||
|
|
||||||
|
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)]
|
||||||
|
return {
|
||||||
|
"stakingKey": worker.staking_public_key,
|
||||||
|
"roundNumber": runner.current_round,
|
||||||
|
"githubUsername": worker.env.get("GITHUB_USERNAME"),
|
||||||
|
"prUrl": round_state["pr_urls"][worker.name],
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def execute(runner, worker, data):
|
||||||
|
"""Execute worker task step"""
|
||||||
|
url = f"{runner.config.middle_server_url}/summarizer/worker/check-todo"
|
||||||
|
response = requests.post(
|
||||||
|
url,
|
||||||
|
json=data,
|
||||||
|
)
|
||||||
|
result = response.json()
|
||||||
|
|
||||||
|
# Handle 409 gracefully - no eligible todos is an expected case
|
||||||
|
if response.status_code == 409:
|
||||||
|
print(
|
||||||
|
f"✓ {result.get('message', 'No eligible todos')} for {worker.name} - continuing"
|
||||||
|
)
|
||||||
|
return {"success": True, "message": result.get("message")}
|
||||||
|
else:
|
||||||
|
response.raise_for_status()
|
||||||
|
|
||||||
|
if result.get("success"):
|
||||||
|
runner.state["repo_url"] = (
|
||||||
|
f"https://github.com/{result['repo_owner']}/{result['repo_name']}"
|
||||||
|
)
|
||||||
|
|
||||||
|
return result
|
Reference in New Issue
Block a user