@ai-sdk/code-mode 1.0.79 → 1.0.80

package/dist/run-code-mode.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/run-code-mode.ts"],"sourcesContent":["import { Buffer } from 'node:buffer';\nimport { createHash } from 'node:crypto';\nimport {\n  createRunner,\n  createSignedContinuationCodec,\n  getHostFunctionContext,\n  RunError,\n  type HostFunctions,\n  type RunInterruption,\n  type RunLimits,\n  type RunResolution,\n} from 'run';\nimport {\n  isCodeModeApprovalInterruptPayload,\n  normalizeApprovalResolution,\n} from './approval.js';\nimport {\n  resolveCodeModeContinuationSecurity,\n  signCodeModeContinuation,\n  verifyCodeModeContinuation,\n} from './continuation-capability.js';\nimport {\n  CodeModeAbortedError,\n  CodeModeBridgeLimitError,\n  CodeModeConcurrencyError,\n  CodeModeDetachedBridgeRequestError,\n  CodeModeError,\n  CodeModeProtocolError,\n  CodeModeSourceTooLargeError,\n  CodeModeTimeoutError,\n  CodeModeToolApprovalDeniedError,\n} from './errors.js';\nimport { invokeHostTool } from './tool-invocation.js';\nimport type {\n  CodeModeContinuation,\n  CodeModeExecutionPolicy,\n  CodeModeInterrupt,\n  CodeModeInterruptExecutionContext,\n  CodeModeInterruptPayload,\n  CodeModePendingInterruption,\n  CodeModePendingResolution,\n  CodeModeToolExecutionOptions,\n  RunCodeModeInput,\n} from './types.js';\nimport { fromJsonPayload, toJsonPayload } from './utils/serialization.js';\n\nconst SOURCE_LINE_OFFSET = 1;\nconst MAX_RUN_LIMIT = 2_147_483_647;\nconst DEFAULT_TIMEOUT_MS = 30_000;\nconst DEFAULT_MEMORY_LIMIT_BYTES = 64 * 1024 * 1024;\nconst DEFAULT_STACK_LIMIT_BYTES = 2 * 1024 * 1024;\nconst DEFAULT_MAX_RESULT_BYTES = 1024 * 1024;\nconst DEFAULT_MAX_CONSOLE_OUTPUT_BYTES = 64 * 1024;\nconst DEFAULT_MAX_SOURCE_BYTES = 256 * 1024;\nconst DEFAULT_MAX_TOOL_INPUT_BYTES = 1024 * 1024;\nconst DEFAULT_MAX_TOOL_OUTPUT_BYTES = 4 * 1024 * 1024;\nconst DEFAULT_MAX_BRIDGE_REQUESTS = 256;\nconst DEFAULT_MAX_IN_FLIGHT_BRIDGE_REQUESTS = 32;\n\nlet invocationCounter = 0;\n\n/**\n * Runs one code-mode invocation directly without wrapping it as an AI SDK tool.\n *\n * The source is executed in a fresh QuickJS context with the provided host tools\n * and execution limits.\n *\n * @param input - Source code, host tools, forwarded tool execution options, and code-mode options.\n * @returns The JSON-serializable value returned by the sandboxed program.\n */\nexport async function runCodeMode(input: RunCodeModeInput): Promise<unknown> {\n  const policy = resolveExecutionPolicy(input.options?.executionPolicy);\n  assertSourceSize(input.js, policy.maxSourceBytes);\n\n  const toolNames = Object.keys(input.tools).sort();\n  const continuationState = prepareContinuation(\n    input,\n    toolNames,\n    policy.maxToolOutputBytes,\n  );\n  if (continuationState.nextInterrupt !== undefined) {\n    return continuationState.nextInterrupt;\n  }\n\n  const outerToolCallId =\n    input.toolExecutionOptions?.toolCallId ??\n    input.continuation?.outerToolCallId ??\n    `code-mode-${++invocationCounter}`;\n  const source = createCodeModeSource(input.js, toolNames);\n  const codeModeErrors: CodeModeError[] = [];\n  const hostFunctions = createHostFunctions({\n    codeModeErrors,\n    input,\n    outerToolCallId,\n    policy,\n    toolNames,\n  });\n  const continuationSecurity = resolveCodeModeContinuationSecurity(\n    input.options?.continuationSecurity,\n  );\n  const runner = createRunner({\n    continuationAudience: '@ai-sdk/code-mode/v2',\n    continuationCodec: createSignedContinuationCodec({\n      maxAgeMs: continuationSecurity.maxAgeMs,\n      secret: createHash('sha256')\n        .update(continuationSecurity.signingKey)\n        .digest(),\n    }),\n  });\n\n  try {\n    const result = await runner.run({\n      source,\n      hostFunctions,\n      limits: toRunLimits(policy, source, input.js),\n      continuationContext: { outerToolCallId, toolNames },\n      ...(input.toolExecutionOptions?.abortSignal === undefined\n        ? {}\n        : { abortSignal: input.toolExecutionOptions.abortSignal }),\n      ...(continuationState.token === undefined\n        ? {}\n        : { continuation: continuationState.token }),\n      ...(continuationState.resolutions === undefined\n        ? {}\n        : { resolutions: continuationState.resolutions }),\n    });\n\n    if (result.status === 'completed') {\n      return fromJsonPayload(\n        toJsonPayload(result.value, policy.maxResultBytes, 'Code mode result'),\n      );\n    }\n\n    const pendingInterruptions = toPendingInterruptions({\n      interruptions: result.interruptions,\n      outerToolCallId,\n      toolNames,\n    });\n    const continuation = signCodeModeContinuation(\n      {\n        version: 2,\n        js: input.js,\n        outerToolCallId,\n        toolNames,\n        token: result.continuation,\n        pendingInterruptions,\n        resolutions: [],\n      },\n      continuationSecurity,\n    );\n    return toCodeModeInterrupt(continuation, 0);\n  } catch (error) {\n    const preserved = findPreservedCodeModeError(error, codeModeErrors);\n    if (preserved !== undefined) {\n      throw translateSourceStack(preserved);\n    }\n    throw translateSourceStack(toCodeModeRuntimeError(error));\n  }\n}\n\ninterface ResolvedExecutionPolicy {\n  timeoutMs: number;\n  memoryLimitBytes: number;\n  maxStackSizeBytes: number;\n  maxResultBytes: number;\n  maxConsoleOutputBytes: number;\n  maxSourceBytes: number;\n  maxToolInputBytes: number;\n  maxToolOutputBytes: number;\n  maxBridgeRequests: number;\n  maxInFlightBridgeRequests: number;\n}\n\nfunction resolveExecutionPolicy(\n  policy: CodeModeExecutionPolicy = {},\n): ResolvedExecutionPolicy {\n  return {\n    timeoutMs: policy.timeoutMs ?? DEFAULT_TIMEOUT_MS,\n    memoryLimitBytes: policy.memoryLimitBytes ?? DEFAULT_MEMORY_LIMIT_BYTES,\n    maxStackSizeBytes: policy.maxStackSizeBytes ?? DEFAULT_STACK_LIMIT_BYTES,\n    maxResultBytes: policy.maxResultBytes ?? DEFAULT_MAX_RESULT_BYTES,\n    maxConsoleOutputBytes:\n      policy.maxConsoleOutputBytes ?? DEFAULT_MAX_CONSOLE_OUTPUT_BYTES,\n    maxSourceBytes: policy.maxSourceBytes ?? DEFAULT_MAX_SOURCE_BYTES,\n    maxToolInputBytes: policy.maxToolInputBytes ?? DEFAULT_MAX_TOOL_INPUT_BYTES,\n    maxToolOutputBytes:\n      policy.maxToolOutputBytes ?? DEFAULT_MAX_TOOL_OUTPUT_BYTES,\n    maxBridgeRequests: policy.maxBridgeRequests ?? DEFAULT_MAX_BRIDGE_REQUESTS,\n    maxInFlightBridgeRequests:\n      policy.maxInFlightBridgeRequests ?? DEFAULT_MAX_IN_FLIGHT_BRIDGE_REQUESTS,\n  };\n}\n\nfunction assertSourceSize(source: string, maxBytes: number): void {\n  if (!isValidRunLimit(maxBytes)) {\n    return;\n  }\n  const bytes = Buffer.byteLength(source);\n  if (bytes > maxBytes) {\n    throw new CodeModeSourceTooLargeError(bytes, maxBytes);\n  }\n}\n\nfunction createCodeModeSource(js: string, toolNames: string[]): string {\n  const bindings = Object.fromEntries(\n    toolNames.map((toolName, index) => [toolName, `__codeMode.tool${index}`]),\n  );\n  const bindingSource = Object.entries(bindings)\n    .map(([name, reference]) => `${JSON.stringify(name)}:${reference}`)\n    .join(',');\n  return `const __codeModeBindings={${bindingSource}};const tools=new Proxy(Object.create(null),{get(_target,name){const binding=__codeModeBindings[name];return typeof binding===\"function\"?(input)=>binding(input):(input)=>__codeMode.missing(String(name),input);}});const __codeModeResult=await(async()=>{\\n${js}\\n})();if(__codeModeResult===undefined)return undefined;return JSON.parse(JSON.stringify(__codeModeResult));`;\n}\n\nfunction createHostFunctions({\n  codeModeErrors,\n  input,\n  outerToolCallId,\n  policy,\n  toolNames,\n}: {\n  codeModeErrors: CodeModeError[];\n  input: RunCodeModeInput;\n  outerToolCallId: string;\n  policy: ResolvedExecutionPolicy;\n  toolNames: string[];\n}): HostFunctions {\n  const group: Record<string, (...args: unknown[]) => Promise<unknown>> = {};\n  for (const [index, toolName] of toolNames.entries()) {\n    group[`tool${index}`] = async (toolInput: unknown) =>\n      await invokeCodeModeTool({\n        codeModeErrors,\n        input,\n        outerToolCallId,\n        policy,\n        toolInput,\n        toolName,\n      });\n  }\n  group.missing = async (toolName: unknown, toolInput: unknown) =>\n    await invokeCodeModeTool({\n      codeModeErrors,\n      input,\n      outerToolCallId,\n      policy,\n      toolInput,\n      toolName: String(toolName),\n    });\n  return { __codeMode: group } as HostFunctions;\n}\n\nasync function invokeCodeModeTool({\n  codeModeErrors,\n  input,\n  outerToolCallId,\n  policy,\n  toolInput,\n  toolName,\n}: {\n  codeModeErrors: CodeModeError[];\n  input: RunCodeModeInput;\n  outerToolCallId: string;\n  policy: ResolvedExecutionPolicy;\n  toolInput: unknown;\n  toolName: string;\n}): Promise<unknown> {\n  const context = getHostFunctionContext();\n  const toolCallId = `${outerToolCallId}:tool-${context.requestIndex}`;\n  const forwardedContext =\n    input.toolExecutionOptions?.context ??\n    input.toolExecutionOptions?.experimental_context;\n  const forwardedExperimentalContext =\n    input.toolExecutionOptions?.experimental_context ??\n    input.toolExecutionOptions?.context;\n  const baseExecutionOptions: CodeModeToolExecutionOptions = {\n    toolCallId: outerToolCallId,\n    messages: input.toolExecutionOptions?.messages ?? [],\n    abortSignal: context.abortSignal,\n    ...(forwardedContext === undefined ? {} : { context: forwardedContext }),\n    ...(forwardedExperimentalContext === undefined\n      ? {}\n      : { experimental_context: forwardedExperimentalContext }),\n  };\n\n  let codeModeInterrupt: CodeModeInterruptExecutionContext | undefined;\n  let skipApproval = false;\n  if (context.resume !== undefined) {\n    const payload = assertInterruptPayload(context.resume.payload);\n    if (isCodeModeApprovalInterruptPayload(payload)) {\n      const decision = normalizeApprovalResolution(context.resume.resolution);\n      if (!decision.approved) {\n        throw new CodeModeToolApprovalDeniedError(\n          toolName,\n          toolInput,\n          toolCallId,\n          decision.reason,\n        );\n      }\n      skipApproval = true;\n    } else {\n      codeModeInterrupt = {\n        interruptId: `${toolCallId}:interrupt`,\n        payload,\n        resolution: context.resume.resolution,\n      };\n    }\n  }\n\n  try {\n    const inputJson = toJsonPayload(\n      toolInput,\n      policy.maxToolInputBytes,\n      `Tool \"${toolName}\" input`,\n    );\n    const outcome = await invokeHostTool({\n      toolName,\n      inputJson,\n      tools: input.tools,\n      baseExecutionOptions,\n      codeModeOptions: input.options ?? {},\n      maxToolInputBytes: policy.maxToolInputBytes,\n      maxToolOutputBytes: policy.maxToolOutputBytes,\n      toolCallId,\n      ...(codeModeInterrupt === undefined ? {} : { codeModeInterrupt }),\n      skipApproval,\n    });\n    if (outcome.type === 'interrupted') {\n      return context.interrupt(outcome.payload);\n    }\n    return fromJsonPayload(outcome.valueJson);\n  } catch (error) {\n    if (error instanceof CodeModeError) {\n      codeModeErrors.push(error);\n      throw new RunError(error.message, error.code, error.details);\n    }\n    if (\n      RunError.isInstance(error) ||\n      (error instanceof Error && error.name === 'HostFunctionInterruptSignal')\n    ) {\n      throw error;\n    }\n    throw new RunError('Host tool failed.', 'CODE_MODE_HOST_TOOL_ERROR');\n  }\n}\n\nfunction assertInterruptPayload(value: unknown): CodeModeInterruptPayload {\n  if (\n    typeof value !== 'object' ||\n    value === null ||\n    Array.isArray(value) ||\n    typeof (value as { kind?: unknown }).kind !== 'string'\n  ) {\n    throw new CodeModeProtocolError(\n      'Code mode interruption payload is malformed.',\n    );\n  }\n  return value as CodeModeInterruptPayload;\n}\n\nfunction toRunLimits(\n  policy: ResolvedExecutionPolicy,\n  source: string,\n  userSource: string,\n): RunLimits {\n  return {\n    timeoutMs: policy.timeoutMs,\n    memoryLimitBytes: policy.memoryLimitBytes,\n    maxStackSizeBytes: policy.maxStackSizeBytes,\n    maxResultBytes: expandedSerializationLimit(policy.maxResultBytes),\n    maxConsoleOutputBytes: policy.maxConsoleOutputBytes,\n    maxSourceBytes: withSerializationOverhead(\n      policy.maxSourceBytes,\n      Buffer.byteLength(source) - Buffer.byteLength(userSource),\n    ),\n    maxHostFunctionArgumentsBytes: expandedSerializationLimit(\n      policy.maxToolInputBytes,\n    ),\n    maxHostFunctionOutputBytes: expandedSerializationLimit(\n      policy.maxToolOutputBytes,\n    ),\n    maxBridgeRequests: policy.maxBridgeRequests,\n    maxInFlightBridgeRequests: policy.maxInFlightBridgeRequests,\n  };\n}\n\nfunction expandedSerializationLimit(value: number): number {\n  if (!isValidRunLimit(value)) {\n    return value;\n  }\n  return Math.min(MAX_RUN_LIMIT, Math.max(4096, value * 2 + 1024));\n}\n\nfunction withSerializationOverhead(value: number, overhead: number): number {\n  return isValidRunLimit(value)\n    ? Math.min(MAX_RUN_LIMIT, value + overhead)\n    : value;\n}\n\nfunction isValidRunLimit(value: number): boolean {\n  return Number.isInteger(value) && value > 0 && value <= MAX_RUN_LIMIT;\n}\n\nfunction prepareContinuation(\n  input: RunCodeModeInput,\n  toolNames: string[],\n  maxToolOutputBytes: number,\n): {\n  token?: string;\n  resolutions?: RunResolution[];\n  nextInterrupt?: CodeModeInterrupt;\n} {\n  if (input.continuation === undefined) {\n    if (input.interruptResolution !== undefined) {\n      throw new CodeModeProtocolError(\n        'A code-mode interrupt resolution was provided without continuation state.',\n      );\n    }\n    return {};\n  }\n\n  verifyCodeModeContinuation(\n    input.continuation,\n    input.options?.continuationSecurity,\n  );\n  if (input.continuation.js !== input.js) {\n    throw new CodeModeProtocolError(\n      'Code mode continuation source does not match the resumed source.',\n    );\n  }\n  if (!stringArraysEqual(input.continuation.toolNames, toolNames)) {\n    throw new CodeModeProtocolError(\n      'Code mode continuation tool names do not match the resumed tools.',\n    );\n  }\n  if (input.interruptResolution === undefined) {\n    throw new CodeModeProtocolError(\n      'A code-mode continuation requires an interrupt resolution.',\n    );\n  }\n\n  const resolutionIndex = input.continuation.resolutions.length;\n  const pending = input.continuation.pendingInterruptions[resolutionIndex];\n  if (\n    pending === undefined ||\n    pending.interruptId !== input.interruptResolution.interruptId\n  ) {\n    throw new CodeModeProtocolError(\n      'Interrupt resolution does not match the next pending code-mode interruption.',\n      { interruptId: input.interruptResolution.interruptId },\n    );\n  }\n  const resolutionValue = normalizeResolutionForPending(\n    pending,\n    input.interruptResolution.resolution,\n    maxToolOutputBytes,\n  );\n  const resolutions: CodeModePendingResolution[] = [\n    ...input.continuation.resolutions,\n    {\n      runInterruptionId: pending.runInterruptionId,\n      value: resolutionValue,\n    },\n  ];\n\n  if (resolutions.length < input.continuation.pendingInterruptions.length) {\n    const continuation = signCodeModeContinuation(\n      {\n        ...withoutAuth(input.continuation),\n        resolutions,\n      },\n      resolveCodeModeContinuationSecurity(input.options?.continuationSecurity),\n    );\n    return {\n      nextInterrupt: toCodeModeInterrupt(continuation, resolutions.length),\n    };\n  }\n\n  assertNoDeniedApproval(input.continuation.pendingInterruptions, resolutions);\n  return {\n    token: input.continuation.token,\n    resolutions: resolutions.map(resolution => ({\n      interruptionId: resolution.runInterruptionId,\n      value: resolution.value,\n    })),\n  };\n}\n\nfunction normalizeResolutionForPending(\n  pending: CodeModePendingInterruption,\n  resolution: unknown,\n  maxToolOutputBytes: number,\n): unknown {\n  return isCodeModeApprovalInterruptPayload(pending.payload)\n    ? normalizeApprovalResolution(resolution)\n    : fromJsonPayload(\n        toJsonPayload(\n          resolution,\n          maxToolOutputBytes,\n          `Resolution \"${pending.interruptId}\"`,\n        ),\n      );\n}\n\nfunction assertNoDeniedApproval(\n  pendingInterruptions: CodeModePendingInterruption[],\n  resolutions: CodeModePendingResolution[],\n): void {\n  for (const [index, pending] of pendingInterruptions.entries()) {\n    if (!isCodeModeApprovalInterruptPayload(pending.payload)) {\n      continue;\n    }\n    const decision = normalizeApprovalResolution(resolutions[index]?.value);\n    if (!decision.approved) {\n      throw new CodeModeToolApprovalDeniedError(\n        pending.toolName,\n        pending.input,\n        pending.toolCallId,\n        decision.reason,\n      );\n    }\n  }\n}\n\nfunction withoutAuth(\n  continuation: CodeModeContinuation,\n): Omit<CodeModeContinuation, 'auth'> {\n  const { auth: _auth, ...unsigned } = continuation;\n  return unsigned;\n}\n\nfunction toPendingInterruptions({\n  interruptions,\n  outerToolCallId,\n  toolNames,\n}: {\n  interruptions: RunInterruption[];\n  outerToolCallId: string;\n  toolNames: string[];\n}): CodeModePendingInterruption[] {\n  return interruptions.map(interruption => {\n    const requestIndex = interruptionIndex(interruption.id);\n    const toolName = toolNameForInterruption(interruption, toolNames);\n    const toolCallId = `${outerToolCallId}:tool-${requestIndex}`;\n    return {\n      runInterruptionId: interruption.id,\n      interruptId: `${toolCallId}:interrupt`,\n      toolName,\n      toolCallId,\n      input:\n        interruption.hostFunctionName === '__codeMode.missing'\n          ? interruption.arguments[1]\n          : interruption.arguments[0],\n      payload: assertInterruptPayload(interruption.payload),\n    };\n  });\n}\n\nfunction interruptionIndex(interruptionId: string): number {\n  const match = /^interrupt-(\\d+)$/u.exec(interruptionId);\n  if (match === null) {\n    throw new CodeModeProtocolError(\n      `Run returned malformed interruption id \"${interruptionId}\".`,\n    );\n  }\n  return Number(match[1]);\n}\n\nfunction toolNameForInterruption(\n  interruption: RunInterruption,\n  toolNames: string[],\n): string {\n  if (interruption.hostFunctionName === '__codeMode.missing') {\n    return String(interruption.arguments[0]);\n  }\n  const match = /^__codeMode\\.tool(\\d+)$/u.exec(interruption.hostFunctionName);\n  const toolName = match === null ? undefined : toolNames[Number(match[1])];\n  if (toolName === undefined) {\n    throw new CodeModeProtocolError(\n      `Run returned an unknown code-mode host function \"${interruption.hostFunctionName}\".`,\n    );\n  }\n  return toolName;\n}\n\nfunction toCodeModeInterrupt(\n  continuation: CodeModeContinuation,\n  index: number,\n): CodeModeInterrupt {\n  const pending = continuation.pendingInterruptions[index];\n  if (pending === undefined) {\n    throw new CodeModeProtocolError(\n      'Code mode continuation has no pending interruption at the requested index.',\n    );\n  }\n  return {\n    type: 'code-mode-interrupt',\n    interruptId: pending.interruptId,\n    toolName: pending.toolName,\n    toolCallId: pending.toolCallId,\n    outerToolCallId: continuation.outerToolCallId,\n    input: structuredClone(pending.input),\n    payload: structuredClone(pending.payload),\n    continuation,\n  };\n}\n\nfunction stringArraysEqual(left: string[], right: string[]): boolean {\n  return (\n    left.length === right.length &&\n    left.every((value, index) => value === right[index])\n  );\n}\n\nfunction findPreservedCodeModeError(\n  error: unknown,\n  errors: CodeModeError[],\n): CodeModeError | undefined {\n  if (!RunError.isInstance(error)) {\n    return undefined;\n  }\n  return errors.find(candidate => candidate.code === error.code);\n}\n\nfunction toCodeModeRuntimeError(error: unknown): unknown {\n  if (error instanceof TypeError) {\n    error.message = translateLimitPath(error.message);\n  }\n  if (\n    error instanceof Error &&\n    error.name === 'SyntaxError' &&\n    !/syntax|unexpected|expression expected/iu.test(error.message)\n  ) {\n    error.message = `Syntax error: ${error.message}`;\n  }\n  if (!RunError.isInstance(error)) {\n    return error;\n  }\n  const details = error.details as Record<string, unknown> | undefined;\n  switch (error.code) {\n    case 'RUN_ABORTED':\n      return copyStack(new CodeModeAbortedError(), error);\n    case 'RUN_TIMEOUT':\n      return copyStack(\n        new CodeModeTimeoutError(Number(details?.timeoutMs ?? 0)),\n        error,\n      );\n    case 'RUN_CONCURRENCY_LIMIT':\n      return copyStack(\n        new CodeModeConcurrencyError(Number(details?.maxWorkers ?? 0)),\n        error,\n      );\n    case 'RUN_SOURCE_TOO_LARGE':\n      return copyStack(\n        new CodeModeSourceTooLargeError(\n          Number(details?.bytes ?? 0),\n          Number(details?.maxBytes ?? 0),\n        ),\n        error,\n      );\n    case 'RUN_BRIDGE_LIMIT':\n      return copyStack(\n        new CodeModeBridgeLimitError(error.message, error.details),\n        error,\n      );\n    case 'RUN_DETACHED_BRIDGE_REQUEST':\n      return copyStack(\n        new CodeModeDetachedBridgeRequestError(error.message, error.details),\n        error,\n      );\n    case 'RUN_PROTOCOL_ERROR':\n      return copyStack(\n        new CodeModeProtocolError(error.message, error.details),\n        error,\n      );\n    default:\n      return error;\n  }\n}\n\nfunction translateLimitPath(message: string): string {\n  const paths: Record<string, string> = {\n    'limits.timeoutMs': 'executionPolicy.timeoutMs',\n    'limits.memoryLimitBytes': 'executionPolicy.memoryLimitBytes',\n    'limits.maxStackSizeBytes': 'executionPolicy.maxStackSizeBytes',\n    'limits.maxResultBytes': 'executionPolicy.maxResultBytes',\n    'limits.maxConsoleOutputBytes': 'executionPolicy.maxConsoleOutputBytes',\n    'limits.maxSourceBytes': 'executionPolicy.maxSourceBytes',\n    'limits.maxHostFunctionArgumentsBytes': 'executionPolicy.maxToolInputBytes',\n    'limits.maxHostFunctionOutputBytes': 'executionPolicy.maxToolOutputBytes',\n    'limits.maxBridgeRequests': 'executionPolicy.maxBridgeRequests',\n    'limits.maxInFlightBridgeRequests':\n      'executionPolicy.maxInFlightBridgeRequests',\n  };\n  for (const [runPath, codeModePath] of Object.entries(paths)) {\n    if (message.includes(runPath)) {\n      return message.replace(runPath, codeModePath);\n    }\n  }\n  return message;\n}\n\nfunction copyStack<T extends Error>(target: T, source: Error): T {\n  if (source.stack !== undefined) {\n    target.stack = source.stack;\n  }\n  return target;\n}\n\nfunction translateSourceStack<T>(error: T): T {\n  if (!(error instanceof Error) || error.stack === undefined) {\n    return error;\n  }\n  error.stack = error.stack.replaceAll(\n    /run\\.js:(\\d+):(\\d+)/gu,\n    (_match, line: string, column: string) =>\n      `code-mode.js:${Math.max(1, Number(line) - SOURCE_LINE_OFFSET)}:${column}`,\n  );\n  return error;\n}\n"],"mappings":"AAAA,SAAS,cAAc;AACvB,SAAS,kBAAkB;AAC3B;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OAKK;AACP;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,sBAAsB;AAY/B,SAAS,iBAAiB,qBAAqB;AAE/C,MAAM,qBAAqB;AAC3B,MAAM,gBAAgB;AACtB,MAAM,qBAAqB;AAC3B,MAAM,6BAA6B,KAAK,OAAO;AAC/C,MAAM,4BAA4B,IAAI,OAAO;AAC7C,MAAM,2BAA2B,OAAO;AACxC,MAAM,mCAAmC,KAAK;AAC9C,MAAM,2BAA2B,MAAM;AACvC,MAAM,+BAA+B,OAAO;AAC5C,MAAM,gCAAgC,IAAI,OAAO;AACjD,MAAM,8BAA8B;AACpC,MAAM,wCAAwC;AAE9C,IAAI,oBAAoB;AAWxB,eAAsB,YAAY,OAA2C;AAC3E,QAAM,SAAS,uBAAuB,MAAM,SAAS,eAAe;AACpE,mBAAiB,MAAM,IAAI,OAAO,cAAc;AAEhD,QAAM,YAAY,OAAO,KAAK,MAAM,KAAK,EAAE,KAAK;AAChD,QAAM,oBAAoB;AAAA,IACxB;AAAA,IACA;AAAA,IACA,OAAO;AAAA,EACT;AACA,MAAI,kBAAkB,kBAAkB,QAAW;AACjD,WAAO,kBAAkB;AAAA,EAC3B;AAEA,QAAM,kBACJ,MAAM,sBAAsB,cAC5B,MAAM,cAAc,mBACpB,aAAa,EAAE,iBAAiB;AAClC,QAAM,SAAS,qBAAqB,MAAM,IAAI,SAAS;AACvD,QAAM,iBAAkC,CAAC;AACzC,QAAM,gBAAgB,oBAAoB;AAAA,IACxC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AACD,QAAM,uBAAuB;AAAA,IAC3B,MAAM,SAAS;AAAA,EACjB;AACA,QAAM,SAAS,aAAa;AAAA,IAC1B,sBAAsB;AAAA,IACtB,mBAAmB,8BAA8B;AAAA,MAC/C,UAAU,qBAAqB;AAAA,MAC/B,QAAQ,WAAW,QAAQ,EACxB,OAAO,qBAAqB,UAAU,EACtC,OAAO;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,MAAI;AACF,UAAM,SAAS,MAAM,OAAO,IAAI;AAAA,MAC9B;AAAA,MACA;AAAA,MACA,QAAQ,YAAY,QAAQ,QAAQ,MAAM,EAAE;AAAA,MAC5C,qBAAqB,EAAE,iBAAiB,UAAU;AAAA,MAClD,GAAI,MAAM,sBAAsB,gBAAgB,SAC5C,CAAC,IACD,EAAE,aAAa,MAAM,qBAAqB,YAAY;AAAA,MAC1D,GAAI,kBAAkB,UAAU,SAC5B,CAAC,IACD,EAAE,cAAc,kBAAkB,MAAM;AAAA,MAC5C,GAAI,kBAAkB,gBAAgB,SAClC,CAAC,IACD,EAAE,aAAa,kBAAkB,YAAY;AAAA,IACnD,CAAC;AAED,QAAI,OAAO,WAAW,aAAa;AACjC,aAAO;AAAA,QACL,cAAc,OAAO,OAAO,OAAO,gBAAgB,kBAAkB;AAAA,MACvE;AAAA,IACF;AAEA,UAAM,uBAAuB,uBAAuB;AAAA,MAClD,eAAe,OAAO;AAAA,MACtB;AAAA,MACA;AAAA,IACF,CAAC;AACD,UAAM,eAAe;AAAA,MACnB;AAAA,QACE,SAAS;AAAA,QACT,IAAI,MAAM;AAAA,QACV;AAAA,QACA;AAAA,QACA,OAAO,OAAO;AAAA,QACd;AAAA,QACA,aAAa,CAAC;AAAA,MAChB;AAAA,MACA;AAAA,IACF;AACA,WAAO,oBAAoB,cAAc,CAAC;AAAA,EAC5C,SAAS,OAAO;AACd,UAAM,YAAY,2BAA2B,OAAO,cAAc;AAClE,QAAI,cAAc,QAAW;AAC3B,YAAM,qBAAqB,SAAS;AAAA,IACtC;AACA,UAAM,qBAAqB,uBAAuB,KAAK,CAAC;AAAA,EAC1D;AACF;AAeA,SAAS,uBACP,SAAkC,CAAC,GACV;AACzB,SAAO;AAAA,IACL,WAAW,OAAO,aAAa;AAAA,IAC/B,kBAAkB,OAAO,oBAAoB;AAAA,IAC7C,mBAAmB,OAAO,qBAAqB;AAAA,IAC/C,gBAAgB,OAAO,kBAAkB;AAAA,IACzC,uBACE,OAAO,yBAAyB;AAAA,IAClC,gBAAgB,OAAO,kBAAkB;AAAA,IACzC,mBAAmB,OAAO,qBAAqB;AAAA,IAC/C,oBACE,OAAO,sBAAsB;AAAA,IAC/B,mBAAmB,OAAO,qBAAqB;AAAA,IAC/C,2BACE,OAAO,6BAA6B;AAAA,EACxC;AACF;AAEA,SAAS,iBAAiB,QAAgB,UAAwB;AAChE,MAAI,CAAC,gBAAgB,QAAQ,GAAG;AAC9B;AAAA,EACF;AACA,QAAM,QAAQ,OAAO,WAAW,MAAM;AACtC,MAAI,QAAQ,UAAU;AACpB,UAAM,IAAI,4BAA4B,OAAO,QAAQ;AAAA,EACvD;AACF;AAEA,SAAS,qBAAqB,IAAY,WAA6B;AACrE,QAAM,WAAW,OAAO;AAAA,IACtB,UAAU,IAAI,CAAC,UAAU,UAAU,CAAC,UAAU,kBAAkB,KAAK,EAAE,CAAC;AAAA,EAC1E;AACA,QAAM,gBAAgB,OAAO,QAAQ,QAAQ,EAC1C,IAAI,CAAC,CAAC,MAAM,SAAS,MAAM,GAAG,KAAK,UAAU,IAAI,CAAC,IAAI,SAAS,EAAE,EACjE,KAAK,GAAG;AACX,SAAO,6BAA6B,aAAa;AAAA,EAAiQ,EAAE;AAAA;AACtT;AAEA,SAAS,oBAAoB;AAAA,EAC3B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAMkB;AAChB,QAAM,QAAkE,CAAC;AACzE,aAAW,CAAC,OAAO,QAAQ,KAAK,UAAU,QAAQ,GAAG;AACnD,UAAM,OAAO,KAAK,EAAE,IAAI,OAAO,cAC7B,MAAM,mBAAmB;AAAA,MACvB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACL;AACA,QAAM,UAAU,OAAO,UAAmB,cACxC,MAAM,mBAAmB;AAAA,IACvB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,UAAU,OAAO,QAAQ;AAAA,EAC3B,CAAC;AACH,SAAO,EAAE,YAAY,MAAM;AAC7B;AAEA,eAAe,mBAAmB;AAAA,EAChC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAOqB;AACnB,QAAM,UAAU,uBAAuB;AACvC,QAAM,aAAa,GAAG,eAAe,SAAS,QAAQ,YAAY;AAClE,QAAM,mBACJ,MAAM,sBAAsB,WAC5B,MAAM,sBAAsB;AAC9B,QAAM,+BACJ,MAAM,sBAAsB,wBAC5B,MAAM,sBAAsB;AAC9B,QAAM,uBAAqD;AAAA,IACzD,YAAY;AAAA,IACZ,UAAU,MAAM,sBAAsB,YAAY,CAAC;AAAA,IACnD,aAAa,QAAQ;AAAA,IACrB,GAAI,qBAAqB,SAAY,CAAC,IAAI,EAAE,SAAS,iBAAiB;AAAA,IACtE,GAAI,iCAAiC,SACjC,CAAC,IACD,EAAE,sBAAsB,6BAA6B;AAAA,EAC3D;AAEA,MAAI;AACJ,MAAI,eAAe;AACnB,MAAI,QAAQ,WAAW,QAAW;AAChC,UAAM,UAAU,uBAAuB,QAAQ,OAAO,OAAO;AAC7D,QAAI,mCAAmC,OAAO,GAAG;AAC/C,YAAM,WAAW,4BAA4B,QAAQ,OAAO,UAAU;AACtE,UAAI,CAAC,SAAS,UAAU;AACtB,cAAM,IAAI;AAAA,UACR;AAAA,UACA;AAAA,UACA;AAAA,UACA,SAAS;AAAA,QACX;AAAA,MACF;AACA,qBAAe;AAAA,IACjB,OAAO;AACL,0BAAoB;AAAA,QAClB,aAAa,GAAG,UAAU;AAAA,QAC1B;AAAA,QACA,YAAY,QAAQ,OAAO;AAAA,MAC7B;AAAA,IACF;AAAA,EACF;AAEA,MAAI;AACF,UAAM,YAAY;AAAA,MAChB;AAAA,MACA,OAAO;AAAA,MACP,SAAS,QAAQ;AAAA,IACnB;AACA,UAAM,UAAU,MAAM,eAAe;AAAA,MACnC;AAAA,MACA;AAAA,MACA,OAAO,MAAM;AAAA,MACb;AAAA,MACA,iBAAiB,MAAM,WAAW,CAAC;AAAA,MACnC,mBAAmB,OAAO;AAAA,MAC1B,oBAAoB,OAAO;AAAA,MAC3B;AAAA,MACA,GAAI,sBAAsB,SAAY,CAAC,IAAI,EAAE,kBAAkB;AAAA,MAC/D;AAAA,IACF,CAAC;AACD,QAAI,QAAQ,SAAS,eAAe;AAClC,aAAO,QAAQ,UAAU,QAAQ,OAAO;AAAA,IAC1C;AACA,WAAO,gBAAgB,QAAQ,SAAS;AAAA,EAC1C,SAAS,OAAO;AACd,QAAI,iBAAiB,eAAe;AAClC,qBAAe,KAAK,KAAK;AACzB,YAAM,IAAI,SAAS,MAAM,SAAS,MAAM,MAAM,MAAM,OAAO;AAAA,IAC7D;AACA,QACE,SAAS,WAAW,KAAK,KACxB,iBAAiB,SAAS,MAAM,SAAS,+BAC1C;AACA,YAAM;AAAA,IACR;AACA,UAAM,IAAI,SAAS,qBAAqB,2BAA2B;AAAA,EACrE;AACF;AAEA,SAAS,uBAAuB,OAA0C;AACxE,MACE,OAAO,UAAU,YACjB,UAAU,QACV,MAAM,QAAQ,KAAK,KACnB,OAAQ,MAA6B,SAAS,UAC9C;AACA,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,YACP,QACA,QACA,YACW;AACX,SAAO;AAAA,IACL,WAAW,OAAO;AAAA,IAClB,kBAAkB,OAAO;AAAA,IACzB,mBAAmB,OAAO;AAAA,IAC1B,gBAAgB,2BAA2B,OAAO,cAAc;AAAA,IAChE,uBAAuB,OAAO;AAAA,IAC9B,gBAAgB;AAAA,MACd,OAAO;AAAA,MACP,OAAO,WAAW,MAAM,IAAI,OAAO,WAAW,UAAU;AAAA,IAC1D;AAAA,IACA,+BAA+B;AAAA,MAC7B,OAAO;AAAA,IACT;AAAA,IACA,4BAA4B;AAAA,MAC1B,OAAO;AAAA,IACT;AAAA,IACA,mBAAmB,OAAO;AAAA,IAC1B,2BAA2B,OAAO;AAAA,EACpC;AACF;AAEA,SAAS,2BAA2B,OAAuB;AACzD,MAAI,CAAC,gBAAgB,KAAK,GAAG;AAC3B,WAAO;AAAA,EACT;AACA,SAAO,KAAK,IAAI,eAAe,KAAK,IAAI,MAAM,QAAQ,IAAI,IAAI,CAAC;AACjE;AAEA,SAAS,0BAA0B,OAAe,UAA0B;AAC1E,SAAO,gBAAgB,KAAK,IACxB,KAAK,IAAI,eAAe,QAAQ,QAAQ,IACxC;AACN;AAEA,SAAS,gBAAgB,OAAwB;AAC/C,SAAO,OAAO,UAAU,KAAK,KAAK,QAAQ,KAAK,SAAS;AAC1D;AAEA,SAAS,oBACP,OACA,WACA,oBAKA;AACA,MAAI,MAAM,iBAAiB,QAAW;AACpC,QAAI,MAAM,wBAAwB,QAAW;AAC3C,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AACA,WAAO,CAAC;AAAA,EACV;AAEA;AAAA,IACE,MAAM;AAAA,IACN,MAAM,SAAS;AAAA,EACjB;AACA,MAAI,MAAM,aAAa,OAAO,MAAM,IAAI;AACtC,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACA,MAAI,CAAC,kBAAkB,MAAM,aAAa,WAAW,SAAS,GAAG;AAC/D,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACA,MAAI,MAAM,wBAAwB,QAAW;AAC3C,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,QAAM,kBAAkB,MAAM,aAAa,YAAY;AACvD,QAAM,UAAU,MAAM,aAAa,qBAAqB,eAAe;AACvE,MACE,YAAY,UACZ,QAAQ,gBAAgB,MAAM,oBAAoB,aAClD;AACA,UAAM,IAAI;AAAA,MACR;AAAA,MACA,EAAE,aAAa,MAAM,oBAAoB,YAAY;AAAA,IACvD;AAAA,EACF;AACA,QAAM,kBAAkB;AAAA,IACtB;AAAA,IACA,MAAM,oBAAoB;AAAA,IAC1B;AAAA,EACF;AACA,QAAM,cAA2C;AAAA,IAC/C,GAAG,MAAM,aAAa;AAAA,IACtB;AAAA,MACE,mBAAmB,QAAQ;AAAA,MAC3B,OAAO;AAAA,IACT;AAAA,EACF;AAEA,MAAI,YAAY,SAAS,MAAM,aAAa,qBAAqB,QAAQ;AACvE,UAAM,eAAe;AAAA,MACnB;AAAA,QACE,GAAG,YAAY,MAAM,YAAY;AAAA,QACjC;AAAA,MACF;AAAA,MACA,oCAAoC,MAAM,SAAS,oBAAoB;AAAA,IACzE;AACA,WAAO;AAAA,MACL,eAAe,oBAAoB,cAAc,YAAY,MAAM;AAAA,IACrE;AAAA,EACF;AAEA,yBAAuB,MAAM,aAAa,sBAAsB,WAAW;AAC3E,SAAO;AAAA,IACL,OAAO,MAAM,aAAa;AAAA,IAC1B,aAAa,YAAY,IAAI,iBAAe;AAAA,MAC1C,gBAAgB,WAAW;AAAA,MAC3B,OAAO,WAAW;AAAA,IACpB,EAAE;AAAA,EACJ;AACF;AAEA,SAAS,8BACP,SACA,YACA,oBACS;AACT,SAAO,mCAAmC,QAAQ,OAAO,IACrD,4BAA4B,UAAU,IACtC;AAAA,IACE;AAAA,MACE;AAAA,MACA;AAAA,MACA,eAAe,QAAQ,WAAW;AAAA,IACpC;AAAA,EACF;AACN;AAEA,SAAS,uBACP,sBACA,aACM;AACN,aAAW,CAAC,OAAO,OAAO,KAAK,qBAAqB,QAAQ,GAAG;AAC7D,QAAI,CAAC,mCAAmC,QAAQ,OAAO,GAAG;AACxD;AAAA,IACF;AACA,UAAM,WAAW,4BAA4B,YAAY,KAAK,GAAG,KAAK;AACtE,QAAI,CAAC,SAAS,UAAU;AACtB,YAAM,IAAI;AAAA,QACR,QAAQ;AAAA,QACR,QAAQ;AAAA,QACR,QAAQ;AAAA,QACR,SAAS;AAAA,MACX;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,YACP,cACoC;AACpC,QAAM,EAAE,MAAM,OAAO,GAAG,SAAS,IAAI;AACrC,SAAO;AACT;AAEA,SAAS,uBAAuB;AAAA,EAC9B;AAAA,EACA;AAAA,EACA;AACF,GAIkC;AAChC,SAAO,cAAc,IAAI,kBAAgB;AACvC,UAAM,eAAe,kBAAkB,aAAa,EAAE;AACtD,UAAM,WAAW,wBAAwB,cAAc,SAAS;AAChE,UAAM,aAAa,GAAG,eAAe,SAAS,YAAY;AAC1D,WAAO;AAAA,MACL,mBAAmB,aAAa;AAAA,MAChC,aAAa,GAAG,UAAU;AAAA,MAC1B;AAAA,MACA;AAAA,MACA,OACE,aAAa,qBAAqB,uBAC9B,aAAa,UAAU,CAAC,IACxB,aAAa,UAAU,CAAC;AAAA,MAC9B,SAAS,uBAAuB,aAAa,OAAO;AAAA,IACtD;AAAA,EACF,CAAC;AACH;AAEA,SAAS,kBAAkB,gBAAgC;AACzD,QAAM,QAAQ,qBAAqB,KAAK,cAAc;AACtD,MAAI,UAAU,MAAM;AAClB,UAAM,IAAI;AAAA,MACR,2CAA2C,cAAc;AAAA,IAC3D;AAAA,EACF;AACA,SAAO,OAAO,MAAM,CAAC,CAAC;AACxB;AAEA,SAAS,wBACP,cACA,WACQ;AACR,MAAI,aAAa,qBAAqB,sBAAsB;AAC1D,WAAO,OAAO,aAAa,UAAU,CAAC,CAAC;AAAA,EACzC;AACA,QAAM,QAAQ,2BAA2B,KAAK,aAAa,gBAAgB;AAC3E,QAAM,WAAW,UAAU,OAAO,SAAY,UAAU,OAAO,MAAM,CAAC,CAAC,CAAC;AACxE,MAAI,aAAa,QAAW;AAC1B,UAAM,IAAI;AAAA,MACR,oDAAoD,aAAa,gBAAgB;AAAA,IACnF;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,oBACP,cACA,OACmB;AACnB,QAAM,UAAU,aAAa,qBAAqB,KAAK;AACvD,MAAI,YAAY,QAAW;AACzB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACA,SAAO;AAAA,IACL,MAAM;AAAA,IACN,aAAa,QAAQ;AAAA,IACrB,UAAU,QAAQ;AAAA,IAClB,YAAY,QAAQ;AAAA,IACpB,iBAAiB,aAAa;AAAA,IAC9B,OAAO,gBAAgB,QAAQ,KAAK;AAAA,IACpC,SAAS,gBAAgB,QAAQ,OAAO;AAAA,IACxC;AAAA,EACF;AACF;AAEA,SAAS,kBAAkB,MAAgB,OAA0B;AACnE,SACE,KAAK,WAAW,MAAM,UACtB,KAAK,MAAM,CAAC,OAAO,UAAU,UAAU,MAAM,KAAK,CAAC;AAEvD;AAEA,SAAS,2BACP,OACA,QAC2B;AAC3B,MAAI,CAAC,SAAS,WAAW,KAAK,GAAG;AAC/B,WAAO;AAAA,EACT;AACA,SAAO,OAAO,KAAK,eAAa,UAAU,SAAS,MAAM,IAAI;AAC/D;AAEA,SAAS,uBAAuB,OAAyB;AACvD,MAAI,iBAAiB,WAAW;AAC9B,UAAM,UAAU,mBAAmB,MAAM,OAAO;AAAA,EAClD;AACA,MACE,iBAAiB,SACjB,MAAM,SAAS,iBACf,CAAC,0CAA0C,KAAK,MAAM,OAAO,GAC7D;AACA,UAAM,UAAU,iBAAiB,MAAM,OAAO;AAAA,EAChD;AACA,MAAI,CAAC,SAAS,WAAW,KAAK,GAAG;AAC/B,WAAO;AAAA,EACT;AACA,QAAM,UAAU,MAAM;AACtB,UAAQ,MAAM,MAAM;AAAA,IAClB,KAAK;AACH,aAAO,UAAU,IAAI,qBAAqB,GAAG,KAAK;AAAA,IACpD,KAAK;AACH,aAAO;AAAA,QACL,IAAI,qBAAqB,OAAO,SAAS,aAAa,CAAC,CAAC;AAAA,QACxD;AAAA,MACF;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,IAAI,yBAAyB,OAAO,SAAS,cAAc,CAAC,CAAC;AAAA,QAC7D;AAAA,MACF;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,IAAI;AAAA,UACF,OAAO,SAAS,SAAS,CAAC;AAAA,UAC1B,OAAO,SAAS,YAAY,CAAC;AAAA,QAC/B;AAAA,QACA;AAAA,MACF;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,IAAI,yBAAyB,MAAM,SAAS,MAAM,OAAO;AAAA,QACzD;AAAA,MACF;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,IAAI,mCAAmC,MAAM,SAAS,MAAM,OAAO;AAAA,QACnE;AAAA,MACF;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,IAAI,sBAAsB,MAAM,SAAS,MAAM,OAAO;AAAA,QACtD;AAAA,MACF;AAAA,IACF;AACE,aAAO;AAAA,EACX;AACF;AAEA,SAAS,mBAAmB,SAAyB;AACnD,QAAM,QAAgC;AAAA,IACpC,oBAAoB;AAAA,IACpB,2BAA2B;AAAA,IAC3B,4BAA4B;AAAA,IAC5B,yBAAyB;AAAA,IACzB,gCAAgC;AAAA,IAChC,yBAAyB;AAAA,IACzB,wCAAwC;AAAA,IACxC,qCAAqC;AAAA,IACrC,4BAA4B;AAAA,IAC5B,oCACE;AAAA,EACJ;AACA,aAAW,CAAC,SAAS,YAAY,KAAK,OAAO,QAAQ,KAAK,GAAG;AAC3D,QAAI,QAAQ,SAAS,OAAO,GAAG;AAC7B,aAAO,QAAQ,QAAQ,SAAS,YAAY;AAAA,IAC9C;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,UAA2B,QAAW,QAAkB;AAC/D,MAAI,OAAO,UAAU,QAAW;AAC9B,WAAO,QAAQ,OAAO;AAAA,EACxB;AACA,SAAO;AACT;AAEA,SAAS,qBAAwB,OAAa;AAC5C,MAAI,EAAE,iBAAiB,UAAU,MAAM,UAAU,QAAW;AAC1D,WAAO;AAAA,EACT;AACA,QAAM,QAAQ,MAAM,MAAM;AAAA,IACxB;AAAA,IACA,CAAC,QAAQ,MAAc,WACrB,gBAAgB,KAAK,IAAI,GAAG,OAAO,IAAI,IAAI,kBAAkB,CAAC,IAAI,MAAM;AAAA,EAC5E;AACA,SAAO;AACT;","names":[]}
+{"version":3,"file":"run-code-mode.js","names":[],"sources":["../src/run-code-mode.ts"],"sourcesContent":["import { Buffer } from 'node:buffer';\nimport { createHash } from 'node:crypto';\nimport {\n  createRunner,\n  createSignedContinuationCodec,\n  getHostFunctionContext,\n  RunError,\n  type HostFunctions,\n  type RunInterruption,\n  type RunLimits,\n  type RunResolution,\n} from 'run';\nimport {\n  isCodeModeApprovalInterruptPayload,\n  normalizeApprovalResolution,\n} from './approval.js';\nimport {\n  resolveCodeModeContinuationSecurity,\n  signCodeModeContinuation,\n  verifyCodeModeContinuation,\n} from './continuation-capability.js';\nimport {\n  CodeModeAbortedError,\n  CodeModeBridgeLimitError,\n  CodeModeConcurrencyError,\n  CodeModeDetachedBridgeRequestError,\n  CodeModeError,\n  CodeModeProtocolError,\n  CodeModeSourceTooLargeError,\n  CodeModeTimeoutError,\n  CodeModeToolApprovalDeniedError,\n} from './errors.js';\nimport { invokeHostTool } from './tool-invocation.js';\nimport type {\n  CodeModeContinuation,\n  CodeModeExecutionPolicy,\n  CodeModeInterrupt,\n  CodeModeInterruptExecutionContext,\n  CodeModeInterruptPayload,\n  CodeModePendingInterruption,\n  CodeModePendingResolution,\n  CodeModeToolExecutionOptions,\n  RunCodeModeInput,\n} from './types.js';\nimport { fromJsonPayload, toJsonPayload } from './utils/serialization.js';\n\nconst SOURCE_LINE_OFFSET = 1;\nconst MAX_RUN_LIMIT = 2_147_483_647;\nconst DEFAULT_TIMEOUT_MS = 30_000;\nconst DEFAULT_MEMORY_LIMIT_BYTES = 64 * 1024 * 1024;\nconst DEFAULT_STACK_LIMIT_BYTES = 2 * 1024 * 1024;\nconst DEFAULT_MAX_RESULT_BYTES = 1024 * 1024;\nconst DEFAULT_MAX_CONSOLE_OUTPUT_BYTES = 64 * 1024;\nconst DEFAULT_MAX_SOURCE_BYTES = 256 * 1024;\nconst DEFAULT_MAX_TOOL_INPUT_BYTES = 1024 * 1024;\nconst DEFAULT_MAX_TOOL_OUTPUT_BYTES = 4 * 1024 * 1024;\nconst DEFAULT_MAX_BRIDGE_REQUESTS = 256;\nconst DEFAULT_MAX_IN_FLIGHT_BRIDGE_REQUESTS = 32;\n\nlet invocationCounter = 0;\n\n/**\n * Runs one code-mode invocation directly without wrapping it as an AI SDK tool.\n *\n * The source is executed in a fresh QuickJS context with the provided host tools\n * and execution limits.\n *\n * @param input - Source code, host tools, forwarded tool execution options, and code-mode options.\n * @returns The JSON-serializable value returned by the sandboxed program.\n */\nexport async function runCodeMode(input: RunCodeModeInput): Promise<unknown> {\n  const policy = resolveExecutionPolicy(input.options?.executionPolicy);\n  assertSourceSize(input.js, policy.maxSourceBytes);\n\n  const toolNames = Object.keys(input.tools).sort();\n  const continuationState = prepareContinuation(\n    input,\n    toolNames,\n    policy.maxToolOutputBytes,\n  );\n  if (continuationState.nextInterrupt !== undefined) {\n    return continuationState.nextInterrupt;\n  }\n\n  const outerToolCallId =\n    input.toolExecutionOptions?.toolCallId ??\n    input.continuation?.outerToolCallId ??\n    `code-mode-${++invocationCounter}`;\n  const source = createCodeModeSource(input.js, toolNames);\n  const codeModeErrors: CodeModeError[] = [];\n  const hostFunctions = createHostFunctions({\n    codeModeErrors,\n    input,\n    outerToolCallId,\n    policy,\n    toolNames,\n  });\n  const continuationSecurity = resolveCodeModeContinuationSecurity(\n    input.options?.continuationSecurity,\n  );\n  const runner = createRunner({\n    continuationAudience: '@ai-sdk/code-mode/v2',\n    continuationCodec: createSignedContinuationCodec({\n      maxAgeMs: continuationSecurity.maxAgeMs,\n      secret: createHash('sha256')\n        .update(continuationSecurity.signingKey)\n        .digest(),\n    }),\n  });\n\n  try {\n    const result = await runner.run({\n      source,\n      hostFunctions,\n      limits: toRunLimits(policy, source, input.js),\n      continuationContext: { outerToolCallId, toolNames },\n      ...(input.toolExecutionOptions?.abortSignal === undefined\n        ? {}\n        : { abortSignal: input.toolExecutionOptions.abortSignal }),\n      ...(continuationState.token === undefined\n        ? {}\n        : { continuation: continuationState.token }),\n      ...(continuationState.resolutions === undefined\n        ? {}\n        : { resolutions: continuationState.resolutions }),\n    });\n\n    if (result.status === 'completed') {\n      return fromJsonPayload(\n        toJsonPayload(result.value, policy.maxResultBytes, 'Code mode result'),\n      );\n    }\n\n    const pendingInterruptions = toPendingInterruptions({\n      interruptions: result.interruptions,\n      outerToolCallId,\n      toolNames,\n    });\n    const continuation = signCodeModeContinuation(\n      {\n        version: 2,\n        js: input.js,\n        outerToolCallId,\n        toolNames,\n        token: result.continuation,\n        pendingInterruptions,\n        resolutions: [],\n      },\n      continuationSecurity,\n    );\n    return toCodeModeInterrupt(continuation, 0);\n  } catch (error) {\n    const preserved = findPreservedCodeModeError(error, codeModeErrors);\n    if (preserved !== undefined) {\n      throw translateSourceStack(preserved);\n    }\n    throw translateSourceStack(toCodeModeRuntimeError(error));\n  }\n}\n\ninterface ResolvedExecutionPolicy {\n  timeoutMs: number;\n  memoryLimitBytes: number;\n  maxStackSizeBytes: number;\n  maxResultBytes: number;\n  maxConsoleOutputBytes: number;\n  maxSourceBytes: number;\n  maxToolInputBytes: number;\n  maxToolOutputBytes: number;\n  maxBridgeRequests: number;\n  maxInFlightBridgeRequests: number;\n}\n\nfunction resolveExecutionPolicy(\n  policy: CodeModeExecutionPolicy = {},\n): ResolvedExecutionPolicy {\n  return {\n    timeoutMs: policy.timeoutMs ?? DEFAULT_TIMEOUT_MS,\n    memoryLimitBytes: policy.memoryLimitBytes ?? DEFAULT_MEMORY_LIMIT_BYTES,\n    maxStackSizeBytes: policy.maxStackSizeBytes ?? DEFAULT_STACK_LIMIT_BYTES,\n    maxResultBytes: policy.maxResultBytes ?? DEFAULT_MAX_RESULT_BYTES,\n    maxConsoleOutputBytes:\n      policy.maxConsoleOutputBytes ?? DEFAULT_MAX_CONSOLE_OUTPUT_BYTES,\n    maxSourceBytes: policy.maxSourceBytes ?? DEFAULT_MAX_SOURCE_BYTES,\n    maxToolInputBytes: policy.maxToolInputBytes ?? DEFAULT_MAX_TOOL_INPUT_BYTES,\n    maxToolOutputBytes:\n      policy.maxToolOutputBytes ?? DEFAULT_MAX_TOOL_OUTPUT_BYTES,\n    maxBridgeRequests: policy.maxBridgeRequests ?? DEFAULT_MAX_BRIDGE_REQUESTS,\n    maxInFlightBridgeRequests:\n      policy.maxInFlightBridgeRequests ?? DEFAULT_MAX_IN_FLIGHT_BRIDGE_REQUESTS,\n  };\n}\n\nfunction assertSourceSize(source: string, maxBytes: number): void {\n  if (!isValidRunLimit(maxBytes)) {\n    return;\n  }\n  const bytes = Buffer.byteLength(source);\n  if (bytes > maxBytes) {\n    throw new CodeModeSourceTooLargeError(bytes, maxBytes);\n  }\n}\n\nfunction createCodeModeSource(js: string, toolNames: string[]): string {\n  const bindings = Object.fromEntries(\n    toolNames.map((toolName, index) => [toolName, `__codeMode.tool${index}`]),\n  );\n  const bindingSource = Object.entries(bindings)\n    .map(([name, reference]) => `${JSON.stringify(name)}:${reference}`)\n    .join(',');\n  return `const __codeModeBindings={${bindingSource}};const tools=new Proxy(Object.create(null),{get(_target,name){const binding=__codeModeBindings[name];return typeof binding===\"function\"?(input)=>binding(input):(input)=>__codeMode.missing(String(name),input);}});const __codeModeResult=await(async()=>{\\n${js}\\n})();if(__codeModeResult===undefined)return undefined;return JSON.parse(JSON.stringify(__codeModeResult));`;\n}\n\nfunction createHostFunctions({\n  codeModeErrors,\n  input,\n  outerToolCallId,\n  policy,\n  toolNames,\n}: {\n  codeModeErrors: CodeModeError[];\n  input: RunCodeModeInput;\n  outerToolCallId: string;\n  policy: ResolvedExecutionPolicy;\n  toolNames: string[];\n}): HostFunctions {\n  const group: Record<string, (...args: unknown[]) => Promise<unknown>> = {};\n  for (const [index, toolName] of toolNames.entries()) {\n    group[`tool${index}`] = async (toolInput: unknown) =>\n      await invokeCodeModeTool({\n        codeModeErrors,\n        input,\n        outerToolCallId,\n        policy,\n        toolInput,\n        toolName,\n      });\n  }\n  group.missing = async (toolName: unknown, toolInput: unknown) =>\n    await invokeCodeModeTool({\n      codeModeErrors,\n      input,\n      outerToolCallId,\n      policy,\n      toolInput,\n      toolName: String(toolName),\n    });\n  return { __codeMode: group } as HostFunctions;\n}\n\nasync function invokeCodeModeTool({\n  codeModeErrors,\n  input,\n  outerToolCallId,\n  policy,\n  toolInput,\n  toolName,\n}: {\n  codeModeErrors: CodeModeError[];\n  input: RunCodeModeInput;\n  outerToolCallId: string;\n  policy: ResolvedExecutionPolicy;\n  toolInput: unknown;\n  toolName: string;\n}): Promise<unknown> {\n  const context = getHostFunctionContext();\n  const toolCallId = `${outerToolCallId}:tool-${context.requestIndex}`;\n  const forwardedContext =\n    input.toolExecutionOptions?.context ??\n    input.toolExecutionOptions?.experimental_context;\n  const forwardedExperimentalContext =\n    input.toolExecutionOptions?.experimental_context ??\n    input.toolExecutionOptions?.context;\n  const baseExecutionOptions: CodeModeToolExecutionOptions = {\n    toolCallId: outerToolCallId,\n    messages: input.toolExecutionOptions?.messages ?? [],\n    abortSignal: context.abortSignal,\n    ...(forwardedContext === undefined ? {} : { context: forwardedContext }),\n    ...(forwardedExperimentalContext === undefined\n      ? {}\n      : { experimental_context: forwardedExperimentalContext }),\n  };\n\n  let codeModeInterrupt: CodeModeInterruptExecutionContext | undefined;\n  let skipApproval = false;\n  if (context.resume !== undefined) {\n    const payload = assertInterruptPayload(context.resume.payload);\n    if (isCodeModeApprovalInterruptPayload(payload)) {\n      const decision = normalizeApprovalResolution(context.resume.resolution);\n      if (!decision.approved) {\n        throw new CodeModeToolApprovalDeniedError(\n          toolName,\n          toolInput,\n          toolCallId,\n          decision.reason,\n        );\n      }\n      skipApproval = true;\n    } else {\n      codeModeInterrupt = {\n        interruptId: `${toolCallId}:interrupt`,\n        payload,\n        resolution: context.resume.resolution,\n      };\n    }\n  }\n\n  try {\n    const inputJson = toJsonPayload(\n      toolInput,\n      policy.maxToolInputBytes,\n      `Tool \"${toolName}\" input`,\n    );\n    const outcome = await invokeHostTool({\n      toolName,\n      inputJson,\n      tools: input.tools,\n      baseExecutionOptions,\n      codeModeOptions: input.options ?? {},\n      maxToolInputBytes: policy.maxToolInputBytes,\n      maxToolOutputBytes: policy.maxToolOutputBytes,\n      toolCallId,\n      ...(codeModeInterrupt === undefined ? {} : { codeModeInterrupt }),\n      skipApproval,\n    });\n    if (outcome.type === 'interrupted') {\n      return context.interrupt(outcome.payload);\n    }\n    return fromJsonPayload(outcome.valueJson);\n  } catch (error) {\n    if (error instanceof CodeModeError) {\n      codeModeErrors.push(error);\n      throw new RunError(error.message, error.code, error.details);\n    }\n    if (\n      RunError.isInstance(error) ||\n      (error instanceof Error && error.name === 'HostFunctionInterruptSignal')\n    ) {\n      throw error;\n    }\n    throw new RunError('Host tool failed.', 'CODE_MODE_HOST_TOOL_ERROR');\n  }\n}\n\nfunction assertInterruptPayload(value: unknown): CodeModeInterruptPayload {\n  if (\n    typeof value !== 'object' ||\n    value === null ||\n    Array.isArray(value) ||\n    typeof (value as { kind?: unknown }).kind !== 'string'\n  ) {\n    throw new CodeModeProtocolError(\n      'Code mode interruption payload is malformed.',\n    );\n  }\n  return value as CodeModeInterruptPayload;\n}\n\nfunction toRunLimits(\n  policy: ResolvedExecutionPolicy,\n  source: string,\n  userSource: string,\n): RunLimits {\n  return {\n    timeoutMs: policy.timeoutMs,\n    memoryLimitBytes: policy.memoryLimitBytes,\n    maxStackSizeBytes: policy.maxStackSizeBytes,\n    maxResultBytes: expandedSerializationLimit(policy.maxResultBytes),\n    maxConsoleOutputBytes: policy.maxConsoleOutputBytes,\n    maxSourceBytes: withSerializationOverhead(\n      policy.maxSourceBytes,\n      Buffer.byteLength(source) - Buffer.byteLength(userSource),\n    ),\n    maxHostFunctionArgumentsBytes: expandedSerializationLimit(\n      policy.maxToolInputBytes,\n    ),\n    maxHostFunctionOutputBytes: expandedSerializationLimit(\n      policy.maxToolOutputBytes,\n    ),\n    maxBridgeRequests: policy.maxBridgeRequests,\n    maxInFlightBridgeRequests: policy.maxInFlightBridgeRequests,\n  };\n}\n\nfunction expandedSerializationLimit(value: number): number {\n  if (!isValidRunLimit(value)) {\n    return value;\n  }\n  return Math.min(MAX_RUN_LIMIT, Math.max(4096, value * 2 + 1024));\n}\n\nfunction withSerializationOverhead(value: number, overhead: number): number {\n  return isValidRunLimit(value)\n    ? Math.min(MAX_RUN_LIMIT, value + overhead)\n    : value;\n}\n\nfunction isValidRunLimit(value: number): boolean {\n  return Number.isInteger(value) && value > 0 && value <= MAX_RUN_LIMIT;\n}\n\nfunction prepareContinuation(\n  input: RunCodeModeInput,\n  toolNames: string[],\n  maxToolOutputBytes: number,\n): {\n  token?: string;\n  resolutions?: RunResolution[];\n  nextInterrupt?: CodeModeInterrupt;\n} {\n  if (input.continuation === undefined) {\n    if (input.interruptResolution !== undefined) {\n      throw new CodeModeProtocolError(\n        'A code-mode interrupt resolution was provided without continuation state.',\n      );\n    }\n    return {};\n  }\n\n  verifyCodeModeContinuation(\n    input.continuation,\n    input.options?.continuationSecurity,\n  );\n  if (input.continuation.js !== input.js) {\n    throw new CodeModeProtocolError(\n      'Code mode continuation source does not match the resumed source.',\n    );\n  }\n  if (!stringArraysEqual(input.continuation.toolNames, toolNames)) {\n    throw new CodeModeProtocolError(\n      'Code mode continuation tool names do not match the resumed tools.',\n    );\n  }\n  if (input.interruptResolution === undefined) {\n    throw new CodeModeProtocolError(\n      'A code-mode continuation requires an interrupt resolution.',\n    );\n  }\n\n  const resolutionIndex = input.continuation.resolutions.length;\n  const pending = input.continuation.pendingInterruptions[resolutionIndex];\n  if (\n    pending === undefined ||\n    pending.interruptId !== input.interruptResolution.interruptId\n  ) {\n    throw new CodeModeProtocolError(\n      'Interrupt resolution does not match the next pending code-mode interruption.',\n      { interruptId: input.interruptResolution.interruptId },\n    );\n  }\n  const resolutionValue = normalizeResolutionForPending(\n    pending,\n    input.interruptResolution.resolution,\n    maxToolOutputBytes,\n  );\n  const resolutions: CodeModePendingResolution[] = [\n    ...input.continuation.resolutions,\n    {\n      runInterruptionId: pending.runInterruptionId,\n      value: resolutionValue,\n    },\n  ];\n\n  if (resolutions.length < input.continuation.pendingInterruptions.length) {\n    const continuation = signCodeModeContinuation(\n      {\n        ...withoutAuth(input.continuation),\n        resolutions,\n      },\n      resolveCodeModeContinuationSecurity(input.options?.continuationSecurity),\n    );\n    return {\n      nextInterrupt: toCodeModeInterrupt(continuation, resolutions.length),\n    };\n  }\n\n  assertNoDeniedApproval(input.continuation.pendingInterruptions, resolutions);\n  return {\n    token: input.continuation.token,\n    resolutions: resolutions.map(resolution => ({\n      interruptionId: resolution.runInterruptionId,\n      value: resolution.value,\n    })),\n  };\n}\n\nfunction normalizeResolutionForPending(\n  pending: CodeModePendingInterruption,\n  resolution: unknown,\n  maxToolOutputBytes: number,\n): unknown {\n  return isCodeModeApprovalInterruptPayload(pending.payload)\n    ? normalizeApprovalResolution(resolution)\n    : fromJsonPayload(\n        toJsonPayload(\n          resolution,\n          maxToolOutputBytes,\n          `Resolution \"${pending.interruptId}\"`,\n        ),\n      );\n}\n\nfunction assertNoDeniedApproval(\n  pendingInterruptions: CodeModePendingInterruption[],\n  resolutions: CodeModePendingResolution[],\n): void {\n  for (const [index, pending] of pendingInterruptions.entries()) {\n    if (!isCodeModeApprovalInterruptPayload(pending.payload)) {\n      continue;\n    }\n    const decision = normalizeApprovalResolution(resolutions[index]?.value);\n    if (!decision.approved) {\n      throw new CodeModeToolApprovalDeniedError(\n        pending.toolName,\n        pending.input,\n        pending.toolCallId,\n        decision.reason,\n      );\n    }\n  }\n}\n\nfunction withoutAuth(\n  continuation: CodeModeContinuation,\n): Omit<CodeModeContinuation, 'auth'> {\n  const { auth: _auth, ...unsigned } = continuation;\n  return unsigned;\n}\n\nfunction toPendingInterruptions({\n  interruptions,\n  outerToolCallId,\n  toolNames,\n}: {\n  interruptions: RunInterruption[];\n  outerToolCallId: string;\n  toolNames: string[];\n}): CodeModePendingInterruption[] {\n  return interruptions.map(interruption => {\n    const requestIndex = interruptionIndex(interruption.id);\n    const toolName = toolNameForInterruption(interruption, toolNames);\n    const toolCallId = `${outerToolCallId}:tool-${requestIndex}`;\n    return {\n      runInterruptionId: interruption.id,\n      interruptId: `${toolCallId}:interrupt`,\n      toolName,\n      toolCallId,\n      input:\n        interruption.hostFunctionName === '__codeMode.missing'\n          ? interruption.arguments[1]\n          : interruption.arguments[0],\n      payload: assertInterruptPayload(interruption.payload),\n    };\n  });\n}\n\nfunction interruptionIndex(interruptionId: string): number {\n  const match = /^interrupt-(\\d+)$/u.exec(interruptionId);\n  if (match === null) {\n    throw new CodeModeProtocolError(\n      `Run returned malformed interruption id \"${interruptionId}\".`,\n    );\n  }\n  return Number(match[1]);\n}\n\nfunction toolNameForInterruption(\n  interruption: RunInterruption,\n  toolNames: string[],\n): string {\n  if (interruption.hostFunctionName === '__codeMode.missing') {\n    return String(interruption.arguments[0]);\n  }\n  const match = /^__codeMode\\.tool(\\d+)$/u.exec(interruption.hostFunctionName);\n  const toolName = match === null ? undefined : toolNames[Number(match[1])];\n  if (toolName === undefined) {\n    throw new CodeModeProtocolError(\n      `Run returned an unknown code-mode host function \"${interruption.hostFunctionName}\".`,\n    );\n  }\n  return toolName;\n}\n\nfunction toCodeModeInterrupt(\n  continuation: CodeModeContinuation,\n  index: number,\n): CodeModeInterrupt {\n  const pending = continuation.pendingInterruptions[index];\n  if (pending === undefined) {\n    throw new CodeModeProtocolError(\n      'Code mode continuation has no pending interruption at the requested index.',\n    );\n  }\n  return {\n    type: 'code-mode-interrupt',\n    interruptId: pending.interruptId,\n    toolName: pending.toolName,\n    toolCallId: pending.toolCallId,\n    outerToolCallId: continuation.outerToolCallId,\n    input: structuredClone(pending.input),\n    payload: structuredClone(pending.payload),\n    continuation,\n  };\n}\n\nfunction stringArraysEqual(left: string[], right: string[]): boolean {\n  return (\n    left.length === right.length &&\n    left.every((value, index) => value === right[index])\n  );\n}\n\nfunction findPreservedCodeModeError(\n  error: unknown,\n  errors: CodeModeError[],\n): CodeModeError | undefined {\n  if (!RunError.isInstance(error)) {\n    return undefined;\n  }\n  return errors.find(candidate => candidate.code === error.code);\n}\n\nfunction toCodeModeRuntimeError(error: unknown): unknown {\n  if (error instanceof TypeError) {\n    error.message = translateLimitPath(error.message);\n  }\n  if (\n    error instanceof Error &&\n    error.name === 'SyntaxError' &&\n    !/syntax|unexpected|expression expected/iu.test(error.message)\n  ) {\n    error.message = `Syntax error: ${error.message}`;\n  }\n  if (!RunError.isInstance(error)) {\n    return error;\n  }\n  const details = error.details as Record<string, unknown> | undefined;\n  switch (error.code) {\n    case 'RUN_ABORTED':\n      return copyStack(new CodeModeAbortedError(), error);\n    case 'RUN_TIMEOUT':\n      return copyStack(\n        new CodeModeTimeoutError(Number(details?.timeoutMs ?? 0)),\n        error,\n      );\n    case 'RUN_CONCURRENCY_LIMIT':\n      return copyStack(\n        new CodeModeConcurrencyError(Number(details?.maxWorkers ?? 0)),\n        error,\n      );\n    case 'RUN_SOURCE_TOO_LARGE':\n      return copyStack(\n        new CodeModeSourceTooLargeError(\n          Number(details?.bytes ?? 0),\n          Number(details?.maxBytes ?? 0),\n        ),\n        error,\n      );\n    case 'RUN_BRIDGE_LIMIT':\n      return copyStack(\n        new CodeModeBridgeLimitError(error.message, error.details),\n        error,\n      );\n    case 'RUN_DETACHED_BRIDGE_REQUEST':\n      return copyStack(\n        new CodeModeDetachedBridgeRequestError(error.message, error.details),\n        error,\n      );\n    case 'RUN_PROTOCOL_ERROR':\n      return copyStack(\n        new CodeModeProtocolError(error.message, error.details),\n        error,\n      );\n    default:\n      return error;\n  }\n}\n\nfunction translateLimitPath(message: string): string {\n  const paths: Record<string, string> = {\n    'limits.timeoutMs': 'executionPolicy.timeoutMs',\n    'limits.memoryLimitBytes': 'executionPolicy.memoryLimitBytes',\n    'limits.maxStackSizeBytes': 'executionPolicy.maxStackSizeBytes',\n    'limits.maxResultBytes': 'executionPolicy.maxResultBytes',\n    'limits.maxConsoleOutputBytes': 'executionPolicy.maxConsoleOutputBytes',\n    'limits.maxSourceBytes': 'executionPolicy.maxSourceBytes',\n    'limits.maxHostFunctionArgumentsBytes': 'executionPolicy.maxToolInputBytes',\n    'limits.maxHostFunctionOutputBytes': 'executionPolicy.maxToolOutputBytes',\n    'limits.maxBridgeRequests': 'executionPolicy.maxBridgeRequests',\n    'limits.maxInFlightBridgeRequests':\n      'executionPolicy.maxInFlightBridgeRequests',\n  };\n  for (const [runPath, codeModePath] of Object.entries(paths)) {\n    if (message.includes(runPath)) {\n      return message.replace(runPath, codeModePath);\n    }\n  }\n  return message;\n}\n\nfunction copyStack<T extends Error>(target: T, source: Error): T {\n  if (source.stack !== undefined) {\n    target.stack = source.stack;\n  }\n  return target;\n}\n\nfunction translateSourceStack<T>(error: T): T {\n  if (!(error instanceof Error) || error.stack === undefined) {\n    return error;\n  }\n  error.stack = error.stack.replaceAll(\n    /run\\.js:(\\d+):(\\d+)/gu,\n    (_match, line: string, column: string) =>\n      `code-mode.js:${Math.max(1, Number(line) - SOURCE_LINE_OFFSET)}:${column}`,\n  );\n  return error;\n}\n"],"mappings":";;;;;;;;;AA8CA,MAAM,qBAAqB;AAC3B,MAAM,gBAAgB;AACtB,MAAM,qBAAqB;AAC3B,MAAM,6BAA6B;AACnC,MAAM,4BAA4B;AAClC,MAAM,2BAA2B;AACjC,MAAM,mCAAmC;AACzC,MAAM,2BAA2B;AACjC,MAAM,+BAA+B;AACrC,MAAM,gCAAgC;AACtC,MAAM,8BAA8B;AACpC,MAAM,wCAAwC;AAE9C,IAAI,oBAAoB;;;;;;;;;;AAWxB,eAAsB,YAAY,OAA2C;CAC3E,MAAM,SAAS,uBAAuB,MAAM,SAAS,eAAe;CACpE,iBAAiB,MAAM,IAAI,OAAO,cAAc;CAEhD,MAAM,YAAY,OAAO,KAAK,MAAM,KAAK,CAAC,CAAC,KAAK;CAChD,MAAM,oBAAoB,oBACxB,OACA,WACA,OAAO,kBACT;CACA,IAAI,kBAAkB,kBAAkB,KAAA,GACtC,OAAO,kBAAkB;CAG3B,MAAM,kBACJ,MAAM,sBAAsB,cAC5B,MAAM,cAAc,mBACpB,aAAa,EAAE;CACjB,MAAM,SAAS,qBAAqB,MAAM,IAAI,SAAS;CACvD,MAAM,iBAAkC,CAAC;CACzC,MAAM,gBAAgB,oBAAoB;EACxC;EACA;EACA;EACA;EACA;CACF,CAAC;CACD,MAAM,uBAAuB,oCAC3B,MAAM,SAAS,oBACjB;CACA,MAAM,SAAS,aAAa;EAC1B,sBAAsB;EACtB,mBAAmB,8BAA8B;GAC/C,UAAU,qBAAqB;GAC/B,QAAQ,WAAW,QAAQ,CAAC,CACzB,OAAO,qBAAqB,UAAU,CAAC,CACvC,OAAO;EACZ,CAAC;CACH,CAAC;CAED,IAAI;EACF,MAAM,SAAS,MAAM,OAAO,IAAI;GAC9B;GACA;GACA,QAAQ,YAAY,QAAQ,QAAQ,MAAM,EAAE;GAC5C,qBAAqB;IAAE;IAAiB;GAAU;GAClD,GAAI,MAAM,sBAAsB,gBAAgB,KAAA,IAC5C,CAAC,IACD,EAAE,aAAa,MAAM,qBAAqB,YAAY;GAC1D,GAAI,kBAAkB,UAAU,KAAA,IAC5B,CAAC,IACD,EAAE,cAAc,kBAAkB,MAAM;GAC5C,GAAI,kBAAkB,gBAAgB,KAAA,IAClC,CAAC,IACD,EAAE,aAAa,kBAAkB,YAAY;EACnD,CAAC;EAED,IAAI,OAAO,WAAW,aACpB,OAAO,gBACL,cAAc,OAAO,OAAO,OAAO,gBAAgB,kBAAkB,CACvE;EAGF,MAAM,uBAAuB,uBAAuB;GAClD,eAAe,OAAO;GACtB;GACA;EACF,CAAC;EAaD,OAAO,oBAZc,yBACnB;GACE,SAAS;GACT,IAAI,MAAM;GACV;GACA;GACA,OAAO,OAAO;GACd;GACA,aAAa,CAAC;EAChB,GACA,oBAEoC,GAAG,CAAC;CAC5C,SAAS,OAAO;EACd,MAAM,YAAY,2BAA2B,OAAO,cAAc;EAClE,IAAI,cAAc,KAAA,GAChB,MAAM,qBAAqB,SAAS;EAEtC,MAAM,qBAAqB,uBAAuB,KAAK,CAAC;CAC1D;AACF;AAeA,SAAS,uBACP,SAAkC,CAAC,GACV;CACzB,OAAO;EACL,WAAW,OAAO,aAAa;EAC/B,kBAAkB,OAAO,oBAAoB;EAC7C,mBAAmB,OAAO,qBAAqB;EAC/C,gBAAgB,OAAO,kBAAkB;EACzC,uBACE,OAAO,yBAAyB;EAClC,gBAAgB,OAAO,kBAAkB;EACzC,mBAAmB,OAAO,qBAAqB;EAC/C,oBACE,OAAO,sBAAsB;EAC/B,mBAAmB,OAAO,qBAAqB;EAC/C,2BACE,OAAO,6BAA6B;CACxC;AACF;AAEA,SAAS,iBAAiB,QAAgB,UAAwB;CAChE,IAAI,CAAC,gBAAgB,QAAQ,GAC3B;CAEF,MAAM,QAAQ,OAAO,WAAW,MAAM;CACtC,IAAI,QAAQ,UACV,MAAM,IAAI,4BAA4B,OAAO,QAAQ;AAEzD;AAEA,SAAS,qBAAqB,IAAY,WAA6B;CACrE,MAAM,WAAW,OAAO,YACtB,UAAU,KAAK,UAAU,UAAU,CAAC,UAAU,kBAAkB,OAAO,CAAC,CAC1E;CAIA,OAAO,6BAHe,OAAO,QAAQ,QAAQ,CAAC,CAC3C,KAAK,CAAC,MAAM,eAAe,GAAG,KAAK,UAAU,IAAI,EAAE,GAAG,WAAW,CAAC,CAClE,KAAK,GACwC,EAAE,gQAAgQ,GAAG;AACvT;AAEA,SAAS,oBAAoB,EAC3B,gBACA,OACA,iBACA,QACA,aAOgB;CAChB,MAAM,QAAkE,CAAC;CACzE,KAAK,MAAM,CAAC,OAAO,aAAa,UAAU,QAAQ,GAChD,MAAM,OAAO,WAAW,OAAO,cAC7B,MAAM,mBAAmB;EACvB;EACA;EACA;EACA;EACA;EACA;CACF,CAAC;CAEL,MAAM,UAAU,OAAO,UAAmB,cACxC,MAAM,mBAAmB;EACvB;EACA;EACA;EACA;EACA;EACA,UAAU,OAAO,QAAQ;CAC3B,CAAC;CACH,OAAO,EAAE,YAAY,MAAM;AAC7B;AAEA,eAAe,mBAAmB,EAChC,gBACA,OACA,iBACA,QACA,WACA,YAQmB;CACnB,MAAM,UAAU,uBAAuB;CACvC,MAAM,aAAa,GAAG,gBAAgB,QAAQ,QAAQ;CACtD,MAAM,mBACJ,MAAM,sBAAsB,WAC5B,MAAM,sBAAsB;CAC9B,MAAM,+BACJ,MAAM,sBAAsB,wBAC5B,MAAM,sBAAsB;CAC9B,MAAM,uBAAqD;EACzD,YAAY;EACZ,UAAU,MAAM,sBAAsB,YAAY,CAAC;EACnD,aAAa,QAAQ;EACrB,GAAI,qBAAqB,KAAA,IAAY,CAAC,IAAI,EAAE,SAAS,iBAAiB;EACtE,GAAI,iCAAiC,KAAA,IACjC,CAAC,IACD,EAAE,sBAAsB,6BAA6B;CAC3D;CAEA,IAAI;CACJ,IAAI,eAAe;CACnB,IAAI,QAAQ,WAAW,KAAA,GAAW;EAChC,MAAM,UAAU,uBAAuB,QAAQ,OAAO,OAAO;EAC7D,IAAI,mCAAmC,OAAO,GAAG;GAC/C,MAAM,WAAW,4BAA4B,QAAQ,OAAO,UAAU;GACtE,IAAI,CAAC,SAAS,UACZ,MAAM,IAAI,gCACR,UACA,WACA,YACA,SAAS,MACX;GAEF,eAAe;EACjB,OACE,oBAAoB;GAClB,aAAa,GAAG,WAAW;GAC3B;GACA,YAAY,QAAQ,OAAO;EAC7B;CAEJ;CAEA,IAAI;EACF,MAAM,YAAY,cAChB,WACA,OAAO,mBACP,SAAS,SAAS,QACpB;EACA,MAAM,UAAU,MAAM,eAAe;GACnC;GACA;GACA,OAAO,MAAM;GACb;GACA,iBAAiB,MAAM,WAAW,CAAC;GACnC,mBAAmB,OAAO;GAC1B,oBAAoB,OAAO;GAC3B;GACA,GAAI,sBAAsB,KAAA,IAAY,CAAC,IAAI,EAAE,kBAAkB;GAC/D;EACF,CAAC;EACD,IAAI,QAAQ,SAAS,eACnB,OAAO,QAAQ,UAAU,QAAQ,OAAO;EAE1C,OAAO,gBAAgB,QAAQ,SAAS;CAC1C,SAAS,OAAO;EACd,IAAI,iBAAiB,eAAe;GAClC,eAAe,KAAK,KAAK;GACzB,MAAM,IAAI,SAAS,MAAM,SAAS,MAAM,MAAM,MAAM,OAAO;EAC7D;EACA,IACE,SAAS,WAAW,KAAK,KACxB,iBAAiB,SAAS,MAAM,SAAS,+BAE1C,MAAM;EAER,MAAM,IAAI,SAAS,qBAAqB,2BAA2B;CACrE;AACF;AAEA,SAAS,uBAAuB,OAA0C;CACxE,IACE,OAAO,UAAU,YACjB,UAAU,QACV,MAAM,QAAQ,KAAK,KACnB,OAAQ,MAA6B,SAAS,UAE9C,MAAM,IAAI,sBACR,8CACF;CAEF,OAAO;AACT;AAEA,SAAS,YACP,QACA,QACA,YACW;CACX,OAAO;EACL,WAAW,OAAO;EAClB,kBAAkB,OAAO;EACzB,mBAAmB,OAAO;EAC1B,gBAAgB,2BAA2B,OAAO,cAAc;EAChE,uBAAuB,OAAO;EAC9B,gBAAgB,0BACd,OAAO,gBACP,OAAO,WAAW,MAAM,IAAI,OAAO,WAAW,UAAU,CAC1D;EACA,+BAA+B,2BAC7B,OAAO,iBACT;EACA,4BAA4B,2BAC1B,OAAO,kBACT;EACA,mBAAmB,OAAO;EAC1B,2BAA2B,OAAO;CACpC;AACF;AAEA,SAAS,2BAA2B,OAAuB;CACzD,IAAI,CAAC,gBAAgB,KAAK,GACxB,OAAO;CAET,OAAO,KAAK,IAAI,eAAe,KAAK,IAAI,MAAM,QAAQ,IAAI,IAAI,CAAC;AACjE;AAEA,SAAS,0BAA0B,OAAe,UAA0B;CAC1E,OAAO,gBAAgB,KAAK,IACxB,KAAK,IAAI,eAAe,QAAQ,QAAQ,IACxC;AACN;AAEA,SAAS,gBAAgB,OAAwB;CAC/C,OAAO,OAAO,UAAU,KAAK,KAAK,QAAQ,KAAK,SAAS;AAC1D;AAEA,SAAS,oBACP,OACA,WACA,oBAKA;CACA,IAAI,MAAM,iBAAiB,KAAA,GAAW;EACpC,IAAI,MAAM,wBAAwB,KAAA,GAChC,MAAM,IAAI,sBACR,2EACF;EAEF,OAAO,CAAC;CACV;CAEA,2BACE,MAAM,cACN,MAAM,SAAS,oBACjB;CACA,IAAI,MAAM,aAAa,OAAO,MAAM,IAClC,MAAM,IAAI,sBACR,kEACF;CAEF,IAAI,CAAC,kBAAkB,MAAM,aAAa,WAAW,SAAS,GAC5D,MAAM,IAAI,sBACR,mEACF;CAEF,IAAI,MAAM,wBAAwB,KAAA,GAChC,MAAM,IAAI,sBACR,4DACF;CAGF,MAAM,kBAAkB,MAAM,aAAa,YAAY;CACvD,MAAM,UAAU,MAAM,aAAa,qBAAqB;CACxD,IACE,YAAY,KAAA,KACZ,QAAQ,gBAAgB,MAAM,oBAAoB,aAElD,MAAM,IAAI,sBACR,gFACA,EAAE,aAAa,MAAM,oBAAoB,YAAY,CACvD;CAEF,MAAM,kBAAkB,8BACtB,SACA,MAAM,oBAAoB,YAC1B,kBACF;CACA,MAAM,cAA2C,CAC/C,GAAG,MAAM,aAAa,aACtB;EACE,mBAAmB,QAAQ;EAC3B,OAAO;CACT,CACF;CAEA,IAAI,YAAY,SAAS,MAAM,aAAa,qBAAqB,QAQ/D,OAAO,EACL,eAAe,oBARI,yBACnB;EACE,GAAG,YAAY,MAAM,YAAY;EACjC;CACF,GACA,oCAAoC,MAAM,SAAS,oBAAoB,CAGzB,GAAG,YAAY,MAAM,EACrE;CAGF,uBAAuB,MAAM,aAAa,sBAAsB,WAAW;CAC3E,OAAO;EACL,OAAO,MAAM,aAAa;EAC1B,aAAa,YAAY,KAAI,gBAAe;GAC1C,gBAAgB,WAAW;GAC3B,OAAO,WAAW;EACpB,EAAE;CACJ;AACF;AAEA,SAAS,8BACP,SACA,YACA,oBACS;CACT,OAAO,mCAAmC,QAAQ,OAAO,IACrD,4BAA4B,UAAU,IACtC,gBACE,cACE,YACA,oBACA,eAAe,QAAQ,YAAY,EACrC,CACF;AACN;AAEA,SAAS,uBACP,sBACA,aACM;CACN,KAAK,MAAM,CAAC,OAAO,YAAY,qBAAqB,QAAQ,GAAG;EAC7D,IAAI,CAAC,mCAAmC,QAAQ,OAAO,GACrD;EAEF,MAAM,WAAW,4BAA4B,YAAY,MAAM,EAAE,KAAK;EACtE,IAAI,CAAC,SAAS,UACZ,MAAM,IAAI,gCACR,QAAQ,UACR,QAAQ,OACR,QAAQ,YACR,SAAS,MACX;CAEJ;AACF;AAEA,SAAS,YACP,cACoC;CACpC,MAAM,EAAE,MAAM,OAAO,GAAG,aAAa;CACrC,OAAO;AACT;AAEA,SAAS,uBAAuB,EAC9B,eACA,iBACA,aAKgC;CAChC,OAAO,cAAc,KAAI,iBAAgB;EACvC,MAAM,eAAe,kBAAkB,aAAa,EAAE;EACtD,MAAM,WAAW,wBAAwB,cAAc,SAAS;EAChE,MAAM,aAAa,GAAG,gBAAgB,QAAQ;EAC9C,OAAO;GACL,mBAAmB,aAAa;GAChC,aAAa,GAAG,WAAW;GAC3B;GACA;GACA,OACE,aAAa,qBAAqB,uBAC9B,aAAa,UAAU,KACvB,aAAa,UAAU;GAC7B,SAAS,uBAAuB,aAAa,OAAO;EACtD;CACF,CAAC;AACH;AAEA,SAAS,kBAAkB,gBAAgC;CACzD,MAAM,QAAQ,qBAAqB,KAAK,cAAc;CACtD,IAAI,UAAU,MACZ,MAAM,IAAI,sBACR,2CAA2C,eAAe,GAC5D;CAEF,OAAO,OAAO,MAAM,EAAE;AACxB;AAEA,SAAS,wBACP,cACA,WACQ;CACR,IAAI,aAAa,qBAAqB,sBACpC,OAAO,OAAO,aAAa,UAAU,EAAE;CAEzC,MAAM,QAAQ,2BAA2B,KAAK,aAAa,gBAAgB;CAC3E,MAAM,WAAW,UAAU,OAAO,KAAA,IAAY,UAAU,OAAO,MAAM,EAAE;CACvE,IAAI,aAAa,KAAA,GACf,MAAM,IAAI,sBACR,oDAAoD,aAAa,iBAAiB,GACpF;CAEF,OAAO;AACT;AAEA,SAAS,oBACP,cACA,OACmB;CACnB,MAAM,UAAU,aAAa,qBAAqB;CAClD,IAAI,YAAY,KAAA,GACd,MAAM,IAAI,sBACR,4EACF;CAEF,OAAO;EACL,MAAM;EACN,aAAa,QAAQ;EACrB,UAAU,QAAQ;EAClB,YAAY,QAAQ;EACpB,iBAAiB,aAAa;EAC9B,OAAO,gBAAgB,QAAQ,KAAK;EACpC,SAAS,gBAAgB,QAAQ,OAAO;EACxC;CACF;AACF;AAEA,SAAS,kBAAkB,MAAgB,OAA0B;CACnE,OACE,KAAK,WAAW,MAAM,UACtB,KAAK,OAAO,OAAO,UAAU,UAAU,MAAM,MAAM;AAEvD;AAEA,SAAS,2BACP,OACA,QAC2B;CAC3B,IAAI,CAAC,SAAS,WAAW,KAAK,GAC5B;CAEF,OAAO,OAAO,MAAK,cAAa,UAAU,SAAS,MAAM,IAAI;AAC/D;AAEA,SAAS,uBAAuB,OAAyB;CACvD,IAAI,iBAAiB,WACnB,MAAM,UAAU,mBAAmB,MAAM,OAAO;CAElD,IACE,iBAAiB,SACjB,MAAM,SAAS,iBACf,CAAC,0CAA0C,KAAK,MAAM,OAAO,GAE7D,MAAM,UAAU,iBAAiB,MAAM;CAEzC,IAAI,CAAC,SAAS,WAAW,KAAK,GAC5B,OAAO;CAET,MAAM,UAAU,MAAM;CACtB,QAAQ,MAAM,MAAd;EACE,KAAK,eACH,OAAO,UAAU,IAAI,qBAAqB,GAAG,KAAK;EACpD,KAAK,eACH,OAAO,UACL,IAAI,qBAAqB,OAAO,SAAS,aAAa,CAAC,CAAC,GACxD,KACF;EACF,KAAK,yBACH,OAAO,UACL,IAAI,yBAAyB,OAAO,SAAS,cAAc,CAAC,CAAC,GAC7D,KACF;EACF,KAAK,wBACH,OAAO,UACL,IAAI,4BACF,OAAO,SAAS,SAAS,CAAC,GAC1B,OAAO,SAAS,YAAY,CAAC,CAC/B,GACA,KACF;EACF,KAAK,oBACH,OAAO,UACL,IAAI,yBAAyB,MAAM,SAAS,MAAM,OAAO,GACzD,KACF;EACF,KAAK,+BACH,OAAO,UACL,IAAI,mCAAmC,MAAM,SAAS,MAAM,OAAO,GACnE,KACF;EACF,KAAK,sBACH,OAAO,UACL,IAAI,sBAAsB,MAAM,SAAS,MAAM,OAAO,GACtD,KACF;EACF,SACE,OAAO;CACX;AACF;AAEA,SAAS,mBAAmB,SAAyB;CAcnD,KAAK,MAAM,CAAC,SAAS,iBAAiB,OAAO,QAAQ;EAZnD,oBAAoB;EACpB,2BAA2B;EAC3B,4BAA4B;EAC5B,yBAAyB;EACzB,gCAAgC;EAChC,yBAAyB;EACzB,wCAAwC;EACxC,qCAAqC;EACrC,4BAA4B;EAC5B,oCACE;CAEqD,CAAC,GACxD,IAAI,QAAQ,SAAS,OAAO,GAC1B,OAAO,QAAQ,QAAQ,SAAS,YAAY;CAGhD,OAAO;AACT;AAEA,SAAS,UAA2B,QAAW,QAAkB;CAC/D,IAAI,OAAO,UAAU,KAAA,GACnB,OAAO,QAAQ,OAAO;CAExB,OAAO;AACT;AAEA,SAAS,qBAAwB,OAAa;CAC5C,IAAI,EAAE,iBAAiB,UAAU,MAAM,UAAU,KAAA,GAC/C,OAAO;CAET,MAAM,QAAQ,MAAM,MAAM,WACxB,0BACC,QAAQ,MAAc,WACrB,gBAAgB,KAAK,IAAI,GAAG,OAAO,IAAI,IAAI,kBAAkB,EAAE,GAAG,QACtE;CACA,OAAO;AACT"}

