skybridge 0.0.0-dev.f76ccdc → 0.0.0-dev.fa642c4

package/dist/src/web/generate-helpers.js

@@ -0,0 +1,111 @@
+import { useCallTool, } from "./hooks/use-call-tool.js";
+import { useToolInfo } from "./hooks/use-tool-info.js";
+/**
+ * Creates typed versions of skybridge hooks with full type inference
+ * for tool names, inputs, and outputs.
+ *
+ * This is the recommended way to use skybridge hooks in your widgets.
+ * Set this up once in a dedicated file and export the typed hooks for use across your app.
+ *
+ * @typeParam ServerType - The type of your McpServer instance (use `typeof server`).
+ *                         Must be a server instance created with method chaining.
+ *                         TypeScript will validate that tools can be inferred from this type.
+ *
+ * @example
+ * ```typescript
+ * // server/src/index.ts
+ * const server = new McpServer({ name: "my-app", version: "1.0" }, {})
+ *   .widget("search-voyage", {}, {
+ *     inputSchema: { destination: z.string() },
+ *     outputSchema: { results: z.array(z.string()) },
+ *   }, async ({ destination }) => {
+ *     return { content: [{ type: "text", text: `Found trips to ${destination}` }] };
+ *   });
+ *
+ * export type AppType = typeof server;
+ * ```
+ *
+ * @example
+ * ```typescript
+ * // web/src/skybridge.ts (one-time setup)
+ * import type { AppType } from "../server";
+ * import { generateHelpers } from "skybridge/web";
+ *
+ * export const { useCallTool, useToolInfo } = generateHelpers<AppType>();
+ * ```
+ *
+ * @example
+ * ```typescript
+ * // web/src/widgets/search.tsx (usage)
+ * import { useCallTool, useToolInfo } from "../skybridge";
+ *
+ * export function SearchWidget() {
+ *   const { callTool, data } = useCallTool("search-voyage");
+ *   //                                      ^ autocomplete for tool names
+ *   callTool({ destination: "Spain" });
+ *   //         ^ autocomplete for input fields
+ *
+ *   const toolInfo = useToolInfo<"search-voyage">();
+ *   //                              ^ autocomplete for widget names
+ *   // toolInfo.input is typed based on widget input schema
+ *   // toolInfo.output.structuredContent is typed based on widget output schema
+ * }
+ * ```
+ */
+export function generateHelpers() {
+    return {
+        /**
+         * Typed version of `useCallTool` that provides autocomplete for tool names
+         * and type inference for inputs and outputs.
+         *
+         * @param name - The name of the widget to call. Autocompletes based on your server's widget registry.
+         * @returns A hook with typed `callTool` function and `data` property.
+         *
+         * @example
+         * ```typescript
+         * const { callTool, data, isPending } = useCallTool("search-voyage");
+         * // TypeScript knows callTool expects { destination: string }
+         * callTool({ destination: "Spain" });
+         *
+         * // data.structuredContent is typed based on your outputSchema
+         * if (data) {
+         *   console.log(data.structuredContent.results);
+         * }
+         * ```
+         */
+        useCallTool: (name) => {
+            return useCallTool(name);
+        },
+        /**
+         * Typed version of `useToolInfo` that provides autocomplete for widget names
+         * and type inference for inputs, outputs, and responseMetadata.
+         *
+         * @typeParam K - The name of the widget. Autocompletes based on your server's widget registry.
+         * @returns A discriminated union with `status: "pending" | "success"` that narrows correctly.
+         *
+         * @example
+         * ```typescript
+         * const toolInfo = useToolInfo<"search-voyage">();
+         * // toolInfo.input is typed as { destination: string; ... }
+         * // toolInfo.output is typed as { results: Array<...>; ... }
+         * // toolInfo.responseMetadata is typed based on _meta in callback return
+         * // toolInfo.status narrows correctly: "pending" | "success"
+         *
+         * if (toolInfo.isPending) {
+         *   // TypeScript knows output and responseMetadata are undefined here
+         *   console.log(toolInfo.input.destination);
+         * }
+         *
+         * if (toolInfo.isSuccess) {
+         *   // TypeScript knows output and responseMetadata are defined here
+         *   console.log(toolInfo.output.results);
+         *   console.log(toolInfo.responseMetadata);
+         * }
+         * ```
+         */
+        useToolInfo: () => {
+            return useToolInfo();
+        },
+    };
+}
+//# sourceMappingURL=generate-helpers.js.map

package/dist/src/web/generate-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-helpers.js","sourceRoot":"","sources":["../../../src/web/generate-helpers.ts"],"names":[],"mappings":"AAMA,OAAO,EAIL,WAAW,GACZ,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAkB,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAoBvE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmDG;AACH,MAAM,UAAU,eAAe;IAI7B,OAAO;QACL;;;;;;;;;;;;;;;;;;WAkBG;QACH,WAAW,EAAE,CACX,IAAc,EAId,EAAE;YACF,OAAO,WAAW,CAAC,IAAI,CAGtB,CAAC;QACJ,CAAC;QAED;;;;;;;;;;;;;;;;;;;;;;;;;;WA0BG;QACH,WAAW,EAAE,GAIX,EAAE;YACF,OAAO,WAAW,EAIjB,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/src/web/generate-helpers.test-d.d.ts

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

