@desplega.ai/agent-swarm 1.166.0 → 1.168.0

package/dist/{cli-b0f70z24.js → cli-mn8qkd8v.js}

@@ -1,6 +1,6 @@
 import {
   lazyApi
-} from "./cli-nexwh8zg.js";
+} from "./cli-g7k7zbcz.js";
 
 // node_modules/@earendil-works/pi-ai/dist/api/bedrock-converse-stream.lazy.js
 var __rewriteRelativeImportExtension = function(path, preserveJsx) {

package/dist/{cli-0qjcsxgz.js → cli-nctfv2yy.js}

@@ -1,10 +1,13 @@
 import {
   runStopHookSessionSummarySubprocess
-} from "./cli-rsy933yz.js";
+} from "./cli-bcspch6h.js";
 import {
   buildOtelTraceparentEnv,
   isHarnessOtelEnabled
 } from "./cli-q2j38vre.js";
+import {
+  fetchInstalledMcpServers
+} from "./cli-tng3131n.js";
 import {
   CTX_MODE_NUDGE_EVERY
 } from "./cli-zm7njwzd.js";
@@ -16,10 +19,7 @@ import {
   clampContextPercent,
   computeContextUsedUnified,
   getContextWindowSize
-} from "./cli-32bca7nr.js";
-import {
-  fetchInstalledMcpServers
-} from "./cli-tng3131n.js";
+} from "./cli-7avy4v8w.js";
 import {
   isClaudeBridgeEffective,
   resolveClaudeTransport
@@ -32,7 +32,7 @@ import {
 } from "./cli-2pdaysab.js";
 import {
   applyReasoningEffort
-} from "./cli-ttbsse3f.js";
+} from "./cli-tmhsxhb8.js";
 import {
   swarmRuntimeInstanceId
 } from "./cli-re8z5p8p.js";
