From b1024b8daa561b42756239dc37feedb6d5405056 Mon Sep 17 00:00:00 2001
From: Marius Ciepluch <11855163+norandom@users.noreply.github.com>
Date: Tue, 6 Oct 2026 08:20:40 +0200
Subject: [PATCH] ax: verify and probe tasks carry the
 search-verify/search-probe name prefixes [skip ci]

The kind default "pin" had produced ousast-engine-pin-search-verify-* names,
which the operator per-host egress rules do not match.

Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
---
 benchmarks/ax/search_probe.py  | 4 ++--
 benchmarks/ax/search_verify.py | 4 ++--
 benchmarks/search/pilot_run.py | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/benchmarks/ax/search_probe.py b/benchmarks/ax/search_probe.py
index e9d5e92..5558b7d 100644
--- a/benchmarks/ax/search_probe.py
+++ b/benchmarks/ax/search_probe.py
@@ -30,12 +30,12 @@ def validate_result(data: bytes, output: Path) -> None:
 
 def probe_workload(image: str, deadline: float = 900) -> Workload:
     validate_image(image)
-    return Workload(image, frozenset({"RESULT_URL"}), deadline, validate_result)
+    return Workload(image, frozenset({"RESULT_URL"}), deadline, validate_result, kind="search-probe")
 
 
 def probe_item(image: str) -> Item:
     return Item(
-        "search-probe",
+        "v1",
         {},
         "result.json",
         ("python3", "-m", "openultrasast.search.probe"),
diff --git a/benchmarks/ax/search_verify.py b/benchmarks/ax/search_verify.py
index 082ad00..791917d 100644
--- a/benchmarks/ax/search_verify.py
+++ b/benchmarks/ax/search_verify.py
@@ -66,7 +66,7 @@ class AXSideDispatcher:
             raise ValueError("verification accepts checkout and declarative inputs only")
         # The trusted host selects an executor for this side. Build/package work
         # finishes there before any verifier worker slot is acquired.
-        workload = Workload(self.image, frozenset({"VERIFY_INPUT_URL", "RESULT_URL"}), timeout_seconds * 6 + 60, validate_result)
+        workload = Workload(self.image, frozenset({"VERIFY_INPUT_URL", "RESULT_URL"}), timeout_seconds * 6 + 60, validate_result, kind="search-verify")
         lane = self.lane or AXLane(self.args, workload)
         spec = {"demo": demo, "family": family, "timeout_seconds": timeout_seconds}
         try:
@@ -79,7 +79,7 @@ class AXSideDispatcher:
             return SideRecord("could_not_build", (), 0, str(exc), isolation_mode="task-boundary")
         records = []
         for _ in range(3):
-            identity = "search-verify-" + uuid.uuid4().hex
+            identity = uuid.uuid4().hex
             item = Item(
                 identity,
                 {"VERIFY_INPUT_URL": archive},
diff --git a/benchmarks/search/pilot_run.py b/benchmarks/search/pilot_run.py
index d604ed5..17a9e73 100644
--- a/benchmarks/search/pilot_run.py
+++ b/benchmarks/search/pilot_run.py
@@ -501,7 +501,7 @@ def main(argv=None):
         try:
             store = open_store(args.board_memory) if args.board_memory else FileStore(root / "boards")
             if not args.dry_run:
-                lane = AXLane(args, Workload(args.image, frozenset({"VERIFY_INPUT_URL", "RESULT_URL"}), 900, validate_result))
+                lane = AXLane(args, Workload(args.image, frozenset({"VERIFY_INPUT_URL", "RESULT_URL"}), 900, validate_result, kind="search-verify"))
         except Exception as exc:
             aborted_reason = exception_reason(exc)
         for index in indices if aborted_reason is None else []: