@assistant-ui/core 0.3.19 → 0.3.20

package/dist/utils/getGroupStatus.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"getGroupStatus.d.ts","names":[],"sources":["../../src/utils/getGroupStatus.ts"],"mappings":";;KAMK;WACM,QAAQ,oBAAoB;;cAG1B,iBACX,iBAAiB,kCAChB,oBAAoB;cAQV,kBACX,iBAAiB,+BACjB;UADiB"}
+{"version":3,"file":"getGroupStatus.d.ts","names":[],"sources":["../../src/utils/getGroupStatus.ts"],"mappings":";;KAMK;WACM,QAAQ,oBAAoB;;qBAG1B,iBACX,iBAAiB,kCAChB,oBAAoB;qBAQV,kBACX,iBAAiB,+BACjB;UADiB"}

package/dist/utils/id.d.ts

@@ -2,9 +2,8 @@
 /**
  * @deprecated This API is experimental and may change without notice.
  */
-declare const generateId: (size?: number) => string;
-declare const generateErrorMessageId: () => string;
-declare const isErrorMessageId: (id: string) => boolean;
+export declare const generateId: (size?: number) => string;
+export declare const generateErrorMessageId: () => string;
+export declare const isErrorMessageId: (id: string) => boolean;
 //#endregion
-export { generateErrorMessageId, generateId, isErrorMessageId };
 //# sourceMappingURL=id.d.ts.map

package/dist/utils/id.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"id.d.ts","names":[],"sources":["../../src/utils/id.ts"],"mappings":";;;;cAKa,aAAU;cAMV;cACA,mBAAoB"}
+{"version":3,"file":"id.d.ts","names":[],"sources":["../../src/utils/id.ts"],"mappings":";;;;qBAKa,aAAU;qBAMV;qBACA,mBAAoB"}

package/dist/utils/image-media-type.d.ts

@@ -7,7 +7,6 @@
  * one on the wire reads it from the bytes rather than assuming a format. Never
  * throws: the input is whatever a caller put on the part.
  */
-declare function detectImageMediaType(base64: string): string | undefined;
+export declare function detectImageMediaType(base64: string): string | undefined;
 //#endregion
-export { detectImageMediaType };
 //# sourceMappingURL=image-media-type.d.ts.map

package/dist/utils/image-media-type.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"image-media-type.d.ts","names":[],"sources":["../../src/utils/image-media-type.ts"],"mappings":";;;;;;;;;iBA4GgB,qBAAqB"}
+{"version":3,"file":"image-media-type.d.ts","names":[],"sources":["../../src/utils/image-media-type.ts"],"mappings":";;;;;;;;;wBA4GgB,qBAAqB"}

package/dist/utils/invoke-user-callback.d.ts

@@ -1,5 +1,4 @@
 //#region src/utils/invoke-user-callback.d.ts
-declare function invokeUserCallback<TArgs extends readonly unknown[]>(tag: string, name: string, callback: ((...args: TArgs) => unknown) | undefined, ...args: TArgs): void | Promise<void>;
+export declare function invokeUserCallback<TArgs extends readonly unknown[]>(tag: string, name: string, callback: ((...args: TArgs) => unknown) | undefined, ...args: TArgs): void | Promise<void>;
 //#endregion
-export { invokeUserCallback };
 //# sourceMappingURL=invoke-user-callback.d.ts.map

package/dist/utils/invoke-user-callback.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"invoke-user-callback.d.ts","names":[],"sources":["../../src/utils/invoke-user-callback.ts"],"mappings":";iBAAgB,mBAAmB,kCACjC,aACA,cACA,eAAe,MAAM,mCAClB,MAAM,eACD"}
+{"version":3,"file":"invoke-user-callback.d.ts","names":[],"sources":["../../src/utils/invoke-user-callback.ts"],"mappings":";wBAAgB,mBAAmB,kCACjC,aACA,cACA,eAAe,MAAM,mCAClB,MAAM,eACD"}

package/dist/utils/json/is-json-equal.d.ts

@@ -1,5 +1,4 @@
 //#region src/utils/json/is-json-equal.d.ts
-declare const isJSONValueEqual: (a: unknown, b: unknown) => boolean;
+export declare const isJSONValueEqual: (a: unknown, b: unknown) => boolean;
 //#endregion