@@ -44,7 +44,7 @@ import {
 import {
   init_file_lock,
   withFileLock
-} from "./cli-jhm6hzjn.js";
+} from "./cli-mxvnnv5q.js";
 import {
   CLAUDE_CREDENTIALS_HINT,
   init_credentials,
@@ -56,6 +56,10 @@ import {
   routeCredentialStatus,
   routeUnsetEnv
 } from "./cli-f9w1semm.js";
+import {
+  getOpenRouterAttributionHeaders,
+  init_openrouter_base_url
+} from "./cli-a35hzdve.js";
 import {
   init_env_flag,
   isEnvFlagEnabled
@@ -95,8 +99,74 @@ import { promisify } from "node:util";
 init_credentials();
 init_env_flag();
 init_file_lock();
+init_openrouter_base_url();
 init_secret_scrubber();
 
+// src/providers/claude-background-keepalive.ts
+var BACKGROUND_KEEPALIVE_INTERVAL_MS = 60000;
+var BACKGROUND_KEEPALIVE_MAX_IDLE_MS = 2 * 60 * 60000;
+
+class ClaudeBackgroundKeepalive {
+  options;
+  liveTaskIds = [];
+  timer;
+  lastMessageAt = Date.now();
+  stopped = false;
+  intervalMs;
+  maxIdleMs;
+  constructor(options) {
+    this.options = options;
+    this.intervalMs = options.intervalMs ?? BACKGROUND_KEEPALIVE_INTERVAL_MS;
+    this.maxIdleMs = options.maxIdleMs ?? BACKGROUND_KEEPALIVE_MAX_IDLE_MS;
+  }
+  observe(message) {
+    if (this.stopped)
+      return;
+    this.lastMessageAt = Date.now();
+    if (message.type === "system" && message.subtype === "background_tasks_changed") {
+      const tasks = Array.isArray(message.tasks) ? message.tasks : [];
+      this.liveTaskIds = tasks.filter((task) => task && task.ambient !== true && typeof task.task_id === "string").map((task) => task.task_id);
+    }
+    if (this.liveTaskIds.length > 0)
+      this.start();
+    else
+      this.clearTimer();
+  }
+  stop() {
+    this.stopped = true;
+    this.liveTaskIds = [];
+    this.clearTimer();
+  }
+  start() {
+    if (this.timer !== undefined)
+      return;
+    if (!this.options.apiUrl || !this.options.taskId)
+      return;
+    this.timer = setInterval(() => this.tick(), this.intervalMs);
+  }
+  tick() {
+    if (Date.now() - this.lastMessageAt > this.maxIdleMs) {
+      this.clearTimer();
+      return;
+    }
+    const { apiUrl, apiKey, agentId, taskId } = this.options;
+    fetch(`${apiUrl}/api/active-sessions/heartbeat/${encodeURIComponent(taskId)}`, {
+      method: "PUT",
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${apiKey}`,
+        "X-Agent-ID": agentId
+      }
+    }).catch(() => {});
+  }
+  clearTimer() {
+    if (this.timer === undefined)
+      return;
+    clearInterval(this.timer);
+    this.timer = undefined;
+  }
+}
+
 // src/providers/claude-session-events.ts
 init_secret_scrubber();
 function finiteNumber(value) {
@@ -628,6 +698,21 @@ function buildClaudeCodeRuntimeEnv(_sourceEnv) {
     DISABLE_BUG_COMMAND: "1"
   };
 }
+function claudeOpenRouterAttributionEnv(env) {
+  const headers = getOpenRouterAttributionHeaders(env.ANTHROPIC_BASE_URL ?? "", env);
+  const names = new Set(Object.keys(headers).map((name) => name.toLowerCase()));
+  if (names.size === 0)
+    return {};
+  const kept = (env.ANTHROPIC_CUSTOM_HEADERS ?? "").split(`
+`).filter((line) => {
+    const colon = line.indexOf(":");
+    const name = colon === -1 ? "" : line.slice(0, colon).trim().toLowerCase();
+    return line.trim() !== "" && !names.has(name);
+  });
+  const ours = Object.entries(headers).map(([name, value]) => `${name}: ${value}`);
+  return { ANTHROPIC_CUSTOM_HEADERS: [...kept, ...ours].join(`
+`) };
+}
 function buildClaudeSessionEnvironment(config, model, taskFilePath) {
   const sourceEnv = { ...withClaudeRouteEnv(config.env || process.env) };
   delete sourceEnv.AGENT_SWARM_CLAUDE_OAUTH_TOKEN;
@@ -636,6 +721,7 @@ function buildClaudeSessionEnvironment(config, model, taskFilePath) {
     env: {
       ENABLE_PROMPT_CACHING_1H: "1",
       ...sourceEnv,
+      ...claudeOpenRouterAttributionEnv(sourceEnv),
       ...buildClaudeCodeRuntimeEnv(sourceEnv),
       ...buildClaudeCodeOtelEnv(sourceEnv),
       ...reasoningApplication.kind === "claude-env" ? reasoningApplication.env : {},
@@ -709,6 +795,7 @@ class ClaudeSession {
   appliedReasoningEffort;
   lastAssistantText = "";
   transcript;
+  backgroundKeepalive;
   deliverSteering;
   constructor(config, model, taskFilePath, taskFileKey, sessionMcpConfig = null, claudeBinaryArgv = ["claude"], systemPromptFile = null, harnessVariant, harnessVariantMeta, queueSteeringSupported = false, runSessionSummary = runStopHookSessionSummarySubprocess) {
     this.config = config;
@@ -723,6 +810,7 @@ class ClaudeSession {
     this.contextWindowSize = getContextWindowSize(model);
     this.systemPromptFile = systemPromptFile;
     this.transcript = [`User: ${scrubSecrets(config.prompt)}`];
+    this.backgroundKeepalive = new ClaudeBackgroundKeepalive(config);
     const cmd = this.buildCommand();
     console.log(`\x1B[2m[${config.role}]\x1B[0m \x1B[36m▸\x1B[0m Spawning Claude (model: ${model}) for task ${config.taskId.slice(0, 8)}`);
     const sessionEnvironment = buildClaudeSessionEnvironment(config, model, taskFilePath);
@@ -897,6 +985,7 @@ class ClaudeSession {
     try {
       await Promise.all([stdoutPromise, stderrPromise]);
     } finally {
+      this.backgroundKeepalive.stop();
       this.closeStdin();
     }
     await logFileHandle.end();
@@ -945,6 +1034,7 @@ ${scrubSecrets(stderrOutput)}`);
       if (json.type === "result") {
         this.closeStdin();
       }
+      this.backgroundKeepalive.observe(json);
       const normalized = normalizeClaudeMessage(json, {
         taskId: this.config.taskId,
         agentId: this.config.agentId,
@@ -1026,7 +1116,7 @@ class ClaudeAdapter {
     if (transport === "sdk" && isClaudeBridgeEffective(sourceEnv)) {
       throw new Error("CLAUDE_TRANSPORT=sdk cannot run with an effective Claude Bridge configuration. Select CLI transport or disable the bridge.");
     }
-    const claudeSdk = transport === "sdk" ? await import("./claude-sdk-session-kkbxtw5q.js") : undefined;
+    const claudeSdk = transport === "sdk" ? await import("./claude-sdk-session-7x7chdrs.js") : undefined;
     claudeSdk?.validateClaudeSdkAdditionalArgs([
       ...claudeBinaryArgv.slice(1),
       ...config.additionalArgs ?? []
@@ -1133,4 +1223,4 @@ ${stderr}`.trim();
   }
 }
 
-export { normalizeClaudeMessage, checkClaudeCredentials, withClaudeRouteEnv, getTaskFilePath, cleanupTaskFile, parseClaudeBinary, resolveClaudeQueueSteeringOverride, resolveClaudeBinary, parseClaudeBridgeEnabled, resolveClaudeBridgeEnabled, resolveClaudeBinaryArgv, preseedClaudeTrustDialog, resolveClaudeTrustDirs, mergeMcpConfig, createSessionMcpConfig, buildClaudeCodeOtelEnv, buildClaudeCodeRuntimeEnv, buildClaudeSessionEnvironment, runClaudeSessionSummary, getSystemPromptFilePath, resolveClaudePrompt, ClaudeAdapter };
+export { ClaudeBackgroundKeepalive, normalizeClaudeMessage, checkClaudeCredentials, withClaudeRouteEnv, getTaskFilePath, cleanupTaskFile, parseClaudeBinary, resolveClaudeQueueSteeringOverride, resolveClaudeBinary, parseClaudeBridgeEnabled, resolveClaudeBridgeEnabled, resolveClaudeBinaryArgv, preseedClaudeTrustDialog, resolveClaudeTrustDirs, mergeMcpConfig, createSessionMcpConfig, buildClaudeCodeOtelEnv, buildClaudeCodeRuntimeEnv, claudeOpenRouterAttributionEnv, buildClaudeSessionEnvironment, runClaudeSessionSummary, getSystemPromptFilePath, resolveClaudePrompt, ClaudeAdapter };

package/dist/{cli-4r1sjfc4.js → cli-ng1pmdx6.js}

@@ -7,7 +7,7 @@ var package_default;
 var init_package = __esm(() => {
   package_default = {
     name: "@desplega.ai/agent-swarm",
-    version: "1.166.0",
+    version: "1.168.0",
     description: "agent-swarm.dev — an open-source operating system for AI work: a lead agent breaks goals into tasks, routes them to specialized workers such as Claude Code or Codex, runs each worker in an isolated container, and preserves shared memory, tools, schedules, and review gates so delegated work compounds across sessions.",
     keywords: [
       "agent-swarm.dev",
@@ -154,8 +154,8 @@ var init_package = __esm(() => {
       "start:portless": "portless api.swarm bun --expose-gc src/http.ts",
       inspector: "bunx @modelcontextprotocol/inspector --transport stdio bun src/stdio.ts",
       "inspector:http": "bunx @modelcontextprotocol/inspector --transport http https://api.swarm.localhost:1355/mcp",
-      lint: "biome check src apps/evals packages/ui-e2e packages/model-routing",
-      "lint:fix": "biome check --write src apps/evals packages/ui-e2e packages/model-routing",
+      lint: "biome check src apps/evals packages/ui-e2e packages/model-routing packages/model-catalog",
+      "lint:fix": "biome check --write src apps/evals packages/ui-e2e packages/model-routing packages/model-catalog",
       format: "biome format --write src",
       "build:binary": "bun build ./src/cli.tsx ./src/providers/pi-binary-root/src/extensions/codemode/worker.ts --root ./src/providers/pi-binary-root --compile --bytecode --format=esm --compile-exec-argv='--expose-gc' --target=bun-linux-x64 --outfile ./dist/agent-swarm",
       "build:binary:arm64": "bun build ./src/cli.tsx ./src/providers/pi-binary-root/src/extensions/codemode/worker.ts --root ./src/providers/pi-binary-root --compile --bytecode --format=esm --compile-exec-argv='--expose-gc' --target=bun-linux-arm64 --outfile ./dist/agent-swarm",
@@ -209,21 +209,21 @@ var init_package = __esm(() => {
     dependencies: {
       "@agentclientprotocol/sdk": "1.7.0",
       "@ai-sdk/openai": "^3.0.41",
-      "@anthropic-ai/claude-agent-sdk": "0.3.293",
-      "@anthropic-ai/sdk": "^0.132.0",
+      "@anthropic-ai/claude-agent-sdk": "0.3.295",
+      "@anthropic-ai/sdk": "^0.132.1",
       "@asteasolutions/zod-to-openapi": "^8.0.0",
       "@aws-sdk/client-bedrock": "3.1048.0",
       "@cursor/sdk": "1.0.37",
       "@desplega.ai/business-use": "^0.4.2",
       "@desplega.ai/localtunnel": "^2.2.0",
-      "@earendil-works/pi-agent-core": "^1.0.4",
-      "@earendil-works/pi-ai": "^1.0.4",
-      "@earendil-works/pi-coding-agent": "^1.0.4",
+      "@earendil-works/pi-agent-core": "^1.1.0",
+      "@earendil-works/pi-ai": "^1.1.0",
+      "@earendil-works/pi-coding-agent": "^1.1.0",
       "@inkjs/ui": "^2.0.0",
       "@jitl/quickjs-singlefile-mjs-release-sync": "0.32.0",
       "@linear/sdk": "^77.0.0",
-      "@modelcontextprotocol/sdk": "1.29.0",
-      "@openai/codex": "0.161.0",
+      "@modelcontextprotocol/sdk": "1.31.0",
+      "@openai/codex": "0.162.0",
       "@opencode-ai/sdk": "^1.18.35",
       "@openfort/openfort-node": "^0.9.1",
       "@opentelemetry/api": "^1.9.1",

package/dist/cli-nkqjj42q.js

@@ -0,0 +1,104 @@
+import {
+  isRecord,
+  parseClassifierUsage,
+  postClassifierRequest,
+  requiredNumber
+} from "./cli-q1vnz8wv.js";
+import {
+  formatProviderError,
+  normalizeProviderError
+} from "./cli-6map995v.js";
+
+// node_modules/@earendil-works/pi-ai/dist/api/system-one-shared.js
+function probabilities(label, value, id) {
+  if (!isRecord(value))
+    throw new Error(`${label} returned invalid probabilities for ${id}`);
+  return Object.fromEntries(Object.entries(value).map(([key, probability]) => [
+    key,
+    requiredNumber(label, probability, `probability for ${id}.${key}`)
+  ]));
+}
+function parseAnswers(label, value, context) {
+  if (!isRecord(value))
+    throw new Error(`${label} returned an unexpected response`);
+  const answers = [];
+  for (const [id, question] of Object.entries(context.questions)) {
+    const answer = value[id];
+    if (!isRecord(answer))
+      throw new Error(`${label} did not return an answer for ${id}`);
+    if (question.type === "choice") {
+      if (answer.type !== "choice" || typeof answer.choice !== "string") {
+        throw new Error(`${label} did not return a choice answer for ${id}`);
+      }
+      answers.push([
+        id,
+        {
+          type: "choice",
+          choice: answer.choice,
+          probabilities: probabilities(label, answer.probabilities, id),
+          confidence: requiredNumber(label, answer.confidence, `confidence for ${id}`)
+        }
+      ]);
+    } else if (question.type === "score") {
+      if (answer.type !== "score")
+        throw new Error(`${label} did not return a score answer for ${id}`);
+      answers.push([
+        id,
+        {
+          type: "score",
+          score: requiredNumber(label, answer.score, `score for ${id}`),
+          confidence: requiredNumber(label, answer.confidence, `confidence for ${id}`)
+        }
+      ]);
+    } else {
+      if (answer.type !== "noul")
+        throw new Error(`${label} did not return a bool answer for ${id}`);
+      answers.push([
+        id,
+        {
+          type: "bool",
+          probability: requiredNumber(label, answer.noul, `probability for ${id}`)
+        }
+      ]);
+    }
+  }
+  return Object.fromEntries(answers);
+}
+function wireRequest(context) {
+  return {
+    state: context.state,
+    questions: Object.fromEntries(Object.entries(context.questions).map(([id, question]) => [
+      id,
+      question.type === "bool" ? { ...question, type: "noul" } : question
+    ]))
+  };
+}
+async function classifySystemOne(transport, model, context, options) {
+  const output = {
+    api: model.api,
+    provider: model.provider,
+    model: model.id,
+    answers: {},
+    stopReason: "stop",
+    timestamp: Date.now()
+  };
+  try {
+    if (model.api !== transport.api)
+      throw new Error(`Unsupported classifier API: ${model.api}`);
+    if (context.images?.length)
+      throw new Error(`${transport.label} does not support image input`);
+    const body = await postClassifierRequest(transport.label, transport.url(model), model, transport.payload(model, wireRequest(context)), options);
+    const result = transport.output(body);
+    const usage = parseClassifierUsage(result.usage, model);
+    if (usage)
+      output.usage = usage;
+    output.answers = parseAnswers(transport.label, result.answers, context);
+    return output;
+  } catch (error) {
+    output.stopReason = options?.signal?.aborted ? "aborted" : "error";
+    output.errorMessage = formatProviderError(normalizeProviderError(error), `${transport.label} error`);
+    return output;
+  }
+}
+
+export { classifySystemOne };

package/dist/{cli-aqcey741.js → cli-p5jqk8fr.js}

@@ -1,7 +1,7 @@
 import {
   init_package,
   package_default
-} from "./cli-4r1sjfc4.js";
+} from "./cli-ng1pmdx6.js";
 
 // src/commands/shared/client-config.ts
 init_package();

package/dist/{cli-66zh26sp.js → cli-pgehz8e1.js}

@@ -18,13 +18,13 @@ import {
   recoverIncompleteRuns,
   repointTrackerSyncBySwarmId,
   resolveLeadOnlyRecoveryAssignment
-} from "./cli-df56tqdw.js";
+} from "./cli-ee0402n1.js";
 import {
   isMultiRuntimeEnabled
 } from "./cli-re8z5p8p.js";
 import {
   createApprovalFollowUpTask
-} from "./cli-jvj6a0cv.js";
+} from "./cli-g6gx2dn1.js";
 import {
   MAX_EMPTY_POLLS,
   assignUnassignedTaskPending,
@@ -75,7 +75,7 @@ import {
   settleSupersededTaskDependents,
   supersedeTask,
   updateAgentStatus
-} from "./cli-yshaazs1.js";
+} from "./cli-74ez9b66.js";
 import {
   TERMINAL_TASK_STATUSES,
   init_types
@@ -90,11 +90,18 @@ import {
   init_env_flag,
   isEnvFlagEnabled
 } from "./cli-ej6gbxfb.js";
+import {
+  init_package,
+  package_default
+} from "./cli-ng1pmdx6.js";
 import {
   init_secret_scrubber,
   scrubSecrets
 } from "./cli-wt7ccy4r.js";
 
+// src/heartbeat/heartbeat.ts
+init_package();
+
 // src/be/approval-sweeps.ts
 init_db();
 var DEFAULT_APPROVAL_REQUEST_AUTO_CANCELLATION_DAYS = 7;
@@ -356,6 +363,9 @@ function heartbeatChecklistIntervalMs() {
 function isHeartbeatChecklistDisabled() {
   return isEnvFlagEnabled("HEARTBEAT_CHECKLIST_DISABLE", false);
 }
+function isBootTriageAlways() {
+  return isEnvFlagEnabled("HEARTBEAT_BOOT_TRIAGE_ALWAYS", false);
+}
 var heartbeatInterval = null;
 var checklistInterval = null;
 var bootTriageTimeout = null;
@@ -1130,19 +1140,7 @@ async function gatherSystemStatus(options) {
       sections.push("- If the retry failed or the work is no longer needed, cancel it");
       sections.push("- Do NOT mark this boot triage as complete until all items are triaged");
     }
-    const orphanedTasks = [];
-    for (const status of ["pending", "offered"]) {
-      const tasks = await getTasksByStatus(status);
-      for (const task of tasks) {
-        const holderId = status === "offered" ? task.offeredTo : task.agentId;
-        if (!holderId)
-          continue;
-        const agent = agents.find((a) => a.id === holderId);
-        if (!agent || agent.status === "offline") {
-          orphanedTasks.push(task);
-        }
-      }
-    }
+    const orphanedTasks = await getOrphanedPendingOrOfferedTasks(agents);
     if (orphanedTasks.length > 0) {
       sections.push("");
       sections.push("## Orphaned Tasks [auto-generated, NEEDS ATTENTION]");
@@ -1160,6 +1158,32 @@ async function gatherSystemStatus(options) {
   return sections.join(`
 `);
 }
+async function getOrphanedPendingOrOfferedTasks(agents) {
+  const orphanedTasks = [];
+  for (const status of ["pending", "offered"]) {
+    const tasks = await getTasksByStatus(status);
+    for (const task of tasks) {
+      const holderId = status === "offered" ? task.offeredTo : task.agentId;
+      if (!holderId)
+        continue;
+      const agent = agents.find((a) => a.id === holderId);
+      if (!agent || agent.status === "offline") {
+        orphanedTasks.push(task);
+      }
+    }
+  }
+  return orphanedTasks;
+}
+async function getBootTriageFindings() {
+  const now = Date.now();
+  const agents = await getAllAgents();
+  return {
+    rebootInterrupted: getRebootAffectedTasks().length,
+    stalled: (await getStalledInProgressTasks(stallThresholdMinutes())).length,
+    orphaned: (await getOrphanedPendingOrOfferedTasks(agents)).length,
+    supersededWithoutResume: (await getSupersededTasksWithoutResume(new Date(now - ORPHAN_SUPERSEDE_MIN_AGE_MS).toISOString(), new Date(now - ORPHAN_SUPERSEDE_MAX_AGE_MS).toISOString())).length
+  };
+}
 async function checkHeartbeatChecklist() {
   const lead = await getLeadAgent();
   if (!lead)
@@ -1323,7 +1347,23 @@ async function createBootTriageTask() {
        LIMIT 1`, [lead.id, ...TERMINAL_TASK_STATUSES]);
   if (existing)
     return;
-  const systemStatus = await gatherSystemStatus({ isBootTriage: true });
+  if (!isBootTriageAlways()) {
+    const findings = await getBootTriageFindings().catch((err) => {
+      console.error("[Heartbeat] Boot triage findings failed, creating Lead task:", scrubSecrets(err instanceof Error ? err.message : String(err)));
+      return null;
+    });
+    if (findings && Object.values(findings).every((count) => count === 0)) {
+      console.log(`[Heartbeat] Clean boot (v${package_default.version}): 0 interrupted, stalled, orphaned, or resume-less superseded tasks; no boot-triage task for lead ${lead.name}`);
+      return;
+    }
+  }
+  const systemStatus = await gatherSystemStatus({ isBootTriage: true }).catch((err) => {
+    const reason = scrubSecrets(err instanceof Error ? err.message : String(err));
+    console.error("[Heartbeat] Boot triage status gathering failed:", reason);
+    return `## System status unavailable
+
+Status gathering failed after this boot (${reason}). Investigate the task, agent, and DB state directly.`;
+  });
   const result = resolveTemplate("heartbeat.boot-triage", {
     system_status: systemStatus,
     heartbeat_content: isEffectivelyEmpty(heartbeatMd) ? "_No standing orders configured._" : heartbeatMd

package/dist/cli-q1vnz8wv.js

@@ -0,0 +1,96 @@
+import {
+  retryProviderRequest
+} from "./cli-zy39d433.js";
+import {
+  headersToRecord,
+  providerHeadersToRecord
+} from "./cli-7kyk9jr3.js";
+import {
+  calculateCost
+} from "./cli-g7k7zbcz.js";
+
+// node_modules/@earendil-works/pi-ai/dist/api/classifier-shared.js
+function httpError(label, response, body) {
+  const error = new Error(`${label} returned ${response.status}`);
+  error.status = response.status;
+  error.headers = response.headers;
+  error.body = body;
+  return error;
+}
+function timeoutError(timeoutMs) {
+  const error = new Error(`Request timed out after ${timeoutMs}ms`);
+  error.name = "TimeoutError";
+  error.status = undefined;
+  error.headers = undefined;
+  error.body = "";
+  return error;
+}
+function isRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function requiredNumber(label, value, field) {
+  if (typeof value !== "number" || !Number.isFinite(value)) {
+    throw new Error(`${label} returned an invalid ${field}`);
+  }
+  return value;
+}
+function requestHeaders(model, apiKey, optionsHeaders) {
+  return providerHeadersToRecord({ authorization: `Bearer ${apiKey}`, "content-type": "application/json" }, model.headers, optionsHeaders) ?? {};
+}
+async function postClassifierRequest(label, url, model, body, options, noRetryStatuses) {
+  if (!options?.apiKey)
+    throw new Error(`No API key for provider: ${model.provider}`);
+  const apiKey = options.apiKey;
+  let payload = body;
+  const transformed = await options.onPayload?.(payload, model);
+  if (transformed !== undefined)
+    payload = transformed;
+  const requestFetch = options.fetch ?? globalThis.fetch;
+  const { response, json } = await retryProviderRequest(async () => {
+    const timeoutSignal = options.timeoutMs !== undefined ? AbortSignal.timeout(options.timeoutMs) : undefined;
+    const signal = options.signal && timeoutSignal ? AbortSignal.any([options.signal, timeoutSignal]) : options.signal ?? timeoutSignal;
+    try {
+      const next = await requestFetch(url, {
+        method: "POST",
+        headers: requestHeaders(model, apiKey, options.headers),
+        body: JSON.stringify(payload),
+        signal
+      });
+      if (!next.ok)
+        throw httpError(label, next, await next.text());
+      return { response: next, json: await next.json() };
+    } catch (error) {
+      if (timeoutSignal?.aborted && !options.signal?.aborted)
+        throw timeoutError(options.timeoutMs);
+      throw error;
+    }
+  }, {
+    maxRetries: options.maxRetries ?? 2,
+    maxRetryDelayMs: options.maxRetryDelayMs,
+    signal: options.signal,
+    noRetryStatuses
+  });
+  await options.onResponse?.({ status: response.status, headers: headersToRecord(response.headers) }, model);
+  return json;
+}
+function tokenCount(value) {
+  return typeof value === "number" && Number.isFinite(value) && value > 0 ? value : 0;
+}
+function parseClassifierUsage(value, model) {
+  if (!isRecord(value) || value.input_tokens === undefined && value.output_tokens === undefined)
+    return;
+  const input = tokenCount(value.input_tokens);
+  const output = tokenCount(value.output_tokens);
+  const usage = {
+    input,
+    output,
+    cacheRead: 0,
+    cacheWrite: 0,
+    totalTokens: input + output,
+    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
+  };
+  calculateCost(model, usage);
+  return usage;
+}
+
+export { isRecord, requiredNumber, postClassifierRequest, parseClassifierUsage };

package/dist/{cli-fj86xr3c.js → cli-qd97v6z2.js}

@@ -8,7 +8,7 @@ import {
   resolveGrammarConstrainedSampling,
   resolveJsonSchemaStrictSampling,
   transformMessages
-} from "./cli-8kt0bmgx.js";
+} from "./cli-xf713zz0.js";
 import {
   sanitizeSurrogates
 } from "./cli-y2apkfe2.js";
@@ -18,7 +18,7 @@ import {
   renderSystemMessageUpdate,
   resolveTranscript,
   resolveTranscriptTools
-} from "./cli-nexwh8zg.js";
+} from "./cli-g7k7zbcz.js";
 import {
   parseStreamingJson
 } from "./cli-1m2e086m.js";

package/dist/{cli-p4ztb80h.js → cli-r7eeddkq.js}

@@ -6,18 +6,20 @@ import {
 import {
   parseModelStr,
   resolveCredential
-} from "./cli-f2n17zhp.js";
-import {
-  DEFAULT_OPENROUTER_BASE_URL,
-  getOpenRouterBaseUrl
-} from "./cli-3dpj7atn.js";
+} from "./cli-7mhy3ths.js";
 import {
   complete,
   getBuiltinModel
-} from "./cli-pkke6geq.js";
+} from "./cli-km4x66j0.js";
 import {
   exports_typebox
 } from "./cli-rb9m6rpe.js";
+import {
+  DEFAULT_OPENROUTER_BASE_URL,
+  getOpenRouterAttributionHeaders,
+  getOpenRouterBaseUrl,
+  init_openrouter_base_url
+} from "./cli-a35hzdve.js";
 import {
   init_zod
 } from "./cli-sd0mxa3c.js";
@@ -31,6 +33,7 @@ import {
 
 // src/utils/internal-ai/complete-structured.ts
 init_zod();
+init_openrouter_base_url();
 init_secret_scrubber();
 var CLAUDE_CLI_TIMEOUT_MS = 30000;
 function stripJsonFences(raw) {
@@ -169,10 +172,12 @@ ${claudeUserPrompt}`, cred.modelDefault, opts.signal, jsonSchema);
     console.error(scrubSecrets(`internal-ai: unknown model ${provider}/${modelId} (callerTag=${callerTag})`));
     return null;
   }
-  const openRouterBaseUrl = getOpenRouterBaseUrl(opts.env ?? process.env);
+  const env = opts.env ?? process.env;
+  const openRouterBaseUrl = getOpenRouterBaseUrl(env);
   if (provider === "openrouter" && openRouterBaseUrl !== DEFAULT_OPENROUTER_BASE_URL) {
     model = { ...model, baseUrl: openRouterBaseUrl };
   }
+  const headers = provider === "openrouter" ? getOpenRouterAttributionHeaders(model.baseUrl, env) : {};
   const completeFn = opts._complete ?? complete;
   const toolChoice = model.api === "anthropic-messages" ? { type: "tool", name: opts.toolName } : "required";
   let userPrompt = opts.userPrompt;
@@ -189,7 +194,7 @@ ${claudeUserPrompt}`, cred.modelDefault, opts.signal, jsonSchema);
             parameters: opts.toolSchema
           }
         ]
-      }, { apiKey: cred.apiKey, signal: opts.signal, toolChoice });
+      }, { apiKey: cred.apiKey, signal: opts.signal, toolChoice, headers });
       const toolCall = msg.content.find((c) => c.type === "toolCall" && c.name === opts.toolName);
       if (!toolCall) {
         const assistantText = msg.content.filter((content) => content.type === "text").map((content) => content.text).join(`