@assistant-ui/core 0.2.10 → 0.2.11

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

@@ -1 +1 @@
-{"version":3,"file":"ToolInvocationTracker.js","names":[],"sources":["../../../src/runtimes/tool-invocations/ToolInvocationTracker.ts"],"sourcesContent":["declare const process: { env: { NODE_ENV?: string } };\n\nimport {\n  createAssistantStreamController,\n  type ToolCallStreamController,\n  ToolResponse,\n  unstable_toolResultStream,\n  type Tool,\n  type ToolModelContentPart,\n} from \"assistant-stream\";\nimport {\n  AssistantMetaTransformStream,\n  type ReadonlyJSONValue,\n} from \"assistant-stream/utils\";\nimport { isJSONValueEqual } from \"../../utils/json/is-json-equal\";\nimport type { ThreadMessage } from \"../../types/message\";\n\n/**\n * Streaming execution state for a frontend tool.\n */\nexport type ToolExecutionStatus =\n  | { type: \"executing\" }\n  | {\n      type: \"interrupt\";\n      payload: { type: \"human\"; payload: unknown };\n    };\n\nexport type AddToolResultCommand = {\n  readonly type: \"add-tool-result\";\n  readonly toolCallId: string;\n  readonly toolName: string;\n  readonly result: ReadonlyJSONValue;\n  readonly isError: boolean;\n  readonly artifact?: ReadonlyJSONValue;\n  readonly modelContent?: readonly ToolModelContentPart[];\n};\n\nexport type ToolInvocationTrackerSnapshot = {\n  readonly messages: readonly ThreadMessage[];\n  /** Whether the producing runtime is currently streaming new output. */\n  readonly isRunning: boolean;\n  /**\n   * Whether the producing runtime is still loading historical state.\n   * When `true`, every snapshot is treated as historical (no `streamCall` /\n   * `execute` fires). When `false`, processing resumes as live.\n   */\n  readonly isLoading?: boolean;\n};\n\nexport type ToolInvocationTrackerCallbacks = {\n  /**\n   * Invoked when a client-side `execute()` returns a result and the runtime\n   * needs to feed it back into the conversation.\n   */\n  onResult: (command: AddToolResultCommand) => void;\n  /**\n   * Invoked whenever the per-tool-call status map changes (executing /\n   * interrupt / cleared). The callback receives a fresh map; mutating the\n   * argument is not supported.\n   */\n  onStatusesChange: (\n    statuses: ReadonlyMap<string, ToolExecutionStatus>,\n  ) => void;\n};\n\ntype ToolCallEntry = {\n  toolName: string;\n  argsText: string;\n  hasResult: boolean;\n} & (\n  | {\n      /** Restored phase — observed during a history-load snapshot. */\n      controller?: undefined;\n      argsComplete?: undefined;\n    }\n  | {\n      /** Active phase — chunks are flowing through `controller`. */\n      controller: ToolCallStreamController;\n      argsComplete: boolean;\n    }\n);\n\nconst isArgsTextComplete = (argsText: string) => {\n  try {\n    JSON.parse(argsText);\n    return true;\n  } catch {\n    return false;\n  }\n};\n\nconst parseArgsText = (argsText: string) => {\n  try {\n    return JSON.parse(argsText);\n  } catch {\n    return undefined;\n  }\n};\n\nconst isEquivalentCompleteArgsText = (previous: string, next: string) => {\n  const previousValue = parseArgsText(previous);\n  const nextValue = parseArgsText(next);\n  if (previousValue === undefined || nextValue === undefined) return false;\n  return isJSONValueEqual(previousValue, nextValue);\n};\n\n/**\n * Plain-class port of the former `useToolInvocations` React hook. Owns the\n * assistant-stream pipeline that drives client-side `streamCall` / `execute`\n * for tool-call parts surfaced by a thread runtime, plus the per-tool-call\n * status map that consumers render against.\n *\n * **Contract**: `streamCall` (and `execute`) fires exactly once per logical\n * `toolCallId`. Args mutations after first completion, result replacement,\n * and result clearing are *not* surfaced through additional `streamCall`\n * invocations — by design — so hosts cannot observe spurious re-fires of\n * side effects. The follow-up `reader.events()` API will expose those\n * post-completion transitions to consumers that opt in.\n *\n * State-transition safety: every public method that observes runtime state\n * (`setState`, `reset`, `abort`, `resume`) wraps its work in try/catch and\n * logs to `console.error` rather than throwing. The tracker is built into\n * the hot message-processing path, so a malformed snapshot must never crash\n * the host runtime. See ./EDGE_CASES.md for the known non-trivial state\n * transitions and what each does today.\n */\nexport class ToolInvocationTracker {\n  private readonly _getTools: () => Record<string, Tool> | undefined;\n  private readonly _callbacks: ToolInvocationTrackerCallbacks;\n\n  private readonly _entries = new Map<string, ToolCallEntry>();\n  /**\n   * Tool call ids whose `execute` should be short-circuited in the wrapper.\n   * Populated when an entry is created with a result already attached\n   * (history reload, mid-run resume, etc.) — `execute` is suppressed so\n   * client-side side effects don't double-run. Membership outlives the\n   * entry: `reset()` deliberately does *not* clear this so post-abort\n   * cancellation `result` chunks for pre-resolved entries can still be\n   * recognized and dropped. Growth is bounded by the number of pre-resolved\n   * tool calls observed in the session.\n   */\n  private readonly _skipExecuteStreamIds = new Set<string>();\n  private readonly _humanInput = new Map<\n    string,\n    {\n      resolve: (payload: unknown) => void;\n      reject: (reason: unknown) => void;\n    }\n  >();\n  /** In-flight `execute` invocations keyed by tool call id. */\n  private readonly _executing = new Set<string>();\n  private readonly _settledResolvers: Array<() => void> = [];\n\n  private _statuses = new Map<string, ToolExecutionStatus>();\n\n  private _ac: AbortController = new AbortController();\n  private _pendingRestore = true;\n\n  /** Cached last snapshot, used to skip processing on identical re-renders. */\n  private _lastSnapshot: ToolInvocationTrackerSnapshot | null = null;\n  private _isRunning = false;\n\n  private _controller!: ReturnType<typeof createAssistantStreamController>[1];\n\n  /**\n   * Set when the assistant-stream pipeline has died (errored out via\n   * `.pipeTo(...).catch(...)`). The next `setState` re-initializes the\n   * pipeline and demotes all active entries to restored so they survive\n   * across the restart without re-firing `streamCall` (preserves the\n   * \"exactly once\" contract). Capped at a single auto-restart per session\n   * — repeated failures keep the tracker dead with a more visible error.\n   */\n  private _pipelineDead = false;\n  private _pipelineRestartUsed = false;\n\n  constructor(\n    getTools: () => Record<string, Tool> | undefined,\n    callbacks: ToolInvocationTrackerCallbacks,\n  ) {\n    this._getTools = getTools;\n    this._callbacks = callbacks;\n\n    this._initPipeline();\n  }\n\n  /**\n   * Build the assistant-stream pipeline. Called once from the constructor\n   * and at most once again if `_pipelineDead` is set (see F.4 in\n   * EDGE_CASES.md).\n   */\n  private _initPipeline(): void {\n    const [stream, controller] = createAssistantStreamController();\n    this._controller = controller;\n\n    const transform = unstable_toolResultStream(\n      () => this._getWrappedTools(),\n      () => this._ac.signal,\n      (toolCallId, payload) => this._onHumanInput(toolCallId, payload),\n      {\n        onExecutionStart: (id) => this._onExecutionStart(id),\n        onExecutionEnd: (id) => this._onExecutionEnd(id),\n      },\n    );\n\n    stream\n      .pipeThrough(transform)\n      .pipeThrough(new AssistantMetaTransformStream())\n      .pipeTo(\n        new WritableStream({\n          write: (chunk) => {\n            try {\n              if (chunk.type !== \"result\") return;\n              this._handleResultChunk(chunk);\n            } catch (err) {\n              console.error(\n                \"[ToolInvocationTracker] result chunk handling failed\",\n                err,\n              );\n            }\n          },\n        }),\n      )\n      .catch((err) => {\n        console.error(\n          \"[ToolInvocationTracker] stream pipeline failed; will attempt single restart on next setState\",\n          err,\n        );\n        this._pipelineDead = true;\n      });\n  }\n\n  // ───────────────────────── public API ─────────────────────────\n\n  /**\n   * Feed the next observed snapshot into the tracker. Called from the host\n   * runtime whenever its message list / running state changes.\n   */\n  public setState(snapshot: ToolInvocationTrackerSnapshot): void {\n    try {\n      // Recover from a dead pipeline before processing anything. We demote\n      // all active entries to \"restored\" so the rebuilt pipeline does not\n      // re-fire `streamCall` for tool calls that already fired pre-death;\n      // preserves the \"exactly once per toolCallId\" contract.\n      if (this._pipelineDead) {\n        if (this._pipelineRestartUsed) {\n          // Already retried once and failed again. Stay dead.\n          return;\n        }\n        this._pipelineRestartUsed = true;\n        this._pipelineDead = false;\n        this._demoteEntriesToRestored();\n        this._executing.clear();\n        this._ac = new AbortController();\n        this._initPipeline();\n        // Fall through and process the snapshot against the fresh pipeline.\n      }\n\n      // Identical snapshot — skip processing entirely. Note: external-store\n      // runtimes rebuild the messages array on every adapter update, so this\n      // fast-path rarely triggers there; it's primarily for the React-hook\n      // shim where state references are stable.\n      if (\n        this._lastSnapshot &&\n        this._lastSnapshot.messages === snapshot.messages &&\n        this._lastSnapshot.isRunning === snapshot.isRunning &&\n        this._lastSnapshot.isLoading === snapshot.isLoading\n      ) {\n        return;\n      }\n\n      // While the host is still loading initial state, treat every snapshot\n      // as historical: tool calls are recorded so the next live snapshot can\n      // diff against them, but `streamCall` / `execute` do not fire.\n      const restoreFromLoading = snapshot.isLoading === true;\n      if (restoreFromLoading) {\n        this._pendingRestore = true;\n      }\n\n      // E.4 / AF3 — only mark `_lastSnapshot`/`_isRunning` as observed after\n      // processing succeeds. If `_processMessages` throws, the next snapshot\n      // (even if identical) gets re-processed against the recovered state.\n      const previousIsRunning = this._isRunning;\n      this._isRunning = snapshot.isRunning;\n      try {\n        this._processMessages(snapshot.messages);\n      } catch (err) {\n        this._isRunning = previousIsRunning;\n        throw err;\n      }\n      this._lastSnapshot = snapshot;\n      this._pendingRestore = false;\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] setState failed; snapshot dropped\",\n        err,\n      );\n    }\n  }\n\n  /**\n   * Reset the tracker so the next observed snapshot is treated as historical.\n   * Clears entries and aborts any in-flight executions. Used by callers like\n   * `importExternalState` to mark a freshly loaded state as restored.\n   */\n  public reset(): void {\n    try {\n      this._pendingRestore = true;\n      this._entries.clear();\n      this._lastSnapshot = null;\n      // `_skipExecuteStreamIds` is intentionally not cleared — see field doc.\n      void this.abort().finally(() => {\n        this._executing.clear();\n      });\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] reset failed\", err);\n    }\n  }\n\n  /**\n   * Abort any in-flight `execute()` invocations. Resolves once all of them\n   * have settled (or immediately if none are running).\n   */\n  public abort(): Promise<void> {\n    try {\n      this._humanInput.forEach(({ reject }) => {\n        try {\n          reject(new Error(\"Tool execution aborted\"));\n        } catch {\n          // host rejection handler threw — already in the abort path,\n          // swallow so we continue cleaning up.\n        }\n      });\n      this._humanInput.clear();\n\n      this._ac.abort();\n      this._ac = new AbortController();\n\n      if (this._executing.size === 0) {\n        return Promise.resolve();\n      }\n      return new Promise<void>((resolve) => {\n        this._settledResolvers.push(resolve);\n      });\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] abort failed\", err);\n      return Promise.resolve();\n    }\n  }\n\n  /**\n   * Resolve a pending human-input request for the given tool call. Returns\n   * `true` if a pending request was resumed, `false` if the tracker has no\n   * outstanding request for that id (the caller should fall back to its own\n   * dispatch path).\n   */\n  public resume(toolCallId: string, payload: unknown): boolean {\n    try {\n      const handlers = this._humanInput.get(toolCallId);\n      if (!handlers) return false;\n      this._humanInput.delete(toolCallId);\n      this._setStatus(toolCallId, { type: \"executing\" });\n      handlers.resolve(payload);\n      return true;\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] resume failed\", err);\n      return false;\n    }\n  }\n\n  /**\n   * Returns the current tool execution status map. The returned `Map` is\n   * the tracker's internal store — do not mutate it. Treat the reference\n   * as a snapshot that may be replaced wholesale on the next status\n   * transition.\n   */\n  public getStatuses(): ReadonlyMap<string, ToolExecutionStatus> {\n    return this._statuses;\n  }\n\n  // ───────────────────── internal: tool wrapping ─────────────────────\n\n  private _getWrappedTools(): Record<string, Tool> | undefined {\n    const tools = this._getTools();\n    if (!tools) return undefined;\n\n    return Object.fromEntries(\n      Object.entries(tools).map(([name, tool]) => {\n        const execute = tool.execute;\n        if (execute === undefined) return [name, tool];\n\n        const wrappedTool = {\n          ...tool,\n          execute: (\n            ...[args, context]: Parameters<NonNullable<typeof execute>>\n          ) => {\n            if (this._skipExecuteStreamIds.has(context.toolCallId)) {\n              // Pre-resolved tool call: never invoke the host's execute.\n              // Returning a never-settling Promise keeps the executor's\n              // pending entry alive but enqueues nothing.\n              return new Promise(() => {}) as never;\n            }\n            return execute(args, context);\n          },\n        } as Tool;\n        return [name, wrappedTool];\n      }),\n    ) as Record<string, Tool>;\n  }\n\n  // ──────────────── internal: execution lifecycle callbacks ────────────────\n\n  private _onHumanInput(\n    toolCallId: string,\n    payload: unknown,\n  ): Promise<unknown> {\n    return new Promise<unknown>((resolve, reject) => {\n      const previous = this._humanInput.get(toolCallId);\n      if (previous) {\n        try {\n          previous.reject(\n            new Error(\"Human input request was superseded by a new request\"),\n          );\n        } catch {\n          // host rejection handler threw; ignore and proceed\n        }\n      }\n      this._humanInput.set(toolCallId, { resolve, reject });\n      this._setStatus(toolCallId, {\n        type: \"interrupt\",\n        payload: { type: \"human\", payload },\n      });\n    });\n  }\n\n  private _onExecutionStart(toolCallId: string): void {\n    if (this._skipExecuteStreamIds.has(toolCallId)) return;\n\n    this._executing.add(toolCallId);\n    this._setStatus(toolCallId, { type: \"executing\" });\n  }\n\n  private _onExecutionEnd(toolCallId: string): void {\n    if (!this._executing.delete(toolCallId)) return;\n\n    this._deleteStatus(toolCallId);\n\n    if (this._executing.size === 0) {\n      const resolvers = this._settledResolvers.splice(0);\n      resolvers.forEach((resolve) => {\n        try {\n          resolve();\n        } catch {\n          // ignore — settled-resolver consumer threw\n        }\n      });\n    }\n  }\n\n  private _handleResultChunk(chunk: {\n    type: \"result\";\n    result: ReadonlyJSONValue;\n    isError: boolean;\n    artifact?: ReadonlyJSONValue;\n    modelContent?: readonly ToolModelContentPart[];\n    meta: { toolCallId: string; toolName: string };\n  }): void {\n    const toolCallId = chunk.meta.toolCallId;\n    const entry = this._entries.get(toolCallId);\n\n    // Pre-resolved tool call whose entry has been cleared by `reset()`.\n    // The post-abort cancellation chunk lands here after the entry is\n    // gone; suppress via the long-lived skip-execute marker.\n    if (!entry && this._skipExecuteStreamIds.has(toolCallId)) {\n      return;\n    }\n\n    // The host already set the result (via the live snapshot's\n    // `setResponse` path). Suppress the executor's redundant emit.\n    if (entry?.hasResult) return;\n\n    this._invokeOnResult({\n      type: \"add-tool-result\",\n      toolCallId,\n      toolName: chunk.meta.toolName,\n      result: chunk.result,\n      isError: chunk.isError,\n      ...(chunk.artifact !== undefined && { artifact: chunk.artifact }),\n      ...(chunk.modelContent !== undefined && {\n        modelContent: chunk.modelContent,\n      }),\n    });\n  }\n\n  // ──────────────── internal: callback invocation (AF1/AF2) ────────────────\n\n  private _invokeOnResult(command: AddToolResultCommand): void {\n    try {\n      this._callbacks.onResult(command);\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] onResult callback threw; result dropped\",\n        err,\n      );\n    }\n  }\n\n  private _invokeOnStatusesChange(): void {\n    try {\n      this._callbacks.onStatusesChange(this._statuses);\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] onStatusesChange callback threw; status change not propagated\",\n        err,\n      );\n    }\n  }\n\n  // ──────────────── internal: status map mutations ────────────────\n\n  private _setStatus(toolCallId: string, status: ToolExecutionStatus): void {\n    const next = new Map(this._statuses);\n    next.set(toolCallId, status);\n    this._statuses = next;\n    this._invokeOnStatusesChange();\n  }\n\n  private _deleteStatus(toolCallId: string): void {\n    if (!this._statuses.has(toolCallId)) return;\n    const next = new Map(this._statuses);\n    next.delete(toolCallId);\n    this._statuses = next;\n    this._invokeOnStatusesChange();\n  }\n\n  // ──────────────── internal: snapshot processing ────────────────\n\n  private _hasExecutableTool(toolName: string): boolean {\n    const tool = this._getTools()?.[toolName];\n    return tool?.execute !== undefined || tool?.streamCall !== undefined;\n  }\n\n  private _shouldCloseArgsStream({\n    toolName,\n    argsText,\n    hasResult,\n  }: {\n    toolName: string;\n    argsText: string;\n    hasResult: boolean;\n  }): boolean {\n    if (hasResult) return true;\n    if (!this._hasExecutableTool(toolName)) {\n      return !this._isRunning && isArgsTextComplete(argsText);\n    }\n    return isArgsTextComplete(argsText);\n  }\n\n  private _startActiveEntry(\n    toolCallId: string,\n    toolName: string,\n    skipExecute: boolean,\n  ): ToolCallEntry {\n    const toolCallController = this._controller.addToolCallPart({\n      toolName,\n      toolCallId,\n    });\n    if (skipExecute) {\n      this._skipExecuteStreamIds.add(toolCallId);\n    }\n    const entry: ToolCallEntry = {\n      toolName,\n      controller: toolCallController,\n      argsText: \"\",\n      hasResult: false,\n      argsComplete: false,\n    };\n    this._entries.set(toolCallId, entry);\n    return entry;\n  }\n\n  /**\n   * Demote every active entry back to the restored phase. Used by the\n   * pipeline-restart path so that, after a fresh pipeline is built, the\n   * next observed snapshot does not re-fire `streamCall` for tool calls\n   * that already fired pre-death. Args / hasResult tracking is preserved\n   * so signature comparisons still work.\n   */\n  private _demoteEntriesToRestored(): void {\n    for (const [toolCallId, entry] of this._entries) {\n      if (!entry.controller) continue;\n      this._entries.set(toolCallId, {\n        toolName: entry.toolName,\n        argsText: entry.argsText,\n        hasResult: entry.hasResult,\n      });\n    }\n  }\n\n  private _processArgsText(\n    entry: ToolCallEntry,\n    content: {\n      toolCallId: string;\n      toolName: string;\n      argsText: string;\n      result?: unknown;\n    },\n  ): void {\n    if (!entry.controller) return;\n    const hasResult = content.result !== undefined;\n\n    if (content.argsText !== entry.argsText) {\n      let shouldWriteArgsText = true;\n\n      if (entry.argsComplete) {\n        if (isEquivalentCompleteArgsText(entry.argsText, content.argsText)) {\n          // A.3 — key reorder. Track new text, no re-fire needed.\n          entry.argsText = content.argsText;\n          shouldWriteArgsText = false;\n        } else {\n          // A.4 — args changed after first completion. Under the\n          // \"exactly once per toolCallId\" contract we do not restart the\n          // stream. The host's existing `streamCall` keeps its original\n          // args view; the snapshot's new text is recorded for diffing\n          // but not surfaced. Events API in a follow-up will expose this\n          // to consumers that opt in.\n          if (process.env.NODE_ENV !== \"production\") {\n            console.warn(\n              \"[ToolInvocationTracker] argsText changed after first completion; not re-firing streamCall (see EDGE_CASES.md A.4)\",\n              {\n                previous: entry.argsText,\n                next: content.argsText,\n                toolCallId: content.toolCallId,\n              },\n            );\n          }\n          shouldWriteArgsText = false;\n        }\n      } else if (!content.argsText.startsWith(entry.argsText)) {\n        if (\n          isArgsTextComplete(entry.argsText) &&\n          isArgsTextComplete(content.argsText) &&\n          isEquivalentCompleteArgsText(entry.argsText, content.argsText)\n        ) {\n          const shouldClose = this._shouldCloseArgsStream({\n            toolName: content.toolName,\n            argsText: content.argsText,\n            hasResult,\n          });\n          if (shouldClose) entry.controller.argsText.close();\n          entry.argsText = content.argsText;\n          entry.argsComplete = shouldClose;\n          shouldWriteArgsText = false;\n        } else {\n          // A.2 — args regressed mid-stream. Under the \"exactly once\"\n          // contract we do not restart. The controller keeps whatever\n          // prefix we already streamed; subsequent prefix-respecting\n          // updates can still flow against it. Snapshots that never\n          // re-converge to a prefix will leave the controller's args\n          // view stale relative to the snapshot. Events API in a\n          // follow-up will expose this to consumers that opt in.\n          if (process.env.NODE_ENV !== \"production\") {\n            console.warn(\n              \"[ToolInvocationTracker] argsText regressed mid-stream; not restarting (see EDGE_CASES.md A.2)\",\n              {\n                previous: entry.argsText,\n                next: content.argsText,\n                toolCallId: content.toolCallId,\n              },\n            );\n          }\n          shouldWriteArgsText = false;\n        }\n      }\n\n      if (shouldWriteArgsText && entry.controller) {\n        const delta = content.argsText.slice(entry.argsText.length);\n        entry.controller.argsText.append(delta);\n        const shouldClose = this._shouldCloseArgsStream({\n          toolName: content.toolName,\n          argsText: content.argsText,\n          hasResult,\n        });\n        if (shouldClose) entry.controller.argsText.close();\n        entry.argsText = content.argsText;\n        entry.argsComplete = shouldClose;\n      }\n    }\n\n    if (!entry.argsComplete && entry.controller) {\n      const shouldClose = this._shouldCloseArgsStream({\n        toolName: content.toolName,\n        argsText: content.argsText,\n        hasResult,\n      });\n      if (shouldClose) {\n        entry.controller.argsText.close();\n        entry.argsText = content.argsText;\n        entry.argsComplete = true;\n      }\n    }\n  }\n\n  private _processMessages(messages: readonly ThreadMessage[]): void {\n    const isRestore = this._pendingRestore;\n\n    for (const message of messages) {\n      if (!message || !Array.isArray((message as ThreadMessage).content)) {\n        continue;\n      }\n      for (const content of message.content as readonly ThreadMessage[\"content\"][number][]) {\n        if (!content || content.type !== \"tool-call\") continue;\n\n        const existing = this._entries.get(content.toolCallId);\n\n        if (isRestore) {\n          // Don't overwrite an already-active entry (e.g. live tool-call\n          // observed before this restore snapshot landed). Restore can\n          // only seed entries the runtime has never seen.\n          if (!existing?.controller) {\n            this._entries.set(content.toolCallId, {\n              toolName: content.toolName,\n              argsText: content.argsText,\n              hasResult: content.result !== undefined,\n            });\n          }\n          if (content.messages) this._processMessages(content.messages);\n          continue;\n        }\n\n        // Live snapshot.\n        let entry = existing;\n\n        if (entry && !entry.controller) {\n          // Restored entry observed in a live snapshot. Promote if its\n          // signature has changed; otherwise treat as still-historical.\n          const signatureChanged =\n            content.argsText !== entry.argsText ||\n            (content.result !== undefined) !== entry.hasResult;\n          if (!signatureChanged) {\n            if (content.messages) this._processMessages(content.messages);\n            continue;\n          }\n          this._entries.delete(content.toolCallId);\n          entry = undefined;\n        }\n\n        if (!entry) {\n          entry = this._startActiveEntry(\n            content.toolCallId,\n            content.toolName,\n            content.result !== undefined,\n          );\n        }\n\n        this._processArgsText(entry, content);\n\n        if (content.result !== undefined && !entry.hasResult) {\n          // `entry` is in active phase from this point — either just\n          // created by `_startActiveEntry`, or pre-existing with a live\n          // controller. Narrow once instead of asserting at every use.\n          const { controller: activeController } = entry;\n          if (!activeController) continue;\n          entry.hasResult = true;\n          entry.argsComplete = true;\n          activeController.setResponse(\n            new ToolResponse({\n              result: content.result as ReadonlyJSONValue,\n              artifact: content.artifact as ReadonlyJSONValue | undefined,\n              isError: content.isError,\n              ...(content.modelContent !== undefined\n                ? { modelContent: content.modelContent }\n                : {}),\n            }),\n          );\n          activeController.close();\n        }\n\n        if (content.messages) this._processMessages(content.messages);\n      }\n    }\n  }\n}\n"],"mappings":";;;;AAkFA,MAAM,sBAAsB,aAAqB;CAC/C,IAAI;EACF,KAAK,MAAM,QAAQ;EACnB,OAAO;CACT,QAAQ;EACN,OAAO;CACT;AACF;AAEA,MAAM,iBAAiB,aAAqB;CAC1C,IAAI;EACF,OAAO,KAAK,MAAM,QAAQ;CAC5B,QAAQ;EACN;CACF;AACF;AAEA,MAAM,gCAAgC,UAAkB,SAAiB;CACvE,MAAM,gBAAgB,cAAc,QAAQ;CAC5C,MAAM,YAAY,cAAc,IAAI;CACpC,IAAI,kBAAkB,KAAA,KAAa,cAAc,KAAA,GAAW,OAAO;CACnE,OAAO,iBAAiB,eAAe,SAAS;AAClD;;;;;;;;;;;;;;;;;;;;;AAsBA,IAAa,wBAAb,MAAmC;CACjC;CACA;CAEA,2BAA4B,IAAI,IAA2B;;;;;;;;;;;CAW3D,wCAAyC,IAAI,IAAY;CACzD,8BAA+B,IAAI,IAMjC;;CAEF,6BAA8B,IAAI,IAAY;CAC9C,oBAAwD,CAAC;CAEzD,4BAAoB,IAAI,IAAiC;CAEzD,MAA+B,IAAI,gBAAgB;CACnD,kBAA0B;;CAG1B,gBAA8D;CAC9D,aAAqB;CAErB;;;;;;;;;CAUA,gBAAwB;CACxB,uBAA+B;CAE/B,YACE,UACA,WACA;EACA,KAAK,YAAY;EACjB,KAAK,aAAa;EAElB,KAAK,cAAc;CACrB;;;;;;CAOA,gBAA8B;EAC5B,MAAM,CAAC,QAAQ,cAAc,gCAAgC;EAC7D,KAAK,cAAc;EAEnB,MAAM,YAAY,gCACV,KAAK,iBAAiB,SACtB,KAAK,IAAI,SACd,YAAY,YAAY,KAAK,cAAc,YAAY,OAAO,GAC/D;GACE,mBAAmB,OAAO,KAAK,kBAAkB,EAAE;GACnD,iBAAiB,OAAO,KAAK,gBAAgB,EAAE;EACjD,CACF;EAEA,OACG,YAAY,SAAS,EACrB,YAAY,IAAI,6BAA6B,CAAC,EAC9C,OACC,IAAI,eAAe,EACjB,QAAQ,UAAU;GAChB,IAAI;IACF,IAAI,MAAM,SAAS,UAAU;IAC7B,KAAK,mBAAmB,KAAK;GAC/B,SAAS,KAAK;IACZ,QAAQ,MACN,wDACA,GACF;GACF;EACF,EACF,CAAC,CACH,EACC,OAAO,QAAQ;GACd,QAAQ,MACN,gGACA,GACF;GACA,KAAK,gBAAgB;EACvB,CAAC;CACL;;;;;CAQA,SAAgB,UAA+C;EAC7D,IAAI;GAKF,IAAI,KAAK,eAAe;IACtB,IAAI,KAAK,sBAEP;IAEF,KAAK,uBAAuB;IAC5B,KAAK,gBAAgB;IACrB,KAAK,yBAAyB;IAC9B,KAAK,WAAW,MAAM;IACtB,KAAK,MAAM,IAAI,gBAAgB;IAC/B,KAAK,cAAc;GAErB;GAMA,IACE,KAAK,iBACL,KAAK,cAAc,aAAa,SAAS,YACzC,KAAK,cAAc,cAAc,SAAS,aAC1C,KAAK,cAAc,cAAc,SAAS,WAE1C;GAOF,IAD2B,SAAS,cAAc,MAEhD,KAAK,kBAAkB;GAMzB,MAAM,oBAAoB,KAAK;GAC/B,KAAK,aAAa,SAAS;GAC3B,IAAI;IACF,KAAK,iBAAiB,SAAS,QAAQ;GACzC,SAAS,KAAK;IACZ,KAAK,aAAa;IAClB,MAAM;GACR;GACA,KAAK,gBAAgB;GACrB,KAAK,kBAAkB;EACzB,SAAS,KAAK;GACZ,QAAQ,MACN,6DACA,GACF;EACF;CACF;;;;;;CAOA,QAAqB;EACnB,IAAI;GACF,KAAK,kBAAkB;GACvB,KAAK,SAAS,MAAM;GACpB,KAAK,gBAAgB;GAErB,KAAU,MAAM,EAAE,cAAc;IAC9B,KAAK,WAAW,MAAM;GACxB,CAAC;EACH,SAAS,KAAK;GACZ,QAAQ,MAAM,wCAAwC,GAAG;EAC3D;CACF;;;;;CAMA,QAA8B;EAC5B,IAAI;GACF,KAAK,YAAY,SAAS,EAAE,aAAa;IACvC,IAAI;KACF,uBAAO,IAAI,MAAM,wBAAwB,CAAC;IAC5C,QAAQ,CAGR;GACF,CAAC;GACD,KAAK,YAAY,MAAM;GAEvB,KAAK,IAAI,MAAM;GACf,KAAK,MAAM,IAAI,gBAAgB;GAE/B,IAAI,KAAK,WAAW,SAAS,GAC3B,OAAO,QAAQ,QAAQ;GAEzB,OAAO,IAAI,SAAe,YAAY;IACpC,KAAK,kBAAkB,KAAK,OAAO;GACrC,CAAC;EACH,SAAS,KAAK;GACZ,QAAQ,MAAM,wCAAwC,GAAG;GACzD,OAAO,QAAQ,QAAQ;EACzB;CACF;;;;;;;CAQA,OAAc,YAAoB,SAA2B;EAC3D,IAAI;GACF,MAAM,WAAW,KAAK,YAAY,IAAI,UAAU;GAChD,IAAI,CAAC,UAAU,OAAO;GACtB,KAAK,YAAY,OAAO,UAAU;GAClC,KAAK,WAAW,YAAY,EAAE,MAAM,YAAY,CAAC;GACjD,SAAS,QAAQ,OAAO;GACxB,OAAO;EACT,SAAS,KAAK;GACZ,QAAQ,MAAM,yCAAyC,GAAG;GAC1D,OAAO;EACT;CACF;;;;;;;CAQA,cAA+D;EAC7D,OAAO,KAAK;CACd;CAIA,mBAA6D;EAC3D,MAAM,QAAQ,KAAK,UAAU;EAC7B,IAAI,CAAC,OAAO,OAAO,KAAA;EAEnB,OAAO,OAAO,YACZ,OAAO,QAAQ,KAAK,EAAE,KAAK,CAAC,MAAM,UAAU;GAC1C,MAAM,UAAU,KAAK;GACrB,IAAI,YAAY,KAAA,GAAW,OAAO,CAAC,MAAM,IAAI;GAgB7C,OAAO,CAAC,MAAM;IAbZ,GAAG;IACH,UACE,GAAG,CAAC,MAAM,aACP;KACH,IAAI,KAAK,sBAAsB,IAAI,QAAQ,UAAU,GAInD,OAAO,IAAI,cAAc,CAAC,CAAC;KAE7B,OAAO,QAAQ,MAAM,OAAO;IAC9B;GAEsB,CAAC;EAC3B,CAAC,CACH;CACF;CAIA,cACE,YACA,SACkB;EAClB,OAAO,IAAI,SAAkB,SAAS,WAAW;GAC/C,MAAM,WAAW,KAAK,YAAY,IAAI,UAAU;GAChD,IAAI,UACF,IAAI;IACF,SAAS,uBACP,IAAI,MAAM,qDAAqD,CACjE;GACF,QAAQ,CAER;GAEF,KAAK,YAAY,IAAI,YAAY;IAAE;IAAS;GAAO,CAAC;GACpD,KAAK,WAAW,YAAY;IAC1B,MAAM;IACN,SAAS;KAAE,MAAM;KAAS;IAAQ;GACpC,CAAC;EACH,CAAC;CACH;CAEA,kBAA0B,YAA0B;EAClD,IAAI,KAAK,sBAAsB,IAAI,UAAU,GAAG;EAEhD,KAAK,WAAW,IAAI,UAAU;EAC9B,KAAK,WAAW,YAAY,EAAE,MAAM,YAAY,CAAC;CACnD;CAEA,gBAAwB,YAA0B;EAChD,IAAI,CAAC,KAAK,WAAW,OAAO,UAAU,GAAG;EAEzC,KAAK,cAAc,UAAU;EAE7B,IAAI,KAAK,WAAW,SAAS,GAE3B,KADuB,kBAAkB,OAAO,CACxC,EAAE,SAAS,YAAY;GAC7B,IAAI;IACF,QAAQ;GACV,QAAQ,CAER;EACF,CAAC;CAEL;CAEA,mBAA2B,OAOlB;EACP,MAAM,aAAa,MAAM,KAAK;EAC9B,MAAM,QAAQ,KAAK,SAAS,IAAI,UAAU;EAK1C,IAAI,CAAC,SAAS,KAAK,sBAAsB,IAAI,UAAU,GACrD;EAKF,IAAI,OAAO,WAAW;EAEtB,KAAK,gBAAgB;GACnB,MAAM;GACN;GACA,UAAU,MAAM,KAAK;GACrB,QAAQ,MAAM;GACd,SAAS,MAAM;GACf,GAAI,MAAM,aAAa,KAAA,KAAa,EAAE,UAAU,MAAM,SAAS;GAC/D,GAAI,MAAM,iBAAiB,KAAA,KAAa,EACtC,cAAc,MAAM,aACtB;EACF,CAAC;CACH;CAIA,gBAAwB,SAAqC;EAC3D,IAAI;GACF,KAAK,WAAW,SAAS,OAAO;EAClC,SAAS,KAAK;GACZ,QAAQ,MACN,mEACA,GACF;EACF;CACF;CAEA,0BAAwC;EACtC,IAAI;GACF,KAAK,WAAW,iBAAiB,KAAK,SAAS;EACjD,SAAS,KAAK;GACZ,QAAQ,MACN,yFACA,GACF;EACF;CACF;CAIA,WAAmB,YAAoB,QAAmC;EACxE,MAAM,OAAO,IAAI,IAAI,KAAK,SAAS;EACnC,KAAK,IAAI,YAAY,MAAM;EAC3B,KAAK,YAAY;EACjB,KAAK,wBAAwB;CAC/B;CAEA,cAAsB,YAA0B;EAC9C,IAAI,CAAC,KAAK,UAAU,IAAI,UAAU,GAAG;EACrC,MAAM,OAAO,IAAI,IAAI,KAAK,SAAS;EACnC,KAAK,OAAO,UAAU;EACtB,KAAK,YAAY;EACjB,KAAK,wBAAwB;CAC/B;CAIA,mBAA2B,UAA2B;EACpD,MAAM,OAAO,KAAK,UAAU,IAAI;EAChC,OAAO,MAAM,YAAY,KAAA,KAAa,MAAM,eAAe,KAAA;CAC7D;CAEA,uBAA+B,EAC7B,UACA,UACA,aAKU;EACV,IAAI,WAAW,OAAO;EACtB,IAAI,CAAC,KAAK,mBAAmB,QAAQ,GACnC,OAAO,CAAC,KAAK,cAAc,mBAAmB,QAAQ;EAExD,OAAO,mBAAmB,QAAQ;CACpC;CAEA,kBACE,YACA,UACA,aACe;EACf,MAAM,qBAAqB,KAAK,YAAY,gBAAgB;GAC1D;GACA;EACF,CAAC;EACD,IAAI,aACF,KAAK,sBAAsB,IAAI,UAAU;EAE3C,MAAM,QAAuB;GAC3B;GACA,YAAY;GACZ,UAAU;GACV,WAAW;GACX,cAAc;EAChB;EACA,KAAK,SAAS,IAAI,YAAY,KAAK;EACnC,OAAO;CACT;;;;;;;;CASA,2BAAyC;EACvC,KAAK,MAAM,CAAC,YAAY,UAAU,KAAK,UAAU;GAC/C,IAAI,CAAC,MAAM,YAAY;GACvB,KAAK,SAAS,IAAI,YAAY;IAC5B,UAAU,MAAM;IAChB,UAAU,MAAM;IAChB,WAAW,MAAM;GACnB,CAAC;EACH;CACF;CAEA,iBACE,OACA,SAMM;EACN,IAAI,CAAC,MAAM,YAAY;EACvB,MAAM,YAAY,QAAQ,WAAW,KAAA;EAErC,IAAI,QAAQ,aAAa,MAAM,UAAU;GACvC,IAAI,sBAAsB;GAE1B,IAAI,MAAM,cACR,IAAI,6BAA6B,MAAM,UAAU,QAAQ,QAAQ,GAAG;IAElE,MAAM,WAAW,QAAQ;IACzB,sBAAsB;GACxB,OAAO;IAOL,IAAI,QAAQ,IAAI,aAAa,cAC3B,QAAQ,KACN,qHACA;KACE,UAAU,MAAM;KAChB,MAAM,QAAQ;KACd,YAAY,QAAQ;IACtB,CACF;IAEF,sBAAsB;GACxB;QACK,IAAI,CAAC,QAAQ,SAAS,WAAW,MAAM,QAAQ,GACpD,IACE,mBAAmB,MAAM,QAAQ,KACjC,mBAAmB,QAAQ,QAAQ,KACnC,6BAA6B,MAAM,UAAU,QAAQ,QAAQ,GAC7D;IACA,MAAM,cAAc,KAAK,uBAAuB;KAC9C,UAAU,QAAQ;KAClB,UAAU,QAAQ;KAClB;IACF,CAAC;IACD,IAAI,aAAa,MAAM,WAAW,SAAS,MAAM;IACjD,MAAM,WAAW,QAAQ;IACzB,MAAM,eAAe;IACrB,sBAAsB;GACxB,OAAO;IAQL,IAAI,QAAQ,IAAI,aAAa,cAC3B,QAAQ,KACN,iGACA;KACE,UAAU,MAAM;KAChB,MAAM,QAAQ;KACd,YAAY,QAAQ;IACtB,CACF;IAEF,sBAAsB;GACxB;GAGF,IAAI,uBAAuB,MAAM,YAAY;IAC3C,MAAM,QAAQ,QAAQ,SAAS,MAAM,MAAM,SAAS,MAAM;IAC1D,MAAM,WAAW,SAAS,OAAO,KAAK;IACtC,MAAM,cAAc,KAAK,uBAAuB;KAC9C,UAAU,QAAQ;KAClB,UAAU,QAAQ;KAClB;IACF,CAAC;IACD,IAAI,aAAa,MAAM,WAAW,SAAS,MAAM;IACjD,MAAM,WAAW,QAAQ;IACzB,MAAM,eAAe;GACvB;EACF;EAEA,IAAI,CAAC,MAAM,gBAAgB,MAAM;OACX,KAAK,uBAAuB;IAC9C,UAAU,QAAQ;IAClB,UAAU,QAAQ;IAClB;GACF,CACc,GAAG;IACf,MAAM,WAAW,SAAS,MAAM;IAChC,MAAM,WAAW,QAAQ;IACzB,MAAM,eAAe;GACvB;;CAEJ;CAEA,iBAAyB,UAA0C;EACjE,MAAM,YAAY,KAAK;EAEvB,KAAK,MAAM,WAAW,UAAU;GAC9B,IAAI,CAAC,WAAW,CAAC,MAAM,QAAS,QAA0B,OAAO,GAC/D;GAEF,KAAK,MAAM,WAAW,QAAQ,SAAwD;IACpF,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa;IAE9C,MAAM,WAAW,KAAK,SAAS,IAAI,QAAQ,UAAU;IAErD,IAAI,WAAW;KAIb,IAAI,CAAC,UAAU,YACb,KAAK,SAAS,IAAI,QAAQ,YAAY;MACpC,UAAU,QAAQ;MAClB,UAAU,QAAQ;MAClB,WAAW,QAAQ,WAAW,KAAA;KAChC,CAAC;KAEH,IAAI,QAAQ,UAAU,KAAK,iBAAiB,QAAQ,QAAQ;KAC5D;IACF;IAGA,IAAI,QAAQ;IAEZ,IAAI,SAAS,CAAC,MAAM,YAAY;KAM9B,IAAI,EAFF,QAAQ,aAAa,MAAM,YAC1B,QAAQ,WAAW,KAAA,MAAe,MAAM,YACpB;MACrB,IAAI,QAAQ,UAAU,KAAK,iBAAiB,QAAQ,QAAQ;MAC5D;KACF;KACA,KAAK,SAAS,OAAO,QAAQ,UAAU;KACvC,QAAQ,KAAA;IACV;IAEA,IAAI,CAAC,OACH,QAAQ,KAAK,kBACX,QAAQ,YACR,QAAQ,UACR,QAAQ,WAAW,KAAA,CACrB;IAGF,KAAK,iBAAiB,OAAO,OAAO;IAEpC,IAAI,QAAQ,WAAW,KAAA,KAAa,CAAC,MAAM,WAAW;KAIpD,MAAM,EAAE,YAAY,qBAAqB;KACzC,IAAI,CAAC,kBAAkB;KACvB,MAAM,YAAY;KAClB,MAAM,eAAe;KACrB,iBAAiB,YACf,IAAI,aAAa;MACf,QAAQ,QAAQ;MAChB,UAAU,QAAQ;MAClB,SAAS,QAAQ;MACjB,GAAI,QAAQ,iBAAiB,KAAA,IACzB,EAAE,cAAc,QAAQ,aAAa,IACrC,CAAC;KACP,CAAC,CACH;KACA,iBAAiB,MAAM;IACzB;IAEA,IAAI,QAAQ,UAAU,KAAK,iBAAiB,QAAQ,QAAQ;GAC9D;EACF;CACF;AACF"}
+{"version":3,"file":"ToolInvocationTracker.js","names":[],"sources":["../../../src/runtimes/tool-invocations/ToolInvocationTracker.ts"],"sourcesContent":["declare const process: { env: { NODE_ENV?: string } };\n\nimport {\n  createAssistantStreamController,\n  type ToolCallStreamController,\n  ToolResponse,\n  unstable_toolResultStream,\n  type Tool,\n  type ToolModelContentPart,\n} from \"assistant-stream\";\nimport {\n  AssistantMetaTransformStream,\n  type ReadonlyJSONValue,\n} from \"assistant-stream/utils\";\nimport { isJSONValueEqual } from \"../../utils/json/is-json-equal\";\nimport type { ThreadMessage } from \"../../types/message\";\n\n/**\n * Streaming execution state for a frontend tool.\n */\nexport type ToolExecutionStatus =\n  | { type: \"executing\" }\n  | {\n      type: \"interrupt\";\n      payload: { type: \"human\"; payload: unknown };\n    };\n\nexport type AddToolResultCommand = {\n  readonly type: \"add-tool-result\";\n  readonly toolCallId: string;\n  readonly toolName: string;\n  readonly result: ReadonlyJSONValue;\n  readonly isError: boolean;\n  readonly artifact?: ReadonlyJSONValue;\n  readonly modelContent?: readonly ToolModelContentPart[];\n};\n\nexport type ToolInvocationTrackerSnapshot = {\n  readonly messages: readonly ThreadMessage[];\n  /** Whether the producing runtime is currently streaming new output. */\n  readonly isRunning: boolean;\n  /**\n   * Whether the producing runtime is still loading historical state.\n   * When `true`, every snapshot is treated as historical (no `streamCall` /\n   * `execute` fires). When `false`, processing resumes as live.\n   */\n  readonly isLoading?: boolean;\n};\n\nexport type ToolInvocationTrackerCallbacks = {\n  /**\n   * Invoked when a client-side `execute()` returns a result and the runtime\n   * needs to feed it back into the conversation.\n   */\n  onResult: (command: AddToolResultCommand) => void;\n  /**\n   * Invoked whenever the per-tool-call status map changes (executing /\n   * interrupt / cleared). The callback receives a fresh map; mutating the\n   * argument is not supported.\n   */\n  onStatusesChange: (\n    statuses: ReadonlyMap<string, ToolExecutionStatus>,\n  ) => void;\n};\n\ntype ToolCallEntry = {\n  toolName: string;\n  argsText: string;\n  hasResult: boolean;\n} & (\n  | {\n      /** Restored phase — observed during a history-load snapshot. */\n      controller?: undefined;\n      argsComplete?: undefined;\n    }\n  | {\n      /** Active phase — chunks are flowing through `controller`. */\n      controller: ToolCallStreamController;\n      argsComplete: boolean;\n    }\n);\n\nconst isArgsTextComplete = (argsText: string) => {\n  try {\n    JSON.parse(argsText);\n    return true;\n  } catch {\n    return false;\n  }\n};\n\nconst parseArgsText = (argsText: string) => {\n  try {\n    return JSON.parse(argsText);\n  } catch {\n    return undefined;\n  }\n};\n\nconst isEquivalentCompleteArgsText = (previous: string, next: string) => {\n  const previousValue = parseArgsText(previous);\n  const nextValue = parseArgsText(next);\n  if (previousValue === undefined || nextValue === undefined) return false;\n  return isJSONValueEqual(previousValue, nextValue);\n};\n\n/**\n * Plain-class port of the former `useToolInvocations` React hook. Owns the\n * assistant-stream pipeline that drives client-side `streamCall` / `execute`\n * for tool-call parts surfaced by a thread runtime, plus the per-tool-call\n * status map that consumers render against.\n *\n * **Contract**: `streamCall` (and `execute`) fires exactly once per logical\n * `toolCallId`. Args mutations after first completion, result replacement,\n * and result clearing are *not* surfaced through additional `streamCall`\n * invocations — by design — so hosts cannot observe spurious re-fires of\n * side effects. The follow-up `reader.events()` API will expose those\n * post-completion transitions to consumers that opt in.\n *\n * State-transition safety: every public method that observes runtime state\n * (`setState`, `reset`, `abort`, `resume`) wraps its work in try/catch and\n * logs to `console.error` rather than throwing. The tracker is built into\n * the hot message-processing path, so a malformed snapshot must never crash\n * the host runtime. See ./EDGE_CASES.md for the known non-trivial state\n * transitions and what each does today.\n */\nexport class ToolInvocationTracker {\n  private readonly _getTools: () => Record<string, Tool> | undefined;\n  private readonly _callbacks: ToolInvocationTrackerCallbacks;\n\n  private readonly _entries = new Map<string, ToolCallEntry>();\n  /**\n   * Tool call ids whose `execute` should be short-circuited in the wrapper.\n   * Populated when an entry is created with a result already attached\n   * (history reload, mid-run resume, etc.) — `execute` is suppressed so\n   * client-side side effects don't double-run. Membership outlives the\n   * entry: `reset()` deliberately does *not* clear this so post-abort\n   * cancellation `result` chunks for pre-resolved entries can still be\n   * recognized and dropped. Growth is bounded by the number of pre-resolved\n   * tool calls observed in the session.\n   */\n  private readonly _skipExecuteStreamIds = new Set<string>();\n  private readonly _humanInput = new Map<\n    string,\n    {\n      resolve: (payload: unknown) => void;\n      reject: (reason: unknown) => void;\n    }\n  >();\n  /** In-flight `execute` invocations keyed by tool call id. */\n  private readonly _executing = new Set<string>();\n  private readonly _settledResolvers: Array<() => void> = [];\n\n  private _statuses = new Map<string, ToolExecutionStatus>();\n\n  private _ac: AbortController = new AbortController();\n  private _pendingRestore = true;\n\n  /** Cached last snapshot, used to skip processing on identical re-renders. */\n  private _lastSnapshot: ToolInvocationTrackerSnapshot | null = null;\n  private _isRunning = false;\n\n  private _controller!: ReturnType<typeof createAssistantStreamController>[1];\n\n  /**\n   * Set when the assistant-stream pipeline has died (errored out via\n   * `.pipeTo(...).catch(...)`). The next `setState` re-initializes the\n   * pipeline and demotes all active entries to restored so they survive\n   * across the restart without re-firing `streamCall` (preserves the\n   * \"exactly once\" contract). Capped at a single auto-restart per session\n   * — repeated failures keep the tracker dead with a more visible error.\n   */\n  private _pipelineDead = false;\n  private _pipelineRestartUsed = false;\n\n  constructor(\n    getTools: () => Record<string, Tool> | undefined,\n    callbacks: ToolInvocationTrackerCallbacks,\n  ) {\n    this._getTools = getTools;\n    this._callbacks = callbacks;\n\n    this._initPipeline();\n  }\n\n  /**\n   * Build the assistant-stream pipeline. Called once from the constructor\n   * and at most once again if `_pipelineDead` is set (see F.4 in\n   * EDGE_CASES.md).\n   */\n  private _initPipeline(): void {\n    const [stream, controller] = createAssistantStreamController();\n    this._controller = controller;\n\n    const transform = unstable_toolResultStream(\n      () => this._getWrappedTools(),\n      () => this._ac.signal,\n      (toolCallId, payload) => this._onHumanInput(toolCallId, payload),\n      {\n        onExecutionStart: (id) => this._onExecutionStart(id),\n        onExecutionEnd: (id) => this._onExecutionEnd(id),\n      },\n    );\n\n    stream\n      .pipeThrough(transform)\n      .pipeThrough(new AssistantMetaTransformStream())\n      .pipeTo(\n        new WritableStream({\n          write: (chunk) => {\n            try {\n              if (chunk.type !== \"result\") return;\n              this._handleResultChunk(chunk);\n            } catch (err) {\n              console.error(\n                \"[ToolInvocationTracker] result chunk handling failed\",\n                err,\n              );\n            }\n          },\n        }),\n      )\n      .catch((err) => {\n        console.error(\n          \"[ToolInvocationTracker] stream pipeline failed; will attempt single restart on next setState\",\n          err,\n        );\n        this._pipelineDead = true;\n      });\n  }\n\n  // ───────────────────────── public API ─────────────────────────\n\n  /**\n   * Feed the next observed snapshot into the tracker. Called from the host\n   * runtime whenever its message list / running state changes.\n   */\n  public setState(snapshot: ToolInvocationTrackerSnapshot): void {\n    try {\n      // Recover from a dead pipeline before processing anything. We demote\n      // all active entries to \"restored\" so the rebuilt pipeline does not\n      // re-fire `streamCall` for tool calls that already fired pre-death;\n      // preserves the \"exactly once per toolCallId\" contract.\n      if (this._pipelineDead) {\n        if (this._pipelineRestartUsed) {\n          // Already retried once and failed again. Stay dead.\n          return;\n        }\n        this._pipelineRestartUsed = true;\n        this._pipelineDead = false;\n        this._demoteEntriesToRestored();\n        this._executing.clear();\n        this._ac = new AbortController();\n        this._initPipeline();\n        // Fall through and process the snapshot against the fresh pipeline.\n      }\n\n      // Identical snapshot — skip processing entirely. Note: external-store\n      // runtimes rebuild the messages array on every adapter update, so this\n      // fast-path rarely triggers there; it's primarily for the React-hook\n      // shim where state references are stable.\n      if (\n        this._lastSnapshot &&\n        this._lastSnapshot.messages === snapshot.messages &&\n        this._lastSnapshot.isRunning === snapshot.isRunning &&\n        this._lastSnapshot.isLoading === snapshot.isLoading\n      ) {\n        return;\n      }\n\n      // While the host is still loading initial state, treat every snapshot\n      // as historical: tool calls are recorded so the next live snapshot can\n      // diff against them, but `streamCall` / `execute` do not fire.\n      const restoreFromLoading = snapshot.isLoading === true;\n      if (restoreFromLoading) {\n        this._pendingRestore = true;\n      }\n\n      // E.4 / AF3 — only mark `_lastSnapshot`/`_isRunning` as observed after\n      // processing succeeds. If `_processMessages` throws, the next snapshot\n      // (even if identical) gets re-processed against the recovered state.\n      const previousIsRunning = this._isRunning;\n      this._isRunning = snapshot.isRunning;\n      try {\n        this._processMessages(snapshot.messages);\n      } catch (err) {\n        this._isRunning = previousIsRunning;\n        throw err;\n      }\n      this._lastSnapshot = snapshot;\n      this._pendingRestore = false;\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] setState failed; snapshot dropped\",\n        err,\n      );\n    }\n  }\n\n  /**\n   * Reset the tracker so the next observed snapshot is treated as historical.\n   * Clears entries and aborts any in-flight executions. Used by callers like\n   * `importExternalState` to mark a freshly loaded state as restored.\n   */\n  public reset(): void {\n    try {\n      this._pendingRestore = true;\n      this._entries.clear();\n      this._lastSnapshot = null;\n      // `_skipExecuteStreamIds` is intentionally not cleared — see field doc.\n      void this.abort().finally(() => {\n        this._executing.clear();\n      });\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] reset failed\", err);\n    }\n  }\n\n  /**\n   * Abort any in-flight `execute()` invocations. Resolves once all of them\n   * have settled (or immediately if none are running).\n   */\n  public abort(): Promise<void> {\n    try {\n      this._humanInput.forEach(({ reject }) => {\n        try {\n          reject(new Error(\"Tool execution aborted\"));\n        } catch {\n          // host rejection handler threw — already in the abort path,\n          // swallow so we continue cleaning up.\n        }\n      });\n      this._humanInput.clear();\n\n      this._ac.abort();\n      this._ac = new AbortController();\n\n      if (this._executing.size === 0) {\n        return Promise.resolve();\n      }\n      return new Promise<void>((resolve) => {\n        this._settledResolvers.push(resolve);\n      });\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] abort failed\", err);\n      return Promise.resolve();\n    }\n  }\n\n  /**\n   * Resolve a pending human-input request for the given tool call. Returns\n   * `true` if a pending request was resumed, `false` if the tracker has no\n   * outstanding request for that id (the caller should fall back to its own\n   * dispatch path).\n   */\n  public resume(toolCallId: string, payload: unknown): boolean {\n    try {\n      const handlers = this._humanInput.get(toolCallId);\n      if (!handlers) return false;\n      this._humanInput.delete(toolCallId);\n      this._setStatus(toolCallId, { type: \"executing\" });\n      handlers.resolve(payload);\n      return true;\n    } catch (err) {\n      console.error(\"[ToolInvocationTracker] resume failed\", err);\n      return false;\n    }\n  }\n\n  /**\n   * Returns the current tool execution status map. The returned `Map` is\n   * the tracker's internal store — do not mutate it. Treat the reference\n   * as a snapshot that may be replaced wholesale on the next status\n   * transition.\n   */\n  public getStatuses(): ReadonlyMap<string, ToolExecutionStatus> {\n    return this._statuses;\n  }\n\n  // ───────────────────── internal: tool wrapping ─────────────────────\n\n  private _getWrappedTools(): Record<string, Tool> | undefined {\n    const tools = this._getTools();\n    if (!tools) return undefined;\n\n    return Object.fromEntries(\n      Object.entries(tools).map(([name, tool]) => {\n        const execute = tool.execute;\n        if (execute === undefined) return [name, tool];\n\n        const wrappedTool = {\n          ...tool,\n          execute: (\n            ...[args, context]: Parameters<NonNullable<typeof execute>>\n          ) => {\n            if (this._skipExecuteStreamIds.has(context.toolCallId)) {\n              // Pre-resolved tool call: never invoke the host's execute.\n              // Returning a never-settling Promise keeps the executor's\n              // pending entry alive but enqueues nothing.\n              return new Promise(() => {}) as never;\n            }\n            return execute(args, context);\n          },\n        } as Tool;\n        return [name, wrappedTool];\n      }),\n    ) as Record<string, Tool>;\n  }\n\n  // ──────────────── internal: execution lifecycle callbacks ────────────────\n\n  private _onHumanInput(\n    toolCallId: string,\n    payload: unknown,\n  ): Promise<unknown> {\n    return new Promise<unknown>((resolve, reject) => {\n      const previous = this._humanInput.get(toolCallId);\n      if (previous) {\n        try {\n          previous.reject(\n            new Error(\"Human input request was superseded by a new request\"),\n          );\n        } catch {\n          // host rejection handler threw; ignore and proceed\n        }\n      }\n      this._humanInput.set(toolCallId, { resolve, reject });\n      this._setStatus(toolCallId, {\n        type: \"interrupt\",\n        payload: { type: \"human\", payload },\n      });\n    });\n  }\n\n  private _onExecutionStart(toolCallId: string): void {\n    if (this._skipExecuteStreamIds.has(toolCallId)) return;\n\n    this._executing.add(toolCallId);\n    this._setStatus(toolCallId, { type: \"executing\" });\n  }\n\n  private _onExecutionEnd(toolCallId: string): void {\n    if (!this._executing.delete(toolCallId)) return;\n\n    this._deleteStatus(toolCallId);\n\n    if (this._executing.size === 0) {\n      const resolvers = this._settledResolvers.splice(0);\n      resolvers.forEach((resolve) => {\n        try {\n          resolve();\n        } catch {\n          // ignore — settled-resolver consumer threw\n        }\n      });\n    }\n  }\n\n  private _handleResultChunk(chunk: {\n    type: \"result\";\n    result: ReadonlyJSONValue;\n    isError: boolean;\n    artifact?: ReadonlyJSONValue;\n    modelContent?: readonly ToolModelContentPart[];\n    meta: { toolCallId: string; toolName: string };\n  }): void {\n    const toolCallId = chunk.meta.toolCallId;\n    const entry = this._entries.get(toolCallId);\n\n    // Pre-resolved tool call whose entry has been cleared by `reset()`.\n    // The post-abort cancellation chunk lands here after the entry is\n    // gone; suppress via the long-lived skip-execute marker.\n    if (!entry && this._skipExecuteStreamIds.has(toolCallId)) {\n      return;\n    }\n\n    // The host already set the result (via the live snapshot's\n    // `setResponse` path). Suppress the executor's redundant emit.\n    if (entry?.hasResult) return;\n\n    this._invokeOnResult({\n      type: \"add-tool-result\",\n      toolCallId,\n      toolName: chunk.meta.toolName,\n      result: chunk.result,\n      isError: chunk.isError,\n      ...(chunk.artifact !== undefined && { artifact: chunk.artifact }),\n      ...(chunk.modelContent !== undefined && {\n        modelContent: chunk.modelContent,\n      }),\n    });\n  }\n\n  // ──────────────── internal: callback invocation (AF1/AF2) ────────────────\n\n  private _invokeOnResult(command: AddToolResultCommand): void {\n    try {\n      this._callbacks.onResult(command);\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] onResult callback threw; result dropped\",\n        err,\n      );\n    }\n  }\n\n  private _invokeOnStatusesChange(): void {\n    try {\n      this._callbacks.onStatusesChange(this._statuses);\n    } catch (err) {\n      console.error(\n        \"[ToolInvocationTracker] onStatusesChange callback threw; status change not propagated\",\n        err,\n      );\n    }\n  }\n\n  // ──────────────── internal: status map mutations ────────────────\n\n  private _setStatus(toolCallId: string, status: ToolExecutionStatus): void {\n    const next = new Map(this._statuses);\n    next.set(toolCallId, status);\n    this._statuses = next;\n    this._invokeOnStatusesChange();\n  }\n\n  private _deleteStatus(toolCallId: string): void {\n    if (!this._statuses.has(toolCallId)) return;\n    const next = new Map(this._statuses);\n    next.delete(toolCallId);\n    this._statuses = next;\n    this._invokeOnStatusesChange();\n  }\n\n  // ──────────────── internal: snapshot processing ────────────────\n\n  private _hasExecutableTool(toolName: string): boolean {\n    const tool = this._getTools()?.[toolName];\n    return tool?.execute !== undefined || tool?.streamCall !== undefined;\n  }\n\n  private _shouldCloseArgsStream({\n    toolName,\n    argsText,\n    hasResult,\n  }: {\n    toolName: string;\n    argsText: string;\n    hasResult: boolean;\n  }): boolean {\n    if (hasResult) return true;\n    if (!this._hasExecutableTool(toolName)) {\n      return !this._isRunning && isArgsTextComplete(argsText);\n    }\n    return isArgsTextComplete(argsText);\n  }\n\n  private _startActiveEntry(\n    toolCallId: string,\n    toolName: string,\n    skipExecute: boolean,\n  ): ToolCallEntry {\n    const toolCallController = this._controller.addToolCallPart({\n      toolName,\n      toolCallId,\n    });\n    if (skipExecute) {\n      this._skipExecuteStreamIds.add(toolCallId);\n    }\n    const entry: ToolCallEntry = {\n      toolName,\n      controller: toolCallController,\n      argsText: \"\",\n      hasResult: false,\n      argsComplete: false,\n    };\n    this._entries.set(toolCallId, entry);\n    return entry;\n  }\n\n  /**\n   * Demote every active entry back to the restored phase. Used by the\n   * pipeline-restart path so that, after a fresh pipeline is built, the\n   * next observed snapshot does not re-fire `streamCall` for tool calls\n   * that already fired pre-death. Args / hasResult tracking is preserved\n   * so signature comparisons still work.\n   */\n  private _demoteEntriesToRestored(): void {\n    for (const [toolCallId, entry] of this._entries) {\n      if (!entry.controller) continue;\n      this._entries.set(toolCallId, {\n        toolName: entry.toolName,\n        argsText: entry.argsText,\n        hasResult: entry.hasResult,\n      });\n    }\n  }\n\n  private _processArgsText(\n    entry: ToolCallEntry,\n    content: {\n      toolCallId: string;\n      toolName: string;\n      argsText: string;\n      result?: unknown;\n    },\n  ): void {\n    if (!entry.controller) return;\n    const hasResult = content.result !== undefined;\n\n    if (content.argsText !== entry.argsText) {\n      let shouldWriteArgsText = true;\n\n      if (entry.argsComplete) {\n        if (isEquivalentCompleteArgsText(entry.argsText, content.argsText)) {\n          // A.3 — key reorder. Track new text, no re-fire needed.\n          entry.argsText = content.argsText;\n          shouldWriteArgsText = false;\n        } else {\n          // A.4 — args changed after first completion. Under the\n          // \"exactly once per toolCallId\" contract we do not restart the\n          // stream. The host's existing `streamCall` keeps its original\n          // args view; the snapshot's new text is recorded for diffing\n          // but not surfaced. Events API in a follow-up will expose this\n          // to consumers that opt in.\n          if (process.env.NODE_ENV !== \"production\") {\n            console.warn(\n              \"[ToolInvocationTracker] argsText changed after first completion; not re-firing streamCall (see EDGE_CASES.md A.4)\",\n              {\n                previous: entry.argsText,\n                next: content.argsText,\n                toolCallId: content.toolCallId,\n              },\n            );\n          }\n          shouldWriteArgsText = false;\n        }\n      } else if (!content.argsText.startsWith(entry.argsText)) {\n        if (\n          isArgsTextComplete(entry.argsText) &&\n          isArgsTextComplete(content.argsText) &&\n          isEquivalentCompleteArgsText(entry.argsText, content.argsText)\n        ) {\n          const shouldClose = this._shouldCloseArgsStream({\n            toolName: content.toolName,\n            argsText: content.argsText,\n            hasResult,\n          });\n          if (shouldClose) entry.controller.argsText.close();\n          entry.argsText = content.argsText;\n          entry.argsComplete = shouldClose;\n          shouldWriteArgsText = false;\n        } else {\n          // A.2 — args regressed mid-stream. Under the \"exactly once\"\n          // contract we do not restart. The controller keeps whatever\n          // prefix we already streamed; subsequent prefix-respecting\n          // updates can still flow against it. Snapshots that never\n          // re-converge to a prefix will leave the controller's args\n          // view stale relative to the snapshot. Events API in a\n          // follow-up will expose this to consumers that opt in.\n          if (process.env.NODE_ENV !== \"production\") {\n            console.warn(\n              \"[ToolInvocationTracker] argsText regressed mid-stream; not restarting (see EDGE_CASES.md A.2)\",\n              {\n                previous: entry.argsText,\n                next: content.argsText,\n                toolCallId: content.toolCallId,\n              },\n            );\n          }\n          shouldWriteArgsText = false;\n        }\n      }\n\n      if (shouldWriteArgsText && entry.controller) {\n        const delta = content.argsText.slice(entry.argsText.length);\n        entry.controller.argsText.append(delta);\n        const shouldClose = this._shouldCloseArgsStream({\n          toolName: content.toolName,\n          argsText: content.argsText,\n          hasResult,\n        });\n        if (shouldClose) entry.controller.argsText.close();\n        entry.argsText = content.argsText;\n        entry.argsComplete = shouldClose;\n      }\n    }\n\n    if (!entry.argsComplete && entry.controller) {\n      const shouldClose = this._shouldCloseArgsStream({\n        toolName: content.toolName,\n        argsText: content.argsText,\n        hasResult,\n      });\n      if (shouldClose) {\n        entry.controller.argsText.close();\n        entry.argsText = content.argsText;\n        entry.argsComplete = true;\n      }\n    }\n  }\n\n  private _processMessages(messages: readonly ThreadMessage[]): void {\n    const isRestore = this._pendingRestore;\n\n    for (const message of messages) {\n      if (!message || !Array.isArray((message as ThreadMessage).content)) {\n        continue;\n      }\n      for (const content of message.content as readonly ThreadMessage[\"content\"][number][]) {\n        if (!content || content.type !== \"tool-call\") continue;\n\n        const existing = this._entries.get(content.toolCallId);\n\n        if (isRestore) {\n          // Don't overwrite an already-active entry (e.g. live tool-call\n          // observed before this restore snapshot landed). Restore can\n          // only seed entries the runtime has never seen.\n          if (!existing?.controller) {\n            this._entries.set(content.toolCallId, {\n              toolName: content.toolName,\n              argsText: content.argsText,\n              hasResult: content.result !== undefined,\n            });\n          }\n          if (content.messages) this._processMessages(content.messages);\n          continue;\n        }\n\n        // Live snapshot.\n        let entry = existing;\n\n        if (entry && !entry.controller) {\n          // Restored entry observed in a live snapshot. Promote if its\n          // signature has changed; otherwise treat as still-historical.\n          const signatureChanged =\n            content.argsText !== entry.argsText ||\n            (content.result !== undefined) !== entry.hasResult;\n          if (!signatureChanged) {\n            if (content.messages) this._processMessages(content.messages);\n            continue;\n          }\n          this._entries.delete(content.toolCallId);\n          entry = undefined;\n        }\n\n        if (!entry) {\n          entry = this._startActiveEntry(\n            content.toolCallId,\n            content.toolName,\n            content.result !== undefined,\n          );\n        }\n\n        this._processArgsText(entry, content);\n\n        if (content.result !== undefined && !entry.hasResult) {\n          // `entry` is in active phase from this point — either just\n          // created by `_startActiveEntry`, or pre-existing with a live\n          // controller. Narrow once instead of asserting at every use.\n          const { controller: activeController } = entry;\n          if (!activeController) continue;\n          entry.hasResult = true;\n          entry.argsComplete = true;\n          activeController.setResponse(\n            new ToolResponse({\n              result: content.result as ReadonlyJSONValue,\n              artifact: content.artifact as ReadonlyJSONValue | undefined,\n              isError: content.isError,\n              ...(content.modelContent !== undefined\n                ? { modelContent: content.modelContent }\n                : {}),\n            }),\n          );\n          activeController.close();\n        }\n\n        if (content.messages) this._processMessages(content.messages);\n      }\n    }\n  }\n}\n"],"mappings":";;;;AAkFA,MAAM,sBAAsB,aAAqB;CAC/C,IAAI;EACF,KAAK,MAAM,QAAQ;EACnB,OAAO;CACT,QAAQ;EACN,OAAO;CACT;AACF;AAEA,MAAM,iBAAiB,aAAqB;CAC1C,IAAI;EACF,OAAO,KAAK,MAAM,QAAQ;CAC5B,QAAQ;EACN;CACF;AACF;AAEA,MAAM,gCAAgC,UAAkB,SAAiB;CACvE,MAAM,gBAAgB,cAAc,QAAQ;CAC5C,MAAM,YAAY,cAAc,IAAI;CACpC,IAAI,kBAAkB,KAAA,KAAa,cAAc,KAAA,GAAW,OAAO;CACnE,OAAO,iBAAiB,eAAe,SAAS;AAClD;;;;;;;;;;;;;;;;;;;;;AAsBA,IAAa,wBAAb,MAAmC;CACjC;CACA;CAEA,2BAA4B,IAAI,IAA2B;;;;;;;;;;;CAW3D,wCAAyC,IAAI,IAAY;CACzD,8BAA+B,IAAI,IAMjC;;CAEF,6BAA8B,IAAI,IAAY;CAC9C,oBAAwD,CAAC;CAEzD,4BAAoB,IAAI,IAAiC;CAEzD,MAA+B,IAAI,gBAAgB;CACnD,kBAA0B;;CAG1B,gBAA8D;CAC9D,aAAqB;CAErB;;;;;;;;;CAUA,gBAAwB;CACxB,uBAA+B;CAE/B,YACE,UACA,WACA;EACA,KAAK,YAAY;EACjB,KAAK,aAAa;EAElB,KAAK,cAAc;CACrB;;;;;;CAOA,gBAA8B;EAC5B,MAAM,CAAC,QAAQ,cAAc,gCAAgC;EAC7D,KAAK,cAAc;EAEnB,MAAM,YAAY,gCACV,KAAK,iBAAiB,SACtB,KAAK,IAAI,SACd,YAAY,YAAY,KAAK,cAAc,YAAY,OAAO,GAC/D;GACE,mBAAmB,OAAO,KAAK,kBAAkB,EAAE;GACnD,iBAAiB,OAAO,KAAK,gBAAgB,EAAE;EACjD,CACF;EAEA,OACG,YAAY,SAAS,CAAC,CACtB,YAAY,IAAI,6BAA6B,CAAC,CAAC,CAC/C,OACC,IAAI,eAAe,EACjB,QAAQ,UAAU;GAChB,IAAI;IACF,IAAI,MAAM,SAAS,UAAU;IAC7B,KAAK,mBAAmB,KAAK;GAC/B,SAAS,KAAK;IACZ,QAAQ,MACN,wDACA,GACF;GACF;EACF,EACF,CAAC,CACH,CAAC,CACA,OAAO,QAAQ;GACd,QAAQ,MACN,gGACA,GACF;GACA,KAAK,gBAAgB;EACvB,CAAC;CACL;;;;;CAQA,SAAgB,UAA+C;EAC7D,IAAI;GAKF,IAAI,KAAK,eAAe;IACtB,IAAI,KAAK,sBAEP;IAEF,KAAK,uBAAuB;IAC5B,KAAK,gBAAgB;IACrB,KAAK,yBAAyB;IAC9B,KAAK,WAAW,MAAM;IACtB,KAAK,MAAM,IAAI,gBAAgB;IAC/B,KAAK,cAAc;GAErB;GAMA,IACE,KAAK,iBACL,KAAK,cAAc,aAAa,SAAS,YACzC,KAAK,cAAc,cAAc,SAAS,aAC1C,KAAK,cAAc,cAAc,SAAS,WAE1C;GAOF,IAD2B,SAAS,cAAc,MAEhD,KAAK,kBAAkB;GAMzB,MAAM,oBAAoB,KAAK;GAC/B,KAAK,aAAa,SAAS;GAC3B,IAAI;IACF,KAAK,iBAAiB,SAAS,QAAQ;GACzC,SAAS,KAAK;IACZ,KAAK,aAAa;IAClB,MAAM;GACR;GACA,KAAK,gBAAgB;GACrB,KAAK,kBAAkB;EACzB,SAAS,KAAK;GACZ,QAAQ,MACN,6DACA,GACF;EACF;CACF;;;;;;CAOA,QAAqB;EACnB,IAAI;GACF,KAAK,kBAAkB;GACvB,KAAK,SAAS,MAAM;GACpB,KAAK,gBAAgB;GAErB,KAAU,MAAM,CAAC,CAAC,cAAc;IAC9B,KAAK,WAAW,MAAM;GACxB,CAAC;EACH,SAAS,KAAK;GACZ,QAAQ,MAAM,wCAAwC,GAAG;EAC3D;CACF;;;;;CAMA,QAA8B;EAC5B,IAAI;GACF,KAAK,YAAY,SAAS,EAAE,aAAa;IACvC,IAAI;KACF,uBAAO,IAAI,MAAM,wBAAwB,CAAC;IAC5C,QAAQ,CAGR;GACF,CAAC;GACD,KAAK,YAAY,MAAM;GAEvB,KAAK,IAAI,MAAM;GACf,KAAK,MAAM,IAAI,gBAAgB;GAE/B,IAAI,KAAK,WAAW,SAAS,GAC3B,OAAO,QAAQ,QAAQ;GAEzB,OAAO,IAAI,SAAe,YAAY;IACpC,KAAK,kBAAkB,KAAK,OAAO;GACrC,CAAC;EACH,SAAS,KAAK;GACZ,QAAQ,MAAM,wCAAwC,GAAG;GACzD,OAAO,QAAQ,QAAQ;EACzB;CACF;;;;;;;CAQA,OAAc,YAAoB,SAA2B;EAC3D,IAAI;GACF,MAAM,WAAW,KAAK,YAAY,IAAI,UAAU;GAChD,IAAI,CAAC,UAAU,OAAO;GACtB,KAAK,YAAY,OAAO,UAAU;GAClC,KAAK,WAAW,YAAY,EAAE,MAAM,YAAY,CAAC;GACjD,SAAS,QAAQ,OAAO;GACxB,OAAO;EACT,SAAS,KAAK;GACZ,QAAQ,MAAM,yCAAyC,GAAG;GAC1D,OAAO;EACT;CACF;;;;;;;CAQA,cAA+D;EAC7D,OAAO,KAAK;CACd;CAIA,mBAA6D;EAC3D,MAAM,QAAQ,KAAK,UAAU;EAC7B,IAAI,CAAC,OAAO,OAAO,KAAA;EAEnB,OAAO,OAAO,YACZ,OAAO,QAAQ,KAAK,CAAC,CAAC,KAAK,CAAC,MAAM,UAAU;GAC1C,MAAM,UAAU,KAAK;GACrB,IAAI,YAAY,KAAA,GAAW,OAAO,CAAC,MAAM,IAAI;GAgB7C,OAAO,CAAC,MAAM;IAbZ,GAAG;IACH,UACE,GAAG,CAAC,MAAM,aACP;KACH,IAAI,KAAK,sBAAsB,IAAI,QAAQ,UAAU,GAInD,OAAO,IAAI,cAAc,CAAC,CAAC;KAE7B,OAAO,QAAQ,MAAM,OAAO;IAC9B;GAEsB,CAAC;EAC3B,CAAC,CACH;CACF;CAIA,cACE,YACA,SACkB;EAClB,OAAO,IAAI,SAAkB,SAAS,WAAW;GAC/C,MAAM,WAAW,KAAK,YAAY,IAAI,UAAU;GAChD,IAAI,UACF,IAAI;IACF,SAAS,uBACP,IAAI,MAAM,qDAAqD,CACjE;GACF,QAAQ,CAER;GAEF,KAAK,YAAY,IAAI,YAAY;IAAE;IAAS;GAAO,CAAC;GACpD,KAAK,WAAW,YAAY;IAC1B,MAAM;IACN,SAAS;KAAE,MAAM;KAAS;IAAQ;GACpC,CAAC;EACH,CAAC;CACH;CAEA,kBAA0B,YAA0B;EAClD,IAAI,KAAK,sBAAsB,IAAI,UAAU,GAAG;EAEhD,KAAK,WAAW,IAAI,UAAU;EAC9B,KAAK,WAAW,YAAY,EAAE,MAAM,YAAY,CAAC;CACnD;CAEA,gBAAwB,YAA0B;EAChD,IAAI,CAAC,KAAK,WAAW,OAAO,UAAU,GAAG;EAEzC,KAAK,cAAc,UAAU;EAE7B,IAAI,KAAK,WAAW,SAAS,GAE3B,KADuB,kBAAkB,OAAO,CACxC,CAAC,CAAC,SAAS,YAAY;GAC7B,IAAI;IACF,QAAQ;GACV,QAAQ,CAER;EACF,CAAC;CAEL;CAEA,mBAA2B,OAOlB;EACP,MAAM,aAAa,MAAM,KAAK;EAC9B,MAAM,QAAQ,KAAK,SAAS,IAAI,UAAU;EAK1C,IAAI,CAAC,SAAS,KAAK,sBAAsB,IAAI,UAAU,GACrD;EAKF,IAAI,OAAO,WAAW;EAEtB,KAAK,gBAAgB;GACnB,MAAM;GACN;GACA,UAAU,MAAM,KAAK;GACrB,QAAQ,MAAM;GACd,SAAS,MAAM;GACf,GAAI,MAAM,aAAa,KAAA,KAAa,EAAE,UAAU,MAAM,SAAS;GAC/D,GAAI,MAAM,iBAAiB,KAAA,KAAa,EACtC,cAAc,MAAM,aACtB;EACF,CAAC;CACH;CAIA,gBAAwB,SAAqC;EAC3D,IAAI;GACF,KAAK,WAAW,SAAS,OAAO;EAClC,SAAS,KAAK;GACZ,QAAQ,MACN,mEACA,GACF;EACF;CACF;CAEA,0BAAwC;EACtC,IAAI;GACF,KAAK,WAAW,iBAAiB,KAAK,SAAS;EACjD,SAAS,KAAK;GACZ,QAAQ,MACN,yFACA,GACF;EACF;CACF;CAIA,WAAmB,YAAoB,QAAmC;EACxE,MAAM,OAAO,IAAI,IAAI,KAAK,SAAS;EACnC,KAAK,IAAI,YAAY,MAAM;EAC3B,KAAK,YAAY;EACjB,KAAK,wBAAwB;CAC/B;CAEA,cAAsB,YAA0B;EAC9C,IAAI,CAAC,KAAK,UAAU,IAAI,UAAU,GAAG;EACrC,MAAM,OAAO,IAAI,IAAI,KAAK,SAAS;EACnC,KAAK,OAAO,UAAU;EACtB,KAAK,YAAY;EACjB,KAAK,wBAAwB;CAC/B;CAIA,mBAA2B,UAA2B;EACpD,MAAM,OAAO,KAAK,UAAU,CAAC,GAAG;EAChC,OAAO,MAAM,YAAY,KAAA,KAAa,MAAM,eAAe,KAAA;CAC7D;CAEA,uBAA+B,EAC7B,UACA,UACA,aAKU;EACV,IAAI,WAAW,OAAO;EACtB,IAAI,CAAC,KAAK,mBAAmB,QAAQ,GACnC,OAAO,CAAC,KAAK,cAAc,mBAAmB,QAAQ;EAExD,OAAO,mBAAmB,QAAQ;CACpC;CAEA,kBACE,YACA,UACA,aACe;EACf,MAAM,qBAAqB,KAAK,YAAY,gBAAgB;GAC1D;GACA;EACF,CAAC;EACD,IAAI,aACF,KAAK,sBAAsB,IAAI,UAAU;EAE3C,MAAM,QAAuB;GAC3B;GACA,YAAY;GACZ,UAAU;GACV,WAAW;GACX,cAAc;EAChB;EACA,KAAK,SAAS,IAAI,YAAY,KAAK;EACnC,OAAO;CACT;;;;;;;;CASA,2BAAyC;EACvC,KAAK,MAAM,CAAC,YAAY,UAAU,KAAK,UAAU;GAC/C,IAAI,CAAC,MAAM,YAAY;GACvB,KAAK,SAAS,IAAI,YAAY;IAC5B,UAAU,MAAM;IAChB,UAAU,MAAM;IAChB,WAAW,MAAM;GACnB,CAAC;EACH;CACF;CAEA,iBACE,OACA,SAMM;EACN,IAAI,CAAC,MAAM,YAAY;EACvB,MAAM,YAAY,QAAQ,WAAW,KAAA;EAErC,IAAI,QAAQ,aAAa,MAAM,UAAU;GACvC,IAAI,sBAAsB;GAE1B,IAAI,MAAM,cACR,IAAI,6BAA6B,MAAM,UAAU,QAAQ,QAAQ,GAAG;IAElE,MAAM,WAAW,QAAQ;IACzB,sBAAsB;GACxB,OAAO;IAOL,IAAI,QAAQ,IAAI,aAAa,cAC3B,QAAQ,KACN,qHACA;KACE,UAAU,MAAM;KAChB,MAAM,QAAQ;KACd,YAAY,QAAQ;IACtB,CACF;IAEF,sBAAsB;GACxB;QACK,IAAI,CAAC,QAAQ,SAAS,WAAW,MAAM,QAAQ,GACpD,IACE,mBAAmB,MAAM,QAAQ,KACjC,mBAAmB,QAAQ,QAAQ,KACnC,6BAA6B,MAAM,UAAU,QAAQ,QAAQ,GAC7D;IACA,MAAM,cAAc,KAAK,uBAAuB;KAC9C,UAAU,QAAQ;KAClB,UAAU,QAAQ;KAClB;IACF,CAAC;IACD,IAAI,aAAa,MAAM,WAAW,SAAS,MAAM;IACjD,MAAM,WAAW,QAAQ;IACzB,MAAM,eAAe;IACrB,sBAAsB;GACxB,OAAO;IAQL,IAAI,QAAQ,IAAI,aAAa,cAC3B,QAAQ,KACN,iGACA;KACE,UAAU,MAAM;KAChB,MAAM,QAAQ;KACd,YAAY,QAAQ;IACtB,CACF;IAEF,sBAAsB;GACxB;GAGF,IAAI,uBAAuB,MAAM,YAAY;IAC3C,MAAM,QAAQ,QAAQ,SAAS,MAAM,MAAM,SAAS,MAAM;IAC1D,MAAM,WAAW,SAAS,OAAO,KAAK;IACtC,MAAM,cAAc,KAAK,uBAAuB;KAC9C,UAAU,QAAQ;KAClB,UAAU,QAAQ;KAClB;IACF,CAAC;IACD,IAAI,aAAa,MAAM,WAAW,SAAS,MAAM;IACjD,MAAM,WAAW,QAAQ;IACzB,MAAM,eAAe;GACvB;EACF;EAEA,IAAI,CAAC,MAAM,gBAAgB,MAAM;OACX,KAAK,uBAAuB;IAC9C,UAAU,QAAQ;IAClB,UAAU,QAAQ;IAClB;GACF,CACc,GAAG;IACf,MAAM,WAAW,SAAS,MAAM;IAChC,MAAM,WAAW,QAAQ;IACzB,MAAM,eAAe;GACvB;;CAEJ;CAEA,iBAAyB,UAA0C;EACjE,MAAM,YAAY,KAAK;EAEvB,KAAK,MAAM,WAAW,UAAU;GAC9B,IAAI,CAAC,WAAW,CAAC,MAAM,QAAS,QAA0B,OAAO,GAC/D;GAEF,KAAK,MAAM,WAAW,QAAQ,SAAwD;IACpF,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa;IAE9C,MAAM,WAAW,KAAK,SAAS,IAAI,QAAQ,UAAU;IAErD,IAAI,WAAW;KAIb,IAAI,CAAC,UAAU,YACb,KAAK,SAAS,IAAI,QAAQ,YAAY;MACpC,UAAU,QAAQ;MAClB,UAAU,QAAQ;MAClB,WAAW,QAAQ,WAAW,KAAA;KAChC,CAAC;KAEH,IAAI,QAAQ,UAAU,KAAK,iBAAiB,QAAQ,QAAQ;KAC5D;IACF;IAGA,IAAI,QAAQ;IAEZ,IAAI,SAAS,CAAC,MAAM,YAAY;KAM9B,IAAI,EAFF,QAAQ,aAAa,MAAM,YAC1B,QAAQ,WAAW,KAAA,MAAe,MAAM,YACpB;MACrB,IAAI,QAAQ,UAAU,KAAK,iBAAiB,QAAQ,QAAQ;MAC5D;KACF;KACA,KAAK,SAAS,OAAO,QAAQ,UAAU;KACvC,QAAQ,KAAA;IACV;IAEA,IAAI,CAAC,OACH,QAAQ,KAAK,kBACX,QAAQ,YACR,QAAQ,UACR,QAAQ,WAAW,KAAA,CACrB;IAGF,KAAK,iBAAiB,OAAO,OAAO;IAEpC,IAAI,QAAQ,WAAW,KAAA,KAAa,CAAC,MAAM,WAAW;KAIpD,MAAM,EAAE,YAAY,qBAAqB;KACzC,IAAI,CAAC,kBAAkB;KACvB,MAAM,YAAY;KAClB,MAAM,eAAe;KACrB,iBAAiB,YACf,IAAI,aAAa;MACf,QAAQ,QAAQ;MAChB,UAAU,QAAQ;MAClB,SAAS,QAAQ;MACjB,GAAI,QAAQ,iBAAiB,KAAA,IACzB,EAAE,cAAc,QAAQ,aAAa,IACrC,CAAC;KACP,CAAC,CACH;KACA,iBAAiB,MAAM;IACzB;IAEA,IAAI,QAAQ,UAAU,KAAK,iBAAiB,QAAQ,QAAQ;GAC9D;EACF;CACF;AACF"}