package/dist/tool-invocation.js

@@ -1,192 +1,114 @@
-import { asSchema } from "ai";
+import { CodeModeProtocolError, CodeModeToolApprovalDeniedError, CodeModeToolApprovalRequiredError, CodeModeToolError } from "./errors.js";
 import { CODE_MODE_TOOL_APPROVAL_KIND } from "./approval.js";
-import {
-  CodeModeProtocolError,
-  CodeModeToolApprovalDeniedError,
-  CodeModeToolApprovalRequiredError,
-  CodeModeToolError
-} from "./errors.js";
-import {
-  assertJsonSerializable,
-  fromJsonPayload,
-  toJsonPayload
-} from "./utils/serialization.js";
-async function invokeHostTool({
-  toolName,
-  inputJson,
-  tools,
-  baseExecutionOptions,
-  codeModeOptions,
-  maxToolInputBytes,
-  maxToolOutputBytes,
-  toolCallId,
-  codeModeInterrupt,
-  skipApproval = false
-}) {
-  throwIfAborted(baseExecutionOptions.abortSignal);
-  const hostTool = tools[toolName];
-  if (!hostTool) {
-    throw new CodeModeToolError(`Unknown tool: ${toolName}`, {
-      toolName,
-      availableTools: Object.keys(tools)
-    });
-  }
-  if (hostTool.execute == null) {
-    throw new CodeModeToolError(`Tool "${toolName}" does not have execute().`, {
-      toolName
-    });
-  }
-  const input = fromJsonPayload(inputJson);
-  assertJsonSerializable(input, maxToolInputBytes, `Tool "${toolName}" input`);
-  const validation = await raceAgainstAbort(
-    validateToolInput(hostTool.inputSchema, input),
-    baseExecutionOptions.abortSignal
-  );
-  if (!validation.success) {
-    throw new CodeModeToolError(
-      `Invalid input for tool "${toolName}": ${validation.error.message}`,
-      { toolName, input, cause: validation.error.message }
-    );
-  }
-  const executionOptions = {
-    ...baseExecutionOptions,
-    toolCallId,
-    ...codeModeInterrupt !== void 0 ? { codeModeInterrupt } : {}
-  };
-  const needsApproval = !skipApproval && await raceAgainstAbort(
-    requiresApproval(hostTool, validation.value, executionOptions),
-    executionOptions.abortSignal
-  );
-  if (needsApproval) {
-    if (codeModeOptions.approval?.mode === "interrupt") {
-      return {
-        type: "interrupted",
-        toolName,
-        input: validation.value,
-        toolCallId,
-        payload: { kind: CODE_MODE_TOOL_APPROVAL_KIND }
-      };
-    }
-    const approval = await raceAgainstAbort(
-      Promise.resolve(
-        codeModeOptions.approval?.onApprovalRequired?.({
-          toolName,
-          input: validation.value,
-          toolCallId
-        })
-      ),
-      baseExecutionOptions.abortSignal
-    );
-    if (approval === void 0) {
-      throw new CodeModeToolApprovalRequiredError(
-        toolName,
-        validation.value,
-        toolCallId
-      );
-    }
-    const approved = typeof approval === "string" ? approval === "approved" : approval?.approved;
-    const reason = typeof approval === "string" ? void 0 : approval?.reason;
-    if (typeof approved !== "boolean") {
-      throw new CodeModeProtocolError(
-        `Tool "${toolName}" approval callback returned a malformed approval decision.`,
-        { toolName, toolCallId }
-      );
-    }
-    if (reason !== void 0 && typeof reason !== "string") {
-      throw new CodeModeProtocolError(
-        `Tool "${toolName}" approval callback returned a malformed approval reason.`,
-        { toolName, toolCallId }
-      );
-    }
-    if (!approved) {
-      throw new CodeModeToolApprovalDeniedError(
-        toolName,
-        validation.value,
-        toolCallId,
-        reason
-      );
-    }
-  }
-  const output = await raceAgainstAbort(
-    executeHostTool(hostTool.execute.bind(hostTool), {
-      input: validation.value,
-      options: executionOptions
-    }),
-    executionOptions.abortSignal
-  );
-  return {
-    type: "success",
-    valueJson: toJsonPayload(
-      output,
-      maxToolOutputBytes,
-      `Tool "${toolName}" output`
-    )
-  };
+import { assertJsonSerializable, fromJsonPayload, toJsonPayload } from "./utils/serialization.js";
+import { asSchema } from "ai";
+//#region src/tool-invocation.ts
+async function invokeHostTool({ toolName, inputJson, tools, baseExecutionOptions, codeModeOptions, maxToolInputBytes, maxToolOutputBytes, toolCallId, codeModeInterrupt, skipApproval = false }) {
+	throwIfAborted(baseExecutionOptions.abortSignal);
+	const hostTool = tools[toolName];
+	if (!hostTool) throw new CodeModeToolError(`Unknown tool: ${toolName}`, {
+		toolName,
+		availableTools: Object.keys(tools)
+	});
+	if (hostTool.execute == null) throw new CodeModeToolError(`Tool "${toolName}" does not have execute().`, { toolName });
+	const input = fromJsonPayload(inputJson);
+	assertJsonSerializable(input, maxToolInputBytes, `Tool "${toolName}" input`);
+	const validation = await raceAgainstAbort(validateToolInput(hostTool.inputSchema, input), baseExecutionOptions.abortSignal);
+	if (!validation.success) throw new CodeModeToolError(`Invalid input for tool "${toolName}": ${validation.error.message}`, {
+		toolName,
+		input,
+		cause: validation.error.message
+	});
+	const executionOptions = {
+		...baseExecutionOptions,
+		toolCallId,
+		...codeModeInterrupt !== void 0 ? { codeModeInterrupt } : {}
+	};
+	if (!skipApproval && await raceAgainstAbort(requiresApproval(hostTool, validation.value, executionOptions), executionOptions.abortSignal)) {
+		if (codeModeOptions.approval?.mode === "interrupt") return {
+			type: "interrupted",
+			toolName,
+			input: validation.value,
+			toolCallId,
+			payload: { kind: CODE_MODE_TOOL_APPROVAL_KIND }
+		};
+		const approval = await raceAgainstAbort(Promise.resolve(codeModeOptions.approval?.onApprovalRequired?.({
+			toolName,
+			input: validation.value,
+			toolCallId
+		})), baseExecutionOptions.abortSignal);
+		if (approval === void 0) throw new CodeModeToolApprovalRequiredError(toolName, validation.value, toolCallId);
+		const approved = typeof approval === "string" ? approval === "approved" : approval?.approved;
+		const reason = typeof approval === "string" ? void 0 : approval?.reason;
+		if (typeof approved !== "boolean") throw new CodeModeProtocolError(`Tool "${toolName}" approval callback returned a malformed approval decision.`, {
+			toolName,
+			toolCallId
+		});
+		if (reason !== void 0 && typeof reason !== "string") throw new CodeModeProtocolError(`Tool "${toolName}" approval callback returned a malformed approval reason.`, {
+			toolName,
+			toolCallId
+		});
+		if (!approved) throw new CodeModeToolApprovalDeniedError(toolName, validation.value, toolCallId, reason);
+	}
+	const output = await raceAgainstAbort(executeHostTool(hostTool.execute.bind(hostTool), {
+		input: validation.value,
+		options: executionOptions
+	}), executionOptions.abortSignal);
+	return {
+		type: "success",
+		valueJson: toJsonPayload(output, maxToolOutputBytes, `Tool "${toolName}" output`)
+	};
 }
 async function requiresApproval(hostTool, input, options) {
-  if (hostTool.needsApproval == null) {
-    return false;
-  }
-  if (typeof hostTool.needsApproval === "boolean") {
-    return hostTool.needsApproval;
-  }
-  return await hostTool.needsApproval(input, options);
+	if (hostTool.needsApproval == null) return false;
+	if (typeof hostTool.needsApproval === "boolean") return hostTool.needsApproval;
+	return await hostTool.needsApproval(input, options);
 }
 async function validateToolInput(schema, input) {
-  const normalizedSchema = asSchema(schema);
-  if (normalizedSchema.validate === void 0) {
-    return { success: true, value: input };
-  }
-  return await normalizedSchema.validate(input);
+	const normalizedSchema = asSchema(schema);
+	if (normalizedSchema.validate === void 0) return {
+		success: true,
+		value: input
+	};
+	return await normalizedSchema.validate(input);
 }
-async function executeHostTool(execute, {
-  input,
-  options
-}) {
-  const output = execute(input, options);
-  if (isAsyncIterable(output)) {
-    let finalOutput;
-    for await (const part of output) {
-      finalOutput = part;
-    }
-    return finalOutput;
-  }
-  return await output;
+async function executeHostTool(execute, { input, options }) {
+	const output = execute(input, options);
+	if (isAsyncIterable(output)) {
+		let finalOutput;
+		for await (const part of output) finalOutput = part;
+		return finalOutput;
+	}
+	return await output;
 }
 function isAsyncIterable(value) {
-  return typeof value === "object" && value !== null && Symbol.asyncIterator in value && typeof value[Symbol.asyncIterator] === "function";
+	return typeof value === "object" && value !== null && Symbol.asyncIterator in value && typeof value[Symbol.asyncIterator] === "function";
 }
 async function raceAgainstAbort(operation, abortSignal) {
-  if (abortSignal === void 0) {
-    return await operation;
-  }
-  throwIfAborted(abortSignal);
-  let rejectOnAbort;
-  const aborted = new Promise((_resolve, reject) => {
-    rejectOnAbort = reject;
-  });
-  const onAbort = () => {
-    rejectOnAbort(abortReason(abortSignal));
-  };
-  abortSignal.addEventListener("abort", onAbort, { once: true });
-  if (abortSignal.aborted) {
-    onAbort();
-  }
-  try {
-    return await Promise.race([operation, aborted]);
-  } finally {
-    abortSignal.removeEventListener("abort", onAbort);
-  }
+	if (abortSignal === void 0) return await operation;
+	throwIfAborted(abortSignal);
+	let rejectOnAbort;
+	const aborted = new Promise((_resolve, reject) => {
+		rejectOnAbort = reject;
+	});
+	const onAbort = () => {
+		rejectOnAbort(abortReason(abortSignal));
+	};
+	abortSignal.addEventListener("abort", onAbort, { once: true });
+	if (abortSignal.aborted) onAbort();
+	try {
+		return await Promise.race([operation, aborted]);
+	} finally {
+		abortSignal.removeEventListener("abort", onAbort);
+	}
 }
 function throwIfAborted(abortSignal) {
-  if (abortSignal?.aborted) {
-    throw abortReason(abortSignal);
-  }
+	if (abortSignal?.aborted) throw abortReason(abortSignal);
 }
 function abortReason(abortSignal) {
-  return abortSignal.reason ?? new DOMException("The operation was aborted.", "AbortError");
+	return abortSignal.reason ?? new DOMException("The operation was aborted.", "AbortError");
 }
-export {
-  invokeHostTool
-};
+//#endregion
+export { invokeHostTool };
+
 //# sourceMappingURL=tool-invocation.js.map

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 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":[]}
+{"version":3,"file":"tool-invocation.js","names":[],"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":";;;;;AA+BA,eAAsB,eAAe,EACnC,UACA,WACA,OACA,sBACA,iBACA,mBACA,oBACA,YACA,mBACA,eAAe,SAYqB;CACpC,eAAe,qBAAqB,WAAW;CAE/C,MAAM,WAAW,MAAM;CACvB,IAAI,CAAC,UACH,MAAM,IAAI,kBAAkB,iBAAiB,YAAY;EACvD;EACA,gBAAgB,OAAO,KAAK,KAAK;CACnC,CAAC;CAEH,IAAI,SAAS,WAAW,MACtB,MAAM,IAAI,kBAAkB,SAAS,SAAS,6BAA6B,EACzE,SACF,CAAC;CAGH,MAAM,QAAQ,gBAAgB,SAAS;CACvC,uBAAuB,OAAO,mBAAmB,SAAS,SAAS,QAAQ;CAE3E,MAAM,aAAa,MAAM,iBACvB,kBAAkB,SAAS,aAAa,KAAK,GAC7C,qBAAqB,WACvB;CACA,IAAI,CAAC,WAAW,SACd,MAAM,IAAI,kBACR,2BAA2B,SAAS,KAAK,WAAW,MAAM,WAC1D;EAAE;EAAU;EAAO,OAAO,WAAW,MAAM;CAAQ,CACrD;CAGF,MAAM,mBAAiD;EACrD,GAAG;EACH;EACA,GAAI,sBAAsB,KAAA,IAAY,EAAE,kBAAkB,IAAI,CAAC;CACjE;CASA,IANE,CAAC,gBACA,MAAM,iBACL,iBAAiB,UAAU,WAAW,OAAO,gBAAgB,GAC7D,iBAAiB,WACnB,GAEiB;EACjB,IAAI,gBAAgB,UAAU,SAAS,aACrC,OAAO;GACL,MAAM;GACN;GACA,OAAO,WAAW;GAClB;GACA,SAAS,EAAE,MAAM,6BAA6B;EAChD;EAGF,MAAM,WAAW,MAAM,iBACrB,QAAQ,QACN,gBAAgB,UAAU,qBAAqB;GAC7C;GACA,OAAO,WAAW;GAClB;EACF,CAAC,CACH,GACA,qBAAqB,WACvB;EACA,IAAI,aAAa,KAAA,GACf,MAAM,IAAI,kCACR,UACA,WAAW,OACX,UACF;EAEF,MAAM,WACJ,OAAO,aAAa,WAChB,aAAa,aACb,UAAU;EAChB,MAAM,SAAS,OAAO,aAAa,WAAW,KAAA,IAAY,UAAU;EACpE,IAAI,OAAO,aAAa,WACtB,MAAM,IAAI,sBACR,SAAS,SAAS,8DAClB;GAAE;GAAU;EAAW,CACzB;EAEF,IAAI,WAAW,KAAA,KAAa,OAAO,WAAW,UAC5C,MAAM,IAAI,sBACR,SAAS,SAAS,4DAClB;GAAE;GAAU;EAAW,CACzB;EAEF,IAAI,CAAC,UACH,MAAM,IAAI,gCACR,UACA,WAAW,OACX,YACA,MACF;CAEJ;CAEA,MAAM,SAAS,MAAM,iBACnB,gBAAgB,SAAS,QAAQ,KAAK,QAAQ,GAAG;EAC/C,OAAO,WAAW;EAClB,SAAS;CACX,CAAC,GACD,iBAAiB,WACnB;CACA,OAAO;EACL,MAAM;EACN,WAAW,cACT,QACA,oBACA,SAAS,SAAS,SACpB;CACF;AACF;AAEA,eAAe,iBACb,UACA,OACA,SACkB;CAClB,IAAI,SAAS,iBAAiB,MAC5B,OAAO;CAET,IAAI,OAAO,SAAS,kBAAkB,WACpC,OAAO,SAAS;CAElB,OAAO,MAAM,SAAS,cAAc,OAAgB,OAAgB;AACtE;AAEA,eAAe,kBACb,QACA,OAGA;CACA,MAAM,mBAAmB,SAAS,MAAM;CACxC,IAAI,iBAAiB,aAAa,KAAA,GAChC,OAAO;EAAE,SAAS;EAAM,OAAO;CAAM;CAEvC,OAAO,MAAM,iBAAiB,SAAS,KAAK;AAC9C;AAEA,eAAe,gBACb,SACA,EACE,OACA,WAKgB;CAClB,MAAM,SAAS,QAAQ,OAAgB,OAAgB;CACvD,IAAI,gBAAgB,MAAM,GAAG;EAC3B,IAAI;EACJ,WAAW,MAAM,QAAQ,QACvB,cAAc;EAEhB,OAAO;CACT;CACA,OAAO,MAAM;AACf;AAEA,SAAS,gBAAgB,OAAiD;CACxE,OACE,OAAO,UAAU,YACjB,UAAU,QACV,OAAO,iBAAiB,SACxB,OAAQ,MACN,OAAO,mBACH;AAEV;AAEA,eAAe,iBACb,WACA,aACY;CACZ,IAAI,gBAAgB,KAAA,GAClB,OAAO,MAAM;CAEf,eAAe,WAAW;CAE1B,IAAI;CACJ,MAAM,UAAU,IAAI,SAAgB,UAAU,WAAW;EACvD,gBAAgB;CAClB,CAAC;CACD,MAAM,gBAAgB;EACpB,cAAc,YAAY,WAAW,CAAC;CACxC;CAEA,YAAY,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;CAC7D,IAAI,YAAY,SACd,QAAQ;CAGV,IAAI;EACF,OAAO,MAAM,QAAQ,KAAK,CAAC,WAAW,OAAO,CAAC;CAChD,UAAU;EACR,YAAY,oBAAoB,SAAS,OAAO;CAClD;AACF;AAEA,SAAS,eAAe,aAA4C;CAClE,IAAI,aAAa,SACf,MAAM,YAAY,WAAW;AAEjC;AAEA,SAAS,YAAY,aAAmC;CACtD,OACE,YAAY,UACZ,IAAI,aAAa,8BAA8B,YAAY;AAE/D"}