@assistant-ui/core 0.2.9 → 0.2.10

package/dist/model-context/tool.d.ts

@@ -6,7 +6,7 @@ import { Tool } from "assistant-stream";
  * optional model-output conversion.
  *
  * This helper keeps reusable tool definitions type-checked and convenient to
- * export for a {@link Toolkit}, {@link Tools}, or {@link useAssistantTool}.
+ * export for a {@link Toolkit} registered with {@link Tools}.
  * Inference from parameter schemas is currently limited, so provide generic
  * arguments when you need precise args or result types.
  *

package/dist/model-context/tool.js

@@ -4,7 +4,7 @@
 * optional model-output conversion.
 *
 * This helper keeps reusable tool definitions type-checked and convenient to
-* export for a {@link Toolkit}, {@link Tools}, or {@link useAssistantTool}.
+* export for a {@link Toolkit} registered with {@link Tools}.
 * Inference from parameter schemas is currently limited, so provide generic
 * arguments when you need precise args or result types.
 *

package/dist/model-context/tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"tool.js","names":[],"sources":["../../src/model-context/tool.ts"],"sourcesContent":["import type { Tool } from \"assistant-stream\";\n\n// TODO re-add the inferrence of the parameters\n\n/**\n * Defines a model tool with its argument schema, execution behavior, and\n * optional model-output conversion.\n *\n * This helper keeps reusable tool definitions type-checked and convenient to\n * export for a {@link Toolkit}, {@link Tools}, or {@link useAssistantTool}.\n * Inference from parameter schemas is currently limited, so provide generic\n * arguments when you need precise args or result types.\n *\n * @param tool - Tool definition to expose to the assistant model.\n *\n * @example\n * ```ts\n * const getWeather = tool<{ city: string }, string>({\n *   type: \"frontend\",\n *   description: \"Get the weather for a city.\",\n *   parameters: {\n *     type: \"object\",\n *     properties: { city: { type: \"string\" } },\n *     required: [\"city\"],\n *   },\n *   execute: async ({ city }) => `Sunny in ${city}`,\n * });\n * ```\n */\nexport function tool<TArgs extends Record<string, unknown>, TResult = any>(\n  tool: Tool<TArgs, TResult>,\n): Tool<TArgs, TResult> {\n  return tool;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AA6BA,SAAgB,KACd,MACsB;CACtB,OAAO;AACT"}
+{"version":3,"file":"tool.js","names":[],"sources":["../../src/model-context/tool.ts"],"sourcesContent":["import type { Tool } from \"assistant-stream\";\n\n// TODO re-add the inferrence of the parameters\n\n/**\n * Defines a model tool with its argument schema, execution behavior, and\n * optional model-output conversion.\n *\n * This helper keeps reusable tool definitions type-checked and convenient to\n * export for a {@link Toolkit} registered with {@link Tools}.\n * Inference from parameter schemas is currently limited, so provide generic\n * arguments when you need precise args or result types.\n *\n * @param tool - Tool definition to expose to the assistant model.\n *\n * @example\n * ```ts\n * const getWeather = tool<{ city: string }, string>({\n *   type: \"frontend\",\n *   description: \"Get the weather for a city.\",\n *   parameters: {\n *     type: \"object\",\n *     properties: { city: { type: \"string\" } },\n *     required: [\"city\"],\n *   },\n *   execute: async ({ city }) => `Sunny in ${city}`,\n * });\n * ```\n */\nexport function tool<TArgs extends Record<string, unknown>, TResult = any>(\n  tool: Tool<TArgs, TResult>,\n): Tool<TArgs, TResult> {\n  return tool;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AA6BA,SAAgB,KACd,MACsB;CACtB,OAAO;AACT"}

package/dist/model-context/types.js

@@ -1,13 +1,28 @@
 //#region src/model-context/types.ts
 const mergeModelContexts = (configSet) => {
-	return Array.from(configSet).map((c) => c.getModelContext()).sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0)).reduce((acc, config) => {
+	const configs = Array.from(configSet).map((c) => c.getModelContext()).sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0));
+	const toolPriorities = {};
+	return configs.reduce((acc, config) => {
+		const priority = config.priority ?? 0;
 		if (config.system) if (acc.system) acc.system += `\n\n${config.system}`;
 		else acc.system = config.system;
 		if (config.tools) for (const [name, tool] of Object.entries(config.tools)) {
 			const existing = acc.tools?.[name];
-			if (existing && existing !== tool) throw new Error(`You tried to define a tool with the name ${name}, but it already exists.`);
+			if (existing && existing !== tool) {
+				const existingPriority = toolPriorities[name];
+				if (existingPriority === priority) throw new Error(`You tried to define a tool with the name ${name}, but it already exists.`);
+				const higherPriorityTool = existingPriority > priority ? existing : tool;
+				const lowerPriorityTool = existingPriority > priority ? tool : existing;
+				acc.tools[name] = {
+					...lowerPriorityTool,
+					...higherPriorityTool
+				};
+				toolPriorities[name] = Math.max(existingPriority, priority);
+				continue;
+			}
 			if (!acc.tools) acc.tools = {};
 			acc.tools[name] = tool;
+			toolPriorities[name] ??= priority;
 		}
 		if (config.config) acc.config = {
 			...acc.config,

package/dist/model-context/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","names":[],"sources":["../../src/model-context/types.ts"],"sourcesContent":["import type { Unsubscribe } from \"../types/unsubscribe\";\nimport type { Tool } from \"assistant-stream\";\n\nexport type LanguageModelV1CallSettings = {\n  maxTokens?: number;\n  temperature?: number;\n  topP?: number;\n  presencePenalty?: number;\n  frequencyPenalty?: number;\n  seed?: number;\n  headers?: Record<string, string | undefined>;\n};\n\nexport type LanguageModelConfig = {\n  apiKey?: string;\n  baseUrl?: string;\n  modelName?: string;\n};\n\nexport type ModelContext = {\n  priority?: number | undefined;\n  system?: string | undefined;\n  tools?: Record<string, Tool<any, any>> | undefined;\n  callSettings?: LanguageModelV1CallSettings | undefined;\n  config?: LanguageModelConfig | undefined;\n};\n\nexport type ModelContextProvider = {\n  getModelContext: () => ModelContext;\n  subscribe?: (callback: () => void) => Unsubscribe;\n};\n\nexport type AssistantToolProps<\n  TArgs extends Record<string, unknown>,\n  TResult,\n> = Tool<TArgs, TResult> & {\n  toolName: string;\n  render?: unknown;\n};\n\nexport type AssistantInstructionsConfig = {\n  disabled?: boolean | undefined;\n  instruction: string;\n};\n\nexport type AssistantContextConfig = {\n  getContext: () => string;\n  disabled?: boolean | undefined;\n};\n\nexport const mergeModelContexts = (\n  configSet: Set<ModelContextProvider>,\n): ModelContext => {\n  const configs = Array.from(configSet)\n    .map((c) => c.getModelContext())\n    .sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0));\n\n  return configs.reduce((acc, config) => {\n    if (config.system) {\n      if (acc.system) {\n        acc.system += `\\n\\n${config.system}`;\n      } else {\n        acc.system = config.system;\n      }\n    }\n    if (config.tools) {\n      for (const [name, tool] of Object.entries(config.tools)) {\n        const existing = acc.tools?.[name];\n        if (existing && existing !== tool) {\n          throw new Error(\n            `You tried to define a tool with the name ${name}, but it already exists.`,\n          );\n        }\n\n        if (!acc.tools) acc.tools = {};\n        acc.tools[name] = tool;\n      }\n    }\n    if (config.config) {\n      acc.config = {\n        ...acc.config,\n        ...config.config,\n      };\n    }\n    if (config.callSettings) {\n      acc.callSettings = {\n        ...acc.callSettings,\n        ...config.callSettings,\n      };\n    }\n    return acc;\n  }, {} as ModelContext);\n};\n"],"mappings":";AAkDA,MAAa,sBACX,cACiB;CAKjB,OAJgB,MAAM,KAAK,SAAS,EACjC,KAAK,MAAM,EAAE,gBAAgB,CAAC,EAC9B,MAAM,GAAG,OAAO,EAAE,YAAY,MAAM,EAAE,YAAY,EAExC,EAAE,QAAQ,KAAK,WAAW;EACrC,IAAI,OAAO,QACT,IAAI,IAAI,QACN,IAAI,UAAU,OAAO,OAAO;OAE5B,IAAI,SAAS,OAAO;EAGxB,IAAI,OAAO,OACT,KAAK,MAAM,CAAC,MAAM,SAAS,OAAO,QAAQ,OAAO,KAAK,GAAG;GACvD,MAAM,WAAW,IAAI,QAAQ;GAC7B,IAAI,YAAY,aAAa,MAC3B,MAAM,IAAI,MACR,4CAA4C,KAAK,yBACnD;GAGF,IAAI,CAAC,IAAI,OAAO,IAAI,QAAQ,CAAC;GAC7B,IAAI,MAAM,QAAQ;EACpB;EAEF,IAAI,OAAO,QACT,IAAI,SAAS;GACX,GAAG,IAAI;GACP,GAAG,OAAO;EACZ;EAEF,IAAI,OAAO,cACT,IAAI,eAAe;GACjB,GAAG,IAAI;GACP,GAAG,OAAO;EACZ;EAEF,OAAO;CACT,GAAG,CAAC,CAAiB;AACvB"}
+{"version":3,"file":"types.js","names":[],"sources":["../../src/model-context/types.ts"],"sourcesContent":["import type { Unsubscribe } from \"../types/unsubscribe\";\nimport type { Tool } from \"assistant-stream\";\n\nexport type LanguageModelV1CallSettings = {\n  maxTokens?: number;\n  temperature?: number;\n  topP?: number;\n  presencePenalty?: number;\n  frequencyPenalty?: number;\n  seed?: number;\n  headers?: Record<string, string | undefined>;\n};\n\nexport type LanguageModelConfig = {\n  apiKey?: string;\n  baseUrl?: string;\n  modelName?: string;\n};\n\nexport type ModelContext = {\n  priority?: number | undefined;\n  system?: string | undefined;\n  tools?: Record<string, Tool<any, any>> | undefined;\n  callSettings?: LanguageModelV1CallSettings | undefined;\n  config?: LanguageModelConfig | undefined;\n};\n\nexport type ModelContextProvider = {\n  getModelContext: () => ModelContext;\n  subscribe?: (callback: () => void) => Unsubscribe;\n};\n\nexport type AssistantToolProps<\n  TArgs extends Record<string, unknown>,\n  TResult,\n> = Tool<TArgs, TResult> & {\n  toolName: string;\n  render?: unknown;\n};\n\nexport type AssistantInstructionsConfig = {\n  disabled?: boolean | undefined;\n  instruction: string;\n};\n\nexport type AssistantContextConfig = {\n  getContext: () => string;\n  disabled?: boolean | undefined;\n};\n\nexport const mergeModelContexts = (\n  configSet: Set<ModelContextProvider>,\n): ModelContext => {\n  const configs = Array.from(configSet)\n    .map((c) => c.getModelContext())\n    .sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0));\n\n  const toolPriorities: Record<string, number> = {};\n\n  return configs.reduce((acc, config) => {\n    const priority = config.priority ?? 0;\n    if (config.system) {\n      if (acc.system) {\n        acc.system += `\\n\\n${config.system}`;\n      } else {\n        acc.system = config.system;\n      }\n    }\n    if (config.tools) {\n      for (const [name, tool] of Object.entries(config.tools)) {\n        const existing = acc.tools?.[name];\n        if (existing && existing !== tool) {\n          const existingPriority = toolPriorities[name]!;\n          if (existingPriority === priority) {\n            throw new Error(\n              `You tried to define a tool with the name ${name}, but it already exists.`,\n            );\n          }\n\n          const higherPriorityTool =\n            existingPriority > priority ? existing : tool;\n          const lowerPriorityTool =\n            existingPriority > priority ? tool : existing;\n          acc.tools![name] = {\n            ...lowerPriorityTool,\n            ...higherPriorityTool,\n          } as Tool<any, any>;\n          toolPriorities[name] = Math.max(existingPriority, priority);\n          continue;\n        }\n\n        if (!acc.tools) acc.tools = {};\n        acc.tools[name] = tool;\n        toolPriorities[name] ??= priority;\n      }\n    }\n    if (config.config) {\n      acc.config = {\n        ...acc.config,\n        ...config.config,\n      };\n    }\n    if (config.callSettings) {\n      acc.callSettings = {\n        ...acc.callSettings,\n        ...config.callSettings,\n      };\n    }\n    return acc;\n  }, {} as ModelContext);\n};\n"],"mappings":";AAkDA,MAAa,sBACX,cACiB;CACjB,MAAM,UAAU,MAAM,KAAK,SAAS,EACjC,KAAK,MAAM,EAAE,gBAAgB,CAAC,EAC9B,MAAM,GAAG,OAAO,EAAE,YAAY,MAAM,EAAE,YAAY,EAAE;CAEvD,MAAM,iBAAyC,CAAC;CAEhD,OAAO,QAAQ,QAAQ,KAAK,WAAW;EACrC,MAAM,WAAW,OAAO,YAAY;EACpC,IAAI,OAAO,QACT,IAAI,IAAI,QACN,IAAI,UAAU,OAAO,OAAO;OAE5B,IAAI,SAAS,OAAO;EAGxB,IAAI,OAAO,OACT,KAAK,MAAM,CAAC,MAAM,SAAS,OAAO,QAAQ,OAAO,KAAK,GAAG;GACvD,MAAM,WAAW,IAAI,QAAQ;GAC7B,IAAI,YAAY,aAAa,MAAM;IACjC,MAAM,mBAAmB,eAAe;IACxC,IAAI,qBAAqB,UACvB,MAAM,IAAI,MACR,4CAA4C,KAAK,yBACnD;IAGF,MAAM,qBACJ,mBAAmB,WAAW,WAAW;IAC3C,MAAM,oBACJ,mBAAmB,WAAW,OAAO;IACvC,IAAI,MAAO,QAAQ;KACjB,GAAG;KACH,GAAG;IACL;IACA,eAAe,QAAQ,KAAK,IAAI,kBAAkB,QAAQ;IAC1D;GACF;GAEA,IAAI,CAAC,IAAI,OAAO,IAAI,QAAQ,CAAC;GAC7B,IAAI,MAAM,QAAQ;GAClB,eAAe,UAAU;EAC3B;EAEF,IAAI,OAAO,QACT,IAAI,SAAS;GACX,GAAG,IAAI;GACP,GAAG,OAAO;EACZ;EAEF,IAAI,OAAO,cACT,IAAI,eAAe;GACjB,GAAG,IAAI;GACP,GAAG,OAAO;EACZ;EAEF,OAAO;CACT,GAAG,CAAC,CAAiB;AACvB"}

package/dist/react/adapters/LocalStorageThreadListAdapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"LocalStorageThreadListAdapter.d.ts","names":[],"sources":["../../../src/react/adapters/LocalStorageThreadListAdapter.tsx"],"mappings":";;;;KAkBY,gBAAA;EACV,OAAA,CAAQ,GAAA,WAAc,OAAA;EACtB,OAAA,CAAQ,GAAA,UAAa,KAAA,WAAgB,OAAA;EACrC,UAAA,CAAW,GAAA,WAAc,OAAA;AAAA;AAAA,KAGtB,0BAAA;EACH,OAAA,EAAS,gBAAA;EACT,MAAA;EACA,cAAA,GAAiB,sBAAsB;AAAA;AAAA,cA4E5B,yBAAA,GACX,OAAA,EAAS,0BAAA,KACR,uBA2HF"}
+{"version":3,"file":"LocalStorageThreadListAdapter.d.ts","names":[],"sources":["../../../src/react/adapters/LocalStorageThreadListAdapter.tsx"],"mappings":";;;;KAkBY,gBAAA;EACV,OAAA,CAAQ,GAAA,WAAc,OAAA;EACtB,OAAA,CAAQ,GAAA,UAAa,KAAA,WAAgB,OAAA;EACrC,UAAA,CAAW,GAAA,WAAc,OAAA;AAAA;AAAA,KAGtB,0BAAA;EACH,OAAA,EAAS,gBAAA;EACT,MAAA;EACA,cAAA,GAAiB,sBAAsB;AAAA;AAAA,cA2E5B,yBAAA,GACX,OAAA,EAAS,0BAAA,KACR,uBAyIF"}

