From e5744c9a06456de8b2c0c2e9f53606794a842cef Mon Sep 17 00:00:00 2001
From: Laura Abro <laura@koii.network>
Date: Thu, 24 Apr 2025 12:12:14 -0300
Subject: [PATCH] add worker check

---
 .../orca-agent/tests/stages/worker_check.py   | 41 +++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 worker/orca-agent/tests/stages/worker_check.py

diff --git a/worker/orca-agent/tests/stages/worker_check.py b/worker/orca-agent/tests/stages/worker_check.py
new file mode 100644
index 0000000..3bf4d36
--- /dev/null
+++ b/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