-export { isJSONValueEqual };
 //# sourceMappingURL=is-json-equal.d.ts.map

package/dist/utils/json/is-json-equal.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"is-json-equal.d.ts","names":[],"sources":["../../../src/utils/json/is-json-equal.ts"],"mappings":";cA4Ca,mBAAoB,YAAY"}
+{"version":3,"file":"is-json-equal.d.ts","names":[],"sources":["../../../src/utils/json/is-json-equal.ts"],"mappings":";qBA4Ca,mBAAoB,YAAY"}

package/dist/utils/json/is-json.d.ts

@@ -1,9 +1,8 @@
 import { ReadonlyJSONArray, ReadonlyJSONObject, ReadonlyJSONValue } from "assistant-stream/utils";
 //#region src/utils/json/is-json.d.ts
-declare function isRecord(value: unknown): value is Record<string, unknown>;
-declare function isJSONValue(value: unknown, currentDepth?: number): value is ReadonlyJSONValue;
-declare function isJSONArray(value: unknown): value is ReadonlyJSONArray;
-declare function isJSONObject(value: unknown): value is ReadonlyJSONObject;
+export declare function isRecord(value: unknown): value is Record<string, unknown>;
+export declare function isJSONValue(value: unknown, currentDepth?: number): value is ReadonlyJSONValue;
+export declare function isJSONArray(value: unknown): value is ReadonlyJSONArray;
+export declare function isJSONObject(value: unknown): value is ReadonlyJSONObject;
 //#endregion
-export { isJSONArray, isJSONObject, isJSONValue, isRecord };
 //# sourceMappingURL=is-json.d.ts.map

package/dist/utils/json/is-json.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"is-json.d.ts","names":[],"sources":["../../../src/utils/json/is-json.ts"],"mappings":";;iBAMgB,SAAS,iBAAiB,SAAS;iBAInC,YACd,gBACA,wBACC,SAAS;iBAiCI,YAAY,iBAAiB,SAAS;iBAItC,aAAa,iBAAiB,SAAS"}
+{"version":3,"file":"is-json.d.ts","names":[],"sources":["../../../src/utils/json/is-json.ts"],"mappings":";;wBAMgB,SAAS,iBAAiB,SAAS;wBAInC,YACd,gBACA,wBACC,SAAS;wBAiCI,YAAY,iBAAiB,SAAS;wBAItC,aAAa,iBAAiB,SAAS"}

package/dist/utils/json/stable-stringify-tool-args.d.ts

@@ -1,7 +1,6 @@
 import { ReadonlyJSONObject } from "assistant-stream/utils";
 //#region src/utils/json/stable-stringify-tool-args.d.ts
-declare const trackToolArgsKeyOrder: (keyOrderCache: Map<string, Map<string, string[]>> | undefined, cacheKey: string, args: ReadonlyJSONObject) => void;
-declare const stableStringifyToolArgs: (keyOrderCache: Map<string, Map<string, string[]>> | undefined, cacheKey: string, args: ReadonlyJSONObject) => string;
+export declare const trackToolArgsKeyOrder: (keyOrderCache: Map<string, Map<string, string[]>> | undefined, cacheKey: string, args: ReadonlyJSONObject) => void;
+export declare const stableStringifyToolArgs: (keyOrderCache: Map<string, Map<string, string[]>> | undefined, cacheKey: string, args: ReadonlyJSONObject) => string;
 //#endregion
-export { stableStringifyToolArgs, trackToolArgsKeyOrder };
 //# sourceMappingURL=stable-stringify-tool-args.d.ts.map

package/dist/utils/json/stable-stringify-tool-args.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stable-stringify-tool-args.d.ts","names":[],"sources":["../../../src/utils/json/stable-stringify-tool-args.ts"],"mappings":";;cAgDa,wBACX,eAAe,YAAY,oCAC3B,kBACA,MAAM;cAMK,0BACX,eAAe,YAAY,oCAC3B,kBACA,MAAM"}
+{"version":3,"file":"stable-stringify-tool-args.d.ts","names":[],"sources":["../../../src/utils/json/stable-stringify-tool-args.ts"],"mappings":";;qBAgDa,wBACX,eAAe,YAAY,oCAC3B,kBACA,MAAM;qBAMK,0BACX,eAAe,YAAY,oCAC3B,kBACA,MAAM"}

