@algosuite/vo-mcp 0.2.0-beta.22 → 0.2.0-beta.24

package/dist/runner-supervisor.js

@@ -61,8 +61,130 @@ async function fetchInstallationToken({ req, required = false, readOnly = false,
   }
 }
 
+// ../../scripts/virtual-office/code-runner/control-plane-task-list.mjs
+var PAGE_SIZE = 500;
+async function listAllPrOpenedTasks(request) {
+  const tasks = [];
+  let beforeCreatedAt = "";
+  let beforeId = "";
+  for (; ; ) {
+    const params = new URLSearchParams({
+      status: "pr_opened",
+      limit: String(PAGE_SIZE),
+      runner_adoption: "1"
+    });
+    if (beforeCreatedAt) {
+      params.set("before_created_at", beforeCreatedAt);
+      params.set("before_id", beforeId);
+    }
+    const res = await request("GET", `/api/v1/code-task?${params}`);
+    if (!res.ok) throw new Error(`listPrOpenedTasks failed: HTTP ${res.status}`);
+    const json = await res.json();
+    const page = Array.isArray(json?.tasks) ? json.tasks : [];
+    tasks.push(...page);
+    if (page.length < PAGE_SIZE) return tasks;
+    const last = page.at(-1);
+    if (!last?.created_at || !last?.code_task_id) {
+      throw new Error("listPrOpenedTasks pagination cursor missing");
+    }
+    beforeCreatedAt = last.created_at;
+    beforeId = last.code_task_id;
+  }
+}
+
+// ../../scripts/virtual-office/code-runner/control-plane-resume.mjs
+async function resumeCodeTaskRequest(req, taskId, { automaticRateLimit = false, automaticContinuation = false } = {}, onUnauthorized = () => {
+}) {
+  const res = await req(
+    "POST",
+    `/api/v1/code-task/${encodeURIComponent(taskId)}/resume`,
+    automaticRateLimit ? { automatic_rate_limit: true } : automaticContinuation ? { automatic_continuation: true } : {}
+  );
+  if (res.status === 401) {
+    onUnauthorized();
+    throw new Error("resume unauthorized (401)");
+  }
+  if (!res.ok) throw new Error(`resume failed: HTTP ${res.status}`);
+  const json = await res.json();
+  return json && json.task ? json.task : null;
+}
+
+// ../../scripts/virtual-office/code-runner/control-plane-autonomous-admission.mjs
+function makeAutonomousDispatchAdmissionClient(req, timeoutMs, onUnauthorized = () => {
+}) {
+  return {
+    async reserveAutonomousDispatchBudget({ requestedBudgetUsd, reservationId, occurrenceKey }) {
+      const res = await req("POST", "/api/v1/autonomous-dispatch/admission", {
+        requested_budget_usd: requestedBudgetUsd,
+        reservation_id: reservationId,
+        dispatch_occurrence_key: occurrenceKey
+      }, { timeoutMs });
+      if (res.status === 401) {
+        onUnauthorized();
+        throw new Error("autonomous dispatch admission unauthorized (401)");
+      }
+      if (!res.ok) throw new Error(`autonomous dispatch admission failed: HTTP ${res.status}`);
+      const body = await res.json();
+      return {
+        allowed: body?.allowed === true,
+        reason: typeof body?.reason === "string" ? body.reason : ""
+      };
+    },
+    async releaseAutonomousDispatchBudget(reservationId) {
+      const res = await req("POST", "/api/v1/autonomous-dispatch/reservation/release", {
+        reservation_id: reservationId
+      }, { timeoutMs });
+      if (res.status === 401) {
+        onUnauthorized();
+        throw new Error("autonomous dispatch release unauthorized (401)");
+      }
+      if (!res.ok) throw new Error(`autonomous dispatch release failed: HTTP ${res.status}`);
+      return true;
+    }
+  };
+}
+
+// ../../scripts/virtual-office/code-runner/control-plane-merge.mjs
+async function mergeVerifiedPrRequest(req, prNumber, automationContext, onUnauthorized) {
+  const res = await req(
+    "POST",
+    "/api/v1/admin/pr/merge",
+    { prNumber, automationContext },
+    { timeoutMs: 12e4 }
+  );
+  if (res.status === 401) {
+    onUnauthorized();
+    throw new Error("gated merge unauthorized (401)");
+  }
+  const json = await res.json().catch(() => ({}));
+  if (res.ok && json?.ok === true) {
+    const result = json?.result && typeof json.result === "object" ? json.result : {};
+    const status = result.merged === true || result.status === "merged" ? "merged" : result.status === "auto-merge-enabled" || String(result.action || "").includes("auto-merge") ? "queued" : "accepted";
+    return {
+      status,
+      detail: typeof result.detail === "string" ? result.detail : null,
+      actionReceiptId: typeof json.action_receipt_id === "string" ? json.action_receipt_id : null
+    };
+  }
+  if (res.status === 503 && json?.error === "verify_unavailable") {
+    return { status: "retry", reason: json.reason || "verification unavailable" };
+  }
+  return {
+    status: "blocked",
+    reason: json?.reason || json?.message || json?.error || `HTTP ${res.status}`,
+    actionReceiptId: typeof json?.action_receipt_id === "string" ? json.action_receipt_id : null
+  };
+}
+
 // ../../scripts/virtual-office/code-runner/control-plane-client.mjs
 var cachedFirebaseToken = null;