package/dist/src/web/generate-helpers.test-d.js

@@ -0,0 +1,180 @@
+import { expectTypeOf, test } from "vitest";
+import { createTestServer } from "../test/utils.js";
+import { generateHelpers } from "./generate-helpers.js";
+const server = createTestServer();
+test("InferTools extracts the tool registry type (widgets + registerTool)", () => {
+    expectTypeOf().toHaveProperty("search-voyage");
+    expectTypeOf().toHaveProperty("get-trip-details");
+    expectTypeOf().toHaveProperty("no-input-widget");
+    expectTypeOf().toHaveProperty("calculate-price");
+    expectTypeOf().toHaveProperty("inferred-output-widget");
+    expectTypeOf().toHaveProperty("inferred-tool");
+    expectTypeOf().toHaveProperty("widget-with-metadata");
+    expectTypeOf().toHaveProperty("tool-with-metadata");
+    expectTypeOf().toHaveProperty("widget-with-mixed-returns");
+});
+test("ToolNames returns a union of tool name literals (widgets + registerTool)", () => {
+    expectTypeOf().toEqualTypeOf();
+});
+test("ToolInput extracts the correct input type from Zod schema", () => {
+    expectTypeOf().toEqualTypeOf();
+    expectTypeOf().toEqualTypeOf();
+    expectTypeOf().toEqualTypeOf();
+});
+test("ToolOutput extracts the correct output type from callback's structuredContent", () => {
+    expectTypeOf().toEqualTypeOf();
+    expectTypeOf().toEqualTypeOf();
+    expectTypeOf().toEqualTypeOf();
+    expectTypeOf().toEqualTypeOf();
+});
+test("ToolOutput extracts the correct output type from callback (inferred)", () => {
+    expectTypeOf().toEqualTypeOf();
+    expectTypeOf().toEqualTypeOf();
+});
+test("generateHelpers provides autocomplete for tool names (widgets + registerTool)", () => {
+    const { useCallTool } = generateHelpers();
+    useCallTool("search-voyage");
+    useCallTool("get-trip-details");
+    useCallTool("no-input-widget");
+    useCallTool("calculate-price");
+    useCallTool("inferred-output-widget");
+    useCallTool("inferred-tool");
+    useCallTool("widget-with-metadata");
+    useCallTool("tool-with-metadata");
+    useCallTool("widget-with-mixed-returns");
+    // @ts-expect-error - "invalid-name" is not a valid tool name
+    useCallTool("invalid-name");
+});
+test("useCallTool returns correctly typed callTool function", () => {
+    const { useCallTool } = generateHelpers();
+    const { callTool } = useCallTool("search-voyage");
+    callTool({ destination: "Spain" });
+    callTool({ destination: "France", departureDate: "2024-06-01" });
+    callTool({ destination: "Italy", maxPrice: 1000 });
+    const { callTool: calculateTool } = useCallTool("calculate-price");
+    calculateTool({ tripId: "123", passengers: 2 });
+});
+test("callTool can be called without args for tools with no required inputs", () => {
+    const { useCallTool } = generateHelpers();
+    const { callTool, callToolAsync } = useCallTool("no-input-widget");
+    callTool();
+    callTool({});
+    callToolAsync();
+    callToolAsync({});
+});
+test("callTool requires args for tools with required inputs", () => {
+    const { useCallTool } = generateHelpers();
+    const { callTool } = useCallTool("search-voyage");
+    // @ts-expect-error - "destination" is required
+    callTool();
+    // @ts-expect-error - "destination" is required
+    callTool({});
+    // This should work
+    callTool({ destination: "Spain" });
+});
+test("callTool supports sideEffects for tools with required inputs", () => {
+    const { useCallTool } = generateHelpers();
+    const { callTool } = useCallTool("search-voyage");
+    callTool({ destination: "Spain" }, {
+        onSuccess: (response, args) => {
+            expectTypeOf(response.structuredContent.results).toBeArray();
+            expectTypeOf(args.destination).toBeString();
+        },
+        onError: (error, args) => {
+            expectTypeOf(error).toBeUnknown();
+            expectTypeOf(args.destination).toBeString();
+        },
+        onSettled: (response, error, args) => {
+            if (response) {
+                expectTypeOf(response.structuredContent.totalCount).toBeNumber();
+            }
+            expectTypeOf(error).toBeUnknown();
+            expectTypeOf(args.destination).toBeString();
+        },
+    });
+});
+test("callTool supports sideEffects for tools with no required inputs", () => {
+    const { useCallTool } = generateHelpers();
+    const { callTool } = useCallTool("no-input-widget");
+    callTool({
+        onSuccess: (response) => {
+            expectTypeOf(response).toHaveProperty("structuredContent");
+        },
+    });
+    callTool({}, {
+        onSuccess: (response) => {
+            expectTypeOf(response).toHaveProperty("structuredContent");
+        },
+    });
+});
+test("callToolAsync returns correctly typed promise", () => {
+    const { useCallTool } = generateHelpers();
+    const { callToolAsync: searchAsync } = useCallTool("search-voyage");
+    const searchPromise = searchAsync({ destination: "Spain" });
+    expectTypeOf(searchPromise).resolves.toHaveProperty("structuredContent");
+    const { callToolAsync: noInputAsync } = useCallTool("no-input-widget");
+    const noInputPromise = noInputAsync();
+    expectTypeOf(noInputPromise).resolves.toHaveProperty("structuredContent");
+});
+test("useCallTool returns correctly typed data", () => {
+    const { useCallTool } = generateHelpers();
+    const { data } = useCallTool("search-voyage");
+    if (data) {
+        expectTypeOf(data.structuredContent).toExtend();
+        expectTypeOf(data.structuredContent.results).toBeArray();
+        expectTypeOf(data.structuredContent.totalCount).toBeNumber();
+    }
+});
+test("useCallTool returns correctly typed data for callback-inferred outputs", () => {
+    const { useCallTool } = generateHelpers();
+    const { data: widgetData } = useCallTool("inferred-output-widget");
+    if (widgetData) {
+        expectTypeOf(widgetData.structuredContent).toExtend();
+    }
+    const { data: toolData } = useCallTool("inferred-tool");
+    if (toolData) {
+        expectTypeOf(toolData.structuredContent).toExtend();
+    }
+});
+test("generateHelpers provides autocomplete for tool names in useToolInfo (widgets + registerTool)", () => {
+    const { useToolInfo } = generateHelpers();
+    useToolInfo();
+    useToolInfo();
+    useToolInfo();
+    useToolInfo();
+    useToolInfo();
+    useToolInfo();
+    useToolInfo();
+    useToolInfo();
+    useToolInfo();
+    // @ts-expect-error - "invalid-name" is not a valid tool name
+    useToolInfo();
+});
+test("useToolInfo infers input and output types", () => {
+    const { useToolInfo } = generateHelpers();
+    const toolInfo = useToolInfo();
+    expectTypeOf(toolInfo.input).toExtend();
+    if (toolInfo.status === "success") {
+        expectTypeOf(toolInfo.output).toExtend();
+        expectTypeOf(toolInfo.output.results).toBeArray();
+        expectTypeOf(toolInfo.output.totalCount).toBeNumber();
+    }
+});
+test("ToolResponseMetadata extracts _meta type from callback", () => {
+    expectTypeOf().toEqualTypeOf();
+    expectTypeOf().toEqualTypeOf();
+    expectTypeOf().toBeUnknown();
+});
+test("useToolInfo infers responseMetadata type from generateHelpers", () => {
+    const { useToolInfo } = generateHelpers();
+    const toolInfo = useToolInfo();
+    if (toolInfo.isSuccess) {
+        expectTypeOf(toolInfo.responseMetadata.requestId).toBeString();
+        expectTypeOf(toolInfo.responseMetadata.timestamp).toBeNumber();
+        expectTypeOf(toolInfo.responseMetadata.cached).toBeBoolean();
+    }
+});
+test("ToolResponseMetadata extracts _meta from mixed return paths", () => {
+    expectTypeOf().toEqualTypeOf();
+});
+//# sourceMappingURL=generate-helpers.test-d.js.map

package/dist/src/web/generate-helpers.test-d.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-helpers.test-d.js","sourceRoot":"","sources":["../../../src/web/generate-helpers.test-d.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAQ5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;AAGlC,IAAI,CAAC,qEAAqE,EAAE,GAAG,EAAE;IAG/E,YAAY,EAAS,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;IACtD,YAAY,EAAS,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;IACzD,YAAY,EAAS,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;IACxD,YAAY,EAAS,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;IACxD,YAAY,EAAS,CAAC,cAAc,CAAC,wBAAwB,CAAC,CAAC;IAC/D,YAAY,EAAS,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;IACtD,YAAY,EAAS,CAAC,cAAc,CAAC,sBAAsB,CAAC,CAAC;IAC7D,YAAY,EAAS,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC;IAC3D,YAAY,EAAS,CAAC,cAAc,CAAC,2BAA2B,CAAC,CAAC;AACpE,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,0EAA0E,EAAE,GAAG,EAAE;IAGpF,YAAY,EAAS,CAAC,aAAa,EAUhC,CAAC;AACN,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2DAA2D,EAAE,GAAG,EAAE;IAGrE,YAAY,EAAe,CAAC,aAAa,EAIrC,CAAC;IAIL,YAAY,EAAgB,CAAC,aAAa,EAEtC,CAAC;IAIL,YAAY,EAAkB,CAAC,aAAa,EAGxC,CAAC;AACP,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+EAA+E,EAAE,GAAG,EAAE;IAGzF,YAAY,EAAgB,CAAC,aAAa,EAOtC,CAAC;IAIL,YAAY,EAAiB,CAAC,aAAa,EAIvC,CAAC;IAML,YAAY,EAAmB,CAAC,aAAa,EAGzC,CAAC;IAGL,YAAY,EAAiB,CAAC,aAAa,EAA0B,CAAC;AACxE,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sEAAsE,EAAE,GAAG,EAAE;IAGhF,YAAY,EAAwB,CAAC,aAAa,EAG9C,CAAC;IAIL,YAAY,EAAsB,CAAC,aAAa,EAG5C,CAAC;AACP,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+EAA+E,EAAE,GAAG,EAAE;IACzF,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IAEtD,WAAW,CAAC,eAAe,CAAC,CAAC;IAC7B,WAAW,CAAC,kBAAkB,CAAC,CAAC;IAChC,WAAW,CAAC,iBAAiB,CAAC,CAAC;IAC/B,WAAW,CAAC,iBAAiB,CAAC,CAAC;IAC/B,WAAW,CAAC,wBAAwB,CAAC,CAAC;IACtC,WAAW,CAAC,eAAe,CAAC,CAAC;IAC7B,WAAW,CAAC,sBAAsB,CAAC,CAAC;IACpC,WAAW,CAAC,oBAAoB,CAAC,CAAC;IAClC,WAAW,CAAC,2BAA2B,CAAC,CAAC;IAEzC,6DAA6D;IAC7D,WAAW,CAAC,cAAc,CAAC,CAAC;AAC9B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,uDAAuD,EAAE,GAAG,EAAE;IACjE,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IACtD,MAAM,EAAE,QAAQ,EAAE,GAAG,WAAW,CAAC,eAAe,CAAC,CAAC;IAElD,QAAQ,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;IACnC,QAAQ,CAAC,EAAE,WAAW,EAAE,QAAQ,EAAE,aAAa,EAAE,YAAY,EAAE,CAAC,CAAC;IACjE,QAAQ,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IAEnD,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,WAAW,CAAC,iBAAiB,CAAC,CAAC;IACnE,aAAa,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;AAClD,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,uEAAuE,EAAE,GAAG,EAAE;IACjF,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IACtD,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,WAAW,CAAC,iBAAiB,CAAC,CAAC;IAEnE,QAAQ,EAAE,CAAC;IAEX,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEb,aAAa,EAAE,CAAC;IAChB,aAAa,CAAC,EAAE,CAAC,CAAC;AACpB,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,uDAAuD,EAAE,GAAG,EAAE;IACjE,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IACtD,MAAM,EAAE,QAAQ,EAAE,GAAG,WAAW,CAAC,eAAe,CAAC,CAAC;IAElD,+CAA+C;IAC/C,QAAQ,EAAE,CAAC;IAEX,+CAA+C;IAC/C,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEb,mBAAmB;IACnB,QAAQ,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;AACrC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8DAA8D,EAAE,GAAG,EAAE;IACxE,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IACtD,MAAM,EAAE,QAAQ,EAAE,GAAG,WAAW,CAAC,eAAe,CAAC,CAAC;IAElD,QAAQ,CACN,EAAE,WAAW,EAAE,OAAO,EAAE,EACxB;QACE,SAAS,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,EAAE;YAC5B,YAAY,CAAC,QAAQ,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;YAC7D,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;QAC9C,CAAC;QACD,OAAO,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YACvB,YAAY,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YAClC,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;QAC9C,CAAC;QACD,SAAS,EAAE,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACnC,IAAI,QAAQ,EAAE,CAAC;gBACb,YAAY,CAAC,QAAQ,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC,UAAU,EAAE,CAAC;YACnE,CAAC;YACD,YAAY,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YAClC,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;QAC9C,CAAC;KACF,CACF,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,iEAAiE,EAAE,GAAG,EAAE;IAC3E,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IACtD,MAAM,EAAE,QAAQ,EAAE,GAAG,WAAW,CAAC,iBAAiB,CAAC,CAAC;IAEpD,QAAQ,CAAC;QACP,SAAS,EAAE,CAAC,QAAQ,EAAE,EAAE;YACtB,YAAY,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC;QAC7D,CAAC;KACF,CAAC,CAAC;IAEH,QAAQ,CACN,EAAE,EACF;QACE,SAAS,EAAE,CAAC,QAAQ,EAAE,EAAE;YACtB,YAAY,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC;QAC7D,CAAC;KACF,CACF,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+CAA+C,EAAE,GAAG,EAAE;IACzD,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IAEtD,MAAM,EAAE,aAAa,EAAE,WAAW,EAAE,GAAG,WAAW,CAAC,eAAe,CAAC,CAAC;IACpE,MAAM,aAAa,GAAG,WAAW,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5D,YAAY,CAAC,aAAa,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC;IAEzE,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,GAAG,WAAW,CAAC,iBAAiB,CAAC,CAAC;IACvE,MAAM,cAAc,GAAG,YAAY,EAAE,CAAC;IACtC,YAAY,CAAC,cAAc,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC;AAC5E,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,0CAA0C,EAAE,GAAG,EAAE;IACpD,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IACtD,MAAM,EAAE,IAAI,EAAE,GAAG,WAAW,CAAC,eAAe,CAAC,CAAC;IAE9C,IAAI,IAAI,EAAE,CAAC;QACT,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,QAAQ,EAOzC,CAAC;QAEL,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACzD,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC,UAAU,EAAE,CAAC;IAC/D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wEAAwE,EAAE,GAAG,EAAE;IAClF,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IAEtD,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,WAAW,CAAC,wBAAwB,CAAC,CAAC;IACnE,IAAI,UAAU,EAAE,CAAC;QACf,YAAY,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,QAAQ,EAG/C,CAAC;IACP,CAAC;IAED,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,WAAW,CAAC,eAAe,CAAC,CAAC;IACxD,IAAI,QAAQ,EAAE,CAAC;QACb,YAAY,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,QAAQ,EAG7C,CAAC;IACP,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8FAA8F,EAAE,GAAG,EAAE;IACxG,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IAEtD,WAAW,EAAmB,CAAC;IAC/B,WAAW,EAAsB,CAAC;IAClC,WAAW,EAAqB,CAAC;IACjC,WAAW,EAAqB,CAAC;IACjC,WAAW,EAA4B,CAAC;IACxC,WAAW,EAAmB,CAAC;IAC/B,WAAW,EAA0B,CAAC;IACtC,WAAW,EAAwB,CAAC;IACpC,WAAW,EAA+B,CAAC;IAE3C,6DAA6D;IAC7D,WAAW,EAAkB,CAAC;AAChC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE;IACrD,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IACtD,MAAM,QAAQ,GAAG,WAAW,EAAmB,CAAC;IAEhD,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,QAAQ,EAElC,CAAC;IAEJ,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAClC,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,EAEnC,CAAC;QACJ,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QAClD,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,UAAU,EAAE,CAAC;IACxD,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wDAAwD,EAAE,GAAG,EAAE;IAElE,YAAY,EAAc,CAAC,aAAa,EAIpC,CAAC;IAGL,YAAY,EAAY,CAAC,aAAa,EAGlC,CAAC;IAGL,YAAY,EAAc,CAAC,WAAW,EAAE,CAAC;AAC3C,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+DAA+D,EAAE,GAAG,EAAE;IACzE,MAAM,EAAE,WAAW,EAAE,GAAG,eAAe,EAAc,CAAC;IACtD,MAAM,QAAQ,GAAG,WAAW,EAA0B,CAAC;IAEvD,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;QACvB,YAAY,CAAC,QAAQ,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,UAAU,EAAE,CAAC;QAC/D,YAAY,CAAC,QAAQ,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,UAAU,EAAE,CAAC;QAC/D,YAAY,CAAC,QAAQ,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;IAC/D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,6DAA6D,EAAE,GAAG,EAAE;IAOvE,YAAY,EAAa,CAAC,aAAa,EAGnC,CAAC;AACP,CAAC,CAAC,CAAC"}

package/dist/src/web/generate-helpers.test.d.ts

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

package/dist/src/web/generate-helpers.test.js

@@ -0,0 +1,17 @@
+import { describe, expect, it } from "vitest";
+import { createMinimalTestServer } from "../test/utils.js";
+import { generateHelpers } from "./generate-helpers.js";
+const server = createMinimalTestServer();
+describe("generateHelpers", () => {
+    it("should return an object with useCallTool hook", () => {
+        const hooks = generateHelpers();
+        expect(hooks).toHaveProperty("useCallTool");
+        expect(typeof hooks.useCallTool).toBe("function");
+    });
+    it("should return an object with useToolInfo hook", () => {
+        const hooks = generateHelpers();
+        expect(hooks).toHaveProperty("useToolInfo");
+        expect(typeof hooks.useToolInfo).toBe("function");
+    });
+});
+//# sourceMappingURL=generate-helpers.test.js.map

package/dist/src/web/generate-helpers.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-helpers.test.js","sourceRoot":"","sources":["../../../src/web/generate-helpers.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,MAAM,MAAM,GAAG,uBAAuB,EAAE,CAAC;AAGzC,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,KAAK,GAAG,eAAe,EAAc,CAAC;QAC5C,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,KAAK,GAAG,eAAe,EAAc,CAAC;QAC5C,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/web/helpers/state.d.ts

