@assistant-ui/react 0.15.21 → 0.15.22

package/src/unstable/webmcp/convertTools.test.ts

@@ -7,7 +7,7 @@ import {
   toWebMcpTool,
 } from "./convertTools";
 
-type WeatherArgs = { city: string };
+type WeatherArgs = { city: string; unit?: "c" | "f" | undefined };
 type FrontendTool = Extract<Tool<WeatherArgs, unknown>, { type: "frontend" }>;
 
 const jsonSchema: FrontendTool["parameters"] = {
@@ -200,6 +200,25 @@ describe("toWebMcpTool schema validation", () => {
     expect(result).toEqual({ content: [text("ok")] });
   });
 
+  it("uses schema output for execution and model content", async () => {
+    const args = { city: " Paris ", extra: true };
+    const result = await descriptorFor({
+      parameters: z.object({
+        city: z.string().trim(),
+        unit: z.enum(["c", "f"]).default("c"),
+      }),
+      execute: ({ city, unit }) => `Weather in ${city} (${unit})`,
+      toModelOutput: ({ input, output }) => [
+        { type: "text", text: `${JSON.stringify(input)}: ${output}` },
+      ],
+    }).execute(args);
+
+    expect(result).toEqual({
+      content: [text('{"city":"Paris","unit":"c"}: Weather in Paris (c)')],
+    });
+    expect(args).toEqual({ city: " Paris ", extra: true });
+  });
+
   it("returns a validation error when the arguments do not validate", async () => {
     const execute = vi.fn(async () => "ok");
     const result = await zodTool({ execute }).execute({ city: 42 });
@@ -214,10 +233,11 @@ describe("toWebMcpTool schema validation", () => {
     const execute = vi.fn(async () => "ok");
     const result = await zodTool({
       execute,
-      experimental_onSchemaValidationError: async () => "recovered",
+      experimental_onSchemaValidationError: async (args) =>
+        `Invalid: ${JSON.stringify(args)}`,
     }).execute({ city: 42 });
     expect(execute).not.toHaveBeenCalled();
-    expect(result).toEqual({ content: [text("recovered")] });
+    expect(result).toEqual({ content: [text('Invalid: {"city":42}')] });
   });
 
   it("awaits a validator that returns a non-Promise thenable", async () => {

package/src/unstable/webmcp/convertTools.ts

@@ -101,16 +101,10 @@ export const toMcpContent = async (
   return response.isError ? { isError: true, content } : { content };
 };
 
-type StandardSchemaLike = {
-  "~standard": {
-    version: number;
-    validate: (
-      value: unknown,
-    ) =>
-      | { issues?: readonly unknown[] | undefined }
-      | Promise<{ issues?: readonly unknown[] | undefined }>;
-  };
-};
+type StandardSchemaLike = Extract<
+  NonNullable<Tool["parameters"]>,
+  { readonly "~standard": unknown }
+>;
 
 const isStandardSchema = (schema: unknown): schema is StandardSchemaLike =>
   typeof schema === "object" &&
@@ -193,7 +187,7 @@ export const toWebMcpTool = (
       return errorResult(`Tool "${name}" is no longer registered`);
     }
     const tool = getTool();
-    const args = (rawArgs ?? {}) as Record<string, unknown>;
+    let args = (rawArgs ?? {}) as Record<string, unknown>;
     const toolCallId = generateId();
     let cleanup: (() => void) | undefined;
     try {
@@ -231,6 +225,8 @@ export const toWebMcpTool = (
                 `Function parameter validation failed. ${JSON.stringify(issues)}`,
               );
             });
+        } else {
+          args = validation.value;
         }
       }
 

package/src/utils/getSelectionMessageId.test.ts

@@ -30,6 +30,34 @@ afterEach(() => {
 });
 
 describe("getSelectionMessageId", () => {
+  it("rejects a message outside the supplied thread root", () => {
+    document.body.innerHTML = `
+      <div id="first-thread">
+        <div data-message-id="message-1"><p id="first">first</p></div>
+      </div>
+      <div id="second-thread">
+        <div data-message-id="message-1"><p>second</p></div>
+      </div>
+    `;
+    const firstThread = document.querySelector("#first-thread");
+    const secondThread = document.querySelector("#second-thread");
+    assert(firstThread);
+    assert(secondThread);
+    const selection = selectText(textNode("#first"));
+
+    expect(getSelectionMessageId(selection, firstThread)).toBe("message-1");
+    expect(getSelectionMessageId(selection, secondThread)).toBeNull();
+  });
+
+  it("falls back to an unscoped selection when a declared root has no element", () => {
+    document.body.innerHTML = `
+      <div data-message-id="message-1"><p id="text">text</p></div>
+    `;
+    const selection = selectText(textNode("#text"));
+
+    expect(getSelectionMessageId(selection, null)).toBe("message-1");
+  });
+
   it("accepts selections anywhere in a message without quote regions", () => {
     document.body.innerHTML = `
       <div data-message-id="message-1">

package/src/utils/getSelectionMessageId.ts

@@ -54,7 +54,10 @@ const intersectsExcluded = (scope: Element, selection: Selection): boolean => {
   return false;
 };
 
-export const getSelectionMessageId = (selection: Selection): string | null => {
+export const getSelectionMessageId = (
+  selection: Selection,
+  root?: Element | null,
+): string | null => {
   const { anchorNode, focusNode } = selection;
   if (!anchorNode || !focusNode) return null;
 
@@ -64,6 +67,9 @@ export const getSelectionMessageId = (selection: Selection): string | null => {
   if (!anchorMessageElement || anchorMessageElement !== focusMessageElement) {
     return null;
   }
+  if (root && !root.contains(anchorMessageElement)) {
+    return null;
+  }
 
   const messageId = anchorMessageElement.getAttribute("data-message-id");
   if (!messageId) return null;

package/dist/tests/remote-thread-list-test-helpers.d.ts

@@ -1,5 +0,0 @@
-import { RemoteThreadListAdapter } from "../index.js";
-//#region src/tests/remote-thread-list-test-helpers.d.ts
-export declare function makeAdapter(overrides?: Partial<RemoteThreadListAdapter>): RemoteThreadListAdapter;
-//#endregion
-//# sourceMappingURL=remote-thread-list-test-helpers.d.ts.map

package/dist/tests/remote-thread-list-test-helpers.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"remote-thread-list-test-helpers.d.ts","names":[],"sources":["../../src/tests/remote-thread-list-test-helpers.ts"],"mappings":";;wBAGgB,YACd,YAAW,QAAQ,2BAClB"}

package/dist/tests/remote-thread-list-test-helpers.js

@@ -1,29 +0,0 @@
-import { vi } from "vitest";
-//#region src/tests/remote-thread-list-test-helpers.ts
-function makeAdapter(overrides = {}) {
-	return {
-		list: vi.fn(async () => ({ threads: [] })),
-		initialize: vi.fn(async (threadId) => ({
-			remoteId: threadId,
-			externalId: threadId
-		})),
-		rename: vi.fn(async () => {}),
-		archive: vi.fn(async () => {}),
-		unarchive: vi.fn(async () => {}),
-		delete: vi.fn(async () => {}),
-		generateTitle: vi.fn(async () => new ReadableStream({ start(c) {
-			c.close();
-		} })),
-		fetch: vi.fn(async (id) => ({
-			status: "regular",
-			remoteId: id,
-			externalId: id,
-			title: "Test"
-		})),
-		...overrides
-	};
-}
-//#endregion
-export { makeAdapter };
-
-//# sourceMappingURL=remote-thread-list-test-helpers.js.map

package/dist/tests/remote-thread-list-test-helpers.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"remote-thread-list-test-helpers.js","names":["vi","RemoteThreadListAdapter","makeAdapter","overrides","Partial","list","fn","threads","initialize","threadId","remoteId","externalId","rename","archive","unarchive","delete","generateTitle","ReadableStream","start","c","close","fetch","id","status","const","title"],"sources":["../../src/tests/remote-thread-list-test-helpers.ts"],"sourcesContent":["import { vi } from \"vitest\";\nimport type { RemoteThreadListAdapter } from \"../index\";\n\nexport function makeAdapter(\n  overrides: Partial<RemoteThreadListAdapter> = {},\n): RemoteThreadListAdapter {\n  return {\n    list: vi.fn(async () => ({ threads: [] })),\n    initialize: vi.fn(async (threadId: string) => ({\n      remoteId: threadId,\n      externalId: threadId,\n    })),\n    rename: vi.fn(async () => {}),\n    archive: vi.fn(async () => {}),\n    unarchive: vi.fn(async () => {}),\n    delete: vi.fn(async () => {}),\n    generateTitle: vi.fn(\n      async () =>\n        new ReadableStream({\n          start(c) {\n            c.close();\n          },\n        }) as never,\n    ),\n    fetch: vi.fn(async (id: string) => ({\n      status: \"regular\" as const,\n      remoteId: id,\n      externalId: id,\n      title: \"Test\",\n    })),\n    ...overrides,\n  };\n}\n"],"mappings":";;AAGA,SAAgBE,YACdC,YAA8C,CAAC,GACtB;CACzB,OAAO;EACLE,MAAML,GAAGM,GAAG,aAAa,EAAEC,SAAS,CAAA,EAAG,EAAE;EACzCC,YAAYR,GAAGM,GAAG,OAAOG,cAAsB;GAC7CC,UAAUD;GACVE,YAAYF;EACd,EAAE;EACFG,QAAQZ,GAAGM,GAAG,YAAY,CAAC,CAAC;EAC5BO,SAASb,GAAGM,GAAG,YAAY,CAAC,CAAC;EAC7BQ,WAAWd,GAAGM,GAAG,YAAY,CAAC,CAAC;EAC/BS,QAAQf,GAAGM,GAAG,YAAY,CAAC,CAAC;EAC5BU,eAAehB,GAAGM,GAChB,YACE,IAAIW,eAAe,EACjBC,MAAMC,GAAG;GACPA,EAAEC,MAAM;EACV,EACF,CAAC,CACL;EACAC,OAAOrB,GAAGM,GAAG,OAAOgB,QAAgB;GAClCC,QAAQ;GACRb,UAAUY;GACVX,YAAYW;GACZG,OAAO;EACT,EAAE;EACF,GAAGtB;CACL;AACF"}

package/dist/tests/setup.d.ts

@@ -1 +0,0 @@
-export {}

package/dist/tests/setup.js

@@ -1,6 +0,0 @@
-import { vi } from "vitest";
-//#region src/tests/setup.ts
-vi.setSystemTime(/* @__PURE__ */ new Date("2023-01-01"));
-//#endregion
-
-//# sourceMappingURL=setup.js.map

package/dist/tests/setup.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"setup.js","names":["vi","setSystemTime","Date"],"sources":["../../src/tests/setup.ts"],"sourcesContent":["// This file contains setup code for tests\nimport { vi } from \"vitest\";\n\n// Use Vitest's built-in to mock Date\nvi.setSystemTime(new Date(\"2023-01-01\"));\n"],"mappings":";;AAIAA,GAAGC,8BAAc,IAAIC,KAAK,YAAY,CAAC"}