+var ClaimAuthorityChangedError = class extends Error {
+  constructor() {
+    super("code-task claim authority changed");
+    this.name = "ClaimAuthorityChangedError";
+    this.code = "code_task_claim_authority_changed";
+  }
+};
 async function resolveBearer(env) {
   const adminToken = env.VO_CONTROL_PLANE_ADMIN_TOKEN;
   if (adminToken) return adminToken;
@@ -84,12 +206,16 @@ function createControlPlaneClient({
   heartbeatTimeoutMs = Math.min(
     Math.max(Number(env.VO_CODE_RUNNER_HEARTBEAT_TIMEOUT_MS) || 15e3, 1e3),
     6e4
-  )
+  ),
+  taskRequestTimeoutMs = Math.min(
+    Math.max(Number(env.VO_CODE_RUNNER_TASK_REQUEST_TIMEOUT_MS) || 5e3, 100),
+    6e4
+  ),
+  runnerId: runnerId2,
+  runnerInstanceId
 } = {}) {
   const resolvedBaseUrl = baseUrl ?? env.VO_CONTROL_PLANE_URL ?? "";
-  if (!resolvedBaseUrl) {
-    throw new Error("VO_CONTROL_PLANE_URL is required for the code-runner daemon");
-  }
+  if (!resolvedBaseUrl) throw new Error("VO_CONTROL_PLANE_URL is required for the code-runner daemon");
   const root = resolvedBaseUrl.replace(/\/+$/, "");
   async function req(method, path2, body, { timeoutMs } = {}) {
     const bearer = await resolveBearer(env);
@@ -117,7 +243,15 @@ function createControlPlaneClient({
       clearTimeout(timeoutId);
     }
   }
+  const taskReq = (method, path2, body, options = {}) => req(method, path2, body, { timeoutMs: taskRequestTimeoutMs, ...options });
   return {
+    ...makeAutonomousDispatchAdmissionClient(
+      req,
+      taskRequestTimeoutMs,
+      () => {
+        cachedFirebaseToken = null;
+      }
+    ),
     /**
      * Claim the next pending task. Returns the task or null (empty queue).
      * `repos` (optional `owner/name` list) and `operatorIds` (optional
@@ -125,17 +259,20 @@ function createControlPlaneClient({
      * serves — the control-plane filters by both (logical AND), so another
      * operator's task never lands on (or bills) this machine.
      */
-    async claim(runnerId2, repos, operatorIds, session = {}) {
-      const body = { runner_id: runnerId2 };
+    async claim(runnerId3, repos, operatorIds, session = {}) {
+      const body = { runner_id: runnerId3 };
       if (Array.isArray(repos) && repos.length > 0) body.repos = repos;
       if (Array.isArray(operatorIds) && operatorIds.length > 0) body.operator_ids = operatorIds;
-      if (session.runnerInstanceId) body.runner_instance_id = session.runnerInstanceId;
+      if (session.runnerInstanceId) {
+        body.runner_instance_id = session.runnerInstanceId;
+        body.runner_progress_protocol_version = 2;
+      }
       if (session.runnerInstanceId && session.reconcileStale) body.reconcile_stale = true;
       if (session.defaultAgent) body.default_agent = session.defaultAgent;
       if (Array.isArray(session.availableAgents)) {
         body.available_agents = session.availableAgents.filter((entry) => entry?.installed === true && entry?.authenticated === true).map((entry) => entry.agent);
       }
-      const res = await req("POST", "/api/v1/code-task/claim", body);
+      const res = await taskReq("POST", "/api/v1/code-task/claim", body);
       if (res.status === 401) {
         cachedFirebaseToken = null;
         throw new Error("claim unauthorized (401)");
@@ -149,11 +286,19 @@ function createControlPlaneClient({
      * Server derives operator/tenant from the daemon's authenticated principal.
      * Returns the created task, or throws on a non-2xx response.
      */
-    async enqueueCodeTask({ repo, prompt, max_budget_usd, max_turns }) {
+    async enqueueCodeTask({ repo, prompt, max_budget_usd, max_turns, dispatch_mode, tier, agent, model, dispatch_occurrence_key, autonomous_reservation_id, on_behalf_of_operator_id, repair_pr_number, repair_kind, repair_head_sha, repair_chain }) {
       const body = { repo, prompt };
       if (typeof max_budget_usd === "number") body.max_budget_usd = max_budget_usd;
       if (typeof max_turns === "number") body.max_turns = max_turns;
-      const res = await req("POST", "/api/v1/code-task", body);
+      for (const [key, value] of Object.entries({ dispatch_mode, tier, agent, model, repair_kind, repair_head_sha })) {
+        if (value) body[key] = value;
+      }
+      if (dispatch_occurrence_key) body.dispatch_occurrence_key = dispatch_occurrence_key;
+      if (autonomous_reservation_id) body.autonomous_reservation_id = autonomous_reservation_id;
+      if (on_behalf_of_operator_id) body.on_behalf_of_operator_id = on_behalf_of_operator_id;
+      if (Number.isInteger(repair_pr_number) && repair_pr_number > 0) body.repair_pr_number = repair_pr_number;
+      if (repair_chain) body.repair_chain = repair_chain;
+      const res = await taskReq("POST", "/api/v1/code-task", body);
       if (res.status === 401) {
         cachedFirebaseToken = null;
         throw new Error("enqueue unauthorized (401)");
@@ -166,15 +311,10 @@ function createControlPlaneClient({
      * Resume a failed/cancelled/max-turn partial code-task. The PR watcher uses
      * this after the runner opens a partial draft PR and CI is no longer pending.
      */
-    async resumeCodeTask(taskId) {
-      const res = await req("POST", `/api/v1/code-task/${encodeURIComponent(taskId)}/resume`, {});
-      if (res.status === 401) {
+    async resumeCodeTask(taskId, { automaticRateLimit = false, automaticContinuation = false } = {}) {
+      return resumeCodeTaskRequest(taskReq, taskId, { automaticRateLimit, automaticContinuation }, () => {
         cachedFirebaseToken = null;
-        throw new Error("resume unauthorized (401)");
-      }
-      if (!res.ok) throw new Error(`resume failed: HTTP ${res.status}`);
-      const json = await res.json();
-      return json && json.task ? json.task : null;
+      });
     },
     /**
      * Send a CI-green PR through the production verify-before-act merge route.
@@ -182,29 +322,14 @@ function createControlPlaneClient({
      * consensus, records a receipt, and direct-merges only the inspected SHA.
      */
     async mergeVerifiedPr(prNumber, automationContext) {
-      const res = await req("POST", "/api/v1/admin/pr/merge", { prNumber, automationContext });
-      if (res.status === 401) {
-        cachedFirebaseToken = null;
-        throw new Error("gated merge unauthorized (401)");
-      }
-      const json = await res.json().catch(() => ({}));
-      if (res.ok && json?.ok === true) {
-        const result = json?.result && typeof json.result === "object" ? json.result : {};
-        const status = result.merged === true || result.status === "merged" ? "merged" : result.status === "auto-merge-enabled" || String(result.action || "").includes("auto-merge") ? "queued" : "accepted";
-        return {
-          status,
-          detail: typeof result.detail === "string" ? result.detail : null,
-          actionReceiptId: typeof json.action_receipt_id === "string" ? json.action_receipt_id : null
-        };
-      }
-      if (res.status === 503 && json?.error === "verify_unavailable") {
-        return { status: "retry", reason: json.reason || "verification unavailable" };
-      }
-      return {
-        status: "blocked",
-        reason: json?.reason || json?.message || json?.error || `HTTP ${res.status}`,
-        actionReceiptId: typeof json?.action_receipt_id === "string" ? json.action_receipt_id : null
-      };
+      return mergeVerifiedPrRequest(
+        req,
+        prNumber,
+        automationContext,
+        () => {
+          cachedFirebaseToken = null;
+        }
+      );
     },
     /**
      * Append progress / set terminal status. Returns
@@ -212,23 +337,37 @@ function createControlPlaneClient({
      *   { terminal: true }  — task already terminal (operator cancelled): STOP
      */
     async postProgress(taskId, patch) {
-      const res = await req("PATCH", `/api/v1/code-task/${taskId}/progress`, patch);
-      if (res.status === 409) return { terminal: true };
+      const progress = {
+        ...patch,
+        ...patch.runner_id ? {} : runnerId2 ? { runner_id: runnerId2 } : {},
+        ...patch.runner_instance_id ? {} : runnerInstanceId ? { runner_instance_id: runnerInstanceId } : {}
+      };
+      const res = await taskReq("PATCH", `/api/v1/code-task/${taskId}/progress`, progress);
+      if (res.status === 409) {
+        const conflict = await res.json().catch(() => ({}));
+        if (conflict?.error === "code_task_claim_authority_changed") {
+          throw new ClaimAuthorityChangedError();
+        }
+        return { terminal: true };
+      }
       if (res.status === 404) return { terminal: true, missing: true };
       if (!res.ok) throw new Error(`progress failed: HTTP ${res.status}`);
       const json = await res.json();
       return { task: json && json.task };
     },
     async getTask(taskId) {
-      const res = await req("GET", `/api/v1/code-task/${taskId}`);
+      const res = await taskReq("GET", `/api/v1/code-task/${taskId}`);
       if (res.status === 404) return null;
       if (!res.ok) throw new Error(`getTask failed: HTTP ${res.status}`);
       const json = await res.json();
       return json ? json.task : null;
     },
+    async listPrOpenedTasks() {
+      return listAllPrOpenedTasks(taskReq);
+    },
     async downloadTaskAttachment(taskId, attachmentId) {
       const path2 = `/api/v1/code-task/${encodeURIComponent(taskId)}/attachment/${encodeURIComponent(attachmentId)}`;
-      const res = await req("GET", path2);
+      const res = await taskReq("GET", path2);
       if (res.status === 401) cachedFirebaseToken = null;
       if (!res.ok) throw new Error(`attachment download failed: HTTP ${res.status}`);
       return Buffer.from(await res.arrayBuffer());
@@ -236,7 +375,7 @@ function createControlPlaneClient({
     async getTaskKnowledgeContext(taskId, { query } = {}) {
       const body = {};
       if (typeof query === "string" && query.trim()) body.query = query;
-      const res = await req("POST", `/api/v1/code-task/${encodeURIComponent(taskId)}/knowledge-context`, body);
+      const res = await taskReq("POST", `/api/v1/code-task/${encodeURIComponent(taskId)}/knowledge-context`, body);
       if (res.status === 401) {
         cachedFirebaseToken = null;
         throw new Error("knowledge-context unauthorized (401)");
@@ -255,10 +394,10 @@ function createControlPlaneClient({
      * `tokens` = { input_tokens, output_tokens, cache_creation_tokens, cache_read_tokens }.
      * Optional: `claudeWeeklyPct` (number) + `claudeWeeklyResetsAt` (ISO string | null).
      */
-    async postWeeklyTokens({ operatorId, runnerId: runnerId2, tokens, claudeWeeklyPct, claudeWeeklyResetsAt }) {
+    async postWeeklyTokens({ operatorId, runnerId: runnerId3, tokens, claudeWeeklyPct, claudeWeeklyResetsAt }) {
       const body = {
         operator_id: operatorId,
-        runner_id: runnerId2,
+        runner_id: runnerId3,
         input_tokens: tokens.input_tokens,
         output_tokens: tokens.output_tokens,
         cache_creation_tokens: tokens.cache_creation_tokens,
@@ -270,7 +409,7 @@ function createControlPlaneClient({
       if (claudeWeeklyResetsAt !== void 0) {
         body.claude_weekly_resets_at = claudeWeeklyResetsAt;
       }
-      const res = await req("POST", "/api/v1/weekly-tokens", body);
+      const res = await taskReq("POST", "/api/v1/weekly-tokens", body);
       if (res.status === 401) {
         cachedFirebaseToken = null;
         throw new Error("weekly-tokens unauthorized (401)");
@@ -283,9 +422,9 @@ function createControlPlaneClient({
      * authenticated operator so the web shows a TRUE "runner online" signal.
      * Best-effort caller; throws on 401/non-ok so the daemon can log + retry.
      */
-    async postHeartbeat({ runnerId: runnerId2, runnerInstanceId, operatorId, uptimeSec, activeTasks, maxConcurrency, effectiveConcurrency, measuredTaskSlots, measuredCpuSlots, measuredMemorySlots, version, daemonVersion, defaultAgent, supervisorInstanceId: supervisorInstanceId2, supervisorVersion: supervisorVersion2, supervisorCapabilities, servedRepos, servedOperators, availableAgents, accountUsage, availableLocalModels }) {
-      const body = { runner_id: runnerId2 };
-      if (runnerInstanceId) body.runner_instance_id = runnerInstanceId;
+    async postHeartbeat({ runnerId: runnerId3, runnerInstanceId: runnerInstanceId2, operatorId, uptimeSec, activeTasks, maxConcurrency, effectiveConcurrency, measuredTaskSlots, measuredCpuSlots, measuredMemorySlots, version, daemonVersion, defaultAgent, supervisorInstanceId: supervisorInstanceId2, supervisorVersion: supervisorVersion2, supervisorCapabilities, servedRepos, servedOperators, availableAgents, accountUsage, availableLocalModels }) {
+      const body = { runner_id: runnerId3 };
+      if (runnerInstanceId2) body.runner_instance_id = runnerInstanceId2;
       if (operatorId) body.operator_id = operatorId;
       if (typeof uptimeSec === "number") body.uptime_sec = uptimeSec;
       if (typeof activeTasks === "number") body.active_tasks = activeTasks;
@@ -322,7 +461,17 @@ function createControlPlaneClient({
         cachedFirebaseToken = null;
         throw new Error("heartbeat unauthorized (401)");
       }
-      if (!res.ok) throw new Error(`heartbeat failed: HTTP ${res.status}`);
+      if (!res.ok) {
+        let detail = "";
+        try {
+          const body2 = await res.json();
+          if (Array.isArray(body2?.issue_paths) && body2.issue_paths.length > 0) {
+            detail = ` (rejected fields: ${body2.issue_paths.join(", ")})`;
+          }
+        } catch {
+        }
+        throw new Error(`heartbeat failed: HTTP ${res.status}${detail}`);
+      }
       return res.json();
     },
     async getRunnerStatus({ operatorId } = {}) {
@@ -338,13 +487,13 @@ function createControlPlaneClient({
       const body = await res.json();
       return Array.isArray(body?.runners) ? body.runners : [];
     },
-    async pollRunnerControl({ runnerId: runnerId2, operatorId, supervisorInstanceId: supervisorInstanceId2, supervisorVersion: supervisorVersion2, capabilities }) {
-      const body = { runner_id: runnerId2 };
+    async pollRunnerControl({ runnerId: runnerId3, operatorId, supervisorInstanceId: supervisorInstanceId2, supervisorVersion: supervisorVersion2, capabilities }) {
+      const body = { runner_id: runnerId3 };
       if (operatorId) body.operator_id = operatorId;
       if (supervisorInstanceId2) body.supervisor_instance_id = supervisorInstanceId2;
       if (supervisorVersion2) body.supervisor_version = supervisorVersion2;
       if (Array.isArray(capabilities) && capabilities.length > 0) body.capabilities = capabilities;
-      const res = await req("POST", "/api/v1/runner/control/poll", body);
+      const res = await taskReq("POST", "/api/v1/runner/control/poll", body);
       if (res.status === 401) {
         cachedFirebaseToken = null;
         throw new Error("runner control poll unauthorized (401)");
@@ -354,14 +503,14 @@ function createControlPlaneClient({
       const action = json?.action;
       return action && typeof action.action_id === "string" && action.action_id ? { ...action, actionId: action.action_id } : null;
     },
-    async completeRunnerControl(actionId, { runnerId: runnerId2, operatorId, supervisorInstanceId: supervisorInstanceId2, supervisorVersion: supervisorVersion2, capabilities, status, detail }) {
-      const body = { runner_id: runnerId2, status };
+    async completeRunnerControl(actionId, { runnerId: runnerId3, operatorId, supervisorInstanceId: supervisorInstanceId2, supervisorVersion: supervisorVersion2, capabilities, status, detail }) {
+      const body = { runner_id: runnerId3, status };
       if (operatorId) body.operator_id = operatorId;
       if (supervisorInstanceId2) body.supervisor_instance_id = supervisorInstanceId2;
       if (supervisorVersion2) body.supervisor_version = supervisorVersion2;
       if (Array.isArray(capabilities) && capabilities.length > 0) body.capabilities = capabilities;
       if (detail) body.detail = detail;
-      const res = await req("POST", `/api/v1/runner/control/${encodeURIComponent(actionId)}/complete`, body);
+      const res = await taskReq("POST", `/api/v1/runner/control/${encodeURIComponent(actionId)}/complete`, body);
       if (res.status === 401) {
         cachedFirebaseToken = null;
         throw new Error("runner control completion unauthorized (401)");
@@ -372,7 +521,7 @@ function createControlPlaneClient({
     },
     /** Mint a GitHub App installation token — see installation-token.mjs. */
     async getInstallationToken({ required = false, readOnly = false, repo = null } = {}) {
-      return fetchInstallationToken({ req, required, readOnly, repo });
+      return fetchInstallationToken({ req: taskReq, required, readOnly, repo });
     },
     /**
      * Read the operator's dispatch-mode config (Fast→Ultracode effort setting).
@@ -381,7 +530,7 @@ function createControlPlaneClient({
      */
     async getDispatchMode() {
       try {
-        const res = await req("GET", "/api/v1/dispatch-mode-config");
+        const res = await taskReq("GET", "/api/v1/dispatch-mode-config");
         if (!res.ok) return "standard";
         const json = await res.json();
         return json?.dispatchMode || "standard";
@@ -2217,8 +2366,9 @@ async function main() {
   let child = null;
   let stopping = false;
   let handling = false;
+  let degraded = false;
   const respawn = () => {
-    if (!stopping && !handling && (!child || child.exitCode !== null)) child = launchChild();
+    if (!stopping && !degraded && !handling && (!child || child.exitCode !== null)) child = launchChild();
   };
   const launchChild = () => {
     const next = spawnChild(childEnv);
@@ -2241,8 +2391,11 @@ async function main() {
     launchPreviousChild: (entry) => spawnPreviousChild(entry, childEnv),
     log: (message) => console.error(`[vo-runner supervisor] ${message}`)
   })) {
-    stopping = true;
-    return;
+    degraded = true;
+    process.exitCode = 1;
+    await stopChild(child);
+    child = null;
+    console.error("[vo-runner supervisor] activation FAILED \u2014 entering degraded mode: not serving tasks, still polling for remote control actions");
   }
   const shutdown = async () => {
     if (stopping) return;
@@ -2308,6 +2461,7 @@ async function main() {
       child = launchChild();
       await sleep(CHILD_START_MS);
       if (child.exitCode !== null || !await waitForLocalRunner(child)) result.ok = false;
+      else degraded = false;
       await client.completeRunnerControl(action.actionId, {
         runnerId,
         ...operatorId ? { operatorId } : {},