@pensar/apex 2.5.0 → 2.6.1-canary.025b1ecf

package/src/core/ai/ai.ts

@@ -0,0 +1,2377 @@
+import { AsyncLocalStorage, AsyncResource } from "node:async_hooks";
+import { setTimeout as delay } from "node:timers/promises";
+import type { AnthropicMessagesModelId } from "@ai-sdk/anthropic/internal";
+import type { OpenAIResponsesProviderOptions } from "@ai-sdk/openai";
+import type { OpenAIChatModelId } from "@ai-sdk/openai/internal";
+import {
+  generateText,
+  type LanguageModel,
+  type LanguageModelMiddleware,
+  type ModelMessage,
+  Output,
+  type StopCondition,
+  type StreamTextOnFinishCallback,
+  type StreamTextOnStepFinishCallback,
+  type StreamTextResult,
+  streamText,
+  type TextStreamPart,
+  type ToolChoice,
+  type ToolSet,
+  wrapLanguageModel,
+} from "ai";
+import type { z } from "zod";
+import {
+  type CustomProviders,
+  parseCustomModelId,
+  resolveCustomModel,
+} from "../config/customProviders";
+import { type HoonifyModel, resolveHoonifyModel } from "../hoonify";
+import { createLogger } from "../logger/structured";
+import {
+  type AiTelemetryOperation,
+  createAiTelemetrySettings,
+  createGenerationSpanTracker,
+  type GenerationSpanTracker,
+  withModelCallDiagnostics,
+} from "../observability";
+import { RunPersistenceError } from "../runtime/persistenceError";
+import type { RunContextRecorder } from "../runtime/runContext";
+import { RunControlInterruption } from "../runtime/runControlStore";
+import { RunLimitError } from "../runtime/runModelStore";
+import { scopedLogger } from "../util/lazyLogger";
+import {
+  cacheBreakpointFor,
+  withCachedLastMessage,
+  withCachedSystemPrompt,
+} from "./caching";
+import {
+  beginCompaction,
+  type CompactionLink,
+  type CompactionTrigger,
+} from "./compactionTelemetry";
+import {
+  estimateMessageTokens,
+  fitMessagesToContext,
+  truncateWithMarker,
+} from "./contextManagement";
+import {
+  getInferenceRecorder,
+  type InferenceRecorder,
+  runWithInferenceRecorder,
+} from "./inference-attempt";
+import {
+  type ClaudeThinkingEffort,
+  getClaudeCapabilities,
+  getMaxOutputTokens,
+  getModelInfo,
+  prefersSequentialToolCalls,
+} from "./models";
+import {
+  runWithNativeRolloutOperation,
+  withNativeRolloutEvidenceModel,
+} from "./native-rollout-evidence";
+import { CONCENTRATE_GLM_5_3_MODEL_ID } from "./providers/concentrate";
+import { OPENAI_PRO_TIMEOUT_MS } from "./providers/openai-pro-fetch";
+import { STREAM_DEBUG } from "./streamTelemetry";
+import {
+  type AIAuthConfig,
+  checkIfContextLengthError,
+  createSummarizationStream,
+  getProviderModel,
+  isAnthropicProvider,
+} from "./utils";
+
+const log = scopedLogger(() => createLogger("ai"));
+
+// Opt-in debug logging for the empty-`{}` / looping response failure; bypasses `silent`.
+const RESPONSE_DEBUG =
+  process.env.RESPONSE_DEBUG === "1" || process.env.RESPONSE_DEBUG === "true";
+const RESPONSE_TOOL_NAME = "response";
+
+function nativeRolloutOperation(
+  operation?: AiTelemetryOperation,
+): "structured.generate" | "context.summarize" | "tool.repair" {
+  if (operation === "apex.context.summarize") return "context.summarize";
+  if (operation === "apex.tool.repair") return "tool.repair";
+  return "structured.generate";
+}
+
+/**
+ * Tools whose arguments carry a security-authorization boundary the model must
+ * never have re-synthesized by tool-call repair. Repair asks a model to invent
+ * schema-valid arguments from a failed call; for these tools a fabricated value
+ * (e.g. a default/localhost `target`) would silently bypass deterministic scope
+ * enforcement. Repair therefore FAILS CLOSED for them — it returns `null` and
+ * the original invalid call is never executed.
+ *
+ * Result-submission tools (`submit_results`, `document_vulnerability`) are
+ * deliberately NOT listed: they carry no authorization boundary, so repair is
+ * safe and desirable — it re-prompts the model to fix a mis-shaped field (e.g.
+ * `method` sent as an array) instead of dropping a completed recon or a
+ * confirmed finding.
+ */
+const REPAIR_FAIL_CLOSED_TOOLS: ReadonlySet<string> = new Set([
+  "spawn_pentest_agent",
+]);
+
+/**
+ * Whether tool-call repair must fail closed (no argument synthesis) for a tool.
+ * Exported for regression tests covering the repaired-invalid-call path.
+ */
+export function isRepairFailClosedTool(toolName: string): boolean {
+  return REPAIR_FAIL_CLOSED_TOOLS.has(toolName);
+}
+
+export type AIModel = AnthropicMessagesModelId | OpenAIChatModelId | string; // For gateway and Bedrock model IDs
+
+export type OpenAIReasoningEffort =
+  | "none"
+  | "low"
+  | "medium"
+  | "high"
+  | "xhigh"
+  | "max"
+  | "ultra";
+
+export const DEFAULT_OPENAI_REASONING_EFFORT: OpenAIReasoningEffort = "medium";
+
+/**
+ * Anthropic adaptive-thinking effort hint. Soft guidance for how much the model
+ * reasons (see AWS "adaptive thinking"): `low` minimizes thinking while
+ * `xhigh` and `max` allow progressively deeper work. Only meaningful for
+ * models that support adaptive thinking; ignored elsewhere.
+ */
+export type ThinkingEffort = ClaudeThinkingEffort;
+
+const OPENAI_REASONING_MODEL_IDS = new Set([
+  "gpt-6.1-sol",
+  "gpt-6-luna",
+  "gpt-6-sol",
+  "gpt-6-astra",
+  "gpt-5.5-pro",
+  "gpt-5.5-pro-2026-04-23",
+  "gpt-5.4-nano",
+  "gpt-5.4-nano-2026-03-17",
+  "gpt-5.4-mini",
+  "gpt-5.4-mini-2026-03-17",
+  "gpt-5",
+  "gpt-5-2025-08-07",
+  "gpt-5.1",
+  "gpt-5.1-2025-11-13",
+  "gpt-5.2",
+  "gpt-5.2-2025-12-11",
+  "gpt-5.2-pro",
+  "gpt-5.2-pro-2025-12-11",
+  "gpt-5.4",
+  "gpt-5.4-2026-03-05",
+  "gpt-5.4-pro",
+  "gpt-5.4-pro-2026-03-05",
+  "gpt-5.5",
+  "gpt-5.5-2026-04-23",
+  "gpt-5.6-sol",
+  "gpt-5.6-terra",
+  "gpt-5.6-luna",
+]);
+
+/** Per-step billing context attached to a usage report, attributing cost to a specific `(sessionId, stepSeq)`. */
+export interface UsageStepContext {
+  sessionId?: string;
+  stepSeq?: number;
+}
+
+/**
+ * Context attached to every usage report. The cache counters are always
+ * present — zero when the provider supplies none — and are already included
+ * in `inputTokens`:
+ * `uncachedInputTokens = inputTokens - cacheReadTokens - cacheWriteTokens`.
+ */
+export interface UsageCallbackContext extends UsageStepContext {
+  cacheReadTokens: number;
+  cacheWriteTokens: number;
+}
+
+/** Callback for reporting token usage from AI operations. */
+export type UsageCallback = (
+  modelId: string,
+  inputTokens: number,
+  outputTokens: number,
+  context: UsageCallbackContext,
+) => void | Promise<void>;
+
+/**
+ * Per-run usage recorder threaded through {@link StreamResponseOpts}. When set
+ * it replaces the ALS run sink and the ambient {@link onUsage} callback for
+ * that call. Async work is awaited before the next model step. Receives the
+ * same cache-aware {@link UsageCallbackContext} as the global callback, so
+ * the cached/uncached split travels on the usage event itself.
+ */
+export type UsageRecorder = (
+  model: string,
+  inputTokens: number,
+  outputTokens: number,
+  context?: UsageCallbackContext,
+) => void | Promise<void>;
+
+let _usageCallback: UsageCallback | null = null;
+
+/** Ambient default usage sink when no run-context sink is set. */
+export function onUsage(cb: UsageCallback | null): void {
+  _usageCallback = cb;
+}
+
+/** Normalized per-step usage: input/output plus cache buckets, provider-agnostic. */
+export interface NormalizedStepUsage {
+  inputTokens: number;
+  outputTokens: number;
+  cacheReadTokens: number;
+  cacheWriteTokens: number;
+}
+
+/**
+ * Extract normalized usage from one completed model step. A single function
+ * feeds both `onCacheMetrics` and the usage callback so provider metadata is
+ * never parsed twice. Anthropic's `providerMetadata.anthropic` wins; the
+ * SDK's normalized `usage.inputTokenDetails` is the fallback (Pensar
+ * provider). Input stays inclusive of cached tokens.
+ */
+export function normalizeStepUsage(step: {
+  usage?: {
+    inputTokens?: number;
+    outputTokens?: number;
+    inputTokenDetails?: {
+      cacheReadTokens?: number;
+      cacheWriteTokens?: number;
+    };
+  };
+  providerMetadata?: unknown;
+}): NormalizedStepUsage {
+  const inputTokens = step.usage?.inputTokens ?? 0;
+  const outputTokens = step.usage?.outputTokens ?? 0;
+
+  // Extract Anthropic cache metrics from providerMetadata (direct Anthropic / Bedrock SDK)
+  const meta = (
+    step.providerMetadata as
+      | { anthropic?: Record<string, unknown> }
+      | null
+      | undefined
+  )?.anthropic;
+  let cacheRead = (meta?.cacheReadInputTokens as number) ?? 0;
+  let cacheWrite = (meta?.cacheCreationInputTokens as number) ?? 0;
+
+  // Pensar provider doesn't populate providerMetadata.anthropic;
+  // fall back to the SDK's normalized usage.inputTokenDetails.
+  const details = step.usage?.inputTokenDetails;
+  if (cacheRead === 0) {
+    cacheRead = details?.cacheReadTokens ?? 0;
+  }
+  if (cacheWrite === 0) {
+    cacheWrite = details?.cacheWriteTokens ?? 0;
+  }
+
+  return {
+    inputTokens,
+    outputTokens,
+    cacheReadTokens: cacheRead,
+    cacheWriteTokens: cacheWrite,
+  };
+}
+
+// Per-run step counter for billing attribution, scoped via AsyncLocalStorage so
+// concurrent runs never race a shared index. Not a strict global sequence —
+// the DB `stepSeq` on `agent_messages` remains authoritative; this only labels
+// live usage reports.
+
+interface StepContext {
+  sessionId?: string;
+  /** Next step index to hand out; mutated in place as steps finish. */
+  next: number;
+  /** Per-run usage sink; wins over the ambient {@link onUsage} callback. */
+  usageSink?: UsageCallback;
+}
+
+const stepContextStore = new AsyncLocalStorage<StepContext>();
+
+/** Runs `fn` in a per-run step-counting context; `streamResponse` steps inside it report usage tagged with `sessionId` and an increasing `stepSeq` starting at `seedStepSeq`. */
+export function runWithStepContext<T>(
+  opts: {
+    sessionId?: string;
+    seedStepSeq?: number;
+    usageSink?: UsageCallback;
+  },
+  fn: () => T,
+): T {
+  const parent = stepContextStore.getStore();
+  return stepContextStore.run(
+    {
+      sessionId: opts.sessionId,
+      next: opts.seedStepSeq ?? 0,
+      usageSink: opts.usageSink ?? parent?.usageSink,
+    },
+    fn,
+  );
+}
+
+/** Takes and advances the next `stepSeq` for the current run; returns `undefined` outside any run context. */
+export function takeStepContext(): UsageStepContext | undefined {
+  const store = stepContextStore.getStore();
+  if (!store) return undefined;
+  const stepSeq = store.next;
+  store.next += 1;
+  return { sessionId: store.sessionId, stepSeq };
+}
+
+/**
+ * Report one finished model call to a usage sink. Advances `stepSeq` whenever
+ * a sink is present (including zero-token steps) so the index stays aligned
+ * with the AI SDK. Skips the callback itself when both token counts are 0.
+ */
+async function emitUsage(
+  model: string,
+  stepUsage: NormalizedStepUsage,
+  sink: UsageRecorder | UsageCallback | null | undefined,
+): Promise<void> {
+  if (!sink) return;
+  const stepCtx = takeStepContext();
+  if (stepUsage.inputTokens <= 0 && stepUsage.outputTokens <= 0) return;
+  await sink(model, stepUsage.inputTokens, stepUsage.outputTokens, {
+    ...stepCtx,
+    cacheReadTokens: stepUsage.cacheReadTokens,
+    cacheWriteTokens: stepUsage.cacheWriteTokens,
+  });
+}
+
+/** explicit recorder > ALS run sink > ambient {@link onUsage}. */
+function resolveUsageSink(
+  explicit?: UsageRecorder,
+): UsageRecorder | UsageCallback | null {
+  return explicit ?? stepContextStore.getStore()?.usageSink ?? _usageCallback;
+}
+
+export type AIModelProvider =
+  | "custom"
+  | "anthropic"
+  | "openai"
+  | "google"
+  | "openrouter"
+  | "concentrate"
+  | "bedrock"
+  | "bedrock-mantle"
+  | "pensar"
+  | "inception"
+  | "hoonify"
+  | "local";
+
+/** Conservative default when `getModelInfo` doesn't have a `contextLength`. */
+export function getContextWindow(
+  modelId: string,
+  customProviders?: CustomProviders,
+  hoonifyModels?: HoonifyModel[],
+): number {
+  if (modelId.startsWith("hoonify:"))
+    return resolveHoonifyModel(modelId, hoonifyModels).contextLength;
+  if (parseCustomModelId(modelId))
+    return resolveCustomModel(modelId, customProviders).model.contextLength;
+  return getModelInfo(modelId).contextLength ?? 200_000;
+}
+
+function checkIfRateLimitError(error: unknown): boolean {
+  const errObj =
+    typeof error === "object" && error !== null
+      ? (error as Record<string, unknown>)
+      : {};
+
+  // Check message — works for both Error instances and plain objects
+  // (Bedrock streaming errors are plain { message: "..." } records)
+  const errorMessage = (
+    typeof errObj.message === "string" ? errObj.message : ""
+  ).toLowerCase();
+  const errorCode = String(
+    typeof errObj.code === "string" ? errObj.code : "",
+  ).toLowerCase();
+  // AWS SDK errors surface the exception type via .name
+  const errorName = (
+    typeof errObj.name === "string" ? errObj.name : ""
+  ).toLowerCase();
+  // Bedrock HTTP-level errors include $metadata.httpStatusCode
+  const httpStatus =
+    typeof errObj.$metadata === "object" && errObj.$metadata !== null
+      ? (errObj.$metadata as Record<string, unknown>).httpStatusCode
+      : undefined;
+  // Stringified fallback for opaque error objects (e.g. Bedrock stream records)
+  const errorString =
+    errorMessage || errorName ? "" : String(error).toLowerCase();
+
+  return (
+    // Message-based detection
+    errorMessage.includes("too many tokens") ||
+    errorMessage.includes("rate limit") ||
+    errorMessage.includes("request rate") ||
+    errorMessage.includes("throttl") ||
+    errorMessage.includes("overloaded") ||
+    errorMessage.includes("too many requests") ||
+    errorMessage.includes("please wait") ||
+    errorMessage.includes("service unavailable") ||
+    // AWS SDK exception names (ThrottlingException, TooManyRequestsException)
+    errorName.includes("throttl") ||
+    errorName.includes("toomanyrequests") ||
+    errorName.includes("serviceunavailable") ||
+    // Error codes
+    errorCode === "rate_limit_exceeded" ||
+    errorCode === "throttling" ||
+    errorCode === "429" ||
+    // HTTP status from AWS SDK $metadata
+    httpStatus === 429 ||
+    httpStatus === 529 ||
+    httpStatus === 503 ||
+    // Fallback: stringified error for opaque Bedrock stream records
+    errorString.includes("throttl") ||
+    errorString.includes("too many") ||
+    errorString.includes("request rate")
+  );
+}
+
+// Injected into the system prompt for models that mis-parse parallel tool
+// calls (DeepSeek V3.1 on Bedrock). See prefersSequentialToolCalls.
+const SEQUENTIAL_TOOL_CALL_INSTRUCTION =
+  "CRITICAL TOOL-CALLING RULE: Emit EXACTLY ONE tool call per response. " +
+  "Never emit more than one tool call in a single turn. After each tool call, " +
+  "wait for its result before making the next one. Emitting multiple tool " +
+  "calls in one turn corrupts their arguments.";
+
+// Returns the system prompt that is actually sent to the provider, accounting
+// for the sequential-tool-call policy appended for models that mis-parse
+// parallel tool calls. Token budgeting (`fitMessagesToContext`) must use this
+// value, not the raw `system`, or it underestimates the system prompt size and
+// overestimates the message budget near the context limit.
+function applySequentialToolCallPolicy(
+  system: string | undefined,
+  tools: ToolSet | undefined,
+  model: AIModel,
+): string | undefined {
+  if (
+    tools &&
+    Object.keys(tools).length > 0 &&
+    prefersSequentialToolCalls(model)
+  ) {
+    return `${system ? `${system}\n\n` : ""}${SEQUENTIAL_TOOL_CALL_INSTRUCTION}`;
+  }
+  return system;
+}
+
+// The structured-output `response` tool is injected by the caller alongside its
+// own tools, so an allowlist of those tools must not hide it from the model.
+// Composed into `resolveEffectiveTools` at the call site so the executable map
+// keeps `response` even when a caller omits it from `activeTools`.
+export function withResponseToolActive(
+  tools: ToolSet | undefined,
+  activeTools: string[] | undefined,
+): string[] | undefined {
+  if (
+    !tools ||
+    !activeTools ||
+    activeTools.length === 0 ||
+    !(RESPONSE_TOOL_NAME in tools) ||
+    activeTools.includes(RESPONSE_TOOL_NAME)
+  ) {
+    return activeTools;
+  }
+  return [...activeTools, RESPONSE_TOOL_NAME];
+}
+
+/**
+ * Resolve the effective toolset from `activeTools` before any context
+ * fitting. The AI SDK treats `activeTools` as advertise-only — it still
+ * parses, executes, and enumerates every tool in the full map — so the
+ * executable map itself must be filtered here for budgeting, provider
+ * exposure, execution, and repair to agree. Built with DefineOwnProperty
+ * semantics so prototype-shaped names like `__proto__` survive as own keys.
+ * Feed it `withResponseToolActive(tools, activeTools)` so the injected
+ * structured-output tool survives the filter.
+ */
+export function resolveEffectiveTools(
+  tools: ToolSet | undefined,
+  activeTools: string[] | undefined,
+): Pick<StreamResponseOpts, "tools" | "activeTools"> {
+  if (!tools || !activeTools) return {};
+  if (activeTools.length === 0) return { tools: {}, activeTools: undefined };
+  const allow = new Set(activeTools);
+  return {
+    tools: Object.fromEntries(
+      Object.entries(tools).filter(([name]) => allow.has(name)),
+    ),
+    activeTools: undefined,
+  };
+}
+
+// Kept for callers/tests that filter a tool map without consuming `activeTools`
+// (e.g. `withResponseToolActive` composition). `resolveEffectiveTools` is the
+// stream-loop entry point; this is the same filter returning just the map.
+export function restrictToolsToActive(
+  tools: ToolSet | undefined,
+  activeTools: string[] | undefined,
+): ToolSet | undefined {
+  if (!tools || !activeTools || activeTools.length === 0) {
+    return tools;
+  }
+  const allow = new Set(activeTools);
+  const restricted: ToolSet = {};
+  for (const [name, tool] of Object.entries(tools)) {
+    if (allow.has(name)) {
+      restricted[name] = tool;
+    }
+  }
+  return restricted;
+}
+
+const MAX_RATE_LIMIT_RETRIES = 20;
+const MAX_IDLE_RESUME_RETRIES = 3;
+const STREAM_IDLE_TIMEOUT_MS = 5 * 60 * 1000; // 5 minutes
+
+// Observe the existing depth allowance only for accepted restarts.
+export async function recordContextRestart(
+  depth: number,
+  opts?: Pick<StreamResponseOpts, "inferenceRecorder">,
+): Promise<void> {
+  if (depth > MAX_RESTART_DEPTH) return;
+  await getInferenceRecorder(opts)?.retry({
+    authority: "context-restart",
+    count: depth,
+    maxRetries: MAX_RESTART_DEPTH,
+    delayMs: 0,
+  });
+}
+
+function getStreamIdleTimeoutMs(model: AIModel): number {
+  // Pro cannot stream progress while reasoning; wait for its completed response.
+  return /^(?:openai\/)?gpt-5\.5-pro(?:-|$)/.test(model)
+    ? OPENAI_PRO_TIMEOUT_MS
+    : STREAM_IDLE_TIMEOUT_MS;
+}
+
+class StreamIdleTimeoutError extends Error {
+  constructor(idleMs: number) {
+    super(`Stream idle for ${Math.round(idleMs / 1000)}s — no chunks received`);
+    this.name = "StreamIdleTimeoutError";
+  }
+}
+
+/**
+ * Wraps an async iterable with a per-chunk idle timeout. If no chunk arrives
+ * within `idleMs`, throws a {@link StreamIdleTimeoutError} so the caller can
+ * resume the conversation from accumulated messages.
+ *
+ * @param isIdleTimerActive Optional predicate evaluated before awaiting each
+ *   chunk. When it returns `false` the timer is skipped for that chunk and we
+ *   wait indefinitely. This lets callers suppress the timeout during expected
+ *   long pauses (e.g. while a tool call is executing) so a slow tool isn't
+ *   misread as a stalled provider stream.
+ */
+async function* withIdleTimeout<T>(
+  stream: AsyncIterable<T>,
+  idleMs: number,
+  isIdleTimerActive?: () => boolean,
+): AsyncGenerator<T> {
+  const iterator = stream[Symbol.asyncIterator]();
+  try {
+    while (true) {
+      const timerActive = isIdleTimerActive?.() ?? true;
+      let result: IteratorResult<T>;
+      if (timerActive) {
+        let timer: ReturnType<typeof setTimeout> | undefined;
+        result = await Promise.race([
+          iterator.next(),
+          new Promise<never>((_, reject) => {
+            timer = setTimeout(
+              () => reject(new StreamIdleTimeoutError(idleMs)),
+              idleMs,
+            );
+            if (typeof timer === "object" && "unref" in timer) timer.unref();
+          }),
+        ]);
+        clearTimeout(timer);
+      } else {
+        result = await iterator.next();
+      }
+
+      if (result.done) return;
+      yield result.value;
+    }
+  } finally {
+    // Best-effort cleanup — don't block on a stuck iterator. The underlying
+    // stream may be wedged (that's why we're here), so a hanging return()
+    // would deadlock the consumer.
+    iterator.return?.()?.catch?.(() => {});
+  }
+}
+
+function isStreamIdleTimeoutError(error: unknown): boolean {
+  // AI SDK may rewrap the error, so walk `cause`.
+  for (let e: unknown = error, depth = 0; e != null && depth < 6; depth++) {
+    if (e instanceof StreamIdleTimeoutError) return true;
+    e = (e as { cause?: unknown }).cause;
+  }
+  return false;
+}
+
+/**
+ * Gates the idle-timeout in {@link withIdleTimeout} so it only fires while
+ * we're waiting on the **model**, not while a tool is executing.
+ *
+ * The provider stream emits nothing between a `tool-call` chunk and the
+ * matching `tool-result`. A slow tool (e.g. `run_attack_surface`, which
+ * spawns a multi-minute sub-agent) would otherwise be misread as a stalled
+ * stream — triggering the auto-resume path and causing the model to re-issue
+ * the same tool call, which spawns a duplicate sub-agent.
+ *
+ * Parallel tool calls are tracked by `toolCallId`. Duplicate `tool-call`
+ * chunks for the same id (defensive — should never happen in practice) are
+ * deduplicated. Terminal chunks (`tool-result` / `tool-error`) without a
+ * matching open call are ignored so the counter can never go negative.
+ *
+ * EXCEPTION: `response` is never gated — its `execute` returns synchronously so
+ * it's not a slow tool, but a Bedrock stream wedging mid-payload while
+ * streaming its (often large) args would otherwise suppress the idle timeout
+ * forever (confirmed hang). Leaving it ungated keeps idle-timeout auto-resume working.
+ */
+const NON_GATED_TOOLS = new Set<string>(["response"]);
+
+function createToolExecutionGate(): {
+  shouldEnforceIdleTimeout: () => boolean;
+  observe: (chunk: {
+    type: string;
+    toolCallId?: string;
+    toolName?: string;
+  }) => void;
+} {
+  let inFlight = 0;
+  const open = new Set<string>();
+
+  return {
+    shouldEnforceIdleTimeout: () => inFlight === 0,
+    observe: (chunk) => {
+      if (chunk.type === "tool-call") {
+        if (chunk.toolName && NON_GATED_TOOLS.has(chunk.toolName)) return;
+        const id = chunk.toolCallId;
+        if (id) {
+          if (!open.has(id)) {
+            open.add(id);
+            inFlight++;
+          }
+        } else {
+          inFlight++;
+        }
+      } else if (chunk.type === "tool-result" || chunk.type === "tool-error") {
+        // A failed tool is just as terminal for the gate as a successful one —
+        // without this, a tool-error leaves its slot open forever, suppressing
+        // the idle timeout for the rest of the run.
+        const id = chunk.toolCallId;
+        if (id) {
+          if (open.delete(id)) {
+            inFlight = Math.max(0, inFlight - 1);
+          }
+        } else if (!(chunk.toolName && NON_GATED_TOOLS.has(chunk.toolName))) {
+          inFlight = Math.max(0, inFlight - 1);
+        }
+      }
+    },
+  };
+}
+
+// Helper function to wrap a stream with error handling for async errors
+function wrapStreamWithErrorHandler(
+  originalStream: StreamTextResult<ToolSet, never>,
+  messagesContainer: { current: ModelMessage[] },
+  opts: StreamResponseOpts,
+  model: LanguageModel,
+  silent?: boolean,
+  rateLimitRetryCount: number = 0,
+  idleResumeCount: number = 0,
+  generationSpans?: GenerationSpanTracker,
+): StreamTextResult<ToolSet, never> {
+  // Create a lazy getter for fullStream that wraps it with error handling
+  let wrappedStream: AsyncIterable<TextStreamPart<ToolSet>> | null = null;
+
+  const handler = {
+    get(_target: StreamTextResult<ToolSet, never>, prop: string | symbol) {
+      // Intercept access to fullStream
+      if (prop === "fullStream") {
+        if (!wrappedStream) {
+          // Tool-execution gate: keeps the idle timer paused while a tool
+          // is in flight (see {@link createToolExecutionGate}). The gate's
+          // predicate is read by `withIdleTimeout` at the top of each
+          // iteration — which always runs AFTER the previous chunk's body
+          // finished mutating the gate, so observations are always visible
+          // before the next idle-race begins.
+          const toolGate = createToolExecutionGate();
+
+          wrappedStream = (async function* () {
+            let streamError: unknown;
+            try {
+              for await (const chunk of withIdleTimeout(
+                originalStream.fullStream,
+                getStreamIdleTimeoutMs(opts.model),
+                toolGate.shouldEnforceIdleTimeout,
+              )) {
+                toolGate.observe(chunk);
+
+                // [response-debug] Log AI-SDK-level response-tool chunks / finish-step, below the consume() loop.
+                if (RESPONSE_DEBUG) {
+                  const c = chunk as {
+                    type: string;
+                    toolName?: string;
+                    toolCallId?: string;
+                    finishReason?: string;
+                    invalid?: boolean;
+                  };
+                  if (
+                    (c.type === "tool-call" ||
+                      c.type === "tool-error" ||
+                      c.type === "tool-result") &&
+                    c.toolName === RESPONSE_TOOL_NAME
+                  ) {
+                    log.warn(
+                      `[response-debug] ai-layer chunk=${c.type} response id=${c.toolCallId} invalid=${c.invalid === true}`,
+                    );
+                  } else if (c.type === "finish-step") {
+                    log.warn(
+                      `[response-debug] ai-layer finish-step finishReason=${c.finishReason}`,
+                    );
+                  }
+                }
+
+                // Only stream-level errors are fatal; tool-level errors
+                // flow through so the UI renders them as failed tool results.
+                // Instead of throwing mid-stream, capture the error and drain
+                // the tail (finish-step/finish): the AI SDK's terminal
+                // lifecycle — root-span end included — only runs when the
+                // stream completes, and aborting the iteration leaks it.
+                if (chunk.type === "error") {
+                  streamError = (chunk as unknown as { error: unknown }).error;
+                  // Mark the root generation span failed NOW: the SDK's
+                  // flush ends it mid-drain, before the loop exits — a
+                  // later mark is a no-op on an ended span.
+                  generationSpans?.markFailed(streamError);
+                  continue;
+                }
+                // Drain the post-error tail silently: the terminal parts
+                // must be consumed for the SDK's finalization, but consumers
+                // must not observe step-close-out after a fatal error.
+                if (streamError !== undefined) {
+                  continue;
+                }
+
+                yield chunk;
+              }
+              if (streamError !== undefined) {
+                // The tail is consumed but the SDK's finalization
+                // (root-span end) settles with the response promise —
+                // await it so the error never overtakes the span lifecycle.
+                await Promise.resolve(originalStream.response).catch(() => {});
+                throw streamError;
+              }
+            } catch (caughtError) {
+              // A tail-drain failure must not replace the provider error that
+              // put the wrapper into drain mode.
+              const error =
+                streamError !== undefined ? streamError : caughtError;
+              // Error-part runs complete normally inside the SDK, so nothing
+              // sets the root generation span failed — mark it here (the
+              // SDK's own thrown-error marking is guarded against below).
+              generationSpans?.markFailed(error);
+
+              const errorMessage =
+                error instanceof Error ? error.message : String(error);
+
+              // SDK RetryError can wrap a critical failure after a prior provider retry.
+              await getInferenceRecorder(opts)?.flush();
+
+              // Check context length FIRST — these should never be retried
+              // as-is; the prompt must be reduced via summarization.
+              const isCtxError = checkIfContextLengthError(error);
+
+              // A latched persistence failure is terminal: retrying would
+              // dispatch a turn whose context was never durably committed.
+              if (
+                error instanceof RunPersistenceError ||
+                error instanceof RunLimitError ||
+                error instanceof RunControlInterruption
+              ) {
+                throw error;
+              }
+
+              // Handle stream idle timeout — resume from accumulated messages
+              if (
+                !isCtxError &&
+                isStreamIdleTimeoutError(error) &&
+                idleResumeCount < MAX_IDLE_RESUME_RETRIES &&
+                messagesContainer.current.length > 0
+              ) {
+                const nextIdleCount = idleResumeCount + 1;
+                await getInferenceRecorder(opts)?.retry({
+                  authority: "stream-idle",
+                  count: nextIdleCount,
+                  maxRetries: MAX_IDLE_RESUME_RETRIES,
+                  delayMs: 0,
+                });
+                // Surfaced on silent sub-agents too when stream-debugging.
+                if (!silent || STREAM_DEBUG) {
+                  log.warn(
+                    `Stream stalled (attempt ${nextIdleCount}/${MAX_IDLE_RESUME_RETRIES}), resuming with ${messagesContainer.current.length} messages: ${errorMessage}`,
+                  );
+                }
+
+                const retriedStream = streamResponse({
+                  ...opts,
+                  messages: messagesContainer.current,
+                });
+
+                const wrappedRetriedStream = wrapStreamWithErrorHandler(
+                  retriedStream,
+                  messagesContainer,
+                  opts,
+                  model,
+                  silent,
+                  rateLimitRetryCount,
+                  nextIdleCount,
+                );
+
+                for await (const chunk of wrappedRetriedStream.fullStream) {
+                  yield chunk;
+                }
+                return;
+              }
+
+              // Handle rate limit errors with exponential backoff retry
+              if (
+                !isCtxError &&
+                checkIfRateLimitError(error) &&
+                rateLimitRetryCount < MAX_RATE_LIMIT_RETRIES
+              ) {
+                const nextRetryCount = rateLimitRetryCount + 1;
+                const delayMs = Math.min(1000 * nextRetryCount, 30000);
+                const recorder = getInferenceRecorder(opts);
+                await recorder?.retry({
+                  authority: "stream-rate-limit",
+                  count: nextRetryCount,
+                  maxRetries: MAX_RATE_LIMIT_RETRIES,
+                  delayMs,
+                });
+
+                if (!silent) {
+                  log.warn(
+                    `Rate limit error (attempt ${nextRetryCount}/${MAX_RATE_LIMIT_RETRIES}), waiting ${delayMs}ms: ${errorMessage}`,
+                  );
+                }
+
+                if (recorder) {
+                  await delay(delayMs, undefined, { signal: opts.abortSignal });
+                } else {
+                  await new Promise((resolve) => setTimeout(resolve, delayMs));
+                }
+
+                const retriedStream = streamResponse({
+                  ...opts,
+                  messages:
+                    messagesContainer.current.length > 0
+                      ? messagesContainer.current
+                      : undefined,
+                });
+
+                const wrappedRetriedStream = wrapStreamWithErrorHandler(
+                  retriedStream,
+                  messagesContainer,
+                  opts,
+                  model,
+                  silent,
+                  nextRetryCount,
+                  idleResumeCount,
+                );
+
+                for await (const chunk of wrappedRetriedStream.fullStream) {
+                  yield chunk;
+                }
+                return;
+              }
+
+              if (isCtxError) {
+                let currentMessages: ModelMessage[] = messagesContainer.current;
+                try {
+                  const response = await originalStream.response;
+                  if (response.messages && response.messages.length > 0) {
+                    currentMessages = response.messages as ModelMessage[];
+                  }
+                } catch {
+                  // Fall back to container messages.
+                }
+
+                const fitted = fitMessagesToContext(currentMessages, {
+                  contextWindow: getContextWindow(
+                    opts.model,
+                    opts.authConfig?.customProviders,
+                    opts.authConfig?.hoonifyModels,
+                  ),
+                  maxOutputTokens: getMaxOutputTokens(
+                    opts.model,
+                    opts.authConfig?.customProviders,
+                    opts.authConfig?.hoonifyModels,
+                  ),
+                  system: applySequentialToolCallPolicy(
+                    opts.system,
+                    opts.tools,
+                    opts.model,
+                  ),
+                  tools: opts.tools,
+                  sessionPath: opts.sessionPath,
+                  telemetry: {
+                    trigger: "context_overflow",
+                    model: opts.model,
+                    sessionId: opts.sessionId,
+                    restartDepth: opts._restartDepth,
+                    previous: opts._compaction?.last,
+                  },
+                });
+
+                if (fitted.compaction && opts._compaction)
+                  opts._compaction.last = fitted.compaction;
+
+                messagesContainer.current = fitted.messages;
+
+                // Track depth across the reactive paths so that a failed
+                // Layer 1+2 retry CONTRIBUTES to the Layer 3 budget too.
+                // Without this, a failed retry's increment is "lost" on
+                // fall-through (Layer 3 reads outer `opts._restartDepth`,
+                // not the post-retry value) and recovery can burn close
+                // to double the intended provider calls before
+                // `MAX_RESTART_DEPTH` finally trips.
+                let postReactiveDepth = opts._restartDepth ?? 0;
+
+                if (fitted.fitsBudget && fitted.modified) {
+                  if (!silent) {
+                    log.warn(
+                      `Context length error — Layer 1+2 reduced to ~${fitted.estimatedInputTokens} tokens, retrying`,
+                    );
+                  }
+                  try {
+                    // Bump `_restartDepth` so a tokenizer-drift loop where
+                    // every reactive retry trips the same provider rejection
+                    // (fitsBudget=true by our estimator, still rejected by
+                    // the provider) eventually exhausts the depth bound at
+                    // the top of `streamResponse` instead of looping
+                    // forever. Without this, only Layer-3 escalation
+                    // increments depth and a drift-driven loop can burn
+                    // arbitrarily many failed provider calls.
+                    await recordContextRestart(postReactiveDepth + 1, opts);
+                    const retried = streamResponse({
+                      ...opts,
+                      messages: fitted.messages,
+                      _restartDepth: postReactiveDepth + 1,
+                    });
+                    const wrappedRetry = wrapStreamWithErrorHandler(
+                      retried,
+                      messagesContainer,
+                      opts,
+                      model,
+                      silent,
+                      rateLimitRetryCount,
+                      idleResumeCount,
+                    );
+                    for await (const chunk of wrappedRetry.fullStream) {
+                      yield chunk;
+                    }
+                    return;
+                  } catch (retryError) {
+                    // Always propagate the typed terminal error and
+                    // cancellations — these MUST escape the cascade
+                    // immediately. Currently `ContextLengthExhaustedError`
+                    // propagates only by accident (its message
+                    // "Context-length recovery exhausted…" uses a hyphen
+                    // that doesn't match any `checkIfContextLengthError`
+                    // pattern); the explicit `instanceof` and abort
+                    // checks are defensive against a future message-text
+                    // tweak that would otherwise silently swallow
+                    // recovery exhaustion. Mirrors the Layer 3
+                    // `summarizeError` catch below for symmetry.
+                    if (
+                      retryError instanceof ContextLengthExhaustedError ||
+                      (retryError instanceof Error &&
+                        retryError.name === "AbortError") ||
+                      opts.abortSignal?.aborted ||
+                      !checkIfContextLengthError(retryError)
+                    ) {
+                      throw retryError;
+                    }
+                    // Account for the depth slot the failed retry just
+                    // consumed so the Layer 3 paths below don't grant a
+                    // fresh budget on top of the already-spent cycle.
+                    postReactiveDepth += 1;
+                  }
+                }
+
+                // Layer 3: full summarization. Wrapped because
+                // `summarizeConversation` itself calls `generateText` and
+                // can throw — fall back to a minimal-context restart.
+                const messagesForSummary = messagesContainer.current;
+                if (!silent) {
+                  log.warn(
+                    `Context length error — summarizing ${messagesForSummary.length} messages`,
+                  );
+                }
+
+                try {
+                  const summarizationStream = createSummarizationStream(
+                    messagesForSummary,
+                    {
+                      ...opts,
+                      _restartDepth: postReactiveDepth,
+                      _compactionTrigger: "context_overflow",
+                    },
+                    model,
+                  );
+                  for await (const chunk of summarizationStream.fullStream) {
+                    yield chunk;
+                  }
+                } catch (summarizeError) {
+                  // Only fall through to minimal-restart for context-length
+                  // errors. For everything else — auth failures, network
+                  // errors, abort signals, the typed terminal
+                  // `ContextLengthExhaustedError`, or non-`Error` throws —
+                  // propagate immediately. Without this, an aborted call
+                  // would spawn a new provider request, and a transient
+                  // auth failure would burn an extra cycle that fails
+                  // identically.
+                  if (
+                    summarizeError instanceof ContextLengthExhaustedError ||
+                    (summarizeError instanceof Error &&
+                      summarizeError.name === "AbortError") ||
+                    opts.abortSignal?.aborted ||
+                    !checkIfContextLengthError(summarizeError)
+                  ) {
+                    throw summarizeError;
+                  }
+                  if (!silent) {
+                    const sumMsg =
+                      summarizeError instanceof Error
+                        ? summarizeError.message
+                        : String(summarizeError);
+                    log.warn(
+                      `Layer 3 summarization failed (${sumMsg}). Falling back to minimal-context restart.`,
+                    );
+                  }
+                  opts.onSummarized?.("");
+                  const minimalPrompt =
+                    typeof opts.prompt === "string" && opts.prompt.length > 0
+                      ? opts.prompt.slice(0, 4_000)
+                      : MINIMAL_RESTART_PROMPT;
+                  const reset = beginCompaction("reset", {
+                    trigger: "summary_overflow",
+                    model: opts.model,
+                    sessionId: opts.sessionId,
+                    restartDepth: postReactiveDepth,
+                    previous: opts._compaction?.last,
+                  });
+                  reset?.capture("before", () => ({
+                    messages: messagesForSummary,
+                    system: opts.system,
+                    prompt: opts.prompt,
+                  }));
+                  reset?.measure(
+                    "before",
+                    () => estimateMessageTokens(messagesForSummary),
+                    messagesForSummary.length,
+                  );
+                  reset?.measure(
+                    "after",
+                    () =>
+                      estimateMessageTokens([
+                        { role: "user", content: minimalPrompt },
+                      ]),
+                    1,
+                  );
+                  reset?.capture("after", () => ({
+                    messages: [{ role: "user", content: minimalPrompt }],
+                    system: opts.system,
+                  }));
+                  reset?.finish("completed");
+                  if (reset && opts._compaction)
+                    opts._compaction.last = reset.link;
+                  await recordContextRestart(postReactiveDepth + 1, opts);
+                  const fallback = streamResponse({
+                    ...opts,
+                    prompt: minimalPrompt,
+                    messages: undefined,
+                    _restartDepth: postReactiveDepth + 1,
+                  });
+                  for await (const chunk of fallback.fullStream) {
+                    yield chunk;
+                  }
+                }
+              } else {
+                if (!silent) {
+                  log.error("Non-recoverable stream error, re-throwing", {
+                    error: errorMessage,
+                  });
+                }
+                throw error;
+              }
+            }
+          })();
+        }
+        return wrappedStream;
+      }
+
+      // For all other properties, return the original
+      return (originalStream as unknown as Record<string | symbol, unknown>)[
+        prop
+      ];
+    },
+  };
+
+  return new Proxy(originalStream, handler);
+}
+
+// Available models with names
+export interface ModelInfo {
+  id: AIModel;
+  name: string;
+  provider: AIModelProvider;
+  contextLength?: number;
+}
+
+export function modelRequiresThinking(modelId: string): boolean {
+  return getClaudeCapabilities(modelId)?.alwaysOnThinking ?? false;
+}
+
+/**
+ * Check whether a model supports extended thinking based on its ID.
+ *
+ * Thinking is available on Claude 3.7 Sonnet and all Claude 4.x+ models
+ * (Sonnet, Opus, Haiku 4.5). Works for Anthropic direct, Bedrock, and
+ * Pensar provider IDs.
+ */
+export function modelSupportsThinking(modelId: string): boolean {
+  if (getClaudeCapabilities(modelId)) return true;
+  // Normalize: strip provider prefixes so we match the base Claude model ID
+  const normalized = modelId
+    .replace(/^pensar:/, "")
+    .replace(/^(us\.|eu\.|global\.|ap\.)?anthropic\./, "");
+
+  return (
+    /^claude-3-7-sonnet/.test(normalized) ||
+    /^claude-sonnet-4/.test(normalized) ||
+    /^claude-opus-4/.test(normalized) ||
+    /^claude-haiku-4-5/.test(normalized)
+  );
+}
+
+/**
+ * Whether a model accepts the `adaptive` thinking type — Claude Opus 4.6+ and
+ * Sonnet 4.6+. Pre-4.6 models (Opus 4.0–4.5, Sonnet 4.5, Haiku 4.5) support
+ * extended thinking but reject `adaptive`, so we don't request thinking for
+ * them at all.
+ */
+export function modelSupportsAdaptiveThinking(modelId: string): boolean {
+  if (getClaudeCapabilities(modelId)) return true;
+  const normalized = modelId
+    .replace(/^pensar:/, "")
+    .replace(/^(us\.|eu\.|global\.|ap\.)?anthropic\./, "");
+
+  return (
+    /^claude-sonnet-4-[6-9]/.test(normalized) ||
+    /^claude-opus-4-[6-9]/.test(normalized)
+  );
+}
+
+export function modelSupportsOpenAIReasoning(modelId: string): boolean {
+  if (modelId === CONCENTRATE_GLM_5_3_MODEL_ID) return true;
+  const { provider } = getModelInfo(modelId);
+  if (!["openai", "openrouter", "concentrate"].includes(provider)) return false;
+  modelId = modelId.replace(/^(openai\/|concentrate:)/, "");
+  return (
+    OPENAI_REASONING_MODEL_IDS.has(modelId) || /^o[134](?:\b|-)/.test(modelId)
+  );
+}
+
+export function modelRequiresReasoning(modelId: string): boolean {
+  return modelId === CONCENTRATE_GLM_5_3_MODEL_ID;
+}
+
+export function getOpenAIReasoningEfforts(
+  modelId: string,
+): OpenAIReasoningEffort[] {
+  if (!modelSupportsOpenAIReasoning(modelId)) return [];
+  if (modelId === CONCENTRATE_GLM_5_3_MODEL_ID) {
+    return ["low", "high", "max"];
+  }
+  if (/^concentrate:gpt-5\.4-(?:mini|nano)$/.test(modelId)) {
+    return ["none", "low", "medium", "high"];
+  }
+  modelId = modelId.replace(/^(openai\/|concentrate:)/, "");
+  if (modelId === "gpt-6-sol" || modelId === "gpt-6-luna") {
+    return ["none", "low", "medium", "high", "xhigh", "max"];
+  }
+  if (/^gpt-6[.-]/.test(modelId)) {
+    return ["low", "medium", "high", "xhigh", "max"];
+  }
+  if (/^gpt-5\.5-pro(?:-|$)/.test(modelId)) {
+    return ["medium", "high", "xhigh"];
+  }
+  if (/^gpt-5\.4-(?:mini|nano)(?:-|$)/.test(modelId)) {
+    return ["none", "low", "medium", "high", "xhigh"];
+  }
+  if (/^gpt-5\.6(?:\b|-)/.test(modelId)) {
+    return ["low", "medium", "high", "xhigh", "max", "ultra"];
+  }
+  if (/^gpt-5\.5(?:\b|-)/.test(modelId)) {
+    return ["none", "low", "medium", "high", "xhigh"];
+  }
+  return ["low", "medium", "high"];
+}
+
+export function normalizeOpenAIReasoningEffort(
+  modelId: string,
+  effort?: OpenAIReasoningEffort | null,
+): OpenAIReasoningEffort | undefined {
+  if (modelId === CONCENTRATE_GLM_5_3_MODEL_ID) {
+    switch (effort) {
+      case "none":
+      case "low":
+        return "low";
+      case "medium":
+      case "high":
+        return "high";
+      case "xhigh":
+      case "max":
+      case "ultra":
+        return "max";
+      default:
+        return "high";
+    }
+  }
+  const efforts = getOpenAIReasoningEfforts(modelId);
+  if (efforts.length === 0) return undefined;
+  if (effort && efforts.includes(effort)) {
+    // ponytail: OpenAI has no distinct ultra level; keep the UI choice as a max alias.
+    return effort === "ultra" ? "max" : effort;
+  }
+  return DEFAULT_OPENAI_REASONING_EFFORT;
+}
+
+/**
+ * Provider-specific options passed to `streamText` to request reasoning /
+ * extended thinking. Each provider reads only its own namespace:
+ *   - `anthropic.thinking` — direct Anthropic provider AND the Pensar gateway
+ *     (the gateway formatter maps it to the Bedrock `thinking` field).
+ *   - `bedrock.reasoningConfig` — the AWS Bedrock provider; it ignores the
+ *     `anthropic` namespace entirely, so Anthropic-on-Bedrock thinking is only
+ *     requested when this key is present. `display: 'summarized'` makes the
+ *     model actually return reasoning text (adaptive can otherwise omit it).
+ *   - `openai.reasoningEffort` — OpenAI/o-series reasoning models.
+ */
+export type ReasoningProviderOptions = {
+  openrouter?: {
+    reasoning: { effort: OpenAIReasoningEffort; enabled?: boolean };
+  };
+  anthropic?: {
+    thinking:
+      | { type: "adaptive"; display?: "summarized" }
+      | { type: "disabled" };
+    // Soft effort hint for adaptive thinking; omitted when no level requested
+    // (model defaults to "high"). Sibling of `thinking` per the AI SDK.
+    effort?: ThinkingEffort;
+  };
+  bedrock?: {
+    reasoningConfig:
+      | {
+          type: "adaptive";
+          display: "summarized";
+          // Bedrock's channel for the adaptive effort hint (maps to Claude's
+          // output_config.effort).
+          maxReasoningEffort?: ThinkingEffort;
+        }
+      | { type: "disabled" };
+    /** Explicitly forwarded for Anthropic because the Bedrock SDK currently
+     * drops reasoningConfig.type="disabled" instead of sending it. */
+    additionalModelRequestFields?: {
+      thinking: { type: "disabled" };
+    };
+  };
+  openai?: OpenAIResponsesProviderOptions;
+};
+
+export type OpenRouterProviderOptions = {
+  openrouter?: {
+    provider:
+      | {
+          only: ["z-ai"];
+          allow_fallbacks: false;
+        }
+      | {
+          order: ["z-ai"];
+          allow_fallbacks: true;
+          require_parameters: true;
+        };
+  };
+};
+
+function isZaiPinnedModel(model: AIModel): boolean {
+  return model === "z-ai/glm-5.2" || model === "z-ai/glm-5.3";
+}
+
+export function buildOpenRouterProviderOptions(
+  model: AIModel,
+): OpenRouterProviderOptions | undefined {
+  if (!isZaiPinnedModel(model)) return undefined;
+  return {
+    openrouter: {
+      provider: {
+        only: ["z-ai"],
+        allow_fallbacks: false,
+      },
+    },
+  };
+}
+
+/**
+ * Z.AI's endpoint is the only `z-ai/*` provider that does not advertise
+ * `structured_outputs`, so under the hard pin above `response_format` is a soft
+ * preference and nothing enforces the schema. Prefer Z.AI, but require an
+ * endpoint that enforces it rather than silently returning unparseable output.
+ */
+export function buildOpenRouterStructuredProviderOptions(
+  model: AIModel,
+): OpenRouterProviderOptions | undefined {
+  if (!isZaiPinnedModel(model)) return undefined;
+  return {
+    openrouter: {
+      provider: {
+        order: ["z-ai"],
+        allow_fallbacks: true,
+        require_parameters: true,
+      },
+    },
+  };
+}
+
+/**
+ * Build the `providerOptions` for a request based on the model and reasoning
+ * settings. Returns `undefined` when no reasoning/thinking is requested.
+ *
+ * When thinking is enabled for an Anthropic-family model we set BOTH the
+ * `anthropic` and `bedrock` keys: the resolved provider (direct Anthropic,
+ * Pensar gateway, or Bedrock) reads only its own namespace and ignores the
+ * other, so this is safe across all three and avoids brittle provider
+ * branching here.
+ */
+export function buildReasoningProviderOptions(
+  model: AIModel,
+  opts: {
+    enableThinking?: boolean;
+    /**
+     * Adaptive-thinking effort hint for Anthropic models that support it
+     * Ignored on models without adaptive support. When omitted, models in the
+     * Claude capability registry use their documented provider default.
+     */
+    thinkingEffort?: ThinkingEffort | null;
+    openAIReasoningEffort?: OpenAIReasoningEffort | null;
+  },
+): ReasoningProviderOptions | undefined {
+  const claude = getClaudeCapabilities(model);
+  if (claude) {
+    const provider = getModelInfo(model).provider;
+    const thinking = !!opts.enableThinking || claude.alwaysOnThinking;
+    const effort = opts.thinkingEffort ?? claude.defaultEffort;
+    if (provider === "openrouter") {
+      return {
+        openrouter: {
+          reasoning: {
+            effort: thinking ? effort : "none",
+            enabled: thinking,
+          },
+        },
+      };
+    }
+    if (provider === "concentrate") {
+      return {
+        openai: {
+          forceReasoning: true,
+          reasoningEffort: thinking ? effort : "none",
+          reasoningSummary: "auto",
+        },
+      };
+    }
+    return {
+      anthropic: {
+        thinking: thinking
+          ? { type: "adaptive", display: "summarized" }
+          : { type: "disabled" },
+        ...(thinking ? { effort } : {}),
+      },
+      ...(provider === "bedrock"
+        ? {
+            bedrock: {
+              reasoningConfig: thinking
+                ? {
+                    type: "adaptive" as const,
+                    display: "summarized" as const,
+                    maxReasoningEffort: effort,
+                  }
+                : { type: "disabled" as const },
+              ...(!thinking
+                ? {
+                    additionalModelRequestFields: {
+                      thinking: { type: "disabled" as const },
+                    },
+                  }
+                : {}),
+            },
+          }
+        : {}),
+    };
+  }
+  const useThinking =
+    !!opts.enableThinking &&
+    isAnthropicProvider(model) &&
+    modelSupportsAdaptiveThinking(model);
+  const normalizedOpenAIEffort = normalizeOpenAIReasoningEffort(
+    model,
+    opts.openAIReasoningEffort,
+  );
+  // The effort hint only rides along when adaptive thinking is actually used.
+  const effort = useThinking ? (opts.thinkingEffort ?? undefined) : undefined;
+
+  if (!useThinking && !normalizedOpenAIEffort) return undefined;
+
+  if (getModelInfo(model).provider === "openrouter" && normalizedOpenAIEffort) {
+    return { openrouter: { reasoning: { effort: normalizedOpenAIEffort } } };
+  }
+
+  return {
+    ...(useThinking
+      ? {
+          anthropic: {
+            thinking: { type: "adaptive" as const },
+            ...(effort ? { effort } : {}),
+          },
+          bedrock: {
+            reasoningConfig: {
+              type: "adaptive" as const,
+              display: "summarized" as const,
+              ...(effort ? { maxReasoningEffort: effort } : {}),
+            },
+          },
+        }
+      : {}),
+    ...(normalizedOpenAIEffort
+      ? { openai: { reasoningEffort: normalizedOpenAIEffort } }
+      : {}),
+  };
+}
+
+/** Cache token metrics extracted from Anthropic providerMetadata */
+export interface CacheMetrics {
+  cacheReadInputTokens: number;
+  cacheCreationInputTokens: number;
+}
+
+export interface StreamResponseOpts {
+  prompt: string;
+  system?: string;
+  model: AIModel;
+  messages?: Array<ModelMessage>;
+  stopWhen?:
+    | StopCondition<NoInfer<ToolSet>>
+    | StopCondition<NoInfer<ToolSet>>[];
+  toolChoice?: ToolChoice<ToolSet>;
+  tools?: ToolSet;
+  onStepFinish?: StreamTextOnStepFinishCallback<ToolSet>;
+  /** Provider middleware applied only to this stream's model calls. */
+  languageModelMiddleware?: LanguageModelMiddleware | LanguageModelMiddleware[];
+  /** Explicit undefined clears an inherited recorder for an unrecorded child. */
+  inferenceRecorder?: InferenceRecorder;
+  /** Per-run usage recorder; when set it replaces the global usage callback for this stream. */
+  usageRecorder?: UsageRecorder;
+  abortSignal?: AbortSignal;
+  activeTools?: string[];
+  silent?: boolean;
+  authConfig?: AIAuthConfig;
+  onFinish?: StreamTextOnFinishCallback<ToolSet>;
+  /**
+   * Called when the context window overflows and the conversation is
+   * summarized. Callers should use this to discard stale message history
+   * so that subsequent persistence writes only include the summary +
+   * new messages (not the full pre-summarization history).
+   */
+  onSummarized?: (summary: string) => void;
+  /**
+   * Deprecated host shim: fires only when a cache counter is `> 0`.
+   * Apex internals read cache from {@link normalizeStepUsage} on the step.
+   */
+  onCacheMetrics?: (metrics: CacheMetrics) => void;
+  /** Enable extended thinking for supported models (Anthropic Claude 3.7+) */
+  enableThinking?: boolean;
+  /**
+   * Adaptive-thinking effort hint for Anthropic models that support it
+   * (Opus/Sonnet 4.6+). Ignored where unsupported; model defaults to `high`.
+   */
+  thinkingEffort?: ThinkingEffort | null;
+  /** OpenAI reasoning effort for GPT/o-series reasoning models. */
+  openAIReasoningEffort?: OpenAIReasoningEffort | null;
+  /** Session root path — used by context management layers to persist truncated tool results */
+  sessionPath?: string;
+  /** Session id (`ses_…`) of the agent making this call; stamped onto AI-span telemetry so traces are filterable by session. */
+  sessionId?: string;
+  /**
+   * Opt-in durable context authority for recorded runs. Turn-0 base and each
+   * step's cumulative context are checkpointed before selection/dispatch;
+   * unset → legacy behavior, no persistence.
+   */
+  contextRecorder?: RunContextRecorder;
+  /**
+   * Internal: recovery-recursion depth. Bumped at each summarize → resume
+   * boundary; throws `ContextLengthExhaustedError` past `MAX_RESTART_DEPTH`.
+   */
+  _restartDepth?: number;
+  /** Shared only by continuations of this stream, never across agents. */
+  _compaction?: { last?: CompactionLink };
+  _compactionTrigger?: CompactionTrigger;
+}
+
+const NATIVE_STREAM_RECOVERY = Symbol("native-stream-recovery");
+
+interface NativeStreamRecovery {
+  model: LanguageModel;
+  run<T>(fn: () => T): T;
+}
+
+type InternalStreamResponseOpts = StreamResponseOpts & {
+  [NATIVE_STREAM_RECOVERY]?: NativeStreamRecovery;
+};
+
+export function streamResponse(
+  opts: StreamResponseOpts,
+): StreamTextResult<ToolSet, never> {
+  opts = { ...opts, inferenceRecorder: getInferenceRecorder(opts) };
+  return runWithInferenceRecorder(opts.inferenceRecorder, () => {
+    const recovery = (opts as InternalStreamResponseOpts)[
+      NATIVE_STREAM_RECOVERY
+    ];
+    if (recovery) {
+      return recovery.run(() => streamResponseWithinOperation(opts, recovery));
+    }
+    return streamResponseWithinOperation(opts);
+  });
+}
+
+function streamResponseWithinOperation(
+  opts: StreamResponseOpts,
+  nativeRecovery?: NativeStreamRecovery,
+): StreamTextResult<ToolSet, never> {
+  const compactionState = opts._compaction ?? {};
+  // One effective tool selection per stream, before any fitting: proactive
+  // fit, provider schemas, execution, repair, and every continuation path
+  // below spread this same opts object, so they must all see the filtered
+  // map. Re-entry is identity (`activeTools` already consumed).
+  opts = {
+    ...opts,
+    _compaction: compactionState,
+    // `activeTools` is authoritative for advertisement and execution: a tool
+    // (including `response`) is exposed only when the caller lists it, so the
+    // structured-output tool is included when listed and inert when omitted.
+    ...resolveEffectiveTools(opts.tools, opts.activeTools),
+  };
+  // Bound recovery recursion (summarize → resume → overflow → …).
+  const restartDepth = opts._restartDepth ?? 0;
+  if (restartDepth > MAX_RESTART_DEPTH) {
+    throw new ContextLengthExhaustedError(
+      `Context-length recovery exhausted after ${restartDepth} restart cycles`,
+      restartDepth,
+    );
+  }
+
+  const {
+    prompt,
+    system,
+    model,
+    messages,
+    stopWhen,
+    toolChoice,
+    tools,
+    onStepFinish: userOnStepFinish,
+    languageModelMiddleware,
+    usageRecorder,
+    abortSignal,
+    activeTools,
+    silent,
+    authConfig,
+    onFinish,
+    onCacheMetrics,
+    enableThinking,
+    thinkingEffort,
+    openAIReasoningEffort,
+    sessionId,
+    contextRecorder,
+  } = opts;
+
+  // Recorded-run base, captured once at the first prepareStep of this
+  // streamText call; only used to reconstruct cumulative prefixes in
+  // onStepFinish (response.messages excludes the input base).
+  let recorderBase: ModelMessage[] | undefined;
+
+  // Wrap onStepFinish to fire cache metrics and the usage callback for every
+  // step. Must be async so that callers returning a Promise (persistence,
+  // issue queuing, or the usage callback itself) are fully awaited before
+  // the AI SDK starts the next step. One normalization feeds both sinks —
+  // provider metadata is never parsed twice.
+  const onStepFinish: typeof userOnStepFinish = async (step) => {
+    const stepUsage = normalizeStepUsage(step);
+    // Canonical commit precedes every downstream use; the recorder latches
+    // rejections (SDK notify() swallows them) until the next turn gate or
+    // agent-side flush turns the latch into a stream failure.
+    // Synthetic summarization/tool-repair events re-report context that
+    // already includes the base (messagesContainer.current) — committing
+    // them would duplicate the prefix.
+    if (
+      contextRecorder &&
+      step.response.id !== "summarization" &&
+      step.response.id !== "tool-repair"
+    ) {
+      if (!recorderBase) {
+        throw new Error("context recorder active without a committed base");
+      }
+      await contextRecorder.checkpoint({
+        messages: [...recorderBase, ...step.response.messages],
+        system: effectiveSystem ?? null,
+      });
+    }
+    if (
+      onCacheMetrics &&
+      (stepUsage.cacheReadTokens > 0 || stepUsage.cacheWriteTokens > 0)
+    ) {
+      await onCacheMetrics({
+        cacheReadInputTokens: stepUsage.cacheReadTokens,
+        cacheCreationInputTokens: stepUsage.cacheWriteTokens,
+      });
+    }
+    await userOnStepFinish?.(step);
+    await emitUsage(model, stepUsage, resolveUsageSink(usageRecorder));
+  };
+  const providerModel =
+    nativeRecovery?.model ??
+    (() => {
+      const baseProviderModel = withNativeRolloutEvidenceModel(
+        withModelCallDiagnostics(getProviderModel(model, authConfig)),
+        {
+          requestedModelId: model,
+          operationKind: "agent.stream",
+          sessionId,
+        },
+      );
+      return languageModelMiddleware
+        ? wrapLanguageModel({
+            model: baseProviderModel,
+            middleware: languageModelMiddleware,
+          })
+        : baseProviderModel;
+    })();
+  // Undefined when the model has no cache breakpoint we can express (non-Claude
+  // Bedrock models, OpenAI/Google/OpenRouter) — those run uncached.
+  const cacheBreakpoint = cacheBreakpointFor(model);
+
+  // Models that mis-parse multiple tool calls per turn (DeepSeek V3.1 on
+  // Bedrock) are constrained to one tool call per turn via system prompt —
+  // single calls parse correctly, parallel calls drop/empty arguments. Only
+  // applied when tools are actually bound. See prefersSequentialToolCalls.
+  // Computed before context fitting so token budgeting accounts for the
+  // appended instruction that is actually sent to the provider.
+  const systemWithToolPolicy = applySequentialToolCallPolicy(
+    system,
+    tools,
+    model,
+  );
+
+  // Proactive context fit: compact before the call. Reactive recovery
+  // below is a safety net for tokenizer drift the estimate doesn't catch.
+  let fittedMessages = messages;
+  let proactiveFitFailed = false;
+  if (messages && messages.length > 0) {
+    const fitted = fitMessagesToContext(messages, {
+      contextWindow: getContextWindow(
+        model,
+        authConfig?.customProviders,
+        authConfig?.hoonifyModels,
+      ),
+      maxOutputTokens: getMaxOutputTokens(
+        model,
+        authConfig?.customProviders,
+        authConfig?.hoonifyModels,
+      ),
+      system: systemWithToolPolicy,
+      tools,
+      sessionPath: opts.sessionPath,
+      telemetry: {
+        trigger: "proactive",
+        model,
+        sessionId,
+        restartDepth,
+        previous: opts._compaction?.last,
+      },
+    });
+    if (fitted.compaction) compactionState.last = fitted.compaction;
+    fittedMessages = fitted.messages;
+    proactiveFitFailed = !fitted.fitsBudget;
+    if (fitted.modified && !silent) {
+      log.warn(
+        `Proactive context fit: compacted messages to ~${fitted.estimatedInputTokens} tokens (fits=${fitted.fitsBudget})`,
+      );
+    }
+  }
+
+  // Container reference is stable; value gets reassigned by `prepareStep`
+  // and the recovery paths so nested retries read the latest compacted view.
+  const messagesContainer = { current: fittedMessages || [] };
+
+  // Layer 1+2 alone can't fit — escalate before sending a doomed request.
+  // Wrap with the same error handler the streamText path uses, otherwise
+  // an overflow inside the summarization call itself escapes uncaught while
+  // the equivalent post-streamText path (line ~880) recovers.
+  if (proactiveFitFailed && fittedMessages) {
+    if (!silent) {
+      log.warn(
+        `Proactive context fit returned fitsBudget=false on ${fittedMessages.length} messages — escalating to summarization before send`,
+      );
+    }
+    // Pass `opts` through unchanged. The slot for this escalation is
+    // consumed by `summarizeConversation`'s internal `_restartDepth + 1`
+    // bump on the resumed `streamResponse` call — pre-bumping here would
+    // double-count vs the reactive Layer 3 path (which also passes
+    // `opts._restartDepth` to `createSummarizationStream` after a
+    // no-retry fall-through). With the post-bump, all three escalation
+    // entry points (proactive, outer-catch, reactive Layer 3) consume
+    // exactly one depth slot per summarization attempt — symmetric.
+    return wrapStreamWithErrorHandler(
+      createSummarizationStream(
+        fittedMessages,
+        { ...opts, _compactionTrigger: "proactive" },
+        providerModel,
+      ),
+      messagesContainer,
+      opts,
+      providerModel,
+      silent,
+    );
+  }
+
+  // For cacheable models, move the system prompt into messages behind a cache
+  // breakpoint. This caches tools + system prompt across multi-turn
+  // conversations (~90% cost reduction on cache hits).
+  let effectiveSystem: string | undefined = systemWithToolPolicy;
+  let effectiveMessages: ModelMessage[] | undefined = fittedMessages;
+
+  if (cacheBreakpoint && systemWithToolPolicy) {
+    const baseMessages: ModelMessage[] = fittedMessages ?? [
+      { role: "user" as const, content: prompt },
+    ];
+    effectiveMessages = withCachedSystemPrompt(
+      systemWithToolPolicy,
+      baseMessages,
+      cacheBreakpoint,
+    );
+    effectiveSystem = undefined;
+  }
+
+  // Build providerOptions for extended thinking when enabled on a supported model.
+  // Caching uses message-level breakpoints (withCachedSystemPrompt / withCachedLastMessage),
+  // not top-level providerOptions, so there is no collision.
+  // Note: when thinking is enabled, temperature must not be set (Anthropic rejects it);
+  // this path never sets temperature, so there is nothing to clear.
+  const reasoningProviderOptions = buildReasoningProviderOptions(model, {
+    enableThinking,
+    thinkingEffort,
+    openAIReasoningEffort,
+  });
+  const openRouterProviderOptions = buildOpenRouterProviderOptions(model);
+  const providerOptions =
+    reasoningProviderOptions || openRouterProviderOptions
+      ? { ...reasoningProviderOptions, ...openRouterProviderOptions }
+      : undefined;
+
+  let rateLimitRetryCount = 0;
+
+  // Make `activeTools` authoritative: advertise, execute, and enumerate only
+  // the allowlisted tools. A no-op when `activeTools` is empty/undefined.
+  // `resolveEffectiveTools` above already filtered the map and cleared
+  // `activeTools`, so this is a no-op on the normal path; kept for callers that
+  // reach streamText with an intact `activeTools` (identity when undefined).
+  const allowedTools = withResponseToolActive(tools, activeTools);
+  const effectiveTools = restrictToolsToActive(tools, allowedTools);
+
+  try {
+    // Create the appropriate provider instance. The span tracker captures
+    // the SDK's root generation span for error marking (see below).
+    const generationSpans = createGenerationSpanTracker();
+    const responseOptions: Parameters<typeof streamText>[0] = {
+      model: providerModel,
+      system: effectiveSystem,
+      ...(effectiveMessages ? { messages: effectiveMessages } : { prompt }),
+      stopWhen,
+      toolChoice,
+      tools: effectiveTools,
+      maxRetries: 3,
+      providerOptions,
+      // Step history would retain a serialized request body per step, and
+      // later steps re-send the whole conversation, so they accumulate.
+      // Native capture reads the provider stream result, not this filter.
+      experimental_include: { requestBody: false },
+      // The forwarding tracer keeps a handle on the SDK's root generation
+      // span: error-part runs complete normally in the SDK, so nothing
+      // marks the span failed — the wrapper's catch does.
+      experimental_telemetry: {
+        ...createAiTelemetrySettings({
+          operation: "apex.agent.stream",
+          sessionId,
+          compaction: opts._compaction?.last,
+        }),
+        tracer: generationSpans.tracer,
+      },
+      // Pin actual emission to the same value `fitMessagesToContext`
+      // reserved for output. Without this, the SDK applies provider
+      // defaults that can exceed our budget — e.g. GPT-4o defaults to
+      // 16K output but our messages were sized assuming a smaller
+      // reservation. Making the value explicit closes that drift class.
+      maxOutputTokens: getMaxOutputTokens(
+        model,
+        authConfig?.customProviders,
+        authConfig?.hoonifyModels,
+      ),
+      // Awaiting inside prepareStep keeps streamResponse synchronous; the
+      // SDK awaits it before provider dispatch, so every turn and every
+      // recovery re-entry commits before use.
+      prepareStep: async (opts) => {
+        if (contextRecorder) {
+          if (recorderBase === undefined) {
+            recorderBase = structuredClone(opts.messages);
+          }
+          // Full logical context each turn — accepted step outputs must
+          // survive into every later commit, and system is explicit (when
+          // cached into messages, effectiveSystem is undefined → null).
+          await contextRecorder.checkpoint({
+            messages: opts.messages,
+            system: effectiveSystem ?? null,
+          });
+        }
+        // Update the container with the latest messages
+        messagesContainer.current = opts.messages;
+        // Mark the last message so the growing conversation caches incrementally
+        if (cacheBreakpoint) {
+          return {
+            messages: withCachedLastMessage(opts.messages, cacheBreakpoint),
+          };
+        }
+        return undefined;
+      },
+      onError: async ({ error }: { error: unknown }) => {
+        // Never throw here: the SDK calls this inside its stream pipeline,
+        // and a thrown callback kills the pipeline's terminal lifecycle —
+        // the root generation span never ends and nothing exports. Return
+        // normally instead; the error part continues downstream and the
+        // consumption wrapper converts it to a thrown error (after the
+        // tail drains), where the retry machinery takes over.
+        const errorMessage =
+          error instanceof Error ? error.message : String(error);
+        if (
+          errorMessage.toLowerCase().includes("too many tokens") ||
+          errorMessage.toLowerCase().includes("overloaded")
+        ) {
+          rateLimitRetryCount++;
+          if (getInferenceRecorder(opts)) {
+            // This SDK callback must not throw, including when cancellation ends its wait.
+            await delay(1000 * rateLimitRetryCount, undefined, {
+              signal: abortSignal,
+            }).catch(() => {});
+          } else {
+            await new Promise((resolve) =>
+              setTimeout(resolve, 1000 * rateLimitRetryCount),
+            );
+          }
+        }
+      },
+      onStepFinish,
+      abortSignal,
+      activeTools: allowedTools,
+      experimental_repairToolCall: async ({
+        toolCall,
+        inputSchema,
+        tools,
+        error,
+      }) => {
+        // [response-debug] Log when repair fires for `response` (streamed args failed Zod validation), bypassing `silent`.
+        if (RESPONSE_DEBUG && toolCall.toolName === RESPONSE_TOOL_NAME) {
+          const raw = toolCall.input;
+          const rawLen =
+            typeof raw === "string" ? raw.length : JSON.stringify(raw).length;
+          log.warn(
+            `[response-debug] experimental_repairToolCall FIRED for response ` +
+              `rawInputChars=${rawLen} error=${(error.message || String(error)).slice(0, 200)}`,
+          );
+        }
+
+        // Fail closed for security-sensitive tools: never let repair invent or
+        // replace arguments (e.g. a spawn `target`). Returning null drops the
+        // invalid call instead of executing a model-synthesized one.
+        if (isRepairFailClosedTool(toolCall.toolName)) {
+          if (!silent) {
+            log.warn(
+              `Tool-call repair disabled for security-sensitive tool "${toolCall.toolName}" — failing closed (no argument synthesis).`,
+            );
+          }
+          return null;
+        }
+
+        try {
+          if (!silent) {
+            log.debug(`Repairing tool call: ${toolCall.toolName}`, {
+              error: error.message || String(error),
+            });
+
+            // Log specific details for common enum errors
+            if (
+              error.message &&
+              (error.message.includes("severity") ||
+                error.message.includes("riskLevel"))
+            ) {
+              log.debug(
+                "Enum validation error — tool call repair will normalize the value",
+              );
+            }
+          }
+
+          // Get the actual tool definition which contains the Zod schema
+          const tool = tools[toolCall.toolName];
+          if (!tool?.inputSchema) {
+            if (!silent) {
+              log.warn(
+                `Cannot repair tool call: ${toolCall.toolName} not found or has no schema`,
+              );
+            }
+            return null;
+          }
+
+          const jsonSchema = inputSchema({ toolName: toolCall.toolName });
+
+          // Bound input + schema so the repair call can't itself overflow.
+          const rawInput = toolCall.input;
+          const inputAsText =
+            typeof rawInput === "string" ? rawInput : JSON.stringify(rawInput);
+          const boundedInput = truncateWithMarker(
+            inputAsText,
+            MAX_TOOL_INPUT_CHARS,
+          );
+          const boundedSchema = truncateWithMarker(
+            JSON.stringify(jsonSchema),
+            MAX_SCHEMA_CHARS,
+            "schema",
+          );
+
+          const {
+            output: repairedArgs,
+            usage: repairUsage,
+            providerMetadata: repairProviderMetadata,
+          } = await runWithNativeRolloutOperation(
+            { operationKind: "tool.repair", sessionId },
+            () =>
+              generateText({
+                model: providerModel,
+                providerOptions: {
+                  ...openRouterProviderOptions,
+                  // Durable runtime: tool-call repair fires only when a
+                  // replayed tool call fails validation — nondeterministically
+                  // vs the first run — so it must stay off the checkpoint
+                  // journal. The agent-runtime middleware reads this namespace
+                  // and runs the call without a durable step; the underlying
+                  // provider ignores it.
+                  pensarRuntime: { ephemeral: true },
+                },
+                output: Output.object({
+                  schema: tool.inputSchema, // Use the actual Zod schema from the tool
+                }),
+                prompt: [
+                  `The model tried to call the tool "${toolCall.toolName}"` +
+                    ` with the following inputs:`,
+                  boundedInput,
+                  `The tool accepts the following schema:`,
+                  boundedSchema,
+                  `Error encountered: ${error}`,
+                  "Please fix the inputs to match the schema.",
+                  "",
+                  "IMPORTANT: For enum fields like 'severity' or 'riskLevel', use ONLY the exact values from the enum (e.g., 'HIGH', 'CRITICAL', 'MEDIUM', 'LOW').",
+                  "Do not add prefixes, suffixes, or formatting characters like '>', '-', '!', etc.",
+                ].join("\n"),
+                abortSignal,
+                experimental_telemetry: createAiTelemetrySettings({
+                  operation: "apex.tool.repair",
+                  sessionId,
+                }),
+              }),
+          );
+
+          // Report tool repair token usage if onStepFinish callback is
+          // provided. Awaited: the callback persists messages and records
+          // usage — a detached invocation would leak unhandled rejections
+          // into the tool loop. Provider metadata and cache details flow
+          // through so repair usage stays attributable.
+          if (onStepFinish && repairUsage) {
+            try {
+              await onStepFinish({
+                text: "",
+                reasoning: undefined,
+                reasoningDetails: [],
+                files: [],
+                sources: [],
+                toolCalls: [],
+                toolResults: [],
+                finishReason: "stop",
+                usage: {
+                  inputTokens: repairUsage.inputTokens ?? 0,
+                  outputTokens: repairUsage.outputTokens ?? 0,
+                  totalTokens: repairUsage.totalTokens ?? 0,
+                  ...(repairUsage.inputTokenDetails
+                    ? { inputTokenDetails: repairUsage.inputTokenDetails }
+                    : {}),
+                },
+                warnings: [],
+                request: {},
+                response: {
+                  id: "tool-repair",
+                  timestamp: new Date(),
+                  modelId: "",
+                  messages: messagesContainer.current,
+                },
+                providerMetadata: repairProviderMetadata,
+                stepType: "initial",
+                isContinued: false,
+              } as unknown as Parameters<
+                StreamTextOnStepFinishCallback<ToolSet>
+              >[0]);
+            } catch (callbackError) {
+              if (!silent) {
+                log.warn("Error reporting tool repair usage", {
+                  error: String(callbackError),
+                });
+              }
+            }
+          }
+
+          if (repairedArgs === undefined || repairedArgs === null) {
+            if (!silent) {
+              log.warn(
+                `Tool call repair for "${toolCall.toolName}" produced no valid output`,
+              );
+            }
+            return null;
+          }
+          return { ...toolCall, input: JSON.stringify(repairedArgs) };
+        } catch (repairError) {
+          if (!silent) {
+            log.warn("Error repairing tool call", {
+              error: String(repairError),
+            });
+          }
+          return null;
+        }
+      },
+      onFinish,
+    };
+    let recovery = nativeRecovery;
+    const response = recovery
+      ? streamText(responseOptions)
+      : runWithNativeRolloutOperation(
+          { operationKind: "agent.stream", sessionId },
+          () => {
+            const operation = new AsyncResource("apex.native-stream-recovery");
+            recovery = {
+              model: providerModel,
+              run: (fn) => operation.runInAsyncScope(fn),
+            };
+            return streamText(responseOptions);
+          },
+        );
+
+    // Wrap the stream to catch async errors during consumption
+    const recoveryOpts: InternalStreamResponseOpts = recovery
+      ? { ...opts, [NATIVE_STREAM_RECOVERY]: recovery }
+      : opts;
+    return wrapStreamWithErrorHandler(
+      response,
+      messagesContainer,
+      recoveryOpts,
+      providerModel,
+      silent,
+      0,
+      0,
+      generationSpans,
+    );
+  } catch (error) {
+    // Check if the error is related to context length
+    const isContextLengthError = checkIfContextLengthError(error);
+    const outerErrorMessage =
+      error instanceof Error ? error.message : String(error);
+
+    if (isContextLengthError) {
+      if (!silent) {
+        log.warn(
+          `Context length error, summarizing ${messagesContainer.current.length} messages`,
+          { error: outerErrorMessage },
+        );
+      }
+      // Wrap so a context error inside the summarization stream itself
+      // (or the resumed `streamResponse` it triggers) is caught and routed
+      // through the same recovery cascade as the proactive and reactive
+      // paths — invariant I4: every context-recovery failure must be caught
+      // at the `streamResponse` boundary, never escape raw to the consumer.
+      //
+      // Pass `opts` through unchanged — the slot is consumed by
+      // `summarizeConversation`'s internal `_restartDepth + 1` bump on
+      // the resumed call. See the proactive escalation comment above
+      // for the symmetry rationale.
+      return wrapStreamWithErrorHandler(
+        createSummarizationStream(
+          messagesContainer.current,
+          { ...opts, _compactionTrigger: "context_overflow" },
+          providerModel,
+        ),
+        messagesContainer,
+        opts,
+        providerModel,
+        silent,
+      );
+    }
+    if (!silent) {
+      log.error("Non-context length error, re-throwing", {
+        error: outerErrorMessage,
+      });
+    }
+
+    // Re-throw if it's not a context length error
+    throw error;
+  }
+}
+
+export interface GenerateObjectOpts<T extends z.ZodType> {
+  model: AIModel;
+  schema: T;
+  prompt: string;
+  system?: string;
+  maxTokens?: number;
+  temperature?: number;
+  openAIReasoningEffort?: OpenAIReasoningEffort | null;
+  authConfig?: AIAuthConfig;
+  abortSignal?: AbortSignal;
+  onTokenUsage?: (inputTokens: number, outputTokens: number) => void;
+  /** Provider middleware applied only to this call's model. */
+  languageModelMiddleware?: LanguageModelMiddleware | LanguageModelMiddleware[];
+  /** Explicit undefined clears an inherited recorder for an unrecorded child. */
+  inferenceRecorder?: InferenceRecorder;
+  /** Per-run usage recorder; when set it replaces the global usage callback. */
+  usageRecorder?: UsageRecorder;
+  /** Session id (`ses_…`) of the caller — stamped onto AI-span telemetry. */
+  sessionId?: string;
+  /** Stable operation id; defaults to `apex.structured.generate`. */
+  operation?: AiTelemetryOperation;
+}
+
+const MAX_OBJECT_RATE_LIMIT_RETRIES = 8;
+
+export async function generateObjectResponse<T extends z.ZodType>(
+  opts: GenerateObjectOpts<T>,
+): Promise<z.infer<T>> {
+  opts = { ...opts, inferenceRecorder: getInferenceRecorder(opts) };
+  return runWithInferenceRecorder(opts.inferenceRecorder, () =>
+    generateObjectResponseWithinRecorder(opts),
+  );
+}
+
+async function generateObjectResponseWithinRecorder<T extends z.ZodType>(
+  opts: GenerateObjectOpts<T>,
+): Promise<z.infer<T>> {
+  const {
+    model,
+    schema,
+    prompt,
+    system,
+    maxTokens,
+    temperature,
+    openAIReasoningEffort,
+    authConfig,
+    abortSignal,
+    onTokenUsage,
+    languageModelMiddleware,
+    usageRecorder,
+    sessionId,
+  } = opts;
+
+  const baseProviderModel = withNativeRolloutEvidenceModel(
+    withModelCallDiagnostics(getProviderModel(model, authConfig)),
+    {
+      requestedModelId: model,
+      operationKind: nativeRolloutOperation(opts.operation),
+      sessionId,
+    },
+  );
+  const providerModel = languageModelMiddleware
+    ? wrapLanguageModel({
+        model: baseProviderModel,
+        middleware: languageModelMiddleware,
+      })
+    : baseProviderModel;
+  const reasoningProviderOptions = buildReasoningProviderOptions(model, {
+    openAIReasoningEffort,
+  });
+  const openRouterProviderOptions =
+    buildOpenRouterStructuredProviderOptions(model);
+
+  return runWithNativeRolloutOperation(
+    {
+      operationKind: nativeRolloutOperation(opts.operation),
+      sessionId,
+    },
+    async () => {
+      let lastError: unknown;
+
+      for (
+        let attempt = 0;
+        attempt <= MAX_OBJECT_RATE_LIMIT_RETRIES;
+        attempt++
+      ) {
+        try {
+          const { output, usage, providerMetadata } = await generateText({
+            model: providerModel,
+            output: Output.object({
+              schema,
+            }),
+            prompt,
+            system,
+            maxOutputTokens: maxTokens,
+            temperature,
+            providerOptions:
+              reasoningProviderOptions || openRouterProviderOptions
+                ? {
+                    ...reasoningProviderOptions,
+                    ...(openRouterProviderOptions
+                      ? {
+                          openrouter: {
+                            ...reasoningProviderOptions?.openrouter,
+                            ...openRouterProviderOptions.openrouter,
+                          },
+                        }
+                      : {}),
+                  }
+                : undefined,
+            maxRetries: 0,
+            abortSignal,
+            experimental_telemetry: createAiTelemetrySettings({
+              operation: opts.operation ?? "apex.structured.generate",
+              sessionId,
+            }),
+          });
+
+          if (onTokenUsage && usage) {
+            onTokenUsage(usage.inputTokens ?? 0, usage.outputTokens ?? 0);
+          }
+
+          if (usage) {
+            await emitUsage(
+              model,
+              normalizeStepUsage({ usage, providerMetadata }),
+              resolveUsageSink(usageRecorder),
+            );
+          }
+
+          // zod v4: the AI SDK's `Output.object` no longer carries the schema's
+          // inferred type through `output` (it widens to `unknown`), so restore it
+          // from the schema generic for callers.
+          return output as z.infer<T>;
+        } catch (error) {
+          // Surface latched persistence/limit failures before any retry
+          // classification — they are terminal, never delayed or retried.
+          await getInferenceRecorder(opts)?.flush();
+          if (
+            error instanceof RunPersistenceError ||
+            error instanceof RunLimitError ||
+            error instanceof RunControlInterruption
+          ) {
+            throw error;
+          }
+          lastError = error;
+
+          if (checkIfContextLengthError(error)) {
+            const msg = error instanceof Error ? error.message : String(error);
+            throw new ContextLengthError(
+              `Prompt exceeds model context window: ${msg}`,
+            );
+          }
+
+          if (
+            checkIfRateLimitError(error) &&
+            attempt < MAX_OBJECT_RATE_LIMIT_RETRIES
+          ) {
+            const delayMs = Math.min(1000 * 2 ** attempt, 60_000);
+            const recorder = getInferenceRecorder(opts);
+            await recorder?.retry({
+              authority: "object-rate-limit",
+              count: attempt + 1,
+              maxRetries: MAX_OBJECT_RATE_LIMIT_RETRIES,
+              delayMs,
+            });
+            if (recorder) {
+              await delay(delayMs, undefined, { signal: abortSignal });
+            } else {
+              await new Promise((resolve) => setTimeout(resolve, delayMs));
+            }
+            continue;
+          }
+
+          throw error;
+        }
+      }
+
+      throw lastError;
+    },
+  );
+}
+
+class ContextLengthError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = "ContextLengthError";
+  }
+}
+
+/**
+ * Terminal: all recovery paths exhausted. The only context-length error
+ * allowed to escape `streamResponse` — others are recoverable by design.
+ */
+export class ContextLengthExhaustedError extends Error {
+  readonly restartDepth: number;
+  constructor(message: string, restartDepth: number) {
+    super(message);
+    this.name = "ContextLengthExhaustedError";
+    this.restartDepth = restartDepth;
+  }
+}
+
+const MAX_RESTART_DEPTH = 3;
+
+const MAX_TOOL_INPUT_CHARS = 8_000;
+const MAX_SCHEMA_CHARS = 6_000;
+
+const MINIMAL_RESTART_PROMPT =
+  "Continue the previous task. Earlier context was discarded due to repeated context-length errors.";
+
+export {
+  applySequentialToolCallPolicy,
+  createToolExecutionGate,
+  getStreamIdleTimeoutMs,
+  SEQUENTIAL_TOOL_CALL_INSTRUCTION,
+  StreamIdleTimeoutError,
+  withIdleTimeout,
+};