skybridge 0.0.0-dev.d8b52a9 → 0.0.0-dev.dc4ed16

package/dist/src/web/data-llm.js

@@ -0,0 +1,68 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+import { createContext, useContext, useEffect, useId, } from "react";
+export const WIDGET_CONTEXT_KEY = "__widget_context";
+const nodes = new Map();
+function setNode(node) {
+    nodes.set(node.id, node);
+    onChange();
+}
+function removeNode(id) {
+    nodes.delete(id);
+    onChange();
+}
+function onChange() {
+    const description = getLLMDescriptionString();
+    window.openai.setWidgetState({
+        ...window.openai.widgetState,
+        [WIDGET_CONTEXT_KEY]: description,
+    });
+}
+const ParentIdContext = createContext(null);
+export function DataLLM({ content, children }) {
+    const parentId = useContext(ParentIdContext);
+    const id = useId();
+    useEffect(() => {
+        if (content) {
+            setNode({
+                id,
+                parentId,
+                content,
+            });
+        }
+        else {
+            removeNode(id);
+        }
+        return () => {
+            removeNode(id);
+        };
+    }, [id, parentId, content]);
+    return (_jsx(ParentIdContext.Provider, { value: id, children: children }));
+}
+function getLLMDescriptionString() {
+    const byParent = new Map();
+    for (const node of Array.from(nodes.values())) {
+        const key = node.parentId ?? null;
+        if (!byParent.has(key))
+            byParent.set(key, []);
+        byParent.get(key).push(node);
+    }
+    for (const list of byParent.values()) {
+        list.sort((a, b) => a.id.localeCompare(b.id));
+    }
+    const lines = [];
+    function traverseTree(parentId, depth) {
+        const children = byParent.get(parentId);
+        if (!children)
+            return;
+        for (const child of children) {
+            if (child.content && child.content.trim()) {
+                const indent = "  ".repeat(depth);
+                lines.push(`${indent}- ${child.content.trim()}`);
+            }
+            traverseTree(child.id, depth + 1);
+        }
+    }
+    traverseTree(null, 0);
+    return lines.join("\n");
+}
+//# sourceMappingURL=data-llm.js.map

package/dist/src/web/data-llm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"data-llm.js","sourceRoot":"","sources":["../../../src/web/data-llm.tsx"],"names":[],"mappings":";AAAA,OAAO,EACL,aAAa,EACb,UAAU,EACV,SAAS,EACT,KAAK,GAEN,MAAM,OAAO,CAAC;AAUf,MAAM,CAAC,MAAM,kBAAkB,GAAG,kBAA2B,CAAC;AAE9D,MAAM,KAAK,GAAG,IAAI,GAAG,EAAuB,CAAC;AAE7C,SAAS,OAAO,CAAC,IAAiB;IAChC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;IACzB,QAAQ,EAAE,CAAC;AACb,CAAC;AAED,SAAS,UAAU,CAAC,EAAU;IAC5B,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACjB,QAAQ,EAAE,CAAC;AACb,CAAC;AAED,SAAS,QAAQ;IACf,MAAM,WAAW,GAAG,uBAAuB,EAAE,CAAC;IAC9C,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC;QAC3B,GAAG,MAAM,CAAC,MAAM,CAAC,WAAW;QAC5B,CAAC,kBAAkB,CAAC,EAAE,WAAW;KAClC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,eAAe,GAAG,aAAa,CAAgB,IAAI,CAAC,CAAC;AAO3D,MAAM,UAAU,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAgB;IACzD,MAAM,QAAQ,GAAG,UAAU,CAAC,eAAe,CAAC,CAAC;IAC7C,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IAEnB,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC;gBACN,EAAE;gBACF,QAAQ;gBACR,OAAO;aACR,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,UAAU,CAAC,EAAE,CAAC,CAAC;QACjB,CAAC;QAED,OAAO,GAAG,EAAE;YACV,UAAU,CAAC,EAAE,CAAC,CAAC;QACjB,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;IAE5B,OAAO,CACL,KAAC,eAAe,CAAC,QAAQ,IAAC,KAAK,EAAE,EAAE,YAAG,QAAQ,GAA4B,CAC3E,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB;IAC9B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAgC,CAAC;IACzD,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;QAC9C,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC;QAClC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC9C,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;QACrC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,SAAS,YAAY,CAAC,QAAuB,EAAE,KAAa;QAC1D,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ;YAAE,OAAO;QAEtB,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC7B,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAClC,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,KAAK,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACnD,CAAC;YACD,YAAY,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAEtB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}

package/dist/src/web/data-llm.test.js

@@ -0,0 +1,76 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { render } from "@testing-library/react";
+import { afterEach, beforeEach, describe, expect, it, vi, } from "vitest";
+import { DataLLM } from "./data-llm.js";
+describe("DataLLM", () => {
+    let OpenaiMock;
+    beforeEach(() => {
+        OpenaiMock = {
+            widgetState: {},
+            setWidgetState: vi.fn(),
+        };
+        // Use Object.defineProperty to ensure it persists
+        Object.defineProperty(globalThis, "openai", {
+            value: OpenaiMock,
+            writable: true,
+            configurable: true,
+        });
+        // Also set on window for browser-like environment
+        if (typeof window !== "undefined") {
+            Object.defineProperty(window, "openai", {
+                value: OpenaiMock,
+                writable: true,
+                configurable: true,
+            });
+        }
+        vi.stubGlobal("openai", OpenaiMock);
+    });
+    afterEach(() => {
+        vi.resetAllMocks();
+        // Keep the mock available for React cleanup, but reset it
+        if (typeof window !== "undefined" && window.openai) {
+            window.openai.setWidgetState = vi.fn();
+            window.openai.widgetState = {};
+        }
+    });
+    it("should register a node with content and call setWidgetState", () => {
+        render(_jsx(DataLLM, { content: "Test content", children: _jsx("div", { children: "Child" }) }));
+        expect(OpenaiMock.setWidgetState).toHaveBeenCalled();
+        const callArgs = OpenaiMock.setWidgetState.mock.calls[0]?.[0];
+        expect(callArgs).toHaveProperty("__widget_context");
+        expect(callArgs?.__widget_context).toContain("- Test content");
+    });
+    it("should preserve existing widgetState when updating context", () => {
+        OpenaiMock.widgetState = { existingKey: "existingValue" };
+        render(_jsx(DataLLM, { content: "Test content", children: _jsx("div", { children: "Child" }) }));
+        const callArgs = OpenaiMock.setWidgetState.mock.calls[0]?.[0];
+        expect(callArgs).toHaveProperty("existingKey", "existingValue");
+        expect(callArgs).toHaveProperty("__widget_context");
+    });
+    it("should handle deeply nested DataLLM components", () => {
+        render(_jsxs(DataLLM, { content: "Level 1", children: [_jsx(DataLLM, { content: "Level 2A" }), _jsx(DataLLM, { content: "Level 2B", children: _jsx(DataLLM, { content: "Level 3", children: _jsx("div", { children: "Content" }) }) })] }));
+        const callArgs = OpenaiMock.setWidgetState.mock.calls[OpenaiMock.setWidgetState.mock.calls.length - 1]?.[0];
+        const context = callArgs?.__widget_context;
+        expect(context).toContain("- Level 1");
+        expect(context).toContain("  - Level 2A");
+        expect(context).toContain("  - Level 2B");
+        expect(context).toContain("    - Level 3");
+    });
+    it("should update context when content changes", () => {
+        const { rerender } = render(_jsx(DataLLM, { content: "Initial content", children: _jsx("div", { children: "Child" }) }));
+        const initialCalls = OpenaiMock.setWidgetState.mock.calls.length;
+        rerender(_jsx(DataLLM, { content: "Updated content", children: _jsx("div", { children: "Child" }) }));
+        expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(initialCalls);
+        const lastCallArgs = OpenaiMock.setWidgetState.mock.calls[OpenaiMock.setWidgetState.mock.calls.length - 1]?.[0];
+        expect(lastCallArgs?.__widget_context).toContain("- Updated content");
+    });
+    it("should remove node and update context when component unmounts", () => {
+        const { unmount } = render(_jsx(DataLLM, { content: "Content to remove", children: _jsx("div", { children: "Child" }) }));
+        const callsBeforeUnmount = OpenaiMock.setWidgetState.mock.calls.length;
+        unmount();
+        expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(callsBeforeUnmount);
+        const lastCallArgs = OpenaiMock.setWidgetState.mock.calls[OpenaiMock.setWidgetState.mock.calls.length - 1]?.[0];
+        expect(lastCallArgs?.__widget_context).not.toContain("Content to remove");
+    });
+});
+//# sourceMappingURL=data-llm.test.js.map