package/dist/store/clients/chain-of-thought-client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chain-of-thought-client.d.ts","names":[],"sources":["../../../src/store/clients/chain-of-thought-client.ts"],"mappings":";;;;;cAaa,oBAAA,GAAoB,KAAA;kBAKb,kBAAA;mBACC,QAAA;IAAY,KAAA;EAAA,MAAoB,WAAA;AAAA;kBADjC,kBAAA;mBACC,QAAA;IAAY,KAAA;EAAA,MAAoB,WAAA;AAAA"}
+{"version":3,"file":"chain-of-thought-client.d.ts","names":[],"sources":["../../../src/store/clients/chain-of-thought-client.ts"],"mappings":";;;;;cAca,oBAAA,GAAoB,KAAA;kBAIf,kBAAA;mBACC,QAAA;IAAY,KAAA;EAAA,MAAoB,WAAA;AAAA;kBADjC,kBAAA;mBACC,QAAA;IAAY,KAAA;EAAA,MAAoB,WAAA;AAAA"}

package/dist/store/clients/chain-of-thought-client.js

@@ -1,12 +1,13 @@
-import { resource, tapMemo, tapState } from "@assistant-ui/tap";
+import { useMemo, useState } from "@assistant-ui/tap/react-shim";
+import { resource } from "@assistant-ui/tap";
 //#region src/store/clients/chain-of-thought-client.ts
 const COMPLETE_STATUS = Object.freeze({ type: "complete" });
