@assistant-ui/core 0.3.14 → 0.3.16

package/dist/runtimes/tool-invocations/ToolInvocationTracker.js.map

@@ -1 +1 @@
-{"version":3,"file":"ToolInvocationTracker.js","names":["process","env","NODE_ENV","createAssistantStreamController","ToolCallStreamController","ToolResponse","unstable_toolResultStream","Tool","ToolModelContentPart","ToolResultStreamOptions","AssistantMetaTransformStream","ReadonlyJSONValue","isJSONValueEqual","ThreadMessage","TOOL_EXECUTION_ID","Symbol","for","ToolExecutionStatus","type","payload","AddToolResultCommand","toolCallId","toolName","result","isError","artifact","modelContent","ToolCallEntry","argsText","hasResult","executionId","skipExecute","controller","argsComplete","SettledResolver","executionIds","ReadonlySet","resolve","isArgsTextComplete","JSON","parse","parseArgsText","undefined","isEquivalentCompleteArgsText","previous","next","previousValue","nextValue","getToolExecutionId","value","Record","PropertyKey","ToolInvocationTracker","_getTools","_callbacks","Callbacks","_entries","Map","_humanInput","reject","reason","_executing","Set","_settledResolvers","_statuses","_ac","AbortController","_pendingRestore","_lastSnapshot","Snapshot","_isRunning","_controller","ReturnType","_pipelineDead","_pipelineRestartUsed","constructor","getTools","callbacks","_initPipeline","stream","human","_onHumanInput","transform","_getWrappedTools","signal","onExecutionStart","id","_name","_onExecutionStart","onExecutionEnd","_onExecutionEnd","pipeThrough","pipeTo","WritableStream","write","chunk","_handleResultChunk","err","console","error","catch","setState","snapshot","_demoteEntriesToRestored","clear","messages","isRunning","isLoading","restoreFromLoading","previousIsRunning","_processMessages","reset","abort","size","_invokeOnStatusesChange","Promise","forEach","Error","push","resume","handlers","get","delete","_setStatus","getStatuses","ReadonlyMap","tools","Object","fromEntries","entries","map","name","tool","execute","streamCall","wrappedTool","args","context","Parameters","entry","_captureExecution","reader","set","add","_deleteStatus","pending","some","has","length","meta","_invokeOnResult","command","onResult","onStatusesChange","status","_hasExecutableTool","_shouldCloseArgsStream","_startActiveEntry","toolCallController","addToolCallPart","_processArgsText","content","shouldWriteArgsText","warn","startsWith","shouldClose","close","delta","slice","append","isRestore","message","Array","isArray","existing","signatureChanged","activeController","setResponse","ExecutionStatus","statuses"],"sources":["../../../src/runtimes/tool-invocations/ToolInvocationTracker.ts"],"sourcesContent":["declare const process: { env: { NODE_ENV?: string } };\n\nimport {\n  createAssistantStreamController,\n  type ToolCallStreamController,\n  ToolResponse,\n  unstable_toolResultStream,\n  type Tool,\n  type ToolModelContentPart,\n  type ToolResultStreamOptions,\n} from \"assistant-stream\";\nimport {\n  AssistantMetaTransformStream,\n  type ReadonlyJSONValue,\n} from \"assistant-stream/utils\";\nimport { isJSONValueEqual } from \"../../utils/json/is-json-equal\";\nimport type { ThreadMessage } from \"../../types/message\";\n\nconst TOOL_EXECUTION_ID = Symbol.for(\"assistant-stream.tool-execution-id\");\n\n/**\n * Streaming execution state for a frontend tool.\n */\nexport type ToolExecutionStatus =\n  | { type: \"executing\" }\n  | {\n      type: \"interrupt\";\n      payload: { type: \"human\"; payload: unknown };\n    };\n\nexport type AddToolResultCommand = {\n  readonly type: \"add-tool-result\";\n  readonly toolCallId: string;\n  readonly toolName: string;\n  readonly result: ReadonlyJSONValue;\n  readonly isError: boolean;\n  readonly artifact?: ReadonlyJSONValue;\n  readonly modelContent?: readonly ToolModelContentPart[];\n};\n\ntype ToolCallEntry = {\n  toolName: string;\n  argsText: string;\n  hasResult: boolean;\n  executionId?: symbol;\n  skipExecute?: boolean;\n} & (\n  | {\n      /** Restored phase — observed during a history-load snapshot. */\n      controller?: undefined;\n      argsComplete?: undefined;\n    }\n  | {\n      /** Active phase — chunks are flowing through `controller`. */\n      controller: ToolCallStreamController;\n      argsComplete: boolean;\n    }\n);\n\ntype SettledResolver = {\n  executionIds: ReadonlySet<symbol>;\n  resolve: () => void;\n};\n\nconst isArgsTextComplete = (argsText: string) => {\n  try {\n    JSON.parse(argsText);\n    return true;\n  } catch {\n    return false;\n  }\n};\n\nconst parseArgsText = (argsText: string) => {\n  try {\n    return JSON.parse(argsText);\n  } catch {\n    return undefined;\n  }\n};\n\nconst isEquivalentCompleteArgsText = (previous: string, next: string) => {\n  const previousValue = parseArgsText(previous);\n  const nextValue = parseArgsText(next);\n  if (previousValue === undefined || nextValue === undefined) return false;\n  return isJSONValueEqual(previousValue, nextValue);\n};\n\nconst getToolExecutionId = (value: object): symbol | undefined =>\n  (value as Record<PropertyKey, unknown>)[TOOL_EXECUTION_ID] as\n    | symbol\n    | undefined;\n\n/**\n * Plain-class port of the former `useToolInvocations` React hook. Owns the\n * assistant-stream pipeline that drives client-side `streamCall` / `execute`\n * for tool-call parts surfaced by a thread runtime, plus the per-tool-call\n * status map that consumers render against.\n *\n * **Contract**: `streamCall` (and `execute`) fires exactly once per logical\n * `toolCallId`. Args mutations after first completion, result replacement,\n * and result clearing are *not* surfaced through additional `streamCall`\n * invocations — by design — so hosts cannot observe spurious re-fires of\n * side effects. The follow-up `reader.events()` API will expose those\n * post-completion transitions to consumers that opt in.\n *\n * State-transition safety: every public method that observes runtime state\n * (`setState`, `reset`, `abort`, `resume`) wraps its work in try/catch and\n * logs to `console.error` rather than throwing. The tracker is built into\n * the hot message-processing path, so a malformed snapshot must never crash\n * the host runtime. See ./EDGE_CASES.md for the known non-trivial state\n * transitions and what each does today.\n *\n * @deprecated Internal — for framework bindings; not a public API. May change without notice.\n */\nexport class ToolInvocationTracker {\n  private readonly _getTools: () => Record<string, Tool> | undefined;\n  private readonly _callbacks: ToolInvocationTracker.Callbacks;\n\n  private readonly _entries = new Map<string, ToolCallEntry>();\n  private readonly _humanInput = new Map<\n    string,\n    {\n      executionId: symbol;\n      resolve: (payload: unknown) => void;\n      reject: (reason: unknown) => void;\n    }\n  >();\n  private readonly _executing = new Set<symbol>();\n  private readonly _settledResolvers: SettledResolver[] = [];\n\n  private _statuses = new Map<string, ToolExecutionStatus>();\n\n  private _ac: AbortController = new AbortController();\n  private _pendingRestore = true;\n\n  /** Cached last snapshot, used to skip processing on identical re-renders. */\n  private _lastSnapshot: ToolInvocationTracker.Snapshot | null = null;\n  private _isRunning = false;\n\n  private _controller!: ReturnType<typeof createAssistantStreamController>[1];\n\n  /**\n   * Set when the assistant-stream pipeline has died (errored out via\n   * `.pipeTo(...).catch(...)`). The next `setState` re-initializes the\n   * pipeline and demotes all active entries to restored so they survive\n   * across the restart without re-firing `streamCall` (preserves the\n   * \"exactly once\" contract). Capped at a single auto-restart per session\n   * — repeated failures keep the tracker dead with a more visible error.\n   */\n  private _pipelineDead = false;\n  private _pipelineRestartUsed = false;\n\n  constructor(\n    getTools: () => Record<string, Tool> | undefined,\n    callbacks: ToolInvocationTracker.Callbacks,\n  ) {\n    this._getTools = getTools;\n    this._callbacks = callbacks;\n\n    this._initPipeline();\n  }\n\n  /**\n   * Build the assistant-stream pipeline. Called once from the constructor\n   * and at most once again if `_pipelineDead` is set (see F.4 in\n   * EDGE_CASES.md).\n   */\n  private _initPipeline(): void {\n    const [stream, controller] = createAssistantStreamController();\n    this._controller = controller;\n\n    const human = (\n      toolCallId: string,\n      payload: unknown,\n      executionId?: symbol,\n    ) => this._onHumanInput(toolCallId, payload, executionId);\n    const transform = unstable_toolResultStream(\n      () => this._getWrappedTools(),\n      () => this._ac.signal,\n      human,\n      {\n        onExecutionStart: (id: string, _name: string, executionId?: symbol) =>\n          this._onExecutionStart(id, executionId),\n        onExecutionEnd: (id: string, _name: string, executionId?: symbol) =>\n          this._onExecutionEnd(id, executionId),\n      } as ToolResultStreamOptions,\n    );\n\n    stream\n      .pipeThrough(transform)\n      .pipeThrough(new AssistantMetaTransformStream())\n      .pipeTo(\n        new WritableStream({\n          write: (chunk) => {\n            try {\n              if (chunk.type !== \"result\") return;\n              this._handleResultChunk(chunk);\n            } catch (err) {\n              console.error(\n                \"[ToolInvocationTracker] result chunk handling failed\",\n                err,\n              );\n            }\n          },\n        }),\n      )\n      .catch((err) => {\n        console.error(\n          \"[ToolInvocationTracker] stream pipeline failed; will attempt single restart on next setState\",\n          err,\n        );\n        this._pipelineDead = true;\n      });\n  }\n\n  // ───────────────────────── public API ─────────────────────────\n\n  /**\n   * Feed the next observed snapshot into the tracker. Called from the host\n   * runtime whenever its message list / running state changes.\n   */\n  public setState(snapshot: ToolInvocationTracker.Snapshot): void {\n    try {\n      // Recover from a dead pipeline before processing anything. We demote\n      // all active entries to \"restored\" so the rebuilt pipeline does not\n      // re-fire `streamCall` for tool calls that already fired pre-death;\n      // preserves the \"exactly once per toolCallId\" contract.\n      if (this._pipelineDead) {\n        if (this._pipelineRestartUsed) {\n          // Already retried once and failed again. Stay dead.\n          return;\n        }\n        this._pipelineRestartUsed = true;\n        this._pipelineDead = false;\n        this._demoteEntriesToRestored();\n        this._executing.clear();\n        this._ac = new AbortController();\n        this._initPipeline();\n        // Fall through and process the snapshot against the fresh pipeline.\n      }\n\n      // Identical snapshot — skip processing entirely. Note: external-store\n      // runtimes rebuild the messages array on every adapter update, so this\n      // fast-path rarely triggers there; it's primarily for the React-hook\n      // shim where state references are stable.\n      if (\n        this._lastSnapshot &&\n        this._lastSnapshot.messages === snapshot.messages &&\n        this._lastSnapshot.isRunning === snapshot.isRunning &&\n        this._lastSnapshot.isLoading === snapshot.isLoading\n      ) {\n        return;\n      }\n\n      // While the host is still loading initial state, treat every snapshot\n      // as historical: tool calls are recorded so the next live snapshot can\n      // diff against them, but `streamCall` / `execute` do not fire.\n      const restoreFromLoading = snapshot.isLoading === true;\n      if (restoreFromLoading) {\n        this._pendingRestore = true;\n      }\n\n      // E.4 / AF3 — only mark `_lastSnapshot`/`_isRunning` as observed after\n      // processing succeeds. If `_processMessages` throws, the next snapshot\n      // (even if identical) gets re-processed against the recovered state.\n      const previousIsRunning = this._isRunning;\n      this._isRunning = snapshot.isRunning;\n      try {\n        this._processMessages(snapshot.messages);\n      } catch (err) {\n        this._isRunning = previousIsRunning;\n        throw err;\n      }\n      this._lastSnapshot = snapshot;\n      this._pendingRestore = false;\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] setState failed; snapshot dropped\",\n        err,\n      );\n    }\n  }\n\n  /**\n   * Reset the tracker so the next observed snapshot is treated as historical.\n   * Clears entries and aborts any in-flight executions. Used by callers like\n   * `importExternalState` to mark a freshly loaded state as restored.\n   */\n  public reset(): void {\n    try {\n      this._pendingRestore = true;\n      this._entries.clear();\n      this._lastSnapshot = null;\n      void this.abort();\n      // Statuses are cleared synchronously: discarded executions may never\n      // settle (aborting hands the signal to the tool, it does not force\n      // settlement), and a late settlement of one sibling must not republish\n      // the others. `_deleteStatus` no-ops for cleared ids, so post-reset\n      // settlements stay silent.\n      if (this._statuses.size > 0) {\n        this._statuses = new Map();\n        this._invokeOnStatusesChange();\n      }\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] reset failed\", err);\n    }\n  }\n\n  /**\n   * Abort any in-flight `execute()` invocations. Resolves once all of them\n   * have settled (or immediately if none are running).\n   */\n  public abort(): Promise<void> {\n    try {\n      this._humanInput.forEach(({ reject }) => {\n        try {\n          reject(new Error(\"Tool execution aborted\"));\n        } catch {\n          // host rejection handler threw — already in the abort path,\n          // swallow so we continue cleaning up.\n        }\n      });\n      this._humanInput.clear();\n\n      this._ac.abort();\n      this._ac = new AbortController();\n\n      if (this._executing.size === 0) {\n        return Promise.resolve();\n      }\n      const executionIds = new Set(this._executing);\n      return new Promise<void>((resolve) => {\n        this._settledResolvers.push({ executionIds, resolve });\n      });\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] abort failed\", err);\n      return Promise.resolve();\n    }\n  }\n\n  /**\n   * Resolve a pending human-input request for the given tool call. Returns\n   * `true` if a pending request was resumed, `false` if the tracker has no\n   * outstanding request for that id (the caller should fall back to its own\n   * dispatch path).\n   */\n  public resume(toolCallId: string, payload: unknown): boolean {\n    try {\n      const handlers = this._humanInput.get(toolCallId);\n      if (!handlers) return false;\n      this._humanInput.delete(toolCallId);\n      this._setStatus(toolCallId, { type: \"executing\" });\n      handlers.resolve(payload);\n      return true;\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] resume failed\", err);\n      return false;\n    }\n  }\n\n  /**\n   * Returns the current tool execution status map. The returned `Map` is\n   * the tracker's internal store — do not mutate it. Treat the reference\n   * as a snapshot that may be replaced wholesale on the next status\n   * transition.\n   */\n  public getStatuses(): ReadonlyMap<string, ToolExecutionStatus> {\n    return this._statuses;\n  }\n\n  // ───────────────────── internal: tool wrapping ─────────────────────\n\n  private _getWrappedTools(): Record<string, Tool> | undefined {\n    const tools = this._getTools();\n    if (!tools) return undefined;\n\n    return Object.fromEntries(\n      Object.entries(tools).map(([name, tool]) => {\n        const execute = tool.execute;\n        const streamCall = tool.streamCall;\n        if (execute === undefined && streamCall === undefined)\n          return [name, tool];\n\n        const wrappedTool = {\n          ...tool,\n          ...(execute !== undefined && {\n            execute: (...[args, context]: Parameters<typeof execute>) => {\n              const executionId = getToolExecutionId(context);\n              const entry = this._captureExecution(\n                context.toolCallId,\n                executionId,\n              );\n              if (!entry || entry.skipExecute) {\n                return new Promise(() => {}) as never;\n              }\n              return execute(args, context);\n            },\n          }),\n          ...(streamCall !== undefined && {\n            streamCall: (\n              ...[reader, context]: Parameters<typeof streamCall>\n            ) => {\n              const executionId = getToolExecutionId(context);\n              const entry = this._captureExecution(\n                context.toolCallId,\n                executionId,\n              );\n              if (!entry) return;\n              return streamCall(reader, context);\n            },\n          }),\n        } as Tool;\n        return [name, wrappedTool];\n      }),\n    ) as Record<string, Tool>;\n  }\n\n  private _captureExecution(\n    toolCallId: string,\n    executionId: symbol | undefined,\n  ): ToolCallEntry | undefined {\n    if (executionId === undefined) return undefined;\n    const entry = this._entries.get(toolCallId);\n    if (!entry?.controller) return undefined;\n    if (entry.executionId === undefined) entry.executionId = executionId;\n    return entry.executionId === executionId ? entry : undefined;\n  }\n\n  // ──────────────── internal: execution lifecycle callbacks ────────────────\n\n  private _onHumanInput(\n    toolCallId: string,\n    payload: unknown,\n    executionId?: symbol,\n  ): Promise<unknown> {\n    return new Promise<unknown>((resolve, reject) => {\n      // A discarded execution must not resurrect a status entry or park an\n      // unanswerable request after its id has been reused.\n      const entry = this._entries.get(toolCallId);\n      if (!entry?.controller || entry.executionId !== executionId) {\n        reject(new Error(\"Tool execution aborted\"));\n        return;\n      }\n      const previous = this._humanInput.get(toolCallId);\n      if (previous) {\n        try {\n          previous.reject(\n            new Error(\"Human input request was superseded by a new request\"),\n          );\n        } catch {\n          // host rejection handler threw; ignore and proceed\n        }\n      }\n      this._humanInput.set(toolCallId, {\n        executionId: executionId!,\n        resolve,\n        reject,\n      });\n      this._setStatus(toolCallId, {\n        type: \"interrupt\",\n        payload: { type: \"human\", payload },\n      });\n    });\n  }\n\n  private _onExecutionStart(\n    toolCallId: string,\n    executionId: symbol | undefined,\n  ): void {\n    if (!this._captureExecution(toolCallId, executionId)) return;\n    const entry = this._entries.get(toolCallId)!;\n    if (entry.skipExecute) return;\n\n    this._executing.add(executionId!);\n    this._setStatus(toolCallId, { type: \"executing\" });\n  }\n\n  private _onExecutionEnd(\n    toolCallId: string,\n    executionId: symbol | undefined,\n  ): void {\n    if (executionId === undefined || !this._executing.delete(executionId))\n      return;\n\n    const entry = this._entries.get(toolCallId);\n    if (entry?.executionId === executionId) this._deleteStatus(toolCallId);\n\n    const pending: SettledResolver[] = [];\n    this._settledResolvers.forEach(({ executionIds, resolve }) => {\n      if ([...executionIds].some((id) => this._executing.has(id))) {\n        pending.push({ executionIds, resolve });\n        return;\n      }\n      try {\n        resolve();\n      } catch {\n        // ignore — settled-resolver consumer threw\n      }\n    });\n    this._settledResolvers.length = 0;\n    this._settledResolvers.push(...pending);\n  }\n\n  private _handleResultChunk(chunk: {\n    type: \"result\";\n    result: ReadonlyJSONValue;\n    isError: boolean;\n    artifact?: ReadonlyJSONValue;\n    modelContent?: readonly ToolModelContentPart[];\n    meta: { toolCallId: string; toolName: string };\n  }): void {\n    const toolCallId = chunk.meta.toolCallId;\n    const executionId = getToolExecutionId(chunk);\n    const entry = this._entries.get(toolCallId);\n\n    if (!entry || entry.executionId !== executionId) return;\n\n    // The host already set the result (via the live snapshot's\n    // `setResponse` path). Suppress the executor's redundant emit.\n    if (entry?.hasResult) return;\n\n    this._invokeOnResult({\n      type: \"add-tool-result\",\n      toolCallId,\n      toolName: chunk.meta.toolName,\n      result: chunk.result,\n      isError: chunk.isError,\n      ...(chunk.artifact !== undefined && { artifact: chunk.artifact }),\n      ...(chunk.modelContent !== undefined && {\n        modelContent: chunk.modelContent,\n      }),\n    });\n  }\n\n  // ──────────────── internal: callback invocation (AF1/AF2) ────────────────\n\n  private _invokeOnResult(command: AddToolResultCommand): void {\n    try {\n      this._callbacks.onResult(command);\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] onResult callback threw; result dropped\",\n        err,\n      );\n    }\n  }\n\n  private _invokeOnStatusesChange(): void {\n    try {\n      this._callbacks.onStatusesChange(this._statuses);\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] onStatusesChange callback threw; status change not propagated\",\n        err,\n      );\n    }\n  }\n\n  // ──────────────── internal: status map mutations ────────────────\n\n  private _setStatus(toolCallId: string, status: ToolExecutionStatus): void {\n    const next = new Map(this._statuses);\n    next.set(toolCallId, status);\n    this._statuses = next;\n    this._invokeOnStatusesChange();\n  }\n\n  private _deleteStatus(toolCallId: string): void {\n    if (!this._statuses.has(toolCallId)) return;\n    const next = new Map(this._statuses);\n    next.delete(toolCallId);\n    this._statuses = next;\n    this._invokeOnStatusesChange();\n  }\n\n  // ──────────────── internal: snapshot processing ────────────────\n\n  private _hasExecutableTool(toolName: string): boolean {\n    const tool = this._getTools()?.[toolName];\n    return tool?.execute !== undefined || tool?.streamCall !== undefined;\n  }\n\n  private _shouldCloseArgsStream({\n    toolName,\n    argsText,\n    hasResult,\n  }: {\n    toolName: string;\n    argsText: string;\n    hasResult: boolean;\n  }): boolean {\n    if (hasResult) return true;\n    if (!this._hasExecutableTool(toolName)) {\n      return !this._isRunning && isArgsTextComplete(argsText);\n    }\n    return isArgsTextComplete(argsText);\n  }\n\n  private _startActiveEntry(\n    toolCallId: string,\n    toolName: string,\n    skipExecute: boolean,\n  ): ToolCallEntry {\n    const toolCallController = this._controller.addToolCallPart({\n      toolName,\n      toolCallId,\n    });\n    const entry: ToolCallEntry = {\n      toolName,\n      controller: toolCallController,\n      argsText: \"\",\n      hasResult: false,\n      skipExecute,\n      argsComplete: false,\n    };\n    this._entries.set(toolCallId, entry);\n    return entry;\n  }\n\n  /**\n   * Demote every active entry back to the restored phase. Used by the\n   * pipeline-restart path so that, after a fresh pipeline is built, the\n   * next observed snapshot does not re-fire `streamCall` for tool calls\n   * that already fired pre-death. Args / hasResult tracking is preserved\n   * so signature comparisons still work.\n   */\n  private _demoteEntriesToRestored(): void {\n    for (const [toolCallId, entry] of this._entries) {\n      if (!entry.controller) continue;\n      this._entries.set(toolCallId, {\n        toolName: entry.toolName,\n        argsText: entry.argsText,\n        hasResult: entry.hasResult,\n      });\n    }\n  }\n\n  private _processArgsText(\n    entry: ToolCallEntry,\n    content: {\n      toolCallId: string;\n      toolName: string;\n      argsText: string;\n      result?: unknown;\n    },\n  ): void {\n    if (!entry.controller) return;\n    const hasResult = content.result !== undefined;\n\n    if (content.argsText !== entry.argsText) {\n      let shouldWriteArgsText = true;\n\n      if (entry.argsComplete) {\n        if (isEquivalentCompleteArgsText(entry.argsText, content.argsText)) {\n          // A.3 — key reorder. Track new text, no re-fire needed.\n          entry.argsText = content.argsText;\n          shouldWriteArgsText = false;\n        } else {\n          // A.4 — args changed after first completion. Under the\n          // \"exactly once per toolCallId\" contract we do not restart the\n          // stream. The host's existing `streamCall` keeps its original\n          // args view; the snapshot's new text is recorded for diffing\n          // but not surfaced. Events API in a follow-up will expose this\n          // to consumers that opt in.\n          if (process.env.NODE_ENV !== \"production\") {\n            console.warn(\n              \"[ToolInvocationTracker] argsText changed after first completion; not re-firing streamCall (see EDGE_CASES.md A.4)\",\n              {\n                previous: entry.argsText,\n                next: content.argsText,\n                toolCallId: content.toolCallId,\n              },\n            );\n          }\n          shouldWriteArgsText = false;\n        }\n      } else if (!content.argsText.startsWith(entry.argsText)) {\n        if (\n          isArgsTextComplete(entry.argsText) &&\n          isArgsTextComplete(content.argsText) &&\n          isEquivalentCompleteArgsText(entry.argsText, content.argsText)\n        ) {\n          const shouldClose = this._shouldCloseArgsStream({\n            toolName: content.toolName,\n            argsText: content.argsText,\n            hasResult,\n          });\n          if (shouldClose) entry.controller.argsText.close();\n          entry.argsText = content.argsText;\n          entry.argsComplete = shouldClose;\n          shouldWriteArgsText = false;\n        } else {\n          // A.2 — args regressed mid-stream. Under the \"exactly once\"\n          // contract we do not restart. The controller keeps whatever\n          // prefix we already streamed; subsequent prefix-respecting\n          // updates can still flow against it. Snapshots that never\n          // re-converge to a prefix will leave the controller's args\n          // view stale relative to the snapshot. Events API in a\n          // follow-up will expose this to consumers that opt in.\n          if (process.env.NODE_ENV !== \"production\") {\n            console.warn(\n              \"[ToolInvocationTracker] argsText regressed mid-stream; not restarting (see EDGE_CASES.md A.2)\",\n              {\n                previous: entry.argsText,\n                next: content.argsText,\n                toolCallId: content.toolCallId,\n              },\n            );\n          }\n          shouldWriteArgsText = false;\n        }\n      }\n\n      if (shouldWriteArgsText && entry.controller) {\n        const delta = content.argsText.slice(entry.argsText.length);\n        entry.controller.argsText.append(delta);\n        const shouldClose = this._shouldCloseArgsStream({\n          toolName: content.toolName,\n          argsText: content.argsText,\n          hasResult,\n        });\n        if (shouldClose) entry.controller.argsText.close();\n        entry.argsText = content.argsText;\n        entry.argsComplete = shouldClose;\n      }\n    }\n\n    if (!entry.argsComplete && entry.controller) {\n      // ToolExecutionStream parses the streamed prefix on close, so the close\n      // gates on the streamed content; a divergent snapshot (A.2) can be\n      // complete while the controller still holds an incomplete stale prefix.\n      const shouldClose = this._shouldCloseArgsStream({\n        toolName: content.toolName,\n        argsText: entry.argsText,\n        hasResult,\n      });\n      if (shouldClose) {\n        entry.controller.argsText.close();\n        entry.argsComplete = true;\n      }\n    }\n  }\n\n  private _processMessages(messages: readonly ThreadMessage[]): void {\n    const isRestore = this._pendingRestore;\n\n    for (const message of messages) {\n      if (!message || !Array.isArray((message as ThreadMessage).content)) {\n        continue;\n      }\n      for (const content of message.content as readonly ThreadMessage[\"content\"][number][]) {\n        if (!content || content.type !== \"tool-call\") continue;\n\n        const existing = this._entries.get(content.toolCallId);\n\n        if (isRestore) {\n          // Don't overwrite an already-active entry (e.g. live tool-call\n          // observed before this restore snapshot landed). Restore can\n          // only seed entries the runtime has never seen.\n          if (!existing?.controller) {\n            this._entries.set(content.toolCallId, {\n              toolName: content.toolName,\n              argsText: content.argsText,\n              hasResult: content.result !== undefined,\n            });\n          }\n          if (content.messages) this._processMessages(content.messages);\n          continue;\n        }\n\n        // Live snapshot.\n        let entry = existing;\n\n        if (entry && !entry.controller) {\n          // Restored entry observed in a live snapshot. Promote if its\n          // signature has changed; otherwise treat as still-historical.\n          const signatureChanged =\n            content.argsText !== entry.argsText ||\n            (content.result !== undefined) !== entry.hasResult;\n          if (!signatureChanged) {\n            if (content.messages) this._processMessages(content.messages);\n            continue;\n          }\n          this._entries.delete(content.toolCallId);\n          entry = undefined;\n        }\n\n        if (!entry) {\n          entry = this._startActiveEntry(\n            content.toolCallId,\n            content.toolName,\n            content.result !== undefined,\n          );\n        }\n\n        this._processArgsText(entry, content);\n\n        if (content.result !== undefined && !entry.hasResult) {\n          // `entry` is in active phase from this point — either just\n          // created by `_startActiveEntry`, or pre-existing with a live\n          // controller. Narrow once instead of asserting at every use.\n          const { controller: activeController } = entry;\n          if (!activeController) continue;\n          entry.hasResult = true;\n          entry.argsComplete = true;\n          activeController.setResponse(\n            new ToolResponse({\n              result: content.result as ReadonlyJSONValue,\n              artifact: content.artifact as ReadonlyJSONValue | undefined,\n              isError: content.isError,\n              ...(content.modelContent !== undefined\n                ? { modelContent: content.modelContent }\n                : {}),\n            }),\n          );\n          activeController.close();\n        }\n\n        if (content.messages) this._processMessages(content.messages);\n      }\n    }\n  }\n}\n\nexport namespace ToolInvocationTracker {\n  export type ExecutionStatus = ToolExecutionStatus;\n\n  export type Snapshot = {\n    readonly messages: readonly ThreadMessage[];\n    /** Whether the producing runtime is currently streaming new output. */\n    readonly isRunning: boolean;\n    /**\n     * Whether the producing runtime is still loading historical state.\n     * When `true`, every snapshot is treated as historical (no `streamCall` /\n     * `execute` fires). When `false`, processing resumes as live.\n     */\n    readonly isLoading?: boolean;\n  };\n\n  export type Callbacks = {\n    /**\n     * Invoked when a client-side `execute()` returns a result and the runtime\n     * needs to feed it back into the conversation.\n     */\n    onResult: (command: AddToolResultCommand) => void;\n    /**\n     * Invoked whenever the per-tool-call status map changes (executing /\n     * interrupt / cleared). The callback receives a fresh map; mutating the\n     * argument is not supported.\n     */\n    onStatusesChange: (\n      statuses: ReadonlyMap<string, ToolExecutionStatus>,\n    ) => void;\n  };\n}\n"],"mappings":";;;;AAkBA,MAAMc,oBAAoBC,OAAOC,IAAI,oCAAoC;AA8CzE,MAAMsB,sBAAsBV,aAAqB;CAC/C,IAAI;EACFW,KAAKC,MAAMZ,QAAQ;EACnB,OAAO;CACT,QAAQ;EACN,OAAO;CACT;AACF;AAEA,MAAMa,iBAAiBb,aAAqB;CAC1C,IAAI;EACF,OAAOW,KAAKC,MAAMZ,QAAQ;CAC5B,QAAQ;EACN;CACF;AACF;AAEA,MAAMe,gCAAgCC,UAAkBC,SAAiB;CACvE,MAAMC,gBAAgBL,cAAcG,QAAQ;CAC5C,MAAMG,YAAYN,cAAcI,IAAI;CACpC,IAAIC,kBAAkBJ,KAAAA,KAAaK,cAAcL,KAAAA,GAAW,OAAO;CACnE,OAAO9B,iBAAiBkC,eAAeC,SAAS;AAClD;AAEA,MAAMC,sBAAsBC,UACzBA,MAAuCnC;;;;;;;;;;;;;;;;;;;;;;;AA0B1C,IAAasC,wBAAb,MAAmC;CACjC;CACA;CAEA,2BAA4B,IAAIK,IAA2B;CAC3D,8BAA+B,IAAIA,IAOjC;CACF,6BAA8B,IAAIK,IAAY;CAC9C,oBAAwD,CAAA;CAExD,4BAAoB,IAAIL,IAAiC;CAEzD,MAA+B,IAAIS,gBAAgB;CACnD,kBAA0B;;CAG1B,gBAA+D;CAC/D,aAAqB;CAErB;;;;;;;;;CAUA,gBAAwB;CACxB,uBAA+B;CAE/BS,YACEC,UACAC,WACA;EACA,KAAKxB,YAAYuB;EACjB,KAAKtB,aAAauB;EAElB,KAAKC,cAAc;CACrB;;;;;;CAOA,gBAA8B;EAC5B,MAAM,CAACC,QAAQ/C,cAAc7B,gCAAgC;EAC7D,KAAKoE,cAAcvC;EAEnB,MAAMgD,SACJ3D,YACAF,SACAW,gBACG,KAAKmD,cAAc5D,YAAYF,SAASW,WAAW;EACxD,MAAMoD,YAAY5E,gCACV,KAAK6E,iBAAiB,SACtB,KAAKlB,IAAImB,QACfJ,OACA;GACEK,mBAAmBC,IAAYC,OAAezD,gBAC5C,KAAK0D,kBAAkBF,IAAIxD,WAAW;GACxC2D,iBAAiBH,IAAYC,OAAezD,gBAC1C,KAAK4D,gBAAgBJ,IAAIxD,WAAW;EACxC,CACF;EAEAiD,OACGY,YAAYT,SAAS,CAAC,CACtBS,YAAY,IAAIjF,6BAA6B,CAAC,CAAC,CAC/CkF,OACC,IAAIC,eAAe,EACjBC,QAAQC,UAAU;GAChB,IAAI;IACF,IAAIA,MAAM7E,SAAS,UAAU;IAC7B,KAAK8E,mBAAmBD,KAAK;GAC/B,SAASE,KAAK;IACZC,QAAQC,MACN,wDACAF,GACF;GACF;EACF,EACF,CAAC,CACH,CAAC,CACAG,OAAOH,QAAQ;GACdC,QAAQC,MACN,gGACAF,GACF;GACA,KAAKxB,gBAAgB;EACvB,CAAC;CACL;;;;;CAQA,SAAgB6B,UAAgD;EAC9D,IAAI;GAKF,IAAI,KAAK7B,eAAe;IACtB,IAAI,KAAKC,sBAEP;IAEF,KAAKA,uBAAuB;IAC5B,KAAKD,gBAAgB;IACrB,KAAK8B,yBAAyB;IAC9B,KAAK1C,WAAW2C,MAAM;IACtB,KAAKvC,MAAM,IAAIC,gBAAgB;IAC/B,KAAKY,cAAc;GAErB;GAMA,IACE,KAAKV,iBACL,KAAKA,cAAcqC,aAAaH,SAASG,YACzC,KAAKrC,cAAcsC,cAAcJ,SAASI,aAC1C,KAAKtC,cAAcuC,cAAcL,SAASK,WAE1C;GAOF,IAD2BL,SAASK,cAAc,MAEhD,KAAKxC,kBAAkB;GAMzB,MAAM0C,oBAAoB,KAAKvC;GAC/B,KAAKA,aAAagC,SAASI;GAC3B,IAAI;IACF,KAAKI,iBAAiBR,SAASG,QAAQ;GACzC,SAASR,KAAK;IACZ,KAAK3B,aAAauC;IAClB,MAAMZ;GACR;GACA,KAAK7B,gBAAgBkC;GACrB,KAAKnC,kBAAkB;EACzB,SAAS8B,KAAK;GACZC,QAAQC,MACN,6DACAF,GACF;EACF;CACF;;;;;;CAOA,QAAqB;EACnB,IAAI;GACF,KAAK9B,kBAAkB;GACvB,KAAKX,SAASgD,MAAM;GACpB,KAAKpC,gBAAgB;GACrB,KAAU4C,MAAM;GAMhB,IAAI,KAAKhD,UAAUiD,OAAO,GAAG;IAC3B,KAAKjD,4BAAY,IAAIP,IAAI;IACzB,KAAKyD,wBAAwB;GAC/B;EACF,SAASjB,KAAK;GACZC,QAAQC,MAAM,wCAAwCF,GAAG;EAC3D;CACF;;;;;CAMA,QAA8B;EAC5B,IAAI;GACF,KAAKvC,YAAY0D,SAAS,EAAEzD,aAAa;IACvC,IAAI;KACFA,uBAAO,IAAI0D,MAAM,wBAAwB,CAAC;IAC5C,QAAQ,CAEN;GAEJ,CAAC;GACD,KAAK3D,YAAY8C,MAAM;GAEvB,KAAKvC,IAAI+C,MAAM;GACf,KAAK/C,MAAM,IAAIC,gBAAgB;GAE/B,IAAI,KAAKL,WAAWoD,SAAS,GAC3B,OAAOE,QAAQ9E,QAAQ;GAEzB,MAAMF,eAAe,IAAI2B,IAAI,KAAKD,UAAU;GAC5C,OAAO,IAAIsD,SAAe9E,YAAY;IACpC,KAAK0B,kBAAkBuD,KAAK;KAAEnF;KAAcE;IAAQ,CAAC;GACvD,CAAC;EACH,SAAS4D,KAAK;GACZC,QAAQC,MAAM,wCAAwCF,GAAG;GACzD,OAAOkB,QAAQ9E,QAAQ;EACzB;CACF;;;;;;;CAQA,OAAchB,YAAoBF,SAA2B;EAC3D,IAAI;GACF,MAAMqG,WAAW,KAAK9D,YAAY+D,IAAIpG,UAAU;GAChD,IAAI,CAACmG,UAAU,OAAO;GACtB,KAAK9D,YAAYgE,OAAOrG,UAAU;GAClC,KAAKsG,WAAWtG,YAAY,EAAEH,MAAM,YAAY,CAAC;GACjDsG,SAASnF,QAAQlB,OAAO;GACxB,OAAO;EACT,SAAS8E,KAAK;GACZC,QAAQC,MAAM,yCAAyCF,GAAG;GAC1D,OAAO;EACT;CACF;;;;;;;CAQA,cAA+D;EAC7D,OAAO,KAAKjC;CACd;CAIA,mBAA6D;EAC3D,MAAM8D,QAAQ,KAAKzE,UAAU;EAC7B,IAAI,CAACyE,OAAO,OAAOpF,KAAAA;EAEnB,OAAOqF,OAAOC,YACZD,OAAOE,QAAQH,KAAK,CAAC,CAACI,KAAK,CAACC,MAAMC,UAAU;GAC1C,MAAMC,UAAUD,KAAKC;GACrB,MAAMC,aAAaF,KAAKE;GACxB,IAAID,YAAY3F,KAAAA,KAAa4F,eAAe5F,KAAAA,GAC1C,OAAO,CAACyF,MAAMC,IAAI;GA+BpB,OAAO,CAACD,MAAMI;IA5BZ,GAAGH;IACH,GAAIC,YAAY3F,KAAAA,KAAa,EAC3B2F,UAAU,GAAG,CAACG,MAAMC,aAAyC;KAC3D,MAAM3G,cAAckB,mBAAmByF,OAAO;KAC9C,MAAME,QAAQ,KAAKC,kBACjBH,QAAQpH,YACRS,WACF;KACA,IAAI,CAAC6G,SAASA,MAAM5G,aAClB,OAAO,IAAIoF,cAAc,CAAC,CAAC;KAE7B,OAAOkB,QAAQG,MAAMC,OAAO;IAC9B,EACF;IACA,GAAIH,eAAe5F,KAAAA,KAAa,EAC9B4F,aACE,GAAG,CAACO,QAAQJ,aACT;KACH,MAAM3G,cAAckB,mBAAmByF,OAAO;KAK9C,IAAI,CAJU,KAAKG,kBACjBH,QAAQpH,YACRS,WAEG6G,GAAO;KACZ,OAAOL,WAAWO,QAAQJ,OAAO;IACnC,EACF;GAEYF,CAAW;EAC3B,CAAC,CACH;CACF;CAEA,kBACElH,YACAS,aAC2B;EAC3B,IAAIA,gBAAgBY,KAAAA,GAAW,OAAOA,KAAAA;EACtC,MAAMiG,QAAQ,KAAKnF,SAASiE,IAAIpG,UAAU;EAC1C,IAAI,CAACsH,OAAO3G,YAAY,OAAOU,KAAAA;EAC/B,IAAIiG,MAAM7G,gBAAgBY,KAAAA,GAAWiG,MAAM7G,cAAcA;EACzD,OAAO6G,MAAM7G,gBAAgBA,cAAc6G,QAAQjG,KAAAA;CACrD;CAIA,cACErB,YACAF,SACAW,aACkB;EAClB,OAAO,IAAIqF,SAAkB9E,SAASsB,WAAW;GAG/C,MAAMgF,QAAQ,KAAKnF,SAASiE,IAAIpG,UAAU;GAC1C,IAAI,CAACsH,OAAO3G,cAAc2G,MAAM7G,gBAAgBA,aAAa;IAC3D6B,uBAAO,IAAI0D,MAAM,wBAAwB,CAAC;IAC1C;GACF;GACA,MAAMzE,WAAW,KAAKc,YAAY+D,IAAIpG,UAAU;GAChD,IAAIuB,UACF,IAAI;IACFA,SAASe,uBACP,IAAI0D,MAAM,qDAAqD,CACjE;GACF,QAAQ,CACN;GAGJ,KAAK3D,YAAYoF,IAAIzH,YAAY;IAClBS;IACbO;IACAsB;GACF,CAAC;GACD,KAAKgE,WAAWtG,YAAY;IAC1BH,MAAM;IACNC,SAAS;KAAED,MAAM;KAASC;IAAQ;GACpC,CAAC;EACH,CAAC;CACH;CAEA,kBACEE,YACAS,aACM;EACN,IAAI,CAAC,KAAK8G,kBAAkBvH,YAAYS,WAAW,GAAG;EAEtD,IADc,KAAK0B,SAASiE,IAAIpG,UAC5BsH,CAAK,CAAC5G,aAAa;EAEvB,KAAK8B,WAAWkF,IAAIjH,WAAY;EAChC,KAAK6F,WAAWtG,YAAY,EAAEH,MAAM,YAAY,CAAC;CACnD;CAEA,gBACEG,YACAS,aACM;EACN,IAAIA,gBAAgBY,KAAAA,KAAa,CAAC,KAAKmB,WAAW6D,OAAO5F,WAAW,GAClE;EAGF,IADc,KAAK0B,SAASiE,IAAIpG,UAC5BsH,CAAK,EAAE7G,gBAAgBA,aAAa,KAAKkH,cAAc3H,UAAU;EAErE,MAAM4H,UAA6B,CAAA;EACnC,KAAKlF,kBAAkBqD,SAAS,EAAEjF,cAAcE,cAAc;GAC5D,IAAI,CAAC,GAAGF,YAAY,CAAC,CAAC+G,MAAM5D,OAAO,KAAKzB,WAAWsF,IAAI7D,EAAE,CAAC,GAAG;IAC3D2D,QAAQ3B,KAAK;KAAEnF;KAAcE;IAAQ,CAAC;IACtC;GACF;GACA,IAAI;IACFA,QAAQ;GACV,QAAQ,CACN;EAEJ,CAAC;EACD,KAAK0B,kBAAkBqF,SAAS;EAChC,KAAKrF,kBAAkBuD,KAAK,GAAG2B,OAAO;CACxC;CAEA,mBAA2BlD,OAOlB;EACP,MAAM1E,aAAa0E,MAAMsD,KAAKhI;EAC9B,MAAMS,cAAckB,mBAAmB+C,KAAK;EAC5C,MAAM4C,QAAQ,KAAKnF,SAASiE,IAAIpG,UAAU;EAE1C,IAAI,CAACsH,SAASA,MAAM7G,gBAAgBA,aAAa;EAIjD,IAAI6G,OAAO9G,WAAW;EAEtB,KAAKyH,gBAAgB;GACnBpI,MAAM;GACNG;GACAC,UAAUyE,MAAMsD,KAAK/H;GACrBC,QAAQwE,MAAMxE;GACdC,SAASuE,MAAMvE;GACf,GAAIuE,MAAMtE,aAAaiB,KAAAA,KAAa,EAAEjB,UAAUsE,MAAMtE,SAAS;GAC/D,GAAIsE,MAAMrE,iBAAiBgB,KAAAA,KAAa,EACtChB,cAAcqE,MAAMrE,aACtB;EACF,CAAC;CACH;CAIA,gBAAwB6H,SAAqC;EAC3D,IAAI;GACF,KAAKjG,WAAWkG,SAASD,OAAO;EAClC,SAAStD,KAAK;GACZC,QAAQC,MACN,mEACAF,GACF;EACF;CACF;CAEA,0BAAwC;EACtC,IAAI;GACF,KAAK3C,WAAWmG,iBAAiB,KAAKzF,SAAS;EACjD,SAASiC,KAAK;GACZC,QAAQC,MACN,yFACAF,GACF;EACF;CACF;CAIA,WAAmB5E,YAAoBqI,QAAmC;EACxE,MAAM7G,OAAO,IAAIY,IAAI,KAAKO,SAAS;EACnCnB,KAAKiG,IAAIzH,YAAYqI,MAAM;EAC3B,KAAK1F,YAAYnB;EACjB,KAAKqE,wBAAwB;CAC/B;CAEA,cAAsB7F,YAA0B;EAC9C,IAAI,CAAC,KAAK2C,UAAUmF,IAAI9H,UAAU,GAAG;EACrC,MAAMwB,OAAO,IAAIY,IAAI,KAAKO,SAAS;EACnCnB,KAAK6E,OAAOrG,UAAU;EACtB,KAAK2C,YAAYnB;EACjB,KAAKqE,wBAAwB;CAC/B;CAIA,mBAA2B5F,UAA2B;EACpD,MAAM8G,OAAO,KAAK/E,UAAU,CAAC,GAAG/B;EAChC,OAAO8G,MAAMC,YAAY3F,KAAAA,KAAa0F,MAAME,eAAe5F,KAAAA;CAC7D;CAEA,uBAA+B,EAC7BpB,UACAM,UACAC,aAKU;EACV,IAAIA,WAAW,OAAO;EACtB,IAAI,CAAC,KAAK8H,mBAAmBrI,QAAQ,GACnC,OAAO,CAAC,KAAKgD,cAAchC,mBAAmBV,QAAQ;EAExD,OAAOU,mBAAmBV,QAAQ;CACpC;CAEA,kBACEP,YACAC,UACAS,aACe;EAKf,MAAM4G,QAAuB;GAC3BrH;GACAU,YANyB,KAAKuC,YAAYwF,gBAAgB;IAC1DzI;IACAD;GACF,CAGcyI;GACZlI,UAAU;GACVC,WAAW;GACXE;GACAE,cAAc;EAChB;EACA,KAAKuB,SAASsF,IAAIzH,YAAYsH,KAAK;EACnC,OAAOA;CACT;;;;;;;;CASA,2BAAyC;EACvC,KAAK,MAAM,CAACtH,YAAYsH,UAAU,KAAKnF,UAAU;GAC/C,IAAI,CAACmF,MAAM3G,YAAY;GACvB,KAAKwB,SAASsF,IAAIzH,YAAY;IAC5BC,UAAUqH,MAAMrH;IAChBM,UAAU+G,MAAM/G;IAChBC,WAAW8G,MAAM9G;GACnB,CAAC;EACH;CACF;CAEA,iBACE8G,OACAsB,SAMM;EACN,IAAI,CAACtB,MAAM3G,YAAY;EACvB,MAAMH,YAAYoI,QAAQ1I,WAAWmB,KAAAA;EAErC,IAAIuH,QAAQrI,aAAa+G,MAAM/G,UAAU;GACvC,IAAIsI,sBAAsB;GAE1B,IAAIvB,MAAM1G,cAAc;IACtB,IAAIU,6BAA6BgG,MAAM/G,UAAUqI,QAAQrI,QAAQ,GAAG;KAElE+G,MAAM/G,WAAWqI,QAAQrI;KACzBsI,sBAAsB;IACxB,OAAO;KAOL,IAAIlK,QAAQC,IAAIC,aAAa,cAC3BgG,QAAQiE,KACN,qHACA;MACEvH,UAAU+F,MAAM/G;MAChBiB,MAAMoH,QAAQrI;MACdP,YAAY4I,QAAQ5I;KACtB,CACF;KAEF6I,sBAAsB;IACxB;GACF,OAAO,IAAI,CAACD,QAAQrI,SAASwI,WAAWzB,MAAM/G,QAAQ,GAAG;IACvD,IACEU,mBAAmBqG,MAAM/G,QAAQ,KACjCU,mBAAmB2H,QAAQrI,QAAQ,KACnCe,6BAA6BgG,MAAM/G,UAAUqI,QAAQrI,QAAQ,GAC7D;KACA,MAAMyI,cAAc,KAAKT,uBAAuB;MAC9CtI,UAAU2I,QAAQ3I;MAClBM,UAAUqI,QAAQrI;MAClBC;KACF,CAAC;KACD,IAAIwI,aAAa1B,MAAM3G,WAAWJ,SAAS0I,MAAM;KACjD3B,MAAM/G,WAAWqI,QAAQrI;KACzB+G,MAAM1G,eAAeoI;KACrBH,sBAAsB;IACxB,OAAO;KAQL,IAAIlK,QAAQC,IAAIC,aAAa,cAC3BgG,QAAQiE,KACN,iGACA;MACEvH,UAAU+F,MAAM/G;MAChBiB,MAAMoH,QAAQrI;MACdP,YAAY4I,QAAQ5I;KACtB,CACF;KAEF6I,sBAAsB;IACxB;GACF;GAEA,IAAIA,uBAAuBvB,MAAM3G,YAAY;IAC3C,MAAMuI,QAAQN,QAAQrI,SAAS4I,MAAM7B,MAAM/G,SAASwH,MAAM;IAC1DT,MAAM3G,WAAWJ,SAAS6I,OAAOF,KAAK;IACtC,MAAMF,cAAc,KAAKT,uBAAuB;KAC9CtI,UAAU2I,QAAQ3I;KAClBM,UAAUqI,QAAQrI;KAClBC;IACF,CAAC;IACD,IAAIwI,aAAa1B,MAAM3G,WAAWJ,SAAS0I,MAAM;IACjD3B,MAAM/G,WAAWqI,QAAQrI;IACzB+G,MAAM1G,eAAeoI;GACvB;EACF;EAEA,IAAI,CAAC1B,MAAM1G,gBAAgB0G,MAAM3G,YAIX;OAAA,KAAK4H,uBAAuB;IAC9CtI,UAAU2I,QAAQ3I;IAClBM,UAAU+G,MAAM/G;IAChBC;GACF,CACIwI,GAAa;IACf1B,MAAM3G,WAAWJ,SAAS0I,MAAM;IAChC3B,MAAM1G,eAAe;GACvB;;CAEJ;CAEA,iBAAyBwE,UAA0C;EACjE,MAAMiE,YAAY,KAAKvG;EAEvB,KAAK,MAAMwG,WAAWlE,UAAU;GAC9B,IAAI,CAACkE,WAAW,CAACC,MAAMC,QAASF,QAA0BV,OAAO,GAC/D;GAEF,KAAK,MAAMA,WAAWU,QAAQV,SAAwD;IACpF,IAAI,CAACA,WAAWA,QAAQ/I,SAAS,aAAa;IAE9C,MAAM4J,WAAW,KAAKtH,SAASiE,IAAIwC,QAAQ5I,UAAU;IAErD,IAAIqJ,WAAW;KAIb,IAAI,CAACI,UAAU9I,YACb,KAAKwB,SAASsF,IAAImB,QAAQ5I,YAAY;MACpCC,UAAU2I,QAAQ3I;MAClBM,UAAUqI,QAAQrI;MAClBC,WAAWoI,QAAQ1I,WAAWmB,KAAAA;KAChC,CAAC;KAEH,IAAIuH,QAAQxD,UAAU,KAAKK,iBAAiBmD,QAAQxD,QAAQ;KAC5D;IACF;IAGA,IAAIkC,QAAQmC;IAEZ,IAAInC,SAAS,CAACA,MAAM3G,YAAY;KAM9B,IAAI,EAFFiI,QAAQrI,aAAa+G,MAAM/G,YAC1BqI,QAAQ1I,WAAWmB,KAAAA,MAAeiG,MAAM9G,YACpB;MACrB,IAAIoI,QAAQxD,UAAU,KAAKK,iBAAiBmD,QAAQxD,QAAQ;MAC5D;KACF;KACA,KAAKjD,SAASkE,OAAOuC,QAAQ5I,UAAU;KACvCsH,QAAQjG,KAAAA;IACV;IAEA,IAAI,CAACiG,OACHA,QAAQ,KAAKkB,kBACXI,QAAQ5I,YACR4I,QAAQ3I,UACR2I,QAAQ1I,WAAWmB,KAAAA,CACrB;IAGF,KAAKsH,iBAAiBrB,OAAOsB,OAAO;IAEpC,IAAIA,QAAQ1I,WAAWmB,KAAAA,KAAa,CAACiG,MAAM9G,WAAW;KAIpD,MAAM,EAAEG,YAAYgJ,qBAAqBrC;KACzC,IAAI,CAACqC,kBAAkB;KACvBrC,MAAM9G,YAAY;KAClB8G,MAAM1G,eAAe;KACrB+I,iBAAiBC,YACf,IAAI5K,aAAa;MACfkB,QAAQ0I,QAAQ1I;MAChBE,UAAUwI,QAAQxI;MAClBD,SAASyI,QAAQzI;MACjB,GAAIyI,QAAQvI,iBAAiBgB,KAAAA,IACzB,EAAEhB,cAAcuI,QAAQvI,aAAa,IACrC,CAAC;KACP,CAAC,CACH;KACAsJ,iBAAiBV,MAAM;IACzB;IAEA,IAAIL,QAAQxD,UAAU,KAAKK,iBAAiBmD,QAAQxD,QAAQ;GAC9D;EACF;CACF;AACF"}
+{"version":3,"file":"ToolInvocationTracker.js","names":["process","env","NODE_ENV","createAssistantStreamController","ToolCallStreamController","ToolResponse","unstable_toolResultStream","Tool","ToolModelContentPart","ToolResultStreamOptions","AssistantMetaTransformStream","ReadonlyJSONValue","isJSONValueEqual","ThreadMessage","TOOL_EXECUTION_ID","Symbol","for","ToolExecutionStatus","type","payload","AddToolResultCommand","toolCallId","toolName","result","isError","artifact","modelContent","ToolCallEntry","argsText","hasResult","executionId","skipExecute","controller","argsComplete","SettledResolver","executionIds","ReadonlySet","resolve","isArgsTextComplete","JSON","parse","parseArgsText","undefined","isEquivalentCompleteArgsText","previous","next","previousValue","nextValue","getToolExecutionId","value","Record","PropertyKey","ToolInvocationTracker","_getTools","_callbacks","Callbacks","_entries","Map","_humanInput","reject","reason","_executing","Set","_settledResolvers","_statuses","_ac","AbortController","_pendingRestore","_lastSnapshot","Snapshot","_isRunning","_controller","ReturnType","_pipelineDead","_pipelineRestartUsed","constructor","getTools","callbacks","_initPipeline","stream","human","_onHumanInput","transform","_getWrappedTools","signal","onExecutionStart","id","_name","_onExecutionStart","onExecutionEnd","_onExecutionEnd","pipeThrough","pipeTo","WritableStream","write","chunk","_handleResultChunk","err","console","error","catch","setState","snapshot","_demoteEntriesToRestored","clear","messages","isRunning","isLoading","restoreFromLoading","previousIsRunning","_processMessages","reset","abort","size","_invokeOnStatusesChange","Promise","forEach","Error","push","resume","handlers","get","delete","_setStatus","getStatuses","ReadonlyMap","tools","Object","fromEntries","entries","map","name","tool","execute","streamCall","wrappedTool","args","context","Parameters","entry","_captureExecution","reader","set","add","_deleteStatus","pending","some","has","length","meta","_invokeOnResult","command","onResult","onStatusesChange","status","_hasExecutableTool","_shouldCloseArgsStream","_startActiveEntry","toolCallController","addToolCallPart","_processArgsText","content","shouldWriteArgsText","warn","startsWith","shouldClose","close","delta","slice","append","isRestore","message","Array","isArray","existing","signatureChanged","approval","activeController","setResponse","ExecutionStatus","statuses"],"sources":["../../../src/runtimes/tool-invocations/ToolInvocationTracker.ts"],"sourcesContent":["declare const process: { env: { NODE_ENV?: string } };\n\nimport {\n  createAssistantStreamController,\n  type ToolCallStreamController,\n  ToolResponse,\n  unstable_toolResultStream,\n  type Tool,\n  type ToolModelContentPart,\n  type ToolResultStreamOptions,\n} from \"assistant-stream\";\nimport {\n  AssistantMetaTransformStream,\n  type ReadonlyJSONValue,\n} from \"assistant-stream/utils\";\nimport { isJSONValueEqual } from \"../../utils/json/is-json-equal\";\nimport type { ThreadMessage } from \"../../types/message\";\n\nconst TOOL_EXECUTION_ID = Symbol.for(\"assistant-stream.tool-execution-id\");\n\n/**\n * Streaming execution state for a frontend tool.\n */\nexport type ToolExecutionStatus =\n  | { type: \"executing\" }\n  | {\n      type: \"interrupt\";\n      payload: { type: \"human\"; payload: unknown };\n    };\n\nexport type AddToolResultCommand = {\n  readonly type: \"add-tool-result\";\n  readonly toolCallId: string;\n  readonly toolName: string;\n  readonly result: ReadonlyJSONValue;\n  readonly isError: boolean;\n  readonly artifact?: ReadonlyJSONValue;\n  readonly modelContent?: readonly ToolModelContentPart[];\n};\n\ntype ToolCallEntry = {\n  toolName: string;\n  argsText: string;\n  hasResult: boolean;\n  executionId?: symbol;\n  skipExecute?: boolean;\n} & (\n  | {\n      /** Restored phase — observed during a history-load snapshot. */\n      controller?: undefined;\n      argsComplete?: undefined;\n    }\n  | {\n      /** Active phase — chunks are flowing through `controller`. */\n      controller: ToolCallStreamController;\n      argsComplete: boolean;\n    }\n);\n\ntype SettledResolver = {\n  executionIds: ReadonlySet<symbol>;\n  resolve: () => void;\n};\n\nconst isArgsTextComplete = (argsText: string) => {\n  try {\n    JSON.parse(argsText);\n    return true;\n  } catch {\n    return false;\n  }\n};\n\nconst parseArgsText = (argsText: string) => {\n  try {\n    return JSON.parse(argsText);\n  } catch {\n    return undefined;\n  }\n};\n\nconst isEquivalentCompleteArgsText = (previous: string, next: string) => {\n  const previousValue = parseArgsText(previous);\n  const nextValue = parseArgsText(next);\n  if (previousValue === undefined || nextValue === undefined) return false;\n  return isJSONValueEqual(previousValue, nextValue);\n};\n\nconst getToolExecutionId = (value: object): symbol | undefined =>\n  (value as Record<PropertyKey, unknown>)[TOOL_EXECUTION_ID] as\n    | symbol\n    | undefined;\n\n/**\n * Plain-class port of the former `useToolInvocations` React hook. Owns the\n * assistant-stream pipeline that drives client-side `streamCall` / `execute`\n * for tool-call parts surfaced by a thread runtime, plus the per-tool-call\n * status map that consumers render against.\n *\n * **Contract**: `streamCall` (and `execute`) fires exactly once per logical\n * `toolCallId`. Args mutations after first completion, result replacement,\n * and result clearing are *not* surfaced through additional `streamCall`\n * invocations — by design — so hosts cannot observe spurious re-fires of\n * side effects. The follow-up `reader.events()` API will expose those\n * post-completion transitions to consumers that opt in.\n *\n * State-transition safety: every public method that observes runtime state\n * (`setState`, `reset`, `abort`, `resume`) wraps its work in try/catch and\n * logs to `console.error` rather than throwing. The tracker is built into\n * the hot message-processing path, so a malformed snapshot must never crash\n * the host runtime. See ./EDGE_CASES.md for the known non-trivial state\n * transitions and what each does today.\n *\n * @deprecated Internal — for framework bindings; not a public API. May change without notice.\n */\nexport class ToolInvocationTracker {\n  private readonly _getTools: () => Record<string, Tool> | undefined;\n  private readonly _callbacks: ToolInvocationTracker.Callbacks;\n\n  private readonly _entries = new Map<string, ToolCallEntry>();\n  private readonly _humanInput = new Map<\n    string,\n    {\n      executionId: symbol;\n      resolve: (payload: unknown) => void;\n      reject: (reason: unknown) => void;\n    }\n  >();\n  private readonly _executing = new Set<symbol>();\n  private readonly _settledResolvers: SettledResolver[] = [];\n\n  private _statuses = new Map<string, ToolExecutionStatus>();\n\n  private _ac: AbortController = new AbortController();\n  private _pendingRestore = true;\n\n  /** Cached last snapshot, used to skip processing on identical re-renders. */\n  private _lastSnapshot: ToolInvocationTracker.Snapshot | null = null;\n  private _isRunning = false;\n\n  private _controller!: ReturnType<typeof createAssistantStreamController>[1];\n\n  /**\n   * Set when the assistant-stream pipeline has died (errored out via\n   * `.pipeTo(...).catch(...)`). The next `setState` re-initializes the\n   * pipeline and demotes all active entries to restored so they survive\n   * across the restart without re-firing `streamCall` (preserves the\n   * \"exactly once\" contract). Capped at a single auto-restart per session\n   * — repeated failures keep the tracker dead with a more visible error.\n   */\n  private _pipelineDead = false;\n  private _pipelineRestartUsed = false;\n\n  constructor(\n    getTools: () => Record<string, Tool> | undefined,\n    callbacks: ToolInvocationTracker.Callbacks,\n  ) {\n    this._getTools = getTools;\n    this._callbacks = callbacks;\n\n    this._initPipeline();\n  }\n\n  /**\n   * Build the assistant-stream pipeline. Called once from the constructor\n   * and at most once again if `_pipelineDead` is set (see F.4 in\n   * EDGE_CASES.md).\n   */\n  private _initPipeline(): void {\n    const [stream, controller] = createAssistantStreamController();\n    this._controller = controller;\n\n    const human = (\n      toolCallId: string,\n      payload: unknown,\n      executionId?: symbol,\n    ) => this._onHumanInput(toolCallId, payload, executionId);\n    const transform = unstable_toolResultStream(\n      () => this._getWrappedTools(),\n      () => this._ac.signal,\n      human,\n      {\n        onExecutionStart: (id: string, _name: string, executionId?: symbol) =>\n          this._onExecutionStart(id, executionId),\n        onExecutionEnd: (id: string, _name: string, executionId?: symbol) =>\n          this._onExecutionEnd(id, executionId),\n      } as ToolResultStreamOptions,\n    );\n\n    stream\n      .pipeThrough(transform)\n      .pipeThrough(new AssistantMetaTransformStream())\n      .pipeTo(\n        new WritableStream({\n          write: (chunk) => {\n            try {\n              if (chunk.type !== \"result\") return;\n              this._handleResultChunk(chunk);\n            } catch (err) {\n              console.error(\n                \"[ToolInvocationTracker] result chunk handling failed\",\n                err,\n              );\n            }\n          },\n        }),\n      )\n      .catch((err) => {\n        console.error(\n          \"[ToolInvocationTracker] stream pipeline failed; will attempt single restart on next setState\",\n          err,\n        );\n        this._pipelineDead = true;\n      });\n  }\n\n  // ───────────────────────── public API ─────────────────────────\n\n  /**\n   * Feed the next observed snapshot into the tracker. Called from the host\n   * runtime whenever its message list / running state changes.\n   */\n  public setState(snapshot: ToolInvocationTracker.Snapshot): void {\n    try {\n      // Recover from a dead pipeline before processing anything. We demote\n      // all active entries to \"restored\" so the rebuilt pipeline does not\n      // re-fire `streamCall` for tool calls that already fired pre-death;\n      // preserves the \"exactly once per toolCallId\" contract.\n      if (this._pipelineDead) {\n        if (this._pipelineRestartUsed) {\n          // Already retried once and failed again. Stay dead.\n          return;\n        }\n        this._pipelineRestartUsed = true;\n        this._pipelineDead = false;\n        this._demoteEntriesToRestored();\n        this._executing.clear();\n        this._ac = new AbortController();\n        this._initPipeline();\n        // Fall through and process the snapshot against the fresh pipeline.\n      }\n\n      // Identical snapshot — skip processing entirely. Note: external-store\n      // runtimes rebuild the messages array on every adapter update, so this\n      // fast-path rarely triggers there; it's primarily for the React-hook\n      // shim where state references are stable.\n      if (\n        this._lastSnapshot &&\n        this._lastSnapshot.messages === snapshot.messages &&\n        this._lastSnapshot.isRunning === snapshot.isRunning &&\n        this._lastSnapshot.isLoading === snapshot.isLoading\n      ) {\n        return;\n      }\n\n      // While the host is still loading initial state, treat every snapshot\n      // as historical: tool calls are recorded so the next live snapshot can\n      // diff against them, but `streamCall` / `execute` do not fire.\n      const restoreFromLoading = snapshot.isLoading === true;\n      if (restoreFromLoading) {\n        this._pendingRestore = true;\n      }\n\n      // E.4 / AF3 — only mark `_lastSnapshot`/`_isRunning` as observed after\n      // processing succeeds. If `_processMessages` throws, the next snapshot\n      // (even if identical) gets re-processed against the recovered state.\n      const previousIsRunning = this._isRunning;\n      this._isRunning = snapshot.isRunning;\n      try {\n        this._processMessages(snapshot.messages);\n      } catch (err) {\n        this._isRunning = previousIsRunning;\n        throw err;\n      }\n      this._lastSnapshot = snapshot;\n      this._pendingRestore = false;\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] setState failed; snapshot dropped\",\n        err,\n      );\n    }\n  }\n\n  /**\n   * Reset the tracker so the next observed snapshot is treated as historical.\n   * Clears entries and aborts any in-flight executions. Used by callers like\n   * `importExternalState` to mark a freshly loaded state as restored.\n   */\n  public reset(): void {\n    try {\n      this._pendingRestore = true;\n      this._entries.clear();\n      this._lastSnapshot = null;\n      void this.abort();\n      // Statuses are cleared synchronously: discarded executions may never\n      // settle (aborting hands the signal to the tool, it does not force\n      // settlement), and a late settlement of one sibling must not republish\n      // the others. `_deleteStatus` no-ops for cleared ids, so post-reset\n      // settlements stay silent.\n      if (this._statuses.size > 0) {\n        this._statuses = new Map();\n        this._invokeOnStatusesChange();\n      }\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] reset failed\", err);\n    }\n  }\n\n  /**\n   * Abort any in-flight `execute()` invocations. Resolves once all of them\n   * have settled (or immediately if none are running).\n   */\n  public abort(): Promise<void> {\n    try {\n      this._humanInput.forEach(({ reject }) => {\n        try {\n          reject(new Error(\"Tool execution aborted\"));\n        } catch {\n          // host rejection handler threw — already in the abort path,\n          // swallow so we continue cleaning up.\n        }\n      });\n      this._humanInput.clear();\n\n      this._ac.abort();\n      this._ac = new AbortController();\n\n      if (this._executing.size === 0) {\n        return Promise.resolve();\n      }\n      const executionIds = new Set(this._executing);\n      return new Promise<void>((resolve) => {\n        this._settledResolvers.push({ executionIds, resolve });\n      });\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] abort failed\", err);\n      return Promise.resolve();\n    }\n  }\n\n  /**\n   * Resolve a pending human-input request for the given tool call. Returns\n   * `true` if a pending request was resumed, `false` if the tracker has no\n   * outstanding request for that id (the caller should fall back to its own\n   * dispatch path).\n   */\n  public resume(toolCallId: string, payload: unknown): boolean {\n    try {\n      const handlers = this._humanInput.get(toolCallId);\n      if (!handlers) return false;\n      this._humanInput.delete(toolCallId);\n      this._setStatus(toolCallId, { type: \"executing\" });\n      handlers.resolve(payload);\n      return true;\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] resume failed\", err);\n      return false;\n    }\n  }\n\n  /**\n   * Returns the current tool execution status map. The returned `Map` is\n   * the tracker's internal store — do not mutate it. Treat the reference\n   * as a snapshot that may be replaced wholesale on the next status\n   * transition.\n   */\n  public getStatuses(): ReadonlyMap<string, ToolExecutionStatus> {\n    return this._statuses;\n  }\n\n  // ───────────────────── internal: tool wrapping ─────────────────────\n\n  private _getWrappedTools(): Record<string, Tool> | undefined {\n    const tools = this._getTools();\n    if (!tools) return undefined;\n\n    return Object.fromEntries(\n      Object.entries(tools).map(([name, tool]) => {\n        const execute = tool.execute;\n        const streamCall = tool.streamCall;\n        if (execute === undefined && streamCall === undefined)\n          return [name, tool];\n\n        const wrappedTool = {\n          ...tool,\n          ...(execute !== undefined && {\n            execute: (...[args, context]: Parameters<typeof execute>) => {\n              const executionId = getToolExecutionId(context);\n              const entry = this._captureExecution(\n                context.toolCallId,\n                executionId,\n              );\n              if (!entry || entry.skipExecute) {\n                return new Promise(() => {}) as never;\n              }\n              return execute(args, context);\n            },\n          }),\n          ...(streamCall !== undefined && {\n            streamCall: (\n              ...[reader, context]: Parameters<typeof streamCall>\n            ) => {\n              const executionId = getToolExecutionId(context);\n              const entry = this._captureExecution(\n                context.toolCallId,\n                executionId,\n              );\n              if (!entry) return;\n              return streamCall(reader, context);\n            },\n          }),\n        } as Tool;\n        return [name, wrappedTool];\n      }),\n    ) as Record<string, Tool>;\n  }\n\n  private _captureExecution(\n    toolCallId: string,\n    executionId: symbol | undefined,\n  ): ToolCallEntry | undefined {\n    if (executionId === undefined) return undefined;\n    const entry = this._entries.get(toolCallId);\n    if (!entry?.controller) return undefined;\n    if (entry.executionId === undefined) entry.executionId = executionId;\n    return entry.executionId === executionId ? entry : undefined;\n  }\n\n  // ──────────────── internal: execution lifecycle callbacks ────────────────\n\n  private _onHumanInput(\n    toolCallId: string,\n    payload: unknown,\n    executionId?: symbol,\n  ): Promise<unknown> {\n    return new Promise<unknown>((resolve, reject) => {\n      // A discarded execution must not resurrect a status entry or park an\n      // unanswerable request after its id has been reused.\n      const entry = this._entries.get(toolCallId);\n      if (!entry?.controller || entry.executionId !== executionId) {\n        reject(new Error(\"Tool execution aborted\"));\n        return;\n      }\n      const previous = this._humanInput.get(toolCallId);\n      if (previous) {\n        try {\n          previous.reject(\n            new Error(\"Human input request was superseded by a new request\"),\n          );\n        } catch {\n          // host rejection handler threw; ignore and proceed\n        }\n      }\n      this._humanInput.set(toolCallId, {\n        executionId: executionId!,\n        resolve,\n        reject,\n      });\n      this._setStatus(toolCallId, {\n        type: \"interrupt\",\n        payload: { type: \"human\", payload },\n      });\n    });\n  }\n\n  private _onExecutionStart(\n    toolCallId: string,\n    executionId: symbol | undefined,\n  ): void {\n    if (!this._captureExecution(toolCallId, executionId)) return;\n    const entry = this._entries.get(toolCallId)!;\n    if (entry.skipExecute) return;\n\n    this._executing.add(executionId!);\n    this._setStatus(toolCallId, { type: \"executing\" });\n  }\n\n  private _onExecutionEnd(\n    toolCallId: string,\n    executionId: symbol | undefined,\n  ): void {\n    if (executionId === undefined || !this._executing.delete(executionId))\n      return;\n\n    const entry = this._entries.get(toolCallId);\n    if (entry?.executionId === executionId) this._deleteStatus(toolCallId);\n\n    const pending: SettledResolver[] = [];\n    this._settledResolvers.forEach(({ executionIds, resolve }) => {\n      if ([...executionIds].some((id) => this._executing.has(id))) {\n        pending.push({ executionIds, resolve });\n        return;\n      }\n      try {\n        resolve();\n      } catch {\n        // ignore — settled-resolver consumer threw\n      }\n    });\n    this._settledResolvers.length = 0;\n    this._settledResolvers.push(...pending);\n  }\n\n  private _handleResultChunk(chunk: {\n    type: \"result\";\n    result: ReadonlyJSONValue;\n    isError: boolean;\n    artifact?: ReadonlyJSONValue;\n    modelContent?: readonly ToolModelContentPart[];\n    meta: { toolCallId: string; toolName: string };\n  }): void {\n    const toolCallId = chunk.meta.toolCallId;\n    const executionId = getToolExecutionId(chunk);\n    const entry = this._entries.get(toolCallId);\n\n    if (!entry || entry.executionId !== executionId) return;\n\n    // The host already set the result (via the live snapshot's\n    // `setResponse` path). Suppress the executor's redundant emit.\n    if (entry?.hasResult) return;\n    if (entry.skipExecute) return;\n\n    this._invokeOnResult({\n      type: \"add-tool-result\",\n      toolCallId,\n      toolName: chunk.meta.toolName,\n      result: chunk.result,\n      isError: chunk.isError,\n      ...(chunk.artifact !== undefined && { artifact: chunk.artifact }),\n      ...(chunk.modelContent !== undefined && {\n        modelContent: chunk.modelContent,\n      }),\n    });\n  }\n\n  // ──────────────── internal: callback invocation (AF1/AF2) ────────────────\n\n  private _invokeOnResult(command: AddToolResultCommand): void {\n    try {\n      this._callbacks.onResult(command);\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] onResult callback threw; result dropped\",\n        err,\n      );\n    }\n  }\n\n  private _invokeOnStatusesChange(): void {\n    try {\n      this._callbacks.onStatusesChange(this._statuses);\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] onStatusesChange callback threw; status change not propagated\",\n        err,\n      );\n    }\n  }\n\n  // ──────────────── internal: status map mutations ────────────────\n\n  private _setStatus(toolCallId: string, status: ToolExecutionStatus): void {\n    const next = new Map(this._statuses);\n    next.set(toolCallId, status);\n    this._statuses = next;\n    this._invokeOnStatusesChange();\n  }\n\n  private _deleteStatus(toolCallId: string): void {\n    if (!this._statuses.has(toolCallId)) return;\n    const next = new Map(this._statuses);\n    next.delete(toolCallId);\n    this._statuses = next;\n    this._invokeOnStatusesChange();\n  }\n\n  // ──────────────── internal: snapshot processing ────────────────\n\n  private _hasExecutableTool(toolName: string): boolean {\n    const tool = this._getTools()?.[toolName];\n    return tool?.execute !== undefined || tool?.streamCall !== undefined;\n  }\n\n  private _shouldCloseArgsStream({\n    toolName,\n    argsText,\n    hasResult,\n  }: {\n    toolName: string;\n    argsText: string;\n    hasResult: boolean;\n  }): boolean {\n    if (hasResult) return true;\n    if (!this._hasExecutableTool(toolName)) {\n      return !this._isRunning && isArgsTextComplete(argsText);\n    }\n    return isArgsTextComplete(argsText);\n  }\n\n  private _startActiveEntry(\n    toolCallId: string,\n    toolName: string,\n    skipExecute: boolean,\n  ): ToolCallEntry {\n    const toolCallController = this._controller.addToolCallPart({\n      toolName,\n      toolCallId,\n    });\n    const entry: ToolCallEntry = {\n      toolName,\n      controller: toolCallController,\n      argsText: \"\",\n      hasResult: false,\n      skipExecute,\n      argsComplete: false,\n    };\n    this._entries.set(toolCallId, entry);\n    return entry;\n  }\n\n  /**\n   * Demote every active entry back to the restored phase. Used by the\n   * pipeline-restart path so that, after a fresh pipeline is built, the\n   * next observed snapshot does not re-fire `streamCall` for tool calls\n   * that already fired pre-death. Args / hasResult tracking is preserved\n   * so signature comparisons still work.\n   */\n  private _demoteEntriesToRestored(): void {\n    for (const [toolCallId, entry] of this._entries) {\n      if (!entry.controller) continue;\n      this._entries.set(toolCallId, {\n        toolName: entry.toolName,\n        argsText: entry.argsText,\n        hasResult: entry.hasResult,\n      });\n    }\n  }\n\n  private _processArgsText(\n    entry: ToolCallEntry,\n    content: {\n      toolCallId: string;\n      toolName: string;\n      argsText: string;\n      result?: unknown;\n    },\n  ): void {\n    if (!entry.controller) return;\n    const hasResult = content.result !== undefined;\n\n    if (content.argsText !== entry.argsText) {\n      let shouldWriteArgsText = true;\n\n      if (entry.argsComplete) {\n        if (isEquivalentCompleteArgsText(entry.argsText, content.argsText)) {\n          // A.3 — key reorder. Track new text, no re-fire needed.\n          entry.argsText = content.argsText;\n          shouldWriteArgsText = false;\n        } else {\n          // A.4 — args changed after first completion. Under the\n          // \"exactly once per toolCallId\" contract we do not restart the\n          // stream. The host's existing `streamCall` keeps its original\n          // args view; the snapshot's new text is recorded for diffing\n          // but not surfaced. Events API in a follow-up will expose this\n          // to consumers that opt in.\n          if (process.env.NODE_ENV !== \"production\") {\n            console.warn(\n              \"[ToolInvocationTracker] argsText changed after first completion; not re-firing streamCall (see EDGE_CASES.md A.4)\",\n              {\n                previous: entry.argsText,\n                next: content.argsText,\n                toolCallId: content.toolCallId,\n              },\n            );\n          }\n          shouldWriteArgsText = false;\n        }\n      } else if (!content.argsText.startsWith(entry.argsText)) {\n        if (\n          isArgsTextComplete(entry.argsText) &&\n          isArgsTextComplete(content.argsText) &&\n          isEquivalentCompleteArgsText(entry.argsText, content.argsText)\n        ) {\n          const shouldClose = this._shouldCloseArgsStream({\n            toolName: content.toolName,\n            argsText: content.argsText,\n            hasResult,\n          });\n          if (shouldClose) entry.controller.argsText.close();\n          entry.argsText = content.argsText;\n          entry.argsComplete = shouldClose;\n          shouldWriteArgsText = false;\n        } else {\n          // A.2 — args regressed mid-stream. Under the \"exactly once\"\n          // contract we do not restart. The controller keeps whatever\n          // prefix we already streamed; subsequent prefix-respecting\n          // updates can still flow against it. Snapshots that never\n          // re-converge to a prefix will leave the controller's args\n          // view stale relative to the snapshot. Events API in a\n          // follow-up will expose this to consumers that opt in.\n          if (process.env.NODE_ENV !== \"production\") {\n            console.warn(\n              \"[ToolInvocationTracker] argsText regressed mid-stream; not restarting (see EDGE_CASES.md A.2)\",\n              {\n                previous: entry.argsText,\n                next: content.argsText,\n                toolCallId: content.toolCallId,\n              },\n            );\n          }\n          shouldWriteArgsText = false;\n        }\n      }\n\n      if (shouldWriteArgsText && entry.controller) {\n        const delta = content.argsText.slice(entry.argsText.length);\n        entry.controller.argsText.append(delta);\n        const shouldClose = this._shouldCloseArgsStream({\n          toolName: content.toolName,\n          argsText: content.argsText,\n          hasResult,\n        });\n        if (shouldClose) entry.controller.argsText.close();\n        entry.argsText = content.argsText;\n        entry.argsComplete = shouldClose;\n      }\n    }\n\n    if (!entry.argsComplete && entry.controller) {\n      // ToolExecutionStream parses the streamed prefix on close, so the close\n      // gates on the streamed content; a divergent snapshot (A.2) can be\n      // complete while the controller still holds an incomplete stale prefix.\n      const shouldClose = this._shouldCloseArgsStream({\n        toolName: content.toolName,\n        argsText: entry.argsText,\n        hasResult,\n      });\n      if (shouldClose) {\n        entry.controller.argsText.close();\n        entry.argsComplete = true;\n      }\n    }\n  }\n\n  private _processMessages(messages: readonly ThreadMessage[]): void {\n    const isRestore = this._pendingRestore;\n\n    for (const message of messages) {\n      if (!message || !Array.isArray((message as ThreadMessage).content)) {\n        continue;\n      }\n      for (const content of message.content as readonly ThreadMessage[\"content\"][number][]) {\n        if (!content || content.type !== \"tool-call\") continue;\n\n        const existing = this._entries.get(content.toolCallId);\n\n        if (isRestore) {\n          // Don't overwrite an already-active entry (e.g. live tool-call\n          // observed before this restore snapshot landed). Restore can\n          // only seed entries the runtime has never seen.\n          if (!existing?.controller) {\n            this._entries.set(content.toolCallId, {\n              toolName: content.toolName,\n              argsText: content.argsText,\n              hasResult: content.result !== undefined,\n            });\n          }\n          if (content.messages) this._processMessages(content.messages);\n          continue;\n        }\n\n        // Live snapshot.\n        let entry = existing;\n\n        if (entry && !entry.controller) {\n          // Restored entry observed in a live snapshot. Promote if its\n          // signature has changed; otherwise treat as still-historical.\n          const signatureChanged =\n            content.argsText !== entry.argsText ||\n            (content.result !== undefined) !== entry.hasResult;\n          if (!signatureChanged) {\n            if (content.messages) this._processMessages(content.messages);\n            continue;\n          }\n          this._entries.delete(content.toolCallId);\n          entry = undefined;\n        }\n\n        if (!entry) {\n          entry = this._startActiveEntry(\n            content.toolCallId,\n            content.toolName,\n            content.result !== undefined,\n          );\n        }\n\n        if (content.approval !== undefined) entry.skipExecute = true;\n\n        this._processArgsText(entry, content);\n\n        if (content.result !== undefined && !entry.hasResult) {\n          // `entry` is in active phase from this point — either just\n          // created by `_startActiveEntry`, or pre-existing with a live\n          // controller. Narrow once instead of asserting at every use.\n          const { controller: activeController } = entry;\n          if (!activeController) continue;\n          entry.hasResult = true;\n          entry.argsComplete = true;\n          activeController.setResponse(\n            new ToolResponse({\n              result: content.result as ReadonlyJSONValue,\n              artifact: content.artifact as ReadonlyJSONValue | undefined,\n              isError: content.isError,\n              ...(content.modelContent !== undefined\n                ? { modelContent: content.modelContent }\n                : {}),\n            }),\n          );\n          activeController.close();\n        }\n\n        if (content.messages) this._processMessages(content.messages);\n      }\n    }\n  }\n}\n\nexport namespace ToolInvocationTracker {\n  export type ExecutionStatus = ToolExecutionStatus;\n\n  export type Snapshot = {\n    readonly messages: readonly ThreadMessage[];\n    /** Whether the producing runtime is currently streaming new output. */\n    readonly isRunning: boolean;\n    /**\n     * Whether the producing runtime is still loading historical state.\n     * When `true`, every snapshot is treated as historical (no `streamCall` /\n     * `execute` fires). When `false`, processing resumes as live.\n     */\n    readonly isLoading?: boolean;\n  };\n\n  export type Callbacks = {\n    /**\n     * Invoked when a client-side `execute()` returns a result and the runtime\n     * needs to feed it back into the conversation.\n     */\n    onResult: (command: AddToolResultCommand) => void;\n    /**\n     * Invoked whenever the per-tool-call status map changes (executing /\n     * interrupt / cleared). The callback receives a fresh map; mutating the\n     * argument is not supported.\n     */\n    onStatusesChange: (\n      statuses: ReadonlyMap<string, ToolExecutionStatus>,\n    ) => void;\n  };\n}\n"],"mappings":";;;;AAkBA,MAAMc,oBAAoBC,OAAOC,IAAI,oCAAoC;AA8CzE,MAAMsB,sBAAsBV,aAAqB;CAC/C,IAAI;EACFW,KAAKC,MAAMZ,QAAQ;EACnB,OAAO;CACT,QAAQ;EACN,OAAO;CACT;AACF;AAEA,MAAMa,iBAAiBb,aAAqB;CAC1C,IAAI;EACF,OAAOW,KAAKC,MAAMZ,QAAQ;CAC5B,QAAQ;EACN;CACF;AACF;AAEA,MAAMe,gCAAgCC,UAAkBC,SAAiB;CACvE,MAAMC,gBAAgBL,cAAcG,QAAQ;CAC5C,MAAMG,YAAYN,cAAcI,IAAI;CACpC,IAAIC,kBAAkBJ,KAAAA,KAAaK,cAAcL,KAAAA,GAAW,OAAO;CACnE,OAAO9B,iBAAiBkC,eAAeC,SAAS;AAClD;AAEA,MAAMC,sBAAsBC,UACzBA,MAAuCnC;;;;;;;;;;;;;;;;;;;;;;;AA0B1C,IAAasC,wBAAb,MAAmC;CACjC;CACA;CAEA,2BAA4B,IAAIK,IAA2B;CAC3D,8BAA+B,IAAIA,IAOjC;CACF,6BAA8B,IAAIK,IAAY;CAC9C,oBAAwD,CAAA;CAExD,4BAAoB,IAAIL,IAAiC;CAEzD,MAA+B,IAAIS,gBAAgB;CACnD,kBAA0B;;CAG1B,gBAA+D;CAC/D,aAAqB;CAErB;;;;;;;;;CAUA,gBAAwB;CACxB,uBAA+B;CAE/BS,YACEC,UACAC,WACA;EACA,KAAKxB,YAAYuB;EACjB,KAAKtB,aAAauB;EAElB,KAAKC,cAAc;CACrB;;;;;;CAOA,gBAA8B;EAC5B,MAAM,CAACC,QAAQ/C,cAAc7B,gCAAgC;EAC7D,KAAKoE,cAAcvC;EAEnB,MAAMgD,SACJ3D,YACAF,SACAW,gBACG,KAAKmD,cAAc5D,YAAYF,SAASW,WAAW;EACxD,MAAMoD,YAAY5E,gCACV,KAAK6E,iBAAiB,SACtB,KAAKlB,IAAImB,QACfJ,OACA;GACEK,mBAAmBC,IAAYC,OAAezD,gBAC5C,KAAK0D,kBAAkBF,IAAIxD,WAAW;GACxC2D,iBAAiBH,IAAYC,OAAezD,gBAC1C,KAAK4D,gBAAgBJ,IAAIxD,WAAW;EACxC,CACF;EAEAiD,OACGY,YAAYT,SAAS,CAAC,CACtBS,YAAY,IAAIjF,6BAA6B,CAAC,CAAC,CAC/CkF,OACC,IAAIC,eAAe,EACjBC,QAAQC,UAAU;GAChB,IAAI;IACF,IAAIA,MAAM7E,SAAS,UAAU;IAC7B,KAAK8E,mBAAmBD,KAAK;GAC/B,SAASE,KAAK;IACZC,QAAQC,MACN,wDACAF,GACF;GACF;EACF,EACF,CAAC,CACH,CAAC,CACAG,OAAOH,QAAQ;GACdC,QAAQC,MACN,gGACAF,GACF;GACA,KAAKxB,gBAAgB;EACvB,CAAC;CACL;;;;;CAQA,SAAgB6B,UAAgD;EAC9D,IAAI;GAKF,IAAI,KAAK7B,eAAe;IACtB,IAAI,KAAKC,sBAEP;IAEF,KAAKA,uBAAuB;IAC5B,KAAKD,gBAAgB;IACrB,KAAK8B,yBAAyB;IAC9B,KAAK1C,WAAW2C,MAAM;IACtB,KAAKvC,MAAM,IAAIC,gBAAgB;IAC/B,KAAKY,cAAc;GAErB;GAMA,IACE,KAAKV,iBACL,KAAKA,cAAcqC,aAAaH,SAASG,YACzC,KAAKrC,cAAcsC,cAAcJ,SAASI,aAC1C,KAAKtC,cAAcuC,cAAcL,SAASK,WAE1C;GAOF,IAD2BL,SAASK,cAAc,MAEhD,KAAKxC,kBAAkB;GAMzB,MAAM0C,oBAAoB,KAAKvC;GAC/B,KAAKA,aAAagC,SAASI;GAC3B,IAAI;IACF,KAAKI,iBAAiBR,SAASG,QAAQ;GACzC,SAASR,KAAK;IACZ,KAAK3B,aAAauC;IAClB,MAAMZ;GACR;GACA,KAAK7B,gBAAgBkC;GACrB,KAAKnC,kBAAkB;EACzB,SAAS8B,KAAK;GACZC,QAAQC,MACN,6DACAF,GACF;EACF;CACF;;;;;;CAOA,QAAqB;EACnB,IAAI;GACF,KAAK9B,kBAAkB;GACvB,KAAKX,SAASgD,MAAM;GACpB,KAAKpC,gBAAgB;GACrB,KAAU4C,MAAM;GAMhB,IAAI,KAAKhD,UAAUiD,OAAO,GAAG;IAC3B,KAAKjD,4BAAY,IAAIP,IAAI;IACzB,KAAKyD,wBAAwB;GAC/B;EACF,SAASjB,KAAK;GACZC,QAAQC,MAAM,wCAAwCF,GAAG;EAC3D;CACF;;;;;CAMA,QAA8B;EAC5B,IAAI;GACF,KAAKvC,YAAY0D,SAAS,EAAEzD,aAAa;IACvC,IAAI;KACFA,uBAAO,IAAI0D,MAAM,wBAAwB,CAAC;IAC5C,QAAQ,CAEN;GAEJ,CAAC;GACD,KAAK3D,YAAY8C,MAAM;GAEvB,KAAKvC,IAAI+C,MAAM;GACf,KAAK/C,MAAM,IAAIC,gBAAgB;GAE/B,IAAI,KAAKL,WAAWoD,SAAS,GAC3B,OAAOE,QAAQ9E,QAAQ;GAEzB,MAAMF,eAAe,IAAI2B,IAAI,KAAKD,UAAU;GAC5C,OAAO,IAAIsD,SAAe9E,YAAY;IACpC,KAAK0B,kBAAkBuD,KAAK;KAAEnF;KAAcE;IAAQ,CAAC;GACvD,CAAC;EACH,SAAS4D,KAAK;GACZC,QAAQC,MAAM,wCAAwCF,GAAG;GACzD,OAAOkB,QAAQ9E,QAAQ;EACzB;CACF;;;;;;;CAQA,OAAchB,YAAoBF,SAA2B;EAC3D,IAAI;GACF,MAAMqG,WAAW,KAAK9D,YAAY+D,IAAIpG,UAAU;GAChD,IAAI,CAACmG,UAAU,OAAO;GACtB,KAAK9D,YAAYgE,OAAOrG,UAAU;GAClC,KAAKsG,WAAWtG,YAAY,EAAEH,MAAM,YAAY,CAAC;GACjDsG,SAASnF,QAAQlB,OAAO;GACxB,OAAO;EACT,SAAS8E,KAAK;GACZC,QAAQC,MAAM,yCAAyCF,GAAG;GAC1D,OAAO;EACT;CACF;;;;;;;CAQA,cAA+D;EAC7D,OAAO,KAAKjC;CACd;CAIA,mBAA6D;EAC3D,MAAM8D,QAAQ,KAAKzE,UAAU;EAC7B,IAAI,CAACyE,OAAO,OAAOpF,KAAAA;EAEnB,OAAOqF,OAAOC,YACZD,OAAOE,QAAQH,KAAK,CAAC,CAACI,KAAK,CAACC,MAAMC,UAAU;GAC1C,MAAMC,UAAUD,KAAKC;GACrB,MAAMC,aAAaF,KAAKE;GACxB,IAAID,YAAY3F,KAAAA,KAAa4F,eAAe5F,KAAAA,GAC1C,OAAO,CAACyF,MAAMC,IAAI;GA+BpB,OAAO,CAACD,MAAMI;IA5BZ,GAAGH;IACH,GAAIC,YAAY3F,KAAAA,KAAa,EAC3B2F,UAAU,GAAG,CAACG,MAAMC,aAAyC;KAC3D,MAAM3G,cAAckB,mBAAmByF,OAAO;KAC9C,MAAME,QAAQ,KAAKC,kBACjBH,QAAQpH,YACRS,WACF;KACA,IAAI,CAAC6G,SAASA,MAAM5G,aAClB,OAAO,IAAIoF,cAAc,CAAC,CAAC;KAE7B,OAAOkB,QAAQG,MAAMC,OAAO;IAC9B,EACF;IACA,GAAIH,eAAe5F,KAAAA,KAAa,EAC9B4F,aACE,GAAG,CAACO,QAAQJ,aACT;KACH,MAAM3G,cAAckB,mBAAmByF,OAAO;KAK9C,IAAI,CAJU,KAAKG,kBACjBH,QAAQpH,YACRS,WAEG6G,GAAO;KACZ,OAAOL,WAAWO,QAAQJ,OAAO;IACnC,EACF;GAEYF,CAAW;EAC3B,CAAC,CACH;CACF;CAEA,kBACElH,YACAS,aAC2B;EAC3B,IAAIA,gBAAgBY,KAAAA,GAAW,OAAOA,KAAAA;EACtC,MAAMiG,QAAQ,KAAKnF,SAASiE,IAAIpG,UAAU;EAC1C,IAAI,CAACsH,OAAO3G,YAAY,OAAOU,KAAAA;EAC/B,IAAIiG,MAAM7G,gBAAgBY,KAAAA,GAAWiG,MAAM7G,cAAcA;EACzD,OAAO6G,MAAM7G,gBAAgBA,cAAc6G,QAAQjG,KAAAA;CACrD;CAIA,cACErB,YACAF,SACAW,aACkB;EAClB,OAAO,IAAIqF,SAAkB9E,SAASsB,WAAW;GAG/C,MAAMgF,QAAQ,KAAKnF,SAASiE,IAAIpG,UAAU;GAC1C,IAAI,CAACsH,OAAO3G,cAAc2G,MAAM7G,gBAAgBA,aAAa;IAC3D6B,uBAAO,IAAI0D,MAAM,wBAAwB,CAAC;IAC1C;GACF;GACA,MAAMzE,WAAW,KAAKc,YAAY+D,IAAIpG,UAAU;GAChD,IAAIuB,UACF,IAAI;IACFA,SAASe,uBACP,IAAI0D,MAAM,qDAAqD,CACjE;GACF,QAAQ,CACN;GAGJ,KAAK3D,YAAYoF,IAAIzH,YAAY;IAClBS;IACbO;IACAsB;GACF,CAAC;GACD,KAAKgE,WAAWtG,YAAY;IAC1BH,MAAM;IACNC,SAAS;KAAED,MAAM;KAASC;IAAQ;GACpC,CAAC;EACH,CAAC;CACH;CAEA,kBACEE,YACAS,aACM;EACN,IAAI,CAAC,KAAK8G,kBAAkBvH,YAAYS,WAAW,GAAG;EAEtD,IADc,KAAK0B,SAASiE,IAAIpG,UAC5BsH,CAAK,CAAC5G,aAAa;EAEvB,KAAK8B,WAAWkF,IAAIjH,WAAY;EAChC,KAAK6F,WAAWtG,YAAY,EAAEH,MAAM,YAAY,CAAC;CACnD;CAEA,gBACEG,YACAS,aACM;EACN,IAAIA,gBAAgBY,KAAAA,KAAa,CAAC,KAAKmB,WAAW6D,OAAO5F,WAAW,GAClE;EAGF,IADc,KAAK0B,SAASiE,IAAIpG,UAC5BsH,CAAK,EAAE7G,gBAAgBA,aAAa,KAAKkH,cAAc3H,UAAU;EAErE,MAAM4H,UAA6B,CAAA;EACnC,KAAKlF,kBAAkBqD,SAAS,EAAEjF,cAAcE,cAAc;GAC5D,IAAI,CAAC,GAAGF,YAAY,CAAC,CAAC+G,MAAM5D,OAAO,KAAKzB,WAAWsF,IAAI7D,EAAE,CAAC,GAAG;IAC3D2D,QAAQ3B,KAAK;KAAEnF;KAAcE;IAAQ,CAAC;IACtC;GACF;GACA,IAAI;IACFA,QAAQ;GACV,QAAQ,CACN;EAEJ,CAAC;EACD,KAAK0B,kBAAkBqF,SAAS;EAChC,KAAKrF,kBAAkBuD,KAAK,GAAG2B,OAAO;CACxC;CAEA,mBAA2BlD,OAOlB;EACP,MAAM1E,aAAa0E,MAAMsD,KAAKhI;EAC9B,MAAMS,cAAckB,mBAAmB+C,KAAK;EAC5C,MAAM4C,QAAQ,KAAKnF,SAASiE,IAAIpG,UAAU;EAE1C,IAAI,CAACsH,SAASA,MAAM7G,gBAAgBA,aAAa;EAIjD,IAAI6G,OAAO9G,WAAW;EACtB,IAAI8G,MAAM5G,aAAa;EAEvB,KAAKuH,gBAAgB;GACnBpI,MAAM;GACNG;GACAC,UAAUyE,MAAMsD,KAAK/H;GACrBC,QAAQwE,MAAMxE;GACdC,SAASuE,MAAMvE;GACf,GAAIuE,MAAMtE,aAAaiB,KAAAA,KAAa,EAAEjB,UAAUsE,MAAMtE,SAAS;GAC/D,GAAIsE,MAAMrE,iBAAiBgB,KAAAA,KAAa,EACtChB,cAAcqE,MAAMrE,aACtB;EACF,CAAC;CACH;CAIA,gBAAwB6H,SAAqC;EAC3D,IAAI;GACF,KAAKjG,WAAWkG,SAASD,OAAO;EAClC,SAAStD,KAAK;GACZC,QAAQC,MACN,mEACAF,GACF;EACF;CACF;CAEA,0BAAwC;EACtC,IAAI;GACF,KAAK3C,WAAWmG,iBAAiB,KAAKzF,SAAS;EACjD,SAASiC,KAAK;GACZC,QAAQC,MACN,yFACAF,GACF;EACF;CACF;CAIA,WAAmB5E,YAAoBqI,QAAmC;EACxE,MAAM7G,OAAO,IAAIY,IAAI,KAAKO,SAAS;EACnCnB,KAAKiG,IAAIzH,YAAYqI,MAAM;EAC3B,KAAK1F,YAAYnB;EACjB,KAAKqE,wBAAwB;CAC/B;CAEA,cAAsB7F,YAA0B;EAC9C,IAAI,CAAC,KAAK2C,UAAUmF,IAAI9H,UAAU,GAAG;EACrC,MAAMwB,OAAO,IAAIY,IAAI,KAAKO,SAAS;EACnCnB,KAAK6E,OAAOrG,UAAU;EACtB,KAAK2C,YAAYnB;EACjB,KAAKqE,wBAAwB;CAC/B;CAIA,mBAA2B5F,UAA2B;EACpD,MAAM8G,OAAO,KAAK/E,UAAU,CAAC,GAAG/B;EAChC,OAAO8G,MAAMC,YAAY3F,KAAAA,KAAa0F,MAAME,eAAe5F,KAAAA;CAC7D;CAEA,uBAA+B,EAC7BpB,UACAM,UACAC,aAKU;EACV,IAAIA,WAAW,OAAO;EACtB,IAAI,CAAC,KAAK8H,mBAAmBrI,QAAQ,GACnC,OAAO,CAAC,KAAKgD,cAAchC,mBAAmBV,QAAQ;EAExD,OAAOU,mBAAmBV,QAAQ;CACpC;CAEA,kBACEP,YACAC,UACAS,aACe;EAKf,MAAM4G,QAAuB;GAC3BrH;GACAU,YANyB,KAAKuC,YAAYwF,gBAAgB;IAC1DzI;IACAD;GACF,CAGcyI;GACZlI,UAAU;GACVC,WAAW;GACXE;GACAE,cAAc;EAChB;EACA,KAAKuB,SAASsF,IAAIzH,YAAYsH,KAAK;EACnC,OAAOA;CACT;;;;;;;;CASA,2BAAyC;EACvC,KAAK,MAAM,CAACtH,YAAYsH,UAAU,KAAKnF,UAAU;GAC/C,IAAI,CAACmF,MAAM3G,YAAY;GACvB,KAAKwB,SAASsF,IAAIzH,YAAY;IAC5BC,UAAUqH,MAAMrH;IAChBM,UAAU+G,MAAM/G;IAChBC,WAAW8G,MAAM9G;GACnB,CAAC;EACH;CACF;CAEA,iBACE8G,OACAsB,SAMM;EACN,IAAI,CAACtB,MAAM3G,YAAY;EACvB,MAAMH,YAAYoI,QAAQ1I,WAAWmB,KAAAA;EAErC,IAAIuH,QAAQrI,aAAa+G,MAAM/G,UAAU;GACvC,IAAIsI,sBAAsB;GAE1B,IAAIvB,MAAM1G,cAAc;IACtB,IAAIU,6BAA6BgG,MAAM/G,UAAUqI,QAAQrI,QAAQ,GAAG;KAElE+G,MAAM/G,WAAWqI,QAAQrI;KACzBsI,sBAAsB;IACxB,OAAO;KAOL,IAAIlK,QAAQC,IAAIC,aAAa,cAC3BgG,QAAQiE,KACN,qHACA;MACEvH,UAAU+F,MAAM/G;MAChBiB,MAAMoH,QAAQrI;MACdP,YAAY4I,QAAQ5I;KACtB,CACF;KAEF6I,sBAAsB;IACxB;GACF,OAAO,IAAI,CAACD,QAAQrI,SAASwI,WAAWzB,MAAM/G,QAAQ,GAAG;IACvD,IACEU,mBAAmBqG,MAAM/G,QAAQ,KACjCU,mBAAmB2H,QAAQrI,QAAQ,KACnCe,6BAA6BgG,MAAM/G,UAAUqI,QAAQrI,QAAQ,GAC7D;KACA,MAAMyI,cAAc,KAAKT,uBAAuB;MAC9CtI,UAAU2I,QAAQ3I;MAClBM,UAAUqI,QAAQrI;MAClBC;KACF,CAAC;KACD,IAAIwI,aAAa1B,MAAM3G,WAAWJ,SAAS0I,MAAM;KACjD3B,MAAM/G,WAAWqI,QAAQrI;KACzB+G,MAAM1G,eAAeoI;KACrBH,sBAAsB;IACxB,OAAO;KAQL,IAAIlK,QAAQC,IAAIC,aAAa,cAC3BgG,QAAQiE,KACN,iGACA;MACEvH,UAAU+F,MAAM/G;MAChBiB,MAAMoH,QAAQrI;MACdP,YAAY4I,QAAQ5I;KACtB,CACF;KAEF6I,sBAAsB;IACxB;GACF;GAEA,IAAIA,uBAAuBvB,MAAM3G,YAAY;IAC3C,MAAMuI,QAAQN,QAAQrI,SAAS4I,MAAM7B,MAAM/G,SAASwH,MAAM;IAC1DT,MAAM3G,WAAWJ,SAAS6I,OAAOF,KAAK;IACtC,MAAMF,cAAc,KAAKT,uBAAuB;KAC9CtI,UAAU2I,QAAQ3I;KAClBM,UAAUqI,QAAQrI;KAClBC;IACF,CAAC;IACD,IAAIwI,aAAa1B,MAAM3G,WAAWJ,SAAS0I,MAAM;IACjD3B,MAAM/G,WAAWqI,QAAQrI;IACzB+G,MAAM1G,eAAeoI;GACvB;EACF;EAEA,IAAI,CAAC1B,MAAM1G,gBAAgB0G,MAAM3G,YAIX;OAAA,KAAK4H,uBAAuB;IAC9CtI,UAAU2I,QAAQ3I;IAClBM,UAAU+G,MAAM/G;IAChBC;GACF,CACIwI,GAAa;IACf1B,MAAM3G,WAAWJ,SAAS0I,MAAM;IAChC3B,MAAM1G,eAAe;GACvB;;CAEJ;CAEA,iBAAyBwE,UAA0C;EACjE,MAAMiE,YAAY,KAAKvG;EAEvB,KAAK,MAAMwG,WAAWlE,UAAU;GAC9B,IAAI,CAACkE,WAAW,CAACC,MAAMC,QAASF,QAA0BV,OAAO,GAC/D;GAEF,KAAK,MAAMA,WAAWU,QAAQV,SAAwD;IACpF,IAAI,CAACA,WAAWA,QAAQ/I,SAAS,aAAa;IAE9C,MAAM4J,WAAW,KAAKtH,SAASiE,IAAIwC,QAAQ5I,UAAU;IAErD,IAAIqJ,WAAW;KAIb,IAAI,CAACI,UAAU9I,YACb,KAAKwB,SAASsF,IAAImB,QAAQ5I,YAAY;MACpCC,UAAU2I,QAAQ3I;MAClBM,UAAUqI,QAAQrI;MAClBC,WAAWoI,QAAQ1I,WAAWmB,KAAAA;KAChC,CAAC;KAEH,IAAIuH,QAAQxD,UAAU,KAAKK,iBAAiBmD,QAAQxD,QAAQ;KAC5D;IACF;IAGA,IAAIkC,QAAQmC;IAEZ,IAAInC,SAAS,CAACA,MAAM3G,YAAY;KAM9B,IAAI,EAFFiI,QAAQrI,aAAa+G,MAAM/G,YAC1BqI,QAAQ1I,WAAWmB,KAAAA,MAAeiG,MAAM9G,YACpB;MACrB,IAAIoI,QAAQxD,UAAU,KAAKK,iBAAiBmD,QAAQxD,QAAQ;MAC5D;KACF;KACA,KAAKjD,SAASkE,OAAOuC,QAAQ5I,UAAU;KACvCsH,QAAQjG,KAAAA;IACV;IAEA,IAAI,CAACiG,OACHA,QAAQ,KAAKkB,kBACXI,QAAQ5I,YACR4I,QAAQ3I,UACR2I,QAAQ1I,WAAWmB,KAAAA,CACrB;IAGF,IAAIuH,QAAQe,aAAatI,KAAAA,GAAWiG,MAAM5G,cAAc;IAExD,KAAKiI,iBAAiBrB,OAAOsB,OAAO;IAEpC,IAAIA,QAAQ1I,WAAWmB,KAAAA,KAAa,CAACiG,MAAM9G,WAAW;KAIpD,MAAM,EAAEG,YAAYiJ,qBAAqBtC;KACzC,IAAI,CAACsC,kBAAkB;KACvBtC,MAAM9G,YAAY;KAClB8G,MAAM1G,eAAe;KACrBgJ,iBAAiBC,YACf,IAAI7K,aAAa;MACfkB,QAAQ0I,QAAQ1I;MAChBE,UAAUwI,QAAQxI;MAClBD,SAASyI,QAAQzI;MACjB,GAAIyI,QAAQvI,iBAAiBgB,KAAAA,IACzB,EAAEhB,cAAcuI,QAAQvI,aAAa,IACrC,CAAC;KACP,CAAC,CACH;KACAuJ,iBAAiBX,MAAM;IACzB;IAEA,IAAIL,QAAQxD,UAAU,KAAKK,iBAAiBmD,QAAQxD,QAAQ;GAC9D;EACF;CACF;AACF"}

