@assistant-ui/core 0.2.23 → 0.3.1

package/dist/runtimes/local/local-thread-runtime-core.js.map

@@ -1 +1 @@
-{"version":3,"file":"local-thread-runtime-core.js","names":["fromThreadMessageLike","generateId","ChatModelAdapter","ChatModelRunResult","shouldContinue","LocalRuntimeOptionsBase","consumeSuggestionResult","AddToolResultOptions","ResumeToolCallOptions","RespondToToolApprovalOptions","ThreadSuggestion","ThreadRuntimeCore","StartRunConfig","ResumeRunConfig","BaseThreadRuntimeCore","AppendMessage","ThreadAssistantMessage","RunConfig","toAssistantError","ModelContextProvider","createMessageQueue","MessageQueueController","EMPTY_QUEUE_ITEMS","QueueItemState","AbortError","Error","name","detach","constructor","message","LocalThreadRuntimeCore","capabilities","switchToBranch","switchBranchDuringRun","edit","delete","reload","cancel","unstable_copy","speech","dictation","voice","attachments","feedback","queue","abortController","AbortController","_queue","_queueRunInFlight","_historyWrites","Map","Promise","_chainHistoryWrite","id","write","next","get","resolve","then","stored","set","_persistPausedMessage","parentId","status","type","history","_options","adapters","update","bind","item","runConfig","_lastRunConfig","catch","isDisabled","isSendDisabled","_isLoading","isLoading","_suggestions","_suggestionsController","suggestions","contextProvider","options","__internal_setOptions","_getThreadId","__internal_setGetThreadId","getThreadId","_getInitializePromise","__internal_setGetInitializePromise","getPromise","extras","undefined","hasUpdates","canSpeak","canDictate","canVoice","canAttach","canFeedback","canDelete","canQueue","unstable_enableMessageQueue","run","_runAppend","finally","notifyIdle","subscribe","_notifySubscribers","adapter","clear","_loadPromise","__internal_load","promise","load","repo","repository","import","messages","length","ensureInitialized","resume","unstable_resume","startRun","headId","sourceId","append","isTail","at","willRun","role","enqueue","steer","getQueueItems","items","steerQueueItem","queueItemId","removeQueueItem","remove","initPromise","newMessage","reason","addOrUpdateMessage","resetHead","deleteMessage","messageId","getMessages","messageIndex","findIndex","m","resumeRun","stream","startConfig","exportExternalState","importExternalState","runCallback","content","metadata","unstable_state","state","unstable_annotations","unstable_data","steps","custom","createdAt","Date","_runLoop","_notifyEventSubscribers","alreadyPersisted","notifyBusy","abort","performRoundtrip","unstable_humanToolNames","queueMicrotask","suggestion","signal","promiseOrGenerator","generate","onUpdate","r","initialContent","initialAnnotations","initialData","initialSteps","initialCustom","updateMessage","Partial","newSteps","newAnnotations","newData","annotations","data","timing","maxSteps","unstable_composerMetadata","_","context","getModelContext","chatModel","abortSignal","threadId","unstable_assistantMessageId","unstable_threadId","unstable_parentId","unstable_getMessage","Symbol","asyncIterator","aborted","e","error","isTerminal","isPausing","cancelRun","addToolResult","toolCallId","result","isError","artifact","messageData","getMessage","added","found","newContent","map","c","resumeToolCall","respondToToolApproval","approvalId","approved","optionId","findLast","some","approval","target","find","resolution","targetApproval"],"sources":["../../../src/runtimes/local/local-thread-runtime-core.ts"],"sourcesContent":["import { fromThreadMessageLike } from \"../../runtime/utils/thread-message-like\";\nimport { generateId } from \"../../utils/id\";\nimport type {\n  ChatModelAdapter,\n  ChatModelRunResult,\n} from \"../../runtime/utils/chat-model-adapter\";\nimport { shouldContinue } from \"./should-continue\";\nimport type { LocalRuntimeOptionsBase } from \"./local-runtime-options\";\nimport { consumeSuggestionResult } from \"../../adapters/suggestion\";\nimport type {\n  AddToolResultOptions,\n  ResumeToolCallOptions,\n  RespondToToolApprovalOptions,\n  ThreadSuggestion,\n  ThreadRuntimeCore,\n  StartRunConfig,\n  ResumeRunConfig,\n} from \"../../runtime/interfaces/thread-runtime-core\";\nimport { BaseThreadRuntimeCore } from \"../../runtime/base/base-thread-runtime-core\";\nimport type {\n  AppendMessage,\n  ThreadAssistantMessage,\n} from \"../../types/message\";\nimport type { RunConfig } from \"../../types/message\";\nimport { toAssistantError } from \"../../types/error\";\nimport type { ModelContextProvider } from \"../../model-context/types\";\nimport {\n  createMessageQueue,\n  type MessageQueueController,\n} from \"../../runtime/queue/message-queue\";\nimport {\n  EMPTY_QUEUE_ITEMS,\n  type QueueItemState,\n} from \"../../store/scopes/queue-item\";\n\nclass AbortError extends Error {\n  override name = \"AbortError\";\n  detach: boolean;\n\n  constructor(detach: boolean, message?: string) {\n    super(message);\n    this.detach = detach;\n  }\n}\n\nexport class LocalThreadRuntimeCore\n  extends BaseThreadRuntimeCore\n  implements ThreadRuntimeCore\n{\n  public readonly capabilities = {\n    switchToBranch: true,\n    switchBranchDuringRun: true,\n    edit: true,\n    delete: false,\n    reload: true,\n    cancel: true,\n    unstable_copy: true,\n    speech: false,\n    dictation: false,\n    voice: false,\n    attachments: false,\n    feedback: false,\n    queue: false,\n  };\n\n  private abortController: AbortController | null = null;\n\n  private _queue: MessageQueueController | null = null;\n  private _queueRunInFlight = false;\n\n  private _historyWrites = new Map<string, Promise<void>>();\n\n  // Writes for one message id must land in issue order; an earlier paused\n  // snapshot arriving after the terminal write would resurrect the pause.\n  private _chainHistoryWrite(\n    id: string,\n    write: () => Promise<void>,\n  ): Promise<void> {\n    const next = (this._historyWrites.get(id) ?? Promise.resolve()).then(\n      write,\n      write,\n    );\n    const stored = next.then(\n      () => {},\n      () => {},\n    );\n    this._historyWrites.set(id, stored);\n    void stored.then(() => {\n      if (this._historyWrites.get(id) === stored) {\n        this._historyWrites.delete(id);\n      }\n    });\n    return next;\n  }\n\n  // A decision recorded on a still-paused message must reach history before\n  // the run resumes, or a refresh would restore the message without it.\n  private _persistPausedMessage(\n    parentId: string | null,\n    message: ThreadAssistantMessage,\n  ) {\n    if (message.status?.type !== \"requires-action\") return;\n    const history = this._options.adapters.history;\n    if (!history?.update) return;\n    const update = history.update.bind(history);\n    const item = { parentId, message, runConfig: this._lastRunConfig };\n    this._chainHistoryWrite(message.id, () => update(item)).catch(() => {});\n  }\n\n  public readonly isDisabled = false;\n  public readonly isSendDisabled = false;\n\n  private _isLoading = false;\n  public get isLoading() {\n    return this._isLoading;\n  }\n\n  private _suggestions: readonly ThreadSuggestion[] = [];\n  private _suggestionsController: AbortController | null = null;\n  public get suggestions(): readonly ThreadSuggestion[] {\n    return this._suggestions;\n  }\n\n  public get adapters() {\n    return this._options.adapters;\n  }\n\n  constructor(\n    contextProvider: ModelContextProvider,\n    options: LocalRuntimeOptionsBase,\n  ) {\n    super(contextProvider);\n    this.__internal_setOptions(options);\n  }\n\n  private _options!: LocalRuntimeOptionsBase;\n\n  private _lastRunConfig: RunConfig = {};\n\n  private _getThreadId?: () => string | undefined;\n\n  public __internal_setGetThreadId(getThreadId: () => string | undefined) {\n    this._getThreadId = getThreadId;\n  }\n\n  private _getInitializePromise?: () => Promise<unknown> | undefined;\n\n  public __internal_setGetInitializePromise(\n    getPromise: () => Promise<unknown> | undefined,\n  ) {\n    this._getInitializePromise = getPromise;\n  }\n\n  public get extras() {\n    return undefined;\n  }\n\n  public __internal_setOptions(options: LocalRuntimeOptionsBase) {\n    if (this._options === options) return;\n\n    this._options = options;\n\n    let hasUpdates = false;\n\n    const canSpeak = options.adapters?.speech !== undefined;\n    if (this.capabilities.speech !== canSpeak) {\n      this.capabilities.speech = canSpeak;\n      hasUpdates = true;\n    }\n\n    const canDictate = options.adapters?.dictation !== undefined;\n    if (this.capabilities.dictation !== canDictate) {\n      this.capabilities.dictation = canDictate;\n      hasUpdates = true;\n    }\n\n    const canVoice = options.adapters?.voice !== undefined;\n    if (this.capabilities.voice !== canVoice) {\n      this.capabilities.voice = canVoice;\n      hasUpdates = true;\n    }\n\n    const canAttach = options.adapters?.attachments !== undefined;\n    if (this.capabilities.attachments !== canAttach) {\n      this.capabilities.attachments = canAttach;\n      hasUpdates = true;\n    }\n\n    const canFeedback = options.adapters?.feedback !== undefined;\n    if (this.capabilities.feedback !== canFeedback) {\n      this.capabilities.feedback = canFeedback;\n      hasUpdates = true;\n    }\n\n    const canDelete = options.adapters?.history?.delete !== undefined;\n    if (this.capabilities.delete !== canDelete) {\n      this.capabilities.delete = canDelete;\n      hasUpdates = true;\n    }\n\n    const canQueue = options.unstable_enableMessageQueue === true;\n    if (canQueue && !this._queue) {\n      this._queue = createMessageQueue({\n        run: (message) => {\n          // release the queue when the dispatch settles, even if it rejects\n          // before reaching startRun's finally, so a failure can't deadlock it\n          this._queueRunInFlight = true;\n          void this._runAppend(message)\n            .finally(() => {\n              this._queueRunInFlight = false;\n              this._queue?.notifyIdle();\n            })\n            .catch(() => {});\n        },\n      });\n      this._queue.subscribe(() => this._notifySubscribers());\n    } else if (!canQueue && this._queue) {\n      this._queue.adapter.clear(\"cancel-run\");\n      this._queue = null;\n    }\n    if (this.capabilities.queue !== canQueue) {\n      this.capabilities.queue = canQueue;\n      hasUpdates = true;\n    }\n\n    if (hasUpdates) this._notifySubscribers();\n  }\n\n  private _loadPromise: Promise<void> | undefined;\n  public __internal_load() {\n    if (this._loadPromise) return this._loadPromise;\n\n    const promise = this.adapters.history?.load() ?? Promise.resolve(null);\n\n    this._isLoading = true;\n    this._notifySubscribers();\n\n    this._loadPromise = promise\n      .then((repo) => {\n        if (!repo) return;\n        this.repository.import(repo);\n        if (repo.messages.length > 0) {\n          this.ensureInitialized();\n        }\n        this._notifySubscribers();\n\n        const resume = this.adapters.history?.resume?.bind(\n          this.adapters.history,\n        );\n        if (repo.unstable_resume && resume) {\n          this.startRun(\n            {\n              parentId: this.repository.headId,\n              sourceId: this.repository.headId,\n              runConfig: this._lastRunConfig,\n            },\n            resume,\n          ).catch(() => {});\n        }\n      })\n      .finally(() => {\n        this._isLoading = false;\n        this._notifySubscribers();\n      });\n\n    return this._loadPromise;\n  }\n\n  public async append(message: AppendMessage): Promise<void> {\n    const isTail = message.parentId === (this.messages.at(-1)?.id ?? null);\n    const willRun = message.startRun ?? message.role === \"user\";\n    if (this._queue && willRun && isTail) {\n      this._queue.adapter.enqueue(message, { steer: message.steer ?? false });\n      return;\n    }\n    if (this._queue && !isTail) this._queue.adapter.clear(\"edit\");\n    return this._runAppend(message);\n  }\n\n  public getQueueItems(): readonly QueueItemState[] {\n    // Reads can arrive during base-thread construction, before the queue field\n    // is assigned, so guard against the unset field.\n    return this._queue?.adapter.items ?? EMPTY_QUEUE_ITEMS;\n  }\n\n  public steerQueueItem(queueItemId: string): void {\n    this._queue?.adapter.steer(queueItemId);\n  }\n\n  public removeQueueItem(queueItemId: string): void {\n    this._queue?.adapter.remove(queueItemId);\n  }\n\n  private async _runAppend(message: AppendMessage): Promise<void> {\n    this.ensureInitialized();\n\n    const initPromise = this._getInitializePromise?.();\n    if (initPromise) {\n      await initPromise;\n    }\n\n    const newMessage = fromThreadMessageLike(message, generateId(), {\n      type: \"complete\",\n      reason: \"unknown\",\n    });\n    this.repository.addOrUpdateMessage(message.parentId, newMessage);\n    this._options.adapters.history?.append({\n      parentId: message.parentId,\n      message: newMessage,\n      ...(message.runConfig !== undefined && { runConfig: message.runConfig }),\n    });\n\n    const startRun = message.startRun ?? message.role === \"user\";\n    if (startRun) {\n      await this.startRun({\n        parentId: newMessage.id,\n        sourceId: message.sourceId,\n        runConfig: message.runConfig ?? {},\n      });\n    } else {\n      this.repository.resetHead(newMessage.id);\n      this._notifySubscribers();\n    }\n  }\n\n  public async deleteMessage(messageId: string): Promise<void> {\n    const adapter = this._options.adapters.history;\n    if (!adapter?.delete)\n      throw new Error(\"Runtime does not support deleting messages.\");\n\n    const messages = this.repository.getMessages();\n    const messageIndex = messages.findIndex((m) => m.id === messageId);\n    if (messageIndex === -1) throw new Error(\"Message not found.\");\n\n    const message = messages[messageIndex]!;\n    const parentId = messages[messageIndex - 1]?.id ?? null;\n    const items = [{ parentId, message }];\n\n    await adapter.delete(items);\n\n    this.repository.deleteMessage(messageId);\n    this._notifySubscribers();\n  }\n\n  public resumeRun({ stream, ...startConfig }: ResumeRunConfig): Promise<void> {\n    if (!stream)\n      throw new Error(\"You must pass a stream parameter to resume runs.\");\n    return this.startRun(startConfig, stream);\n  }\n\n  public exportExternalState(): any {\n    throw new Error(\"Runtime does not support exporting external states.\");\n  }\n\n  public importExternalState(): void {\n    throw new Error(\"Runtime does not support importing external states.\");\n  }\n\n  public async startRun(\n    { parentId, runConfig }: StartRunConfig,\n    runCallback?: ChatModelAdapter[\"run\"],\n  ): Promise<void> {\n    this.ensureInitialized();\n\n    // add assistant message\n    const id = generateId();\n    const message: ThreadAssistantMessage = {\n      id,\n      role: \"assistant\",\n      status: { type: \"running\" },\n      content: [],\n      metadata: {\n        unstable_state: this.state,\n        unstable_annotations: [],\n        unstable_data: [],\n        steps: [],\n        custom: {},\n      },\n      createdAt: new Date(),\n    };\n\n    return this._runLoop(parentId, message, runConfig, runCallback);\n  }\n\n  private async _runLoop(\n    parentId: string | null,\n    message: ThreadAssistantMessage,\n    runConfig: RunConfig | undefined,\n    runCallback?: ChatModelAdapter[\"run\"],\n  ): Promise<void> {\n    this._notifyEventSubscribers(\"runStart\", {});\n\n    // A run entered on a requires-action message resumes a pause an\n    // update-capable adapter already holds (written at pause time or loaded).\n    const alreadyPersisted =\n      message.status?.type === \"requires-action\" &&\n      this._options.adapters.history?.update !== undefined;\n\n    try {\n      // mark busy for runs not started through the queue (regenerate, resume)\n      this._queue?.notifyBusy();\n      this._suggestions = [];\n      this._suggestionsController?.abort();\n      this._suggestionsController = null;\n      this._notifySubscribers();\n\n      do {\n        message = await this.performRoundtrip(\n          parentId,\n          message,\n          runConfig,\n          alreadyPersisted,\n          runCallback,\n        );\n        runCallback = undefined;\n      } while (shouldContinue(message, this._options.unstable_humanToolNames));\n    } finally {\n      this._notifyEventSubscribers(\"runEnd\", {});\n      // queue-driven runs release from the driver settle handler; a direct\n      // run (regenerate, resume) releases here\n      if (!this._queueRunInFlight) {\n        queueMicrotask(() => this._queue?.notifyIdle());\n      }\n    }\n\n    if (\n      this.adapters.suggestion &&\n      message.status?.type !== \"requires-action\"\n    ) {\n      this._suggestionsController = new AbortController();\n      const signal = this._suggestionsController.signal;\n      const adapter = this.adapters.suggestion;\n      void (async () => {\n        try {\n          const promiseOrGenerator = adapter.generate({\n            messages: this.messages,\n            signal,\n          });\n          await consumeSuggestionResult(promiseOrGenerator, {\n            signal,\n            onUpdate: (r) => {\n              this._suggestions = r;\n              this._notifySubscribers();\n            },\n          });\n        } catch {}\n      })();\n    }\n  }\n\n  private async performRoundtrip(\n    parentId: string | null,\n    message: ThreadAssistantMessage,\n    runConfig: RunConfig | undefined,\n    alreadyPersisted: boolean,\n    runCallback?: ChatModelAdapter[\"run\"],\n  ) {\n    const messages = parentId ? this.repository.getMessages(parentId) : [];\n\n    // abort existing run\n    this.abortController?.abort();\n    this.abortController = new AbortController();\n\n    const initialContent = message.content;\n    const initialAnnotations = message.metadata?.unstable_annotations;\n    const initialData = message.metadata?.unstable_data;\n    const initialSteps = message.metadata?.steps;\n    const initialCustom = message.metadata?.custom;\n    const updateMessage = (m: Partial<ChatModelRunResult>) => {\n      const newSteps = m.metadata?.steps;\n      const steps = newSteps\n        ? [...(initialSteps ?? []), ...newSteps]\n        : undefined;\n\n      const newAnnotations = m.metadata?.unstable_annotations;\n      const newData = m.metadata?.unstable_data;\n      const annotations = newAnnotations\n        ? [...(initialAnnotations ?? []), ...newAnnotations]\n        : undefined;\n      const data = newData ? [...(initialData ?? []), ...newData] : undefined;\n\n      message = {\n        ...message,\n        ...(m.content\n          ? { content: [...initialContent, ...(m.content ?? [])] }\n          : undefined),\n        status: m.status ?? message.status,\n        ...(m.metadata\n          ? {\n              metadata: {\n                ...message.metadata,\n                ...(m.metadata.unstable_state !== undefined\n                  ? { unstable_state: m.metadata.unstable_state }\n                  : undefined),\n                ...(annotations\n                  ? { unstable_annotations: annotations }\n                  : undefined),\n                ...(data ? { unstable_data: data } : undefined),\n                ...(steps ? { steps } : undefined),\n                ...(m.metadata?.timing\n                  ? { timing: m.metadata.timing }\n                  : undefined),\n                ...(m.metadata?.custom\n                  ? {\n                      custom: {\n                        ...(initialCustom ?? {}),\n                        ...m.metadata.custom,\n                      },\n                    }\n                  : undefined),\n              },\n            }\n          : undefined),\n      };\n      this.repository.addOrUpdateMessage(parentId, message);\n      this._notifySubscribers();\n    };\n\n    const maxSteps = this._options.maxSteps ?? 2;\n\n    try {\n      const steps = message.metadata?.steps?.length ?? 0;\n      if (steps >= maxSteps) {\n        updateMessage({\n          status: {\n            type: \"incomplete\",\n            reason: \"tool-calls\",\n          },\n        });\n        return message;\n      }\n\n      updateMessage({\n        status: {\n          type: \"running\",\n        },\n      });\n\n      // Switch to the new message branch right after adding it for the first time\n      this.repository.resetHead(message.id);\n      this._notifySubscribers();\n\n      this._lastRunConfig = runConfig ?? {};\n      // unstable_composerMetadata is composer-only (stamped onto the outgoing\n      // message); never expose it to the chat-model adapter's run context.\n      const { unstable_composerMetadata: _, ...context } =\n        this.getModelContext();\n\n      runCallback =\n        runCallback ??\n        this.adapters.chatModel.run.bind(this.adapters.chatModel);\n\n      const abortSignal = this.abortController.signal;\n      const threadId = this._getThreadId?.();\n      const promiseOrGenerator = runCallback({\n        messages,\n        runConfig: this._lastRunConfig,\n        abortSignal,\n        context,\n        unstable_assistantMessageId: message.id,\n        unstable_threadId: threadId,\n        unstable_parentId: parentId,\n        unstable_getMessage() {\n          return message;\n        },\n      });\n\n      // handle async iterator for streaming results\n      if (Symbol.asyncIterator in promiseOrGenerator) {\n        for await (const r of promiseOrGenerator) {\n          if (abortSignal.aborted) {\n            updateMessage({\n              status: { type: \"incomplete\", reason: \"cancelled\" },\n            });\n            break;\n          }\n\n          updateMessage(r);\n        }\n      } else {\n        updateMessage(await promiseOrGenerator);\n      }\n\n      if (message.status.type === \"running\") {\n        updateMessage({\n          status: abortSignal.aborted\n            ? { type: \"incomplete\", reason: \"cancelled\" }\n            : { type: \"complete\", reason: \"unknown\" },\n        });\n      }\n    } catch (e) {\n      if (e instanceof AbortError) {\n        updateMessage({\n          status: { type: \"incomplete\", reason: \"cancelled\" },\n        });\n      } else if (e instanceof Error && e.name === \"AbortError\") {\n        updateMessage({\n          status: { type: \"incomplete\", reason: \"cancelled\" },\n        });\n      } else {\n        updateMessage({\n          status: {\n            type: \"incomplete\",\n            reason: \"error\",\n            error: toAssistantError(e),\n          },\n        });\n\n        throw e;\n      }\n    } finally {\n      this.abortController = null;\n\n      const history = this._options.adapters.history;\n      const item = {\n        parentId,\n        message,\n        runConfig: this._lastRunConfig,\n      };\n      const isTerminal =\n        message.status.type === \"complete\" ||\n        message.status.type === \"incomplete\";\n      const isPausing =\n        message.status.type === \"requires-action\" &&\n        !shouldContinue(message, this._options.unstable_humanToolNames);\n\n      // Pauses are written only for adapters that can rewrite the entry later;\n      // an append-only adapter would strand a half-finished run in history.\n      if (isTerminal || (isPausing && history?.update)) {\n        const write =\n          alreadyPersisted && history?.update\n            ? history.update.bind(history)\n            : history?.append.bind(history);\n        if (write) {\n          await this._chainHistoryWrite(message.id, () => write(item));\n        }\n      }\n    }\n    return message;\n  }\n\n  public detach() {\n    this._queue?.adapter.clear(\"cancel-run\");\n    const error = new AbortError(true);\n    this.abortController?.abort(error);\n    this.abortController = null;\n    this._suggestionsController?.abort();\n    this._suggestionsController = null;\n  }\n\n  public cancelRun() {\n    this._queue?.adapter.clear(\"cancel-run\");\n    const error = new AbortError(false);\n    this.abortController?.abort(error);\n    this.abortController = null;\n    this._suggestionsController?.abort();\n    this._suggestionsController = null;\n  }\n\n  public addToolResult({\n    messageId,\n    toolCallId,\n    result,\n    isError,\n    artifact,\n  }: AddToolResultOptions) {\n    const messageData = this.repository.getMessage(messageId);\n    const { parentId } = messageData;\n    let { message } = messageData;\n\n    if (message.role !== \"assistant\")\n      throw new Error(\"Tried to add tool result to non-assistant message\");\n\n    let added = false;\n    let found = false;\n    const newContent = message.content.map((c) => {\n      if (c.type !== \"tool-call\") return c;\n      if (c.toolCallId !== toolCallId) return c;\n      found = true;\n      if (!c.result) added = true;\n      return {\n        ...c,\n        result,\n        artifact,\n        isError,\n      };\n    });\n\n    if (!found)\n      throw new Error(\"Tried to add tool result to non-existing tool call\");\n\n    message = {\n      ...message,\n      content: newContent,\n    };\n    this.repository.addOrUpdateMessage(parentId, message);\n    this._notifySubscribers();\n\n    // a result may arrive mid-run or on a non-head message; the resume\n    // intentionally aborts any in-flight run, unlike respondToToolApproval\n    if (\n      added &&\n      shouldContinue(message, this._options.unstable_humanToolNames)\n    ) {\n      this._runLoop(parentId, message, this._lastRunConfig).catch(() => {});\n    } else if (added) {\n      this._persistPausedMessage(parentId, message);\n    }\n  }\n\n  public resumeToolCall(_options: ResumeToolCallOptions) {\n    throw new Error(\n      \"Local runtime does not support resuming tool calls. For human-in-the-loop tools, list the tool in unstable_humanToolNames and complete the call with addToolResult.\",\n    );\n  }\n\n  public respondToToolApproval({\n    approvalId,\n    approved,\n    optionId,\n    reason,\n  }: RespondToToolApprovalOptions) {\n    let message = this.repository\n      .getMessages()\n      .findLast(\n        (m): m is ThreadAssistantMessage =>\n          m.role === \"assistant\" &&\n          m.content.some(\n            (c) => c.type === \"tool-call\" && c.approval?.id === approvalId,\n          ),\n      );\n\n    if (!message)\n      throw new Error(\"Tried to respond to a non-existing tool approval\");\n\n    if (this.abortController !== null)\n      throw new Error(\n        \"Tried to respond to a tool approval while a run is in progress\",\n      );\n\n    if (message.status?.type !== \"requires-action\")\n      throw new Error(\n        \"Tried to respond to a tool approval on a message whose status is not requires-action\",\n      );\n\n    const target = message.content.find(\n      (c) => c.type === \"tool-call\" && c.approval?.id === approvalId,\n    );\n    if (target?.type !== \"tool-call\" || !target.approval)\n      throw new Error(\"Tried to respond to a non-existing tool approval\");\n    if (target.approval.resolution !== undefined)\n      throw new Error(\n        \"Tried to respond to a tool approval that was cancelled or expired\",\n      );\n    if (target.approval.approved !== undefined)\n      throw new Error(\"Tried to respond to an already decided tool approval\");\n\n    const targetApproval = target.approval;\n    const newContent = message.content.map((c) => {\n      if (c !== target) return c;\n      const approval = {\n        ...targetApproval,\n        approved,\n        ...(optionId != null && { optionId }),\n        ...(reason != null && { reason }),\n      };\n      if (approved) return { ...c, approval };\n      return {\n        ...c,\n        approval,\n        result: { error: reason || \"Tool approval denied\" },\n        isError: true,\n      };\n    });\n\n    message = { ...message, content: newContent };\n    const { parentId } = this.repository.getMessage(message.id);\n    this.repository.addOrUpdateMessage(parentId, message);\n    this._notifySubscribers();\n\n    if (\n      this.repository.headId === message.id &&\n      shouldContinue(message, this._options.unstable_humanToolNames)\n    ) {\n      this._runLoop(parentId, message, this._lastRunConfig).catch(() => {});\n    } else {\n      this._persistPausedMessage(parentId, message);\n    }\n  }\n}\n"],"mappings":";;;;;;;;;AAmCA,IAAMwB,aAAN,cAAyBC,MAAM;CAC7B,OAAgB;CAChBE;CAEAC,YAAYD,QAAiBE,SAAkB;EAC7C,MAAMA,OAAO;EACb,KAAKF,SAASA;CAChB;AACF;AAEA,IAAaG,yBAAb,cACUhB,sBAEV;CACE,eAA+B;EAC7BkB,gBAAgB;EAChBC,uBAAuB;EACvBC,MAAM;EACNC,QAAQ;EACRC,QAAQ;EACRC,QAAQ;EACRC,eAAe;EACfC,QAAQ;EACRC,WAAW;EACXC,OAAO;EACPC,aAAa;EACbC,UAAU;EACVC,OAAO;CACT;CAEA,kBAAkD;CAElD,SAAgD;CAChD,oBAA4B;CAE5B,iCAAyB,IAAIM,IAA2B;CAIxD,mBACEG,IACAC,OACe;EACf,MAAMC,QAAQ,KAAKN,eAAeO,IAAIH,EAAE,KAAKF,QAAQM,QAAQ,EAAA,CAAGC,KAC9DJ,OACAA,KACF;EACA,MAAMK,SAASJ,KAAKG,WACZ,CAAC,SACD,CAAC,CACT;EACA,KAAKT,eAAeW,IAAIP,IAAIM,MAAM;EAClC,OAAYD,WAAW;GACrB,IAAI,KAAKT,eAAeO,IAAIH,EAAE,MAAMM,QAClC,KAAKV,eAAed,OAAOkB,EAAE;EAEjC,CAAC;EACD,OAAOE;CACT;CAIA,sBACEO,UACAjC,SACA;EACA,IAAIA,QAAQkC,QAAQC,SAAS,mBAAmB;EAChD,MAAMC,UAAU,KAAKC,SAASC,SAASF;EACvC,IAAI,CAACA,SAASG,QAAQ;EACtB,MAAMA,SAASH,QAAQG,OAAOC,KAAKJ,OAAO;EAC1C,MAAMK,OAAO;GAAER;GAAUjC;GAAS0C,WAAW,KAAKC;EAAe;EACjE,KAAKpB,mBAAmBvB,QAAQwB,UAAUe,OAAOE,IAAI,CAAC,CAAC,CAACG,YAAY,CAAC,CAAC;CACxE;CAEA,aAA6B;CAC7B,iBAAiC;CAEjC,aAAqB;CACrB,IAAWI,YAAY;EACrB,OAAO,KAAKD;CACd;CAEA,eAAoD,CAAA;CACpD,yBAAyD;CACzD,IAAWI,cAA2C;EACpD,OAAO,KAAKF;CACd;CAEA,IAAWX,WAAW;EACpB,OAAO,KAAKD,SAASC;CACvB;CAEAvC,YACEqD,iBACAC,SACA;EACA,MAAMD,eAAe;EACrB,KAAKE,sBAAsBD,OAAO;CACpC;CAEA;CAEA,iBAAoC,CAAC;CAErC;CAEA,0BAAiCI,aAAuC;EACtE,KAAKF,eAAeE;CACtB;CAEA;CAEA,mCACEG,YACA;EACA,KAAKF,wBAAwBE;CAC/B;CAEA,IAAWC,SAAS,CAEpB;CAEA,sBAA6BR,SAAkC;EAC7D,IAAI,KAAKhB,aAAagB,SAAS;EAE/B,KAAKhB,WAAWgB;EAEhB,IAAIU,aAAa;EAEjB,MAAMC,WAAWX,QAAQf,UAAU5B,WAAWoD,KAAAA;EAC9C,IAAI,KAAK5D,aAAaQ,WAAWsD,UAAU;GACzC,KAAK9D,aAAaQ,SAASsD;GAC3BD,aAAa;EACf;EAEA,MAAME,aAAaZ,QAAQf,UAAU3B,cAAcmD,KAAAA;EACnD,IAAI,KAAK5D,aAAaS,cAAcsD,YAAY;GAC9C,KAAK/D,aAAaS,YAAYsD;GAC9BF,aAAa;EACf;EAEA,MAAMG,WAAWb,QAAQf,UAAU1B,UAAUkD,KAAAA;EAC7C,IAAI,KAAK5D,aAAaU,UAAUsD,UAAU;GACxC,KAAKhE,aAAaU,QAAQsD;GAC1BH,aAAa;EACf;EAEA,MAAMI,YAAYd,QAAQf,UAAUzB,gBAAgBiD,KAAAA;EACpD,IAAI,KAAK5D,aAAaW,gBAAgBsD,WAAW;GAC/C,KAAKjE,aAAaW,cAAcsD;GAChCJ,aAAa;EACf;EAEA,MAAMK,cAAcf,QAAQf,UAAUxB,aAAagD,KAAAA;EACnD,IAAI,KAAK5D,aAAaY,aAAasD,aAAa;GAC9C,KAAKlE,aAAaY,WAAWsD;GAC7BL,aAAa;EACf;EAEA,MAAMM,YAAYhB,QAAQf,UAAUF,SAAS9B,WAAWwD,KAAAA;EACxD,IAAI,KAAK5D,aAAaI,WAAW+D,WAAW;GAC1C,KAAKnE,aAAaI,SAAS+D;GAC3BN,aAAa;EACf;EAEA,MAAMO,WAAWjB,QAAQkB,gCAAgC;EACzD,IAAID,YAAY,CAAC,KAAKpD,QAAQ;GAC5B,KAAKA,SAAS3B,mBAAmB,EAC/BiF,MAAMxE,YAAY;IAGhB,KAAKmB,oBAAoB;IACzB,KAAUsD,WAAWzE,OAAO,CAAC,CAC1B0E,cAAc;KACb,KAAKvD,oBAAoB;KACzB,KAAKD,QAAQyD,WAAW;IAC1B,CAAC,CAAC,CACD/B,YAAY,CAAC,CAAC;GACnB,EACF,CAAC;GACD,KAAK1B,OAAO0D,gBAAgB,KAAKC,mBAAmB,CAAC;EACvD,OAAO,IAAI,CAACP,YAAY,KAAKpD,QAAQ;GACnC,KAAKA,OAAO4D,QAAQC,MAAM,YAAY;GACtC,KAAK7D,SAAS;EAChB;EACA,IAAI,KAAKhB,aAAaa,UAAUuD,UAAU;GACxC,KAAKpE,aAAaa,QAAQuD;GAC1BP,aAAa;EACf;EAEA,IAAIA,YAAY,KAAKc,mBAAmB;CAC1C;CAEA;CACA,kBAAyB;EACvB,IAAI,KAAKG,cAAc,OAAO,KAAKA;EAEnC,MAAME,UAAU,KAAK5C,SAASF,SAAS+C,KAAK,KAAK7D,QAAQM,QAAQ,IAAI;EAErE,KAAKmB,aAAa;EAClB,KAAK8B,mBAAmB;EAExB,KAAKG,eAAeE,QACjBrD,MAAMuD,SAAS;GACd,IAAI,CAACA,MAAM;GACX,KAAKC,WAAWC,OAAOF,IAAI;GAC3B,IAAIA,KAAKG,SAASC,SAAS,GACzB,KAAKC,kBAAkB;GAEzB,KAAKZ,mBAAmB;GAExB,MAAMa,SAAS,KAAKpD,SAASF,SAASsD,QAAQlD,KAC5C,KAAKF,SAASF,OAChB;GACA,IAAIgD,KAAKO,mBAAmBD,QAC1B,KAAKE,SACH;IACE3D,UAAU,KAAKoD,WAAWQ;IAC1BC,UAAU,KAAKT,WAAWQ;IAC1BnD,WAAW,KAAKC;GAClB,GACA+C,MACF,CAAC,CAAC9C,YAAY,CAAC,CAAC;EAEpB,CAAC,CAAC,CACD8B,cAAc;GACb,KAAK3B,aAAa;GAClB,KAAK8B,mBAAmB;EAC1B,CAAC;EAEH,OAAO,KAAKG;CACd;CAEA,MAAae,OAAO/F,SAAuC;EACzD,MAAMgG,SAAShG,QAAQiC,cAAc,KAAKsD,SAASU,GAAG,EAAE,CAAC,EAAEzE,MAAM;EACjE,MAAM0E,UAAUlG,QAAQ4F,YAAY5F,QAAQmG,SAAS;EACrD,IAAI,KAAKjF,UAAUgF,WAAWF,QAAQ;GACpC,KAAK9E,OAAO4D,QAAQsB,QAAQpG,SAAS,EAAEqG,OAAOrG,QAAQqG,SAAS,MAAM,CAAC;GACtE;EACF;EACA,IAAI,KAAKnF,UAAU,CAAC8E,QAAQ,KAAK9E,OAAO4D,QAAQC,MAAM,MAAM;EAC5D,OAAO,KAAKN,WAAWzE,OAAO;CAChC;CAEA,gBAAkD;EAGhD,OAAO,KAAKkB,QAAQ4D,QAAQyB,SAAS9G;CACvC;CAEA,eAAsBgH,aAA2B;EAC/C,KAAKvF,QAAQ4D,QAAQuB,MAAMI,WAAW;CACxC;CAEA,gBAAuBA,aAA2B;EAChD,KAAKvF,QAAQ4D,QAAQ6B,OAAOF,WAAW;CACzC;CAEA,MAAchC,WAAWzE,SAAuC;EAC9D,KAAKyF,kBAAkB;EAEvB,MAAMmB,cAAc,KAAKlD,wBAAwB;EACjD,IAAIkD,aACF,MAAMA;EAGR,MAAMC,aAAa1I,sBAAsB6B,SAAS5B,WAAW,GAAG;GAC9D+D,MAAM;GACN2E,QAAQ;EACV,CAAC;EACD,KAAKzB,WAAW0B,mBAAmB/G,QAAQiC,UAAU4E,UAAU;EAC/D,KAAKxE,SAASC,SAASF,SAAS2D,OAAO;GACrC9D,UAAUjC,QAAQiC;GAClBjC,SAAS6G;GACT,GAAI7G,QAAQ0C,cAAcoB,KAAAA,KAAa,EAAEpB,WAAW1C,QAAQ0C,UAAU;EACxE,CAAC;EAGD,IADiB1C,QAAQ4F,YAAY5F,QAAQmG,SAAS,QAEpD,MAAM,KAAKP,SAAS;GAClB3D,UAAU4E,WAAWrF;GACrBsE,UAAU9F,QAAQ8F;GAClBpD,WAAW1C,QAAQ0C,aAAa,CAAC;EACnC,CAAC;OACI;GACL,KAAK2C,WAAW2B,UAAUH,WAAWrF,EAAE;GACvC,KAAKqD,mBAAmB;EAC1B;CACF;CAEA,MAAaoC,cAAcC,WAAkC;EAC3D,MAAMpC,UAAU,KAAKzC,SAASC,SAASF;EACvC,IAAI,CAAC0C,SAASxE,QACZ,MAAM,IAAIV,MAAM,6CAA6C;EAE/D,MAAM2F,WAAW,KAAKF,WAAW8B,YAAY;EAC7C,MAAMC,eAAe7B,SAAS8B,WAAWC,MAAMA,EAAE9F,OAAO0F,SAAS;EACjE,IAAIE,iBAAiB,IAAI,MAAM,IAAIxH,MAAM,oBAAoB;EAE7D,MAAMI,UAAUuF,SAAS6B;EAEzB,MAAMb,QAAQ,CAAC;GAAEtE,UADAsD,SAAS6B,eAAe,EAAE,EAAE5F,MAAM;GACxBxB;EAAQ,CAAC;EAEpC,MAAM8E,QAAQxE,OAAOiG,KAAK;EAE1B,KAAKlB,WAAW4B,cAAcC,SAAS;EACvC,KAAKrC,mBAAmB;CAC1B;CAEA,UAAiB,EAAE2C,QAAQ,GAAGC,eAA+C;EAC3E,IAAI,CAACD,QACH,MAAM,IAAI5H,MAAM,kDAAkD;EACpE,OAAO,KAAKgG,SAAS6B,aAAaD,MAAM;CAC1C;CAEA,sBAAkC;EAChC,MAAM,IAAI5H,MAAM,qDAAqD;CACvE;CAEA,sBAAmC;EACjC,MAAM,IAAIA,MAAM,qDAAqD;CACvE;CAEA,MAAagG,SACX,EAAE3D,UAAUS,aACZkF,aACe;EACf,KAAKnC,kBAAkB;EAIvB,MAAMzF,UAAkC;GACtCwB,IAFSpD,WAEToD;GACA2E,MAAM;GACNjE,QAAQ,EAAEC,MAAM,UAAU;GAC1B0F,SAAS,CAAA;GACTC,UAAU;IACRC,gBAAgB,KAAKC;IACrBC,sBAAsB,CAAA;IACtBC,eAAe,CAAA;IACfC,OAAO,CAAA;IACPC,QAAQ,CAAC;GACX;GACAC,2BAAW,IAAIC,KAAK;EACtB;EAEA,OAAO,KAAKC,SAAStG,UAAUjC,SAAS0C,WAAWkF,WAAW;CAChE;CAEA,MAAcW,SACZtG,UACAjC,SACA0C,WACAkF,aACe;EACf,KAAKY,wBAAwB,YAAY,CAAC,CAAC;EAI3C,MAAMC,mBACJzI,QAAQkC,QAAQC,SAAS,qBACzB,KAAKE,SAASC,SAASF,SAASG,WAAWuB,KAAAA;EAE7C,IAAI;GAEF,KAAK5C,QAAQwH,WAAW;GACxB,KAAKzF,eAAe,CAAA;GACpB,KAAKC,wBAAwByF,MAAM;GACnC,KAAKzF,yBAAyB;GAC9B,KAAK2B,mBAAmB;GAExB,GAAG;IACD7E,UAAU,MAAM,KAAK4I,iBACnB3G,UACAjC,SACA0C,WACA+F,kBACAb,WACF;IACAA,cAAc9D,KAAAA;GAChB,SAASvF,eAAeyB,SAAS,KAAKqC,SAASwG,uBAAuB;EACxE,UAAU;GACR,KAAKL,wBAAwB,UAAU,CAAC,CAAC;GAGzC,IAAI,CAAC,KAAKrH,mBACR2H,qBAAqB,KAAK5H,QAAQyD,WAAW,CAAC;EAElD;EAEA,IACE,KAAKrC,SAASyG,cACd/I,QAAQkC,QAAQC,SAAS,mBACzB;GACA,KAAKe,yBAAyB,IAAIjC,gBAAgB;GAClD,MAAM+H,SAAS,KAAK9F,uBAAuB8F;GAC3C,MAAMlE,UAAU,KAAKxC,SAASyG;GAC9B,CAAM,YAAY;IAChB,IAAI;KAKF,MAAMtK,wBAJqBqG,QAAQoE,SAAS;MAC1C3D,UAAU,KAAKA;MACfyD;KACF,CAC8BC,GAAoB;MAChDD;MACAG,WAAWC,MAAM;OACf,KAAKnG,eAAemG;OACpB,KAAKvE,mBAAmB;MAC1B;KACF,CAAC;IACH,QAAQ,CAAC;GACX,EAAA,CAAG;EACL;CACF;CAEA,MAAc+D,iBACZ3G,UACAjC,SACA0C,WACA+F,kBACAb,aACA;EACA,MAAMrC,WAAWtD,WAAW,KAAKoD,WAAW8B,YAAYlF,QAAQ,IAAI,CAAA;EAGpE,KAAKjB,iBAAiB2H,MAAM;EAC5B,KAAK3H,kBAAkB,IAAIC,gBAAgB;EAE3C,MAAMoI,iBAAiBrJ,QAAQ6H;EAC/B,MAAMyB,qBAAqBtJ,QAAQ8H,UAAUG;EAC7C,MAAMsB,cAAcvJ,QAAQ8H,UAAUI;EACtC,MAAMsB,eAAexJ,QAAQ8H,UAAUK;EACvC,MAAMsB,gBAAgBzJ,QAAQ8H,UAAUM;EACxC,MAAMsB,iBAAiBpC,MAAmC;GACxD,MAAMsC,WAAWtC,EAAEQ,UAAUK;GAC7B,MAAMA,QAAQyB,WACV,CAAC,GAAIJ,gBAAgB,CAAA,GAAK,GAAGI,QAAQ,IACrC9F,KAAAA;GAEJ,MAAM+F,iBAAiBvC,EAAEQ,UAAUG;GACnC,MAAM6B,UAAUxC,EAAEQ,UAAUI;GAC5B,MAAM6B,cAAcF,iBAChB,CAAC,GAAIP,sBAAsB,CAAA,GAAK,GAAGO,cAAc,IACjD/F,KAAAA;GACJ,MAAMkG,OAAOF,UAAU,CAAC,GAAIP,eAAe,CAAA,GAAK,GAAGO,OAAO,IAAIhG,KAAAA;GAE9D9D,UAAU;IACR,GAAGA;IACH,GAAIsH,EAAEO,UACF,EAAEA,SAAS,CAAC,GAAGwB,gBAAgB,GAAI/B,EAAEO,WAAW,CAAA,CAAG,EAAE,IACrD/D,KAAAA;IACJ5B,QAAQoF,EAAEpF,UAAUlC,QAAQkC;IAC5B,GAAIoF,EAAEQ,WACF,EACEA,UAAU;KACR,GAAG9H,QAAQ8H;KACX,GAAIR,EAAEQ,SAASC,mBAAmBjE,KAAAA,IAC9B,EAAEiE,gBAAgBT,EAAEQ,SAASC,eAAe,IAC5CjE,KAAAA;KACJ,GAAIiG,cACA,EAAE9B,sBAAsB8B,YAAY,IACpCjG,KAAAA;KACJ,GAAIkG,OAAO,EAAE9B,eAAe8B,KAAK,IAAIlG,KAAAA;KACrC,GAAIqE,QAAQ,EAAEA,MAAM,IAAIrE,KAAAA;KACxB,GAAIwD,EAAEQ,UAAUmC,SACZ,EAAEA,QAAQ3C,EAAEQ,SAASmC,OAAO,IAC5BnG,KAAAA;KACJ,GAAIwD,EAAEQ,UAAUM,SACZ,EACEA,QAAQ;MACN,GAAIqB,iBAAiB,CAAC;MACtB,GAAGnC,EAAEQ,SAASM;KAChB,EACF,IACAtE,KAAAA;IACN,EACF,IACAA,KAAAA;GACN;GACA,KAAKuB,WAAW0B,mBAAmB9E,UAAUjC,OAAO;GACpD,KAAK6E,mBAAmB;EAC1B;EAEA,MAAMqF,WAAW,KAAK7H,SAAS6H,YAAY;EAE3C,IAAI;GAEF,KADclK,QAAQ8H,UAAUK,OAAO3C,UAAU,MACpC0E,UAAU;IACrBR,cAAc,EACZxH,QAAQ;KACNC,MAAM;KACN2E,QAAQ;IACV,EACF,CAAC;IACD,OAAO9G;GACT;GAEA0J,cAAc,EACZxH,QAAQ,EACNC,MAAM,UACR,EACF,CAAC;GAGD,KAAKkD,WAAW2B,UAAUhH,QAAQwB,EAAE;GACpC,KAAKqD,mBAAmB;GAExB,KAAKlC,iBAAiBD,aAAa,CAAC;GAGpC,MAAM,EAAEyH,2BAA2BC,GAAG,GAAGC,YACvC,KAAKC,gBAAgB;GAEvB1C,cACEA,eACA,KAAKtF,SAASiI,UAAU/F,IAAIhC,KAAK,KAAKF,SAASiI,SAAS;GAE1D,MAAMC,cAAc,KAAKxJ,gBAAgBgI;GACzC,MAAMyB,WAAW,KAAKlH,eAAe;GACrC,MAAM0F,qBAAqBrB,YAAY;IACrCrC;IACA7C,WAAW,KAAKC;IAChB6H;IACAH;IACAK,6BAA6B1K,QAAQwB;IACrCmJ,mBAAmBF;IACnBG,mBAAmB3I;IACnB4I,sBAAsB;KACpB,OAAO7K;IACT;GACF,CAAC;GAGD,IAAI8K,OAAOC,iBAAiB9B,oBAC1B,WAAW,MAAMG,KAAKH,oBAAoB;IACxC,IAAIuB,YAAYQ,SAAS;KACvBtB,cAAc,EACZxH,QAAQ;MAAEC,MAAM;MAAc2E,QAAQ;KAAY,EACpD,CAAC;KACD;IACF;IAEA4C,cAAcN,CAAC;GACjB;QAEAM,cAAc,MAAMT,kBAAkB;GAGxC,IAAIjJ,QAAQkC,OAAOC,SAAS,WAC1BuH,cAAc,EACZxH,QAAQsI,YAAYQ,UAChB;IAAE7I,MAAM;IAAc2E,QAAQ;GAAY,IAC1C;IAAE3E,MAAM;IAAY2E,QAAQ;GAAU,EAC5C,CAAC;EAEL,SAASmE,GAAG;GACV,IAAIA,aAAatL,YACf+J,cAAc,EACZxH,QAAQ;IAAEC,MAAM;IAAc2E,QAAQ;GAAY,EACpD,CAAC;QACI,IAAImE,aAAarL,SAASqL,EAAEpL,SAAS,cAC1C6J,cAAc,EACZxH,QAAQ;IAAEC,MAAM;IAAc2E,QAAQ;GAAY,EACpD,CAAC;QACI;IACL4C,cAAc,EACZxH,QAAQ;KACNC,MAAM;KACN2E,QAAQ;KACRoE,OAAO7L,iBAAiB4L,CAAC;IAC3B,EACF,CAAC;IAED,MAAMA;GACR;EACF,UAAU;GACR,KAAKjK,kBAAkB;GAEvB,MAAMoB,UAAU,KAAKC,SAASC,SAASF;GACvC,MAAMK,OAAO;IACXR;IACAjC;IACA0C,WAAW,KAAKC;GAClB;GACA,MAAMwI,aACJnL,QAAQkC,OAAOC,SAAS,cACxBnC,QAAQkC,OAAOC,SAAS;GAC1B,MAAMiJ,YACJpL,QAAQkC,OAAOC,SAAS,qBACxB,CAAC5D,eAAeyB,SAAS,KAAKqC,SAASwG,uBAAuB;GAIhE,IAAIsC,cAAeC,aAAahJ,SAASG,QAAS;IAChD,MAAMd,QACJgH,oBAAoBrG,SAASG,SACzBH,QAAQG,OAAOC,KAAKJ,OAAO,IAC3BA,SAAS2D,OAAOvD,KAAKJ,OAAO;IAClC,IAAIX,OACF,MAAM,KAAKF,mBAAmBvB,QAAQwB,UAAUC,MAAMgB,IAAI,CAAC;GAE/D;EACF;EACA,OAAOzC;CACT;CAEA,SAAgB;EACd,KAAKkB,QAAQ4D,QAAQC,MAAM,YAAY;EACvC,MAAMmG,QAAQ,IAAIvL,WAAW,IAAI;EACjC,KAAKqB,iBAAiB2H,MAAMuC,KAAK;EACjC,KAAKlK,kBAAkB;EACvB,KAAKkC,wBAAwByF,MAAM;EACnC,KAAKzF,yBAAyB;CAChC;CAEA,YAAmB;EACjB,KAAKhC,QAAQ4D,QAAQC,MAAM,YAAY;EACvC,MAAMmG,QAAQ,IAAIvL,WAAW,KAAK;EAClC,KAAKqB,iBAAiB2H,MAAMuC,KAAK;EACjC,KAAKlK,kBAAkB;EACvB,KAAKkC,wBAAwByF,MAAM;EACnC,KAAKzF,yBAAyB;CAChC;CAEA,cAAqB,EACnBgE,WACAqE,YACAC,QACAC,SACAC,YACuB;EACvB,MAAMC,cAAc,KAAKtG,WAAWuG,WAAW1E,SAAS;EACxD,MAAM,EAAEjF,aAAa0J;EACrB,IAAI,EAAE3L,YAAY2L;EAElB,IAAI3L,QAAQmG,SAAS,aACnB,MAAM,IAAIvG,MAAM,mDAAmD;EAErE,IAAIiM,QAAQ;EACZ,IAAIC,QAAQ;EACZ,MAAMC,aAAa/L,QAAQ6H,QAAQmE,KAAKC,MAAM;GAC5C,IAAIA,EAAE9J,SAAS,aAAa,OAAO8J;GACnC,IAAIA,EAAEV,eAAeA,YAAY,OAAOU;GACxCH,QAAQ;GACR,IAAI,CAACG,EAAET,QAAQK,QAAQ;GACvB,OAAO;IACL,GAAGI;IACHT;IACAE;IACAD;GACF;EACF,CAAC;EAED,IAAI,CAACK,OACH,MAAM,IAAIlM,MAAM,oDAAoD;EAEtEI,UAAU;GACR,GAAGA;GACH6H,SAASkE;EACX;EACA,KAAK1G,WAAW0B,mBAAmB9E,UAAUjC,OAAO;EACpD,KAAK6E,mBAAmB;EAIxB,IACEgH,SACAtN,eAAeyB,SAAS,KAAKqC,SAASwG,uBAAuB,GAE7D,KAAKN,SAAStG,UAAUjC,SAAS,KAAK2C,cAAc,CAAC,CAACC,YAAY,CAAC,CAAC;OAC/D,IAAIiJ,OACT,KAAK7J,sBAAsBC,UAAUjC,OAAO;CAEhD;CAEA,eAAsBqC,UAAiC;EACrD,MAAM,IAAIzC,MACR,qKACF;CACF;CAEA,sBAA6B,EAC3BwM,YACAC,UACAC,UACAxF,UAC+B;EAC/B,IAAI9G,UAAU,KAAKqF,WAChB8B,YAAY,CAAC,CACboF,UACEjF,MACCA,EAAEnB,SAAS,eACXmB,EAAEO,QAAQ2E,MACPP,MAAMA,EAAE9J,SAAS,eAAe8J,EAAEQ,UAAUjL,OAAO4K,UACtD,CACJ;EAEF,IAAI,CAACpM,SACH,MAAM,IAAIJ,MAAM,kDAAkD;EAEpE,IAAI,KAAKoB,oBAAoB,MAC3B,MAAM,IAAIpB,MACR,gEACF;EAEF,IAAII,QAAQkC,QAAQC,SAAS,mBAC3B,MAAM,IAAIvC,MACR,sFACF;EAEF,MAAM8M,SAAS1M,QAAQ6H,QAAQ8E,MAC5BV,MAAMA,EAAE9J,SAAS,eAAe8J,EAAEQ,UAAUjL,OAAO4K,UACtD;EACA,IAAIM,QAAQvK,SAAS,eAAe,CAACuK,OAAOD,UAC1C,MAAM,IAAI7M,MAAM,kDAAkD;EACpE,IAAI8M,OAAOD,SAASG,eAAe9I,KAAAA,GACjC,MAAM,IAAIlE,MACR,mEACF;EACF,IAAI8M,OAAOD,SAASJ,aAAavI,KAAAA,GAC/B,MAAM,IAAIlE,MAAM,sDAAsD;EAExE,MAAMiN,iBAAiBH,OAAOD;EAC9B,MAAMV,aAAa/L,QAAQ6H,QAAQmE,KAAKC,MAAM;GAC5C,IAAIA,MAAMS,QAAQ,OAAOT;GACzB,MAAMQ,WAAW;IACf,GAAGI;IACHR;IACA,GAAIC,YAAY,QAAQ,EAAEA,SAAS;IACnC,GAAIxF,UAAU,QAAQ,EAAEA,OAAO;GACjC;GACA,IAAIuF,UAAU,OAAO;IAAE,GAAGJ;IAAGQ;GAAS;GACtC,OAAO;IACL,GAAGR;IACHQ;IACAjB,QAAQ,EAAEN,OAAOpE,UAAU,uBAAuB;IAClD2E,SAAS;GACX;EACF,CAAC;EAEDzL,UAAU;GAAE,GAAGA;GAAS6H,SAASkE;EAAW;EAC5C,MAAM,EAAE9J,aAAa,KAAKoD,WAAWuG,WAAW5L,QAAQwB,EAAE;EAC1D,KAAK6D,WAAW0B,mBAAmB9E,UAAUjC,OAAO;EACpD,KAAK6E,mBAAmB;EAExB,IACE,KAAKQ,WAAWQ,WAAW7F,QAAQwB,MACnCjD,eAAeyB,SAAS,KAAKqC,SAASwG,uBAAuB,GAE7D,KAAKN,SAAStG,UAAUjC,SAAS,KAAK2C,cAAc,CAAC,CAACC,YAAY,CAAC,CAAC;OAEpE,KAAKZ,sBAAsBC,UAAUjC,OAAO;CAEhD;AACF"}
+{"version":3,"file":"local-thread-runtime-core.js","names":["fromThreadMessageLike","generateId","ChatModelAdapter","ChatModelRunResult","shouldContinue","LocalRuntimeOptionsBase","consumeSuggestionResult","AddToolResultOptions","ResumeToolCallOptions","RespondToToolApprovalOptions","ThreadSuggestion","ThreadRuntimeCore","StartRunConfig","ResumeRunConfig","BaseThreadRuntimeCore","AppendMessage","ThreadAssistantMessage","RunConfig","toAssistantError","ModelContextProvider","createMessageQueue","MessageQueueController","EMPTY_QUEUE_ITEMS","QueueItemState","AbortError","Error","name","detach","constructor","message","LocalThreadRuntimeCore","capabilities","switchToBranch","switchBranchDuringRun","edit","delete","reload","cancel","unstable_copy","speech","dictation","voice","attachments","feedback","queue","abortController","AbortController","_queue","_queueRunInFlight","_historyWrites","Map","Promise","_chainHistoryWrite","id","write","next","get","resolve","then","stored","set","_persistPausedMessage","parentId","status","type","history","_options","adapters","update","bind","item","runConfig","_lastRunConfig","catch","isDisabled","isSendDisabled","_isLoading","isLoading","_suggestions","_suggestionsController","suggestions","contextProvider","options","__internal_setOptions","_getThreadId","__internal_setGetThreadId","getThreadId","_getInitializePromise","__internal_setGetInitializePromise","getPromise","extras","undefined","hasUpdates","canSpeak","canDictate","canVoice","canAttach","canFeedback","canDelete","canQueue","unstable_enableMessageQueue","run","_runAppend","finally","notifyIdle","subscribe","_notifySubscribers","adapter","clear","_loadPromise","__internal_load","promise","load","repo","repository","import","messages","length","ensureInitialized","resume","unstable_resume","startRun","headId","sourceId","append","isTail","at","willRun","role","enqueue","steer","getQueueItems","items","steerQueueItem","queueItemId","removeQueueItem","remove","initPromise","newMessage","reason","addOrUpdateMessage","resetHead","deleteMessage","messageId","getMessages","messageIndex","findIndex","m","resumeRun","stream","startConfig","exportExternalState","importExternalState","runCallback","content","metadata","unstable_state","state","unstable_annotations","unstable_data","steps","custom","createdAt","Date","_runLoop","_notifyEventSubscribers","alreadyPersisted","notifyBusy","abort","performRoundtrip","unstable_humanToolNames","queueMicrotask","suggestion","signal","promiseOrGenerator","generate","onUpdate","r","initialContent","initialAnnotations","initialData","initialSteps","initialCustom","updateMessage","Partial","newSteps","newAnnotations","newData","annotations","data","timing","maxSteps","unstable_composerMetadata","_","context","getModelContext","chatModel","abortSignal","threadId","unstable_assistantMessageId","unstable_threadId","unstable_parentId","unstable_getMessage","Symbol","asyncIterator","aborted","e","error","isTerminal","isPausing","cancelRun","addToolResult","toolCallId","result","isError","artifact","messageData","getMessage","added","found","newContent","map","c","resumeToolCall","respondToToolApproval","approvalId","approved","optionId","findLast","some","approval","target","find","resolution","targetApproval"],"sources":["../../../src/runtimes/local/local-thread-runtime-core.ts"],"sourcesContent":["import { fromThreadMessageLike } from \"../../runtime/utils/thread-message-like\";\nimport { generateId } from \"../../utils/id\";\nimport type {\n  ChatModelAdapter,\n  ChatModelRunResult,\n} from \"../../runtime/utils/chat-model-adapter\";\nimport { shouldContinue } from \"./should-continue\";\nimport type { LocalRuntimeOptionsBase } from \"./local-runtime-options\";\nimport { consumeSuggestionResult } from \"../../adapters/suggestion\";\nimport type {\n  AddToolResultOptions,\n  ResumeToolCallOptions,\n  RespondToToolApprovalOptions,\n  ThreadSuggestion,\n  ThreadRuntimeCore,\n  StartRunConfig,\n  ResumeRunConfig,\n} from \"../../runtime/interfaces/thread-runtime-core\";\nimport { BaseThreadRuntimeCore } from \"../../runtime/base/base-thread-runtime-core\";\nimport type {\n  AppendMessage,\n  ThreadAssistantMessage,\n} from \"../../types/message\";\nimport type { RunConfig } from \"../../types/message\";\nimport { toAssistantError } from \"../../types/error\";\nimport type { ModelContextProvider } from \"../../model-context/types\";\nimport {\n  createMessageQueue,\n  type MessageQueueController,\n} from \"../../runtime/queue/message-queue\";\nimport {\n  EMPTY_QUEUE_ITEMS,\n  type QueueItemState,\n} from \"../../store/scopes/queue-item\";\n\nclass AbortError extends Error {\n  override name = \"AbortError\";\n  detach: boolean;\n\n  constructor(detach: boolean, message?: string) {\n    super(message);\n    this.detach = detach;\n  }\n}\n\nexport class LocalThreadRuntimeCore\n  extends BaseThreadRuntimeCore\n  implements ThreadRuntimeCore\n{\n  public readonly capabilities = {\n    switchToBranch: true,\n    switchBranchDuringRun: true,\n    edit: true,\n    delete: false,\n    reload: true,\n    cancel: true,\n    unstable_copy: true,\n    speech: false,\n    dictation: false,\n    voice: false,\n    attachments: false,\n    feedback: false,\n    queue: false,\n  };\n\n  private abortController: AbortController | null = null;\n\n  private _queue: MessageQueueController | null = null;\n  private _queueRunInFlight = false;\n\n  private _historyWrites = new Map<string, Promise<void>>();\n\n  // Writes for one message id must land in issue order; an earlier paused\n  // snapshot arriving after the terminal write would resurrect the pause.\n  private _chainHistoryWrite(\n    id: string,\n    write: () => Promise<void>,\n  ): Promise<void> {\n    const next = (this._historyWrites.get(id) ?? Promise.resolve()).then(\n      write,\n      write,\n    );\n    const stored = next.then(\n      () => {},\n      () => {},\n    );\n    this._historyWrites.set(id, stored);\n    void stored.then(() => {\n      if (this._historyWrites.get(id) === stored) {\n        this._historyWrites.delete(id);\n      }\n    });\n    return next;\n  }\n\n  // A decision recorded on a still-paused message must reach history before\n  // the run resumes, or a refresh would restore the message without it.\n  private _persistPausedMessage(\n    parentId: string | null,\n    message: ThreadAssistantMessage,\n  ) {\n    if (message.status?.type !== \"requires-action\") return;\n    const history = this._options.adapters.history;\n    if (!history?.update) return;\n    const update = history.update.bind(history);\n    const item = { parentId, message, runConfig: this._lastRunConfig };\n    this._chainHistoryWrite(message.id, () => update(item)).catch(() => {});\n  }\n\n  public readonly isDisabled = false;\n  public readonly isSendDisabled = false;\n\n  private _isLoading = false;\n  public get isLoading() {\n    return this._isLoading;\n  }\n\n  private _suggestions: readonly ThreadSuggestion[] = [];\n  private _suggestionsController: AbortController | null = null;\n  public get suggestions(): readonly ThreadSuggestion[] {\n    return this._suggestions;\n  }\n\n  public get adapters() {\n    return this._options.adapters;\n  }\n\n  constructor(\n    contextProvider: ModelContextProvider,\n    options: LocalRuntimeOptionsBase,\n  ) {\n    super(contextProvider);\n    this.__internal_setOptions(options);\n  }\n\n  private _options!: LocalRuntimeOptionsBase;\n\n  private _lastRunConfig: RunConfig = {};\n\n  private _getThreadId?: () => string | undefined;\n\n  public __internal_setGetThreadId(getThreadId: () => string | undefined) {\n    this._getThreadId = getThreadId;\n  }\n\n  private _getInitializePromise?: () => Promise<unknown> | undefined;\n\n  public __internal_setGetInitializePromise(\n    getPromise: () => Promise<unknown> | undefined,\n  ) {\n    this._getInitializePromise = getPromise;\n  }\n\n  public get extras() {\n    return undefined;\n  }\n\n  public __internal_setOptions(options: LocalRuntimeOptionsBase) {\n    if (this._options === options) return;\n\n    this._options = options;\n\n    let hasUpdates = false;\n\n    const canSpeak = options.adapters?.speech !== undefined;\n    if (this.capabilities.speech !== canSpeak) {\n      this.capabilities.speech = canSpeak;\n      hasUpdates = true;\n    }\n\n    const canDictate = options.adapters?.dictation !== undefined;\n    if (this.capabilities.dictation !== canDictate) {\n      this.capabilities.dictation = canDictate;\n      hasUpdates = true;\n    }\n\n    const canVoice = options.adapters?.voice !== undefined;\n    if (this.capabilities.voice !== canVoice) {\n      this.capabilities.voice = canVoice;\n      hasUpdates = true;\n    }\n\n    const canAttach = options.adapters?.attachments !== undefined;\n    if (this.capabilities.attachments !== canAttach) {\n      this.capabilities.attachments = canAttach;\n      hasUpdates = true;\n    }\n\n    const canFeedback = options.adapters?.feedback !== undefined;\n    if (this.capabilities.feedback !== canFeedback) {\n      this.capabilities.feedback = canFeedback;\n      hasUpdates = true;\n    }\n\n    const canDelete = options.adapters?.history?.delete !== undefined;\n    if (this.capabilities.delete !== canDelete) {\n      this.capabilities.delete = canDelete;\n      hasUpdates = true;\n    }\n\n    const canQueue = options.unstable_enableMessageQueue === true;\n    if (canQueue && !this._queue) {\n      this._queue = createMessageQueue({\n        run: (message) => {\n          // release the queue when the dispatch settles, even if it rejects\n          // before reaching startRun's finally, so a failure can't deadlock it\n          this._queueRunInFlight = true;\n          void this._runAppend(message)\n            .finally(() => {\n              this._queueRunInFlight = false;\n              this._queue?.notifyIdle();\n            })\n            .catch(() => {});\n        },\n      });\n      this._queue.subscribe(() => this._notifySubscribers());\n    } else if (!canQueue && this._queue) {\n      this._queue.adapter.clear(\"cancel-run\");\n      this._queue = null;\n    }\n    if (this.capabilities.queue !== canQueue) {\n      this.capabilities.queue = canQueue;\n      hasUpdates = true;\n    }\n\n    if (hasUpdates) this._notifySubscribers();\n  }\n\n  private _loadPromise: Promise<void> | undefined;\n  public __internal_load() {\n    if (this._loadPromise) return this._loadPromise;\n\n    const promise = this.adapters.history?.load() ?? Promise.resolve(null);\n\n    this._isLoading = true;\n    this._notifySubscribers();\n\n    this._loadPromise = promise\n      .then((repo) => {\n        if (!repo) return;\n        this.repository.import(repo);\n        if (repo.messages.length > 0) {\n          this.ensureInitialized();\n        }\n        this._notifySubscribers();\n\n        const resume = this.adapters.history?.resume?.bind(\n          this.adapters.history,\n        );\n        if (repo.unstable_resume && resume) {\n          this.startRun(\n            {\n              parentId: this.repository.headId,\n              sourceId: this.repository.headId,\n              runConfig: this._lastRunConfig,\n            },\n            resume,\n          ).catch(() => {});\n        }\n      })\n      .finally(() => {\n        this._isLoading = false;\n        this._notifySubscribers();\n      });\n\n    return this._loadPromise;\n  }\n\n  public async append(message: AppendMessage): Promise<void> {\n    const isTail = message.parentId === (this.messages.at(-1)?.id ?? null);\n    const willRun = message.startRun ?? message.role === \"user\";\n    if (this._queue && willRun && isTail) {\n      this._queue.adapter.enqueue(message, { steer: message.steer ?? false });\n      return;\n    }\n    if (this._queue && !isTail) this._queue.adapter.clear(\"edit\");\n    return this._runAppend(message);\n  }\n\n  public getQueueItems(): readonly QueueItemState[] {\n    // Reads can arrive during base-thread construction, before the queue field\n    // is assigned, so guard against the unset field.\n    return this._queue?.adapter.items ?? EMPTY_QUEUE_ITEMS;\n  }\n\n  public steerQueueItem(queueItemId: string): void {\n    this._queue?.adapter.steer(queueItemId);\n  }\n\n  public removeQueueItem(queueItemId: string): void {\n    this._queue?.adapter.remove(queueItemId);\n  }\n\n  private async _runAppend(message: AppendMessage): Promise<void> {\n    this.ensureInitialized();\n\n    const initPromise = this._getInitializePromise?.();\n    if (initPromise) {\n      await initPromise;\n    }\n\n    const newMessage = fromThreadMessageLike(message, generateId(), {\n      type: \"complete\",\n      reason: \"unknown\",\n    });\n    this.repository.addOrUpdateMessage(message.parentId, newMessage);\n    this._options.adapters.history?.append({\n      parentId: message.parentId,\n      message: newMessage,\n      ...(message.runConfig !== undefined && { runConfig: message.runConfig }),\n    });\n\n    const startRun = message.startRun ?? message.role === \"user\";\n    if (startRun) {\n      await this.startRun({\n        parentId: newMessage.id,\n        sourceId: message.sourceId,\n        runConfig: message.runConfig ?? {},\n      });\n    } else {\n      this.repository.resetHead(newMessage.id);\n      this._notifySubscribers();\n    }\n  }\n\n  public async deleteMessage(messageId: string): Promise<void> {\n    const adapter = this._options.adapters.history;\n    if (!adapter?.delete)\n      throw new Error(\"Runtime does not support deleting messages.\");\n\n    const messages = this.repository.getMessages();\n    const messageIndex = messages.findIndex((m) => m.id === messageId);\n    if (messageIndex === -1) throw new Error(\"Message not found.\");\n\n    const message = messages[messageIndex]!;\n    const parentId = messages[messageIndex - 1]?.id ?? null;\n    const items = [{ parentId, message }];\n\n    await adapter.delete(items);\n\n    this.repository.deleteMessage(messageId);\n    this._notifySubscribers();\n  }\n\n  public resumeRun({ stream, ...startConfig }: ResumeRunConfig): Promise<void> {\n    if (!stream)\n      throw new Error(\"You must pass a stream parameter to resume runs.\");\n    return this.startRun(startConfig, stream);\n  }\n\n  public exportExternalState(): any {\n    throw new Error(\"Runtime does not support exporting external states.\");\n  }\n\n  public importExternalState(): void {\n    throw new Error(\"Runtime does not support importing external states.\");\n  }\n\n  public async startRun(\n    { parentId, runConfig }: StartRunConfig,\n    runCallback?: ChatModelAdapter[\"run\"],\n  ): Promise<void> {\n    this.ensureInitialized();\n\n    // add assistant message\n    const id = generateId();\n    const message: ThreadAssistantMessage = {\n      id,\n      role: \"assistant\",\n      status: { type: \"running\" },\n      content: [],\n      metadata: {\n        unstable_state: this.state,\n        unstable_annotations: [],\n        unstable_data: [],\n        steps: [],\n        custom: {},\n      },\n      createdAt: new Date(),\n    };\n\n    return this._runLoop(parentId, message, runConfig, runCallback);\n  }\n\n  private async _runLoop(\n    parentId: string | null,\n    message: ThreadAssistantMessage,\n    runConfig: RunConfig | undefined,\n    runCallback?: ChatModelAdapter[\"run\"],\n  ): Promise<void> {\n    this._notifyEventSubscribers(\"runStart\", {});\n\n    // A run entered on a requires-action message resumes a pause an\n    // update-capable adapter already holds (written at pause time or loaded).\n    const alreadyPersisted =\n      message.status?.type === \"requires-action\" &&\n      this._options.adapters.history?.update !== undefined;\n\n    try {\n      // mark busy for runs not started through the queue (regenerate, resume)\n      this._queue?.notifyBusy();\n      this._suggestions = [];\n      this._suggestionsController?.abort();\n      this._suggestionsController = null;\n      this._notifySubscribers();\n\n      do {\n        message = await this.performRoundtrip(\n          parentId,\n          message,\n          runConfig,\n          alreadyPersisted,\n          runCallback,\n        );\n        runCallback = undefined;\n      } while (shouldContinue(message, this._options.unstable_humanToolNames));\n    } finally {\n      this._notifyEventSubscribers(\"runEnd\", {});\n      // queue-driven runs release from the driver settle handler; a direct\n      // run (regenerate, resume) releases here\n      if (!this._queueRunInFlight) {\n        queueMicrotask(() => this._queue?.notifyIdle());\n      }\n    }\n\n    if (\n      this.adapters.suggestion &&\n      message.status?.type !== \"requires-action\"\n    ) {\n      this._suggestionsController = new AbortController();\n      const signal = this._suggestionsController.signal;\n      const adapter = this.adapters.suggestion;\n      void (async () => {\n        try {\n          const promiseOrGenerator = adapter.generate({\n            messages: this.messages,\n            signal,\n          });\n          await consumeSuggestionResult(promiseOrGenerator, {\n            signal,\n            onUpdate: (r) => {\n              this._suggestions = r;\n              this._notifySubscribers();\n            },\n          });\n        } catch {}\n      })();\n    }\n  }\n\n  private async performRoundtrip(\n    parentId: string | null,\n    message: ThreadAssistantMessage,\n    runConfig: RunConfig | undefined,\n    alreadyPersisted: boolean,\n    runCallback?: ChatModelAdapter[\"run\"],\n  ) {\n    const messages = parentId ? this.repository.getMessages(parentId) : [];\n\n    // abort existing run\n    this.abortController?.abort();\n    const abortController = new AbortController();\n    this.abortController = abortController;\n\n    const initialContent = message.content;\n    const initialAnnotations = message.metadata?.unstable_annotations;\n    const initialData = message.metadata?.unstable_data;\n    const initialSteps = message.metadata?.steps;\n    const initialCustom = message.metadata?.custom;\n    const updateMessage = (m: Partial<ChatModelRunResult>) => {\n      const newSteps = m.metadata?.steps;\n      const steps = newSteps\n        ? [...(initialSteps ?? []), ...newSteps]\n        : undefined;\n\n      const newAnnotations = m.metadata?.unstable_annotations;\n      const newData = m.metadata?.unstable_data;\n      const annotations = newAnnotations\n        ? [...(initialAnnotations ?? []), ...newAnnotations]\n        : undefined;\n      const data = newData ? [...(initialData ?? []), ...newData] : undefined;\n\n      message = {\n        ...message,\n        ...(m.content\n          ? { content: [...initialContent, ...(m.content ?? [])] }\n          : undefined),\n        status: m.status ?? message.status,\n        ...(m.metadata\n          ? {\n              metadata: {\n                ...message.metadata,\n                ...(m.metadata.unstable_state !== undefined\n                  ? { unstable_state: m.metadata.unstable_state }\n                  : undefined),\n                ...(annotations\n                  ? { unstable_annotations: annotations }\n                  : undefined),\n                ...(data ? { unstable_data: data } : undefined),\n                ...(steps ? { steps } : undefined),\n                ...(m.metadata?.timing\n                  ? { timing: m.metadata.timing }\n                  : undefined),\n                ...(m.metadata?.custom\n                  ? {\n                      custom: {\n                        ...(initialCustom ?? {}),\n                        ...m.metadata.custom,\n                      },\n                    }\n                  : undefined),\n              },\n            }\n          : undefined),\n      };\n      this.repository.addOrUpdateMessage(parentId, message);\n      this._notifySubscribers();\n    };\n\n    const maxSteps = this._options.maxSteps ?? 2;\n\n    try {\n      const steps = message.metadata?.steps?.length ?? 0;\n      if (steps >= maxSteps) {\n        updateMessage({\n          status: {\n            type: \"incomplete\",\n            reason: \"tool-calls\",\n          },\n        });\n        return message;\n      }\n\n      updateMessage({\n        status: {\n          type: \"running\",\n        },\n      });\n\n      // Switch to the new message branch right after adding it for the first time\n      this.repository.resetHead(message.id);\n      this._notifySubscribers();\n\n      this._lastRunConfig = runConfig ?? {};\n      // unstable_composerMetadata is composer-only (stamped onto the outgoing\n      // message); never expose it to the chat-model adapter's run context.\n      const { unstable_composerMetadata: _, ...context } =\n        this.getModelContext();\n\n      runCallback =\n        runCallback ??\n        this.adapters.chatModel.run.bind(this.adapters.chatModel);\n\n      const abortSignal = abortController.signal;\n      const threadId = this._getThreadId?.();\n      const promiseOrGenerator = runCallback({\n        messages,\n        runConfig: this._lastRunConfig,\n        abortSignal,\n        context,\n        unstable_assistantMessageId: message.id,\n        unstable_threadId: threadId,\n        unstable_parentId: parentId,\n        unstable_getMessage() {\n          return message;\n        },\n      });\n\n      // handle async iterator for streaming results\n      if (Symbol.asyncIterator in promiseOrGenerator) {\n        for await (const r of promiseOrGenerator) {\n          if (abortSignal.aborted) {\n            updateMessage({\n              status: { type: \"incomplete\", reason: \"cancelled\" },\n            });\n            break;\n          }\n\n          updateMessage(r);\n        }\n      } else {\n        updateMessage(await promiseOrGenerator);\n      }\n\n      if (message.status.type === \"running\") {\n        updateMessage({\n          status: abortSignal.aborted\n            ? { type: \"incomplete\", reason: \"cancelled\" }\n            : { type: \"complete\", reason: \"unknown\" },\n        });\n      }\n    } catch (e) {\n      if (e instanceof AbortError) {\n        updateMessage({\n          status: { type: \"incomplete\", reason: \"cancelled\" },\n        });\n      } else if (e instanceof Error && e.name === \"AbortError\") {\n        updateMessage({\n          status: { type: \"incomplete\", reason: \"cancelled\" },\n        });\n      } else {\n        updateMessage({\n          status: {\n            type: \"incomplete\",\n            reason: \"error\",\n            error: toAssistantError(e),\n          },\n        });\n\n        throw e;\n      }\n    } finally {\n      if (this.abortController === abortController) {\n        this.abortController = null;\n      }\n\n      const history = this._options.adapters.history;\n      const item = {\n        parentId,\n        message,\n        runConfig: this._lastRunConfig,\n      };\n      const isTerminal =\n        message.status.type === \"complete\" ||\n        message.status.type === \"incomplete\";\n      const isPausing =\n        message.status.type === \"requires-action\" &&\n        !shouldContinue(message, this._options.unstable_humanToolNames);\n\n      // Pauses are written only for adapters that can rewrite the entry later;\n      // an append-only adapter would strand a half-finished run in history.\n      if (isTerminal || (isPausing && history?.update)) {\n        const write =\n          alreadyPersisted && history?.update\n            ? history.update.bind(history)\n            : history?.append.bind(history);\n        if (write) {\n          await this._chainHistoryWrite(message.id, () => write(item));\n        }\n      }\n    }\n    return message;\n  }\n\n  public detach() {\n    this._queue?.adapter.clear(\"cancel-run\");\n    const error = new AbortError(true);\n    this.abortController?.abort(error);\n    this.abortController = null;\n    this._suggestionsController?.abort();\n    this._suggestionsController = null;\n  }\n\n  public cancelRun() {\n    this._queue?.adapter.clear(\"cancel-run\");\n    const error = new AbortError(false);\n    this.abortController?.abort(error);\n    this.abortController = null;\n    this._suggestionsController?.abort();\n    this._suggestionsController = null;\n  }\n\n  public addToolResult({\n    messageId,\n    toolCallId,\n    result,\n    isError,\n    artifact,\n  }: AddToolResultOptions) {\n    const messageData = this.repository.getMessage(messageId);\n    const { parentId } = messageData;\n    let { message } = messageData;\n\n    if (message.role !== \"assistant\")\n      throw new Error(\"Tried to add tool result to non-assistant message\");\n\n    let added = false;\n    let found = false;\n    const newContent = message.content.map((c) => {\n      if (c.type !== \"tool-call\") return c;\n      if (c.toolCallId !== toolCallId) return c;\n      found = true;\n      if (!c.result) added = true;\n      return {\n        ...c,\n        result,\n        artifact,\n        isError,\n      };\n    });\n\n    if (!found)\n      throw new Error(\"Tried to add tool result to non-existing tool call\");\n\n    message = {\n      ...message,\n      content: newContent,\n    };\n    this.repository.addOrUpdateMessage(parentId, message);\n    this._notifySubscribers();\n\n    // a result may arrive mid-run or on a non-head message; the resume\n    // intentionally aborts any in-flight run, unlike respondToToolApproval\n    if (\n      added &&\n      shouldContinue(message, this._options.unstable_humanToolNames)\n    ) {\n      this._runLoop(parentId, message, this._lastRunConfig).catch(() => {});\n    } else if (added) {\n      this._persistPausedMessage(parentId, message);\n    }\n  }\n\n  public resumeToolCall(_options: ResumeToolCallOptions) {\n    throw new Error(\n      \"Local runtime does not support resuming tool calls. For human-in-the-loop tools, list the tool in unstable_humanToolNames and complete the call with addToolResult.\",\n    );\n  }\n\n  public respondToToolApproval({\n    approvalId,\n    approved,\n    optionId,\n    reason,\n  }: RespondToToolApprovalOptions) {\n    let message = this.repository\n      .getMessages()\n      .findLast(\n        (m): m is ThreadAssistantMessage =>\n          m.role === \"assistant\" &&\n          m.content.some(\n            (c) => c.type === \"tool-call\" && c.approval?.id === approvalId,\n          ),\n      );\n\n    if (!message)\n      throw new Error(\"Tried to respond to a non-existing tool approval\");\n\n    if (this.abortController !== null)\n      throw new Error(\n        \"Tried to respond to a tool approval while a run is in progress\",\n      );\n\n    if (message.status?.type !== \"requires-action\")\n      throw new Error(\n        \"Tried to respond to a tool approval on a message whose status is not requires-action\",\n      );\n\n    const target = message.content.find(\n      (c) => c.type === \"tool-call\" && c.approval?.id === approvalId,\n    );\n    if (target?.type !== \"tool-call\" || !target.approval)\n      throw new Error(\"Tried to respond to a non-existing tool approval\");\n    if (target.approval.resolution !== undefined)\n      throw new Error(\n        \"Tried to respond to a tool approval that was cancelled or expired\",\n      );\n    if (target.approval.approved !== undefined)\n      throw new Error(\"Tried to respond to an already decided tool approval\");\n\n    const targetApproval = target.approval;\n    const newContent = message.content.map((c) => {\n      if (c !== target) return c;\n      const approval = {\n        ...targetApproval,\n        approved,\n        ...(optionId != null && { optionId }),\n        ...(reason != null && { reason }),\n      };\n      if (approved) return { ...c, approval };\n      return {\n        ...c,\n        approval,\n        result: { error: reason || \"Tool approval denied\" },\n        isError: true,\n      };\n    });\n\n    message = { ...message, content: newContent };\n    const { parentId } = this.repository.getMessage(message.id);\n    this.repository.addOrUpdateMessage(parentId, message);\n    this._notifySubscribers();\n\n    if (\n      this.repository.headId === message.id &&\n      shouldContinue(message, this._options.unstable_humanToolNames)\n    ) {\n      this._runLoop(parentId, message, this._lastRunConfig).catch(() => {});\n    } else {\n      this._persistPausedMessage(parentId, message);\n    }\n  }\n}\n"],"mappings":";;;;;;;;;AAmCA,IAAMwB,aAAN,cAAyBC,MAAM;CAC7B,OAAgB;CAChBE;CAEAC,YAAYD,QAAiBE,SAAkB;EAC7C,MAAMA,OAAO;EACb,KAAKF,SAASA;CAChB;AACF;AAEA,IAAaG,yBAAb,cACUhB,sBAEV;CACE,eAA+B;EAC7BkB,gBAAgB;EAChBC,uBAAuB;EACvBC,MAAM;EACNC,QAAQ;EACRC,QAAQ;EACRC,QAAQ;EACRC,eAAe;EACfC,QAAQ;EACRC,WAAW;EACXC,OAAO;EACPC,aAAa;EACbC,UAAU;EACVC,OAAO;CACT;CAEA,kBAAkD;CAElD,SAAgD;CAChD,oBAA4B;CAE5B,iCAAyB,IAAIM,IAA2B;CAIxD,mBACEG,IACAC,OACe;EACf,MAAMC,QAAQ,KAAKN,eAAeO,IAAIH,EAAE,KAAKF,QAAQM,QAAQ,EAAA,CAAGC,KAC9DJ,OACAA,KACF;EACA,MAAMK,SAASJ,KAAKG,WACZ,CAAC,SACD,CAAC,CACT;EACA,KAAKT,eAAeW,IAAIP,IAAIM,MAAM;EAClC,OAAYD,WAAW;GACrB,IAAI,KAAKT,eAAeO,IAAIH,EAAE,MAAMM,QAClC,KAAKV,eAAed,OAAOkB,EAAE;EAEjC,CAAC;EACD,OAAOE;CACT;CAIA,sBACEO,UACAjC,SACA;EACA,IAAIA,QAAQkC,QAAQC,SAAS,mBAAmB;EAChD,MAAMC,UAAU,KAAKC,SAASC,SAASF;EACvC,IAAI,CAACA,SAASG,QAAQ;EACtB,MAAMA,SAASH,QAAQG,OAAOC,KAAKJ,OAAO;EAC1C,MAAMK,OAAO;GAAER;GAAUjC;GAAS0C,WAAW,KAAKC;EAAe;EACjE,KAAKpB,mBAAmBvB,QAAQwB,UAAUe,OAAOE,IAAI,CAAC,CAAC,CAACG,YAAY,CAAC,CAAC;CACxE;CAEA,aAA6B;CAC7B,iBAAiC;CAEjC,aAAqB;CACrB,IAAWI,YAAY;EACrB,OAAO,KAAKD;CACd;CAEA,eAAoD,CAAA;CACpD,yBAAyD;CACzD,IAAWI,cAA2C;EACpD,OAAO,KAAKF;CACd;CAEA,IAAWX,WAAW;EACpB,OAAO,KAAKD,SAASC;CACvB;CAEAvC,YACEqD,iBACAC,SACA;EACA,MAAMD,eAAe;EACrB,KAAKE,sBAAsBD,OAAO;CACpC;CAEA;CAEA,iBAAoC,CAAC;CAErC;CAEA,0BAAiCI,aAAuC;EACtE,KAAKF,eAAeE;CACtB;CAEA;CAEA,mCACEG,YACA;EACA,KAAKF,wBAAwBE;CAC/B;CAEA,IAAWC,SAAS,CAEpB;CAEA,sBAA6BR,SAAkC;EAC7D,IAAI,KAAKhB,aAAagB,SAAS;EAE/B,KAAKhB,WAAWgB;EAEhB,IAAIU,aAAa;EAEjB,MAAMC,WAAWX,QAAQf,UAAU5B,WAAWoD,KAAAA;EAC9C,IAAI,KAAK5D,aAAaQ,WAAWsD,UAAU;GACzC,KAAK9D,aAAaQ,SAASsD;GAC3BD,aAAa;EACf;EAEA,MAAME,aAAaZ,QAAQf,UAAU3B,cAAcmD,KAAAA;EACnD,IAAI,KAAK5D,aAAaS,cAAcsD,YAAY;GAC9C,KAAK/D,aAAaS,YAAYsD;GAC9BF,aAAa;EACf;EAEA,MAAMG,WAAWb,QAAQf,UAAU1B,UAAUkD,KAAAA;EAC7C,IAAI,KAAK5D,aAAaU,UAAUsD,UAAU;GACxC,KAAKhE,aAAaU,QAAQsD;GAC1BH,aAAa;EACf;EAEA,MAAMI,YAAYd,QAAQf,UAAUzB,gBAAgBiD,KAAAA;EACpD,IAAI,KAAK5D,aAAaW,gBAAgBsD,WAAW;GAC/C,KAAKjE,aAAaW,cAAcsD;GAChCJ,aAAa;EACf;EAEA,MAAMK,cAAcf,QAAQf,UAAUxB,aAAagD,KAAAA;EACnD,IAAI,KAAK5D,aAAaY,aAAasD,aAAa;GAC9C,KAAKlE,aAAaY,WAAWsD;GAC7BL,aAAa;EACf;EAEA,MAAMM,YAAYhB,QAAQf,UAAUF,SAAS9B,WAAWwD,KAAAA;EACxD,IAAI,KAAK5D,aAAaI,WAAW+D,WAAW;GAC1C,KAAKnE,aAAaI,SAAS+D;GAC3BN,aAAa;EACf;EAEA,MAAMO,WAAWjB,QAAQkB,gCAAgC;EACzD,IAAID,YAAY,CAAC,KAAKpD,QAAQ;GAC5B,KAAKA,SAAS3B,mBAAmB,EAC/BiF,MAAMxE,YAAY;IAGhB,KAAKmB,oBAAoB;IACzB,KAAUsD,WAAWzE,OAAO,CAAC,CAC1B0E,cAAc;KACb,KAAKvD,oBAAoB;KACzB,KAAKD,QAAQyD,WAAW;IAC1B,CAAC,CAAC,CACD/B,YAAY,CAAC,CAAC;GACnB,EACF,CAAC;GACD,KAAK1B,OAAO0D,gBAAgB,KAAKC,mBAAmB,CAAC;EACvD,OAAO,IAAI,CAACP,YAAY,KAAKpD,QAAQ;GACnC,KAAKA,OAAO4D,QAAQC,MAAM,YAAY;GACtC,KAAK7D,SAAS;EAChB;EACA,IAAI,KAAKhB,aAAaa,UAAUuD,UAAU;GACxC,KAAKpE,aAAaa,QAAQuD;GAC1BP,aAAa;EACf;EAEA,IAAIA,YAAY,KAAKc,mBAAmB;CAC1C;CAEA;CACA,kBAAyB;EACvB,IAAI,KAAKG,cAAc,OAAO,KAAKA;EAEnC,MAAME,UAAU,KAAK5C,SAASF,SAAS+C,KAAK,KAAK7D,QAAQM,QAAQ,IAAI;EAErE,KAAKmB,aAAa;EAClB,KAAK8B,mBAAmB;EAExB,KAAKG,eAAeE,QACjBrD,MAAMuD,SAAS;GACd,IAAI,CAACA,MAAM;GACX,KAAKC,WAAWC,OAAOF,IAAI;GAC3B,IAAIA,KAAKG,SAASC,SAAS,GACzB,KAAKC,kBAAkB;GAEzB,KAAKZ,mBAAmB;GAExB,MAAMa,SAAS,KAAKpD,SAASF,SAASsD,QAAQlD,KAC5C,KAAKF,SAASF,OAChB;GACA,IAAIgD,KAAKO,mBAAmBD,QAC1B,KAAKE,SACH;IACE3D,UAAU,KAAKoD,WAAWQ;IAC1BC,UAAU,KAAKT,WAAWQ;IAC1BnD,WAAW,KAAKC;GAClB,GACA+C,MACF,CAAC,CAAC9C,YAAY,CAAC,CAAC;EAEpB,CAAC,CAAC,CACD8B,cAAc;GACb,KAAK3B,aAAa;GAClB,KAAK8B,mBAAmB;EAC1B,CAAC;EAEH,OAAO,KAAKG;CACd;CAEA,MAAae,OAAO/F,SAAuC;EACzD,MAAMgG,SAAShG,QAAQiC,cAAc,KAAKsD,SAASU,GAAG,EAAE,CAAC,EAAEzE,MAAM;EACjE,MAAM0E,UAAUlG,QAAQ4F,YAAY5F,QAAQmG,SAAS;EACrD,IAAI,KAAKjF,UAAUgF,WAAWF,QAAQ;GACpC,KAAK9E,OAAO4D,QAAQsB,QAAQpG,SAAS,EAAEqG,OAAOrG,QAAQqG,SAAS,MAAM,CAAC;GACtE;EACF;EACA,IAAI,KAAKnF,UAAU,CAAC8E,QAAQ,KAAK9E,OAAO4D,QAAQC,MAAM,MAAM;EAC5D,OAAO,KAAKN,WAAWzE,OAAO;CAChC;CAEA,gBAAkD;EAGhD,OAAO,KAAKkB,QAAQ4D,QAAQyB,SAAS9G;CACvC;CAEA,eAAsBgH,aAA2B;EAC/C,KAAKvF,QAAQ4D,QAAQuB,MAAMI,WAAW;CACxC;CAEA,gBAAuBA,aAA2B;EAChD,KAAKvF,QAAQ4D,QAAQ6B,OAAOF,WAAW;CACzC;CAEA,MAAchC,WAAWzE,SAAuC;EAC9D,KAAKyF,kBAAkB;EAEvB,MAAMmB,cAAc,KAAKlD,wBAAwB;EACjD,IAAIkD,aACF,MAAMA;EAGR,MAAMC,aAAa1I,sBAAsB6B,SAAS5B,WAAW,GAAG;GAC9D+D,MAAM;GACN2E,QAAQ;EACV,CAAC;EACD,KAAKzB,WAAW0B,mBAAmB/G,QAAQiC,UAAU4E,UAAU;EAC/D,KAAKxE,SAASC,SAASF,SAAS2D,OAAO;GACrC9D,UAAUjC,QAAQiC;GAClBjC,SAAS6G;GACT,GAAI7G,QAAQ0C,cAAcoB,KAAAA,KAAa,EAAEpB,WAAW1C,QAAQ0C,UAAU;EACxE,CAAC;EAGD,IADiB1C,QAAQ4F,YAAY5F,QAAQmG,SAAS,QAEpD,MAAM,KAAKP,SAAS;GAClB3D,UAAU4E,WAAWrF;GACrBsE,UAAU9F,QAAQ8F;GAClBpD,WAAW1C,QAAQ0C,aAAa,CAAC;EACnC,CAAC;OACI;GACL,KAAK2C,WAAW2B,UAAUH,WAAWrF,EAAE;GACvC,KAAKqD,mBAAmB;EAC1B;CACF;CAEA,MAAaoC,cAAcC,WAAkC;EAC3D,MAAMpC,UAAU,KAAKzC,SAASC,SAASF;EACvC,IAAI,CAAC0C,SAASxE,QACZ,MAAM,IAAIV,MAAM,6CAA6C;EAE/D,MAAM2F,WAAW,KAAKF,WAAW8B,YAAY;EAC7C,MAAMC,eAAe7B,SAAS8B,WAAWC,MAAMA,EAAE9F,OAAO0F,SAAS;EACjE,IAAIE,iBAAiB,IAAI,MAAM,IAAIxH,MAAM,oBAAoB;EAE7D,MAAMI,UAAUuF,SAAS6B;EAEzB,MAAMb,QAAQ,CAAC;GAAEtE,UADAsD,SAAS6B,eAAe,EAAE,EAAE5F,MAAM;GACxBxB;EAAQ,CAAC;EAEpC,MAAM8E,QAAQxE,OAAOiG,KAAK;EAE1B,KAAKlB,WAAW4B,cAAcC,SAAS;EACvC,KAAKrC,mBAAmB;CAC1B;CAEA,UAAiB,EAAE2C,QAAQ,GAAGC,eAA+C;EAC3E,IAAI,CAACD,QACH,MAAM,IAAI5H,MAAM,kDAAkD;EACpE,OAAO,KAAKgG,SAAS6B,aAAaD,MAAM;CAC1C;CAEA,sBAAkC;EAChC,MAAM,IAAI5H,MAAM,qDAAqD;CACvE;CAEA,sBAAmC;EACjC,MAAM,IAAIA,MAAM,qDAAqD;CACvE;CAEA,MAAagG,SACX,EAAE3D,UAAUS,aACZkF,aACe;EACf,KAAKnC,kBAAkB;EAIvB,MAAMzF,UAAkC;GACtCwB,IAFSpD,WAEToD;GACA2E,MAAM;GACNjE,QAAQ,EAAEC,MAAM,UAAU;GAC1B0F,SAAS,CAAA;GACTC,UAAU;IACRC,gBAAgB,KAAKC;IACrBC,sBAAsB,CAAA;IACtBC,eAAe,CAAA;IACfC,OAAO,CAAA;IACPC,QAAQ,CAAC;GACX;GACAC,2BAAW,IAAIC,KAAK;EACtB;EAEA,OAAO,KAAKC,SAAStG,UAAUjC,SAAS0C,WAAWkF,WAAW;CAChE;CAEA,MAAcW,SACZtG,UACAjC,SACA0C,WACAkF,aACe;EACf,KAAKY,wBAAwB,YAAY,CAAC,CAAC;EAI3C,MAAMC,mBACJzI,QAAQkC,QAAQC,SAAS,qBACzB,KAAKE,SAASC,SAASF,SAASG,WAAWuB,KAAAA;EAE7C,IAAI;GAEF,KAAK5C,QAAQwH,WAAW;GACxB,KAAKzF,eAAe,CAAA;GACpB,KAAKC,wBAAwByF,MAAM;GACnC,KAAKzF,yBAAyB;GAC9B,KAAK2B,mBAAmB;GAExB,GAAG;IACD7E,UAAU,MAAM,KAAK4I,iBACnB3G,UACAjC,SACA0C,WACA+F,kBACAb,WACF;IACAA,cAAc9D,KAAAA;GAChB,SAASvF,eAAeyB,SAAS,KAAKqC,SAASwG,uBAAuB;EACxE,UAAU;GACR,KAAKL,wBAAwB,UAAU,CAAC,CAAC;GAGzC,IAAI,CAAC,KAAKrH,mBACR2H,qBAAqB,KAAK5H,QAAQyD,WAAW,CAAC;EAElD;EAEA,IACE,KAAKrC,SAASyG,cACd/I,QAAQkC,QAAQC,SAAS,mBACzB;GACA,KAAKe,yBAAyB,IAAIjC,gBAAgB;GAClD,MAAM+H,SAAS,KAAK9F,uBAAuB8F;GAC3C,MAAMlE,UAAU,KAAKxC,SAASyG;GAC9B,CAAM,YAAY;IAChB,IAAI;KAKF,MAAMtK,wBAJqBqG,QAAQoE,SAAS;MAC1C3D,UAAU,KAAKA;MACfyD;KACF,CAC8BC,GAAoB;MAChDD;MACAG,WAAWC,MAAM;OACf,KAAKnG,eAAemG;OACpB,KAAKvE,mBAAmB;MAC1B;KACF,CAAC;IACH,QAAQ,CAAC;GACX,EAAA,CAAG;EACL;CACF;CAEA,MAAc+D,iBACZ3G,UACAjC,SACA0C,WACA+F,kBACAb,aACA;EACA,MAAMrC,WAAWtD,WAAW,KAAKoD,WAAW8B,YAAYlF,QAAQ,IAAI,CAAA;EAGpE,KAAKjB,iBAAiB2H,MAAM;EAC5B,MAAM3H,kBAAkB,IAAIC,gBAAgB;EAC5C,KAAKD,kBAAkBA;EAEvB,MAAMqI,iBAAiBrJ,QAAQ6H;EAC/B,MAAMyB,qBAAqBtJ,QAAQ8H,UAAUG;EAC7C,MAAMsB,cAAcvJ,QAAQ8H,UAAUI;EACtC,MAAMsB,eAAexJ,QAAQ8H,UAAUK;EACvC,MAAMsB,gBAAgBzJ,QAAQ8H,UAAUM;EACxC,MAAMsB,iBAAiBpC,MAAmC;GACxD,MAAMsC,WAAWtC,EAAEQ,UAAUK;GAC7B,MAAMA,QAAQyB,WACV,CAAC,GAAIJ,gBAAgB,CAAA,GAAK,GAAGI,QAAQ,IACrC9F,KAAAA;GAEJ,MAAM+F,iBAAiBvC,EAAEQ,UAAUG;GACnC,MAAM6B,UAAUxC,EAAEQ,UAAUI;GAC5B,MAAM6B,cAAcF,iBAChB,CAAC,GAAIP,sBAAsB,CAAA,GAAK,GAAGO,cAAc,IACjD/F,KAAAA;GACJ,MAAMkG,OAAOF,UAAU,CAAC,GAAIP,eAAe,CAAA,GAAK,GAAGO,OAAO,IAAIhG,KAAAA;GAE9D9D,UAAU;IACR,GAAGA;IACH,GAAIsH,EAAEO,UACF,EAAEA,SAAS,CAAC,GAAGwB,gBAAgB,GAAI/B,EAAEO,WAAW,CAAA,CAAG,EAAE,IACrD/D,KAAAA;IACJ5B,QAAQoF,EAAEpF,UAAUlC,QAAQkC;IAC5B,GAAIoF,EAAEQ,WACF,EACEA,UAAU;KACR,GAAG9H,QAAQ8H;KACX,GAAIR,EAAEQ,SAASC,mBAAmBjE,KAAAA,IAC9B,EAAEiE,gBAAgBT,EAAEQ,SAASC,eAAe,IAC5CjE,KAAAA;KACJ,GAAIiG,cACA,EAAE9B,sBAAsB8B,YAAY,IACpCjG,KAAAA;KACJ,GAAIkG,OAAO,EAAE9B,eAAe8B,KAAK,IAAIlG,KAAAA;KACrC,GAAIqE,QAAQ,EAAEA,MAAM,IAAIrE,KAAAA;KACxB,GAAIwD,EAAEQ,UAAUmC,SACZ,EAAEA,QAAQ3C,EAAEQ,SAASmC,OAAO,IAC5BnG,KAAAA;KACJ,GAAIwD,EAAEQ,UAAUM,SACZ,EACEA,QAAQ;MACN,GAAIqB,iBAAiB,CAAC;MACtB,GAAGnC,EAAEQ,SAASM;KAChB,EACF,IACAtE,KAAAA;IACN,EACF,IACAA,KAAAA;GACN;GACA,KAAKuB,WAAW0B,mBAAmB9E,UAAUjC,OAAO;GACpD,KAAK6E,mBAAmB;EAC1B;EAEA,MAAMqF,WAAW,KAAK7H,SAAS6H,YAAY;EAE3C,IAAI;GAEF,KADclK,QAAQ8H,UAAUK,OAAO3C,UAAU,MACpC0E,UAAU;IACrBR,cAAc,EACZxH,QAAQ;KACNC,MAAM;KACN2E,QAAQ;IACV,EACF,CAAC;IACD,OAAO9G;GACT;GAEA0J,cAAc,EACZxH,QAAQ,EACNC,MAAM,UACR,EACF,CAAC;GAGD,KAAKkD,WAAW2B,UAAUhH,QAAQwB,EAAE;GACpC,KAAKqD,mBAAmB;GAExB,KAAKlC,iBAAiBD,aAAa,CAAC;GAGpC,MAAM,EAAEyH,2BAA2BC,GAAG,GAAGC,YACvC,KAAKC,gBAAgB;GAEvB1C,cACEA,eACA,KAAKtF,SAASiI,UAAU/F,IAAIhC,KAAK,KAAKF,SAASiI,SAAS;GAE1D,MAAMC,cAAcxJ,gBAAgBgI;GACpC,MAAMyB,WAAW,KAAKlH,eAAe;GACrC,MAAM0F,qBAAqBrB,YAAY;IACrCrC;IACA7C,WAAW,KAAKC;IAChB6H;IACAH;IACAK,6BAA6B1K,QAAQwB;IACrCmJ,mBAAmBF;IACnBG,mBAAmB3I;IACnB4I,sBAAsB;KACpB,OAAO7K;IACT;GACF,CAAC;GAGD,IAAI8K,OAAOC,iBAAiB9B,oBAC1B,WAAW,MAAMG,KAAKH,oBAAoB;IACxC,IAAIuB,YAAYQ,SAAS;KACvBtB,cAAc,EACZxH,QAAQ;MAAEC,MAAM;MAAc2E,QAAQ;KAAY,EACpD,CAAC;KACD;IACF;IAEA4C,cAAcN,CAAC;GACjB;QAEAM,cAAc,MAAMT,kBAAkB;GAGxC,IAAIjJ,QAAQkC,OAAOC,SAAS,WAC1BuH,cAAc,EACZxH,QAAQsI,YAAYQ,UAChB;IAAE7I,MAAM;IAAc2E,QAAQ;GAAY,IAC1C;IAAE3E,MAAM;IAAY2E,QAAQ;GAAU,EAC5C,CAAC;EAEL,SAASmE,GAAG;GACV,IAAIA,aAAatL,YACf+J,cAAc,EACZxH,QAAQ;IAAEC,MAAM;IAAc2E,QAAQ;GAAY,EACpD,CAAC;QACI,IAAImE,aAAarL,SAASqL,EAAEpL,SAAS,cAC1C6J,cAAc,EACZxH,QAAQ;IAAEC,MAAM;IAAc2E,QAAQ;GAAY,EACpD,CAAC;QACI;IACL4C,cAAc,EACZxH,QAAQ;KACNC,MAAM;KACN2E,QAAQ;KACRoE,OAAO7L,iBAAiB4L,CAAC;IAC3B,EACF,CAAC;IAED,MAAMA;GACR;EACF,UAAU;GACR,IAAI,KAAKjK,oBAAoBA,iBAC3B,KAAKA,kBAAkB;GAGzB,MAAMoB,UAAU,KAAKC,SAASC,SAASF;GACvC,MAAMK,OAAO;IACXR;IACAjC;IACA0C,WAAW,KAAKC;GAClB;GACA,MAAMwI,aACJnL,QAAQkC,OAAOC,SAAS,cACxBnC,QAAQkC,OAAOC,SAAS;GAC1B,MAAMiJ,YACJpL,QAAQkC,OAAOC,SAAS,qBACxB,CAAC5D,eAAeyB,SAAS,KAAKqC,SAASwG,uBAAuB;GAIhE,IAAIsC,cAAeC,aAAahJ,SAASG,QAAS;IAChD,MAAMd,QACJgH,oBAAoBrG,SAASG,SACzBH,QAAQG,OAAOC,KAAKJ,OAAO,IAC3BA,SAAS2D,OAAOvD,KAAKJ,OAAO;IAClC,IAAIX,OACF,MAAM,KAAKF,mBAAmBvB,QAAQwB,UAAUC,MAAMgB,IAAI,CAAC;GAE/D;EACF;EACA,OAAOzC;CACT;CAEA,SAAgB;EACd,KAAKkB,QAAQ4D,QAAQC,MAAM,YAAY;EACvC,MAAMmG,QAAQ,IAAIvL,WAAW,IAAI;EACjC,KAAKqB,iBAAiB2H,MAAMuC,KAAK;EACjC,KAAKlK,kBAAkB;EACvB,KAAKkC,wBAAwByF,MAAM;EACnC,KAAKzF,yBAAyB;CAChC;CAEA,YAAmB;EACjB,KAAKhC,QAAQ4D,QAAQC,MAAM,YAAY;EACvC,MAAMmG,QAAQ,IAAIvL,WAAW,KAAK;EAClC,KAAKqB,iBAAiB2H,MAAMuC,KAAK;EACjC,KAAKlK,kBAAkB;EACvB,KAAKkC,wBAAwByF,MAAM;EACnC,KAAKzF,yBAAyB;CAChC;CAEA,cAAqB,EACnBgE,WACAqE,YACAC,QACAC,SACAC,YACuB;EACvB,MAAMC,cAAc,KAAKtG,WAAWuG,WAAW1E,SAAS;EACxD,MAAM,EAAEjF,aAAa0J;EACrB,IAAI,EAAE3L,YAAY2L;EAElB,IAAI3L,QAAQmG,SAAS,aACnB,MAAM,IAAIvG,MAAM,mDAAmD;EAErE,IAAIiM,QAAQ;EACZ,IAAIC,QAAQ;EACZ,MAAMC,aAAa/L,QAAQ6H,QAAQmE,KAAKC,MAAM;GAC5C,IAAIA,EAAE9J,SAAS,aAAa,OAAO8J;GACnC,IAAIA,EAAEV,eAAeA,YAAY,OAAOU;GACxCH,QAAQ;GACR,IAAI,CAACG,EAAET,QAAQK,QAAQ;GACvB,OAAO;IACL,GAAGI;IACHT;IACAE;IACAD;GACF;EACF,CAAC;EAED,IAAI,CAACK,OACH,MAAM,IAAIlM,MAAM,oDAAoD;EAEtEI,UAAU;GACR,GAAGA;GACH6H,SAASkE;EACX;EACA,KAAK1G,WAAW0B,mBAAmB9E,UAAUjC,OAAO;EACpD,KAAK6E,mBAAmB;EAIxB,IACEgH,SACAtN,eAAeyB,SAAS,KAAKqC,SAASwG,uBAAuB,GAE7D,KAAKN,SAAStG,UAAUjC,SAAS,KAAK2C,cAAc,CAAC,CAACC,YAAY,CAAC,CAAC;OAC/D,IAAIiJ,OACT,KAAK7J,sBAAsBC,UAAUjC,OAAO;CAEhD;CAEA,eAAsBqC,UAAiC;EACrD,MAAM,IAAIzC,MACR,qKACF;CACF;CAEA,sBAA6B,EAC3BwM,YACAC,UACAC,UACAxF,UAC+B;EAC/B,IAAI9G,UAAU,KAAKqF,WAChB8B,YAAY,CAAC,CACboF,UACEjF,MACCA,EAAEnB,SAAS,eACXmB,EAAEO,QAAQ2E,MACPP,MAAMA,EAAE9J,SAAS,eAAe8J,EAAEQ,UAAUjL,OAAO4K,UACtD,CACJ;EAEF,IAAI,CAACpM,SACH,MAAM,IAAIJ,MAAM,kDAAkD;EAEpE,IAAI,KAAKoB,oBAAoB,MAC3B,MAAM,IAAIpB,MACR,gEACF;EAEF,IAAII,QAAQkC,QAAQC,SAAS,mBAC3B,MAAM,IAAIvC,MACR,sFACF;EAEF,MAAM8M,SAAS1M,QAAQ6H,QAAQ8E,MAC5BV,MAAMA,EAAE9J,SAAS,eAAe8J,EAAEQ,UAAUjL,OAAO4K,UACtD;EACA,IAAIM,QAAQvK,SAAS,eAAe,CAACuK,OAAOD,UAC1C,MAAM,IAAI7M,MAAM,kDAAkD;EACpE,IAAI8M,OAAOD,SAASG,eAAe9I,KAAAA,GACjC,MAAM,IAAIlE,MACR,mEACF;EACF,IAAI8M,OAAOD,SAASJ,aAAavI,KAAAA,GAC/B,MAAM,IAAIlE,MAAM,sDAAsD;EAExE,MAAMiN,iBAAiBH,OAAOD;EAC9B,MAAMV,aAAa/L,QAAQ6H,QAAQmE,KAAKC,MAAM;GAC5C,IAAIA,MAAMS,QAAQ,OAAOT;GACzB,MAAMQ,WAAW;IACf,GAAGI;IACHR;IACA,GAAIC,YAAY,QAAQ,EAAEA,SAAS;IACnC,GAAIxF,UAAU,QAAQ,EAAEA,OAAO;GACjC;GACA,IAAIuF,UAAU,OAAO;IAAE,GAAGJ;IAAGQ;GAAS;GACtC,OAAO;IACL,GAAGR;IACHQ;IACAjB,QAAQ,EAAEN,OAAOpE,UAAU,uBAAuB;IAClD2E,SAAS;GACX;EACF,CAAC;EAEDzL,UAAU;GAAE,GAAGA;GAAS6H,SAASkE;EAAW;EAC5C,MAAM,EAAE9J,aAAa,KAAKoD,WAAWuG,WAAW5L,QAAQwB,EAAE;EAC1D,KAAK6D,WAAW0B,mBAAmB9E,UAAUjC,OAAO;EACpD,KAAK6E,mBAAmB;EAExB,IACE,KAAKQ,WAAWQ,WAAW7F,QAAQwB,MACnCjD,eAAeyB,SAAS,KAAKqC,SAASwG,uBAAuB,GAE7D,KAAKN,SAAStG,UAAUjC,SAAS,KAAK2C,cAAc,CAAC,CAACC,YAAY,CAAC,CAAC;OAEpE,KAAKZ,sBAAsBC,UAAUjC,OAAO;CAEhD;AACF"}