@@ -0,0 +1,7 @@
+import { type SuperJSONResult } from "superjson";
+import type { UnknownObject } from "../types.js";
+export declare function filterWidgetContext<T extends UnknownObject>(state?: T | null): T | null;
+export declare function injectWidgetContext<T extends UnknownObject>(newState: T | null): T | null;
+export declare function serializeState(value: UnknownObject): unknown;
+export declare function deserializeState(value: SuperJSONResult): unknown;
+export declare function getInitialState<State extends UnknownObject>(defaultState?: State | (() => State)): State | null;

package/dist/src/web/helpers/state.js

@@ -0,0 +1,40 @@
+import superjson, {} from "superjson";
+import { WIDGET_CONTEXT_KEY } from "../data-llm.js";
+export function filterWidgetContext(state) {
+    if (state === null || state === undefined) {
+        return null;
+    }
+    const { [WIDGET_CONTEXT_KEY]: _, ...filteredState } = state;
+    return filteredState;
+}
+export function injectWidgetContext(newState) {
+    if (newState === null) {
+        return null;
+    }
+    const currentWindowState = window.openai?.widgetState;
+    if (currentWindowState !== null &&
+        currentWindowState !== undefined &&
+        WIDGET_CONTEXT_KEY in currentWindowState) {
+        return {
+            ...newState,
+            [WIDGET_CONTEXT_KEY]: currentWindowState[WIDGET_CONTEXT_KEY],
+        };
+    }
+    return newState;
+}
+export function serializeState(value) {
+    return superjson.parse(superjson.stringify(value)); // Strips functions
+}
+export function deserializeState(value) {
+    return superjson.deserialize(value);
+}
+export function getInitialState(defaultState) {
+    const widgetStateFromWindow = window.openai?.widgetState;
+    if (widgetStateFromWindow !== null && widgetStateFromWindow !== undefined) {
+        return filterWidgetContext(widgetStateFromWindow);
+    }
+    return typeof defaultState === "function"
+        ? defaultState()
+        : (defaultState ?? null);
+}
+//# sourceMappingURL=state.js.map