package/dist/store/clients/external-thread.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"external-thread.d.ts","names":[],"sources":["../../../src/store/clients/external-thread.ts"],"mappings":";;;;;;;;;;KAyDY,wBAAwB;EAClC;;KAgBU;EACV,mBAAmB;EACnB;EACA;EACA,QAAQ;EACR;;;;;;;EAOA;;;;;EAKA,SAAS,SAAS;EAClB,UAAU,SAAS;EACnB,YAAY;EACZ;EACA;EACA;;;;;;;EAOA,yBAAyB;EACzB,oBAAoB,SAAS;;EAE7B,qBAAqB,SAAS;EAC9B,wBAAwB;EACxB,oBAAoB;EACpB,kBAAkB;EAClB,gBAAgB;;EAEhB,QAAQ;;EAER,WAAW;;EAEX,2BAA2B,SAAS;;cAqkCzB,4CAAc,SAAA,yBAAA"}
+{"version":3,"file":"external-thread.d.ts","names":[],"sources":["../../../src/store/clients/external-thread.ts"],"mappings":";;;;;;;;;;KA6DY,wBAAwB;EAClC;;KAgBU;EACV,mBAAmB;EACnB;EACA;EACA,QAAQ;EACR;;;;;;;EAOA;;;;;EAKA,SAAS,SAAS;EAClB,UAAU,SAAS;EACnB,YAAY;EACZ;EACA;EACA;;;;;;;EAOA,yBAAyB;EACzB,oBAAoB,SAAS;;EAE7B,qBAAqB,SAAS;EAC9B,wBAAwB;EACxB,oBAAoB;EACpB,kBAAkB;EAClB,gBAAgB;;EAEhB,QAAQ;;EAER,WAAW;;EAEX,2BAA2B,SAAS;;cAohCzB,4CAAc,SAAA,yBAAA"}

