add worker check

worker/orca-agent/tests/stages/worker_check.py

@@ -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