@ai-sdk/code-mode 1.0.21 → 1.0.22

package/dist/tool-invocation.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/tool-invocation.ts"],"sourcesContent":["import { asSchema } from 'ai';\nimport { CODE_MODE_TOOL_APPROVAL_KIND } from './approval.js';\nimport {\n  CodeModeProtocolError,\n  CodeModeToolApprovalDeniedError,\n  CodeModeToolApprovalRequiredError,\n  CodeModeToolError,\n} from './errors.js';\nimport { isCodeModeHostInterruptSignal } from './host-interrupt.js';\nimport type {\n  CodeModeInterruptExecutionContext,\n  CodeModeInterruptPayload,\n  CodeModeOptions,\n  CodeModeToolExecutionOptions,\n  CodeModeToolSet,\n} from './types.js';\nimport {\n  assertJsonSerializable,\n  toJsonPayload,\n} from './utils/serialization.js';\n\nexport type HostToolInvocationResult =\n  | { type: 'success'; valueJson: string }\n  | {\n      type: 'interrupted';\n      toolName: string;\n      input: unknown;\n      toolCallId: string;\n      payload: CodeModeInterruptPayload;\n    };\n\nexport async function invokeHostTool({\n  toolName,\n  inputJson,\n  tools,\n  baseExecutionOptions,\n  codeModeOptions,\n  maxToolInputBytes,\n  maxToolOutputBytes,\n  toolCallId,\n  codeModeInterrupt,\n  skipApproval = false,\n}: {\n  toolName: string;\n  inputJson: string;\n  tools: CodeModeToolSet;\n  baseExecutionOptions: CodeModeToolExecutionOptions;\n  codeModeOptions: CodeModeOptions;\n  maxToolInputBytes: number;\n  maxToolOutputBytes: number;\n  toolCallId: string;\n  codeModeInterrupt?: CodeModeInterruptExecutionContext;\n  skipApproval?: boolean;\n}): Promise<HostToolInvocationResult> {\n  throwIfAborted(baseExecutionOptions.abortSignal);\n\n  const hostTool = tools[toolName];\n  if (!hostTool) {\n    throw new CodeModeToolError(`Unknown tool: ${toolName}`, {\n      toolName,\n      availableTools: Object.keys(tools),\n    });\n  }\n  if (hostTool.execute == null) {\n    throw new CodeModeToolError(`Tool \"${toolName}\" does not have execute().`, {\n      toolName,\n    });\n  }\n\n  const input = inputJson === '' ? undefined : JSON.parse(inputJson);\n  assertJsonSerializable(input, maxToolInputBytes, `Tool \"${toolName}\" input`);\n\n  const validation = await raceAgainstAbort(\n    validateToolInput(hostTool.inputSchema, input),\n    baseExecutionOptions.abortSignal,\n  );\n  if (!validation.success) {\n    throw new CodeModeToolError(\n      `Invalid input for tool \"${toolName}\": ${validation.error.message}`,\n      { toolName, input, cause: validation.error.message },\n    );\n  }\n\n  const executionOptions: CodeModeToolExecutionOptions = {\n    ...baseExecutionOptions,\n    toolCallId,\n    ...(codeModeInterrupt !== undefined ? { codeModeInterrupt } : {}),\n  };\n\n  const needsApproval =\n    !skipApproval &&\n    (await raceAgainstAbort(\n      requiresApproval(hostTool, validation.value, executionOptions),\n      executionOptions.abortSignal,\n    ));\n\n  if (needsApproval) {\n    if (codeModeOptions.approval?.mode === 'interrupt') {\n      return {\n        type: 'interrupted',\n        toolName,\n        input: validation.value,\n        toolCallId,\n        payload: { kind: CODE_MODE_TOOL_APPROVAL_KIND },\n      };\n    }\n\n    const approval = await raceAgainstAbort(\n      Promise.resolve(\n        codeModeOptions.approval?.onApprovalRequired?.({\n          toolName,\n          input: validation.value,\n          toolCallId,\n        }),\n      ),\n      baseExecutionOptions.abortSignal,\n    );\n    if (approval === undefined) {\n      throw new CodeModeToolApprovalRequiredError(\n        toolName,\n        validation.value,\n        toolCallId,\n      );\n    }\n    const approved =\n      typeof approval === 'string'\n        ? approval === 'approved'\n        : approval?.approved;\n    const reason = typeof approval === 'string' ? undefined : approval?.reason;\n    if (typeof approved !== 'boolean') {\n      throw new CodeModeProtocolError(\n        `Tool \"${toolName}\" approval callback returned a malformed approval decision.`,\n        { toolName, toolCallId },\n      );\n    }\n    if (reason !== undefined && typeof reason !== 'string') {\n      throw new CodeModeProtocolError(\n        `Tool \"${toolName}\" approval callback returned a malformed approval reason.`,\n        { toolName, toolCallId },\n      );\n    }\n    if (!approved) {\n      throw new CodeModeToolApprovalDeniedError(\n        toolName,\n        validation.value,\n        toolCallId,\n        reason,\n      );\n    }\n  }\n\n  let output: unknown;\n  try {\n    output = await raceAgainstAbort(\n      executeHostTool(hostTool.execute.bind(hostTool), {\n        input: validation.value,\n        options: executionOptions,\n      }),\n      executionOptions.abortSignal,\n    );\n  } catch (error) {\n    if (isCodeModeHostInterruptSignal(error)) {\n      return {\n        type: 'interrupted',\n        toolName,\n        input: validation.value,\n        toolCallId,\n        payload: error.payload,\n      };\n    }\n    throw error;\n  }\n  return {\n    type: 'success',\n    valueJson: toJsonPayload(\n      output,\n      maxToolOutputBytes,\n      `Tool \"${toolName}\" output`,\n    ),\n  };\n}\n\nasync function requiresApproval(\n  hostTool: CodeModeToolSet[string],\n  input: unknown,\n  options: CodeModeToolExecutionOptions,\n): Promise<boolean> {\n  if (hostTool.needsApproval == null) {\n    return false;\n  }\n  if (typeof hostTool.needsApproval === 'boolean') {\n    return hostTool.needsApproval;\n  }\n  return await hostTool.needsApproval(input as never, options as never);\n}\n\nasync function validateToolInput(\n  schema: CodeModeToolSet[string]['inputSchema'],\n  input: unknown,\n): Promise<\n  { success: true; value: unknown } | { success: false; error: Error }\n> {\n  const normalizedSchema = asSchema(schema);\n  if (normalizedSchema.validate === undefined) {\n    return { success: true, value: input };\n  }\n  return await normalizedSchema.validate(input);\n}\n\nasync function executeHostTool(\n  execute: NonNullable<CodeModeToolSet[string]['execute']>,\n  {\n    input,\n    options,\n  }: {\n    input: unknown;\n    options: CodeModeToolExecutionOptions;\n  },\n): Promise<unknown> {\n  const output = execute(input as never, options as never);\n  if (isAsyncIterable(output)) {\n    let finalOutput: unknown;\n    for await (const part of output) {\n      finalOutput = part;\n    }\n    return finalOutput;\n  }\n  return await output;\n}\n\nfunction isAsyncIterable(value: unknown): value is AsyncIterable<unknown> {\n  return (\n    typeof value === 'object' &&\n    value !== null &&\n    Symbol.asyncIterator in value &&\n    typeof (value as { [Symbol.asyncIterator]?: unknown })[\n      Symbol.asyncIterator\n    ] === 'function'\n  );\n}\n\nasync function raceAgainstAbort<T>(\n  operation: Promise<T>,\n  abortSignal: AbortSignal | undefined,\n): Promise<T> {\n  if (abortSignal === undefined) {\n    return await operation;\n  }\n  throwIfAborted(abortSignal);\n\n  let rejectOnAbort!: (reason?: unknown) => void;\n  const aborted = new Promise<never>((_resolve, reject) => {\n    rejectOnAbort = reject;\n  });\n  const onAbort = () => {\n    rejectOnAbort(abortReason(abortSignal));\n  };\n\n  abortSignal.addEventListener('abort', onAbort, { once: true });\n  if (abortSignal.aborted) {\n    onAbort();\n  }\n\n  try {\n    return await Promise.race([operation, aborted]);\n  } finally {\n    abortSignal.removeEventListener('abort', onAbort);\n  }\n}\n\nfunction throwIfAborted(abortSignal: AbortSignal | undefined): void {\n  if (abortSignal?.aborted) {\n    throw abortReason(abortSignal);\n  }\n}\n\nfunction abortReason(abortSignal: AbortSignal): unknown {\n  return (\n    abortSignal.reason ??\n    new DOMException('The operation was aborted.', 'AbortError')\n  );\n}\n"],"mappings":"AAAA,SAAS,gBAAgB;AACzB,SAAS,oCAAoC;AAC7C;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,qCAAqC;AAQ9C;AAAA,EACE;AAAA,EACA;AAAA,OACK;AAYP,eAAsB,eAAe;AAAA,EACnC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,eAAe;AACjB,GAWsC;AACpC,iBAAe,qBAAqB,WAAW;AAE/C,QAAM,WAAW,MAAM,QAAQ;AAC/B,MAAI,CAAC,UAAU;AACb,UAAM,IAAI,kBAAkB,iBAAiB,QAAQ,IAAI;AAAA,MACvD;AAAA,MACA,gBAAgB,OAAO,KAAK,KAAK;AAAA,IACnC,CAAC;AAAA,EACH;AACA,MAAI,SAAS,WAAW,MAAM;AAC5B,UAAM,IAAI,kBAAkB,SAAS,QAAQ,8BAA8B;AAAA,MACzE;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,QAAQ,cAAc,KAAK,SAAY,KAAK,MAAM,SAAS;AACjE,yBAAuB,OAAO,mBAAmB,SAAS,QAAQ,SAAS;AAE3E,QAAM,aAAa,MAAM;AAAA,IACvB,kBAAkB,SAAS,aAAa,KAAK;AAAA,IAC7C,qBAAqB;AAAA,EACvB;AACA,MAAI,CAAC,WAAW,SAAS;AACvB,UAAM,IAAI;AAAA,MACR,2BAA2B,QAAQ,MAAM,WAAW,MAAM,OAAO;AAAA,MACjE,EAAE,UAAU,OAAO,OAAO,WAAW,MAAM,QAAQ;AAAA,IACrD;AAAA,EACF;AAEA,QAAM,mBAAiD;AAAA,IACrD,GAAG;AAAA,IACH;AAAA,IACA,GAAI,sBAAsB,SAAY,EAAE,kBAAkB,IAAI,CAAC;AAAA,EACjE;AAEA,QAAM,gBACJ,CAAC,gBACA,MAAM;AAAA,IACL,iBAAiB,UAAU,WAAW,OAAO,gBAAgB;AAAA,IAC7D,iBAAiB;AAAA,EACnB;AAEF,MAAI,eAAe;AACjB,QAAI,gBAAgB,UAAU,SAAS,aAAa;AAClD,aAAO;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA,OAAO,WAAW;AAAA,QAClB;AAAA,QACA,SAAS,EAAE,MAAM,6BAA6B;AAAA,MAChD;AAAA,IACF;AAEA,UAAM,WAAW,MAAM;AAAA,MACrB,QAAQ;AAAA,QACN,gBAAgB,UAAU,qBAAqB;AAAA,UAC7C;AAAA,UACA,OAAO,WAAW;AAAA,UAClB;AAAA,QACF,CAAC;AAAA,MACH;AAAA,MACA,qBAAqB;AAAA,IACvB;AACA,QAAI,aAAa,QAAW;AAC1B,YAAM,IAAI;AAAA,QACR;AAAA,QACA,WAAW;AAAA,QACX;AAAA,MACF;AAAA,IACF;AACA,UAAM,WACJ,OAAO,aAAa,WAChB,aAAa,aACb,UAAU;AAChB,UAAM,SAAS,OAAO,aAAa,WAAW,SAAY,UAAU;AACpE,QAAI,OAAO,aAAa,WAAW;AACjC,YAAM,IAAI;AAAA,QACR,SAAS,QAAQ;AAAA,QACjB,EAAE,UAAU,WAAW;AAAA,MACzB;AAAA,IACF;AACA,QAAI,WAAW,UAAa,OAAO,WAAW,UAAU;AACtD,YAAM,IAAI;AAAA,QACR,SAAS,QAAQ;AAAA,QACjB,EAAE,UAAU,WAAW;AAAA,MACzB;AAAA,IACF;AACA,QAAI,CAAC,UAAU;AACb,YAAM,IAAI;AAAA,QACR;AAAA,QACA,WAAW;AAAA,QACX;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,MAAI;AACJ,MAAI;AACF,aAAS,MAAM;AAAA,MACb,gBAAgB,SAAS,QAAQ,KAAK,QAAQ,GAAG;AAAA,QAC/C,OAAO,WAAW;AAAA,QAClB,SAAS;AAAA,MACX,CAAC;AAAA,MACD,iBAAiB;AAAA,IACnB;AAAA,EACF,SAAS,OAAO;AACd,QAAI,8BAA8B,KAAK,GAAG;AACxC,aAAO;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA,OAAO,WAAW;AAAA,QAClB;AAAA,QACA,SAAS,MAAM;AAAA,MACjB;AAAA,IACF;AACA,UAAM;AAAA,EACR;AACA,SAAO;AAAA,IACL,MAAM;AAAA,IACN,WAAW;AAAA,MACT;AAAA,MACA;AAAA,MACA,SAAS,QAAQ;AAAA,IACnB;AAAA,EACF;AACF;AAEA,eAAe,iBACb,UACA,OACA,SACkB;AAClB,MAAI,SAAS,iBAAiB,MAAM;AAClC,WAAO;AAAA,EACT;AACA,MAAI,OAAO,SAAS,kBAAkB,WAAW;AAC/C,WAAO,SAAS;AAAA,EAClB;AACA,SAAO,MAAM,SAAS,cAAc,OAAgB,OAAgB;AACtE;AAEA,eAAe,kBACb,QACA,OAGA;AACA,QAAM,mBAAmB,SAAS,MAAM;AACxC,MAAI,iBAAiB,aAAa,QAAW;AAC3C,WAAO,EAAE,SAAS,MAAM,OAAO,MAAM;AAAA,EACvC;AACA,SAAO,MAAM,iBAAiB,SAAS,KAAK;AAC9C;AAEA,eAAe,gBACb,SACA;AAAA,EACE;AAAA,EACA;AACF,GAIkB;AAClB,QAAM,SAAS,QAAQ,OAAgB,OAAgB;AACvD,MAAI,gBAAgB,MAAM,GAAG;AAC3B,QAAI;AACJ,qBAAiB,QAAQ,QAAQ;AAC/B,oBAAc;AAAA,IAChB;AACA,WAAO;AAAA,EACT;AACA,SAAO,MAAM;AACf;AAEA,SAAS,gBAAgB,OAAiD;AACxE,SACE,OAAO,UAAU,YACjB,UAAU,QACV,OAAO,iBAAiB,SACxB,OAAQ,MACN,OAAO,aACT,MAAM;AAEV;AAEA,eAAe,iBACb,WACA,aACY;AACZ,MAAI,gBAAgB,QAAW;AAC7B,WAAO,MAAM;AAAA,EACf;AACA,iBAAe,WAAW;AAE1B,MAAI;AACJ,QAAM,UAAU,IAAI,QAAe,CAAC,UAAU,WAAW;AACvD,oBAAgB;AAAA,EAClB,CAAC;AACD,QAAM,UAAU,MAAM;AACpB,kBAAc,YAAY,WAAW,CAAC;AAAA,EACxC;AAEA,cAAY,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;AAC7D,MAAI,YAAY,SAAS;AACvB,YAAQ;AAAA,EACV;AAEA,MAAI;AACF,WAAO,MAAM,QAAQ,KAAK,CAAC,WAAW,OAAO,CAAC;AAAA,EAChD,UAAE;AACA,gBAAY,oBAAoB,SAAS,OAAO;AAAA,EAClD;AACF;AAEA,SAAS,eAAe,aAA4C;AAClE,MAAI,aAAa,SAAS;AACxB,UAAM,YAAY,WAAW;AAAA,EAC/B;AACF;AAEA,SAAS,YAAY,aAAmC;AACtD,SACE,YAAY,UACZ,IAAI,aAAa,8BAA8B,YAAY;AAE/D;","names":[]}
+{"version":3,"sources":["../src/tool-invocation.ts"],"sourcesContent":["import { asSchema } from 'ai';\nimport { CODE_MODE_TOOL_APPROVAL_KIND } from './approval.js';\nimport {\n  CodeModeProtocolError,\n  CodeModeToolApprovalDeniedError,\n  CodeModeToolApprovalRequiredError,\n  CodeModeToolError,\n} from './errors.js';\nimport type {\n  CodeModeInterruptExecutionContext,\n  CodeModeInterruptPayload,\n  CodeModeOptions,\n  CodeModeToolExecutionOptions,\n  CodeModeToolSet,\n} from './types.js';\nimport {\n  assertJsonSerializable,\n  fromJsonPayload,\n  toJsonPayload,\n} from './utils/serialization.js';\n\nexport type HostToolInvocationResult =\n  | { type: 'success'; valueJson: string }\n  | {\n      type: 'interrupted';\n      toolName: string;\n      input: unknown;\n      toolCallId: string;\n      payload: CodeModeInterruptPayload;\n    };\n\nexport async function invokeHostTool({\n  toolName,\n  inputJson,\n  tools,\n  baseExecutionOptions,\n  codeModeOptions,\n  maxToolInputBytes,\n  maxToolOutputBytes,\n  toolCallId,\n  codeModeInterrupt,\n  skipApproval = false,\n}: {\n  toolName: string;\n  inputJson: string;\n  tools: CodeModeToolSet;\n  baseExecutionOptions: CodeModeToolExecutionOptions;\n  codeModeOptions: CodeModeOptions;\n  maxToolInputBytes: number;\n  maxToolOutputBytes: number;\n  toolCallId: string;\n  codeModeInterrupt?: CodeModeInterruptExecutionContext;\n  skipApproval?: boolean;\n}): Promise<HostToolInvocationResult> {\n  throwIfAborted(baseExecutionOptions.abortSignal);\n\n  const hostTool = tools[toolName];\n  if (!hostTool) {\n    throw new CodeModeToolError(`Unknown tool: ${toolName}`, {\n      toolName,\n      availableTools: Object.keys(tools),\n    });\n  }\n  if (hostTool.execute == null) {\n    throw new CodeModeToolError(`Tool \"${toolName}\" does not have execute().`, {\n      toolName,\n    });\n  }\n\n  const input = fromJsonPayload(inputJson);\n  assertJsonSerializable(input, maxToolInputBytes, `Tool \"${toolName}\" input`);\n\n  const validation = await raceAgainstAbort(\n    validateToolInput(hostTool.inputSchema, input),\n    baseExecutionOptions.abortSignal,\n  );\n  if (!validation.success) {\n    throw new CodeModeToolError(\n      `Invalid input for tool \"${toolName}\": ${validation.error.message}`,\n      { toolName, input, cause: validation.error.message },\n    );\n  }\n\n  const executionOptions: CodeModeToolExecutionOptions = {\n    ...baseExecutionOptions,\n    toolCallId,\n    ...(codeModeInterrupt !== undefined ? { codeModeInterrupt } : {}),\n  };\n\n  const needsApproval =\n    !skipApproval &&\n    (await raceAgainstAbort(\n      requiresApproval(hostTool, validation.value, executionOptions),\n      executionOptions.abortSignal,\n    ));\n\n  if (needsApproval) {\n    if (codeModeOptions.approval?.mode === 'interrupt') {\n      return {\n        type: 'interrupted',\n        toolName,\n        input: validation.value,\n        toolCallId,\n        payload: { kind: CODE_MODE_TOOL_APPROVAL_KIND },\n      };\n    }\n\n    const approval = await raceAgainstAbort(\n      Promise.resolve(\n        codeModeOptions.approval?.onApprovalRequired?.({\n          toolName,\n          input: validation.value,\n          toolCallId,\n        }),\n      ),\n      baseExecutionOptions.abortSignal,\n    );\n    if (approval === undefined) {\n      throw new CodeModeToolApprovalRequiredError(\n        toolName,\n        validation.value,\n        toolCallId,\n      );\n    }\n    const approved =\n      typeof approval === 'string'\n        ? approval === 'approved'\n        : approval?.approved;\n    const reason = typeof approval === 'string' ? undefined : approval?.reason;\n    if (typeof approved !== 'boolean') {\n      throw new CodeModeProtocolError(\n        `Tool \"${toolName}\" approval callback returned a malformed approval decision.`,\n        { toolName, toolCallId },\n      );\n    }\n    if (reason !== undefined && typeof reason !== 'string') {\n      throw new CodeModeProtocolError(\n        `Tool \"${toolName}\" approval callback returned a malformed approval reason.`,\n        { toolName, toolCallId },\n      );\n    }\n    if (!approved) {\n      throw new CodeModeToolApprovalDeniedError(\n        toolName,\n        validation.value,\n        toolCallId,\n        reason,\n      );\n    }\n  }\n\n  const output = await raceAgainstAbort(\n    executeHostTool(hostTool.execute.bind(hostTool), {\n      input: validation.value,\n      options: executionOptions,\n    }),\n    executionOptions.abortSignal,\n  );\n  return {\n    type: 'success',\n    valueJson: toJsonPayload(\n      output,\n      maxToolOutputBytes,\n      `Tool \"${toolName}\" output`,\n    ),\n  };\n}\n\nasync function requiresApproval(\n  hostTool: CodeModeToolSet[string],\n  input: unknown,\n  options: CodeModeToolExecutionOptions,\n): Promise<boolean> {\n  if (hostTool.needsApproval == null) {\n    return false;\n  }\n  if (typeof hostTool.needsApproval === 'boolean') {\n    return hostTool.needsApproval;\n  }\n  return await hostTool.needsApproval(input as never, options as never);\n}\n\nasync function validateToolInput(\n  schema: CodeModeToolSet[string]['inputSchema'],\n  input: unknown,\n): Promise<\n  { success: true; value: unknown } | { success: false; error: Error }\n> {\n  const normalizedSchema = asSchema(schema);\n  if (normalizedSchema.validate === undefined) {\n    return { success: true, value: input };\n  }\n  return await normalizedSchema.validate(input);\n}\n\nasync function executeHostTool(\n  execute: NonNullable<CodeModeToolSet[string]['execute']>,\n  {\n    input,\n    options,\n  }: {\n    input: unknown;\n    options: CodeModeToolExecutionOptions;\n  },\n): Promise<unknown> {\n  const output = execute(input as never, options as never);\n  if (isAsyncIterable(output)) {\n    let finalOutput: unknown;\n    for await (const part of output) {\n      finalOutput = part;\n    }\n    return finalOutput;\n  }\n  return await output;\n}\n\nfunction isAsyncIterable(value: unknown): value is AsyncIterable<unknown> {\n  return (\n    typeof value === 'object' &&\n    value !== null &&\n    Symbol.asyncIterator in value &&\n    typeof (value as { [Symbol.asyncIterator]?: unknown })[\n      Symbol.asyncIterator\n    ] === 'function'\n  );\n}\n\nasync function raceAgainstAbort<T>(\n  operation: Promise<T>,\n  abortSignal: AbortSignal | undefined,\n): Promise<T> {\n  if (abortSignal === undefined) {\n    return await operation;\n  }\n  throwIfAborted(abortSignal);\n\n  let rejectOnAbort!: (reason?: unknown) => void;\n  const aborted = new Promise<never>((_resolve, reject) => {\n    rejectOnAbort = reject;\n  });\n  const onAbort = () => {\n    rejectOnAbort(abortReason(abortSignal));\n  };\n\n  abortSignal.addEventListener('abort', onAbort, { once: true });\n  if (abortSignal.aborted) {\n    onAbort();\n  }\n\n  try {\n    return await Promise.race([operation, aborted]);\n  } finally {\n    abortSignal.removeEventListener('abort', onAbort);\n  }\n}\n\nfunction throwIfAborted(abortSignal: AbortSignal | undefined): void {\n  if (abortSignal?.aborted) {\n    throw abortReason(abortSignal);\n  }\n}\n\nfunction abortReason(abortSignal: AbortSignal): unknown {\n  return (\n    abortSignal.reason ??\n    new DOMException('The operation was aborted.', 'AbortError')\n  );\n}\n"],"mappings":"AAAA,SAAS,gBAAgB;AACzB,SAAS,oCAAoC;AAC7C;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAQP;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAYP,eAAsB,eAAe;AAAA,EACnC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,eAAe;AACjB,GAWsC;AACpC,iBAAe,qBAAqB,WAAW;AAE/C,QAAM,WAAW,MAAM,QAAQ;AAC/B,MAAI,CAAC,UAAU;AACb,UAAM,IAAI,kBAAkB,iBAAiB,QAAQ,IAAI;AAAA,MACvD;AAAA,MACA,gBAAgB,OAAO,KAAK,KAAK;AAAA,IACnC,CAAC;AAAA,EACH;AACA,MAAI,SAAS,WAAW,MAAM;AAC5B,UAAM,IAAI,kBAAkB,SAAS,QAAQ,8BAA8B;AAAA,MACzE;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,QAAQ,gBAAgB,SAAS;AACvC,yBAAuB,OAAO,mBAAmB,SAAS,QAAQ,SAAS;AAE3E,QAAM,aAAa,MAAM;AAAA,IACvB,kBAAkB,SAAS,aAAa,KAAK;AAAA,IAC7C,qBAAqB;AAAA,EACvB;AACA,MAAI,CAAC,WAAW,SAAS;AACvB,UAAM,IAAI;AAAA,MACR,2BAA2B,QAAQ,MAAM,WAAW,MAAM,OAAO;AAAA,MACjE,EAAE,UAAU,OAAO,OAAO,WAAW,MAAM,QAAQ;AAAA,IACrD;AAAA,EACF;AAEA,QAAM,mBAAiD;AAAA,IACrD,GAAG;AAAA,IACH;AAAA,IACA,GAAI,sBAAsB,SAAY,EAAE,kBAAkB,IAAI,CAAC;AAAA,EACjE;AAEA,QAAM,gBACJ,CAAC,gBACA,MAAM;AAAA,IACL,iBAAiB,UAAU,WAAW,OAAO,gBAAgB;AAAA,IAC7D,iBAAiB;AAAA,EACnB;AAEF,MAAI,eAAe;AACjB,QAAI,gBAAgB,UAAU,SAAS,aAAa;AAClD,aAAO;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA,OAAO,WAAW;AAAA,QAClB;AAAA,QACA,SAAS,EAAE,MAAM,6BAA6B;AAAA,MAChD;AAAA,IACF;AAEA,UAAM,WAAW,MAAM;AAAA,MACrB,QAAQ;AAAA,QACN,gBAAgB,UAAU,qBAAqB;AAAA,UAC7C;AAAA,UACA,OAAO,WAAW;AAAA,UAClB;AAAA,QACF,CAAC;AAAA,MACH;AAAA,MACA,qBAAqB;AAAA,IACvB;AACA,QAAI,aAAa,QAAW;AAC1B,YAAM,IAAI;AAAA,QACR;AAAA,QACA,WAAW;AAAA,QACX;AAAA,MACF;AAAA,IACF;AACA,UAAM,WACJ,OAAO,aAAa,WAChB,aAAa,aACb,UAAU;AAChB,UAAM,SAAS,OAAO,aAAa,WAAW,SAAY,UAAU;AACpE,QAAI,OAAO,aAAa,WAAW;AACjC,YAAM,IAAI;AAAA,QACR,SAAS,QAAQ;AAAA,QACjB,EAAE,UAAU,WAAW;AAAA,MACzB;AAAA,IACF;AACA,QAAI,WAAW,UAAa,OAAO,WAAW,UAAU;AACtD,YAAM,IAAI;AAAA,QACR,SAAS,QAAQ;AAAA,QACjB,EAAE,UAAU,WAAW;AAAA,MACzB;AAAA,IACF;AACA,QAAI,CAAC,UAAU;AACb,YAAM,IAAI;AAAA,QACR;AAAA,QACA,WAAW;AAAA,QACX;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,QAAM,SAAS,MAAM;AAAA,IACnB,gBAAgB,SAAS,QAAQ,KAAK,QAAQ,GAAG;AAAA,MAC/C,OAAO,WAAW;AAAA,MAClB,SAAS;AAAA,IACX,CAAC;AAAA,IACD,iBAAiB;AAAA,EACnB;AACA,SAAO;AAAA,IACL,MAAM;AAAA,IACN,WAAW;AAAA,MACT;AAAA,MACA;AAAA,MACA,SAAS,QAAQ;AAAA,IACnB;AAAA,EACF;AACF;AAEA,eAAe,iBACb,UACA,OACA,SACkB;AAClB,MAAI,SAAS,iBAAiB,MAAM;AAClC,WAAO;AAAA,EACT;AACA,MAAI,OAAO,SAAS,kBAAkB,WAAW;AAC/C,WAAO,SAAS;AAAA,EAClB;AACA,SAAO,MAAM,SAAS,cAAc,OAAgB,OAAgB;AACtE;AAEA,eAAe,kBACb,QACA,OAGA;AACA,QAAM,mBAAmB,SAAS,MAAM;AACxC,MAAI,iBAAiB,aAAa,QAAW;AAC3C,WAAO,EAAE,SAAS,MAAM,OAAO,MAAM;AAAA,EACvC;AACA,SAAO,MAAM,iBAAiB,SAAS,KAAK;AAC9C;AAEA,eAAe,gBACb,SACA;AAAA,EACE;AAAA,EACA;AACF,GAIkB;AAClB,QAAM,SAAS,QAAQ,OAAgB,OAAgB;AACvD,MAAI,gBAAgB,MAAM,GAAG;AAC3B,QAAI;AACJ,qBAAiB,QAAQ,QAAQ;AAC/B,oBAAc;AAAA,IAChB;AACA,WAAO;AAAA,EACT;AACA,SAAO,MAAM;AACf;AAEA,SAAS,gBAAgB,OAAiD;AACxE,SACE,OAAO,UAAU,YACjB,UAAU,QACV,OAAO,iBAAiB,SACxB,OAAQ,MACN,OAAO,aACT,MAAM;AAEV;AAEA,eAAe,iBACb,WACA,aACY;AACZ,MAAI,gBAAgB,QAAW;AAC7B,WAAO,MAAM;AAAA,EACf;AACA,iBAAe,WAAW;AAE1B,MAAI;AACJ,QAAM,UAAU,IAAI,QAAe,CAAC,UAAU,WAAW;AACvD,oBAAgB;AAAA,EAClB,CAAC;AACD,QAAM,UAAU,MAAM;AACpB,kBAAc,YAAY,WAAW,CAAC;AAAA,EACxC;AAEA,cAAY,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;AAC7D,MAAI,YAAY,SAAS;AACvB,YAAQ;AAAA,EACV;AAEA,MAAI;AACF,WAAO,MAAM,QAAQ,KAAK,CAAC,WAAW,OAAO,CAAC;AAAA,EAChD,UAAE;AACA,gBAAY,oBAAoB,SAAS,OAAO;AAAA,EAClD;AACF;AAEA,SAAS,eAAe,aAA4C;AAClE,MAAI,aAAa,SAAS;AACxB,UAAM,YAAY,WAAW;AAAA,EAC/B;AACF;AAEA,SAAS,YAAY,aAAmC;AACtD,SACE,YAAY,UACZ,IAAI,aAAa,8BAA8B,YAAY;AAE/D;","names":[]}

