diff --git a/worker/orca-agent/tests/stages/update_audit.py b/worker/orca-agent/tests/stages/update_audit.py new file mode 100644 index 0000000..edcaf1c --- /dev/null +++ b/worker/orca-agent/tests/stages/update_audit.py @@ -0,0 +1,24 @@ +"""Stage for executing worker tasks.""" + +import requests + + +def prepare(runner, worker): + """Prepare data for worker task""" + + return { + "taskId": runner.config.task_id, + "round": runner.current_round, + } + + +def execute(runner, worker, data): + """Execute worker task step""" + url = f"{runner.config.middle_server_url}/summarizer/worker/update-audit-result" + response = requests.post( + url, + json=data, + ) + response.raise_for_status() + + return response.json() diff --git a/worker/orca-agent/tests/stages/worker_update.py b/worker/orca-agent/tests/stages/worker_update.py deleted file mode 100644 index e69de29..0000000