-const ChainOfThoughtClient = resource(({ parts, getMessagePart }) => {
-	const [collapsed, setCollapsed] = tapState(true);
-	const status = tapMemo(() => {
+const ChainOfThoughtClient = resource(function ChainOfThoughtClient({ parts, getMessagePart }) {
+	const [collapsed, setCollapsed] = useState(true);
+	const status = useMemo(() => {
 		return parts[parts.length - 1]?.status ?? COMPLETE_STATUS;
 	}, [parts]);
-	const state = tapMemo(() => ({
+	const state = useMemo(() => ({
 		parts,
 		collapsed,
 		status

package/dist/store/clients/chain-of-thought-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"chain-of-thought-client.js","names":[],"sources":["../../../src/store/clients/chain-of-thought-client.ts"],"sourcesContent":["import { resource, tapMemo, tapState } from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport type {\n  ChainOfThoughtState,\n  ChainOfThoughtPart,\n} from \"../scopes/chain-of-thought\";\nimport type { MessagePartStatus } from \"../../types/message\";\nimport type { PartMethods } from \"../scopes/part\";\n\nconst COMPLETE_STATUS: MessagePartStatus = Object.freeze({\n  type: \"complete\",\n});\n\nexport const ChainOfThoughtClient = resource(\n  ({\n    parts,\n    getMessagePart,\n  }: {\n    parts: readonly ChainOfThoughtPart[];\n    getMessagePart: (selector: { index: number }) => PartMethods;\n  }): ClientOutput<\"chainOfThought\"> => {\n    const [collapsed, setCollapsed] = tapState(true);\n\n    const status = tapMemo(() => {\n      const lastPart = parts[parts.length - 1];\n      return lastPart?.status ?? COMPLETE_STATUS;\n    }, [parts]);\n\n    const state = tapMemo<ChainOfThoughtState>(\n      () => ({ parts, collapsed, status }),\n      [parts, collapsed, status],\n    );\n\n    return {\n      getState: () => state,\n      setCollapsed,\n      part: getMessagePart,\n    };\n  },\n);\n"],"mappings":";;AASA,MAAM,kBAAqC,OAAO,OAAO,EACvD,MAAM,WACR,CAAC;AAED,MAAa,uBAAuB,UACjC,EACC,OACA,qBAIoC;CACpC,MAAM,CAAC,WAAW,gBAAgB,SAAS,IAAI;CAE/C,MAAM,SAAS,cAAc;EAE3B,OADiB,MAAM,MAAM,SAAS,IACrB,UAAU;CAC7B,GAAG,CAAC,KAAK,CAAC;CAEV,MAAM,QAAQ,eACL;EAAE;EAAO;EAAW;CAAO,IAClC;EAAC;EAAO;EAAW;CAAM,CAC3B;CAEA,OAAO;EACL,gBAAgB;EAChB;EACA,MAAM;CACR;AACF,CACF"}
+{"version":3,"file":"chain-of-thought-client.js","names":[],"sources":["../../../src/store/clients/chain-of-thought-client.ts"],"sourcesContent":["import { useMemo, useState } from \"react\";\nimport { resource } from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport type {\n  ChainOfThoughtState,\n  ChainOfThoughtPart,\n} from \"../scopes/chain-of-thought\";\nimport type { MessagePartStatus } from \"../../types/message\";\nimport type { PartMethods } from \"../scopes/part\";\n\nconst COMPLETE_STATUS: MessagePartStatus = Object.freeze({\n  type: \"complete\",\n});\n\nexport const ChainOfThoughtClient = resource(function ChainOfThoughtClient({\n  parts,\n  getMessagePart,\n}: {\n  parts: readonly ChainOfThoughtPart[];\n  getMessagePart: (selector: { index: number }) => PartMethods;\n}): ClientOutput<\"chainOfThought\"> {\n  const [collapsed, setCollapsed] = useState(true);\n\n  const status = useMemo(() => {\n    const lastPart = parts[parts.length - 1];\n    return lastPart?.status ?? COMPLETE_STATUS;\n  }, [parts]);\n\n  const state = useMemo<ChainOfThoughtState>(\n    () => ({ parts, collapsed, status }),\n    [parts, collapsed, status],\n  );\n\n  return {\n    getState: () => state,\n    setCollapsed,\n    part: getMessagePart,\n  };\n});\n"],"mappings":";;;AAUA,MAAM,kBAAqC,OAAO,OAAO,EACvD,MAAM,WACR,CAAC;AAED,MAAa,uBAAuB,SAAS,SAAS,qBAAqB,EACzE,OACA,kBAIiC;CACjC,MAAM,CAAC,WAAW,gBAAgB,SAAS,IAAI;CAE/C,MAAM,SAAS,cAAc;EAE3B,OADiB,MAAM,MAAM,SAAS,EACvB,EAAE,UAAU;CAC7B,GAAG,CAAC,KAAK,CAAC;CAEV,MAAM,QAAQ,eACL;EAAE;EAAO;EAAW;CAAO,IAClC;EAAC;EAAO;EAAW;CAAM,CAC3B;CAEA,OAAO;EACL,gBAAgB;EAChB;EACA,MAAM;CACR;AACF,CAAC"}

package/dist/store/clients/model-context-client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"model-context-client.d.ts","names":[],"sources":["../../../src/store/clients/model-context-client.ts"],"mappings":";;;cA8Ba,YAAA,oCAAY,eAAA,CAAA,YAAA"}
+{"version":3,"file":"model-context-client.d.ts","names":[],"sources":["../../../src/store/clients/model-context-client.ts"],"mappings":";;;cA+Ba,YAAA,oCAAY,eAAA,CAAA,YAAA"}

package/dist/store/clients/model-context-client.js

@@ -1,5 +1,6 @@
 import { CompositeContextProvider } from "../../utils/composite-context-provider.js";
-import { resource, tapEffect, tapMemo, tapState } from "@assistant-ui/tap";
+import { useEffect, useMemo, useState } from "@assistant-ui/tap/react-shim";
+import { resource } from "@assistant-ui/tap";
 //#region src/store/clients/model-context-client.ts
 const EMPTY_TOOL_NAMES = [];
 const INITIAL_STATE = {
@@ -18,10 +19,10 @@ const deriveState = (composite, prev) => {
 		toolNames
 	};
 };
-const ModelContext = resource(() => {
-	const composite = tapMemo(() => new CompositeContextProvider(), []);
-	const [state, setState] = tapState(() => deriveState(composite, INITIAL_STATE));
-	tapEffect(() => {
+const ModelContext = resource(function ModelContext() {
+	const composite = useMemo(() => new CompositeContextProvider(), []);
+	const [state, setState] = useState(() => deriveState(composite, INITIAL_STATE));
+	useEffect(() => {
 		setState((prev) => deriveState(composite, prev));
 		return composite.subscribe(() => {
 			setState((prev) => deriveState(composite, prev));

package/dist/store/clients/model-context-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"model-context-client.js","names":[],"sources":["../../../src/store/clients/model-context-client.ts"],"sourcesContent":["import { resource, tapEffect, tapMemo, tapState } from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport { CompositeContextProvider } from \"../../utils/composite-context-provider\";\nimport type { ModelContextState } from \"../scopes/model-context\";\n\nconst EMPTY_TOOL_NAMES: readonly string[] = [];\n\nconst INITIAL_STATE: ModelContextState = {\n  modelName: undefined,\n  toolNames: EMPTY_TOOL_NAMES,\n};\n\nconst toolNamesEqual = (a: readonly string[], b: readonly string[]): boolean =>\n  a === b || (a.length === b.length && a.every((v, i) => v === b[i]));\n\nconst deriveState = (\n  composite: CompositeContextProvider,\n  prev: ModelContextState,\n): ModelContextState => {\n  const ctx = composite.getModelContext();\n  const modelName = ctx.config?.modelName;\n  const keys = ctx.tools ? Object.keys(ctx.tools).sort() : EMPTY_TOOL_NAMES;\n  const toolNames = keys.length ? keys : EMPTY_TOOL_NAMES;\n\n  if (modelName === prev.modelName && toolNamesEqual(toolNames, prev.toolNames))\n    return prev;\n\n  return { modelName, toolNames };\n};\n\nexport const ModelContext = resource((): ClientOutput<\"modelContext\"> => {\n  const composite = tapMemo(() => new CompositeContextProvider(), []);\n  const [state, setState] = tapState<ModelContextState>(() =>\n    deriveState(composite, INITIAL_STATE),\n  );\n\n  tapEffect(() => {\n    setState((prev) => deriveState(composite, prev));\n    return composite.subscribe(() => {\n      setState((prev) => deriveState(composite, prev));\n    });\n  }, [composite]);\n\n  return {\n    getState: () => deriveState(composite, state),\n    getModelContext: () => composite.getModelContext(),\n    subscribe: (callback) => composite.subscribe(callback),\n    register: (provider) => composite.registerModelContextProvider(provider),\n  };\n});\n"],"mappings":";;;AAKA,MAAM,mBAAsC,CAAC;AAE7C,MAAM,gBAAmC;CACvC,WAAW,KAAA;CACX,WAAW;AACb;AAEA,MAAM,kBAAkB,GAAsB,MAC5C,MAAM,KAAM,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,GAAG,MAAM,MAAM,EAAE,EAAE;AAEnE,MAAM,eACJ,WACA,SACsB;CACtB,MAAM,MAAM,UAAU,gBAAgB;CACtC,MAAM,YAAY,IAAI,QAAQ;CAC9B,MAAM,OAAO,IAAI,QAAQ,OAAO,KAAK,IAAI,KAAK,EAAE,KAAK,IAAI;CACzD,MAAM,YAAY,KAAK,SAAS,OAAO;CAEvC,IAAI,cAAc,KAAK,aAAa,eAAe,WAAW,KAAK,SAAS,GAC1E,OAAO;CAET,OAAO;EAAE;EAAW;CAAU;AAChC;AAEA,MAAa,eAAe,eAA6C;CACvE,MAAM,YAAY,cAAc,IAAI,yBAAyB,GAAG,CAAC,CAAC;CAClE,MAAM,CAAC,OAAO,YAAY,eACxB,YAAY,WAAW,aAAa,CACtC;CAEA,gBAAgB;EACd,UAAU,SAAS,YAAY,WAAW,IAAI,CAAC;EAC/C,OAAO,UAAU,gBAAgB;GAC/B,UAAU,SAAS,YAAY,WAAW,IAAI,CAAC;EACjD,CAAC;CACH,GAAG,CAAC,SAAS,CAAC;CAEd,OAAO;EACL,gBAAgB,YAAY,WAAW,KAAK;EAC5C,uBAAuB,UAAU,gBAAgB;EACjD,YAAY,aAAa,UAAU,UAAU,QAAQ;EACrD,WAAW,aAAa,UAAU,6BAA6B,QAAQ;CACzE;AACF,CAAC"}
+{"version":3,"file":"model-context-client.js","names":[],"sources":["../../../src/store/clients/model-context-client.ts"],"sourcesContent":["import { useEffect, useMemo, useState } from \"react\";\nimport { resource } from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport { CompositeContextProvider } from \"../../utils/composite-context-provider\";\nimport type { ModelContextState } from \"../scopes/model-context\";\n\nconst EMPTY_TOOL_NAMES: readonly string[] = [];\n\nconst INITIAL_STATE: ModelContextState = {\n  modelName: undefined,\n  toolNames: EMPTY_TOOL_NAMES,\n};\n\nconst toolNamesEqual = (a: readonly string[], b: readonly string[]): boolean =>\n  a === b || (a.length === b.length && a.every((v, i) => v === b[i]));\n\nconst deriveState = (\n  composite: CompositeContextProvider,\n  prev: ModelContextState,\n): ModelContextState => {\n  const ctx = composite.getModelContext();\n  const modelName = ctx.config?.modelName;\n  const keys = ctx.tools ? Object.keys(ctx.tools).sort() : EMPTY_TOOL_NAMES;\n  const toolNames = keys.length ? keys : EMPTY_TOOL_NAMES;\n\n  if (modelName === prev.modelName && toolNamesEqual(toolNames, prev.toolNames))\n    return prev;\n\n  return { modelName, toolNames };\n};\n\nexport const ModelContext = resource(\n  function ModelContext(): ClientOutput<\"modelContext\"> {\n    const composite = useMemo(() => new CompositeContextProvider(), []);\n    const [state, setState] = useState<ModelContextState>(() =>\n      deriveState(composite, INITIAL_STATE),\n    );\n\n    useEffect(() => {\n      setState((prev) => deriveState(composite, prev));\n      return composite.subscribe(() => {\n        setState((prev) => deriveState(composite, prev));\n      });\n    }, [composite]);\n\n    return {\n      getState: () => deriveState(composite, state),\n      getModelContext: () => composite.getModelContext(),\n      subscribe: (callback) => composite.subscribe(callback),\n      register: (provider) => composite.registerModelContextProvider(provider),\n    };\n  },\n);\n"],"mappings":";;;;AAMA,MAAM,mBAAsC,CAAC;AAE7C,MAAM,gBAAmC;CACvC,WAAW,KAAA;CACX,WAAW;AACb;AAEA,MAAM,kBAAkB,GAAsB,MAC5C,MAAM,KAAM,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,GAAG,MAAM,MAAM,EAAE,EAAE;AAEnE,MAAM,eACJ,WACA,SACsB;CACtB,MAAM,MAAM,UAAU,gBAAgB;CACtC,MAAM,YAAY,IAAI,QAAQ;CAC9B,MAAM,OAAO,IAAI,QAAQ,OAAO,KAAK,IAAI,KAAK,CAAC,CAAC,KAAK,IAAI;CACzD,MAAM,YAAY,KAAK,SAAS,OAAO;CAEvC,IAAI,cAAc,KAAK,aAAa,eAAe,WAAW,KAAK,SAAS,GAC1E,OAAO;CAET,OAAO;EAAE;EAAW;CAAU;AAChC;AAEA,MAAa,eAAe,SAC1B,SAAS,eAA6C;CACpD,MAAM,YAAY,cAAc,IAAI,yBAAyB,GAAG,CAAC,CAAC;CAClE,MAAM,CAAC,OAAO,YAAY,eACxB,YAAY,WAAW,aAAa,CACtC;CAEA,gBAAgB;EACd,UAAU,SAAS,YAAY,WAAW,IAAI,CAAC;EAC/C,OAAO,UAAU,gBAAgB;GAC/B,UAAU,SAAS,YAAY,WAAW,IAAI,CAAC;EACjD,CAAC;CACH,GAAG,CAAC,SAAS,CAAC;CAEd,OAAO;EACL,gBAAgB,YAAY,WAAW,KAAK;EAC5C,uBAAuB,UAAU,gBAAgB;EACjD,YAAY,aAAa,UAAU,UAAU,QAAQ;EACrD,WAAW,aAAa,UAAU,6BAA6B,QAAQ;CACzE;AACF,CACF"}

package/dist/store/clients/no-op-composer-client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"no-op-composer-client.d.ts","names":[],"sources":["../../../src/store/clients/no-op-composer-client.ts"],"mappings":";;;cAIa,kBAAA,GAAkB,KAAA;;kDAkE9B,YAAA"}
+{"version":3,"file":"no-op-composer-client.d.ts","names":[],"sources":["../../../src/store/clients/no-op-composer-client.ts"],"mappings":";;;cAKa,kBAAA,GAAkB,KAAA;;kDAoE7B,YAAA"}

package/dist/store/clients/no-op-composer-client.js

@@ -1,7 +1,8 @@
-import { resource, tapMemo } from "@assistant-ui/tap";
+import { useMemo } from "@assistant-ui/tap/react-shim";
+import { resource } from "@assistant-ui/tap";
 //#region src/store/clients/no-op-composer-client.ts
-const NoOpComposerClient = resource(({ type }) => {
-	const state = tapMemo(() => {
+const NoOpComposerClient = resource(function NoOpComposerClient({ type }) {
+	const state = useMemo(() => {
 		return {
 			isEditing: false,
 			isEmpty: true,

package/dist/store/clients/no-op-composer-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"no-op-composer-client.js","names":[],"sources":["../../../src/store/clients/no-op-composer-client.ts"],"sourcesContent":["import { resource, tapMemo } from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport type { ComposerState } from \"../scopes/composer\";\n\nexport const NoOpComposerClient = resource(\n  ({ type }: { type: \"edit\" | \"thread\" }): ClientOutput<\"composer\"> => {\n    const state = tapMemo<ComposerState>(() => {\n      return {\n        isEditing: false,\n        isEmpty: true,\n        text: \"\",\n        attachmentAccept: \"*\",\n        attachments: [],\n        role: \"user\",\n        runConfig: {},\n        canCancel: false,\n        canSend: false,\n        type: type,\n        dictation: undefined,\n        quote: undefined,\n        queue: [],\n      };\n    }, [type]);\n\n    return {\n      getState: () => state,\n      setText: () => {\n        throw new Error(\"Not supported\");\n      },\n      setRole: () => {\n        throw new Error(\"Not supported\");\n      },\n      setRunConfig: () => {\n        throw new Error(\"Not supported\");\n      },\n      addAttachment: () => {\n        throw new Error(\"Not supported\");\n      },\n      clearAttachments: () => {\n        throw new Error(\"Not supported\");\n      },\n      attachment: () => {\n        throw new Error(\"Not supported\");\n      },\n      reset: () => {\n        throw new Error(\"Not supported\");\n      },\n      send: () => {\n        throw new Error(\"Not supported\");\n      },\n      cancel: () => {\n        throw new Error(\"Not supported\");\n      },\n      startDictation: () => {\n        throw new Error(\"Not supported\");\n      },\n      stopDictation: () => {\n        throw new Error(\"Not supported\");\n      },\n      beginEdit: () => {\n        throw new Error(\"Not supported\");\n      },\n      setQuote: () => {\n        throw new Error(\"Not supported\");\n      },\n      queueItem: () => {\n        throw new Error(\"Not supported\");\n      },\n    };\n  },\n);\n"],"mappings":";;AAIA,MAAa,qBAAqB,UAC/B,EAAE,WAAkE;CACnE,MAAM,QAAQ,cAA6B;EACzC,OAAO;GACL,WAAW;GACX,SAAS;GACT,MAAM;GACN,kBAAkB;GAClB,aAAa,CAAC;GACd,MAAM;GACN,WAAW,CAAC;GACZ,WAAW;GACX,SAAS;GACH;GACN,WAAW,KAAA;GACX,OAAO,KAAA;GACP,OAAO,CAAC;EACV;CACF,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;EACL,gBAAgB;EAChB,eAAe;GACb,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,eAAe;GACb,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,oBAAoB;GAClB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,qBAAqB;GACnB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,wBAAwB;GACtB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,kBAAkB;GAChB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,aAAa;GACX,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,YAAY;GACV,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,cAAc;GACZ,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,sBAAsB;GACpB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,qBAAqB;GACnB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,iBAAiB;GACf,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,gBAAgB;GACd,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,iBAAiB;GACf,MAAM,IAAI,MAAM,eAAe;EACjC;CACF;AACF,CACF"}
+{"version":3,"file":"no-op-composer-client.js","names":[],"sources":["../../../src/store/clients/no-op-composer-client.ts"],"sourcesContent":["import { useMemo } from \"react\";\nimport { resource } from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport type { ComposerState } from \"../scopes/composer\";\n\nexport const NoOpComposerClient = resource(function NoOpComposerClient({\n  type,\n}: {\n  type: \"edit\" | \"thread\";\n}): ClientOutput<\"composer\"> {\n  const state = useMemo<ComposerState>(() => {\n    return {\n      isEditing: false,\n      isEmpty: true,\n      text: \"\",\n      attachmentAccept: \"*\",\n      attachments: [],\n      role: \"user\",\n      runConfig: {},\n      canCancel: false,\n      canSend: false,\n      type: type,\n      dictation: undefined,\n      quote: undefined,\n      queue: [],\n    };\n  }, [type]);\n\n  return {\n    getState: () => state,\n    setText: () => {\n      throw new Error(\"Not supported\");\n    },\n    setRole: () => {\n      throw new Error(\"Not supported\");\n    },\n    setRunConfig: () => {\n      throw new Error(\"Not supported\");\n    },\n    addAttachment: () => {\n      throw new Error(\"Not supported\");\n    },\n    clearAttachments: () => {\n      throw new Error(\"Not supported\");\n    },\n    attachment: () => {\n      throw new Error(\"Not supported\");\n    },\n    reset: () => {\n      throw new Error(\"Not supported\");\n    },\n    send: () => {\n      throw new Error(\"Not supported\");\n    },\n    cancel: () => {\n      throw new Error(\"Not supported\");\n    },\n    startDictation: () => {\n      throw new Error(\"Not supported\");\n    },\n    stopDictation: () => {\n      throw new Error(\"Not supported\");\n    },\n    beginEdit: () => {\n      throw new Error(\"Not supported\");\n    },\n    setQuote: () => {\n      throw new Error(\"Not supported\");\n    },\n    queueItem: () => {\n      throw new Error(\"Not supported\");\n    },\n  };\n});\n"],"mappings":";;;AAKA,MAAa,qBAAqB,SAAS,SAAS,mBAAmB,EACrE,QAG2B;CAC3B,MAAM,QAAQ,cAA6B;EACzC,OAAO;GACL,WAAW;GACX,SAAS;GACT,MAAM;GACN,kBAAkB;GAClB,aAAa,CAAC;GACd,MAAM;GACN,WAAW,CAAC;GACZ,WAAW;GACX,SAAS;GACH;GACN,WAAW,KAAA;GACX,OAAO,KAAA;GACP,OAAO,CAAC;EACV;CACF,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;EACL,gBAAgB;EAChB,eAAe;GACb,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,eAAe;GACb,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,oBAAoB;GAClB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,qBAAqB;GACnB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,wBAAwB;GACtB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,kBAAkB;GAChB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,aAAa;GACX,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,YAAY;GACV,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,cAAc;GACZ,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,sBAAsB;GACpB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,qBAAqB;GACnB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,iBAAiB;GACf,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,gBAAgB;GACd,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,iBAAiB;GACf,MAAM,IAAI,MAAM,eAAe;EACjC;CACF;AACF,CAAC"}

package/dist/store/clients/runtime-adapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runtime-adapter.d.ts","names":[],"sources":["../../../src/store/clients/runtime-adapter.ts"],"mappings":";;;;cAYa,sBAAA,GAAsB,KAAA,EAAA,gBAAA,iCAAA,eAAA,+BAAA,YAAA,aAAA,gBAAA;AAAA,cAiBtB,iCAAA,GACX,MAAA,EAAQ,YAAA,EACR,MAAA,EAAQ,eAAe"}
+{"version":3,"file":"runtime-adapter.d.ts","names":[],"sources":["../../../src/store/clients/runtime-adapter.ts"],"mappings":";;;;cAaa,sBAAA,GAAsB,KAAA,EAAA,gBAAA,iCAAA,eAAA,+BAAA,YAAA,aAAA,gBAAA;AAAA,cAmBtB,iCAAA,GACX,MAAA,EAAQ,YAAA,EACR,MAAA,EAAQ,eAAe"}

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

@@ -1,15 +1,16 @@
 import { ThreadListClient } from "../runtime-clients/thread-list-runtime-client.js";
 import { ModelContext } from "./model-context-client.js";
 import { Suggestions } from "./suggestions.js";
-import { Derived, tapAssistantClientRef } from "@assistant-ui/store";
-import { resource, tapEffect, tapResource } from "@assistant-ui/tap";
+import { useEffect } from "@assistant-ui/tap/react-shim";
+import { Derived, useAssistantClientRef } from "@assistant-ui/store";
+import { resource, useResource } from "@assistant-ui/tap";
 //#region src/store/clients/runtime-adapter.ts
-const RuntimeAdapterResource = resource((runtime) => {
-	const clientRef = tapAssistantClientRef();
-	tapEffect(() => {
+const RuntimeAdapterResource = resource(function RuntimeAdapterResource(runtime) {
+	const clientRef = useAssistantClientRef();
+	useEffect(() => {
 		return runtime.registerModelContextProvider(clientRef.current.modelContext());
 	}, [runtime, clientRef]);
-	return tapResource(ThreadListClient({
+	return useResource(ThreadListClient({
 		runtime: runtime.threads,
 		__internal_assistantRuntime: runtime
 	}));

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

@@ -1 +1 @@
-{"version":3,"file":"runtime-adapter.js","names":[],"sources":["../../../src/store/clients/runtime-adapter.ts"],"sourcesContent":["import { resource, tapEffect, tapResource } from \"@assistant-ui/tap\";\nimport type { AssistantRuntime } from \"../../runtime/api/assistant-runtime\";\nimport { ThreadListClient } from \"../runtime-clients/thread-list-runtime-client\";\nimport {\n  tapAssistantClientRef,\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 RuntimeAdapterResource = resource((runtime: AssistantRuntime) => {\n  const clientRef = tapAssistantClientRef();\n\n  tapEffect(() => {\n    return runtime.registerModelContextProvider(\n      clientRef.current!.modelContext(),\n    );\n  }, [runtime, clientRef]);\n\n  return tapResource(\n    ThreadListClient({\n      runtime: runtime.threads,\n      __internal_assistantRuntime: runtime,\n    }),\n  );\n});\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":";;;;;;AAYA,MAAa,yBAAyB,UAAU,YAA8B;CAC5E,MAAM,YAAY,sBAAsB;CAExC,gBAAgB;EACd,OAAO,QAAQ,6BACb,UAAU,QAAS,aAAa,CAClC;CACF,GAAG,CAAC,SAAS,SAAS,CAAC;CAEvB,OAAO,YACL,iBAAiB;EACf,SAAS,QAAQ;EACjB,6BAA6B;CAC/B,CAAC,CACH;AACF,CAAC;AAED,MAAa,qCACX,QACA,WACS;CACT,OAAO,WAAW,QAAQ;EACxB,QAAQ;EACR,OAAO,EAAE,MAAM,OAAO;EACtB,MAAM,QAAQ,IAAI,QAAQ,EAAE,OAAO,MAAM;CAC3C,CAAC;CACD,OAAO,mBAAmB,QAAQ;EAChC,QAAQ;EACR,OAAO,EAAE,MAAM,OAAO;EACtB,MAAM,QAAQ,IAAI,QAAQ,EAAE,KAAK,MAAM;CACzC,CAAC;CACD,OAAO,aAAa,QAAQ;EAC1B,QAAQ;EACR,OAAO,CAAC;EACR,MAAM,QAAQ,IAAI,QAAQ,EAAE,OAAO,MAAM,EAAE,SAAS;CACtD,CAAC;CAED,IAAI,CAAC,OAAO,gBAAgB,OAAO,aAAa,WAAW,MACzD,OAAO,eAAe,aAAa;CAErC,IAAI,CAAC,OAAO,eAAe,OAAO,YAAY,WAAW,MACvD,OAAO,cAAc,YAAY;AAErC"}
+{"version":3,"file":"runtime-adapter.js","names":[],"sources":["../../../src/store/clients/runtime-adapter.ts"],"sourcesContent":["import { useEffect } from \"react\";\nimport { useResource, resource } from \"@assistant-ui/tap\";\nimport type { AssistantRuntime } from \"../../runtime/api/assistant-runtime\";\nimport { ThreadListClient } from \"../runtime-clients/thread-list-runtime-client\";\nimport {\n  useAssistantClientRef,\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 RuntimeAdapterResource = resource(function RuntimeAdapterResource(\n  runtime: AssistantRuntime,\n) {\n  const clientRef = useAssistantClientRef();\n\n  useEffect(() => {\n    return runtime.registerModelContextProvider(\n      clientRef.current!.modelContext(),\n    );\n  }, [runtime, clientRef]);\n\n  return useResource(\n    ThreadListClient({\n      runtime: runtime.threads,\n      __internal_assistantRuntime: runtime,\n    }),\n  );\n});\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":";;;;;;;AAaA,MAAa,yBAAyB,SAAS,SAAS,uBACtD,SACA;CACA,MAAM,YAAY,sBAAsB;CAExC,gBAAgB;EACd,OAAO,QAAQ,6BACb,UAAU,QAAS,aAAa,CAClC;CACF,GAAG,CAAC,SAAS,SAAS,CAAC;CAEvB,OAAO,YACL,iBAAiB;EACf,SAAS,QAAQ;EACjB,6BAA6B;CAC/B,CAAC,CACH;AACF,CAAC;AAED,MAAa,qCACX,QACA,WACS;CACT,OAAO,WAAW,QAAQ;EACxB,QAAQ;EACR,OAAO,EAAE,MAAM,OAAO;EACtB,MAAM,QAAQ,IAAI,QAAQ,CAAC,CAAC,OAAO,MAAM;CAC3C,CAAC;CACD,OAAO,mBAAmB,QAAQ;EAChC,QAAQ;EACR,OAAO,EAAE,MAAM,OAAO;EACtB,MAAM,QAAQ,IAAI,QAAQ,CAAC,CAAC,KAAK,MAAM;CACzC,CAAC;CACD,OAAO,aAAa,QAAQ;EAC1B,QAAQ;EACR,OAAO,CAAC;EACR,MAAM,QAAQ,IAAI,QAAQ,CAAC,CAAC,OAAO,MAAM,CAAC,CAAC,SAAS;CACtD,CAAC;CAED,IAAI,CAAC,OAAO,gBAAgB,OAAO,aAAa,WAAW,MACzD,OAAO,eAAe,aAAa;CAErC,IAAI,CAAC,OAAO,eAAe,OAAO,YAAY,WAAW,MACvD,OAAO,cAAc,YAAY;AAErC"}

package/dist/store/clients/suggestions.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"suggestions.d.ts","names":[],"sources":["../../../src/store/clients/suggestions.ts"],"mappings":";;;KAMY,gBAAA;EAEN,KAAA;EAAe,KAAA;EAAe,MAAA;AAAA;AAAA,cAkDvB,WAAA;EAAA,gCACqB,eAAA,CAC9B,YAAA;EAAA,CAIA,WAAA,EAAa,gBAAA,iCACgB,eAAA,CAC7B,YAAA,iBACA,gBAAA;AAAA"}
+{"version":3,"file":"suggestions.d.ts","names":[],"sources":["../../../src/store/clients/suggestions.ts"],"mappings":";;;KAOY,gBAAA;EAEN,KAAA;EAAe,KAAA;EAAe,MAAA;AAAA;AAAA,cAkDvB,WAAA;EAAA,gCACqB,eAAA,CAC9B,YAAA;EAAA,CAIA,WAAA,EAAa,gBAAA,iCACgB,eAAA,CAC7B,YAAA,iBACA,gBAAA;AAAA"}

package/dist/store/clients/suggestions.js

@@ -1,11 +1,12 @@
-import { tapClientLookup } from "@assistant-ui/store";
-import { resource, tapState, withKey } from "@assistant-ui/tap";
+import { useState } from "@assistant-ui/tap/react-shim";
+import { useClientLookup } from "@assistant-ui/store";
+import { resource, withKey } from "@assistant-ui/tap";
 //#region src/store/clients/suggestions.ts
-const SuggestionClient = resource((state) => {
+const SuggestionClient = resource(function SuggestionClient(state) {
 	return { getState: () => state };
 });
-const Suggestions = resource((suggestions) => {
-	const [state] = tapState(() => {
+const Suggestions = resource(function SuggestionsResource(suggestions) {
+	const [state] = useState(() => {
 		return { suggestions: (suggestions ?? []).map((s) => {
 			if (typeof s === "string") return {
 				title: s,
@@ -19,7 +20,7 @@ const Suggestions = resource((suggestions) => {
 			};
 		}) };
 	});
-	const suggestionClients = tapClientLookup(() => state.suggestions.map((suggestion, index) => withKey(index, SuggestionClient(suggestion))), [state.suggestions]);
+	const suggestionClients = useClientLookup(() => state.suggestions.map((suggestion, index) => withKey(index, SuggestionClient(suggestion))), [state.suggestions]);
 	return {
 		getState: () => state,
 		suggestion: ({ index }) => {

package/dist/store/clients/suggestions.js.map

@@ -1 +1 @@
-{"version":3,"file":"suggestions.js","names":[],"sources":["../../../src/store/clients/suggestions.ts"],"sourcesContent":["import { resource, tapState, withKey } from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport { tapClientLookup } from \"@assistant-ui/store\";\nimport type { SuggestionsState } from \"../scopes/suggestions\";\nimport type { SuggestionState } from \"../scopes/suggestion\";\n\nexport type SuggestionConfig =\n  | string\n  | { title: string; label: string; prompt: string };\n\nconst SuggestionClient = resource(\n  (state: SuggestionState): ClientOutput<\"suggestion\"> => {\n    return {\n      getState: () => state,\n    };\n  },\n);\n\nconst SuggestionsResource = resource(\n  (suggestions?: SuggestionConfig[]): ClientOutput<\"suggestions\"> => {\n    const [state] = tapState<SuggestionsState>(() => {\n      const normalizedSuggestions = (suggestions ?? []).map((s) => {\n        if (typeof s === \"string\") {\n          return {\n            title: s,\n            label: \"\",\n            prompt: s,\n          };\n        }\n        return {\n          title: s.title,\n          label: s.label,\n          prompt: s.prompt,\n        };\n      });\n\n      return {\n        suggestions: normalizedSuggestions,\n      };\n    });\n\n    const suggestionClients = tapClientLookup(\n      () =>\n        state.suggestions.map((suggestion, index) =>\n          withKey(index, SuggestionClient(suggestion)),\n        ),\n      [state.suggestions],\n    );\n\n    return {\n      getState: () => state,\n      suggestion: ({ index }: { index: number }) => {\n        return suggestionClients.get({ index });\n      },\n    };\n  },\n);\n\nexport const Suggestions: {\n  (): import(\"@assistant-ui/tap\").ResourceElement<\n    ClientOutput<\"suggestions\">,\n    undefined\n  >;\n  (\n    suggestions: SuggestionConfig[],\n  ): import(\"@assistant-ui/tap\").ResourceElement<\n    ClientOutput<\"suggestions\">,\n    SuggestionConfig[]\n  >;\n} = SuggestionsResource as any;\n"],"mappings":";;;AAUA,MAAM,mBAAmB,UACtB,UAAuD;CACtD,OAAO,EACL,gBAAgB,MAClB;AACF,CACF;AA0CA,MAAa,cAxCe,UACzB,gBAAkE;CACjE,MAAM,CAAC,SAAS,eAAiC;EAgB/C,OAAO,EACL,cAhB6B,eAAe,CAAC,GAAG,KAAK,MAAM;GAC3D,IAAI,OAAO,MAAM,UACf,OAAO;IACL,OAAO;IACP,OAAO;IACP,QAAQ;GACV;GAEF,OAAO;IACL,OAAO,EAAE;IACT,OAAO,EAAE;IACT,QAAQ,EAAE;GACZ;EACF,CAGmC,EACnC;CACF,CAAC;CAED,MAAM,oBAAoB,sBAEtB,MAAM,YAAY,KAAK,YAAY,UACjC,QAAQ,OAAO,iBAAiB,UAAU,CAAC,CAC7C,GACF,CAAC,MAAM,WAAW,CACpB;CAEA,OAAO;EACL,gBAAgB;EAChB,aAAa,EAAE,YAA+B;GAC5C,OAAO,kBAAkB,IAAI,EAAE,MAAM,CAAC;EACxC;CACF;AACF,CAcE"}
+{"version":3,"file":"suggestions.js","names":[],"sources":["../../../src/store/clients/suggestions.ts"],"sourcesContent":["import { useState } from \"react\";\nimport { resource, withKey } from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport { useClientLookup } from \"@assistant-ui/store\";\nimport type { SuggestionsState } from \"../scopes/suggestions\";\nimport type { SuggestionState } from \"../scopes/suggestion\";\n\nexport type SuggestionConfig =\n  | string\n  | { title: string; label: string; prompt: string };\n\nconst SuggestionClient = resource(function SuggestionClient(\n  state: SuggestionState,\n): ClientOutput<\"suggestion\"> {\n  return {\n    getState: () => state,\n  };\n});\n\nconst SuggestionsResource = resource(function SuggestionsResource(\n  suggestions?: SuggestionConfig[],\n): ClientOutput<\"suggestions\"> {\n  const [state] = useState<SuggestionsState>(() => {\n    const normalizedSuggestions = (suggestions ?? []).map((s) => {\n      if (typeof s === \"string\") {\n        return {\n          title: s,\n          label: \"\",\n          prompt: s,\n        };\n      }\n      return {\n        title: s.title,\n        label: s.label,\n        prompt: s.prompt,\n      };\n    });\n\n    return {\n      suggestions: normalizedSuggestions,\n    };\n  });\n\n  const suggestionClients = useClientLookup(\n    () =>\n      state.suggestions.map((suggestion, index) =>\n        withKey(index, SuggestionClient(suggestion)),\n      ),\n    [state.suggestions],\n  );\n\n  return {\n    getState: () => state,\n    suggestion: ({ index }: { index: number }) => {\n      return suggestionClients.get({ index });\n    },\n  };\n});\n\nexport const Suggestions: {\n  (): import(\"@assistant-ui/tap\").ResourceElement<\n    ClientOutput<\"suggestions\">,\n    undefined\n  >;\n  (\n    suggestions: SuggestionConfig[],\n  ): import(\"@assistant-ui/tap\").ResourceElement<\n    ClientOutput<\"suggestions\">,\n    SuggestionConfig[]\n  >;\n} = SuggestionsResource as any;\n"],"mappings":";;;;AAWA,MAAM,mBAAmB,SAAS,SAAS,iBACzC,OAC4B;CAC5B,OAAO,EACL,gBAAgB,MAClB;AACF,CAAC;AA0CD,MAAa,cAxCe,SAAS,SAAS,oBAC5C,aAC6B;CAC7B,MAAM,CAAC,SAAS,eAAiC;EAgB/C,OAAO,EACL,cAhB6B,eAAe,CAAC,EAAA,CAAG,KAAK,MAAM;GAC3D,IAAI,OAAO,MAAM,UACf,OAAO;IACL,OAAO;IACP,OAAO;IACP,QAAQ;GACV;GAEF,OAAO;IACL,OAAO,EAAE;IACT,OAAO,EAAE;IACT,QAAQ,EAAE;GACZ;EACF,CAGmC,EACnC;CACF,CAAC;CAED,MAAM,oBAAoB,sBAEtB,MAAM,YAAY,KAAK,YAAY,UACjC,QAAQ,OAAO,iBAAiB,UAAU,CAAC,CAC7C,GACF,CAAC,MAAM,WAAW,CACpB;CAEA,OAAO;EACL,gBAAgB;EAChB,aAAa,EAAE,YAA+B;GAC5C,OAAO,kBAAkB,IAAI,EAAE,MAAM,CAAC;EACxC;CACF;AACF,CAaI"}

package/dist/store/clients/thread-message-client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"thread-message-client.d.ts","names":[],"sources":["../../../src/store/clients/thread-message-client.ts"],"mappings":";;;;KAyDY,wBAAA;EACV,OAAA,EAAS,aAAa;EACtB,KAAA;EACA,MAAA;EACA,YAAA;EACA,WAAA;AAAA;AAAA,cAGW,mBAAA,GAAmB,KAAA,EAAA,wBAAA,iCAAA,eAAA,CAAA,YAAA,aAAA,wBAAA"}
+{"version":3,"file":"thread-message-client.d.ts","names":[],"sources":["../../../src/store/clients/thread-message-client.ts"],"mappings":";;;;KAsDY,wBAAA;EACV,OAAA,EAAS,aAAa;EACtB,KAAA;EACA,MAAA;EACA,YAAA;EACA,WAAA;AAAA;AAAA,cAGW,mBAAA,GAAmB,KAAA,EAAA,wBAAA,iCAAA,eAAA,CAAA,YAAA,aAAA,wBAAA"}

package/dist/store/clients/thread-message-client.js

@@ -1,9 +1,10 @@
 import { NoOpComposerClient } from "./no-op-composer-client.js";
-import { tapClientLookup } from "@assistant-ui/store";
-import { resource, tapMemo, tapResource, tapState, withKey } from "@assistant-ui/tap";
+import { useMemo, useState } from "@assistant-ui/tap/react-shim";
+import { useClientLookup } from "@assistant-ui/store";
+import { resource, useResource, withKey } from "@assistant-ui/tap";
 //#region src/store/clients/thread-message-client.ts
-const ThreadMessagePartClient = resource(({ part }) => {
-	const state = tapMemo(() => {
+const ThreadMessagePartClient = resource(function ThreadMessagePartClient({ part }) {
+	const state = useMemo(() => {
 		return {
 			...part,
 			status: { type: "complete" }
@@ -22,7 +23,7 @@ const ThreadMessagePartClient = resource(({ part }) => {
 		}
 	};
 });
-const ThreadMessageAttachmentClient = resource(({ attachment }) => {
+const ThreadMessageAttachmentClient = resource(function ThreadMessageAttachmentClient({ attachment }) {
 	return {
 		getState: () => attachment,
 		remove: () => {
@@ -30,14 +31,14 @@ const ThreadMessageAttachmentClient = resource(({ attachment }) => {
 		}
 	};
 });
-const ThreadMessageClient = resource(({ message, index, isLast = true, branchNumber = 1, branchCount = 1 }) => {
-	const [isCopiedState, setIsCopied] = tapState(false);
-	const [isHoveringState, setIsHovering] = tapState(false);
-	const parts = tapClientLookup(() => message.content.map((part, idx) => withKey("toolCallId" in part && part.toolCallId != null ? `toolCallId-${part.toolCallId}` : `index-${idx}`, ThreadMessagePartClient({ part }))), [message.content]);
-	const attachments = tapClientLookup(() => (message.attachments ?? []).map((attachment) => withKey(attachment.id, ThreadMessageAttachmentClient({ attachment }))), [message.attachments]);
-	const composer = tapResource(NoOpComposerClient({ type: "edit" }));
+const ThreadMessageClient = resource(function ThreadMessageClient({ message, index, isLast = true, branchNumber = 1, branchCount = 1 }) {
+	const [isCopiedState, setIsCopied] = useState(false);
+	const [isHoveringState, setIsHovering] = useState(false);
+	const parts = useClientLookup(() => message.content.map((part, idx) => withKey("toolCallId" in part && part.toolCallId != null ? `toolCallId-${part.toolCallId}` : `index-${idx}`, ThreadMessagePartClient({ part }))), [message.content]);
+	const attachments = useClientLookup(() => (message.attachments ?? []).map((attachment) => withKey(attachment.id, ThreadMessageAttachmentClient({ attachment }))), [message.attachments]);
+	const composer = useResource(NoOpComposerClient({ type: "edit" }));
 	const composerState = composer.getState();
-	const state = tapMemo(() => {
+	const state = useMemo(() => {
 		return {
 			...message,
 			parts: parts.state,

package/dist/store/clients/thread-message-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"thread-message-client.js","names":[],"sources":["../../../src/store/clients/thread-message-client.ts"],"sourcesContent":["import type {\n  ThreadAssistantMessagePart,\n  ThreadUserMessagePart,\n  ThreadMessage,\n} from \"../../types/message\";\nimport type { Attachment } from \"../../types/attachment\";\nimport {\n  resource,\n  tapMemo,\n  tapState,\n  tapResource,\n  withKey,\n} from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport { tapClientLookup } from \"@assistant-ui/store\";\nimport type { MessageState } from \"../scopes/message\";\nimport type { PartState } from \"../scopes/part\";\nimport { NoOpComposerClient } from \"./no-op-composer-client\";\n\nconst ThreadMessagePartClient = resource(\n  ({\n    part,\n  }: {\n    part: ThreadAssistantMessagePart | ThreadUserMessagePart;\n  }): ClientOutput<\"part\"> => {\n    const state = tapMemo<PartState>(() => {\n      return {\n        ...part,\n        status: { type: \"complete\" },\n      };\n    }, [part]);\n\n    return {\n      getState: () => state,\n      addToolResult: () => {\n        throw new Error(\"Not supported\");\n      },\n      resumeToolCall: () => {\n        throw new Error(\"Not supported\");\n      },\n      respondToToolApproval: () => {\n        throw new Error(\"Not supported\");\n      },\n    };\n  },\n);\n\nconst ThreadMessageAttachmentClient = resource(\n  ({ attachment }: { attachment: Attachment }): ClientOutput<\"attachment\"> => {\n    return {\n      getState: () => attachment,\n      remove: () => {\n        throw new Error(\"Not supported\");\n      },\n    };\n  },\n);\nexport type ThreadMessageClientProps = {\n  message: ThreadMessage;\n  index: number;\n  isLast?: boolean;\n  branchNumber?: number;\n  branchCount?: number;\n};\n\nexport const ThreadMessageClient = resource(\n  ({\n    message,\n    index,\n    isLast = true,\n    branchNumber = 1,\n    branchCount = 1,\n  }: ThreadMessageClientProps): ClientOutput<\"message\"> => {\n    const [isCopiedState, setIsCopied] = tapState(false);\n    const [isHoveringState, setIsHovering] = tapState(false);\n\n    const parts = tapClientLookup(\n      () =>\n        message.content.map((part, idx) =>\n          withKey(\n            \"toolCallId\" in part && part.toolCallId != null\n              ? `toolCallId-${part.toolCallId}`\n              : `index-${idx}`,\n            ThreadMessagePartClient({ part }),\n          ),\n        ),\n      [message.content],\n    );\n\n    const attachments = tapClientLookup(\n      () =>\n        (message.attachments ?? []).map((attachment) =>\n          withKey(attachment.id, ThreadMessageAttachmentClient({ attachment })),\n        ),\n      [message.attachments],\n    );\n\n    const composer = tapResource(NoOpComposerClient({ type: \"edit\" }));\n    const composerState = composer.getState();\n\n    const state = tapMemo<MessageState>(() => {\n      return {\n        ...message,\n        parts: parts.state,\n        composer: composerState,\n        parentId: null,\n        index,\n        isLast,\n        branchNumber,\n        branchCount,\n        speech: undefined,\n        isCopied: isCopiedState,\n        isHovering: isHoveringState,\n      };\n    }, [\n      message,\n      index,\n      isCopiedState,\n      isHoveringState,\n      isLast,\n      parts.state,\n      composerState,\n      branchNumber,\n      branchCount,\n    ]);\n\n    return {\n      getState: () => state,\n      composer: () => composer,\n      part: (selector) => {\n        if (\"index\" in selector) {\n          return parts.get({ index: selector.index });\n        } else {\n          return parts.get({ key: `toolCallId-${selector.toolCallId}` });\n        }\n      },\n      attachment: (selector) => {\n        if (\"id\" in selector) {\n          return attachments.get({ key: selector.id });\n        } else {\n          return attachments.get(selector);\n        }\n      },\n      reload: () => {\n        throw new Error(\"Not supported in ThreadMessageProvider\");\n      },\n      speak: () => {\n        throw new Error(\"Not supported in ThreadMessageProvider\");\n      },\n      stopSpeaking: () => {\n        throw new Error(\"Not supported in ThreadMessageProvider\");\n      },\n      submitFeedback: () => {\n        throw new Error(\"Not supported in ThreadMessageProvider\");\n      },\n      switchToBranch: () => {\n        throw new Error(\"Not supported in ThreadMessageProvider\");\n      },\n      getCopyText: () => {\n        return message.content\n          .map((part) => {\n            if (\"text\" in part && typeof part.text === \"string\") {\n              return part.text;\n            }\n            return \"\";\n          })\n          .join(\"\\n\");\n      },\n      setIsCopied,\n      setIsHovering,\n    };\n  },\n);\n"],"mappings":";;;;AAmBA,MAAM,0BAA0B,UAC7B,EACC,WAG0B;CAC1B,MAAM,QAAQ,cAAyB;EACrC,OAAO;GACL,GAAG;GACH,QAAQ,EAAE,MAAM,WAAW;EAC7B;CACF,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;EACL,gBAAgB;EAChB,qBAAqB;GACnB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,sBAAsB;GACpB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,6BAA6B;GAC3B,MAAM,IAAI,MAAM,eAAe;EACjC;CACF;AACF,CACF;AAEA,MAAM,gCAAgC,UACnC,EAAE,iBAAyE;CAC1E,OAAO;EACL,gBAAgB;EAChB,cAAc;GACZ,MAAM,IAAI,MAAM,eAAe;EACjC;CACF;AACF,CACF;AASA,MAAa,sBAAsB,UAChC,EACC,SACA,OACA,SAAS,MACT,eAAe,GACf,cAAc,QACyC;CACvD,MAAM,CAAC,eAAe,eAAe,SAAS,KAAK;CACnD,MAAM,CAAC,iBAAiB,iBAAiB,SAAS,KAAK;CAEvD,MAAM,QAAQ,sBAEV,QAAQ,QAAQ,KAAK,MAAM,QACzB,QACE,gBAAgB,QAAQ,KAAK,cAAc,OACvC,cAAc,KAAK,eACnB,SAAS,OACb,wBAAwB,EAAE,KAAK,CAAC,CAClC,CACF,GACF,CAAC,QAAQ,OAAO,CAClB;CAEA,MAAM,cAAc,uBAEf,QAAQ,eAAe,CAAC,GAAG,KAAK,eAC/B,QAAQ,WAAW,IAAI,8BAA8B,EAAE,WAAW,CAAC,CAAC,CACtE,GACF,CAAC,QAAQ,WAAW,CACtB;CAEA,MAAM,WAAW,YAAY,mBAAmB,EAAE,MAAM,OAAO,CAAC,CAAC;CACjE,MAAM,gBAAgB,SAAS,SAAS;CAExC,MAAM,QAAQ,cAA4B;EACxC,OAAO;GACL,GAAG;GACH,OAAO,MAAM;GACb,UAAU;GACV,UAAU;GACV;GACA;GACA;GACA;GACA,QAAQ,KAAA;GACR,UAAU;GACV,YAAY;EACd;CACF,GAAG;EACD;EACA;EACA;EACA;EACA;EACA,MAAM;EACN;EACA;EACA;CACF,CAAC;CAED,OAAO;EACL,gBAAgB;EAChB,gBAAgB;EAChB,OAAO,aAAa;GAClB,IAAI,WAAW,UACb,OAAO,MAAM,IAAI,EAAE,OAAO,SAAS,MAAM,CAAC;QAE1C,OAAO,MAAM,IAAI,EAAE,KAAK,cAAc,SAAS,aAAa,CAAC;EAEjE;EACA,aAAa,aAAa;GACxB,IAAI,QAAQ,UACV,OAAO,YAAY,IAAI,EAAE,KAAK,SAAS,GAAG,CAAC;QAE3C,OAAO,YAAY,IAAI,QAAQ;EAEnC;EACA,cAAc;GACZ,MAAM,IAAI,MAAM,wCAAwC;EAC1D;EACA,aAAa;GACX,MAAM,IAAI,MAAM,wCAAwC;EAC1D;EACA,oBAAoB;GAClB,MAAM,IAAI,MAAM,wCAAwC;EAC1D;EACA,sBAAsB;GACpB,MAAM,IAAI,MAAM,wCAAwC;EAC1D;EACA,sBAAsB;GACpB,MAAM,IAAI,MAAM,wCAAwC;EAC1D;EACA,mBAAmB;GACjB,OAAO,QAAQ,QACZ,KAAK,SAAS;IACb,IAAI,UAAU,QAAQ,OAAO,KAAK,SAAS,UACzC,OAAO,KAAK;IAEd,OAAO;GACT,CAAC,EACA,KAAK,IAAI;EACd;EACA;EACA;CACF;AACF,CACF"}
+{"version":3,"file":"thread-message-client.js","names":[],"sources":["../../../src/store/clients/thread-message-client.ts"],"sourcesContent":["import type {\n  ThreadAssistantMessagePart,\n  ThreadUserMessagePart,\n  ThreadMessage,\n} from \"../../types/message\";\nimport type { Attachment } from \"../../types/attachment\";\nimport { useMemo, useState } from \"react\";\nimport { useResource, resource, withKey } from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport { useClientLookup } from \"@assistant-ui/store\";\nimport type { MessageState } from \"../scopes/message\";\nimport type { PartState } from \"../scopes/part\";\nimport { NoOpComposerClient } from \"./no-op-composer-client\";\n\nconst ThreadMessagePartClient = resource(function ThreadMessagePartClient({\n  part,\n}: {\n  part: ThreadAssistantMessagePart | ThreadUserMessagePart;\n}): ClientOutput<\"part\"> {\n  const state = useMemo<PartState>(() => {\n    return {\n      ...part,\n      status: { type: \"complete\" },\n    };\n  }, [part]);\n\n  return {\n    getState: () => state,\n    addToolResult: () => {\n      throw new Error(\"Not supported\");\n    },\n    resumeToolCall: () => {\n      throw new Error(\"Not supported\");\n    },\n    respondToToolApproval: () => {\n      throw new Error(\"Not supported\");\n    },\n  };\n});\n\nconst ThreadMessageAttachmentClient = resource(\n  function ThreadMessageAttachmentClient({\n    attachment,\n  }: {\n    attachment: Attachment;\n  }): ClientOutput<\"attachment\"> {\n    return {\n      getState: () => attachment,\n      remove: () => {\n        throw new Error(\"Not supported\");\n      },\n    };\n  },\n);\nexport type ThreadMessageClientProps = {\n  message: ThreadMessage;\n  index: number;\n  isLast?: boolean;\n  branchNumber?: number;\n  branchCount?: number;\n};\n\nexport const ThreadMessageClient = resource(function ThreadMessageClient({\n  message,\n  index,\n  isLast = true,\n  branchNumber = 1,\n  branchCount = 1,\n}: ThreadMessageClientProps): ClientOutput<\"message\"> {\n  const [isCopiedState, setIsCopied] = useState(false);\n  const [isHoveringState, setIsHovering] = useState(false);\n\n  const parts = useClientLookup(\n    () =>\n      message.content.map((part, idx) =>\n        withKey(\n          \"toolCallId\" in part && part.toolCallId != null\n            ? `toolCallId-${part.toolCallId}`\n            : `index-${idx}`,\n          ThreadMessagePartClient({ part }),\n        ),\n      ),\n    [message.content],\n  );\n\n  const attachments = useClientLookup(\n    () =>\n      (message.attachments ?? []).map((attachment) =>\n        withKey(attachment.id, ThreadMessageAttachmentClient({ attachment })),\n      ),\n    [message.attachments],\n  );\n\n  const composer = useResource(NoOpComposerClient({ type: \"edit\" }));\n  const composerState = composer.getState();\n\n  const state = useMemo<MessageState>(() => {\n    return {\n      ...message,\n      parts: parts.state,\n      composer: composerState,\n      parentId: null,\n      index,\n      isLast,\n      branchNumber,\n      branchCount,\n      speech: undefined,\n      isCopied: isCopiedState,\n      isHovering: isHoveringState,\n    };\n  }, [\n    message,\n    index,\n    isCopiedState,\n    isHoveringState,\n    isLast,\n    parts.state,\n    composerState,\n    branchNumber,\n    branchCount,\n  ]);\n\n  return {\n    getState: () => state,\n    composer: () => composer,\n    part: (selector) => {\n      if (\"index\" in selector) {\n        return parts.get({ index: selector.index });\n      } else {\n        return parts.get({ key: `toolCallId-${selector.toolCallId}` });\n      }\n    },\n    attachment: (selector) => {\n      if (\"id\" in selector) {\n        return attachments.get({ key: selector.id });\n      } else {\n        return attachments.get(selector);\n      }\n    },\n    reload: () => {\n      throw new Error(\"Not supported in ThreadMessageProvider\");\n    },\n    speak: () => {\n      throw new Error(\"Not supported in ThreadMessageProvider\");\n    },\n    stopSpeaking: () => {\n      throw new Error(\"Not supported in ThreadMessageProvider\");\n    },\n    submitFeedback: () => {\n      throw new Error(\"Not supported in ThreadMessageProvider\");\n    },\n    switchToBranch: () => {\n      throw new Error(\"Not supported in ThreadMessageProvider\");\n    },\n    getCopyText: () => {\n      return message.content\n        .map((part) => {\n          if (\"text\" in part && typeof part.text === \"string\") {\n            return part.text;\n          }\n          return \"\";\n        })\n        .join(\"\\n\");\n    },\n    setIsCopied,\n    setIsHovering,\n  };\n});\n"],"mappings":";;;;;AAcA,MAAM,0BAA0B,SAAS,SAAS,wBAAwB,EACxE,QAGuB;CACvB,MAAM,QAAQ,cAAyB;EACrC,OAAO;GACL,GAAG;GACH,QAAQ,EAAE,MAAM,WAAW;EAC7B;CACF,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;EACL,gBAAgB;EAChB,qBAAqB;GACnB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,sBAAsB;GACpB,MAAM,IAAI,MAAM,eAAe;EACjC;EACA,6BAA6B;GAC3B,MAAM,IAAI,MAAM,eAAe;EACjC;CACF;AACF,CAAC;AAED,MAAM,gCAAgC,SACpC,SAAS,8BAA8B,EACrC,cAG6B;CAC7B,OAAO;EACL,gBAAgB;EAChB,cAAc;GACZ,MAAM,IAAI,MAAM,eAAe;EACjC;CACF;AACF,CACF;AASA,MAAa,sBAAsB,SAAS,SAAS,oBAAoB,EACvE,SACA,OACA,SAAS,MACT,eAAe,GACf,cAAc,KACsC;CACpD,MAAM,CAAC,eAAe,eAAe,SAAS,KAAK;CACnD,MAAM,CAAC,iBAAiB,iBAAiB,SAAS,KAAK;CAEvD,MAAM,QAAQ,sBAEV,QAAQ,QAAQ,KAAK,MAAM,QACzB,QACE,gBAAgB,QAAQ,KAAK,cAAc,OACvC,cAAc,KAAK,eACnB,SAAS,OACb,wBAAwB,EAAE,KAAK,CAAC,CAClC,CACF,GACF,CAAC,QAAQ,OAAO,CAClB;CAEA,MAAM,cAAc,uBAEf,QAAQ,eAAe,CAAC,EAAA,CAAG,KAAK,eAC/B,QAAQ,WAAW,IAAI,8BAA8B,EAAE,WAAW,CAAC,CAAC,CACtE,GACF,CAAC,QAAQ,WAAW,CACtB;CAEA,MAAM,WAAW,YAAY,mBAAmB,EAAE,MAAM,OAAO,CAAC,CAAC;CACjE,MAAM,gBAAgB,SAAS,SAAS;CAExC,MAAM,QAAQ,cAA4B;EACxC,OAAO;GACL,GAAG;GACH,OAAO,MAAM;GACb,UAAU;GACV,UAAU;GACV;GACA;GACA;GACA;GACA,QAAQ,KAAA;GACR,UAAU;GACV,YAAY;EACd;CACF,GAAG;EACD;EACA;EACA;EACA;EACA;EACA,MAAM;EACN;EACA;EACA;CACF,CAAC;CAED,OAAO;EACL,gBAAgB;EAChB,gBAAgB;EAChB,OAAO,aAAa;GAClB,IAAI,WAAW,UACb,OAAO,MAAM,IAAI,EAAE,OAAO,SAAS,MAAM,CAAC;QAE1C,OAAO,MAAM,IAAI,EAAE,KAAK,cAAc,SAAS,aAAa,CAAC;EAEjE;EACA,aAAa,aAAa;GACxB,IAAI,QAAQ,UACV,OAAO,YAAY,IAAI,EAAE,KAAK,SAAS,GAAG,CAAC;QAE3C,OAAO,YAAY,IAAI,QAAQ;EAEnC;EACA,cAAc;GACZ,MAAM,IAAI,MAAM,wCAAwC;EAC1D;EACA,aAAa;GACX,MAAM,IAAI,MAAM,wCAAwC;EAC1D;EACA,oBAAoB;GAClB,MAAM,IAAI,MAAM,wCAAwC;EAC1D;EACA,sBAAsB;GACpB,MAAM,IAAI,MAAM,wCAAwC;EAC1D;EACA,sBAAsB;GACpB,MAAM,IAAI,MAAM,wCAAwC;EAC1D;EACA,mBAAmB;GACjB,OAAO,QAAQ,QACZ,KAAK,SAAS;IACb,IAAI,UAAU,QAAQ,OAAO,KAAK,SAAS,UACzC,OAAO,KAAK;IAEd,OAAO;GACT,CAAC,CAAC,CACD,KAAK,IAAI;EACd;EACA;EACA;CACF;AACF,CAAC"}

package/dist/store/index.d.ts

@@ -1,7 +1,7 @@
 /// <reference path="./scope-registration.d.ts" />
+import { QueueItemClientSchema, QueueItemMeta, QueueItemMethods, QueueItemState } from "./scopes/queue-item.js";
 import { ThreadListItemClientSchema, ThreadListItemEvents, ThreadListItemMeta, ThreadListItemMethods, ThreadListItemState } from "./scopes/thread-list-item.js";
 import { AttachmentClientSchema, AttachmentMeta, AttachmentMethods, AttachmentState } from "./scopes/attachment.js";
-import { QueueItemClientSchema, QueueItemMeta, QueueItemMethods, QueueItemState } from "./scopes/queue-item.js";
 import { ComposerClientSchema, ComposerEvents, ComposerMeta, ComposerMethods, ComposerSendOptions, ComposerState } from "./scopes/composer.js";
 import { PartClientSchema, PartMeta, PartMethods, PartState } from "./scopes/part.js";
 import { MessageClientSchema, MessageMeta, MessageMethods, MessageState } from "./scopes/message.js";

package/dist/store/runtime-clients/attachment-runtime-client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"attachment-runtime-client.d.ts","names":[],"sources":["../../../src/store/runtime-clients/attachment-runtime-client.ts"],"mappings":";;;;cAKa,uBAAA,GAAuB,KAAA;WACT,iBAAA;AAAA;WAAA,iBAAA;AAAA"}
+{"version":3,"file":"attachment-runtime-client.d.ts","names":[],"sources":["../../../src/store/runtime-clients/attachment-runtime-client.ts"],"mappings":";;;;cAKa,uBAAA,GAAuB,KAAA;WAIvB,iBAAA;AAAA;WAAA,iBAAA;AAAA"}

package/dist/store/runtime-clients/attachment-runtime-client.js

@@ -1,8 +1,8 @@
-import { tapSubscribable } from "./tap-subscribable.js";
+import { useSubscribable } from "./useSubscribable.js";
 import { resource } from "@assistant-ui/tap";
 //#region src/store/runtime-clients/attachment-runtime-client.ts
-const AttachmentRuntimeClient = resource(({ runtime }) => {
-	const state = tapSubscribable(runtime);
+const AttachmentRuntimeClient = resource(function AttachmentRuntimeClient({ runtime }) {
+	const state = useSubscribable(runtime);
 	return {
 		getState: () => state,
 		remove: runtime.remove,

package/dist/store/runtime-clients/attachment-runtime-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"attachment-runtime-client.js","names":[],"sources":["../../../src/store/runtime-clients/attachment-runtime-client.ts"],"sourcesContent":["import { resource } from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport type { AttachmentRuntime } from \"../../runtime/api/attachment-runtime\";\nimport { tapSubscribable } from \"./tap-subscribable\";\n\nexport const AttachmentRuntimeClient = resource(\n  ({ runtime }: { runtime: AttachmentRuntime }): ClientOutput<\"attachment\"> => {\n    const state = tapSubscribable(runtime);\n\n    return {\n      getState: () => state,\n      remove: runtime.remove,\n      __internal_getRuntime: () => runtime,\n    };\n  },\n);\n"],"mappings":";;;AAKA,MAAa,0BAA0B,UACpC,EAAE,cAA0E;CAC3E,MAAM,QAAQ,gBAAgB,OAAO;CAErC,OAAO;EACL,gBAAgB;EAChB,QAAQ,QAAQ;EAChB,6BAA6B;CAC/B;AACF,CACF"}
+{"version":3,"file":"attachment-runtime-client.js","names":[],"sources":["../../../src/store/runtime-clients/attachment-runtime-client.ts"],"sourcesContent":["import { resource } from \"@assistant-ui/tap\";\nimport type { ClientOutput } from \"@assistant-ui/store\";\nimport type { AttachmentRuntime } from \"../../runtime/api/attachment-runtime\";\nimport { useSubscribable } from \"./useSubscribable\";\n\nexport const AttachmentRuntimeClient = resource(\n  function AttachmentRuntimeClient({\n    runtime,\n  }: {\n    runtime: AttachmentRuntime;\n  }): ClientOutput<\"attachment\"> {\n    const state = useSubscribable(runtime);\n\n    return {\n      getState: () => state,\n      remove: runtime.remove,\n      __internal_getRuntime: () => runtime,\n    };\n  },\n);\n"],"mappings":";;;AAKA,MAAa,0BAA0B,SACrC,SAAS,wBAAwB,EAC/B,WAG6B;CAC7B,MAAM,QAAQ,gBAAgB,OAAO;CAErC,OAAO;EACL,gBAAgB;EAChB,QAAQ,QAAQ;EAChB,6BAA6B;CAC/B;AACF,CACF"}

package/dist/store/runtime-clients/composer-runtime-client.d.ts

@@ -1,15 +1,22 @@
 import { ComposerRuntime } from "../../runtime/api/composer-runtime.js";
 import { ClientOutput } from "@assistant-ui/store";
-import { tapRef } from "@assistant-ui/tap";
 
 //#region src/store/runtime-clients/composer-runtime-client.d.ts
 declare const ComposerClient: (props: {
-  threadIdRef: tapRef.RefObject<string>;
-  messageIdRef?: tapRef.RefObject<string>;
+  threadIdRef: {
+    current: string;
+  };
+  messageIdRef?: {
+    current: string;
+  };
   runtime: ComposerRuntime;
 }) => import("@assistant-ui/tap").ResourceElement<ClientOutput<"composer">, {
-  threadIdRef: tapRef.RefObject<string>;
-  messageIdRef?: tapRef.RefObject<string>;
+  threadIdRef: {
+    current: string;
+  };
+  messageIdRef?: {
+    current: string;
+  };
   runtime: ComposerRuntime;
 }>;
 //#endregion

package/dist/store/runtime-clients/composer-runtime-client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"composer-runtime-client.d.ts","names":[],"sources":["../../../src/store/runtime-clients/composer-runtime-client.ts"],"mappings":";;;;;cAqCa,cAAA,GAAc,KAAA;eAMV,MAAA,CAAO,SAAA;iBACL,MAAA,CAAO,SAAA;WACb,eAAA;AAAA;eAFI,MAAA,CAAO,SAAA;iBACL,MAAA,CAAO,SAAA;WACb,eAAA;AAAA"}
+{"version":3,"file":"composer-runtime-client.d.ts","names":[],"sources":["../../../src/store/runtime-clients/composer-runtime-client.ts"],"mappings":";;;;cAsDa,cAAA,GAAc,KAAA;;IAKV,OAAA;EAAA;;IACE,OAAA;EAAA;WACR,eAAA;AAAA;;IAFM,OAAA;EAAA;;IACE,OAAA;EAAA;WACR,eAAA;AAAA"}