package/dist/store/clients/runtime-adapter.js

@@ -6,17 +6,17 @@ const baseRuntimeAdapterTransformScopes = (scopes, parent) => {
 	scopes.thread ??= Derived({
 		source: "threads",
 		query: { type: "main" },
-		get: (aui) => aui.threads().thread("main")
+		get: (aui) => aui.threads.thread("main")
 	});
 	scopes.threadListItem ??= Derived({
 		source: "threads",
 		query: { type: "main" },
-		get: (aui) => aui.threads().item("main")
+		get: (aui) => aui.threads.item("main")
 	});
 	scopes.composer ??= Derived({
 		source: "thread",
 		query: {},
-		get: (aui) => aui.threads().thread("main").composer()
+		get: (aui) => aui.threads.thread("main").composer()
 	});
 	if (!scopes.modelContext && parent.modelContext.source === null) scopes.modelContext = ModelContext();
 	if (!scopes.suggestions && parent.suggestions.source === null) scopes.suggestions = Suggestions();

package/dist/store/clients/runtime-adapter.js.map

@@ -1 +1 @@
-{"version":3,"file":"runtime-adapter.js","names":["Derived","ScopesConfig","AssistantClient","ModelContext","Suggestions","baseRuntimeAdapterTransformScopes","scopes","parent","thread","source","query","type","get","aui","threads","threadListItem","item","composer","modelContext","suggestions"],"sources":["../../../src/store/clients/runtime-adapter.ts"],"sourcesContent":["import {\n  Derived,\n  type ScopesConfig,\n  type AssistantClient,\n} from \"@assistant-ui/store\";\nimport { ModelContext } from \"./model-context-client\";\nimport { Suggestions } from \"./suggestions\";\n\nexport const baseRuntimeAdapterTransformScopes = (\n  scopes: ScopesConfig,\n  parent: AssistantClient,\n): void => {\n  scopes.thread ??= Derived({\n    source: \"threads\",\n    query: { type: \"main\" },\n    get: (aui) => aui.threads().thread(\"main\"),\n  });\n  scopes.threadListItem ??= Derived({\n    source: \"threads\",\n    query: { type: \"main\" },\n    get: (aui) => aui.threads().item(\"main\"),\n  });\n  scopes.composer ??= Derived({\n    source: \"thread\",\n    query: {},\n    get: (aui) => aui.threads().thread(\"main\").composer(),\n  });\n\n  if (!scopes.modelContext && parent.modelContext.source === null) {\n    scopes.modelContext = ModelContext();\n  }\n  if (!scopes.suggestions && parent.suggestions.source === null) {\n    scopes.suggestions = Suggestions();\n  }\n};\n"],"mappings":";;;;AAQA,MAAaK,qCACXC,QACAC,WACS;CACTD,OAAOE,WAAWR,QAAQ;EACxBS,QAAQ;EACRC,OAAO,EAAEC,MAAM,OAAO;EACtBC,MAAMC,QAAQA,IAAIC,QAAQ,CAAC,CAACN,OAAO,MAAM;CAC3C,CAAC;CACDF,OAAOS,mBAAmBf,QAAQ;EAChCS,QAAQ;EACRC,OAAO,EAAEC,MAAM,OAAO;EACtBC,MAAMC,QAAQA,IAAIC,QAAQ,CAAC,CAACE,KAAK,MAAM;CACzC,CAAC;CACDV,OAAOW,aAAajB,QAAQ;EAC1BS,QAAQ;EACRC,OAAO,CAAC;EACRE,MAAMC,QAAQA,IAAIC,QAAQ,CAAC,CAACN,OAAO,MAAM,CAAC,CAACS,SAAS;CACtD,CAAC;CAED,IAAI,CAACX,OAAOY,gBAAgBX,OAAOW,aAAaT,WAAW,MACzDH,OAAOY,eAAef,aAAa;CAErC,IAAI,CAACG,OAAOa,eAAeZ,OAAOY,YAAYV,WAAW,MACvDH,OAAOa,cAAcf,YAAY;AAErC"}
+{"version":3,"file":"runtime-adapter.js","names":["Derived","ScopesConfig","AssistantClient","ModelContext","Suggestions","baseRuntimeAdapterTransformScopes","scopes","parent","thread","source","query","type","get","aui","threads","threadListItem","item","composer","modelContext","suggestions"],"sources":["../../../src/store/clients/runtime-adapter.ts"],"sourcesContent":["import {\n  Derived,\n  type ScopesConfig,\n  type AssistantClient,\n} from \"@assistant-ui/store\";\nimport { ModelContext } from \"./model-context-client\";\nimport { Suggestions } from \"./suggestions\";\n\nexport const baseRuntimeAdapterTransformScopes = (\n  scopes: ScopesConfig,\n  parent: AssistantClient,\n): void => {\n  scopes.thread ??= Derived({\n    source: \"threads\",\n    query: { type: \"main\" },\n    get: (aui) => aui.threads.thread(\"main\"),\n  });\n  scopes.threadListItem ??= Derived({\n    source: \"threads\",\n    query: { type: \"main\" },\n    get: (aui) => aui.threads.item(\"main\"),\n  });\n  scopes.composer ??= Derived({\n    source: \"thread\",\n    query: {},\n    get: (aui) => aui.threads.thread(\"main\").composer(),\n  });\n\n  if (!scopes.modelContext && parent.modelContext.source === null) {\n    scopes.modelContext = ModelContext();\n  }\n  if (!scopes.suggestions && parent.suggestions.source === null) {\n    scopes.suggestions = Suggestions();\n  }\n};\n"],"mappings":";;;;AAQA,MAAaK,qCACXC,QACAC,WACS;CACTD,OAAOE,WAAWR,QAAQ;EACxBS,QAAQ;EACRC,OAAO,EAAEC,MAAM,OAAO;EACtBC,MAAMC,QAAQA,IAAIC,QAAQN,OAAO,MAAM;CACzC,CAAC;CACDF,OAAOS,mBAAmBf,QAAQ;EAChCS,QAAQ;EACRC,OAAO,EAAEC,MAAM,OAAO;EACtBC,MAAMC,QAAQA,IAAIC,QAAQE,KAAK,MAAM;CACvC,CAAC;CACDV,OAAOW,aAAajB,QAAQ;EAC1BS,QAAQ;EACRC,OAAO,CAAC;EACRE,MAAMC,QAAQA,IAAIC,QAAQN,OAAO,MAAM,CAAC,CAACS,SAAS;CACpD,CAAC;CAED,IAAI,CAACX,OAAOY,gBAAgBX,OAAOW,aAAaT,WAAW,MACzDH,OAAOY,eAAef,aAAa;CAErC,IAAI,CAACG,OAAOa,eAAeZ,OAAOY,YAAYV,WAAW,MACvDH,OAAOa,cAAcf,YAAY;AAErC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/core",
-  "version": "0.2.23",
+  "version": "0.3.1",
   "description": "Framework-agnostic core runtime for assistant-ui",
   "keywords": [
     "assistant",
@@ -59,14 +59,14 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "assistant-stream": "^0.3.28",
+    "assistant-stream": "^0.3.30",
     "nanoid": "^6.0.0"
   },
   "optionalDevDependencies": {
     "@assistant-ui/x-generative-compiler": ">=0.0.3"
   },
   "peerDependencies": {
-    "@assistant-ui/store": "^0.2.13",
+    "@assistant-ui/store": "^0.3.0",
     "@assistant-ui/tap": "^0.9.0",
     "@types/react": "*",
     "react": "^18 || ^19",
@@ -95,8 +95,8 @@
     "react-dom": "^19.2.8",
     "vitest": "^4.1.10",
     "zustand": "^5.0.14",
-    "@assistant-ui/store": "0.2.22",
-    "@assistant-ui/tap": "0.9.6",
+    "@assistant-ui/store": "0.3.1",
+    "@assistant-ui/tap": "0.9.7",
     "@assistant-ui/vite": "0.0.11",
     "@assistant-ui/x-buildutils": "0.0.20",
     "assistant-cloud": "0.1.37"

package/src/react/adapters/LocalStorageThreadListAdapter.tsx

@@ -297,7 +297,7 @@ class AsyncStorageHistoryAdapter implements ThreadHistoryAdapter {
   }
 
   async load(): Promise<ExportedMessageRepository> {
-    const remoteId = this.aui.threadListItem().getState().remoteId;
+    const remoteId = this.aui.threadListItem.getState().remoteId;
     if (!remoteId) return { messages: [] };
 
     const raw = await this.storage.getItem(this._messagesKey(remoteId));
@@ -305,7 +305,7 @@ class AsyncStorageHistoryAdapter implements ThreadHistoryAdapter {
   }
 
   async append(item: ExportedMessageRepositoryItem): Promise<void> {
-    const { remoteId } = await this.aui.threadListItem().initialize();
+    const { remoteId } = await this.aui.threadListItem.initialize();
 
     const key = this._messagesKey(remoteId);
     await this.mutationQueue.run(key, async () => {

package/src/react/client/Tools.ts

@@ -48,13 +48,6 @@ const useTools = ({
     (): ToolsState => ({
       toolUIs,
       mcpApp: mcpAppOutput,
-      // Deprecated component-only view, derived from `toolUIs`. Removed in v0.15.
-      tools: Object.fromEntries(
-        Object.entries(toolUIs).map(([name, regs]) => [
-          name,
-          regs.map((r) => r.render),
-        ]),
-      ),
     }),
     [toolUIs, mcpAppOutput],
   );

package/src/react/interactables-legacy/useAssistantInteractable.ts

@@ -42,7 +42,7 @@ export const useAssistantInteractable = (
   initialStateRef.current = config.initialState;
 
   useEffect(() => {
-    return aui.interactables().register({
+    return aui.interactables.register({
       id,
       name,
       description: config.description,

package/src/react/interactables-legacy/useInteractableState.ts

@@ -35,7 +35,7 @@ export const useInteractableState = <TState>(
 
   const setState = useCallback(
     (updater: StateUpdater<TState>) => {
-      aui.interactables().setState(id, (prev) => {
+      aui.interactables.setState(id, (prev) => {
         if (typeof updater === "function") {
           return (updater as (prev: TState) => TState)(prev as TState);
         }
@@ -47,12 +47,12 @@ export const useInteractableState = <TState>(
 
   const setSelected = useCallback(
     (selected: boolean) => {
-      aui.interactables().setSelected(id, selected);
+      aui.interactables.setSelected(id, selected);
     },
     [aui, id],
   );
 
-  const flush = useCallback(() => aui.interactables().flush(), [aui]);
+  const flush = useCallback(() => aui.interactables.flush(), [aui]);
 
   return [
     state,

package/src/react/model-context/useAssistantContext.ts

@@ -13,7 +13,7 @@ export const useAssistantContext = (config: AssistantContextConfig) => {
   useEffect(() => {
     if (disabled) return;
 
-    return aui.modelContext().register({
+    return aui.modelContext.register({
       getModelContext: () => ({
         system: getContextRef.current(),
       }),

package/src/react/model-context/useAssistantDataUI.ts

@@ -20,6 +20,6 @@ export const useAssistantDataUI = (dataUI: AssistantDataUIProps | null) => {
   const aui = useAui();
   useEffect(() => {
     if (!dataUI?.name || !dataUI?.render) return undefined;
-    return aui.dataRenderers().setDataUI(dataUI.name, dataUI.render);
+    return aui.dataRenderers.setDataUI(dataUI.name, dataUI.render);
   }, [aui, dataUI?.name, dataUI?.render]);
 };

package/src/react/model-context/useAssistantInstructions.ts

@@ -23,7 +23,7 @@ export const useAssistantInstructions = (
     const config = {
       system: instruction,
     };
-    return aui.modelContext().register({
+    return aui.modelContext.register({
       getModelContext: () => config,
     });
   }, [aui, instruction, disabled]);

package/src/react/model-context/useAssistantTool.ts

@@ -77,7 +77,7 @@ export const useAssistantTool = <
 
   useEffect(() => {
     if (!render) return undefined;
-    return aui.tools().setToolUI(tool.toolName, render, { standalone });
+    return aui.tools.setToolUI(tool.toolName, render, { standalone });
   }, [aui, tool.toolName, render, standalone]);
 
   useEffect(() => {
@@ -89,7 +89,7 @@ export const useAssistantTool = <
         [toolName]: rest,
       },
     };
-    return aui.modelContext().register({
+    return aui.modelContext.register({
       getModelContext: () => context,
     });
   }, [aui, tool]);

package/src/react/model-context/useAssistantToolUI.ts

@@ -41,6 +41,6 @@ export const useAssistantToolUI = (
   const standalone = tool?.display === "standalone";
   useEffect(() => {
     if (!tool?.toolName || !tool?.render) return undefined;
-    return aui.tools().setToolUI(tool.toolName, tool.render, { standalone });
+    return aui.tools.setToolUI(tool.toolName, tool.render, { standalone });
   }, [aui, tool?.toolName, tool?.render, standalone]);
 };

package/src/react/model-context/useAuiToolOverrides.ts

@@ -28,7 +28,7 @@ export function useAuiToolOverrides(overrides: AuiToolOverrides): void {
   overridesRef.current = overrides;
 
   useEffect(() => {
-    return aui.modelContext().register({
+    return aui.modelContext.register({
       getModelContext: () => ({
         priority: 1000,
         tools: overridesRef.current as Record<string, Tool<any, any>>,

package/src/react/model-context/useInteractable.ts

@@ -90,14 +90,10 @@ const useInteractable = <TSchema extends Unstable_InteractableStateSchema>(
 
   const autoId = useId().replace(/[^a-zA-Z0-9]/g, "");
 
-  // Whether this component renders inside a message part is fixed for its
-  // lifetime, so conditioning the selectors on it is safe.
-  const inPart = aui.part.source != null;
   const updateToolName = interactableToolName(name);
   const part = useAuiState((s) => {
-    if (!inPart) return undefined;
-    const p = s.part;
-    return p.type === "tool-call" ? p : undefined;
+    const p = s.optional.part;
+    return p?.type === "tool-call" ? p : undefined;
   });
 
   // Inside an update_{name} part, the instance id comes from the call itself:
@@ -123,7 +119,7 @@ const useInteractable = <TSchema extends Unstable_InteractableStateSchema>(
   const initialStateRef = useRef(config.initialState);
   initialStateRef.current = config.initialState;
 
-  const interactables = useAuiState(() => aui.unstable_interactables());
+  const interactables = useAuiState(() => aui.unstable_interactables);
 
   useEffect(() => {
     return interactables.register({

package/src/react/model-context/useInteractableState.ts

@@ -28,7 +28,7 @@ const useInteractableState = <TState>(
 
   const setState = useCallback(
     (updater: StateUpdater<TState>) => {
-      aui.unstable_interactables().setState(id, (prev) => {
+      aui.unstable_interactables.setState(id, (prev) => {
         if (typeof updater === "function") {
           return (updater as (prev: TState) => TState)(prev as TState);
         }
@@ -38,7 +38,7 @@ const useInteractableState = <TState>(
     [aui, id],
   );
 
-  const flush = useCallback(() => aui.unstable_interactables().flush(), [aui]);
+  const flush = useCallback(() => aui.unstable_interactables.flush(), [aui]);
 
   return [
     state,

package/src/react/primitive-hooks/useActionBarCopy.test.ts

@@ -18,10 +18,10 @@ const mocks = vi.hoisted(() => {
       },
     },
     aui: {
-      message: () => ({
+      message: {
         getCopyText: () => "Hello",
         setIsCopied,
-      }),
+      },
     },
   };
 });

package/src/react/primitive-hooks/useActionBarCopy.ts

@@ -25,15 +25,15 @@ export const useActionBarCopy = ({
   const copy = useCallback(() => {
     if (!copyToClipboard) return;
 
-    const valueToCopy = isEditing ? composerValue : aui.message().getCopyText();
+    const valueToCopy = isEditing ? composerValue : aui.message.getCopyText();
     if (!valueToCopy) return;
 
     // The rejection handler swallows clipboard write failures (permission denied,
     // API unavailable) so they don't surface as unhandled promise rejections.
     Promise.resolve(copyToClipboard(valueToCopy)).then(
       () => {
-        aui.message().setIsCopied(true);
-        setTimeout(() => aui.message().setIsCopied(false), copiedDuration);
+        aui.message.setIsCopied(true);
+        setTimeout(() => aui.message.setIsCopied(false), copiedDuration);
       },
       () => {},
     );

package/src/react/primitive-hooks/useActionBarEdit.ts

@@ -6,7 +6,7 @@ export const useActionBarEdit = () => {
   const disabled = useAuiState((s) => s.composer.isEditing);
 
   const edit = useCallback(() => {
-    aui.composer().beginEdit();
+    aui.composer.beginEdit();
   }, [aui]);
 
   return { edit, disabled };

package/src/react/primitive-hooks/useActionBarFeedback.ts

@@ -8,7 +8,7 @@ export const useActionBarFeedbackPositive = () => {
   );
 
   const submit = useCallback(() => {
-    aui.message().submitFeedback({ type: "positive" });
+    aui.message.submitFeedback({ type: "positive" });
   }, [aui]);
 
   return { submit, isSubmitted };
@@ -21,7 +21,7 @@ export const useActionBarFeedbackNegative = () => {
   );
 
   const submit = useCallback(() => {
-    aui.message().submitFeedback({ type: "negative" });
+    aui.message.submitFeedback({ type: "negative" });
   }, [aui]);
 
   return { submit, isSubmitted };

package/src/react/primitive-hooks/useActionBarReload.ts

@@ -11,7 +11,7 @@ export const useActionBarReload = () => {
   );
 
   const reload = useCallback(() => {
-    aui.message().reload();
+    aui.message.reload();
   }, [aui]);
 
   return { reload, disabled };

package/src/react/primitive-hooks/useActionBarSpeak.ts

@@ -13,7 +13,7 @@ export const useActionBarSpeak = () => {
   });
 
   const speak = useCallback(async () => {
-    aui.message().speak();
+    aui.message.speak();
   }, [aui]);
 
   return { speak, disabled };

package/src/react/primitive-hooks/useActionBarStopSpeaking.ts

@@ -6,7 +6,7 @@ export const useActionBarStopSpeaking = () => {
   const disabled = useAuiState((s) => s.message.speech == null);
 
   const stopSpeaking = useCallback(() => {
-    aui.message().stopSpeaking();
+    aui.message.stopSpeaking();
   }, [aui]);
 
   return { stopSpeaking, disabled };

package/src/react/primitive-hooks/useBranchPickerNext.ts

@@ -12,7 +12,7 @@ export const useBranchPickerNext = () => {
   });
 
   const next = useCallback(() => {
-    aui.message().switchToBranch({ position: "next" });
+    aui.message.switchToBranch({ position: "next" });
   }, [aui]);
 
   return { next, disabled };

package/src/react/primitive-hooks/useBranchPickerPrevious.ts

@@ -12,7 +12,7 @@ export const useBranchPickerPrevious = () => {
   });
 
   const previous = useCallback(() => {
-    aui.message().switchToBranch({ position: "previous" });
+    aui.message.switchToBranch({ position: "previous" });
   }, [aui]);
 
   return { previous, disabled };

package/src/react/primitive-hooks/useComposerAddAttachment.ts

@@ -8,7 +8,7 @@ export const useComposerAddAttachment = () => {
 
   const addAttachment = useCallback(
     (file: File | CreateAttachment) => {
-      return aui.composer().addAttachment(file);
+      return aui.composer.addAttachment(file);
     },
     [aui],
   );

package/src/react/primitive-hooks/useComposerCancel.ts

@@ -6,7 +6,7 @@ export const useComposerCancel = () => {
   const disabled = useAuiState((s) => !s.composer.canCancel);
 
   const cancel = useCallback(() => {
-    aui.composer().cancel();
+    aui.composer.cancel();
   }, [aui]);
 
   return { cancel, disabled };

package/src/react/primitive-hooks/useComposerDictate.ts

@@ -11,7 +11,7 @@ export const useComposerDictate = () => {
   );
 
   const startDictation = useCallback(() => {
-    aui.composer().startDictation();
+    aui.composer.startDictation();
   }, [aui]);
 
   return { startDictation, disabled };

package/src/react/primitive-hooks/useComposerSend.ts

@@ -12,7 +12,7 @@ export const useComposerSend = () => {
 
   const send = useCallback(
     (opts?: ComposerSendOptions) => {
-      aui.composer().send(opts);
+      aui.composer.send(opts);
     },
     [aui],
   );

package/src/react/primitive-hooks/useEditComposerCancel.ts

@@ -5,7 +5,7 @@ export const useEditComposerCancel = () => {
   const aui = useAui();
 
   const cancel = useCallback(() => {
-    aui.composer().cancel();
+    aui.composer.cancel();
   }, [aui]);
 
   return { cancel };