@assistant-ui/core 0.3.15 → 0.3.17

package/dist/utils/normalizePartStatus.js.map

@@ -1 +1 @@
-{"version":3,"file":"normalizePartStatus.js","names":["MessagePartStatus","MessagePartStreamStatus","ThreadAssistantMessagePart","ThreadMessage","ThreadUserMessagePart","ToolCallMessagePartStatus","COMPLETE_STATUS","Object","freeze","type","RUNNING_STATUS","IncompleteMessagePartStatus","Extract","INCOMPLETE_STATUSES","Readonly","Record","cancelled","reason","length","other","error","normalizePartStatus","part","status","undefined","normalizedReason","toMessagePartStatus","message","partIndex","role","result","isLastPart","Math","max","content"],"sources":["../../src/utils/normalizePartStatus.ts"],"sourcesContent":["import type {\n  MessagePartStatus,\n  MessagePartStreamStatus,\n  ThreadAssistantMessagePart,\n  ThreadMessage,\n  ThreadUserMessagePart,\n  ToolCallMessagePartStatus,\n} from \"../types/message\";\n\nexport const COMPLETE_STATUS: MessagePartStatus = Object.freeze({\n  type: \"complete\",\n});\nexport const RUNNING_STATUS: MessagePartStatus = Object.freeze({\n  type: \"running\",\n});\ntype IncompleteMessagePartStatus = Extract<\n  MessagePartStreamStatus,\n  { readonly type: \"incomplete\" }\n>;\nconst INCOMPLETE_STATUSES: Readonly<\n  Record<IncompleteMessagePartStatus[\"reason\"], IncompleteMessagePartStatus>\n> = Object.freeze({\n  cancelled: Object.freeze({ type: \"incomplete\", reason: \"cancelled\" }),\n  length: Object.freeze({ type: \"incomplete\", reason: \"length\" }),\n  \"content-filter\": Object.freeze({\n    type: \"incomplete\",\n    reason: \"content-filter\",\n  }),\n  other: Object.freeze({ type: \"incomplete\", reason: \"other\" }),\n  error: Object.freeze({ type: \"incomplete\", reason: \"error\" }),\n});\n\nexport const normalizePartStatus = (\n  part: ThreadUserMessagePart | ThreadAssistantMessagePart,\n): MessagePartStatus | undefined => {\n  const status = (part as { readonly status?: unknown }).status;\n  if (!status || typeof status !== \"object\") return undefined;\n\n  const { type } = status as { readonly type?: unknown };\n  if (type === \"running\") return RUNNING_STATUS;\n  if (type === \"complete\") return COMPLETE_STATUS;\n  if (type !== \"incomplete\") return undefined;\n\n  const { reason } = status as {\n    readonly reason?: unknown;\n  };\n  const normalizedReason: IncompleteMessagePartStatus[\"reason\"] =\n    reason === \"cancelled\" ||\n    reason === \"length\" ||\n    reason === \"content-filter\" ||\n    reason === \"other\" ||\n    reason === \"error\"\n      ? reason\n      : \"other\";\n\n  return INCOMPLETE_STATUSES[normalizedReason];\n};\n\nexport const toMessagePartStatus = (\n  message: ThreadMessage,\n  partIndex: number,\n  part: ThreadUserMessagePart | ThreadAssistantMessagePart,\n): ToolCallMessagePartStatus => {\n  if (message.role !== \"assistant\") return COMPLETE_STATUS;\n\n  if (part.type === \"tool-call\") {\n    if (part.result === undefined) {\n      return message.status as ToolCallMessagePartStatus;\n    } else {\n      return COMPLETE_STATUS;\n    }\n  }\n\n  if (message.status.type === \"running\") {\n    const status = normalizePartStatus(part);\n    if (status) return status;\n  }\n\n  const isLastPart = partIndex === Math.max(0, message.content.length - 1);\n  if (message.status.type === \"requires-action\") return COMPLETE_STATUS;\n  return isLastPart ? (message.status as MessagePartStatus) : COMPLETE_STATUS;\n};\n"],"mappings":";AASA,MAAaM,kBAAqCC,OAAOC,OAAO,EAC9DC,MAAM,WACR,CAAC;AACD,MAAaC,iBAAoCH,OAAOC,OAAO,EAC7DC,MAAM,UACR,CAAC;AAKD,MAAMI,sBAEFN,OAAOC,OAAO;CAChBQ,WAAWT,OAAOC,OAAO;EAAEC,MAAM;EAAcQ,QAAQ;CAAY,CAAC;CACpEC,QAAQX,OAAOC,OAAO;EAAEC,MAAM;EAAcQ,QAAQ;CAAS,CAAC;CAC9D,kBAAkBV,OAAOC,OAAO;EAC9BC,MAAM;EACNQ,QAAQ;CACV,CAAC;CACDE,OAAOZ,OAAOC,OAAO;EAAEC,MAAM;EAAcQ,QAAQ;CAAQ,CAAC;CAC5DG,OAAOb,OAAOC,OAAO;EAAEC,MAAM;EAAcQ,QAAQ;CAAQ,CAAC;AAC9D,CAAC;AAED,MAAaI,uBACXC,SACkC;CAClC,MAAMC,SAAUD,KAAuCC;CACvD,IAAI,CAACA,UAAU,OAAOA,WAAW,UAAU,OAAOC,KAAAA;CAElD,MAAM,EAAEf,SAASc;CACjB,IAAId,SAAS,WAAW,OAAOC;CAC/B,IAAID,SAAS,YAAY,OAAOH;CAChC,IAAIG,SAAS,cAAc,OAAOe,KAAAA;CAElC,MAAM,EAAEP,WAAWM;CAYnB,OAAOV,oBARLI,WAAW,eACXA,WAAW,YACXA,WAAW,oBACXA,WAAW,WACXA,WAAW,UACPA,SACA;AAGR;AAEA,MAAaS,uBACXC,SACAC,WACAN,SAC8B;CAC9B,IAAIK,QAAQE,SAAS,aAAa,OAAOvB;CAEzC,IAAIgB,KAAKb,SAAS,aAAa;EAC7B,IAAIa,KAAKQ,WAAWN,KAAAA,GAClB,OAAOG,QAAQJ;OAEf,OAAOjB;CAEX;CAEA,IAAIqB,QAAQJ,OAAOd,SAAS,WAAW;EACrC,MAAMc,SAASF,oBAAoBC,IAAI;EACvC,IAAIC,QAAQ,OAAOA;CACrB;CAEA,MAAMQ,aAAaH,cAAcI,KAAKC,IAAI,GAAGN,QAAQO,QAAQhB,SAAS,CAAC;CACvE,IAAIS,QAAQJ,OAAOd,SAAS,mBAAmB,OAAOH;CACtD,OAAOyB,aAAcJ,QAAQJ,SAA+BjB;AAC9D"}
+{"version":3,"file":"normalizePartStatus.js","names":["MessagePartStatus","MessagePartStreamStatus","ThreadAssistantMessagePart","ThreadMessage","ThreadUserMessagePart","ToolCallMessagePartStatus","COMPLETE_STATUS","Object","freeze","type","RUNNING_STATUS","IncompleteMessagePartStatus","Extract","INCOMPLETE_STATUSES","Readonly","Record","cancelled","reason","length","other","error","normalizePartStatus","part","status","undefined","normalizedReason","toMessagePartStatus","message","partIndex","role","result","isLastPart","Math","max","content"],"sources":["../../src/utils/normalizePartStatus.ts"],"sourcesContent":["import type {\n  MessagePartStatus,\n  MessagePartStreamStatus,\n  ThreadAssistantMessagePart,\n  ThreadMessage,\n  ThreadUserMessagePart,\n  ToolCallMessagePartStatus,\n} from \"../types/message\";\n\nexport const COMPLETE_STATUS: MessagePartStatus = Object.freeze({\n  type: \"complete\",\n});\nexport const RUNNING_STATUS: MessagePartStatus = Object.freeze({\n  type: \"running\",\n});\ntype IncompleteMessagePartStatus = Extract<\n  MessagePartStreamStatus,\n  { readonly type: \"incomplete\" }\n>;\nconst INCOMPLETE_STATUSES: Readonly<\n  Record<IncompleteMessagePartStatus[\"reason\"], IncompleteMessagePartStatus>\n> = Object.freeze({\n  cancelled: Object.freeze({ type: \"incomplete\", reason: \"cancelled\" }),\n  length: Object.freeze({ type: \"incomplete\", reason: \"length\" }),\n  \"content-filter\": Object.freeze({\n    type: \"incomplete\",\n    reason: \"content-filter\",\n  }),\n  other: Object.freeze({ type: \"incomplete\", reason: \"other\" }),\n  error: Object.freeze({ type: \"incomplete\", reason: \"error\" }),\n});\n\nexport const normalizePartStatus = (\n  part: ThreadUserMessagePart | ThreadAssistantMessagePart,\n): MessagePartStatus | undefined => {\n  const status = (part as { readonly status?: unknown }).status;\n  if (!status || typeof status !== \"object\") return undefined;\n\n  const { type } = status as { readonly type?: unknown };\n  if (type === \"running\") return RUNNING_STATUS;\n  if (type === \"complete\") return COMPLETE_STATUS;\n  if (type !== \"incomplete\") return undefined;\n\n  const { reason } = status as {\n    readonly reason?: unknown;\n  };\n  const normalizedReason: IncompleteMessagePartStatus[\"reason\"] =\n    reason === \"cancelled\" ||\n    reason === \"length\" ||\n    reason === \"content-filter\" ||\n    reason === \"other\" ||\n    reason === \"error\"\n      ? reason\n      : \"other\";\n\n  return INCOMPLETE_STATUSES[normalizedReason];\n};\n\nexport const toMessagePartStatus = (\n  message: ThreadMessage,\n  partIndex: number,\n  part: ThreadUserMessagePart | ThreadAssistantMessagePart,\n): ToolCallMessagePartStatus => {\n  if (message.role !== \"assistant\") return COMPLETE_STATUS;\n\n  if (part.type === \"tool-call\") {\n    if (part.result === undefined) {\n      return message.status;\n    } else {\n      return COMPLETE_STATUS;\n    }\n  }\n\n  if (message.status.type === \"running\") {\n    const status = normalizePartStatus(part);\n    if (status) return status;\n  }\n\n  const isLastPart = partIndex === Math.max(0, message.content.length - 1);\n  if (message.status.type === \"requires-action\") return COMPLETE_STATUS;\n  return isLastPart ? (message.status as MessagePartStatus) : COMPLETE_STATUS;\n};\n"],"mappings":";AASA,MAAaM,kBAAqCC,OAAOC,OAAO,EAC9DC,MAAM,WACR,CAAC;AACD,MAAaC,iBAAoCH,OAAOC,OAAO,EAC7DC,MAAM,UACR,CAAC;AAKD,MAAMI,sBAEFN,OAAOC,OAAO;CAChBQ,WAAWT,OAAOC,OAAO;EAAEC,MAAM;EAAcQ,QAAQ;CAAY,CAAC;CACpEC,QAAQX,OAAOC,OAAO;EAAEC,MAAM;EAAcQ,QAAQ;CAAS,CAAC;CAC9D,kBAAkBV,OAAOC,OAAO;EAC9BC,MAAM;EACNQ,QAAQ;CACV,CAAC;CACDE,OAAOZ,OAAOC,OAAO;EAAEC,MAAM;EAAcQ,QAAQ;CAAQ,CAAC;CAC5DG,OAAOb,OAAOC,OAAO;EAAEC,MAAM;EAAcQ,QAAQ;CAAQ,CAAC;AAC9D,CAAC;AAED,MAAaI,uBACXC,SACkC;CAClC,MAAMC,SAAUD,KAAuCC;CACvD,IAAI,CAACA,UAAU,OAAOA,WAAW,UAAU,OAAOC,KAAAA;CAElD,MAAM,EAAEf,SAASc;CACjB,IAAId,SAAS,WAAW,OAAOC;CAC/B,IAAID,SAAS,YAAY,OAAOH;CAChC,IAAIG,SAAS,cAAc,OAAOe,KAAAA;CAElC,MAAM,EAAEP,WAAWM;CAYnB,OAAOV,oBARLI,WAAW,eACXA,WAAW,YACXA,WAAW,oBACXA,WAAW,WACXA,WAAW,UACPA,SACA;AAGR;AAEA,MAAaS,uBACXC,SACAC,WACAN,SAC8B;CAC9B,IAAIK,QAAQE,SAAS,aAAa,OAAOvB;CAEzC,IAAIgB,KAAKb,SAAS,aAAa;EAC7B,IAAIa,KAAKQ,WAAWN,KAAAA,GAClB,OAAOG,QAAQJ;OAEf,OAAOjB;CAEX;CAEA,IAAIqB,QAAQJ,OAAOd,SAAS,WAAW;EACrC,MAAMc,SAASF,oBAAoBC,IAAI;EACvC,IAAIC,QAAQ,OAAOA;CACrB;CAEA,MAAMQ,aAAaH,cAAcI,KAAKC,IAAI,GAAGN,QAAQO,QAAQhB,SAAS,CAAC;CACvE,IAAIS,QAAQJ,OAAOd,SAAS,mBAAmB,OAAOH;CACtD,OAAOyB,aAAcJ,QAAQJ,SAA+BjB;AAC9D"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/core",
-  "version": "0.3.15",
+  "version": "0.3.17",
   "description": "Framework-agnostic core runtime for assistant-ui",
   "keywords": [
     "assistant",
@@ -59,19 +59,18 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "assistant-stream": "^0.3.39",
+    "assistant-stream": "^0.3.41",
     "nanoid": "^6.0.1"
   },
   "optionalDevDependencies": {
     "@assistant-ui/x-generative-compiler": ">=0.0.3"
   },
   "peerDependencies": {
-    "@assistant-ui/store": "^0.3.0",
-    "@assistant-ui/tap": "^0.9.0",
+    "@assistant-ui/store": "^0.3.12",
+    "@assistant-ui/tap": "^0.9.16",
     "@types/react": "*",
     "react": "^18 || ^19",
-    "assistant-cloud": "^0.1.31",
-    "zustand": "^5.0.11"
+    "assistant-cloud": "^0.1.42"
   },
   "peerDependenciesMeta": {
     "@types/react": {
@@ -82,24 +81,20 @@
     },
     "react": {
       "optional": true
-    },
-    "zustand": {
-      "optional": true
     }
   },
   "devDependencies": {
-    "@testing-library/react": "^16.3.2",
+    "@assistant-ui/store": "0.3.12",
+    "@assistant-ui/tap": "0.9.16",
+    "@assistant-ui/vite": "0.0.16",
+    "@assistant-ui/x-buildutils": "0.0.25",
+    "@testing-library/react": "^16.3.3",
     "@types/react": "^19.2.18",
+    "assistant-cloud": "0.1.43",
     "jsdom": "^30.0.1",
     "react": "^19.2.8",
     "react-dom": "^19.2.8",
-    "vitest": "^4.1.10",
-    "zustand": "^5.0.15",
-    "@assistant-ui/store": "0.3.10",
-    "@assistant-ui/tap": "0.9.14",
-    "@assistant-ui/vite": "0.0.14",
-    "@assistant-ui/x-buildutils": "0.0.23",
-    "assistant-cloud": "0.1.41"
+    "vitest": "^4.1.11"
   },
   "publishConfig": {
     "access": "public",

package/src/index.ts

@@ -22,6 +22,7 @@ export type {
   GenerativeUINode,
   GenerativeUISpec,
   Unstable_AudioMessagePart,
+  ToolApprovalDisplay,
   ToolApprovalOption,
   ToolApprovalOptionKind,
   ToolApprovalResponse,
@@ -50,7 +51,11 @@ export type {
   AppendMessage,
 } from "./types/message";
 
-export { MCP_APP_URI_SCHEME, isMcpAppUri } from "./types/message";
+export {
+  MCP_APP_URI_SCHEME,
+  isMcpAppUri,
+  toolApprovalAcceptsText,
+} from "./types/message";
 
 export type {
   Attachment,

package/src/internal.ts

@@ -60,16 +60,36 @@ export { isJSONValue, isRecord } from "./utils/json/is-json";
 // outbound part conversion lives in one place.
 export {
   dataUrlMediaType,
+  type FilePartSource,
   httpUrlPattern,
   isParsableUrl,
   parseDataUrl,
+  resolveFilePartSource,
 } from "./utils/data-url";
+export { invokeUserCallback } from "./utils/invoke-user-callback";
 export { detectImageMediaType } from "./utils/image-media-type";
 export {
   resolveFileMediaType,
   resolveImageMediaType,
   toMediaWireUrl,
 } from "./utils/wire-media";
+export {
+  createToolCallCancellationStub,
+  scanPendingToolCalls,
+} from "./runtime/utils/pending-tool-calls";
+// Nested tool-call traversal, reused by framework adapters so the
+// ToolCallMessagePart.messages tree is walked one way.
+export {
+  iterateToolCallParts,
+  mapToolCallPartsDeep,
+  type ToolCallTreeEntry,
+  walkToolCallTree,
+} from "./runtime/utils/tool-call-tree";
+export {
+  createAbortableThreadLoad,
+  type AbortableThreadLoadPurpose,
+} from "./runtime/utils/abortable-thread-load";
+export { createCloudThreadListAdapterCreateFallback } from "./react/runtimes/cloud/createCloudThreadListAdapterCreateFallback";
 
 export * from "./runtime/internal";
 export * from "./runtimes/internal";

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

@@ -8,7 +8,9 @@ const executionContext = {
   human: async () => undefined,
 };
 
-const createHost = () => {
+const DEFAULT_ORIGIN = "https://host.example";
+
+const createHost = (targetOrigin?: string) => {
   let handleMessage: ((event: MessageEvent) => void) | undefined;
   const addEventListener = vi.fn(
     (_type: string, listener: EventListenerOrEventListenerObject) => {
@@ -16,15 +18,23 @@ const createHost = () => {
     },
   );
   const removeEventListener = vi.fn();
-  vi.stubGlobal("window", { addEventListener, removeEventListener });
+  vi.stubGlobal("window", {
+    addEventListener,
+    removeEventListener,
+    location: { origin: DEFAULT_ORIGIN },
+  });
 
   const postMessage = vi.fn();
   const iframeWindow = { postMessage } as unknown as Window;
-  const host = new AssistantFrameHost(iframeWindow);
+  const host = new AssistantFrameHost(iframeWindow, targetOrigin);
 
-  const dispatchMessage = (message: FrameMessage) =>
+  const dispatchMessage = (
+    message: FrameMessage,
+    origin = targetOrigin ?? DEFAULT_ORIGIN,
+  ) =>
     handleMessage?.({
       source: iframeWindow,
+      origin,
       data: {
         channel: FRAME_MESSAGE_CHANNEL,
         message,
@@ -67,6 +77,45 @@ afterEach(() => {
 });
 
 describe("AssistantFrameHost", () => {
+  it("defaults to the current origin", () => {
+    const { host, postMessage } = createHost();
+
+    expect(postMessage).toHaveBeenCalledWith(expect.anything(), DEFAULT_ORIGIN);
+
+    host.dispose();
+  });
+
+  it("ignores context updates from another origin", () => {
+    const { dispatchMessage, host } = createHost();
+
+    dispatchMessage(
+      {
+        type: "model-context-update",
+        context: { system: "untrusted instructions" },
+      },
+      "https://untrusted.example",
+    );
+
+    expect(host.getModelContext().system).toBeUndefined();
+    host.dispose();
+  });
+
+  it("allows an explicit wildcard origin", () => {
+    const { dispatchMessage, host, postMessage } = createHost("*");
+
+    dispatchMessage(
+      {
+        type: "model-context-update",
+        context: { system: "cross-origin instructions" },
+      },
+      "https://iframe.example",
+    );
+
+    expect(postMessage).toHaveBeenCalledWith(expect.anything(), "*");
+    expect(host.getModelContext().system).toBe("cross-origin instructions");
+    host.dispose();
+  });
+
   it("resolves tool calls from frame results", async () => {
     const { dispatchMessage, execute, getToolCallId, host } = createHost();
     const result = Promise.resolve(
@@ -84,6 +133,41 @@ describe("AssistantFrameHost", () => {
     host.dispose();
   });
 
+  it("rejects tool results that carry an empty error message", async () => {
+    const { dispatchMessage, execute, getToolCallId, host } = createHost();
+    const result = Promise.resolve(
+      execute({ query: "weather" }, executionContext),
+    );
+
+    dispatchMessage({
+      type: "tool-result",
+      id: getToolCallId(),
+      error: "",
+    });
+
+    await expect(result).rejects.toThrow();
+    expect(vi.getTimerCount()).toBe(0);
+    host.dispose();
+  });
+
+  it("resolves results from guests that serialize an absent error as null", async () => {
+    const { dispatchMessage, execute, getToolCallId, host } = createHost();
+    const result = Promise.resolve(
+      execute({ query: "weather" }, executionContext),
+    );
+
+    dispatchMessage({
+      type: "tool-result",
+      id: getToolCallId(),
+      result: "sunny",
+      error: null,
+    } as unknown as FrameMessage);
+
+    await expect(result).resolves.toBe("sunny");
+    expect(vi.getTimerCount()).toBe(0);
+    host.dispose();
+  });
+
   it("rejects pending tool calls when disposed", async () => {
     const { execute, getToolCallId, host, postMessage } = createHost();
     const result = Promise.resolve(execute({}, executionContext));
@@ -99,7 +183,7 @@ describe("AssistantFrameHost", () => {
         channel: FRAME_MESSAGE_CHANNEL,
         message: { type: "tool-cancel", id: toolCallId },
       },
-      "*",
+      DEFAULT_ORIGIN,
     );
     expect(vi.getTimerCount()).toBe(0);
   });
@@ -131,7 +215,7 @@ describe("AssistantFrameHost", () => {
         channel: FRAME_MESSAGE_CHANNEL,
         message: { type: "tool-cancel", id: toolCallId },
       },
-      "*",
+      DEFAULT_ORIGIN,
     );
     expect(vi.getTimerCount()).toBe(0);
     host.dispose();
@@ -153,7 +237,7 @@ describe("AssistantFrameHost", () => {
         channel: FRAME_MESSAGE_CHANNEL,
         message: { type: "tool-cancel", id: toolCallId },
       },
-      "*",
+      DEFAULT_ORIGIN,
     );
     expect(vi.getTimerCount()).toBe(0);
     host.dispose();

package/src/model-context/frame/host.ts

@@ -10,6 +10,8 @@ import {
   type SerializedTool,
 } from "./types";
 
+const getDefaultTargetOrigin = () => window.location.origin;
+
 /**
  * Deserializes tools from JSON Schema format back to Tool objects
  */
@@ -63,7 +65,10 @@ export class AssistantFrameHost implements ModelContextProvider {
   private _targetOrigin: string;
   private _disposed = false;
 
-  constructor(iframeWindow: Window, targetOrigin: string = "*") {
+  constructor(
+    iframeWindow: Window,
+    targetOrigin: string = getDefaultTargetOrigin(),
+  ) {
     this._iframeWindow = iframeWindow;
     this._targetOrigin = targetOrigin;
 
@@ -90,7 +95,7 @@ export class AssistantFrameHost implements ModelContextProvider {
       case "tool-result": {
         const pending = this._pendingRequests.get(message.id);
         if (pending) {
-          if (message.error) {
+          if (typeof message.error === "string") {
             pending.reject(new Error(message.error));
           } else {
             pending.resolve(message.result);