package/dist/react/adapters/LocalStorageThreadListAdapter.js

@@ -64,7 +64,8 @@ const createLocalStorageAdapter = (options) => {
 				remoteId: t.remoteId,
 				externalId: t.externalId,
 				status: t.status,
-				title: t.title
+				title: t.title,
+				custom: t.custom
 			})) };
 		},
 		async initialize(threadId) {
@@ -90,6 +91,14 @@ const createLocalStorageAdapter = (options) => {
 				await saveThreadMetadata(threads);
 			}
 		},
+		async updateCustom(remoteId, custom) {
+			const threads = await loadThreadMetadata();
+			const thread = threads.find((t) => t.remoteId === remoteId);
+			if (thread) {
+				thread.custom = custom;
+				await saveThreadMetadata(threads);
+			}
+		},
 		async archive(remoteId) {
 			const threads = await loadThreadMetadata();
 			const thread = threads.find((t) => t.remoteId === remoteId);
@@ -117,7 +126,8 @@ const createLocalStorageAdapter = (options) => {
 				remoteId: thread.remoteId,
 				externalId: thread.externalId,
 				status: thread.status,
-				title: thread.title
+				title: thread.title,
+				custom: thread.custom
 			};
 		},
 		async generateTitle(remoteId, messages) {

package/dist/react/adapters/LocalStorageThreadListAdapter.js.map

@@ -1 +1 @@
-{"version":3,"file":"LocalStorageThreadListAdapter.js","names":[],"sources":["../../../src/react/adapters/LocalStorageThreadListAdapter.tsx"],"sourcesContent":["import { type AssistantStream, createAssistantStream } from \"assistant-stream\";\nimport { type FC, type PropsWithChildren, useMemo } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\nimport type {\n  RemoteThreadInitializeResponse,\n  RemoteThreadListAdapter,\n  RemoteThreadListResponse,\n  RemoteThreadMetadata,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n} from \"../../index\";\nimport type {\n  ExportedMessageRepository,\n  ExportedMessageRepositoryItem,\n} from \"../../internal\";\nimport { RuntimeAdapterProvider } from \"../runtimes/RuntimeAdapterProvider\";\nimport type { TitleGenerationAdapter } from \"./TitleGenerationAdapter\";\n\nexport type AsyncStorageLike = {\n  getItem(key: string): Promise<string | null>;\n  setItem(key: string, value: string): Promise<void>;\n  removeItem(key: string): Promise<void>;\n};\n\ntype LocalStorageAdapterOptions = {\n  storage: AsyncStorageLike;\n  prefix?: string | undefined;\n  titleGenerator?: TitleGenerationAdapter | undefined;\n};\n\n// `RemoteThreadMetadata.custom` is intentionally not persisted; consumers that\n// need it across reloads should fork this adapter or use a remote backend.\ntype StoredThreadMetadata = {\n  remoteId: string;\n  externalId?: string;\n  status: \"regular\" | \"archived\";\n  title?: string;\n};\n\nclass AsyncStorageHistoryAdapter implements ThreadHistoryAdapter {\n  constructor(\n    private storage: AsyncStorageLike,\n    private aui: ReturnType<typeof useAui>,\n    private prefix: string,\n  ) {}\n\n  private _messagesKey(remoteId: string) {\n    return `${this.prefix}messages:${remoteId}`;\n  }\n\n  async load(): Promise<ExportedMessageRepository> {\n    const remoteId = this.aui.threadListItem().getState().remoteId;\n    if (!remoteId) return { messages: [] };\n\n    const raw = await this.storage.getItem(this._messagesKey(remoteId));\n    if (!raw) return { messages: [] };\n    return JSON.parse(raw) as ExportedMessageRepository;\n  }\n\n  async append(item: ExportedMessageRepositoryItem): Promise<void> {\n    const { remoteId } = await this.aui.threadListItem().initialize();\n\n    const key = this._messagesKey(remoteId);\n    const raw = await this.storage.getItem(key);\n    const repo: ExportedMessageRepository = raw\n      ? (JSON.parse(raw) as ExportedMessageRepository)\n      : { messages: [] };\n\n    const idx = repo.messages.findIndex(\n      (m) => m.message.id === item.message.id,\n    );\n    if (idx >= 0) {\n      repo.messages[idx] = item;\n    } else {\n      repo.messages.push(item);\n    }\n    repo.headId = item.message.id;\n\n    await this.storage.setItem(key, JSON.stringify(repo));\n  }\n}\n\nconst createHistoryProvider = (\n  storage: AsyncStorageLike,\n  prefix: string,\n): FC<PropsWithChildren> => {\n  const Provider: FC<PropsWithChildren> = ({ children }) => {\n    const aui = useAui();\n    const history = useMemo(\n      () => new AsyncStorageHistoryAdapter(storage, aui, prefix),\n      [aui],\n    );\n    const adapters = useMemo(() => ({ history }), [history]);\n\n    return (\n      <RuntimeAdapterProvider adapters={adapters}>\n        {children}\n      </RuntimeAdapterProvider>\n    );\n  };\n  return Provider;\n};\n\nexport const createLocalStorageAdapter = (\n  options: LocalStorageAdapterOptions,\n): RemoteThreadListAdapter => {\n  const { storage, prefix = \"@assistant-ui:\", titleGenerator } = options;\n\n  const threadsKey = `${prefix}threads`;\n  const messagesKey = (threadId: string) => `${prefix}messages:${threadId}`;\n\n  const loadThreadMetadata = async (): Promise<StoredThreadMetadata[]> => {\n    const raw = await storage.getItem(threadsKey);\n    return raw ? (JSON.parse(raw) as StoredThreadMetadata[]) : [];\n  };\n\n  const saveThreadMetadata = async (\n    threads: StoredThreadMetadata[],\n  ): Promise<void> => {\n    await storage.setItem(threadsKey, JSON.stringify(threads));\n  };\n\n  const adapter: RemoteThreadListAdapter = {\n    unstable_Provider: createHistoryProvider(storage, prefix),\n\n    async list(): Promise<RemoteThreadListResponse> {\n      const threads = await loadThreadMetadata();\n      return {\n        threads: threads.map((t) => ({\n          remoteId: t.remoteId,\n          externalId: t.externalId,\n          status: t.status,\n          title: t.title,\n        })),\n      };\n    },\n\n    async initialize(\n      threadId: string,\n    ): Promise<RemoteThreadInitializeResponse> {\n      const remoteId = threadId;\n      const threads = await loadThreadMetadata();\n\n      // Only add if not already present\n      if (!threads.some((t) => t.remoteId === remoteId)) {\n        threads.unshift({\n          remoteId,\n          status: \"regular\",\n        });\n        await saveThreadMetadata(threads);\n      }\n\n      return { remoteId, externalId: undefined };\n    },\n\n    async rename(remoteId: string, newTitle: string): Promise<void> {\n      const threads = await loadThreadMetadata();\n      const thread = threads.find((t) => t.remoteId === remoteId);\n      if (thread) {\n        thread.title = newTitle;\n        await saveThreadMetadata(threads);\n      }\n    },\n\n    async archive(remoteId: string): Promise<void> {\n      const threads = await loadThreadMetadata();\n      const thread = threads.find((t) => t.remoteId === remoteId);\n      if (thread) {\n        thread.status = \"archived\";\n        await saveThreadMetadata(threads);\n      }\n    },\n\n    async unarchive(remoteId: string): Promise<void> {\n      const threads = await loadThreadMetadata();\n      const thread = threads.find((t) => t.remoteId === remoteId);\n      if (thread) {\n        thread.status = \"regular\";\n        await saveThreadMetadata(threads);\n      }\n    },\n\n    async delete(remoteId: string): Promise<void> {\n      const threads = await loadThreadMetadata();\n      const filtered = threads.filter((t) => t.remoteId !== remoteId);\n      await saveThreadMetadata(filtered);\n      await storage.removeItem(messagesKey(remoteId));\n    },\n\n    async fetch(threadId: string): Promise<RemoteThreadMetadata> {\n      const threads = await loadThreadMetadata();\n      const thread = threads.find((t) => t.remoteId === threadId);\n      if (!thread) throw new Error(\"Thread not found\");\n      return {\n        remoteId: thread.remoteId,\n        externalId: thread.externalId,\n        status: thread.status,\n        title: thread.title,\n      };\n    },\n\n    async generateTitle(\n      remoteId: string,\n      messages: readonly ThreadMessage[],\n    ): Promise<AssistantStream> {\n      if (titleGenerator) {\n        const title = await titleGenerator.generateTitle(messages);\n\n        // Update the stored title\n        const threads = await loadThreadMetadata();\n        const thread = threads.find((t) => t.remoteId === remoteId);\n        if (thread) {\n          thread.title = title;\n          await saveThreadMetadata(threads);\n        }\n\n        // Return a stream with a single text part\n        return createAssistantStream((controller) => {\n          controller.appendText(title);\n        });\n      }\n\n      // No title generator — return empty stream\n      return createAssistantStream(() => {});\n    },\n  };\n\n  return adapter;\n};\n"],"mappings":";;;;;;AAuCA,IAAM,6BAAN,MAAiE;CAErD;CACA;CACA;CAHV,YACE,SACA,KACA,QACA;EAHQ,KAAA,UAAA;EACA,KAAA,MAAA;EACA,KAAA,SAAA;CACP;CAEH,aAAqB,UAAkB;EACrC,OAAO,GAAG,KAAK,OAAO,WAAW;CACnC;CAEA,MAAM,OAA2C;EAC/C,MAAM,WAAW,KAAK,IAAI,eAAe,EAAE,SAAS,EAAE;EACtD,IAAI,CAAC,UAAU,OAAO,EAAE,UAAU,CAAC,EAAE;EAErC,MAAM,MAAM,MAAM,KAAK,QAAQ,QAAQ,KAAK,aAAa,QAAQ,CAAC;EAClE,IAAI,CAAC,KAAK,OAAO,EAAE,UAAU,CAAC,EAAE;EAChC,OAAO,KAAK,MAAM,GAAG;CACvB;CAEA,MAAM,OAAO,MAAoD;EAC/D,MAAM,EAAE,aAAa,MAAM,KAAK,IAAI,eAAe,EAAE,WAAW;EAEhE,MAAM,MAAM,KAAK,aAAa,QAAQ;EACtC,MAAM,MAAM,MAAM,KAAK,QAAQ,QAAQ,GAAG;EAC1C,MAAM,OAAkC,MACnC,KAAK,MAAM,GAAG,IACf,EAAE,UAAU,CAAC,EAAE;EAEnB,MAAM,MAAM,KAAK,SAAS,WACvB,MAAM,EAAE,QAAQ,OAAO,KAAK,QAAQ,EACvC;EACA,IAAI,OAAO,GACT,KAAK,SAAS,OAAO;OAErB,KAAK,SAAS,KAAK,IAAI;EAEzB,KAAK,SAAS,KAAK,QAAQ;EAE3B,MAAM,KAAK,QAAQ,QAAQ,KAAK,KAAK,UAAU,IAAI,CAAC;CACtD;AACF;AAEA,MAAM,yBACJ,SACA,WAC0B;CAC1B,MAAM,YAAmC,EAAE,eAAe;EACxD,MAAM,MAAM,OAAO;EACnB,MAAM,UAAU,cACR,IAAI,2BAA2B,SAAS,KAAK,MAAM,GACzD,CAAC,GAAG,CACN;EAGA,OACE,oBAAC,wBAAD;GAAwB,UAHT,eAAe,EAAE,QAAQ,IAAI,CAAC,OAAO,CAGX;GACtC;EACqB,CAAA;CAE5B;CACA,OAAO;AACT;AAEA,MAAa,6BACX,YAC4B;CAC5B,MAAM,EAAE,SAAS,SAAS,kBAAkB,mBAAmB;CAE/D,MAAM,aAAa,GAAG,OAAO;CAC7B,MAAM,eAAe,aAAqB,GAAG,OAAO,WAAW;CAE/D,MAAM,qBAAqB,YAA6C;EACtE,MAAM,MAAM,MAAM,QAAQ,QAAQ,UAAU;EAC5C,OAAO,MAAO,KAAK,MAAM,GAAG,IAA+B,CAAC;CAC9D;CAEA,MAAM,qBAAqB,OACzB,YACkB;EAClB,MAAM,QAAQ,QAAQ,YAAY,KAAK,UAAU,OAAO,CAAC;CAC3D;CA2GA,OAAO;EAxGL,mBAAmB,sBAAsB,SAAS,MAAM;EAExD,MAAM,OAA0C;GAE9C,OAAO,EACL,UAAS,MAFW,mBAAmB,GAEtB,KAAK,OAAO;IAC3B,UAAU,EAAE;IACZ,YAAY,EAAE;IACd,QAAQ,EAAE;IACV,OAAO,EAAE;GACX,EAAE,EACJ;EACF;EAEA,MAAM,WACJ,UACyC;GACzC,MAAM,WAAW;GACjB,MAAM,UAAU,MAAM,mBAAmB;GAGzC,IAAI,CAAC,QAAQ,MAAM,MAAM,EAAE,aAAa,QAAQ,GAAG;IACjD,QAAQ,QAAQ;KACd;KACA,QAAQ;IACV,CAAC;IACD,MAAM,mBAAmB,OAAO;GAClC;GAEA,OAAO;IAAE;IAAU,YAAY,KAAA;GAAU;EAC3C;EAEA,MAAM,OAAO,UAAkB,UAAiC;GAC9D,MAAM,UAAU,MAAM,mBAAmB;GACzC,MAAM,SAAS,QAAQ,MAAM,MAAM,EAAE,aAAa,QAAQ;GAC1D,IAAI,QAAQ;IACV,OAAO,QAAQ;IACf,MAAM,mBAAmB,OAAO;GAClC;EACF;EAEA,MAAM,QAAQ,UAAiC;GAC7C,MAAM,UAAU,MAAM,mBAAmB;GACzC,MAAM,SAAS,QAAQ,MAAM,MAAM,EAAE,aAAa,QAAQ;GAC1D,IAAI,QAAQ;IACV,OAAO,SAAS;IAChB,MAAM,mBAAmB,OAAO;GAClC;EACF;EAEA,MAAM,UAAU,UAAiC;GAC/C,MAAM,UAAU,MAAM,mBAAmB;GACzC,MAAM,SAAS,QAAQ,MAAM,MAAM,EAAE,aAAa,QAAQ;GAC1D,IAAI,QAAQ;IACV,OAAO,SAAS;IAChB,MAAM,mBAAmB,OAAO;GAClC;EACF;EAEA,MAAM,OAAO,UAAiC;GAG5C,MAAM,oBADW,MADK,mBAAmB,GAChB,QAAQ,MAAM,EAAE,aAAa,QACtB,CAAC;GACjC,MAAM,QAAQ,WAAW,YAAY,QAAQ,CAAC;EAChD;EAEA,MAAM,MAAM,UAAiD;GAE3D,MAAM,UAAS,MADO,mBAAmB,GAClB,MAAM,MAAM,EAAE,aAAa,QAAQ;GAC1D,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,kBAAkB;GAC/C,OAAO;IACL,UAAU,OAAO;IACjB,YAAY,OAAO;IACnB,QAAQ,OAAO;IACf,OAAO,OAAO;GAChB;EACF;EAEA,MAAM,cACJ,UACA,UAC0B;GAC1B,IAAI,gBAAgB;IAClB,MAAM,QAAQ,MAAM,eAAe,cAAc,QAAQ;IAGzD,MAAM,UAAU,MAAM,mBAAmB;IACzC,MAAM,SAAS,QAAQ,MAAM,MAAM,EAAE,aAAa,QAAQ;IAC1D,IAAI,QAAQ;KACV,OAAO,QAAQ;KACf,MAAM,mBAAmB,OAAO;IAClC;IAGA,OAAO,uBAAuB,eAAe;KAC3C,WAAW,WAAW,KAAK;IAC7B,CAAC;GACH;GAGA,OAAO,4BAA4B,CAAC,CAAC;EACvC;CAGW;AACf"}
+{"version":3,"file":"LocalStorageThreadListAdapter.js","names":[],"sources":["../../../src/react/adapters/LocalStorageThreadListAdapter.tsx"],"sourcesContent":["import { type AssistantStream, createAssistantStream } from \"assistant-stream\";\nimport { type FC, type PropsWithChildren, useMemo } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\nimport type {\n  RemoteThreadInitializeResponse,\n  RemoteThreadListAdapter,\n  RemoteThreadListResponse,\n  RemoteThreadMetadata,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n} from \"../../index\";\nimport type {\n  ExportedMessageRepository,\n  ExportedMessageRepositoryItem,\n} from \"../../internal\";\nimport { RuntimeAdapterProvider } from \"../runtimes/RuntimeAdapterProvider\";\nimport type { TitleGenerationAdapter } from \"./TitleGenerationAdapter\";\n\nexport type AsyncStorageLike = {\n  getItem(key: string): Promise<string | null>;\n  setItem(key: string, value: string): Promise<void>;\n  removeItem(key: string): Promise<void>;\n};\n\ntype LocalStorageAdapterOptions = {\n  storage: AsyncStorageLike;\n  prefix?: string | undefined;\n  titleGenerator?: TitleGenerationAdapter | undefined;\n};\n\ntype StoredThreadMetadata = {\n  remoteId: string;\n  externalId?: string;\n  status: \"regular\" | \"archived\";\n  title?: string;\n  custom?: Record<string, unknown> | undefined;\n};\n\nclass AsyncStorageHistoryAdapter implements ThreadHistoryAdapter {\n  constructor(\n    private storage: AsyncStorageLike,\n    private aui: ReturnType<typeof useAui>,\n    private prefix: string,\n  ) {}\n\n  private _messagesKey(remoteId: string) {\n    return `${this.prefix}messages:${remoteId}`;\n  }\n\n  async load(): Promise<ExportedMessageRepository> {\n    const remoteId = this.aui.threadListItem().getState().remoteId;\n    if (!remoteId) return { messages: [] };\n\n    const raw = await this.storage.getItem(this._messagesKey(remoteId));\n    if (!raw) return { messages: [] };\n    return JSON.parse(raw) as ExportedMessageRepository;\n  }\n\n  async append(item: ExportedMessageRepositoryItem): Promise<void> {\n    const { remoteId } = await this.aui.threadListItem().initialize();\n\n    const key = this._messagesKey(remoteId);\n    const raw = await this.storage.getItem(key);\n    const repo: ExportedMessageRepository = raw\n      ? (JSON.parse(raw) as ExportedMessageRepository)\n      : { messages: [] };\n\n    const idx = repo.messages.findIndex(\n      (m) => m.message.id === item.message.id,\n    );\n    if (idx >= 0) {\n      repo.messages[idx] = item;\n    } else {\n      repo.messages.push(item);\n    }\n    repo.headId = item.message.id;\n\n    await this.storage.setItem(key, JSON.stringify(repo));\n  }\n}\n\nconst createHistoryProvider = (\n  storage: AsyncStorageLike,\n  prefix: string,\n): FC<PropsWithChildren> => {\n  const Provider: FC<PropsWithChildren> = ({ children }) => {\n    const aui = useAui();\n    const history = useMemo(\n      () => new AsyncStorageHistoryAdapter(storage, aui, prefix),\n      [aui],\n    );\n    const adapters = useMemo(() => ({ history }), [history]);\n\n    return (\n      <RuntimeAdapterProvider adapters={adapters}>\n        {children}\n      </RuntimeAdapterProvider>\n    );\n  };\n  return Provider;\n};\n\nexport const createLocalStorageAdapter = (\n  options: LocalStorageAdapterOptions,\n): RemoteThreadListAdapter => {\n  const { storage, prefix = \"@assistant-ui:\", titleGenerator } = options;\n\n  const threadsKey = `${prefix}threads`;\n  const messagesKey = (threadId: string) => `${prefix}messages:${threadId}`;\n\n  const loadThreadMetadata = async (): Promise<StoredThreadMetadata[]> => {\n    const raw = await storage.getItem(threadsKey);\n    return raw ? (JSON.parse(raw) as StoredThreadMetadata[]) : [];\n  };\n\n  const saveThreadMetadata = async (\n    threads: StoredThreadMetadata[],\n  ): Promise<void> => {\n    await storage.setItem(threadsKey, JSON.stringify(threads));\n  };\n\n  const adapter: RemoteThreadListAdapter = {\n    unstable_Provider: createHistoryProvider(storage, prefix),\n\n    async list(): Promise<RemoteThreadListResponse> {\n      const threads = await loadThreadMetadata();\n      return {\n        threads: threads.map((t) => ({\n          remoteId: t.remoteId,\n          externalId: t.externalId,\n          status: t.status,\n          title: t.title,\n          custom: t.custom,\n        })),\n      };\n    },\n\n    async initialize(\n      threadId: string,\n    ): Promise<RemoteThreadInitializeResponse> {\n      const remoteId = threadId;\n      const threads = await loadThreadMetadata();\n\n      // Only add if not already present\n      if (!threads.some((t) => t.remoteId === remoteId)) {\n        threads.unshift({\n          remoteId,\n          status: \"regular\",\n        });\n        await saveThreadMetadata(threads);\n      }\n\n      return { remoteId, externalId: undefined };\n    },\n\n    async rename(remoteId: string, newTitle: string): Promise<void> {\n      const threads = await loadThreadMetadata();\n      const thread = threads.find((t) => t.remoteId === remoteId);\n      if (thread) {\n        thread.title = newTitle;\n        await saveThreadMetadata(threads);\n      }\n    },\n\n    async updateCustom(\n      remoteId: string,\n      custom: Record<string, unknown> | undefined,\n    ): Promise<void> {\n      const threads = await loadThreadMetadata();\n      const thread = threads.find((t) => t.remoteId === remoteId);\n      if (thread) {\n        thread.custom = custom;\n        await saveThreadMetadata(threads);\n      }\n    },\n\n    async archive(remoteId: string): Promise<void> {\n      const threads = await loadThreadMetadata();\n      const thread = threads.find((t) => t.remoteId === remoteId);\n      if (thread) {\n        thread.status = \"archived\";\n        await saveThreadMetadata(threads);\n      }\n    },\n\n    async unarchive(remoteId: string): Promise<void> {\n      const threads = await loadThreadMetadata();\n      const thread = threads.find((t) => t.remoteId === remoteId);\n      if (thread) {\n        thread.status = \"regular\";\n        await saveThreadMetadata(threads);\n      }\n    },\n\n    async delete(remoteId: string): Promise<void> {\n      const threads = await loadThreadMetadata();\n      const filtered = threads.filter((t) => t.remoteId !== remoteId);\n      await saveThreadMetadata(filtered);\n      await storage.removeItem(messagesKey(remoteId));\n    },\n\n    async fetch(threadId: string): Promise<RemoteThreadMetadata> {\n      const threads = await loadThreadMetadata();\n      const thread = threads.find((t) => t.remoteId === threadId);\n      if (!thread) throw new Error(\"Thread not found\");\n      return {\n        remoteId: thread.remoteId,\n        externalId: thread.externalId,\n        status: thread.status,\n        title: thread.title,\n        custom: thread.custom,\n      };\n    },\n\n    async generateTitle(\n      remoteId: string,\n      messages: readonly ThreadMessage[],\n    ): Promise<AssistantStream> {\n      if (titleGenerator) {\n        const title = await titleGenerator.generateTitle(messages);\n\n        // Update the stored title\n        const threads = await loadThreadMetadata();\n        const thread = threads.find((t) => t.remoteId === remoteId);\n        if (thread) {\n          thread.title = title;\n          await saveThreadMetadata(threads);\n        }\n\n        // Return a stream with a single text part\n        return createAssistantStream((controller) => {\n          controller.appendText(title);\n        });\n      }\n\n      // No title generator — return empty stream\n      return createAssistantStream(() => {});\n    },\n  };\n\n  return adapter;\n};\n"],"mappings":";;;;;;AAsCA,IAAM,6BAAN,MAAiE;CAErD;CACA;CACA;CAHV,YACE,SACA,KACA,QACA;EAHQ,KAAA,UAAA;EACA,KAAA,MAAA;EACA,KAAA,SAAA;CACP;CAEH,aAAqB,UAAkB;EACrC,OAAO,GAAG,KAAK,OAAO,WAAW;CACnC;CAEA,MAAM,OAA2C;EAC/C,MAAM,WAAW,KAAK,IAAI,eAAe,EAAE,SAAS,EAAE;EACtD,IAAI,CAAC,UAAU,OAAO,EAAE,UAAU,CAAC,EAAE;EAErC,MAAM,MAAM,MAAM,KAAK,QAAQ,QAAQ,KAAK,aAAa,QAAQ,CAAC;EAClE,IAAI,CAAC,KAAK,OAAO,EAAE,UAAU,CAAC,EAAE;EAChC,OAAO,KAAK,MAAM,GAAG;CACvB;CAEA,MAAM,OAAO,MAAoD;EAC/D,MAAM,EAAE,aAAa,MAAM,KAAK,IAAI,eAAe,EAAE,WAAW;EAEhE,MAAM,MAAM,KAAK,aAAa,QAAQ;EACtC,MAAM,MAAM,MAAM,KAAK,QAAQ,QAAQ,GAAG;EAC1C,MAAM,OAAkC,MACnC,KAAK,MAAM,GAAG,IACf,EAAE,UAAU,CAAC,EAAE;EAEnB,MAAM,MAAM,KAAK,SAAS,WACvB,MAAM,EAAE,QAAQ,OAAO,KAAK,QAAQ,EACvC;EACA,IAAI,OAAO,GACT,KAAK,SAAS,OAAO;OAErB,KAAK,SAAS,KAAK,IAAI;EAEzB,KAAK,SAAS,KAAK,QAAQ;EAE3B,MAAM,KAAK,QAAQ,QAAQ,KAAK,KAAK,UAAU,IAAI,CAAC;CACtD;AACF;AAEA,MAAM,yBACJ,SACA,WAC0B;CAC1B,MAAM,YAAmC,EAAE,eAAe;EACxD,MAAM,MAAM,OAAO;EACnB,MAAM,UAAU,cACR,IAAI,2BAA2B,SAAS,KAAK,MAAM,GACzD,CAAC,GAAG,CACN;EAGA,OACE,oBAAC,wBAAD;GAAwB,UAHT,eAAe,EAAE,QAAQ,IAAI,CAAC,OAAO,CAGX;GACtC;EACqB,CAAA;CAE5B;CACA,OAAO;AACT;AAEA,MAAa,6BACX,YAC4B;CAC5B,MAAM,EAAE,SAAS,SAAS,kBAAkB,mBAAmB;CAE/D,MAAM,aAAa,GAAG,OAAO;CAC7B,MAAM,eAAe,aAAqB,GAAG,OAAO,WAAW;CAE/D,MAAM,qBAAqB,YAA6C;EACtE,MAAM,MAAM,MAAM,QAAQ,QAAQ,UAAU;EAC5C,OAAO,MAAO,KAAK,MAAM,GAAG,IAA+B,CAAC;CAC9D;CAEA,MAAM,qBAAqB,OACzB,YACkB;EAClB,MAAM,QAAQ,QAAQ,YAAY,KAAK,UAAU,OAAO,CAAC;CAC3D;CAyHA,OAAO;EAtHL,mBAAmB,sBAAsB,SAAS,MAAM;EAExD,MAAM,OAA0C;GAE9C,OAAO,EACL,UAAS,MAFW,mBAAmB,GAEtB,KAAK,OAAO;IAC3B,UAAU,EAAE;IACZ,YAAY,EAAE;IACd,QAAQ,EAAE;IACV,OAAO,EAAE;IACT,QAAQ,EAAE;GACZ,EAAE,EACJ;EACF;EAEA,MAAM,WACJ,UACyC;GACzC,MAAM,WAAW;GACjB,MAAM,UAAU,MAAM,mBAAmB;GAGzC,IAAI,CAAC,QAAQ,MAAM,MAAM,EAAE,aAAa,QAAQ,GAAG;IACjD,QAAQ,QAAQ;KACd;KACA,QAAQ;IACV,CAAC;IACD,MAAM,mBAAmB,OAAO;GAClC;GAEA,OAAO;IAAE;IAAU,YAAY,KAAA;GAAU;EAC3C;EAEA,MAAM,OAAO,UAAkB,UAAiC;GAC9D,MAAM,UAAU,MAAM,mBAAmB;GACzC,MAAM,SAAS,QAAQ,MAAM,MAAM,EAAE,aAAa,QAAQ;GAC1D,IAAI,QAAQ;IACV,OAAO,QAAQ;IACf,MAAM,mBAAmB,OAAO;GAClC;EACF;EAEA,MAAM,aACJ,UACA,QACe;GACf,MAAM,UAAU,MAAM,mBAAmB;GACzC,MAAM,SAAS,QAAQ,MAAM,MAAM,EAAE,aAAa,QAAQ;GAC1D,IAAI,QAAQ;IACV,OAAO,SAAS;IAChB,MAAM,mBAAmB,OAAO;GAClC;EACF;EAEA,MAAM,QAAQ,UAAiC;GAC7C,MAAM,UAAU,MAAM,mBAAmB;GACzC,MAAM,SAAS,QAAQ,MAAM,MAAM,EAAE,aAAa,QAAQ;GAC1D,IAAI,QAAQ;IACV,OAAO,SAAS;IAChB,MAAM,mBAAmB,OAAO;GAClC;EACF;EAEA,MAAM,UAAU,UAAiC;GAC/C,MAAM,UAAU,MAAM,mBAAmB;GACzC,MAAM,SAAS,QAAQ,MAAM,MAAM,EAAE,aAAa,QAAQ;GAC1D,IAAI,QAAQ;IACV,OAAO,SAAS;IAChB,MAAM,mBAAmB,OAAO;GAClC;EACF;EAEA,MAAM,OAAO,UAAiC;GAG5C,MAAM,oBADW,MADK,mBAAmB,GAChB,QAAQ,MAAM,EAAE,aAAa,QACtB,CAAC;GACjC,MAAM,QAAQ,WAAW,YAAY,QAAQ,CAAC;EAChD;EAEA,MAAM,MAAM,UAAiD;GAE3D,MAAM,UAAS,MADO,mBAAmB,GAClB,MAAM,MAAM,EAAE,aAAa,QAAQ;GAC1D,IAAI,CAAC,QAAQ,MAAM,IAAI,MAAM,kBAAkB;GAC/C,OAAO;IACL,UAAU,OAAO;IACjB,YAAY,OAAO;IACnB,QAAQ,OAAO;IACf,OAAO,OAAO;IACd,QAAQ,OAAO;GACjB;EACF;EAEA,MAAM,cACJ,UACA,UAC0B;GAC1B,IAAI,gBAAgB;IAClB,MAAM,QAAQ,MAAM,eAAe,cAAc,QAAQ;IAGzD,MAAM,UAAU,MAAM,mBAAmB;IACzC,MAAM,SAAS,QAAQ,MAAM,MAAM,EAAE,aAAa,QAAQ;IAC1D,IAAI,QAAQ;KACV,OAAO,QAAQ;KACf,MAAM,mBAAmB,OAAO;IAClC;IAGA,OAAO,uBAAuB,eAAe;KAC3C,WAAW,WAAW,KAAK;IAC7B,CAAC;GACH;GAGA,OAAO,4BAA4B,CAAC,CAAC;EACvC;CAGW;AACf"}

package/dist/react/client/Tools.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Tools.d.ts","names":[],"sources":["../../../src/react/client/Tools.ts"],"mappings":";;;;;;;AAkCA;;;;;;;cAAa,KAAA,GAAK,KAAA;EAQW,gFAFf,OAAA;WAED,eAAA,CAAgB,oBAAA;AAAA;kFAFf,OAAA;WAED,eAAA,CAAgB,oBAAA;AAAA"}
+{"version":3,"file":"Tools.d.ts","names":[],"sources":["../../../src/react/client/Tools.ts"],"mappings":";;;;;;;AAmCA;;;;;;;cAAa,KAAA,GAAK,KAAA;EAQW,gFAFf,OAAA;WAED,eAAA,CAAgB,oBAAA;AAAA;kFAFf,OAAA;WAED,eAAA,CAAgB,oBAAA;AAAA"}

package/dist/react/client/Tools.js

@@ -1,5 +1,5 @@
 import { ModelContext } from "../../store/clients/model-context-client.js";
-import { isStandaloneToolDisplay } from "../model-context/toolbox.js";
+import { isStandaloneToolDisplay, makeToolCallTextComponent } from "../model-context/toolbox.js";
 import { attachTransformScopes, tapAssistantClientRef } from "@assistant-ui/store";
 import { resource, tapCallback, tapEffect, tapMemo, tapResources, tapState, withKey } from "@assistant-ui/tap";
 //#region src/react/client/Tools.ts
@@ -44,9 +44,15 @@ const Tools = resource(({ toolkit, mcpApp }) => {
 	tapEffect(() => {
 		if (!toolkit) return;
 		const unsubscribes = [];
-		for (const [toolName, tool] of Object.entries(toolkit)) if (tool.render) unsubscribes.push(setToolUI(toolName, tool.render, { standalone: isStandaloneToolDisplay(tool) }));
+		for (const [toolName, tool] of Object.entries(toolkit)) {
+			const toolRender = "render" in tool ? tool.render : void 0;
+			const toolRenderText = "renderText" in tool ? tool.renderText : void 0;
+			const render = toolRender ?? (toolRenderText ? makeToolCallTextComponent(toolRenderText) : void 0);
+			if (render) unsubscribes.push(setToolUI(toolName, render, { standalone: isStandaloneToolDisplay(tool) }));
+		}
 		const toolsWithoutRender = Object.entries(toolkit).reduce((acc, [name, tool]) => {
-			const { render, display, ...rest } = tool;
+			if (tool.type === "mcp") return acc;
+			const { display: _display, render: _render, renderText: _renderText, ...rest } = tool;
 			acc[name] = rest;
 			return acc;
 		}, {});

package/dist/react/client/Tools.js.map

@@ -1 +1 @@
-{"version":3,"file":"Tools.js","names":[],"sources":["../../../src/react/client/Tools.ts"],"sourcesContent":["import {\n  resource,\n  tapState,\n  tapEffect,\n  tapCallback,\n  tapMemo,\n  tapResources,\n  withKey,\n  type ResourceElement,\n} from \"@assistant-ui/tap\";\nimport {\n  tapAssistantClientRef,\n  type ClientOutput,\n  attachTransformScopes,\n} from \"@assistant-ui/store\";\nimport type { McpAppResourceOutput, ToolsState } from \"../types/scopes/tools\";\nimport type { Tool } from \"assistant-stream\";\nimport {\n  isStandaloneToolDisplay,\n  type Toolkit,\n} from \"../model-context/toolbox\";\nimport type { ToolCallMessagePartComponent } from \"../types/MessagePartComponentTypes\";\nimport { ModelContext } from \"../../store\";\n\nexport type { McpAppResourceOutput };\n\n/**\n * Registers tools with model context and installs tool-call renderers.\n *\n * Mount this resource near an assistant subtree when you want to expose a\n * group of tools declaratively. Tool definitions are registered with model\n * context, while each tool renderer is registered with the tools scope for\n * message rendering.\n */\nexport const Tools = resource(\n  ({\n    toolkit,\n    mcpApp,\n  }: {\n    /** Tools to expose to the model and optional renderers to install. */\n    toolkit?: Toolkit;\n    /** Optional MCP app resource whose tools should be merged into context. */\n    mcpApp?: ResourceElement<McpAppResourceOutput> | undefined;\n  }): ClientOutput<\"tools\"> => {\n    const mcpAppOutputs = tapResources(\n      () => (mcpApp ? [withKey(\"mcpApp\", mcpApp)] : []),\n      [mcpApp],\n    );\n    const mcpAppOutput = mcpAppOutputs[0];\n\n    const [toolUIs, setToolUIs] = tapState<ToolsState[\"toolUIs\"]>(() => ({}));\n\n    const state = tapMemo(\n      (): ToolsState => ({\n        toolUIs,\n        mcpApp: mcpAppOutput,\n        // Deprecated component-only view, derived from `toolUIs`. Removed in v0.15.\n        tools: Object.fromEntries(\n          Object.entries(toolUIs).map(([name, regs]) => [\n            name,\n            regs.map((r) => r.render),\n          ]),\n        ),\n      }),\n      [toolUIs, mcpAppOutput],\n    );\n\n    const clientRef = tapAssistantClientRef();\n\n    const setToolUI = tapCallback(\n      (\n        toolName: string,\n        render: ToolCallMessagePartComponent,\n        options?: { standalone?: boolean },\n      ) => {\n        // One registration object per call; identity is the removal key, so\n        // the per-name list stays correctly ref-counted across re-registers.\n        const registration = {\n          render,\n          standalone: options?.standalone ?? false,\n        };\n\n        setToolUIs((prev) => ({\n          ...prev,\n          [toolName]: [...(prev[toolName] ?? []), registration],\n        }));\n\n        return () => {\n          setToolUIs((prev) => {\n            const next =\n              prev[toolName]?.filter((r) => r !== registration) ?? [];\n            if (next.length > 0) return { ...prev, [toolName]: next };\n            // Drop the key entirely so repeatedly mounted/unmounted tools\n            // don't leave empty arrays accumulating across a long session.\n            const { [toolName]: _removed, ...rest } = prev;\n            return rest;\n          });\n        };\n      },\n      [],\n    );\n\n    tapEffect(() => {\n      if (!toolkit) return;\n      const unsubscribes: (() => void)[] = [];\n\n      // Register tool UIs (exclude symbols)\n      for (const [toolName, tool] of Object.entries(toolkit)) {\n        if (tool.render) {\n          unsubscribes.push(\n            setToolUI(toolName, tool.render, {\n              standalone: isStandaloneToolDisplay(tool),\n            }),\n          );\n        }\n      }\n\n      // Register tools with model context (exclude symbols). `render` and\n      // `display` are client-only presentation concerns and never reach the\n      // model.\n      const toolsWithoutRender = Object.entries(toolkit).reduce(\n        (acc, [name, tool]) => {\n          const { render, display, ...rest } = tool;\n          acc[name] = rest;\n          return acc;\n        },\n        {} as Record<string, Tool<any, any>>,\n      );\n\n      const modelContextProvider = {\n        getModelContext: () => ({\n          tools: toolsWithoutRender,\n        }),\n      };\n\n      unsubscribes.push(\n        clientRef.current!.modelContext().register(modelContextProvider),\n      );\n\n      return () => {\n        unsubscribes.forEach((fn) => fn());\n      };\n    }, [toolkit, setToolUI, clientRef]);\n\n    return {\n      getState: () => state,\n      setToolUI,\n    };\n  },\n);\n\nattachTransformScopes(Tools, (scopes, parent) => {\n  if (!scopes.modelContext && parent.modelContext.source === null) {\n    scopes.modelContext = ModelContext();\n  }\n});\n"],"mappings":";;;;;;;;;;;;;AAkCA,MAAa,QAAQ,UAClB,EACC,SACA,aAM2B;CAK3B,MAAM,eAJgB,mBACb,SAAS,CAAC,QAAQ,UAAU,MAAM,CAAC,IAAI,CAAC,GAC/C,CAAC,MAAM,CAEwB,EAAE;CAEnC,MAAM,CAAC,SAAS,cAAc,gBAAuC,CAAC,EAAE;CAExE,MAAM,QAAQ,eACO;EACjB;EACA,QAAQ;EAER,OAAO,OAAO,YACZ,OAAO,QAAQ,OAAO,EAAE,KAAK,CAAC,MAAM,UAAU,CAC5C,MACA,KAAK,KAAK,MAAM,EAAE,MAAM,CAC1B,CAAC,CACH;CACF,IACA,CAAC,SAAS,YAAY,CACxB;CAEA,MAAM,YAAY,sBAAsB;CAExC,MAAM,YAAY,aAEd,UACA,QACA,YACG;EAGH,MAAM,eAAe;GACnB;GACA,YAAY,SAAS,cAAc;EACrC;EAEA,YAAY,UAAU;GACpB,GAAG;IACF,WAAW,CAAC,GAAI,KAAK,aAAa,CAAC,GAAI,YAAY;EACtD,EAAE;EAEF,aAAa;GACX,YAAY,SAAS;IACnB,MAAM,OACJ,KAAK,WAAW,QAAQ,MAAM,MAAM,YAAY,KAAK,CAAC;IACxD,IAAI,KAAK,SAAS,GAAG,OAAO;KAAE,GAAG;MAAO,WAAW;IAAK;IAGxD,MAAM,GAAG,WAAW,UAAU,GAAG,SAAS;IAC1C,OAAO;GACT,CAAC;EACH;CACF,GACA,CAAC,CACH;CAEA,gBAAgB;EACd,IAAI,CAAC,SAAS;EACd,MAAM,eAA+B,CAAC;EAGtC,KAAK,MAAM,CAAC,UAAU,SAAS,OAAO,QAAQ,OAAO,GACnD,IAAI,KAAK,QACP,aAAa,KACX,UAAU,UAAU,KAAK,QAAQ,EAC/B,YAAY,wBAAwB,IAAI,EAC1C,CAAC,CACH;EAOJ,MAAM,qBAAqB,OAAO,QAAQ,OAAO,EAAE,QAChD,KAAK,CAAC,MAAM,UAAU;GACrB,MAAM,EAAE,QAAQ,SAAS,GAAG,SAAS;GACrC,IAAI,QAAQ;GACZ,OAAO;EACT,GACA,CAAC,CACH;EAQA,aAAa,KACX,UAAU,QAAS,aAAa,EAAE,SAAS,EAN3C,wBAAwB,EACtB,OAAO,mBACT,GAI8D,CAAC,CACjE;EAEA,aAAa;GACX,aAAa,SAAS,OAAO,GAAG,CAAC;EACnC;CACF,GAAG;EAAC;EAAS;EAAW;CAAS,CAAC;CAElC,OAAO;EACL,gBAAgB;EAChB;CACF;AACF,CACF;AAEA,sBAAsB,QAAQ,QAAQ,WAAW;CAC/C,IAAI,CAAC,OAAO,gBAAgB,OAAO,aAAa,WAAW,MACzD,OAAO,eAAe,aAAa;AAEvC,CAAC"}
+{"version":3,"file":"Tools.js","names":[],"sources":["../../../src/react/client/Tools.ts"],"sourcesContent":["import {\n  resource,\n  tapState,\n  tapEffect,\n  tapCallback,\n  tapMemo,\n  tapResources,\n  withKey,\n  type ResourceElement,\n} from \"@assistant-ui/tap\";\nimport {\n  tapAssistantClientRef,\n  type ClientOutput,\n  attachTransformScopes,\n} from \"@assistant-ui/store\";\nimport type { McpAppResourceOutput, ToolsState } from \"../types/scopes/tools\";\nimport type { Tool } from \"assistant-stream\";\nimport {\n  isStandaloneToolDisplay,\n  makeToolCallTextComponent,\n  type Toolkit,\n} from \"../model-context/toolbox\";\nimport type { ToolCallMessagePartComponent } from \"../types/MessagePartComponentTypes\";\nimport { ModelContext } from \"../../store\";\n\nexport type { McpAppResourceOutput };\n\n/**\n * Registers tools with model context and installs tool-call renderers.\n *\n * Mount this resource near an assistant subtree when you want to expose a\n * group of tools declaratively. Tool definitions are registered with model\n * context, while each tool renderer is registered with the tools scope for\n * message rendering.\n */\nexport const Tools = resource(\n  ({\n    toolkit,\n    mcpApp,\n  }: {\n    /** Tools to expose to the model and optional renderers to install. */\n    toolkit?: Toolkit;\n    /** Optional MCP app resource whose tools should be merged into context. */\n    mcpApp?: ResourceElement<McpAppResourceOutput> | undefined;\n  }): ClientOutput<\"tools\"> => {\n    const mcpAppOutputs = tapResources(\n      () => (mcpApp ? [withKey(\"mcpApp\", mcpApp)] : []),\n      [mcpApp],\n    );\n    const mcpAppOutput = mcpAppOutputs[0];\n\n    const [toolUIs, setToolUIs] = tapState<ToolsState[\"toolUIs\"]>(() => ({}));\n\n    const state = tapMemo(\n      (): ToolsState => ({\n        toolUIs,\n        mcpApp: mcpAppOutput,\n        // Deprecated component-only view, derived from `toolUIs`. Removed in v0.15.\n        tools: Object.fromEntries(\n          Object.entries(toolUIs).map(([name, regs]) => [\n            name,\n            regs.map((r) => r.render),\n          ]),\n        ),\n      }),\n      [toolUIs, mcpAppOutput],\n    );\n\n    const clientRef = tapAssistantClientRef();\n\n    const setToolUI = tapCallback(\n      (\n        toolName: string,\n        render: ToolCallMessagePartComponent,\n        options?: { standalone?: boolean },\n      ) => {\n        // One registration object per call; identity is the removal key, so\n        // the per-name list stays correctly ref-counted across re-registers.\n        const registration = {\n          render,\n          standalone: options?.standalone ?? false,\n        };\n\n        setToolUIs((prev) => ({\n          ...prev,\n          [toolName]: [...(prev[toolName] ?? []), registration],\n        }));\n\n        return () => {\n          setToolUIs((prev) => {\n            const next =\n              prev[toolName]?.filter((r) => r !== registration) ?? [];\n            if (next.length > 0) return { ...prev, [toolName]: next };\n            // Drop the key entirely so repeatedly mounted/unmounted tools\n            // don't leave empty arrays accumulating across a long session.\n            const { [toolName]: _removed, ...rest } = prev;\n            return rest;\n          });\n        };\n      },\n      [],\n    );\n\n    tapEffect(() => {\n      if (!toolkit) return;\n      const unsubscribes: (() => void)[] = [];\n\n      // Register tool UIs (exclude symbols)\n      for (const [toolName, tool] of Object.entries(toolkit)) {\n        const toolRender = \"render\" in tool ? tool.render : undefined;\n        const toolRenderText =\n          \"renderText\" in tool ? tool.renderText : undefined;\n        const render =\n          toolRender ??\n          (toolRenderText\n            ? makeToolCallTextComponent(toolRenderText)\n            : undefined);\n        if (render) {\n          unsubscribes.push(\n            setToolUI(toolName, render, {\n              standalone: isStandaloneToolDisplay(tool),\n            }),\n          );\n        }\n      }\n\n      // Register tools with model context (exclude symbols). `render`,\n      // `renderText`, and `display` are client-only presentation concerns and\n      // never reach the model.\n      const toolsWithoutRender = Object.entries(toolkit).reduce(\n        (acc, [name, tool]) => {\n          if (tool.type === \"mcp\") return acc;\n          const {\n            display: _display,\n            render: _render,\n            renderText: _renderText,\n            ...rest\n          } = tool as typeof tool & { renderText?: unknown };\n          acc[name] = rest as Tool<any, any>;\n          return acc;\n        },\n        {} as Record<string, Tool<any, any>>,\n      );\n\n      const modelContextProvider = {\n        getModelContext: () => ({\n          tools: toolsWithoutRender,\n        }),\n      };\n\n      unsubscribes.push(\n        clientRef.current!.modelContext().register(modelContextProvider),\n      );\n\n      return () => {\n        unsubscribes.forEach((fn) => fn());\n      };\n    }, [toolkit, setToolUI, clientRef]);\n\n    return {\n      getState: () => state,\n      setToolUI,\n    };\n  },\n);\n\nattachTransformScopes(Tools, (scopes, parent) => {\n  if (!scopes.modelContext && parent.modelContext.source === null) {\n    scopes.modelContext = ModelContext();\n  }\n});\n"],"mappings":";;;;;;;;;;;;;AAmCA,MAAa,QAAQ,UAClB,EACC,SACA,aAM2B;CAK3B,MAAM,eAJgB,mBACb,SAAS,CAAC,QAAQ,UAAU,MAAM,CAAC,IAAI,CAAC,GAC/C,CAAC,MAAM,CAEwB,EAAE;CAEnC,MAAM,CAAC,SAAS,cAAc,gBAAuC,CAAC,EAAE;CAExE,MAAM,QAAQ,eACO;EACjB;EACA,QAAQ;EAER,OAAO,OAAO,YACZ,OAAO,QAAQ,OAAO,EAAE,KAAK,CAAC,MAAM,UAAU,CAC5C,MACA,KAAK,KAAK,MAAM,EAAE,MAAM,CAC1B,CAAC,CACH;CACF,IACA,CAAC,SAAS,YAAY,CACxB;CAEA,MAAM,YAAY,sBAAsB;CAExC,MAAM,YAAY,aAEd,UACA,QACA,YACG;EAGH,MAAM,eAAe;GACnB;GACA,YAAY,SAAS,cAAc;EACrC;EAEA,YAAY,UAAU;GACpB,GAAG;IACF,WAAW,CAAC,GAAI,KAAK,aAAa,CAAC,GAAI,YAAY;EACtD,EAAE;EAEF,aAAa;GACX,YAAY,SAAS;IACnB,MAAM,OACJ,KAAK,WAAW,QAAQ,MAAM,MAAM,YAAY,KAAK,CAAC;IACxD,IAAI,KAAK,SAAS,GAAG,OAAO;KAAE,GAAG;MAAO,WAAW;IAAK;IAGxD,MAAM,GAAG,WAAW,UAAU,GAAG,SAAS;IAC1C,OAAO;GACT,CAAC;EACH;CACF,GACA,CAAC,CACH;CAEA,gBAAgB;EACd,IAAI,CAAC,SAAS;EACd,MAAM,eAA+B,CAAC;EAGtC,KAAK,MAAM,CAAC,UAAU,SAAS,OAAO,QAAQ,OAAO,GAAG;GACtD,MAAM,aAAa,YAAY,OAAO,KAAK,SAAS,KAAA;GACpD,MAAM,iBACJ,gBAAgB,OAAO,KAAK,aAAa,KAAA;GAC3C,MAAM,SACJ,eACC,iBACG,0BAA0B,cAAc,IACxC,KAAA;GACN,IAAI,QACF,aAAa,KACX,UAAU,UAAU,QAAQ,EAC1B,YAAY,wBAAwB,IAAI,EAC1C,CAAC,CACH;EAEJ;EAKA,MAAM,qBAAqB,OAAO,QAAQ,OAAO,EAAE,QAChD,KAAK,CAAC,MAAM,UAAU;GACrB,IAAI,KAAK,SAAS,OAAO,OAAO;GAChC,MAAM,EACJ,SAAS,UACT,QAAQ,SACR,YAAY,aACZ,GAAG,SACD;GACJ,IAAI,QAAQ;GACZ,OAAO;EACT,GACA,CAAC,CACH;EAQA,aAAa,KACX,UAAU,QAAS,aAAa,EAAE,SAAS,EAN3C,wBAAwB,EACtB,OAAO,mBACT,GAI8D,CAAC,CACjE;EAEA,aAAa;GACX,aAAa,SAAS,OAAO,GAAG,CAAC;EACnC;CACF,GAAG;EAAC;EAAS;EAAW;CAAS,CAAC;CAElC,OAAO;EACL,gBAAgB;EAChB;CACF;AACF,CACF;AAEA,sBAAsB,QAAQ,QAAQ,WAAW;CAC/C,IAAI,CAAC,OAAO,gBAAgB,OAAO,aAAa,WAAW,MACzD,OAAO,eAAe,aAAa;AAEvC,CAAC"}

package/dist/react/index.d.ts

@@ -10,7 +10,7 @@ import { DataRenderers } from "./client/DataRenderers.js";
 import { Interactables } from "./client/Interactables.js";
 import { DataMessagePartComponent, DataMessagePartProps, EmptyMessagePartComponent, EmptyMessagePartProps, FileMessagePartComponent, FileMessagePartProps, GenerativeUIComponentRegistry, GenerativeUIMessagePartComponent, GenerativeUIMessagePartProps, GenerativeUIRenderProps, ImageMessagePartComponent, ImageMessagePartProps, QuoteMessagePartComponent, QuoteMessagePartProps, ReasoningGroupComponent, ReasoningGroupProps, ReasoningMessagePartComponent, ReasoningMessagePartProps, SourceMessagePartComponent, SourceMessagePartProps, TextMessagePartComponent, TextMessagePartProps, ToolCallMessagePartComponent, ToolCallMessagePartProps, Unstable_AudioMessagePartComponent, Unstable_AudioMessagePartProps } from "./types/MessagePartComponentTypes.js";
 import { McpAppResourceOutput, ToolsClientSchema, ToolsMethods, ToolsState } from "./types/scopes/tools.js";
-import { ToolDefinition, Toolkit, ToolkitDeclaration, ToolkitDeclarationDefinition } from "./model-context/toolbox.js";
+import { ToolCallText, ToolDefinition, Toolkit, ToolkitDefinition, ToolkitDefinitionEntry } from "./model-context/toolbox.js";
 import { Tools } from "./client/Tools.js";
 import { InteractableDefinition, InteractablePersistedState, InteractablePersistenceAdapter, InteractablePersistenceStatus, InteractableRegistration, InteractableStateSchema, InteractablesClientSchema, InteractablesMethods, InteractablesState } from "./types/scopes/interactables.js";
 import { AssistantToolProps, useAssistantTool } from "./model-context/useAssistantTool.js";
@@ -23,7 +23,11 @@ import { useAssistantInstructions } from "./model-context/useAssistantInstructio
 import { useAssistantContext } from "./model-context/useAssistantContext.js";
 import { useInlineRender } from "./model-context/useInlineRender.js";
 import { defineToolkit } from "./model-context/define-toolkit.js";
-import { hitl } from "./model-context/hitl.js";
+import { stubTool } from "./model-context/stub-tool.js";
+import { useAuiToolOverrides } from "./model-context/useAuiToolOverrides.js";
+import { hitl, hitlTool } from "./model-context/hitl.js";
+import { ProviderToolConfig, providerTool } from "./model-context/provider-tool.js";
+import { McpToolkitDefinition, defineMcpToolkit } from "./model-context/define-mcp-toolkit.js";
 import { AssistantInteractableProps, useAssistantInteractable } from "./model-context/useAssistantInteractable.js";
 import { useInteractableState } from "./model-context/useInteractableState.js";
 import { ToolArgsStatus, useToolArgsStatus } from "./model-context/useToolArgsStatus.js";
@@ -97,4 +101,4 @@ import { useEditComposerCancel } from "./primitive-hooks/useEditComposerCancel.j
 import { useEditComposerSend } from "./primitive-hooks/useEditComposerSend.js";
 import { useMessageError } from "./primitive-hooks/useMessageError.js";
 import { LocalRuntimeOptions, splitLocalRuntimeOptions, useLocalRuntime } from "./runtimes/useLocalRuntime.js";
-export { type AssistantContextConfig, type AssistantDataUI, type AssistantDataUIProps, type AssistantInteractableProps, AssistantProviderBase, type AssistantProviderBaseProps, AssistantRuntimeProvider, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AsyncStorageLike, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, type DataMessagePartComponent, type DataMessagePartProps, DataRenderers, type DataRenderersClientSchema, type DataRenderersMethods, type DataRenderersState, type EmptyMessagePartComponent, type EmptyMessagePartProps, type EnrichedPartState, type FileMessagePartComponent, type FileMessagePartProps, type GenerativeUIComponentRegistry, type GenerativeUIMessagePartComponent, type GenerativeUIMessagePartProps, GenerativeUIRender, GenerativeUIRenderError, type GenerativeUIRenderProps, type GroupByContext, type ImageMessagePartComponent, type ImageMessagePartProps, type InteractableDefinition, type InteractablePersistedState, type InteractablePersistenceAdapter, type InteractablePersistenceStatus, type InteractableRegistration, type InteractableStateSchema, Interactables, type InteractablesClientSchema, type InteractablesMethods, type InteractablesState, type LocalRuntimeOptions, type McpAppResourceOutput, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, type PartState, QueueItemByIndexProvider, type QueueItemByIndexProviderProps, type QuoteMessagePartComponent, type QuoteMessagePartProps, ReadonlyThreadProvider, type ReasoningGroupComponent, type ReasoningGroupProps, type ReasoningMessagePartComponent, type ReasoningMessagePartProps, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, type RuntimeAdapters, type SourceMessagePartComponent, type SourceMessagePartProps, SuggestionByIndexProvider, type SuggestionByIndexProviderProps, type TextMessagePartComponent, type TextMessagePartProps, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, type TitleGenerationAdapter, type ToolArgsStatus, type ToolCallMessagePartComponent, type ToolCallMessagePartProps, type ToolDefinition, type Toolkit, type ToolkitDeclaration, type ToolkitDeclarationDefinition, Tools, type ToolsClientSchema, type ToolsMethods, type ToolsState, type Unstable_AudioMessagePartComponent, type Unstable_AudioMessagePartProps, type UseActionBarCopyOptions, type UseComposerIfProps, type UseSuggestionTriggerOptions, convertExternalMessages, createLocalStorageAdapter, createMessageConverter, createSimpleTitleAdapter, defineToolkit, getMessageQuote, getRenderComponent, groupPartByType, hitl, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, splitLocalRuntimeOptions, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume };
+export { type AssistantContextConfig, type AssistantDataUI, type AssistantDataUIProps, type AssistantInteractableProps, AssistantProviderBase, type AssistantProviderBaseProps, AssistantRuntimeProvider, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AsyncStorageLike, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, type DataMessagePartComponent, type DataMessagePartProps, DataRenderers, type DataRenderersClientSchema, type DataRenderersMethods, type DataRenderersState, type EmptyMessagePartComponent, type EmptyMessagePartProps, type EnrichedPartState, type FileMessagePartComponent, type FileMessagePartProps, type GenerativeUIComponentRegistry, type GenerativeUIMessagePartComponent, type GenerativeUIMessagePartProps, GenerativeUIRender, GenerativeUIRenderError, type GenerativeUIRenderProps, type GroupByContext, type ImageMessagePartComponent, type ImageMessagePartProps, type InteractableDefinition, type InteractablePersistedState, type InteractablePersistenceAdapter, type InteractablePersistenceStatus, type InteractableRegistration, type InteractableStateSchema, Interactables, type InteractablesClientSchema, type InteractablesMethods, type InteractablesState, type LocalRuntimeOptions, type McpAppResourceOutput, type McpToolkitDefinition, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, type PartState, type ProviderToolConfig, QueueItemByIndexProvider, type QueueItemByIndexProviderProps, type QuoteMessagePartComponent, type QuoteMessagePartProps, ReadonlyThreadProvider, type ReasoningGroupComponent, type ReasoningGroupProps, type ReasoningMessagePartComponent, type ReasoningMessagePartProps, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, type RuntimeAdapters, type SourceMessagePartComponent, type SourceMessagePartProps, SuggestionByIndexProvider, type SuggestionByIndexProviderProps, type TextMessagePartComponent, type TextMessagePartProps, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, type TitleGenerationAdapter, type ToolArgsStatus, type ToolCallMessagePartComponent, type ToolCallMessagePartProps, type ToolCallText, type ToolDefinition, type Toolkit, type ToolkitDefinition, type ToolkitDefinitionEntry, Tools, type ToolsClientSchema, type ToolsMethods, type ToolsState, type Unstable_AudioMessagePartComponent, type Unstable_AudioMessagePartProps, type UseActionBarCopyOptions, type UseComposerIfProps, type UseSuggestionTriggerOptions, convertExternalMessages, createLocalStorageAdapter, createMessageConverter, createSimpleTitleAdapter, defineMcpToolkit, defineToolkit, getMessageQuote, getRenderComponent, groupPartByType, hitl, hitlTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, providerTool, splitLocalRuntimeOptions, stubTool, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAuiToolOverrides, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume };

package/dist/react/index.js

@@ -13,7 +13,11 @@ import { useAssistantInstructions } from "./model-context/useAssistantInstructio
 import { useAssistantContext } from "./model-context/useAssistantContext.js";
 import { useInlineRender } from "./model-context/useInlineRender.js";
 import { defineToolkit } from "./model-context/define-toolkit.js";
-import { hitl } from "./model-context/hitl.js";
+import { stubTool } from "./model-context/stub-tool.js";
+import { useAuiToolOverrides } from "./model-context/useAuiToolOverrides.js";
+import { hitl, hitlTool } from "./model-context/hitl.js";
+import { providerTool } from "./model-context/provider-tool.js";
+import { defineMcpToolkit } from "./model-context/define-mcp-toolkit.js";
 import { useAssistantInteractable } from "./model-context/useAssistantInteractable.js";
 import { useInteractableState } from "./model-context/useInteractableState.js";
 import { useToolArgsStatus } from "./model-context/useToolArgsStatus.js";
@@ -88,4 +92,4 @@ import { useEditComposerCancel } from "./primitive-hooks/useEditComposerCancel.j
 import { useEditComposerSend } from "./primitive-hooks/useEditComposerSend.js";
 import { useMessageError } from "./primitive-hooks/useMessageError.js";
 import { splitLocalRuntimeOptions, useLocalRuntime } from "./runtimes/useLocalRuntime.js";
-export { AssistantProviderBase, AssistantRuntimeProvider, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, DataRenderers, GenerativeUIRender, GenerativeUIRenderError, Interactables, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, QueueItemByIndexProvider, ReadonlyThreadProvider, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, SuggestionByIndexProvider, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, Tools, convertExternalMessages, createLocalStorageAdapter, createMessageConverter, createSimpleTitleAdapter, defineToolkit, getMessageQuote, getRenderComponent, groupPartByType, hitl, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, splitLocalRuntimeOptions, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume };
+export { AssistantProviderBase, AssistantRuntimeProvider, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, DataRenderers, GenerativeUIRender, GenerativeUIRenderError, Interactables, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, QueueItemByIndexProvider, ReadonlyThreadProvider, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, SuggestionByIndexProvider, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, Tools, convertExternalMessages, createLocalStorageAdapter, createMessageConverter, createSimpleTitleAdapter, defineMcpToolkit, defineToolkit, getMessageQuote, getRenderComponent, groupPartByType, hitl, hitlTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, providerTool, splitLocalRuntimeOptions, stubTool, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAuiToolOverrides, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume };

package/dist/react/model-context/define-mcp-toolkit.d.ts

@@ -0,0 +1,12 @@
+import { Toolkit } from "./toolbox.js";
+import { McpServerConfig } from "assistant-stream";
+
+//#region src/react/model-context/define-mcp-toolkit.d.ts
+type McpToolkitDefinition = Record<string, McpServerConfig>;
+/**
+ * Defines MCP server tools as a spreadable toolkit fragment.
+ */
+declare function defineMcpToolkit(definition: McpToolkitDefinition): Toolkit;
+//#endregion
+export { McpToolkitDefinition, defineMcpToolkit };
+//# sourceMappingURL=define-mcp-toolkit.d.ts.map

package/dist/react/model-context/define-mcp-toolkit.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"define-mcp-toolkit.d.ts","names":[],"sources":["../../../src/react/model-context/define-mcp-toolkit.ts"],"mappings":";;;;KAGY,oBAAA,GAAuB,MAAM,SAAS,eAAA;;AAAlD;;iBAKgB,gBAAA,CAAiB,UAAA,EAAY,oBAAA,GAAuB,OAAO"}

package/dist/react/model-context/define-mcp-toolkit.js

@@ -0,0 +1,14 @@
+//#region src/react/model-context/define-mcp-toolkit.ts
+/**
+* Defines MCP server tools as a spreadable toolkit fragment.
+*/
+function defineMcpToolkit(definition) {
+	return Object.fromEntries(Object.entries(definition).map(([name, server]) => [name, {
+		type: "mcp",
+		server
+	}]));
+}
+//#endregion
+export { defineMcpToolkit };
+
+//# sourceMappingURL=define-mcp-toolkit.js.map

package/dist/react/model-context/define-mcp-toolkit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"define-mcp-toolkit.js","names":[],"sources":["../../../src/react/model-context/define-mcp-toolkit.ts"],"sourcesContent":["import type { McpServerConfig } from \"assistant-stream\";\nimport type { Toolkit } from \"./toolbox\";\n\nexport type McpToolkitDefinition = Record<string, McpServerConfig>;\n\n/**\n * Defines MCP server tools as a spreadable toolkit fragment.\n */\nexport function defineMcpToolkit(definition: McpToolkitDefinition): Toolkit {\n  return Object.fromEntries(\n    Object.entries(definition).map(([name, server]) => [\n      name,\n      { type: \"mcp\", server },\n    ]),\n  ) as Toolkit;\n}\n"],"mappings":";;;;AAQA,SAAgB,iBAAiB,YAA2C;CAC1E,OAAO,OAAO,YACZ,OAAO,QAAQ,UAAU,EAAE,KAAK,CAAC,MAAM,YAAY,CACjD,MACA;EAAE,MAAM;EAAO;CAAO,CACxB,CAAC,CACH;AACF"}

package/dist/react/model-context/define-toolkit.d.ts

@@ -1,9 +1,9 @@
-import { Toolkit, ToolkitDeclaration } from "./toolbox.js";
+import { Toolkit, ToolkitDefinition, ToolkitDefinitionEntryWithParameters } from "./toolbox.js";
 
 //#region src/react/model-context/define-toolkit.d.ts
 /**
  * Authoring helper for a `"use generative"` toolkit. Accepts the permissive
- * {@link ToolkitDeclaration} (a `backend` tool may carry its server `execute`)
+ * {@link ToolkitDefinition} (a `backend` tool may carry its server `execute`)
  * and types the result as the canonical {@link Toolkit}.
  *
  * It has **no runtime implementation**. A `"use generative"` compiler (e.g.
@@ -14,7 +14,8 @@ import { Toolkit, ToolkitDeclaration } from "./toolbox.js";
  * outside a `"use generative"` file — which would ship a backend `execute` to the
  * client. So it throws instead of silently leaking.
  */
-declare function defineToolkit(_declaration: ToolkitDeclaration): Toolkit;
+declare function defineToolkit<TArgsByName extends { [K in keyof TArgsByName]: Record<string, unknown> }, TResultByName extends { [K in keyof TArgsByName]: unknown } = { [K in keyof TArgsByName]: unknown }>(_definition: { [K in keyof TArgsByName]: ToolkitDefinitionEntryWithParameters<TArgsByName[K], TResultByName[K]> }): Toolkit;
+declare function defineToolkit(_definition: ToolkitDefinition): Toolkit;
 //#endregion
 export { defineToolkit };
 //# sourceMappingURL=define-toolkit.d.ts.map

package/dist/react/model-context/define-toolkit.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"define-toolkit.d.ts","names":[],"sources":["../../../src/react/model-context/define-toolkit.ts"],"mappings":";;;;;AAeA;;;;;;;;AAAwE;;;iBAAxD,aAAA,CAAc,YAAA,EAAc,kBAAA,GAAqB,OAAO"}
+{"version":3,"file":"define-toolkit.d.ts","names":[],"sources":["../../../src/react/model-context/define-toolkit.ts"],"mappings":";;;;;AAmBA;;;;;;;;;;;iBAAgB,aAAA,mCAEA,WAAA,GAAc,MAAA,yDAEQ,WAAA,6BACtB,WAAA,cAEd,WAAA,gBACY,WAAA,GAAc,oCAAA,CACxB,WAAA,CAAY,CAAA,GACZ,aAAA,CAAc,CAAA,OAEd,OAAA;AAAA,iBACY,aAAA,CAAc,WAAA,EAAa,iBAAA,GAAoB,OAAO"}

package/dist/react/model-context/define-toolkit.js

@@ -1,18 +1,5 @@
 //#region src/react/model-context/define-toolkit.ts
-/**
-* Authoring helper for a `"use generative"` toolkit. Accepts the permissive
-* {@link ToolkitDeclaration} (a `backend` tool may carry its server `execute`)
-* and types the result as the canonical {@link Toolkit}.
-*
-* It has **no runtime implementation**. A `"use generative"` compiler (e.g.
-* `@assistant-ui/next` or `@assistant-ui/vite`) strips the `defineToolkit(...)`
-* wrapper (and its import) per build, so a correctly compiled
-* `export default defineToolkit({...})` never calls this. If it *does* run, the
-* module was not compiled by a use-generative loader — e.g. `defineToolkit` used
-* outside a `"use generative"` file — which would ship a backend `execute` to the
-* client. So it throws instead of silently leaking.
-*/
-function defineToolkit(_declaration) {
+function defineToolkit(_definition) {
 	throw new Error("[assistant-ui] defineToolkit() has no runtime implementation — it is stripped at build time by the use-generative compiler. Reaching it means this module was not compiled (e.g. defineToolkit used outside a \"use generative\" file). Add the directive, or do not use defineToolkit here.");
 }
 //#endregion

package/dist/react/model-context/define-toolkit.js.map

@@ -1 +1 @@
-{"version":3,"file":"define-toolkit.js","names":[],"sources":["../../../src/react/model-context/define-toolkit.ts"],"sourcesContent":["import type { Toolkit, ToolkitDeclaration } from \"./toolbox\";\n\n/**\n * Authoring helper for a `\"use generative\"` toolkit. Accepts the permissive\n * {@link ToolkitDeclaration} (a `backend` tool may carry its server `execute`)\n * and types the result as the canonical {@link Toolkit}.\n *\n * It has **no runtime implementation**. A `\"use generative\"` compiler (e.g.\n * `@assistant-ui/next` or `@assistant-ui/vite`) strips the `defineToolkit(...)`\n * wrapper (and its import) per build, so a correctly compiled\n * `export default defineToolkit({...})` never calls this. If it *does* run, the\n * module was not compiled by a use-generative loader — e.g. `defineToolkit` used\n * outside a `\"use generative\"` file — which would ship a backend `execute` to the\n * client. So it throws instead of silently leaking.\n */\nexport function defineToolkit(_declaration: ToolkitDeclaration): Toolkit {\n  throw new Error(\n    \"[assistant-ui] defineToolkit() has no runtime implementation — it is \" +\n      \"stripped at build time by the use-generative compiler. Reaching it means \" +\n      'this module was not compiled (e.g. defineToolkit used outside a \"use ' +\n      'generative\" file). Add the directive, or do not use defineToolkit here.',\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;AAeA,SAAgB,cAAc,cAA2C;CACvE,MAAM,IAAI,MACR,8RAIF;AACF"}
+{"version":3,"file":"define-toolkit.js","names":[],"sources":["../../../src/react/model-context/define-toolkit.ts"],"sourcesContent":["import type {\n  Toolkit,\n  ToolkitDefinition,\n  ToolkitDefinitionEntryWithParameters,\n} from \"./toolbox\";\n\n/**\n * Authoring helper for a `\"use generative\"` toolkit. Accepts the permissive\n * {@link ToolkitDefinition} (a `backend` tool may carry its server `execute`)\n * and types the result as the canonical {@link Toolkit}.\n *\n * It has **no runtime implementation**. A `\"use generative\"` compiler (e.g.\n * `@assistant-ui/next` or `@assistant-ui/vite`) strips the `defineToolkit(...)`\n * wrapper (and its import) per build, so a correctly compiled\n * `export default defineToolkit({...})` never calls this. If it *does* run, the\n * module was not compiled by a use-generative loader — e.g. `defineToolkit` used\n * outside a `\"use generative\"` file — which would ship a backend `execute` to the\n * client. So it throws instead of silently leaking.\n */\nexport function defineToolkit<\n  TArgsByName extends {\n    [K in keyof TArgsByName]: Record<string, unknown>;\n  },\n  TResultByName extends { [K in keyof TArgsByName]: unknown } = {\n    [K in keyof TArgsByName]: unknown;\n  },\n>(_definition: {\n  [K in keyof TArgsByName]: ToolkitDefinitionEntryWithParameters<\n    TArgsByName[K],\n    TResultByName[K]\n  >;\n}): Toolkit;\nexport function defineToolkit(_definition: ToolkitDefinition): Toolkit;\nexport function defineToolkit(_definition: ToolkitDefinition): Toolkit {\n  throw new Error(\n    \"[assistant-ui] defineToolkit() has no runtime implementation — it is \" +\n      \"stripped at build time by the use-generative compiler. Reaching it means \" +\n      'this module was not compiled (e.g. defineToolkit used outside a \"use ' +\n      'generative\" file). Add the directive, or do not use defineToolkit here.',\n  );\n}\n"],"mappings":";AAiCA,SAAgB,cAAc,aAAyC;CACrE,MAAM,IAAI,MACR,8RAIF;AACF"}

package/dist/react/model-context/hitl.d.ts

@@ -4,16 +4,20 @@
  * supplies the result instead of code. Use it as the tool's `execute`:
  *
  * ```tsx
- * confirm: { execute: hitl(), render: (props) => <Confirm {...props} /> }
+ * confirm: { execute: hitlTool(), render: (props) => <Confirm {...props} /> }
  * ```
  *
  * Like {@link defineToolkit}, it has **no runtime implementation**: a
  * `"use generative"` compiler (e.g. `@assistant-ui/next` or `@assistant-ui/vite`)
- * detects `execute: hitl()`, drops it, and stamps the tool `type: "human"`.
+ * detects `execute: hitlTool()`, drops it, and stamps the tool `type: "human"`.
  * Reaching it at runtime means the module wasn't compiled (used outside a
  * `"use generative"` file), so it throws.
  */
-declare function hitl(): never;
+declare function hitlTool(): never;
+/**
+ * @deprecated Use {@link hitlTool}.
+ */
+declare const hitl: typeof hitlTool;
 //#endregion
-export { hitl };
+export { hitl, hitlTool };
 //# sourceMappingURL=hitl.d.ts.map

package/dist/react/model-context/hitl.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"hitl.d.ts","names":[],"sources":["../../../src/react/model-context/hitl.ts"],"mappings":";;AAcA;;;;AAAoB;;;;;;;;;iBAAJ,IAAA"}
+{"version":3,"file":"hitl.d.ts","names":[],"sources":["../../../src/react/model-context/hitl.ts"],"mappings":";;AAcA;;;;AAAwB;AAYxB;;;;AAA4B;;;;iBAZZ,QAAA;;;;cAYH,IAAA,SAAI,QAAW"}

package/dist/react/model-context/hitl.js

@@ -4,19 +4,23 @@
 * supplies the result instead of code. Use it as the tool's `execute`:
 *
 * ```tsx
-* confirm: { execute: hitl(), render: (props) => <Confirm {...props} /> }
+* confirm: { execute: hitlTool(), render: (props) => <Confirm {...props} /> }
 * ```
 *
 * Like {@link defineToolkit}, it has **no runtime implementation**: a
 * `"use generative"` compiler (e.g. `@assistant-ui/next` or `@assistant-ui/vite`)
-* detects `execute: hitl()`, drops it, and stamps the tool `type: "human"`.
+* detects `execute: hitlTool()`, drops it, and stamps the tool `type: "human"`.
 * Reaching it at runtime means the module wasn't compiled (used outside a
 * `"use generative"` file), so it throws.
 */
-function hitl() {
-	throw new Error("[assistant-ui] hitl() has no runtime implementation — it marks a human-in-the-loop tool and is stripped at build time by the use-generative compiler. Reaching it means this module was not compiled (e.g. hitl() used outside a \"use generative\" file).");
+function hitlTool() {
+	throw new Error("[assistant-ui] hitlTool() has no runtime implementation — it marks a human-in-the-loop tool and is stripped at build time by the use-generative compiler. Reaching it means this module was not compiled (e.g. hitlTool() used outside a \"use generative\" file).");
 }
+/**
+* @deprecated Use {@link hitlTool}.
+*/
+const hitl = hitlTool;
 //#endregion
-export { hitl };
+export { hitl, hitlTool };
 
 //# sourceMappingURL=hitl.js.map

package/dist/react/model-context/hitl.js.map

@@ -1 +1 @@
-{"version":3,"file":"hitl.js","names":[],"sources":["../../../src/react/model-context/hitl.ts"],"sourcesContent":["/**\n * Marks a tool as **human-in-the-loop**: the agent pauses and the UI (`render`)\n * supplies the result instead of code. Use it as the tool's `execute`:\n *\n * ```tsx\n * confirm: { execute: hitl(), render: (props) => <Confirm {...props} /> }\n * ```\n *\n * Like {@link defineToolkit}, it has **no runtime implementation**: a\n * `\"use generative\"` compiler (e.g. `@assistant-ui/next` or `@assistant-ui/vite`)\n * detects `execute: hitl()`, drops it, and stamps the tool `type: \"human\"`.\n * Reaching it at runtime means the module wasn't compiled (used outside a\n * `\"use generative\"` file), so it throws.\n */\nexport function hitl(): never {\n  throw new Error(\n    \"[assistant-ui] hitl() has no runtime implementation — it marks a \" +\n      \"human-in-the-loop tool and is stripped at build time by the \" +\n      \"use-generative compiler. Reaching it means this module was not compiled \" +\n      '(e.g. hitl() used outside a \"use generative\" file).',\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;;AAcA,SAAgB,OAAc;CAC5B,MAAM,IAAI,MACR,4PAIF;AACF"}
+{"version":3,"file":"hitl.js","names":[],"sources":["../../../src/react/model-context/hitl.ts"],"sourcesContent":["/**\n * Marks a tool as **human-in-the-loop**: the agent pauses and the UI (`render`)\n * supplies the result instead of code. Use it as the tool's `execute`:\n *\n * ```tsx\n * confirm: { execute: hitlTool(), render: (props) => <Confirm {...props} /> }\n * ```\n *\n * Like {@link defineToolkit}, it has **no runtime implementation**: a\n * `\"use generative\"` compiler (e.g. `@assistant-ui/next` or `@assistant-ui/vite`)\n * detects `execute: hitlTool()`, drops it, and stamps the tool `type: \"human\"`.\n * Reaching it at runtime means the module wasn't compiled (used outside a\n * `\"use generative\"` file), so it throws.\n */\nexport function hitlTool(): never {\n  throw new Error(\n    \"[assistant-ui] hitlTool() has no runtime implementation — it marks a \" +\n      \"human-in-the-loop tool and is stripped at build time by the \" +\n      \"use-generative compiler. Reaching it means this module was not compiled \" +\n      '(e.g. hitlTool() used outside a \"use generative\" file).',\n  );\n}\n\n/**\n * @deprecated Use {@link hitlTool}.\n */\nexport const hitl = hitlTool;\n"],"mappings":";;;;;;;;;;;;;;;AAcA,SAAgB,WAAkB;CAChC,MAAM,IAAI,MACR,oQAIF;AACF;;;;AAKA,MAAa,OAAO"}

package/dist/react/model-context/makeAssistantTool.d.ts

@@ -7,6 +7,10 @@ import { FC } from "react";
  *
  * Rendering the component registers its tool for the lifetime of that render
  * subtree.
+ *
+ * @deprecated Use a toolkit with `Tools({ toolkit })` and register it via
+ * `useAui({ tools: Tools({ toolkit }) })` instead. See
+ * https://assistant-ui.com/docs/migrations/toolkit-tools.
  */
 type AssistantTool = FC & {
   /** Tool definition registered by this component. */unstable_tool: AssistantToolProps<any, any>;
@@ -18,6 +22,10 @@ type AssistantTool = FC & {
  * rather than calling {@link useAssistantTool} directly.
  *
  * @param tool - Tool definition and name to register.
+ *
+ * @deprecated Use a toolkit with `Tools({ toolkit })` and register it via
+ * `useAui({ tools: Tools({ toolkit }) })` instead. See
+ * https://assistant-ui.com/docs/migrations/toolkit-tools.
  */
 declare const makeAssistantTool: <TArgs extends Record<string, unknown>, TResult>(tool: AssistantToolProps<TArgs, TResult>) => AssistantTool;
 //#endregion

package/dist/react/model-context/makeAssistantTool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"makeAssistantTool.d.ts","names":[],"sources":["../../../src/react/model-context/makeAssistantTool.ts"],"mappings":";;;;;;AASA;;;;KAAY,aAAA,GAAgB,EAAA;EAE1B,oDAAA,aAAA,EAAe,kBAAkB;AAAA;AAAA;AAWnC;;;;;;;AAXmC,cAWtB,iBAAA,iBACG,MAAA,4BAGd,IAAA,EAAM,kBAAA,CAAmB,KAAA,EAAO,OAAA,MAAQ,aAAA"}
+{"version":3,"file":"makeAssistantTool.d.ts","names":[],"sources":["../../../src/react/model-context/makeAssistantTool.ts"],"mappings":";;;;;;AAaA;;;;;;;;KAAY,aAAA,GAAgB,EAAA;EAiBf,oDAfX,aAAA,EAAe,kBAAkB;AAAA;;;;;;;;;;;;;cAetB,iBAAA,iBACG,MAAA,4BAGd,IAAA,EAAM,kBAAA,CAAmB,KAAA,EAAO,OAAA,MAAQ,aAAA"}

package/dist/react/model-context/makeAssistantTool.js

@@ -7,6 +7,10 @@ import { useAssistantTool } from "./useAssistantTool.js";
 * rather than calling {@link useAssistantTool} directly.
 *
 * @param tool - Tool definition and name to register.
+*
+* @deprecated Use a toolkit with `Tools({ toolkit })` and register it via
+* `useAui({ tools: Tools({ toolkit }) })` instead. See
+* https://assistant-ui.com/docs/migrations/toolkit-tools.
 */
 const makeAssistantTool = (tool) => {
 	const Tool = () => {

package/dist/react/model-context/makeAssistantTool.js.map

@@ -1 +1 @@
-{"version":3,"file":"makeAssistantTool.js","names":[],"sources":["../../../src/react/model-context/makeAssistantTool.ts"],"sourcesContent":["import type { FC } from \"react\";\nimport { type AssistantToolProps, useAssistantTool } from \"./useAssistantTool\";\n\n/**\n * Component returned by {@link makeAssistantTool}.\n *\n * Rendering the component registers its tool for the lifetime of that render\n * subtree.\n */\nexport type AssistantTool = FC & {\n  /** Tool definition registered by this component. */\n  unstable_tool: AssistantToolProps<any, any>;\n};\n\n/**\n * Creates a React component that registers an assistant tool when rendered.\n *\n * Use this when exporting reusable tool modules that can be included in JSX\n * rather than calling {@link useAssistantTool} directly.\n *\n * @param tool - Tool definition and name to register.\n */\nexport const makeAssistantTool = <\n  TArgs extends Record<string, unknown>,\n  TResult,\n>(\n  tool: AssistantToolProps<TArgs, TResult>,\n) => {\n  const Tool: AssistantTool = () => {\n    useAssistantTool(tool);\n    return null;\n  };\n  Tool.unstable_tool = tool;\n  return Tool;\n};\n"],"mappings":";;;;;;;;;;AAsBA,MAAa,qBAIX,SACG;CACH,MAAM,aAA4B;EAChC,iBAAiB,IAAI;EACrB,OAAO;CACT;CACA,KAAK,gBAAgB;CACrB,OAAO;AACT"}
+{"version":3,"file":"makeAssistantTool.js","names":[],"sources":["../../../src/react/model-context/makeAssistantTool.ts"],"sourcesContent":["import type { FC } from \"react\";\nimport { type AssistantToolProps, useAssistantTool } from \"./useAssistantTool\";\n\n/**\n * Component returned by {@link makeAssistantTool}.\n *\n * Rendering the component registers its tool for the lifetime of that render\n * subtree.\n *\n * @deprecated Use a toolkit with `Tools({ toolkit })` and register it via\n * `useAui({ tools: Tools({ toolkit }) })` instead. See\n * https://assistant-ui.com/docs/migrations/toolkit-tools.\n */\nexport type AssistantTool = FC & {\n  /** Tool definition registered by this component. */\n  unstable_tool: AssistantToolProps<any, any>;\n};\n\n/**\n * Creates a React component that registers an assistant tool when rendered.\n *\n * Use this when exporting reusable tool modules that can be included in JSX\n * rather than calling {@link useAssistantTool} directly.\n *\n * @param tool - Tool definition and name to register.\n *\n * @deprecated Use a toolkit with `Tools({ toolkit })` and register it via\n * `useAui({ tools: Tools({ toolkit }) })` instead. See\n * https://assistant-ui.com/docs/migrations/toolkit-tools.\n */\nexport const makeAssistantTool = <\n  TArgs extends Record<string, unknown>,\n  TResult,\n>(\n  tool: AssistantToolProps<TArgs, TResult>,\n) => {\n  const Tool: AssistantTool = () => {\n    useAssistantTool(tool);\n    return null;\n  };\n  Tool.unstable_tool = tool;\n  return Tool;\n};\n"],"mappings":";;;;;;;;;;;;;;AA8BA,MAAa,qBAIX,SACG;CACH,MAAM,aAA4B;EAChC,iBAAiB,IAAI;EACrB,OAAO;CACT;CACA,KAAK,gBAAgB;CACrB,OAAO;AACT"}

package/dist/react/model-context/makeAssistantToolUI.d.ts

@@ -7,6 +7,10 @@ import { FC } from "react";
  *
  * Rendering the component registers a renderer for matching tool-call message
  * parts.
+ *
+ * @deprecated Put `render`/`renderText` on the matching toolkit entry, or use
+ * `MessagePrimitive.Parts` inline tool render overrides for per-message UI.
+ * See https://assistant-ui.com/docs/migrations/toolkit-tools.
  */
 type AssistantToolUI = FC & {
   /** Tool renderer registered by this component. */unstable_tool: AssistantToolUIProps<any, any>;
@@ -18,6 +22,10 @@ type AssistantToolUI = FC & {
  * are registered elsewhere.
  *
  * @param tool - Tool renderer registration.
+ *
+ * @deprecated Put `render`/`renderText` on the matching toolkit entry, or use
+ * `MessagePrimitive.Parts` inline tool render overrides for per-message UI.
+ * See https://assistant-ui.com/docs/migrations/toolkit-tools.
  */
 declare const makeAssistantToolUI: <TArgs, TResult>(tool: AssistantToolUIProps<TArgs, TResult>) => AssistantToolUI;
 //#endregion

package/dist/react/model-context/makeAssistantToolUI.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"makeAssistantToolUI.d.ts","names":[],"sources":["../../../src/react/model-context/makeAssistantToolUI.ts"],"mappings":";;;;;;AAYA;;;;KAAY,eAAA,GAAkB,EAAA;EAE5B,kDAAA,aAAA,EAAe,oBAAoB;AAAA;AAAA;AAWrC;;;;;;;AAXqC,cAWxB,mBAAA,mBACX,IAAA,EAAM,oBAAA,CAAqB,KAAA,EAAO,OAAA,MAAQ,eAAA"}
+{"version":3,"file":"makeAssistantToolUI.d.ts","names":[],"sources":["../../../src/react/model-context/makeAssistantToolUI.ts"],"mappings":";;;;;;AAgBA;;;;;;;;KAAY,eAAA,GAAkB,EAAA;EAiBjB,kDAfX,aAAA,EAAe,oBAAoB;AAAA;;;;;;;;;;;;;cAexB,mBAAA,mBACX,IAAA,EAAM,oBAAA,CAAqB,KAAA,EAAO,OAAA,MAAQ,eAAA"}

package/dist/react/model-context/makeAssistantToolUI.js

@@ -7,6 +7,10 @@ import { useAssistantToolUI } from "./useAssistantToolUI.js";
 * are registered elsewhere.
 *
 * @param tool - Tool renderer registration.
+*
+* @deprecated Put `render`/`renderText` on the matching toolkit entry, or use
+* `MessagePrimitive.Parts` inline tool render overrides for per-message UI.
+* See https://assistant-ui.com/docs/migrations/toolkit-tools.
 */
 const makeAssistantToolUI = (tool) => {
 	const ToolUI = () => {