package/dist/utils/normalizePartStatus.d.ts

@@ -1,9 +1,8 @@
 import { MessagePartStatus, ThreadAssistantMessagePart, ThreadMessage, ThreadUserMessagePart, ToolCallMessagePartStatus } from "../types/message.js";
 //#region src/utils/normalizePartStatus.d.ts
-declare const COMPLETE_STATUS: MessagePartStatus;
-declare const RUNNING_STATUS: MessagePartStatus;
-declare const normalizePartStatus: (part: ThreadUserMessagePart | ThreadAssistantMessagePart) => MessagePartStatus | undefined;
-declare const toMessagePartStatus: (message: ThreadMessage, partIndex: number, part: ThreadUserMessagePart | ThreadAssistantMessagePart) => ToolCallMessagePartStatus;
+export declare const COMPLETE_STATUS: MessagePartStatus;
+export declare const RUNNING_STATUS: MessagePartStatus;
+export declare const normalizePartStatus: (part: ThreadUserMessagePart | ThreadAssistantMessagePart) => MessagePartStatus | undefined;
+export declare const toMessagePartStatus: (message: ThreadMessage, partIndex: number, part: ThreadUserMessagePart | ThreadAssistantMessagePart) => ToolCallMessagePartStatus;
 //#endregion
-export { COMPLETE_STATUS, RUNNING_STATUS, normalizePartStatus, toMessagePartStatus };
 //# sourceMappingURL=normalizePartStatus.d.ts.map

package/dist/utils/normalizePartStatus.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"normalizePartStatus.d.ts","names":[],"sources":["../../src/utils/normalizePartStatus.ts"],"mappings":";;cASa,iBAAiB;cAGjB,gBAAgB;cAoBhB,sBACX,MAAM,wBAAwB,+BAC7B;cAwBU,sBACX,SAAS,eACT,mBACA,MAAM,wBAAwB,+BAC7B"}
+{"version":3,"file":"normalizePartStatus.d.ts","names":[],"sources":["../../src/utils/normalizePartStatus.ts"],"mappings":";;qBASa,iBAAiB;qBAGjB,gBAAgB;qBAoBhB,sBACX,MAAM,wBAAwB,+BAC7B;qBAwBU,sBACX,SAAS,eACT,mBACA,MAAM,wBAAwB,+BAC7B"}

package/dist/utils/notify-event-listeners.d.ts

@@ -1,6 +1,5 @@
 //#region src/utils/notify-event-listeners.d.ts
 type EventListener<T> = (payload: T) => unknown;
-declare const notifyEventListeners: <T>(listeners: Iterable<EventListener<T>>, payloadOrFactory: T | (() => T), errorContext: string) => void;
+export declare const notifyEventListeners: <T>(listeners: Iterable<EventListener<T>>, payloadOrFactory: T | (() => T), errorContext: string) => void;
 //#endregion
-export { notifyEventListeners };
 //# sourceMappingURL=notify-event-listeners.d.ts.map

package/dist/utils/notify-event-listeners.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"notify-event-listeners.d.ts","names":[],"sources":["../../src/utils/notify-event-listeners.ts"],"mappings":";KAAK,cAAc,MAAM,SAAS;cAGrB,uBAAwB,GACnC,WAAW,SAAS,cAAc,KAClC,kBAAkB,WAAW,IAC7B"}
+{"version":3,"file":"notify-event-listeners.d.ts","names":[],"sources":["../../src/utils/notify-event-listeners.ts"],"mappings":";KAAK,cAAc,MAAM,SAAS;qBAGrB,uBAAwB,GACnC,WAAW,SAAS,cAAc,KAClC,kBAAkB,WAAW,IAC7B"}

package/dist/utils/record.d.ts

@@ -5,7 +5,6 @@
  * `Object.prototype` setter instead of defining an own property, so a record
  * assigned into by a key the app does not control is built this way.
  */
-declare const nullProtoRecord: <T>(base?: Readonly<Record<string, T>>, ...rest: readonly (Readonly<Record<string, NoInfer<T>>> | undefined)[]) => Record<string, T>;
+export declare const nullProtoRecord: <T>(base?: Readonly<Record<string, T>>, ...rest: readonly (Readonly<Record<string, NoInfer<T>>> | undefined)[]) => Record<string, T>;
 //#endregion