package/dist/src/web/data-llm.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"data-llm.test.js","sourceRoot":"","sources":["../../../src/web/data-llm.test.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAChD,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,EAAE,EACF,EAAE,GAEH,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAExC,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;IACvB,IAAI,UAA0D,CAAC;IAE/D,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,WAAW,EAAE,EAAE;YACf,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;SACxB,CAAC;QACF,kDAAkD;QAClD,MAAM,CAAC,cAAc,CAAC,UAAU,EAAE,QAAQ,EAAE;YAC1C,KAAK,EAAE,UAAU;YACjB,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QACH,kDAAkD;QAClD,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;YAClC,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,QAAQ,EAAE;gBACtC,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE,IAAI;gBACd,YAAY,EAAE,IAAI;aACnB,CAAC,CAAC;QACL,CAAC;QACD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,0DAA0D;QAC1D,IAAI,OAAO,MAAM,KAAK,WAAW,IAAK,MAAc,CAAC,MAAM,EAAE,CAAC;YAC3D,MAAc,CAAC,MAAM,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/C,MAAc,CAAC,MAAM,CAAC,WAAW,GAAG,EAAE,CAAC;QAC1C,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;QAEF,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACrD,MAAM,QAAQ,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;QACpD,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,UAAU,CAAC,WAAW,GAAG,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC;QAE1D,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;QAEF,MAAM,QAAQ,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;QAChE,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,CACJ,MAAC,OAAO,IAAC,OAAO,EAAC,SAAS,aACxB,KAAC,OAAO,IAAC,OAAO,EAAC,UAAU,GAAG,EAC9B,KAAC,OAAO,IAAC,OAAO,EAAC,UAAU,YACzB,KAAC,OAAO,IAAC,OAAO,EAAC,SAAS,YACxB,oCAAkB,GACV,GACF,IACF,CACX,CAAC;QAEF,MAAM,QAAQ,GACZ,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,CAAC;QACT,MAAM,OAAO,GAAG,QAAQ,EAAE,gBAA0B,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CACzB,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,CACX,CAAC;QAEF,MAAM,YAAY,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAEjE,QAAQ,CACN,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,CACX,CAAC;QAEF,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CACjE,YAAY,CACb,CAAC;QACF,MAAM,YAAY,GAChB,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,CAAC;QACT,MAAM,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,OAAO,IAAC,OAAO,EAAC,mBAAmB,YAClC,kCAAgB,GACR,CACX,CAAC;QAEF,MAAM,kBAAkB,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAEvE,OAAO,EAAE,CAAC;QAEV,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CACjE,kBAAkB,CACnB,CAAC;QACF,MAAM,YAAY,GAChB,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,CAAC;QACT,MAAM,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/web/{typed-hooks.d.ts → generate-helpers.d.ts}

@@ -1,11 +1,15 @@
-import { useCallTool } from "./hooks/use-call-tool.js";
-import type { ToolPendingState, ToolSuccessState } from "./hooks/use-tool-info.js";
-import type { McpServer, InferWidgets, AnyWidgetRegistry, WidgetInput, WidgetOutput } from "../server/index.js";
-import type { CallToolArgs, UnknownObject } from "./types.js";
-type TypedCallToolReturn<TInput, TOutput> = ReturnType<typeof useCallTool<TInput & CallToolArgs, {
-    structuredContent: TOutput & UnknownObject;
-}>>;
-type TypedToolInfoReturn<TInput extends UnknownObject, TOutput extends UnknownObject, TResponseMetadata extends UnknownObject> = ToolPendingState<TInput> | ToolSuccessState<TInput, TOutput, TResponseMetadata>;
+import { type CallToolState, type CallToolFn, type CallToolAsyncFn } from "./hooks/use-call-tool.js";
+import { type ToolState } from "./hooks/use-tool-info.js";
+import type { InferTools, ToolInput, ToolOutput } from "../server/index.js";
+import type { CallToolResponse, Prettify, Objectify } from "./types.js";
+type TypedCallToolResponse<TOutput> = CallToolResponse & {
+    structuredContent: TOutput;
+};
+type TypedCallToolReturn<TInput, TOutput> = Prettify<CallToolState<TypedCallToolResponse<TOutput>> & {
+    callTool: CallToolFn<TInput, TypedCallToolResponse<TOutput>>;
+    callToolAsync: CallToolAsyncFn<TInput, TypedCallToolResponse<TOutput>>;
+}>;
+type TypedToolInfoReturn<TInput, TOutput> = ToolState<Objectify<TInput>, Objectify<TOutput>, Objectify<{}>>;
 /**
  * Creates typed versions of skybridge hooks with full type inference
  * for tool names, inputs, and outputs.
@@ -13,7 +17,9 @@ type TypedToolInfoReturn<TInput extends UnknownObject, TOutput extends UnknownOb
  * 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 T - The type of your McpServer instance. Use `typeof server`.
+ * @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
@@ -33,9 +39,9 @@ type TypedToolInfoReturn<TInput extends UnknownObject, TOutput extends UnknownOb
  * ```typescript
  * // web/src/skybridge.ts (one-time setup)
  * import type { AppType } from "../server";
- * import { createTypedHooks } from "skybridge/web";
+ * import { generateHelpers } from "skybridge/web";
  *
- * export const { useCallTool, useToolInfo } = createTypedHooks<AppType>();
+ * export const { useCallTool, useToolInfo } = generateHelpers<AppType>();
  * ```
  *
  * @example
@@ -52,11 +58,11 @@ type TypedToolInfoReturn<TInput extends UnknownObject, TOutput extends UnknownOb
  *   const toolInfo = useToolInfo<"search-voyage">();
  *   //                              ^ autocomplete for widget names
  *   // toolInfo.input is typed based on widget input schema
- *   // toolInfo.output is typed based on widget output schema
+ *   // toolInfo.output.structuredContent is typed based on widget output schema
  * }
  * ```
  */
-export declare function createTypedHooks<T extends McpServer<AnyWidgetRegistry>>(): {
+export declare function generateHelpers<ServerType = never>(): {
     /**
      * Typed version of `useCallTool` that provides autocomplete for tool names
      * and type inference for inputs and outputs.
@@ -76,7 +82,7 @@ export declare function createTypedHooks<T extends McpServer<AnyWidgetRegistry>>
      * }
      * ```
      */
-    useCallTool: <K extends keyof InferWidgets<T> & string>(name: K) => TypedCallToolReturn<InferWidgets<T>[K]["input"], InferWidgets<T>[K]["output"]>;
+    useCallTool: <ToolName extends keyof InferTools<ServerType> & string>(name: ToolName) => TypedCallToolReturn<ToolInput<ServerType, ToolName>, ToolOutput<ServerType, ToolName>>;
     /**
      * Typed version of `useToolInfo` that provides autocomplete for widget names
      * and type inference for inputs, outputs, and responseMetadata.
@@ -88,7 +94,7 @@ export declare function createTypedHooks<T extends McpServer<AnyWidgetRegistry>>
      * ```typescript
      * const toolInfo = useToolInfo<"search-voyage">();
      * // toolInfo.input is typed as { destination: string; ... }
-     * // toolInfo.output is typed as { results: Array<...>; ... } | undefined
+     * // toolInfo.output is typed as { results: Array<...>; ... }
      * // toolInfo.status narrows correctly: "pending" | "success"
      *
      * if (toolInfo.isPending) {
@@ -102,6 +108,6 @@ export declare function createTypedHooks<T extends McpServer<AnyWidgetRegistry>>
      * }
      * ```
      */
-    useToolInfo: <K extends keyof InferWidgets<T> & string>() => TypedToolInfoReturn<WidgetInput<T, K> & UnknownObject, WidgetOutput<T, K> & UnknownObject, UnknownObject>;
+    useToolInfo: <ToolName extends keyof InferTools<ServerType> & string>() => TypedToolInfoReturn<ToolInput<ServerType, ToolName>, ToolOutput<ServerType, ToolName>>;
 };
 export {};

package/dist/src/web/{typed-hooks.js → generate-helpers.js}

@@ -1,4 +1,4 @@
-import { useCallTool } from "./hooks/use-call-tool.js";
+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
@@ -7,7 +7,9 @@ import { useToolInfo } from "./hooks/use-tool-info.js";
  * 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 T - The type of your McpServer instance. Use `typeof server`.
+ * @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
@@ -27,9 +29,9 @@ import { useToolInfo } from "./hooks/use-tool-info.js";
  * ```typescript
  * // web/src/skybridge.ts (one-time setup)
  * import type { AppType } from "../server";
- * import { createTypedHooks } from "skybridge/web";
+ * import { generateHelpers } from "skybridge/web";
  *
- * export const { useCallTool, useToolInfo } = createTypedHooks<AppType>();
+ * export const { useCallTool, useToolInfo } = generateHelpers<AppType>();
  * ```
  *
  * @example
@@ -46,11 +48,11 @@ import { useToolInfo } from "./hooks/use-tool-info.js";
  *   const toolInfo = useToolInfo<"search-voyage">();
  *   //                              ^ autocomplete for widget names
  *   // toolInfo.input is typed based on widget input schema
- *   // toolInfo.output is typed based on widget output schema
+ *   // toolInfo.output.structuredContent is typed based on widget output schema
  * }
  * ```
  */
-export function createTypedHooks() {
+export function generateHelpers() {
     return {
         /**
          * Typed version of `useCallTool` that provides autocomplete for tool names
@@ -72,8 +74,6 @@ export function createTypedHooks() {
          * ```
          */
         useCallTool: (name) => {
-            // Type assertion is safe here because the runtime types are compatible.
-            // The underlying hook accepts broader types, but we expose narrower, more specific types.
             return useCallTool(name);
         },
         /**
@@ -87,7 +87,7 @@ export function createTypedHooks() {
          * ```typescript
          * const toolInfo = useToolInfo<"search-voyage">();
          * // toolInfo.input is typed as { destination: string; ... }
-         * // toolInfo.output is typed as { results: Array<...>; ... } | undefined
+         * // toolInfo.output is typed as { results: Array<...>; ... }
          * // toolInfo.status narrows correctly: "pending" | "success"
          *
          * if (toolInfo.isPending) {
@@ -102,10 +102,8 @@ export function createTypedHooks() {
          * ```
          */
         useToolInfo: () => {
-            // Type assertion is safe here because the runtime types are compatible.
-            // The underlying hook accepts broader types, but we expose narrower, more specific types.
             return useToolInfo();
         },
     };
 }
-//# sourceMappingURL=typed-hooks.js.map
+//# 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":"AAAA,OAAO,EACL,WAAW,GAIZ,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,WAAW,EAAkB,MAAM,0BAA0B,CAAC;AAmBvE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmDG;AACH,MAAM,UAAU,eAAe;IAI7B,OAAO;QACL;;;;;;;;;;;;;;;;;;WAkBG;QACH,WAAW,EAAE,CACX,IAAc,EAC0E,EAAE;YAC1F,OAAO,WAAW,CAAC,IAAI,CAGtB,CAAC;QACJ,CAAC;QAED;;;;;;;;;;;;;;;;;;;;;;;;WAwBG;QACH,WAAW,EAAE,GAGX,EAAE;YACF,OAAO,WAAW,EAGjB,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}

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

@@ -0,0 +1,153 @@
+import { expectTypeOf, test } from "vitest";
+import { generateHelpers } from "./generate-helpers.js";
+import { createTestServer } from "../test/utils.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");
+});
+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");
+    // @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, data } = 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(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();
+    // @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();
+    }
+});
+//# 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;AAC5C,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAEpD,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;AACxD,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,0EAA0E,EAAE,GAAG,EAAE;IAGpF,YAAY,EAAS,CAAC,aAAa,EAOhC,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,EAAM,CAAC;AACpD,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;IAE7B,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,IAAI,EAAE,GAAG,WAAW,CAAC,eAAe,CAAC,CAAC;IAExD,QAAQ,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE;QACjC,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,IAAI,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;QAC9C,CAAC;KACF,CAAC,CAAC;AACL,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,CAAC,EAAE,EAAE;QACX,SAAS,EAAE,CAAC,QAAQ,EAAE,EAAE;YACtB,YAAY,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC;QAC7D,CAAC;KACF,CAAC,CAAC;AACL,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;IAE/B,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,EAA0C,CAAC;IAEhF,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAClC,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,EAA2C,CAAC;QAClF,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"}

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 { generateHelpers } from "./generate-helpers.js";
+import { createMinimalTestServer } from "../test/utils.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,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAE3D,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,YAAY,IAAI,IAAI,CAAC;AAC3B,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,46 @@
+import { describe, it, expect, vi, beforeEach, afterEach, } from "vitest";
+import { filterWidgetContext, injectWidgetContext, serializeState, getInitialState, } from "./state.js";
+import { WIDGET_CONTEXT_KEY } from "../data-llm.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 stateWithCtxAndOthers = {
+                a: 1,
+                b: "two",
+                c: { nested: true },
+                [WIDGET_CONTEXT_KEY]: "context",
+            };
+            const filteredWithCtxAndOthers = filterWidgetContext(stateWithCtxAndOthers);
+            expect(filteredWithCtxAndOthers).toEqual({
+                a: 1,
+                b: "two",
+                c: { nested: true },
+            });
+            const stateNoCtx = { count: 5, name: "test" };
+            const filteredNoCtx = filterWidgetContext(stateNoCtx);
+            expect(filteredNoCtx).toEqual(stateNoCtx);
+        });
+    });
+    describe("serializeState", () => {
+        it("should serialize plain objects", () => {
+            const arr = [1, "two", { three: 3 }];
+            const date = new Date("2023-01-01T00:00:00Z");
+            const obj = { a: 1, b: "test", c: true, arr, date, fn: () => "test" };
+            const result = serializeState(obj);
+            expect(result).toEqual({
+                a: 1,
+                b: "test",
+                c: true,
+                arr: [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,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAChB,OAAO,EACL,mBAAmB,EACnB,mBAAmB,EACnB,cAAc,EACd,eAAe,GAChB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAEpD,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,qBAAqB,GAAG;gBAC5B,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,wBAAwB,GAAG,mBAAmB,CAClD,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,wBAAwB,CAAC,CAAC,OAAO,CAAC;gBACvC,CAAC,EAAE,CAAC;gBACJ,CAAC,EAAE,KAAK;gBACR,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;aACpB,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YAC9C,MAAM,aAAa,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;YACtD,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YACrC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,sBAAsB,CAAC,CAAC;YAC9C,MAAM,GAAG,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC;YACtE,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;YAEnC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,CAAC,EAAE,CAAC;gBACJ,CAAC,EAAE,MAAM;gBACT,CAAC,EAAE,IAAI;gBACP,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;gBAC7B,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,4 +1,4 @@
-export { useCallTool } from "./use-call-tool.js";
+export { useCallTool, type CallToolState, type SideEffects, type CallToolFn, type CallToolAsyncFn, } from "./use-call-tool.js";
 export { useDisplayMode } from "./use-display-mode.js";
 export { useFiles } from "./use-files.js";
 export { useLocale } from "./use-locale.js";
@@ -8,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";