Herman/Yiqun
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use new version of test framework with updated worker management

Browse Source
This commit is contained in:
Laura Abro
2025-04-30 10:04:49 -03:00
parent b1c53693a5
commit aa936faf92
6 changed files with 32 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
worker/orca-agent/requirements.txt
View File

@ -17,4 +17,4 @@ sqlmodel>=0.0.22
openai>=0.28.0 openai>=0.28.0
colorama>=0.4.67 colorama>=0.4.67
prometheus-swarm>=0.1.7 prometheus-swarm>=0.1.7
prometheus-test>=0.1.4 prometheus-test>=0.1.5

16
worker/orca-agent/tests/stages/worker_audit.py
View File

@ -23,8 +23,8 @@ def prepare(runner, worker, target_name):
# "taskId": runner.get("task_id"), # "taskId": runner.get("task_id"),
# "roundNumber": runner.get("current_round"), # "roundNumber": runner.get("current_round"),
# "prUrl": pr_url, # "prUrl": pr_url,
# "stakingKey": worker.staking_public_key, # "stakingKey": worker.get_key("staking_public"),
# "pubKey": worker.public_key, # "pubKey": worker.get_key("main_public"),
# } # }
# Structure the payload according to what the server expects # Structure the payload according to what the server expects
@ -47,13 +47,15 @@ def prepare(runner, worker, target_name):
# "prUrl": pr_url, # "prUrl": pr_url,
# "repoOwner": submission_data.get("repoOwner"), # "repoOwner": submission_data.get("repoOwner"),
# "repoName": submission_data.get("repoName"), # "repoName": submission_data.get("repoName"),
# "githubUsername": worker.env.get("GITHUB_USERNAME"), # "githubUsername": worker.get_env("GITHUB_USERNAME"),
# "stakingKey": worker.staking_public_key, # "stakingKey": worker.get_key("staking_public"),
# "pubKey": worker.public_key, # "pubKey": worker.get_key("main_public"),
# "stakingSignature": create_signature( # "stakingSignature": create_signature(
# worker.staking_signing_key, auditor_payload # worker.get_key("staking_signing"), auditor_payload
# ),
# "publicSignature": create_signature(
# worker.get_key("main_signing"), auditor_payload
# ), # ),
# "publicSignature": create_signature(worker.public_signing_key, auditor_payload),
# } # }
return {"submission": submission_data} return {"submission": submission_data}

4
worker/orca-agent/tests/stages/worker_check.py
View File

@ -11,9 +11,9 @@ def prepare(runner, worker):
return None return None
return { return {
"stakingKey": worker.staking_public_key, "stakingKey": worker.get_key("staking_public"),
"roundNumber": runner.get("current_round"), "roundNumber": runner.get("current_round"),
"githubUsername": worker.env.get("GITHUB_USERNAME"), "githubUsername": worker.get_env("GITHUB_USERNAME"),
"prUrl": pr_url, "prUrl": pr_url,
} }

16
worker/orca-agent/tests/stages/worker_fetch.py
View File

@ -11,18 +11,20 @@ def prepare(runner, worker):
"taskId": runner.get("task_id"), "taskId": runner.get("task_id"),
"roundNumber": runner.get("current_round"), "roundNumber": runner.get("current_round"),
"action": "fetch-todo", "action": "fetch-todo",
"githubUsername": worker.env.get("GITHUB_USERNAME"), "githubUsername": worker.get_env("GITHUB_USERNAME"),
"stakingKey": worker.staking_public_key, "stakingKey": worker.get_key("staking_public"),
"pubKey": worker.public_key, "pubKey": worker.get_key("main_public"),
} }
return { return {
"taskId": runner.get("task_id"), "taskId": runner.get("task_id"),
"roundNumber": runner.get("current_round"), "roundNumber": runner.get("current_round"),
"stakingKey": worker.staking_public_key, "stakingKey": worker.get_key("staking_public"),
"pubKey": worker.public_key, "pubKey": worker.get_key("main_public"),
"stakingSignature": create_signature(worker.staking_signing_key, payload), "stakingSignature": create_signature(
"publicSignature": create_signature(worker.public_signing_key, payload), worker.get_key("staking_signing"), payload
),
"publicSignature": create_signature(worker.get_key("main_signing"), payload),
} }

8
worker/orca-agent/tests/stages/worker_pr.py
View File

@ -13,12 +13,12 @@ def prepare(runner, worker):
"action": "add-todo-pr", "action": "add-todo-pr",
"roundNumber": runner.get("current_round"), "roundNumber": runner.get("current_round"),
"prUrl": pr_url, "prUrl": pr_url,
"stakingKey": worker.staking_public_key, "stakingKey": worker.get_key("staking_public"),
"pubKey": worker.public_key, "pubKey": worker.get_key("main_public"),
} }
return { return {
"signature": create_signature(worker.staking_signing_key, payload), "signature": create_signature(worker.get_key("staking_signing"), payload),
"stakingKey": worker.staking_public_key, "stakingKey": worker.get_key("staking_public"),
} }

12
worker/orca-agent/tests/stages/worker_submission.py
View File

@ -21,17 +21,19 @@ def prepare(runner, worker):
submitter_payload = { submitter_payload = {
"taskId": runner.get("task_id"), "taskId": runner.get("task_id"),
"roundNumber": runner.get("current_round"), "roundNumber": runner.get("current_round"),
"stakingKey": worker.staking_public_key, "stakingKey": worker.get_key("staking_public"),
"pubKey": worker.public_key, "pubKey": worker.get_key("main_public"),
"action": "audit", "action": "audit",
**submission_data, **submission_data,
} }
return { return {
**submission_data, **submission_data,
"signature": create_signature(worker.staking_signing_key, submitter_payload), "signature": create_signature(
"stakingKey": worker.staking_public_key, worker.get_key("staking_signing"), submitter_payload
"pubKey": worker.public_key, ),
"stakingKey": worker.get_key("staking_public"),
"pubKey": worker.get_key("main_public"),
} }