-export { nullProtoRecord };
 //# sourceMappingURL=record.d.ts.map

package/dist/utils/record.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"record.d.ts","names":[],"sources":["../../src/utils/record.ts"],"mappings":";;;;;;;cAMa,kBAAmB,GAC9B,OAAO,SAAS,eAAe,QAC5B,gBAAgB,SAAS,eAAe,QAAQ,wBAClD,eAAe"}
+{"version":3,"file":"record.d.ts","names":[],"sources":["../../src/utils/record.ts"],"mappings":";;;;;;;qBAMa,kBAAmB,GAC9B,OAAO,SAAS,eAAe,QAC5B,gBAAgB,SAAS,eAAe,QAAQ,wBAClD,eAAe"}

package/dist/utils/silent-runtime-action.d.ts

@@ -1,6 +1,5 @@
 //#region src/utils/silent-runtime-action.d.ts
-declare const SILENT_RUNTIME_ACTION: unique symbol;
-declare const isSilentRuntimeAction: (error: unknown) => boolean;
+export declare const SILENT_RUNTIME_ACTION: unique symbol;
+export declare const isSilentRuntimeAction: (error: unknown) => boolean;
 //#endregion
-export { SILENT_RUNTIME_ACTION, isSilentRuntimeAction };
 //# sourceMappingURL=silent-runtime-action.d.ts.map

package/dist/utils/silent-runtime-action.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"silent-runtime-action.d.ts","names":[],"sources":["../../src/utils/silent-runtime-action.ts"],"mappings":";cAAa;cAIA,wBAAyB"}
+{"version":3,"file":"silent-runtime-action.d.ts","names":[],"sources":["../../src/utils/silent-runtime-action.ts"],"mappings":";qBAAa;qBAIA,wBAAyB"}

package/dist/utils/text.d.ts

@@ -1,6 +1,5 @@
 import { AppendMessage, ThreadMessage } from "../types/message.js";
 //#region src/utils/text.d.ts
-declare const getThreadMessageText: (message: ThreadMessage | AppendMessage) => string;
+export declare const getThreadMessageText: (message: ThreadMessage | AppendMessage) => string;
 //#endregion
-export { getThreadMessageText };
 //# sourceMappingURL=text.d.ts.map

package/dist/utils/text.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"text.d.ts","names":[],"sources":["../../src/utils/text.ts"],"mappings":";;cAGa,uBACX,SAAS,gBAAgB"}
+{"version":3,"file":"text.d.ts","names":[],"sources":["../../src/utils/text.ts"],"mappings":";;qBAGa,uBACX,SAAS,gBAAgB"}

package/dist/utils/wire-media.d.ts

@@ -8,7 +8,7 @@
  * for a url source and whenever inline bytes fail to sniff, so it fails exactly
  * where a fallback is needed.
  */
-declare function resolveImageMediaType(image: string, contentType?: string | undefined): string;
+export declare function resolveImageMediaType(image: string, contentType?: string | undefined): string;
 /**
  * The media type to declare for a `FileMessagePart`: its own `mimeType`, then
  * the payload's data URL declaration, then `application/octet-stream`.
@@ -16,7 +16,7 @@ declare function resolveImageMediaType(image: string, contentType?: string | und
  * `mimeType` is a plain string, and an adapter reading `file.type` on a file
  * the OS cannot type yields `""`, so the declared value is not always present.
  */
-declare function resolveFileMediaType(data: string, mimeType?: string | undefined): string;
+export declare function resolveFileMediaType(data: string, mimeType?: string | undefined): string;
 /**
  * A payload placed into a wire field that is contractually a url.
  *
@@ -29,7 +29,6 @@ declare function resolveFileMediaType(data: string, mimeType?: string | undefine
  * cannot be corrected without transcoding the payload and is authoritative for
  * the bytes it carries anyway.
  */
-declare function toMediaWireUrl(payload: string, mediaType: string): string;
+export declare function toMediaWireUrl(payload: string, mediaType: string): string;
 //#endregion
-export { resolveFileMediaType, resolveImageMediaType, toMediaWireUrl };
 //# sourceMappingURL=wire-media.d.ts.map

package/dist/utils/wire-media.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"wire-media.d.ts","names":[],"sources":["../../src/utils/wire-media.ts"],"mappings":";;;;;;;;;;iBAYgB,sBACd,eACA;;;;;;;;iBA+Bc,qBACd,cACA;;;;;;;;;;;;;iBAiBc,eAAe,iBAAiB"}
+{"version":3,"file":"wire-media.d.ts","names":[],"sources":["../../src/utils/wire-media.ts"],"mappings":";;;;;;;;;;wBAYgB,sBACd,eACA;;;;;;;;wBA+Bc,qBACd,cACA;;;;;;;;;;;;;wBAiBc,eAAe,iBAAiB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/core",
-  "version": "0.3.19",
+  "version": "0.3.20",
   "description": "Framework-agnostic core runtime for assistant-ui",
   "keywords": [
     "assistant",
@@ -59,15 +59,15 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "assistant-stream": "^0.3.43",
+    "assistant-stream": "^0.3.44",
     "nanoid": "^6.0.1"
   },
   "optionalDevDependencies": {
     "@assistant-ui/x-generative-compiler": ">=0.0.3"
   },
   "peerDependencies": {
-    "@assistant-ui/store": "^0.3.13",
-    "@assistant-ui/tap": "^0.9.17",
+    "@assistant-ui/store": "^0.3.14",
+    "@assistant-ui/tap": "^0.9.18",
     "@types/react": "*",
     "react": "^18 || ^19",
     "assistant-cloud": "^0.2.1"
@@ -84,17 +84,17 @@
     }
   },
   "devDependencies": {
-    "@assistant-ui/store": "0.3.13",
-    "@assistant-ui/tap": "0.9.17",
-    "@assistant-ui/vite": "0.0.17",
-    "@assistant-ui/x-buildutils": "0.0.27",
+    "@assistant-ui/store": "0.3.14",
+    "@assistant-ui/tap": "0.9.18",
+    "@assistant-ui/vite": "0.0.18",
+    "@assistant-ui/x-buildutils": "0.0.28",
     "@testing-library/react": "^16.3.3",
-    "@types/react": "^19.2.18",
-    "assistant-cloud": "0.2.1",
+    "@types/react": "^19.3.0",
+    "assistant-cloud": "0.2.2",
     "jsdom": "^30.0.1",
-    "react": "^19.2.8",
-    "react-dom": "^19.2.8",
-    "vitest": "^5.0.0"
+    "react": "^19.3.0",
+    "react-dom": "^19.3.0",
+    "vitest": "^5.0.1"
   },
   "publishConfig": {
     "access": "public",
@@ -112,6 +112,7 @@
   "scripts": {
     "build": "aui-build",
     "test": "vitest run",
+    "typecheck": "tsc --noEmit && tsc --noEmit -p tsconfig.test.json",
     "test:react-compiler": "vitest run --config vitest.react-compiler.config.ts",
     "test:watch": "vitest"
   }

package/src/adapters/attachment.ts

@@ -83,6 +83,13 @@ export const getFileDataURL = async (file: File): Promise<string> => {
   });
 };
 
+const escapeAttachmentName = (name: string) =>
+  name
+    .replaceAll("&", "&amp;")
+    .replaceAll('"', "&quot;")
+    .replaceAll("<", "&lt;")
+    .replaceAll(">", "&gt;");
+
 export class SimpleTextAttachmentAdapter implements AttachmentAdapter {
   public accept =
     "text/plain,text/html,text/markdown,text/csv,text/xml,text/json,application/json,text/css";
@@ -107,7 +114,7 @@ export class SimpleTextAttachmentAdapter implements AttachmentAdapter {
       content: [
         {
           type: "text",
-          text: `<attachment name=${attachment.name}>\n${await getFileText(attachment.file)}\n</attachment>`,
+          text: `<attachment name="${escapeAttachmentName(attachment.name)}">\n${await getFileText(attachment.file)}\n</attachment>`,
         },
       ],
     };

package/src/adapters/feedback.ts

@@ -3,6 +3,7 @@ import type { ThreadMessage } from "../types/message";
 type FeedbackAdapterFeedback = {
   message: ThreadMessage;
   type: "positive" | "negative";
+  comment?: string;
 };
 
 export type FeedbackAdapter = {

package/src/model-context/frame/provider.test.ts

@@ -2,9 +2,24 @@
  * @vitest-environment jsdom
  */
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import type { Tool } from "assistant-stream";
 import { AssistantFrameProvider } from "./provider";
 import { FRAME_MESSAGE_CHANNEL } from "./types";
 
+const createTool = <TResult>(
+  execute: NonNullable<
+    Extract<
+      Tool<Record<string, unknown>, TResult>,
+      { type: "frontend" }
+    >["execute"]
+  >,
+) =>
+  ({
+    type: "frontend",
+    parameters: { type: "object", properties: {} },
+    execute,
+  }) satisfies Tool<Record<string, unknown>, TResult>;
+
 describe("AssistantFrameProvider", () => {
   let messageHandler: ((event: MessageEvent) => void) | undefined;
   let parentWindow: Window;
@@ -80,7 +95,7 @@ describe("AssistantFrameProvider", () => {
       {
         getModelContext: () => ({
           tools: {
-            sensitiveTool: { execute },
+            sensitiveTool: createTool(execute),
           },
         }),
       },
@@ -104,7 +119,7 @@ describe("AssistantFrameProvider", () => {
     const execute = vi.fn(async () => "result");
     AssistantFrameProvider.addModelContextProvider({
       getModelContext: () => ({
-        tools: { sensitiveTool: { execute } },
+        tools: { sensitiveTool: createTool(execute) },
       }),
     });
 
@@ -163,7 +178,7 @@ describe("AssistantFrameProvider", () => {
     AssistantFrameProvider.addModelContextProvider({
       getModelContext: () => ({
         tools: {
-          sensitiveTool: { execute },
+          sensitiveTool: createTool(execute),
         },
       }),
     });
@@ -192,7 +207,7 @@ describe("AssistantFrameProvider", () => {
     });
     AssistantFrameProvider.addModelContextProvider({
       getModelContext: () => ({
-        tools: { sensitiveTool: { execute } },
+        tools: { sensitiveTool: createTool(execute) },
       }),
     });
 
@@ -233,7 +248,7 @@ describe("AssistantFrameProvider", () => {
     );
     AssistantFrameProvider.addModelContextProvider({
       getModelContext: () => ({
-        tools: { sensitiveTool: { execute } },
+        tools: { sensitiveTool: createTool(execute) },
       }),
     });
 
@@ -270,7 +285,9 @@ describe("AssistantFrameProvider", () => {
       },
     );
     AssistantFrameProvider.addModelContextProvider({
-      getModelContext: () => ({ tools: { sensitiveTool: { execute } } }),
+      getModelContext: () => ({
+        tools: { sensitiveTool: createTool(execute) },
+      }),
     });
 
     dispatchToolCall(window.location.origin, parentWindow, "tool-a");
@@ -298,7 +315,9 @@ describe("AssistantFrameProvider", () => {
       },
     );
     AssistantFrameProvider.addModelContextProvider({
-      getModelContext: () => ({ tools: { sensitiveTool: { execute } } }),
+      getModelContext: () => ({
+        tools: { sensitiveTool: createTool(execute) },
+      }),
     });
 
     dispatchToolCall(window.location.origin, parentWindow, "duplicate");
@@ -325,7 +344,9 @@ describe("AssistantFrameProvider", () => {
       },
     );
     AssistantFrameProvider.addModelContextProvider({
-      getModelContext: () => ({ tools: { sensitiveTool: { execute } } }),
+      getModelContext: () => ({
+        tools: { sensitiveTool: createTool(execute) },
+      }),
     });
 
     dispatchToolCall(window.location.origin);
@@ -371,7 +392,9 @@ describe("AssistantFrameProvider", () => {
       },
     );
     const removeProvider = AssistantFrameProvider.addModelContextProvider({
-      getModelContext: () => ({ tools: { sensitiveTool: { execute } } }),
+      getModelContext: () => ({
+        tools: { sensitiveTool: createTool(execute) },
+      }),
     });
 
     dispatchToolCall(window.location.origin);
@@ -417,7 +440,9 @@ describe("AssistantFrameProvider", () => {
       },
     );
     const provider = {
-      getModelContext: () => ({ tools: { sensitiveTool: { execute } } }),
+      getModelContext: () => ({
+        tools: { sensitiveTool: createTool(execute) },
+      }),
     };
     const removeFirst =
       AssistantFrameProvider.addModelContextProvider(provider);
@@ -438,7 +463,7 @@ describe("AssistantFrameProvider", () => {
     const shadowedExecute = vi.fn(async () => "shadowed");
     const removeShadowed = AssistantFrameProvider.addModelContextProvider({
       getModelContext: () => ({
-        tools: { sensitiveTool: { execute: shadowedExecute } },
+        tools: { sensitiveTool: createTool(shadowedExecute) },
       }),
     });
 
@@ -457,7 +482,7 @@ describe("AssistantFrameProvider", () => {
     );
     const removeOwner = AssistantFrameProvider.addModelContextProvider({
       getModelContext: () => ({
-        tools: { sensitiveTool: { execute } },
+        tools: { sensitiveTool: createTool(execute) },
       }),
     });
 
@@ -490,10 +515,10 @@ describe("AssistantFrameProvider", () => {
       },
     );
     const removeFirst = AssistantFrameProvider.addModelContextProvider({
-      getModelContext: () => ({ tools: { firstTool: { execute } } }),
+      getModelContext: () => ({ tools: { firstTool: createTool(execute) } }),
     });
     const removeSecond = AssistantFrameProvider.addModelContextProvider({
-      getModelContext: () => ({ tools: { secondTool: { execute } } }),
+      getModelContext: () => ({ tools: { secondTool: createTool(execute) } }),
     });
 
     dispatchToolCall(
@@ -540,7 +565,7 @@ describe("AssistantFrameProvider", () => {
       {
         getModelContext: () => ({
           tools: {
-            sensitiveTool: { execute },
+            sensitiveTool: createTool(execute),
           },
         }),
       },
@@ -567,7 +592,7 @@ describe("AssistantFrameProvider", () => {
       {
         getModelContext: () => ({
           tools: {
-            sensitiveTool: { execute },
+            sensitiveTool: createTool(execute),
           },
         }),
       },
@@ -609,7 +634,7 @@ describe("AssistantFrameProvider", () => {
       AssistantFrameProvider.addModelContextProvider(
         {
           getModelContext: () => ({
-            tools: { sensitiveTool: { execute } },
+            tools: { sensitiveTool: createTool(execute) },
           }),
           subscribe: () => {
             throw new Error("subscribe failed");
@@ -648,7 +673,7 @@ describe("AssistantFrameProvider", () => {
     expect(() =>
       AssistantFrameProvider.addModelContextProvider({
         getModelContext: () => ({
-          tools: { sensitiveTool: { execute } },
+          tools: { sensitiveTool: createTool(execute) },
         }),
         subscribe: () => {
           dispatchToolCall(window.location.origin);
@@ -678,7 +703,7 @@ describe("AssistantFrameProvider", () => {
     let subscriptionCount = 0;
     const provider = {
       getModelContext: () => ({
-        tools: { sensitiveTool: { execute } },
+        tools: { sensitiveTool: createTool(execute) },
       }),
       subscribe: () => {
         subscriptionCount += 1;
@@ -934,7 +959,7 @@ describe("AssistantFrameProvider", () => {
       {
         getModelContext: () => ({
           tools: {
-            sensitiveTool: { execute },
+            sensitiveTool: createTool(execute),
           },
         }),
       },

package/src/react/RuntimeAdapter.test.ts

@@ -5,13 +5,16 @@ import { RuntimeAdapter } from "../store";
 import { ExternalStoreRuntimeCore } from "../runtimes/internal";
 import { AssistantRuntimeImpl } from "../runtime/internal";
 import type { ExternalStoreAdapter } from "../runtimes/external-store/external-store-adapter";
+import type { AppendMessage } from "../types/message";
 
 type DemoMessage = { role: "user" | "assistant"; text: string };
 
 const createRuntime = (
   overrides?: Partial<ExternalStoreAdapter<DemoMessage>>,
 ) => {
-  const onNew = vi.fn(async () => {});
+  const onNew = vi.fn<(message: AppendMessage) => Promise<void>>(
+    async () => {},
+  );
   const adapter: ExternalStoreAdapter<DemoMessage> = {
     messages: [{ role: "user", text: "hi" }],
     convertMessage: (message) => ({