package/dist/store/clients/external-thread.js

@@ -3,6 +3,7 @@ import { getThreadMessageText } from "../../utils/text.js";
 import { toMessagePartStatus } from "../../utils/normalizePartStatus.js";
 import { resolveToolApprovalResponse } from "../../runtime/utils/resolveToolApprovalResponse.js";
 import { isCreateAttachment } from "../../types/attachment.js";
+import { AttachmentAddOperations, drainAttachmentAdd } from "../../runtime/utils/attachment-add-operations.js";
 import { ThreadSuggestions } from "./suggestions.js";
 import { ModelContext } from "./model-context-client.js";
 import { DataRenderers } from "../../react/client/DataRenderers.js";
@@ -353,39 +354,6 @@ const useAttachmentResource = ({ attachment, onRemove }) => {
 	};
 };
 const AttachmentResource = resource(useAttachmentResource);
-var AttachmentAddOperations = class {
-	operations = /* @__PURE__ */ new Set();
-	start() {
-		const operation = {
-			cancelled: false,
-			attachmentIds: /* @__PURE__ */ new Set()
-		};
-		this.operations.add(operation);
-		return operation;
-	}
-	accept(operation, attachmentId) {
-		if (operation.cancelled) return false;
-		operation.attachmentIds.add(attachmentId);
-		return true;
-	}
-	finish(operation) {
-		this.operations.delete(operation);
-	}
-	isCancelled(operation) {
-		return operation.cancelled;
-	}
-	cancel(attachmentId) {
-		for (const operation of [...this.operations]) {
-			if (!operation.attachmentIds.has(attachmentId)) continue;
-			operation.cancelled = true;
-			this.operations.delete(operation);
-		}
-	}
-	cancelAll() {
-		for (const operation of this.operations) operation.cancelled = true;
-		this.operations.clear();
-	}
-};
 const useQueueItemClient = ({ item, onMove, onRemove }) => {
 	return {
 		getState: () => item,
@@ -398,11 +366,6 @@ const useQueueItemClient = ({ item, onMove, onRemove }) => {
 	};
 };
 const QueueItemClient = resource(useQueueItemClient);
-const drainAdapterAdd = async (result, upsert) => {
-	if (Symbol.asyncIterator in result) {
-		for await (const attachment of result) if (!upsert(attachment)) break;
-	} else upsert(await result);
-};
 const useLiveState = (initial) => {
 	const $ = c(3);
 	const [state, setState] = useState(initial);
@@ -429,7 +392,7 @@ const useLiveState = (initial) => {
 	return t1;
 };
 const useComposerClientResource = (t0) => {
-	const $ = c(105);
+	const $ = c(108);
 	const { type, canCancel, isRunning: t1, isSendDisabled: t2, onCancel, onBeginEdit, onSend, message, queue, attachmentAdapter } = t0;
 	const isRunning = t1 === void 0 ? false : t1;
 	const isSendDisabled = t2 === void 0 ? false : t2;
@@ -611,7 +574,7 @@ const useComposerClientResource = (t0) => {
 			if (!isCreateAttachment(fileOrAttachment) && attachmentAdapter) {
 				const operation = attachmentAddOperations.start();
 				try {
-					await drainAdapterAdd(attachmentAdapter.add({ file: fileOrAttachment }), (attachment_1) => {
+					await drainAttachmentAdd(attachmentAdapter.add({ file: fileOrAttachment }), (attachment_1) => {
 						if (!attachmentAddOperations.accept(operation, attachment_1.id)) return false;
 						upsertAttachment(attachment_1);
 						return true;
@@ -756,18 +719,27 @@ const useComposerClientResource = (t0) => {
 		$[78] = t19;
 	} else t19 = $[78];
 	let t20;
-	if ($[79] !== onCancel || $[80] !== setIsEditing || $[81] !== type) {
+	if ($[79] !== attachmentsRef || $[80] !== onCancel || $[81] !== removePendingAttachments || $[82] !== setAttachments || $[83] !== setIsEditing || $[84] !== type) {
 		t20 = () => {
+			if (type === "edit") {
+				attachmentAddOperations.cancelAll();
+				const removed_2 = attachmentsRef.current;
+				setAttachments([]);
+				removePendingAttachments(removed_2).catch(_temp7);
+			}
 			onCancel?.();
 			if (type === "edit") setIsEditing(false);
 		};
-		$[79] = onCancel;
-		$[80] = setIsEditing;
-		$[81] = type;
-		$[82] = t20;
-	} else t20 = $[82];
+		$[79] = attachmentsRef;
+		$[80] = onCancel;
+		$[81] = removePendingAttachments;
+		$[82] = setAttachments;
+		$[83] = setIsEditing;
+		$[84] = type;
+		$[85] = t20;
+	} else t20 = $[85];
 	let t21;
-	if ($[83] !== isEditingRef || $[84] !== onBeginEdit || $[85] !== setIsEditing || $[86] !== type || $[87] !== updateFromMessage) {
+	if ($[86] !== isEditingRef || $[87] !== onBeginEdit || $[88] !== setIsEditing || $[89] !== type || $[90] !== updateFromMessage) {
 		t21 = () => {
 			onBeginEdit?.();
 			if (type === "thread") return;
@@ -775,24 +747,24 @@ const useComposerClientResource = (t0) => {
 			setIsEditing(true);
 			updateFromMessage();
 		};
-		$[83] = isEditingRef;
-		$[84] = onBeginEdit;
-		$[85] = setIsEditing;
-		$[86] = type;
-		$[87] = updateFromMessage;
-		$[88] = t21;
-	} else t21 = $[88];
+		$[86] = isEditingRef;
+		$[87] = onBeginEdit;
+		$[88] = setIsEditing;
+		$[89] = type;
+		$[90] = updateFromMessage;
+		$[91] = t21;
+	} else t21 = $[91];
 	let t22;
-	if ($[89] !== queueItemClients) {
+	if ($[92] !== queueItemClients) {
 		t22 = (selector_0) => {
 			if ("id" in selector_0) return queueItemClients.get({ key: selector_0.id });
 			return queueItemClients.get(selector_0);
 		};
-		$[89] = queueItemClients;
-		$[90] = t22;
-	} else t22 = $[90];
+		$[92] = queueItemClients;
+		$[93] = t22;
+	} else t22 = $[93];
 	let t23;
-	if ($[91] !== setQuote || $[92] !== setRole || $[93] !== setRunConfig || $[94] !== setText || $[95] !== t14 || $[96] !== t15 || $[97] !== t16 || $[98] !== t17 || $[99] !== t18 || $[100] !== t19 || $[101] !== t20 || $[102] !== t21 || $[103] !== t22) {
+	if ($[94] !== setQuote || $[95] !== setRole || $[96] !== setRunConfig || $[97] !== setText || $[98] !== t14 || $[99] !== t15 || $[100] !== t16 || $[101] !== t17 || $[102] !== t18 || $[103] !== t19 || $[104] !== t20 || $[105] !== t21 || $[106] !== t22) {
 		t23 = {
 			getState: t14,
 			setText,
@@ -805,26 +777,26 @@ const useComposerClientResource = (t0) => {
 			send: t19,
 			cancel: t20,
 			beginEdit: t21,
-			startDictation: _temp7,
-			stopDictation: _temp8,
+			startDictation: _temp8,
+			stopDictation: _temp9,
 			setQuote,
 			queueItem: t22
 		};
-		$[91] = setQuote;
-		$[92] = setRole;
-		$[93] = setRunConfig;
-		$[94] = setText;
-		$[95] = t14;
-		$[96] = t15;
-		$[97] = t16;
-		$[98] = t17;
-		$[99] = t18;
-		$[100] = t19;
-		$[101] = t20;
-		$[102] = t21;
-		$[103] = t22;
-		$[104] = t23;
-	} else t23 = $[104];
+		$[94] = setQuote;
+		$[95] = setRole;
+		$[96] = setRunConfig;
+		$[97] = setText;
+		$[98] = t14;
+		$[99] = t15;
+		$[100] = t16;
+		$[101] = t17;
+		$[102] = t18;
+		$[103] = t19;
+		$[104] = t20;
+		$[105] = t21;
+		$[106] = t22;
+		$[107] = t23;
+	} else t23 = $[107];
 	return t23;
 };
 const ComposerClientResource = resource(useComposerClientResource);
@@ -1157,12 +1129,12 @@ const useExternalThread = (t0) => {
 	const hasSpeech = !!speechAdapter;
 	let t25;
 	if ($[59] !== messageClients.state) {
-		t25 = messageClients.state.map(_temp9);
+		t25 = messageClients.state.map(_temp0);
 		$[59] = messageClients.state;
 		$[60] = t25;
 	} else t25 = $[60];
 	const messageStates = t25;
-	const t26 = messages.length === 0;
+	const t26 = messages.length === 0 && !isLoading;
 	let t27;
 	if ($[61] !== hasAttachments || $[62] !== hasBranches || $[63] !== hasCancel || $[64] !== hasEdit || $[65] !== hasFeedback || $[66] !== hasQueue || $[67] !== hasRefetchThread || $[68] !== hasReload || $[69] !== hasSpeech) {
 		t27 = {
@@ -1326,24 +1298,24 @@ const useExternalThread = (t0) => {
 			composer: t31,
 			suggestions: t32,
 			append: t33,
-			deleteMessage: _temp0,
+			deleteMessage: _temp1,
 			startRun: t34,
 			resumeRun: t35,
 			cancelRun: handleCancelRun,
 			...t36,
 			importExternalState: t37,
-			getModelContext: _temp1,
-			export: _temp10,
-			import: _temp11,
-			reset: _temp12,
+			getModelContext: _temp10,
+			export: _temp11,
+			import: _temp12,
+			reset: _temp13,
 			message: t38,
 			stopSpeaking: speechController.stop,
-			connectVoice: _temp13,
-			disconnectVoice: _temp14,
-			getVoiceVolume: _temp15,
-			subscribeVoiceVolume: _temp17,
-			muteVoice: _temp18,
-			unmuteVoice: _temp19
+			connectVoice: _temp14,
+			disconnectVoice: _temp15,
+			getVoiceVolume: _temp16,
+			subscribeVoiceVolume: _temp18,
+			muteVoice: _temp19,
+			unmuteVoice: _temp20
 		};
 		$[103] = handleCancelRun;
 		$[104] = speechController.stop;
@@ -1408,37 +1380,40 @@ function _temp5(part_0) {
 function _temp6(a_0) {
 	return a_0.status.type !== "complete";
 }
-function _temp7() {}
+function _temp7(error_1) {
+	console.error("Failed to remove cancelled edit attachments", error_1);
+}
 function _temp8() {}
-function _temp9(s, idx, arr) {
+function _temp9() {}
+function _temp0(s, idx, arr) {
 	return {
 		...s,
 		isLast: idx === arr.length - 1
 	};
 }
-function _temp0() {}
-function _temp1() {
+function _temp1() {}
+function _temp10() {
 	return {
 		tools: {},
 		config: {}
 	};
 }
-function _temp10() {
+function _temp11() {
 	return { messages: [] };
 }
-function _temp11() {}
 function _temp12() {}
 function _temp13() {}
 function _temp14() {}
-function _temp15() {
+function _temp15() {}
+function _temp16() {
 	return 0;
 }
-function _temp16() {}
-function _temp17() {
-	return _temp16;
+function _temp17() {}
+function _temp18() {
+	return _temp17;
 }
-function _temp18() {}
 function _temp19() {}
+function _temp20() {}
 //#endregion
 export { ExternalThread };