package/dist/utils/serialization.js

@@ -8,6 +8,9 @@ function toJsonPayload(value, maxBytes, label) {
   }
   return toStrictJsonPayload(value, maxBytes, label);
 }
+function fromJsonPayload(valueJson) {
+  return valueJson === "" ? void 0 : JSON.parse(valueJson);
+}
 function toStrictJsonPayload(value, maxBytes, label) {
   let encoded;
   try {
@@ -43,6 +46,7 @@ function getErrorMessage(error) {
 export {
   assertJsonPayloadSize,
   assertJsonSerializable,
+  fromJsonPayload,
   toJsonPayload,
   toStrictJsonPayload
 };

package/dist/utils/serialization.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/utils/serialization.ts"],"sourcesContent":["import { CodeModeError } from '../errors.js';\n\nexport function assertJsonSerializable(\n  value: unknown,\n  maxBytes: number,\n  label: string,\n): void {\n  void toJsonPayload(value, maxBytes, label);\n}\n\nexport function toJsonPayload(\n  value: unknown,\n  maxBytes: number,\n  label: string,\n): string {\n  if (value === undefined) {\n    return '';\n  }\n\n  return toStrictJsonPayload(value, maxBytes, label);\n}\n\nexport function toStrictJsonPayload(\n  value: unknown,\n  maxBytes: number,\n  label: string,\n): string {\n  let encoded: string | undefined;\n  try {\n    encoded = JSON.stringify(value);\n  } catch (error) {\n    throw new CodeModeError(\n      `${label} is not JSON-serializable: ${getErrorMessage(error)}`,\n      'CODE_MODE_SERIALIZATION_ERROR',\n    );\n  }\n\n  if (encoded === undefined) {\n    throw new CodeModeError(\n      `${label} is not JSON-serializable.`,\n      'CODE_MODE_SERIALIZATION_ERROR',\n    );\n  }\n\n  assertJsonPayloadSize(encoded, maxBytes, label);\n  return encoded;\n}\n\nexport function assertJsonPayloadSize(\n  valueJson: string,\n  maxBytes: number,\n  label: string,\n): void {\n  const bytes = new TextEncoder().encode(valueJson).byteLength;\n  if (bytes > maxBytes) {\n    throw new CodeModeError(\n      `${label} exceeds the ${maxBytes} byte size limit.`,\n      'CODE_MODE_SERIALIZATION_ERROR',\n      { bytes, maxBytes },\n    );\n  }\n}\n\nfunction getErrorMessage(error: unknown): string {\n  return error instanceof Error ? error.message : String(error);\n}\n"],"mappings":"AAAA,SAAS,qBAAqB;AAEvB,SAAS,uBACd,OACA,UACA,OACM;AACN,OAAK,cAAc,OAAO,UAAU,KAAK;AAC3C;AAEO,SAAS,cACd,OACA,UACA,OACQ;AACR,MAAI,UAAU,QAAW;AACvB,WAAO;AAAA,EACT;AAEA,SAAO,oBAAoB,OAAO,UAAU,KAAK;AACnD;AAEO,SAAS,oBACd,OACA,UACA,OACQ;AACR,MAAI;AACJ,MAAI;AACF,cAAU,KAAK,UAAU,KAAK;AAAA,EAChC,SAAS,OAAO;AACd,UAAM,IAAI;AAAA,MACR,GAAG,KAAK,8BAA8B,gBAAgB,KAAK,CAAC;AAAA,MAC5D;AAAA,IACF;AAAA,EACF;AAEA,MAAI,YAAY,QAAW;AACzB,UAAM,IAAI;AAAA,MACR,GAAG,KAAK;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,wBAAsB,SAAS,UAAU,KAAK;AAC9C,SAAO;AACT;AAEO,SAAS,sBACd,WACA,UACA,OACM;AACN,QAAM,QAAQ,IAAI,YAAY,EAAE,OAAO,SAAS,EAAE;AAClD,MAAI,QAAQ,UAAU;AACpB,UAAM,IAAI;AAAA,MACR,GAAG,KAAK,gBAAgB,QAAQ;AAAA,MAChC;AAAA,MACA,EAAE,OAAO,SAAS;AAAA,IACpB;AAAA,EACF;AACF;AAEA,SAAS,gBAAgB,OAAwB;AAC/C,SAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAC9D;","names":[]}
+{"version":3,"sources":["../../src/utils/serialization.ts"],"sourcesContent":["import { CodeModeError } from '../errors.js';\n\nexport function assertJsonSerializable(\n  value: unknown,\n  maxBytes: number,\n  label: string,\n): void {\n  void toJsonPayload(value, maxBytes, label);\n}\n\nexport function toJsonPayload(\n  value: unknown,\n  maxBytes: number,\n  label: string,\n): string {\n  if (value === undefined) {\n    return '';\n  }\n\n  return toStrictJsonPayload(value, maxBytes, label);\n}\n\nexport function fromJsonPayload(valueJson: string): unknown {\n  return valueJson === '' ? undefined : JSON.parse(valueJson);\n}\n\nexport function toStrictJsonPayload(\n  value: unknown,\n  maxBytes: number,\n  label: string,\n): string {\n  let encoded: string | undefined;\n  try {\n    encoded = JSON.stringify(value);\n  } catch (error) {\n    throw new CodeModeError(\n      `${label} is not JSON-serializable: ${getErrorMessage(error)}`,\n      'CODE_MODE_SERIALIZATION_ERROR',\n    );\n  }\n\n  if (encoded === undefined) {\n    throw new CodeModeError(\n      `${label} is not JSON-serializable.`,\n      'CODE_MODE_SERIALIZATION_ERROR',\n    );\n  }\n\n  assertJsonPayloadSize(encoded, maxBytes, label);\n  return encoded;\n}\n\nexport function assertJsonPayloadSize(\n  valueJson: string,\n  maxBytes: number,\n  label: string,\n): void {\n  const bytes = new TextEncoder().encode(valueJson).byteLength;\n  if (bytes > maxBytes) {\n    throw new CodeModeError(\n      `${label} exceeds the ${maxBytes} byte size limit.`,\n      'CODE_MODE_SERIALIZATION_ERROR',\n      { bytes, maxBytes },\n    );\n  }\n}\n\nfunction getErrorMessage(error: unknown): string {\n  return error instanceof Error ? error.message : String(error);\n}\n"],"mappings":"AAAA,SAAS,qBAAqB;AAEvB,SAAS,uBACd,OACA,UACA,OACM;AACN,OAAK,cAAc,OAAO,UAAU,KAAK;AAC3C;AAEO,SAAS,cACd,OACA,UACA,OACQ;AACR,MAAI,UAAU,QAAW;AACvB,WAAO;AAAA,EACT;AAEA,SAAO,oBAAoB,OAAO,UAAU,KAAK;AACnD;AAEO,SAAS,gBAAgB,WAA4B;AAC1D,SAAO,cAAc,KAAK,SAAY,KAAK,MAAM,SAAS;AAC5D;AAEO,SAAS,oBACd,OACA,UACA,OACQ;AACR,MAAI;AACJ,MAAI;AACF,cAAU,KAAK,UAAU,KAAK;AAAA,EAChC,SAAS,OAAO;AACd,UAAM,IAAI;AAAA,MACR,GAAG,KAAK,8BAA8B,gBAAgB,KAAK,CAAC;AAAA,MAC5D;AAAA,IACF;AAAA,EACF;AAEA,MAAI,YAAY,QAAW;AACzB,UAAM,IAAI;AAAA,MACR,GAAG,KAAK;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,wBAAsB,SAAS,UAAU,KAAK;AAC9C,SAAO;AACT;AAEO,SAAS,sBACd,WACA,UACA,OACM;AACN,QAAM,QAAQ,IAAI,YAAY,EAAE,OAAO,SAAS,EAAE;AAClD,MAAI,QAAQ,UAAU;AACpB,UAAM,IAAI;AAAA,MACR,GAAG,KAAK,gBAAgB,QAAQ;AAAA,MAChC;AAAA,MACA,EAAE,OAAO,SAAS;AAAA,IACpB;AAAA,EACF;AACF;AAEA,SAAS,gBAAgB,OAAwB;AAC/C,SAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAC9D;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/code-mode",
-  "version": "1.0.21",
+  "version": "1.0.22",
   "description": "QuickJS-backed code mode tool for AI SDK",
   "keywords": [
     "ai",
@@ -37,20 +37,18 @@
     "README.md"
   ],
   "peerDependencies": {
-    "ai": "7.0.64"
+    "ai": "7.0.65"
   },
   "devDependencies": {
     "@types/node": "22.19.19",
-    "esbuild": "^0.28.1",
-    "quickjs-emscripten": "^0.32.0",
     "tsup": "^8.5.1",
     "typescript": "5.8.3",
     "zod": "3.25.76",
     "@vercel/ai-tsconfig": "0.0.0",
-    "ai": "7.0.64"
+    "ai": "7.0.65"
   },
   "engines": {
-    "node": ">=22"
+    "node": ">=22.13.0"
   },
   "publishConfig": {
     "access": "public",
@@ -65,8 +63,11 @@
   "bugs": {
     "url": "https://github.com/vercel/ai/issues"
   },
+  "dependencies": {
+    "run": "^2.0.0"
+  },
   "scripts": {
-    "build": "pnpm clean && tsup --tsconfig tsconfig.build.json && node scripts/embed-inline-worker.mjs",
+    "build": "pnpm clean && tsup --tsconfig tsconfig.build.json",
     "clean": "del-cli build dist *.tsbuildinfo",
     "type-check": "tsc --build && tsc -p src/tsconfig.json",
     "test": "pnpm test:node",

package/src/continuation-capability.ts

@@ -82,6 +82,24 @@ export function verifyCodeModeContinuation(
   continuation: CodeModeContinuation,
   security: CodeModeContinuationSecurityOptions = {},
 ): void {
+  if (
+    typeof continuation !== 'object' ||
+    continuation === null ||
+    continuation.version !== 2 ||
+    typeof continuation.js !== 'string' ||
+    typeof continuation.outerToolCallId !== 'string' ||
+    !Array.isArray(continuation.toolNames) ||
+    !continuation.toolNames.every(name => typeof name === 'string') ||
+    typeof continuation.token !== 'string' ||
+    continuation.token.length === 0 ||
+    !Array.isArray(continuation.pendingInterruptions) ||
+    continuation.pendingInterruptions.length === 0 ||
+    !Array.isArray(continuation.resolutions)
+  ) {
+    throw new CodeModeProtocolError(
+      'Code mode continuation envelope is malformed.',
+    );
+  }
   assertAuthShape(continuation.auth);
   const now = Date.now();
   if (continuation.auth.expiresAtMs < now) {

package/src/errors.ts

@@ -1,5 +1,3 @@
-import type { SerializableError } from './types.js';
-
 /**
  * Base class for errors raised by code mode.
  *
@@ -142,191 +140,3 @@ export class CodeModeToolApprovalDeniedError extends CodeModeToolError {
     this.code = 'CODE_MODE_TOOL_APPROVAL_DENIED';
   }
 }
-
-/**
- * Converts an unknown thrown value into a worker-safe serializable shape.
- *
- * @internal
- */
-export function serializeError(error: unknown): SerializableError {
-  if (error instanceof CodeModeError) {
-    const result = compactError({
-      name: error.name,
-      message: error.message,
-      code: error.code,
-    });
-    if (error.stack !== undefined) {
-      result.stack = error.stack;
-    }
-    if (error.details !== undefined) {
-      result.details = error.details;
-    }
-    return result;
-  }
-
-  if (error instanceof Error) {
-    const result = compactError({
-      name: error.name,
-      message: error.message,
-    });
-    if (error.stack !== undefined) {
-      result.stack = error.stack;
-    }
-    const maybeCoded = error as Error & { code?: unknown; details?: unknown };
-    if (typeof maybeCoded.code === 'string') {
-      result.code = maybeCoded.code;
-    }
-    if (maybeCoded.details !== undefined) {
-      result.details = maybeCoded.details;
-    }
-    return result;
-  }
-
-  return {
-    name: 'Error',
-    message: String(error),
-  };
-}
-
-/**
- * Converts host bridge failures into a sandbox-visible sanitized shape.
- *
- * This intentionally omits stack traces and diagnostic details. Full host
- * diagnostics remain on the host side.
- *
- * @internal
- */
-export function serializeBridgeErrorForGuest(
-  error: unknown,
-  context: 'tool' | 'bridge',
-): SerializableError {
-  if (error instanceof CodeModeError) {
-    return compactError({
-      name: error.name,
-      message: error.message,
-      code: error.code,
-    });
-  }
-
-  const fallback =
-    context === 'tool'
-      ? {
-          message: 'Host tool failed.',
-          code: 'CODE_MODE_HOST_TOOL_ERROR',
-        }
-      : {
-          message: 'Host bridge request failed.',
-          code: 'CODE_MODE_HOST_BRIDGE_ERROR',
-        };
-
-  return {
-    name: 'Error',
-    message: fallback.message,
-    code: fallback.code,
-  };
-}
-
-function compactError(error: {
-  name: string;
-  message: string;
-  stack?: string;
-  code?: string;
-  details?: unknown;
-}): SerializableError {
-  return {
-    name: error.name,
-    message: error.message,
-    ...(error.stack !== undefined ? { stack: error.stack } : {}),
-    ...(error.code !== undefined ? { code: error.code } : {}),
-    ...(error.details !== undefined ? { details: error.details } : {}),
-  };
-}
-
-/**
- * Rehydrates a serialized worker error into an Error instance.
- *
- * @internal
- */
-export function deserializeError(error: SerializableError): Error {
-  if (error.code === 'CODE_MODE_SOURCE_TOO_LARGE') {
-    const details = error.details as
-      | { bytes?: number; maxBytes?: number }
-      | undefined;
-    const result = new CodeModeSourceTooLargeError(
-      details?.bytes ?? 0,
-      details?.maxBytes ?? 0,
-    );
-    restoreStack(result, error);
-    return result;
-  }
-
-  if (error.code === 'CODE_MODE_BRIDGE_LIMIT') {
-    const result = new CodeModeBridgeLimitError(error.message, error.details);
-    restoreStack(result, error);
-    return result;
-  }
-
-  if (error.code === 'CODE_MODE_DETACHED_BRIDGE_REQUEST') {
-    const result = new CodeModeDetachedBridgeRequestError(
-      error.message,
-      error.details,
-    );
-    restoreStack(result, error);
-    return result;
-  }
-
-  if (error.code === 'CODE_MODE_PROTOCOL_ERROR') {
-    const result = new CodeModeProtocolError(error.message, error.details);
-    restoreStack(result, error);
-    return result;
-  }
-
-  if (error.code === 'CODE_MODE_TOOL_APPROVAL_REQUIRED') {
-    const details = error.details as
-      | { toolName?: string; input?: unknown; toolCallId?: string }
-      | undefined;
-    const result = new CodeModeToolApprovalRequiredError(
-      details?.toolName ?? 'unknown',
-      details?.input,
-      details?.toolCallId ?? 'unknown',
-    );
-    result.message = error.message;
-    restoreStack(result, error);
-    return result;
-  }
-
-  if (error.code === 'CODE_MODE_TOOL_APPROVAL_DENIED') {
-    const details = error.details as
-      | {
-          toolName?: string;
-          input?: unknown;
-          toolCallId?: string;
-          reason?: string;
-        }
-      | undefined;
-    const result = new CodeModeToolApprovalDeniedError(
-      details?.toolName ?? 'unknown',
-      details?.input,
-      details?.toolCallId ?? 'unknown',
-      details?.reason,
-    );
-    result.message = error.message;
-    restoreStack(result, error);
-    return result;
-  }
-
-  const result = new CodeModeError(
-    error.message,
-    error.code ?? 'CODE_MODE_ERROR',
-    error.details,
-  );
-  result.name = error.name;
-  restoreStack(result, error);
-  return result;
-}
-
-function restoreStack(error: Error, serialized: SerializableError): void {
-  if (serialized.stack) {
-    error.stack = serialized.stack;
-  }
-}

package/src/host-interrupt.ts

@@ -1,14 +1,6 @@
+import { getHostFunctionContext } from 'run';
 import type { CodeModeInterruptPayload } from './types.js';
 
-const CODE_MODE_HOST_INTERRUPT_SIGNAL = Symbol.for(
-  '@ai-sdk/code-mode.host-interrupt',
-);
-
-interface CodeModeHostInterruptSignal extends Error {
-  readonly [CODE_MODE_HOST_INTERRUPT_SIGNAL]: true;
-  readonly payload: CodeModeInterruptPayload;
-}
-
 export function requestCodeModeInterrupt<
   TPayload extends CodeModeInterruptPayload,
 >(payload: TPayload): never {
@@ -24,26 +16,5 @@ export function requestCodeModeInterrupt<
       'Code mode interrupt payload must include a string kind.',
     );
   }
-
-  const error = new Error('Code mode host interruption requested.');
-  error.name = 'CodeModeHostInterruptSignal';
-  Object.defineProperties(error, {
-    [CODE_MODE_HOST_INTERRUPT_SIGNAL]: { value: true },
-    payload: { value: payload },
-  });
-  throw error;
-}
-
-export function isCodeModeHostInterruptSignal(
-  value: unknown,
-): value is CodeModeHostInterruptSignal {
-  return (
-    typeof value === 'object' &&
-    value !== null &&
-    (value as { [CODE_MODE_HOST_INTERRUPT_SIGNAL]?: unknown })[
-      CODE_MODE_HOST_INTERRUPT_SIGNAL
-    ] === true &&
-    typeof (value as { payload?: unknown }).payload === 'object' &&
-    (value as { payload?: unknown }).payload !== null
-  );
+  return getHostFunctionContext().interrupt(payload);
 }

package/src/index.ts

@@ -31,7 +31,7 @@ export {
 } from './interrupt-continuation.js';
 export { runCodeMode as experimental_runCodeMode } from './run-code-mode.js';
 export { setCodeModeContinuationSigningKey as experimental_setCodeModeContinuationSigningKey } from './continuation-capability.js';
-export { setMaxWorkers as experimental_setMaxWorkers } from './runtime/max-workers.js';
+export { setMaxWorkers as experimental_setMaxWorkers } from 'run';
 export type {
   ApprovalDecision,
   CodeModeApprovalInterrupt,

package/src/interrupt-continuation.ts

@@ -143,19 +143,21 @@ function assertInterruptMatchesLedger(interrupt: CodeModeInterrupt): void {
       'Code-mode interrupt outer tool call id does not match its continuation.',
     );
   }
-  const matches = interrupt.continuation.ledger.filter(
-    entry =>
-      entry.status === 'interrupted' &&
-      entry.interruptId === interrupt.interruptId &&
-      entry.toolCallId === interrupt.toolCallId &&
-      entry.name === interrupt.toolName &&
-      jsonEqual(fromJsonPayload(entry.inputJson), interrupt.input) &&
-      jsonEqual(entry.interruptPayload, interrupt.payload),
-  );
-  if (matches.length !== 1) {
+  const pending =
+    interrupt.continuation.pendingInterruptions[
+      interrupt.continuation.resolutions.length
+    ];
+  if (
+    pending === undefined ||
+    pending.interruptId !== interrupt.interruptId ||
+    pending.toolCallId !== interrupt.toolCallId ||
+    pending.toolName !== interrupt.toolName ||
+    !jsonEqual(pending.input, interrupt.input) ||
+    !jsonEqual(pending.payload, interrupt.payload)
+  ) {
     throw new CodeModeProtocolError(
       'Code-mode interrupt metadata does not match the signed continuation ledger.',
-      { interruptId: interrupt.interruptId, matches: matches.length },
+      { interruptId: interrupt.interruptId },
     );
   }
 }
@@ -164,10 +166,6 @@ function jsonEqual(left: unknown, right: unknown): boolean {
   return JSON.stringify(left) === JSON.stringify(right);
 }
 
-function fromJsonPayload(valueJson: string): unknown {
-  return valueJson === '' ? undefined : JSON.parse(valueJson);
-}
-
 function isRecord(value: unknown): value is Record<string, unknown> {
   return typeof value === 'object' && value !== null && !Array.isArray(value);
 }