package/dist/src/web/helpers/state.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"state.js","sourceRoot":"","sources":["../../../../src/web/helpers/state.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,EAAE,EAAwB,MAAM,WAAW,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAGpD,MAAM,UAAU,mBAAmB,CACjC,KAAgB;IAEhB,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,EAAE,CAAC,kBAAkB,CAAC,EAAE,CAAC,EAAE,GAAG,aAAa,EAAE,GAAG,KAErD,CAAC;IAEF,OAAO,aAAkB,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,QAAkB;IAElB,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,kBAAkB,GAAG,MAAM,CAAC,MAAM,EAAE,WAG7B,CAAC;IAEd,IACE,kBAAkB,KAAK,IAAI;QAC3B,kBAAkB,KAAK,SAAS;QAChC,kBAAkB,IAAI,kBAAkB,EACxC,CAAC;QACD,OAAO;YACL,GAAG,QAAQ;YACX,CAAC,kBAAkB,CAAC,EAAE,kBAAkB,CAAC,kBAAkB,CAAC;SACxD,CAAC;IACT,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,KAAoB;IACjD,OAAO,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,mBAAmB;AACzE,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,KAAsB;IACrD,OAAO,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,YAAoC;IAEpC,MAAM,qBAAqB,GAAG,MAAM,CAAC,MAAM,EAAE,WAGhC,CAAC;IAEd,IAAI,qBAAqB,KAAK,IAAI,IAAI,qBAAqB,KAAK,SAAS,EAAE,CAAC;QAC1E,OAAO,mBAAmB,CAAC,qBAAqB,CAAC,CAAC;IACpD,CAAC;IAED,OAAO,OAAO,YAAY,KAAK,UAAU;QACvC,CAAC,CAAC,YAAY,EAAE;QAChB,CAAC,CAAC,CAAC,YAAY,IAAI,IAAI,CAAC,CAAC;AAC7B,CAAC"}

package/dist/src/web/helpers/state.test.d.ts

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

package/dist/src/web/helpers/state.test.js

@@ -0,0 +1,53 @@
+import { describe, expect, it } from "vitest";
+import { WIDGET_CONTEXT_KEY } from "../data-llm.js";
+import { filterWidgetContext, serializeState } from "./state.js";
+describe("state helpers", () => {
+    describe("filterWidgetContext", () => {
+        it("should return null when state is null", () => {
+            expect(filterWidgetContext(null)).toBe(null);
+        });
+        it("should return null when state is undefined", () => {
+            expect(filterWidgetContext(undefined)).toBe(null);
+        });
+        it("should correctly filter WIDGET_CONTEXT_KEY and preserve other properties", () => {
+            const stateWithContextAndOthers = {
+                a: 1,
+                b: "two",
+                c: { nested: true },
+                [WIDGET_CONTEXT_KEY]: "context",
+            };
+            const filteredWithContextAndOthers = filterWidgetContext(stateWithContextAndOthers);
+            expect(filteredWithContextAndOthers).toEqual({
+                a: 1,
+                b: "two",
+                c: { nested: true },
+            });
+            const stateNoContext = { count: 5, name: "test" };
+            const filteredNoContext = filterWidgetContext(stateNoContext);
+            expect(filteredNoContext).toEqual(stateNoContext);
+        });
+    });
+    describe("serializeState", () => {
+        it("should serialize plain objects", () => {
+            const array = [1, "two", { three: 3 }];
+            const date = new Date("2023-01-01T00:00:00Z");
+            const object = {
+                a: 1,
+                b: "test",
+                c: true,
+                array,
+                date,
+                function: () => "test",
+            };
+            const result = serializeState(object);
+            expect(result).toEqual({
+                a: 1,
+                b: "test",
+                c: true,
+                array: [1, "two", { three: 3 }],
+                date: new Date("2023-01-01T00:00:00.000Z"),
+            });
+        });
+    });
+});
+//# sourceMappingURL=state.test.js.map

