From 6deb63b34250009ec29b7b9b0aeeffef2d72c429 Mon Sep 17 00:00:00 2001 From: Laura Abro Date: Thu, 24 Apr 2025 12:19:53 -0300 Subject: [PATCH] add update audit --- .../orca-agent/tests/stages/update_audit.py | 24 +++++++++++++++++++ .../orca-agent/tests/stages/worker_update.py | 0 2 files changed, 24 insertions(+) create mode 100644 worker/orca-agent/tests/stages/update_audit.py delete mode 100644 worker/orca-agent/tests/stages/worker_update.py 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