package/dist/src/web/helpers/state.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"state.test.js","sourceRoot":"","sources":["../../../../src/web/helpers/state.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjE,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,MAAM,yBAAyB,GAAG;gBAChC,CAAC,EAAE,CAAC;gBACJ,CAAC,EAAE,KAAK;gBACR,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;gBACnB,CAAC,kBAAkB,CAAC,EAAE,SAAS;aAChC,CAAC;YACF,MAAM,4BAA4B,GAAG,mBAAmB,CACtD,yBAAyB,CAC1B,CAAC;YACF,MAAM,CAAC,4BAA4B,CAAC,CAAC,OAAO,CAAC;gBAC3C,CAAC,EAAE,CAAC;gBACJ,CAAC,EAAE,KAAK;gBACR,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;aACpB,CAAC,CAAC;YAEH,MAAM,cAAc,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YAClD,MAAM,iBAAiB,GAAG,mBAAmB,CAAC,cAAc,CAAC,CAAC;YAC9D,MAAM,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,KAAK,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YACvC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,sBAAsB,CAAC,CAAC;YAC9C,MAAM,MAAM,GAAG;gBACb,CAAC,EAAE,CAAC;gBACJ,CAAC,EAAE,MAAM;gBACT,CAAC,EAAE,IAAI;gBACP,KAAK;gBACL,IAAI;gBACJ,QAAQ,EAAE,GAAG,EAAE,CAAC,MAAM;aACvB,CAAC;YACF,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;YAEtC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,CAAC,EAAE,CAAC;gBACJ,CAAC,EAAE,MAAM;gBACT,CAAC,EAAE,IAAI;gBACP,KAAK,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;gBAC/B,IAAI,EAAE,IAAI,IAAI,CAAC,0BAA0B,CAAC;aAC3C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/web/hooks/index.d.ts

@@ -1,5 +1,6 @@
-export { useCallTool } from "./use-call-tool.js";
+export { type CallToolAsyncFn, type CallToolFn, type CallToolState, type SideEffects, useCallTool, } from "./use-call-tool.js";
 export { useDisplayMode } from "./use-display-mode.js";
+export { useFiles } from "./use-files.js";
 export { useLocale } from "./use-locale.js";
 export { useOpenExternal } from "./use-open-external.js";
 export { useOpenAiGlobal } from "./use-openai-global.js";
@@ -7,7 +8,5 @@ export { useRequestModal } from "./use-request-modal.js";
 export { useSendFollowUpMessage } from "./use-send-follow-up-message.js";
 export { useTheme } from "./use-theme.js";
 export { useToolInfo } from "./use-tool-info.js";
-export { useToolOutput } from "./use-tool-output.js";
-export { useToolResponseMetadata } from "./use-tool-response-metadata.js";
 export { useUserAgent } from "./use-user-agent.js";
 export { useWidgetState } from "./use-widget-state.js";

package/dist/src/web/hooks/index.js

@@ -1,5 +1,6 @@
-export { useCallTool } from "./use-call-tool.js";
+export { useCallTool, } from "./use-call-tool.js";
 export { useDisplayMode } from "./use-display-mode.js";
+export { useFiles } from "./use-files.js";
 export { useLocale } from "./use-locale.js";
 export { useOpenExternal } from "./use-open-external.js";
 export { useOpenAiGlobal } from "./use-openai-global.js";
@@ -7,8 +8,6 @@ export { useRequestModal } from "./use-request-modal.js";
 export { useSendFollowUpMessage } from "./use-send-follow-up-message.js";
 export { useTheme } from "./use-theme.js";
 export { useToolInfo } from "./use-tool-info.js";
-export { useToolOutput } from "./use-tool-output.js";
-export { useToolResponseMetadata } from "./use-tool-response-metadata.js";
 export { useUserAgent } from "./use-user-agent.js";
 export { useWidgetState } from "./use-widget-state.js";
 //# sourceMappingURL=index.js.map

package/dist/src/web/hooks/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/web/hooks/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AACzE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAC1E,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/web/hooks/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAKL,WAAW,GACZ,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AACzE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC"}

package/dist/src/web/hooks/use-call-tool.d.ts

@@ -1,10 +1,64 @@
-import type { CallToolArgs, CallToolResponse } from "../types.js";
-export declare const useCallTool: <ToolArgs extends CallToolArgs = null, ToolResponse extends CallToolResponse = CallToolResponse>(name: string) => {
-    callTool: (toolArgs: ToolArgs, sideEffects?: {
-        onSuccess?: (data: ToolResponse, toolArgs: ToolArgs) => void;
-        onError?: (error: unknown, toolArgs: ToolArgs) => void;
-    }) => void;
-    callToolAsync: (toolArgs: ToolArgs) => Promise<ToolResponse>;
+import type { CallToolArgs, CallToolResponse, HasRequiredKeys } from "../types.js";
+type CallToolIdleState = {
+    status: "idle";
+    isIdle: true;
+    isPending: false;
+    isSuccess: false;
+    isError: false;
+    data: undefined;
+    error: undefined;
+};
+type CallToolPendingState = {
+    status: "pending";
+    isIdle: false;
+    isPending: true;
+    isSuccess: false;
+    isError: false;
+    data: undefined;
+    error: undefined;
+};
+type CallToolSuccessState<TData extends CallToolResponse = CallToolResponse> = {
+    status: "success";
+    isIdle: false;
+    isPending: false;
+    isSuccess: true;
+    isError: false;
+    data: TData;
+    error: undefined;
+};
+type CallToolErrorState = {
+    status: "error";
+    isIdle: false;
+    isPending: false;
+    isSuccess: false;
+    isError: true;
+    data: undefined;
+    error: unknown;
+};
+export type CallToolState<TData extends CallToolResponse = CallToolResponse> = CallToolIdleState | CallToolPendingState | CallToolSuccessState<TData> | CallToolErrorState;
+export type SideEffects<ToolArgs, ToolResponse> = {
+    onSuccess?: (data: ToolResponse, toolArgs: ToolArgs) => void;
+    onError?: (error: unknown, toolArgs: ToolArgs) => void;
+    onSettled?: (data: ToolResponse | undefined, error: unknown | undefined, toolArgs: ToolArgs) => void;
+};
+type IsArgsOptional<T> = [T] extends [null] ? true : HasRequiredKeys<T> extends false ? true : false;
+export type CallToolFn<TArgs, TResponse> = IsArgsOptional<TArgs> extends true ? {
+    (): void;
+    (sideEffects: SideEffects<TArgs, TResponse>): void;
+    (args: TArgs): void;
+    (args: TArgs, sideEffects: SideEffects<TArgs, TResponse>): void;
+} : {
+    (args: TArgs): void;
+    (args: TArgs, sideEffects: SideEffects<TArgs, TResponse>): void;
+};
+export type CallToolAsyncFn<TArgs, TResponse> = IsArgsOptional<TArgs> extends true ? {
+    (): Promise<TResponse>;
+    (args: TArgs): Promise<TResponse>;
+} : (args: TArgs) => Promise<TResponse>;
+type ToolResponseSignature = Pick<CallToolResponse, "structuredContent" | "meta">;
+export declare const useCallTool: <ToolArgs extends CallToolArgs = null, ToolResponse extends Partial<ToolResponseSignature> = Record<string, never>>(name: string) => {
+    callTool: CallToolFn<ToolArgs, CallToolResponse & ToolResponse>;
+    callToolAsync: CallToolAsyncFn<ToolArgs, CallToolResponse & ToolResponse>;
     status: "idle";
     isIdle: true;
     isPending: false;
@@ -13,11 +67,8 @@ export declare const useCallTool: <ToolArgs extends CallToolArgs = null, ToolRes
     data: undefined;
     error: undefined;
 } | {
-    callTool: (toolArgs: ToolArgs, sideEffects?: {
-        onSuccess?: (data: ToolResponse, toolArgs: ToolArgs) => void;
-        onError?: (error: unknown, toolArgs: ToolArgs) => void;
-    }) => void;
-    callToolAsync: (toolArgs: ToolArgs) => Promise<ToolResponse>;
+    callTool: CallToolFn<ToolArgs, CallToolResponse & ToolResponse>;
+    callToolAsync: CallToolAsyncFn<ToolArgs, CallToolResponse & ToolResponse>;
     status: "pending";
     isIdle: false;
     isPending: true;
@@ -26,11 +77,8 @@ export declare const useCallTool: <ToolArgs extends CallToolArgs = null, ToolRes
     data: undefined;
     error: undefined;
 } | {
-    callTool: (toolArgs: ToolArgs, sideEffects?: {
-        onSuccess?: (data: ToolResponse, toolArgs: ToolArgs) => void;
-        onError?: (error: unknown, toolArgs: ToolArgs) => void;
-    }) => void;
-    callToolAsync: (toolArgs: ToolArgs) => Promise<ToolResponse>;
+    callTool: CallToolFn<ToolArgs, CallToolResponse & ToolResponse>;
+    callToolAsync: CallToolAsyncFn<ToolArgs, CallToolResponse & ToolResponse>;
     status: "error";
     isIdle: false;
     isPending: false;
@@ -39,16 +87,14 @@ export declare const useCallTool: <ToolArgs extends CallToolArgs = null, ToolRes
     data: undefined;
     error: unknown;
 } | {
-    callTool: (toolArgs: ToolArgs, sideEffects?: {
-        onSuccess?: (data: ToolResponse, toolArgs: ToolArgs) => void;
-        onError?: (error: unknown, toolArgs: ToolArgs) => void;
-    }) => void;
-    callToolAsync: (toolArgs: ToolArgs) => Promise<ToolResponse>;
+    callTool: CallToolFn<ToolArgs, CallToolResponse & ToolResponse>;
+    callToolAsync: CallToolAsyncFn<ToolArgs, CallToolResponse & ToolResponse>;
     status: "success";
     isIdle: false;
     isPending: false;
     isSuccess: true;
     isError: false;
-    data: ToolResponse;
+    data: CallToolResponse & ToolResponse;
     error: undefined;
 };
+export {};