@tambo-ai/react 0.48.0 → 0.50.0

package/dist/hooks/__tests__/use-message-images.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=use-message-images.test.d.ts.map

package/dist/hooks/__tests__/use-message-images.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-message-images.test.d.ts","sourceRoot":"","sources":["../../../src/hooks/__tests__/use-message-images.test.ts"],"names":[],"mappings":""}

package/dist/hooks/__tests__/use-message-images.test.js

@@ -0,0 +1,66 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const react_1 = require("@testing-library/react");
+const use_message_images_1 = require("../use-message-images");
+// Mock crypto.randomUUID
+global.crypto = {
+    randomUUID: jest.fn(() => "mock-uuid-" + Math.random()),
+};
+// Mock FileReader
+const mockFileReader = {
+    readAsDataURL: jest.fn(),
+    onload: null,
+    onerror: null,
+    result: "data:image/png;base64,mock-data",
+};
+global.FileReader = jest.fn(() => {
+    const reader = { ...mockFileReader };
+    reader.readAsDataURL = jest.fn(() => {
+        setTimeout(() => {
+            if (reader.onload) {
+                reader.onload({});
+            }
+        }, 0);
+    });
+    return reader;
+});
+describe("useMessageImages", () => {
+    beforeEach(() => {
+        jest.clearAllMocks();
+    });
+    it("should initialize with empty images array", () => {
+        const { result } = (0, react_1.renderHook)(() => (0, use_message_images_1.useMessageImages)());
+        expect(result.current.images).toEqual([]);
+    });
+    it("should reject non-image files", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, use_message_images_1.useMessageImages)());
+        const mockFile = new File(["test"], "test-document.pdf", {
+            type: "application/pdf",
+        });
+        await expect(result.current.addImage(mockFile)).rejects.toThrow("Only image files are allowed");
+    });
+    it("should clear all images", () => {
+        const { result } = (0, react_1.renderHook)(() => (0, use_message_images_1.useMessageImages)());
+        (0, react_1.act)(() => {
+            result.current.clearImages();
+        });
+        expect(result.current.images).toHaveLength(0);
+    });
+    it("should handle image validation correctly", () => {
+        const { result } = (0, react_1.renderHook)(() => (0, use_message_images_1.useMessageImages)());
+        // Test that hooks are available
+        expect(typeof result.current.addImage).toBe("function");
+        expect(typeof result.current.addImages).toBe("function");
+        expect(typeof result.current.removeImage).toBe("function");
+        expect(typeof result.current.clearImages).toBe("function");
+    });
+    it("should reject when no valid image files provided to addImages", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, use_message_images_1.useMessageImages)());
+        const mockFiles = [
+            new File(["test"], "document.pdf", { type: "application/pdf" }),
+            new File(["test"], "text.txt", { type: "text/plain" }),
+        ];
+        await expect(result.current.addImages(mockFiles)).rejects.toThrow("No valid image files provided");
+    });
+});
+//# sourceMappingURL=use-message-images.test.js.map

package/dist/hooks/__tests__/use-message-images.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-message-images.test.js","sourceRoot":"","sources":["../../../src/hooks/__tests__/use-message-images.test.ts"],"names":[],"mappings":";;AAAA,kDAAyD;AACzD,8DAAyD;AAEzD,yBAAyB;AACzB,MAAM,CAAC,MAAM,GAAG;IACd,UAAU,EAAE,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,YAAY,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;CACjD,CAAC;AAET,kBAAkB;AAClB,MAAM,cAAc,GAAG;IACrB,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;IACxB,MAAM,EAAE,IAAW;IACnB,OAAO,EAAE,IAAW;IACpB,MAAM,EAAE,iCAAiC;CAC1C,CAAC;AAED,MAAc,CAAC,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE;IACxC,MAAM,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,CAAC;IACrC,MAAM,CAAC,aAAa,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE;QAClC,UAAU,CAAC,GAAG,EAAE;YACd,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClB,MAAM,CAAC,MAAM,CAAC,EAAS,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC,EAAE,CAAC,CAAC,CAAC;IACR,CAAC,CAAC,CAAC;IACH,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAgB,GAAE,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAgB,GAAE,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,CAAC,MAAM,CAAC,EAAE,mBAAmB,EAAE;YACvD,IAAI,EAAE,iBAAiB;SACxB,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC7D,8BAA8B,CAC/B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAgB,GAAE,CAAC,CAAC;QAExD,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAgB,GAAE,CAAC,CAAC;QAExD,gCAAgC;QAChC,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACxD,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzD,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3D,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAgB,GAAE,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG;YAChB,IAAI,IAAI,CAAC,CAAC,MAAM,CAAC,EAAE,cAAc,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;YAC/D,IAAI,IAAI,CAAC,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;SACvD,CAAC;QAEF,MAAM,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC/D,+BAA+B,CAChC,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { act, renderHook } from \"@testing-library/react\";\nimport { useMessageImages } from \"../use-message-images\";\n\n// Mock crypto.randomUUID\nglobal.crypto = {\n  randomUUID: jest.fn(() => \"mock-uuid-\" + Math.random()),\n} as any;\n\n// Mock FileReader\nconst mockFileReader = {\n  readAsDataURL: jest.fn(),\n  onload: null as any,\n  onerror: null as any,\n  result: \"data:image/png;base64,mock-data\",\n};\n\n(global as any).FileReader = jest.fn(() => {\n  const reader = { ...mockFileReader };\n  reader.readAsDataURL = jest.fn(() => {\n    setTimeout(() => {\n      if (reader.onload) {\n        reader.onload({} as any);\n      }\n    }, 0);\n  });\n  return reader;\n});\n\ndescribe(\"useMessageImages\", () => {\n  beforeEach(() => {\n    jest.clearAllMocks();\n  });\n\n  it(\"should initialize with empty images array\", () => {\n    const { result } = renderHook(() => useMessageImages());\n    expect(result.current.images).toEqual([]);\n  });\n\n  it(\"should reject non-image files\", async () => {\n    const { result } = renderHook(() => useMessageImages());\n    const mockFile = new File([\"test\"], \"test-document.pdf\", {\n      type: \"application/pdf\",\n    });\n\n    await expect(result.current.addImage(mockFile)).rejects.toThrow(\n      \"Only image files are allowed\",\n    );\n  });\n\n  it(\"should clear all images\", () => {\n    const { result } = renderHook(() => useMessageImages());\n\n    act(() => {\n      result.current.clearImages();\n    });\n\n    expect(result.current.images).toHaveLength(0);\n  });\n\n  it(\"should handle image validation correctly\", () => {\n    const { result } = renderHook(() => useMessageImages());\n\n    // Test that hooks are available\n    expect(typeof result.current.addImage).toBe(\"function\");\n    expect(typeof result.current.addImages).toBe(\"function\");\n    expect(typeof result.current.removeImage).toBe(\"function\");\n    expect(typeof result.current.clearImages).toBe(\"function\");\n  });\n\n  it(\"should reject when no valid image files provided to addImages\", async () => {\n    const { result } = renderHook(() => useMessageImages());\n    const mockFiles = [\n      new File([\"test\"], \"document.pdf\", { type: \"application/pdf\" }),\n      new File([\"test\"], \"text.txt\", { type: \"text/plain\" }),\n    ];\n\n    await expect(result.current.addImages(mockFiles)).rejects.toThrow(\n      \"No valid image files provided\",\n    );\n  });\n});\n"]}

package/dist/hooks/use-message-images.d.ts

@@ -0,0 +1,25 @@
+/**
+ * Represents an image staged for upload
+ */
+export interface StagedImage {
+    id: string;
+    name: string;
+    dataUrl: string;
+    file: File;
+    size: number;
+    type: string;
+}
+interface UseMessageImagesReturn {
+    images: StagedImage[];
+    addImage: (file: File) => Promise<void>;
+    addImages: (files: File[]) => Promise<void>;
+    removeImage: (id: string) => void;
+    clearImages: () => void;
+}
+/**
+ * Hook for managing images in message input
+ * @returns Object with images array and management functions
+ */
+export declare function useMessageImages(): UseMessageImagesReturn;
+export {};
+//# sourceMappingURL=use-message-images.d.ts.map

package/dist/hooks/use-message-images.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-message-images.d.ts","sourceRoot":"","sources":["../../src/hooks/use-message-images.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,IAAI,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd;AAED,UAAU,sBAAsB;IAC9B,MAAM,EAAE,WAAW,EAAE,CAAC;IACtB,QAAQ,EAAE,CAAC,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,SAAS,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5C,WAAW,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,IAAI,CAAC;IAClC,WAAW,EAAE,MAAM,IAAI,CAAC;CACzB;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,IAAI,sBAAsB,CAqEzD"}

package/dist/hooks/use-message-images.js

@@ -0,0 +1,66 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useMessageImages = useMessageImages;
+const react_1 = require("react");
+/**
+ * Hook for managing images in message input
+ * @returns Object with images array and management functions
+ */
+function useMessageImages() {
+    const [images, setImages] = (0, react_1.useState)([]);
+    const fileToDataUrl = async (file) => {
+        return await new Promise((resolve, reject) => {
+            const reader = new FileReader();
+            reader.onload = () => resolve(reader.result);
+            reader.onerror = reject;
+            reader.readAsDataURL(file);
+        });
+    };
+    const addImage = (0, react_1.useCallback)(async (file) => {
+        if (!file.type.startsWith("image/")) {
+            throw new Error("Only image files are allowed");
+        }
+        const dataUrl = await fileToDataUrl(file);
+        const newImage = {
+            id: crypto.randomUUID(),
+            name: file.name,
+            dataUrl,
+            file,
+            size: file.size,
+            type: file.type,
+        };
+        setImages((prev) => [...prev, newImage]);
+    }, []);
+    const addImages = (0, react_1.useCallback)(async (files) => {
+        const imageFiles = files.filter((file) => file.type.startsWith("image/"));
+        if (imageFiles.length === 0) {
+            throw new Error("No valid image files provided");
+        }
+        const newImages = await Promise.all(imageFiles.map(async (file) => {
+            const dataUrl = await fileToDataUrl(file);
+            return {
+                id: crypto.randomUUID(),
+                name: file.name,
+                dataUrl,
+                file,
+                size: file.size,
+                type: file.type,
+            };
+        }));
+        setImages((prev) => [...prev, ...newImages]);
+    }, []);
+    const removeImage = (0, react_1.useCallback)((id) => {
+        setImages((prev) => prev.filter((img) => img.id !== id));
+    }, []);
+    const clearImages = (0, react_1.useCallback)(() => {
+        setImages([]);
+    }, []);
+    return {
+        images,
+        addImage,
+        addImages,
+        removeImage,
+        clearImages,
+    };
+}
+//# sourceMappingURL=use-message-images.js.map

package/dist/hooks/use-message-images.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-message-images.js","sourceRoot":"","sources":["../../src/hooks/use-message-images.ts"],"names":[],"mappings":";;AA0BA,4CAqEC;AA/FD,iCAA8C;AAsB9C;;;GAGG;AACH,SAAgB,gBAAgB;IAC9B,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,IAAA,gBAAQ,EAAgB,EAAE,CAAC,CAAC;IAExD,MAAM,aAAa,GAAG,KAAK,EAAE,IAAU,EAAmB,EAAE;QAC1D,OAAO,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3C,MAAM,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;YAChC,MAAM,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,MAAgB,CAAC,CAAC;YACvD,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC;YACxB,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,QAAQ,GAAG,IAAA,mBAAW,EAAC,KAAK,EAAE,IAAU,EAAE,EAAE;QAChD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAgB;YAC5B,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO;YACP,IAAI;YACJ,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;SAChB,CAAC;QAEF,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;IAC3C,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,SAAS,GAAG,IAAA,mBAAW,EAAC,KAAK,EAAE,KAAa,EAAE,EAAE;QACpD,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;QAE1E,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,GAAG,CACjC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YAC5B,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,CAAC;YAC1C,OAAO;gBACL,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;gBACvB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,OAAO;gBACP,IAAI;gBACJ,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,IAAI,CAAC,IAAI;aAChB,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;QAEF,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC;IAC/C,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,WAAW,GAAG,IAAA,mBAAW,EAAC,CAAC,EAAU,EAAE,EAAE;QAC7C,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,WAAW,GAAG,IAAA,mBAAW,EAAC,GAAG,EAAE;QACnC,SAAS,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO;QACL,MAAM;QACN,QAAQ;QACR,SAAS;QACT,WAAW;QACX,WAAW;KACZ,CAAC;AACJ,CAAC","sourcesContent":["import { useCallback, useState } from \"react\";\n\n/**\n * Represents an image staged for upload\n */\nexport interface StagedImage {\n  id: string;\n  name: string;\n  dataUrl: string;\n  file: File;\n  size: number;\n  type: string;\n}\n\ninterface UseMessageImagesReturn {\n  images: StagedImage[];\n  addImage: (file: File) => Promise<void>;\n  addImages: (files: File[]) => Promise<void>;\n  removeImage: (id: string) => void;\n  clearImages: () => void;\n}\n\n/**\n * Hook for managing images in message input\n * @returns Object with images array and management functions\n */\nexport function useMessageImages(): UseMessageImagesReturn {\n  const [images, setImages] = useState<StagedImage[]>([]);\n\n  const fileToDataUrl = async (file: File): Promise<string> => {\n    return await new Promise((resolve, reject) => {\n      const reader = new FileReader();\n      reader.onload = () => resolve(reader.result as string);\n      reader.onerror = reject;\n      reader.readAsDataURL(file);\n    });\n  };\n\n  const addImage = useCallback(async (file: File) => {\n    if (!file.type.startsWith(\"image/\")) {\n      throw new Error(\"Only image files are allowed\");\n    }\n\n    const dataUrl = await fileToDataUrl(file);\n    const newImage: StagedImage = {\n      id: crypto.randomUUID(),\n      name: file.name,\n      dataUrl,\n      file,\n      size: file.size,\n      type: file.type,\n    };\n\n    setImages((prev) => [...prev, newImage]);\n  }, []);\n\n  const addImages = useCallback(async (files: File[]) => {\n    const imageFiles = files.filter((file) => file.type.startsWith(\"image/\"));\n\n    if (imageFiles.length === 0) {\n      throw new Error(\"No valid image files provided\");\n    }\n\n    const newImages = await Promise.all(\n      imageFiles.map(async (file) => {\n        const dataUrl = await fileToDataUrl(file);\n        return {\n          id: crypto.randomUUID(),\n          name: file.name,\n          dataUrl,\n          file,\n          size: file.size,\n          type: file.type,\n        };\n      }),\n    );\n\n    setImages((prev) => [...prev, ...newImages]);\n  }, []);\n\n  const removeImage = useCallback((id: string) => {\n    setImages((prev) => prev.filter((img) => img.id !== id));\n  }, []);\n\n  const clearImages = useCallback(() => {\n    setImages([]);\n  }, []);\n\n  return {\n    images,\n    addImage,\n    addImages,\n    removeImage,\n    clearImages,\n  };\n}\n"]}

package/dist/index.d.ts

@@ -1,6 +1,7 @@
 /** Exports for the library. Only publically available exports are re-exported here. Anything not exported here is not supported and may change or break at any time. */
 export { useTamboComponentState } from "./hooks/use-component-state";
 export { TamboMessageProvider, useTamboCurrentMessage, } from "./hooks/use-current-message";
+export { useMessageImages, type StagedImage } from "./hooks/use-message-images";
 export { useTamboStreamingProps } from "./hooks/use-streaming-props";
 export * from "./hooks/use-suggestions";
 export { useTamboStreamStatus, type PropStatus, type StreamStatus, } from "./hooks/use-tambo-stream-status";

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,wKAAwK;AAExK,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EACL,oBAAoB,EACpB,sBAAsB,GACvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,cAAc,yBAAyB,CAAC;AACxC,OAAO,EACL,oBAAoB,EACpB,KAAK,UAAU,EACf,KAAK,YAAY,GAClB,MAAM,iCAAiC,CAAC;AAGzC,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,2BAA2B,EAC3B,uBAAuB,EACvB,aAAa,EACb,iBAAiB,EACjB,wBAAwB,EACxB,mBAAmB,EACnB,uBAAuB,EACvB,QAAQ,EACR,cAAc,EACd,sBAAsB,EACtB,uBAAuB,EACvB,cAAc,EACd,cAAc,EACd,mBAAmB,EACnB,KAAK,cAAc,EACnB,KAAK,+BAA+B,EACpC,KAAK,gCAAgC,EACrC,KAAK,oBAAoB,EACzB,KAAK,sBAAsB,EAC3B,KAAK,4BAA4B,GAClC,MAAM,aAAa,CAAC;AAGrB,YAAY,EACV,QAAQ,EACR,cAAc,EACd,YAAY,GACb,MAAM,0BAA0B,CAAC;AAClC,YAAY,EACV,UAAU,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,sBAAsB,GACvB,MAAM,6DAA6D,CAAC;AACrE,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EACL,KAAK,4BAA4B,EACjC,KAAK,iBAAiB,EACtB,KAAK,aAAa,EAClB,KAAK,mBAAmB,EACxB,KAAK,SAAS,GACf,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,eAAe,EACf,KAAK,kBAAkB,GACxB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,KAAK,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD,YAAY,EACV,0BAA0B,IAAI,qBAAqB,EACnD,wBAAwB,GACzB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,qBAAqB,IAAI,gBAAgB,EACzC,KAAK,kBAAkB,EACvB,KAAK,0BAA0B,GAChC,MAAM,yCAAyC,CAAC;AACjD,OAAO,EACL,oBAAoB,EACpB,+BAA+B,GAChC,MAAM,yCAAyC,CAAC;AAGjD,OAAO,EACL,wBAAwB,EACxB,wBAAwB,GACzB,MAAM,mBAAmB,CAAC;AAC3B,YAAY,EACV,iBAAiB,EACjB,eAAe,EACf,cAAc,GACf,MAAM,mBAAmB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,wKAAwK;AAExK,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EACL,oBAAoB,EACpB,sBAAsB,GACvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,gBAAgB,EAAE,KAAK,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAChF,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,cAAc,yBAAyB,CAAC;AACxC,OAAO,EACL,oBAAoB,EACpB,KAAK,UAAU,EACf,KAAK,YAAY,GAClB,MAAM,iCAAiC,CAAC;AAGzC,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,2BAA2B,EAC3B,uBAAuB,EACvB,aAAa,EACb,iBAAiB,EACjB,wBAAwB,EACxB,mBAAmB,EACnB,uBAAuB,EACvB,QAAQ,EACR,cAAc,EACd,sBAAsB,EACtB,uBAAuB,EACvB,cAAc,EACd,cAAc,EACd,mBAAmB,EACnB,KAAK,cAAc,EACnB,KAAK,+BAA+B,EACpC,KAAK,gCAAgC,EACrC,KAAK,oBAAoB,EACzB,KAAK,sBAAsB,EAC3B,KAAK,4BAA4B,GAClC,MAAM,aAAa,CAAC;AAGrB,YAAY,EACV,QAAQ,EACR,cAAc,EACd,YAAY,GACb,MAAM,0BAA0B,CAAC;AAClC,YAAY,EACV,UAAU,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,sBAAsB,GACvB,MAAM,6DAA6D,CAAC;AACrE,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EACL,KAAK,4BAA4B,EACjC,KAAK,iBAAiB,EACtB,KAAK,aAAa,EAClB,KAAK,mBAAmB,EACxB,KAAK,SAAS,GACf,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,eAAe,EACf,KAAK,kBAAkB,GACxB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,KAAK,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD,YAAY,EACV,0BAA0B,IAAI,qBAAqB,EACnD,wBAAwB,GACzB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,qBAAqB,IAAI,gBAAgB,EACzC,KAAK,kBAAkB,EACvB,KAAK,0BAA0B,GAChC,MAAM,yCAAyC,CAAC;AACjD,OAAO,EACL,oBAAoB,EACpB,+BAA+B,GAChC,MAAM,yCAAyC,CAAC;AAGjD,OAAO,EACL,wBAAwB,EACxB,wBAAwB,GACzB,MAAM,mBAAmB,CAAC;AAC3B,YAAY,EACV,iBAAiB,EACjB,eAAe,EACf,cAAc,GACf,MAAM,mBAAmB,CAAC"}

package/dist/index.js

@@ -15,12 +15,14 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.currentTimeContextHelper = exports.currentPageContextHelper = exports.useCurrentInteractablesSnapshot = exports.useTamboInteractable = exports.withInteractable = exports.GenerationStage = exports.useTamboThreadList = exports.useTamboThreadInput = exports.useTamboThread = exports.useTamboStream = exports.useTamboGenerationStage = exports.useTamboContextHelpers = exports.useTamboClient = exports.useTambo = exports.useIsTamboTokenUpdating = exports.TamboThreadProvider = exports.TamboThreadInputProvider = exports.TamboStubProvider = exports.TamboProvider = exports.TamboPropStreamProvider = exports.TamboContextHelpersProvider = exports.TamboComponentProvider = exports.TamboClientProvider = exports.useTamboStreamStatus = exports.useTamboStreamingProps = exports.useTamboCurrentMessage = exports.TamboMessageProvider = exports.useTamboComponentState = void 0;
+exports.currentTimeContextHelper = exports.currentPageContextHelper = exports.useCurrentInteractablesSnapshot = exports.useTamboInteractable = exports.withInteractable = exports.GenerationStage = exports.useTamboThreadList = exports.useTamboThreadInput = exports.useTamboThread = exports.useTamboStream = exports.useTamboGenerationStage = exports.useTamboContextHelpers = exports.useTamboClient = exports.useTambo = exports.useIsTamboTokenUpdating = exports.TamboThreadProvider = exports.TamboThreadInputProvider = exports.TamboStubProvider = exports.TamboProvider = exports.TamboPropStreamProvider = exports.TamboContextHelpersProvider = exports.TamboComponentProvider = exports.TamboClientProvider = exports.useTamboStreamStatus = exports.useTamboStreamingProps = exports.useMessageImages = exports.useTamboCurrentMessage = exports.TamboMessageProvider = exports.useTamboComponentState = void 0;
 var use_component_state_1 = require("./hooks/use-component-state");
 Object.defineProperty(exports, "useTamboComponentState", { enumerable: true, get: function () { return use_component_state_1.useTamboComponentState; } });
 var use_current_message_1 = require("./hooks/use-current-message");
 Object.defineProperty(exports, "TamboMessageProvider", { enumerable: true, get: function () { return use_current_message_1.TamboMessageProvider; } });
 Object.defineProperty(exports, "useTamboCurrentMessage", { enumerable: true, get: function () { return use_current_message_1.useTamboCurrentMessage; } });
+var use_message_images_1 = require("./hooks/use-message-images");
+Object.defineProperty(exports, "useMessageImages", { enumerable: true, get: function () { return use_message_images_1.useMessageImages; } });
 var use_streaming_props_1 = require("./hooks/use-streaming-props");
 Object.defineProperty(exports, "useTamboStreamingProps", { enumerable: true, get: function () { return use_streaming_props_1.useTamboStreamingProps; } });
 __exportStar(require("./hooks/use-suggestions"), exports);

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA,wKAAwK;;;;;;;;;;;;;;;;;AAExK,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,mEAGqC;AAFnC,2HAAA,oBAAoB,OAAA;AACpB,6HAAA,sBAAsB,OAAA;AAExB,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,0DAAwC;AACxC,2EAIyC;AAHvC,+HAAA,oBAAoB,OAAA;AAKtB,gCAAgC;AAChC,yCAuBqB;AAtBnB,gHAAA,mBAAmB,OAAA;AACnB,mHAAA,sBAAsB,OAAA;AACtB,wHAAA,2BAA2B,OAAA;AAC3B,oHAAA,uBAAuB,OAAA;AACvB,0GAAA,aAAa,OAAA;AACb,8GAAA,iBAAiB,OAAA;AACjB,qHAAA,wBAAwB,OAAA;AACxB,gHAAA,mBAAmB,OAAA;AACnB,oHAAA,uBAAuB,OAAA;AACvB,qGAAA,QAAQ,OAAA;AACR,2GAAA,cAAc,OAAA;AACd,mHAAA,sBAAsB,OAAA;AACtB,oHAAA,uBAAuB,OAAA;AACvB,2GAAA,cAAc,OAAA;AACd,2GAAA,cAAc,OAAA;AACd,gHAAA,mBAAmB,OAAA;AAqBrB,+DAA+D;AAAtD,uHAAA,kBAAkB,OAAA;AAQ3B,mFAG6C;AAF3C,8HAAA,eAAe,OAAA;AASjB,mFAIiD;AAH/C,2HAAA,qBAAqB,OAAoB;AAI3C,uFAGiD;AAF/C,mIAAA,oBAAoB,OAAA;AACpB,8IAAA,+BAA+B,OAAA;AAGjC,0BAA0B;AAC1B,qDAG2B;AAFzB,2HAAA,wBAAwB,OAAA;AACxB,2HAAA,wBAAwB,OAAA","sourcesContent":["/** Exports for the library. Only publically available exports are re-exported here. Anything not exported here is not supported and may change or break at any time. */\n\nexport { useTamboComponentState } from \"./hooks/use-component-state\";\nexport {\n  TamboMessageProvider,\n  useTamboCurrentMessage,\n} from \"./hooks/use-current-message\";\nexport { useTamboStreamingProps } from \"./hooks/use-streaming-props\";\nexport * from \"./hooks/use-suggestions\";\nexport {\n  useTamboStreamStatus,\n  type PropStatus,\n  type StreamStatus,\n} from \"./hooks/use-tambo-stream-status\";\n\n// Re-export provider components\nexport {\n  TamboClientProvider,\n  TamboComponentProvider,\n  TamboContextHelpersProvider,\n  TamboPropStreamProvider,\n  TamboProvider,\n  TamboStubProvider,\n  TamboThreadInputProvider,\n  TamboThreadProvider,\n  useIsTamboTokenUpdating,\n  useTambo,\n  useTamboClient,\n  useTamboContextHelpers,\n  useTamboGenerationStage,\n  useTamboStream,\n  useTamboThread,\n  useTamboThreadInput,\n  type TamboComponent,\n  type TamboContextHelpersContextProps,\n  type TamboContextHelpersProviderProps,\n  type TamboRegistryContext,\n  type TamboStubProviderProps,\n  type TamboThreadInputContextProps,\n} from \"./providers\";\n\n// Re-export types from Tambo Node SDK\nexport type {\n  APIError,\n  RateLimitError,\n  TamboAIError,\n} from \"@tambo-ai/typescript-sdk\";\nexport type {\n  Suggestion,\n  SuggestionGenerateParams,\n  SuggestionGenerateResponse,\n  SuggestionListResponse,\n} from \"@tambo-ai/typescript-sdk/resources/beta/threads/suggestions\";\nexport { useTamboThreadList } from \"./hooks/use-tambo-threads\";\nexport {\n  type ComponentContextToolMetadata,\n  type ComponentRegistry,\n  type ParameterSpec,\n  type RegisteredComponent,\n  type TamboTool,\n} from \"./model/component-metadata\";\nexport {\n  GenerationStage,\n  type TamboThreadMessage,\n} from \"./model/generate-component-response\";\nexport { type TamboThread } from \"./model/tambo-thread\";\n\nexport type {\n  TamboInteractableComponent as InteractableComponent,\n  TamboInteractableContext,\n} from \"./model/tambo-interactable\";\nexport {\n  withTamboInteractable as withInteractable,\n  type InteractableConfig,\n  type WithTamboInteractableProps,\n} from \"./providers/hoc/with-tambo-interactable\";\nexport {\n  useTamboInteractable,\n  useCurrentInteractablesSnapshot,\n} from \"./providers/tambo-interactable-provider\";\n\n// Context helpers exports\nexport {\n  currentPageContextHelper,\n  currentTimeContextHelper,\n} from \"./context-helpers\";\nexport type {\n  AdditionalContext,\n  ContextHelperFn,\n  ContextHelpers,\n} from \"./context-helpers\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA,wKAAwK;;;;;;;;;;;;;;;;;AAExK,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,mEAGqC;AAFnC,2HAAA,oBAAoB,OAAA;AACpB,6HAAA,sBAAsB,OAAA;AAExB,iEAAgF;AAAvE,sHAAA,gBAAgB,OAAA;AACzB,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,0DAAwC;AACxC,2EAIyC;AAHvC,+HAAA,oBAAoB,OAAA;AAKtB,gCAAgC;AAChC,yCAuBqB;AAtBnB,gHAAA,mBAAmB,OAAA;AACnB,mHAAA,sBAAsB,OAAA;AACtB,wHAAA,2BAA2B,OAAA;AAC3B,oHAAA,uBAAuB,OAAA;AACvB,0GAAA,aAAa,OAAA;AACb,8GAAA,iBAAiB,OAAA;AACjB,qHAAA,wBAAwB,OAAA;AACxB,gHAAA,mBAAmB,OAAA;AACnB,oHAAA,uBAAuB,OAAA;AACvB,qGAAA,QAAQ,OAAA;AACR,2GAAA,cAAc,OAAA;AACd,mHAAA,sBAAsB,OAAA;AACtB,oHAAA,uBAAuB,OAAA;AACvB,2GAAA,cAAc,OAAA;AACd,2GAAA,cAAc,OAAA;AACd,gHAAA,mBAAmB,OAAA;AAqBrB,+DAA+D;AAAtD,uHAAA,kBAAkB,OAAA;AAQ3B,mFAG6C;AAF3C,8HAAA,eAAe,OAAA;AASjB,mFAIiD;AAH/C,2HAAA,qBAAqB,OAAoB;AAI3C,uFAGiD;AAF/C,mIAAA,oBAAoB,OAAA;AACpB,8IAAA,+BAA+B,OAAA;AAGjC,0BAA0B;AAC1B,qDAG2B;AAFzB,2HAAA,wBAAwB,OAAA;AACxB,2HAAA,wBAAwB,OAAA","sourcesContent":["/** Exports for the library. Only publically available exports are re-exported here. Anything not exported here is not supported and may change or break at any time. */\n\nexport { useTamboComponentState } from \"./hooks/use-component-state\";\nexport {\n  TamboMessageProvider,\n  useTamboCurrentMessage,\n} from \"./hooks/use-current-message\";\nexport { useMessageImages, type StagedImage } from \"./hooks/use-message-images\";\nexport { useTamboStreamingProps } from \"./hooks/use-streaming-props\";\nexport * from \"./hooks/use-suggestions\";\nexport {\n  useTamboStreamStatus,\n  type PropStatus,\n  type StreamStatus,\n} from \"./hooks/use-tambo-stream-status\";\n\n// Re-export provider components\nexport {\n  TamboClientProvider,\n  TamboComponentProvider,\n  TamboContextHelpersProvider,\n  TamboPropStreamProvider,\n  TamboProvider,\n  TamboStubProvider,\n  TamboThreadInputProvider,\n  TamboThreadProvider,\n  useIsTamboTokenUpdating,\n  useTambo,\n  useTamboClient,\n  useTamboContextHelpers,\n  useTamboGenerationStage,\n  useTamboStream,\n  useTamboThread,\n  useTamboThreadInput,\n  type TamboComponent,\n  type TamboContextHelpersContextProps,\n  type TamboContextHelpersProviderProps,\n  type TamboRegistryContext,\n  type TamboStubProviderProps,\n  type TamboThreadInputContextProps,\n} from \"./providers\";\n\n// Re-export types from Tambo Node SDK\nexport type {\n  APIError,\n  RateLimitError,\n  TamboAIError,\n} from \"@tambo-ai/typescript-sdk\";\nexport type {\n  Suggestion,\n  SuggestionGenerateParams,\n  SuggestionGenerateResponse,\n  SuggestionListResponse,\n} from \"@tambo-ai/typescript-sdk/resources/beta/threads/suggestions\";\nexport { useTamboThreadList } from \"./hooks/use-tambo-threads\";\nexport {\n  type ComponentContextToolMetadata,\n  type ComponentRegistry,\n  type ParameterSpec,\n  type RegisteredComponent,\n  type TamboTool,\n} from \"./model/component-metadata\";\nexport {\n  GenerationStage,\n  type TamboThreadMessage,\n} from \"./model/generate-component-response\";\nexport { type TamboThread } from \"./model/tambo-thread\";\n\nexport type {\n  TamboInteractableComponent as InteractableComponent,\n  TamboInteractableContext,\n} from \"./model/tambo-interactable\";\nexport {\n  withTamboInteractable as withInteractable,\n  type InteractableConfig,\n  type WithTamboInteractableProps,\n} from \"./providers/hoc/with-tambo-interactable\";\nexport {\n  useTamboInteractable,\n  useCurrentInteractablesSnapshot,\n} from \"./providers/tambo-interactable-provider\";\n\n// Context helpers exports\nexport {\n  currentPageContextHelper,\n  currentTimeContextHelper,\n} from \"./context-helpers\";\nexport type {\n  AdditionalContext,\n  ContextHelperFn,\n  ContextHelpers,\n} from \"./context-helpers\";\n"]}

package/dist/providers/__tests__/tambo-registry-provider.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=tambo-registry-provider.test.d.ts.map

package/dist/providers/__tests__/tambo-registry-provider.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-registry-provider.test.d.ts","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-registry-provider.test.tsx"],"names":[],"mappings":""}

package/dist/providers/__tests__/tambo-registry-provider.test.js

@@ -0,0 +1,224 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const react_1 = require("@testing-library/react");
+const react_2 = __importDefault(require("react"));
+const zod_1 = require("zod");
+const tambo_registry_provider_1 = require("../tambo-registry-provider");
+// Shared tool registry for all tests
+const createMockTools = () => [
+    {
+        name: "test-tool-1",
+        description: "First test tool",
+        tool: jest.fn().mockResolvedValue("test-tool-1-result"),
+        toolSchema: zod_1.z
+            .function()
+            .args(zod_1.z.string().describe("input parameter"))
+            .returns(zod_1.z.string()),
+    },
+    {
+        name: "test-tool-2",
+        description: "Second test tool",
+        tool: jest.fn().mockResolvedValue("test-tool-2-result"),
+        toolSchema: zod_1.z
+            .function()
+            .args(zod_1.z.number().describe("number parameter"))
+            .returns(zod_1.z.string()),
+    },
+];
+const createMockComponents = (tools) => [
+    {
+        name: "TestComponent",
+        component: () => react_2.default.createElement("div", null, "TestComponent"),
+        description: "Test component",
+        propsSchema: zod_1.z.object({
+            test: zod_1.z.string(),
+        }),
+        associatedTools: tools,
+    },
+];
+describe("TamboRegistryProvider", () => {
+    const mockTools = createMockTools();
+    const mockComponents = createMockComponents(mockTools);
+    beforeEach(() => {
+        jest.clearAllMocks();
+    });
+    describe("Component and tool registration", () => {
+        it("should register components and tools correctly", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockComponents }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            expect(result.current.componentList).toHaveProperty("TestComponent");
+            expect(result.current.toolRegistry).toHaveProperty("test-tool-1");
+            expect(result.current.toolRegistry).toHaveProperty("test-tool-2");
+            expect(result.current.componentToolAssociations).toHaveProperty("TestComponent");
+            expect(result.current.componentToolAssociations.TestComponent).toEqual([
+                "test-tool-1",
+                "test-tool-2",
+            ]);
+        });
+        it("should provide onCallUnregisteredTool in context", () => {
+            const mockonCallUnregisteredTool = jest
+                .fn()
+                .mockResolvedValue("test-result");
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockComponents, onCallUnregisteredTool: mockonCallUnregisteredTool }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // The onCallUnregisteredTool should be available in the context
+            expect(result.current.onCallUnregisteredTool).toBeDefined();
+            expect(typeof result.current.onCallUnregisteredTool).toBe("function");
+        });
+        it("should handle tool registration and association", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // Register a new tool
+            (0, react_1.act)(() => {
+                result.current.registerTool(mockTools[0]);
+            });
+            expect(result.current.toolRegistry).toHaveProperty("test-tool-1");
+            // Register a new component
+            (0, react_1.act)(() => {
+                result.current.registerComponent(mockComponents[0]);
+            });
+            expect(result.current.componentList).toHaveProperty("TestComponent");
+            expect(result.current.componentToolAssociations).toHaveProperty("TestComponent");
+        });
+        it("should handle multiple tool registration", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // Register multiple tools
+            (0, react_1.act)(() => {
+                result.current.registerTools(mockTools);
+            });
+            expect(result.current.toolRegistry).toHaveProperty("test-tool-1");
+            expect(result.current.toolRegistry).toHaveProperty("test-tool-2");
+            expect(Object.keys(result.current.toolRegistry)).toHaveLength(2);
+        });
+        it("should handle tool association with components", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockComponents }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // Add a new tool association
+            (0, react_1.act)(() => {
+                const newTool = {
+                    name: "new-tool",
+                    description: "New tool",
+                    tool: jest.fn().mockResolvedValue("new-tool-result"),
+                    toolSchema: zod_1.z
+                        .function()
+                        .args(zod_1.z.string().describe("input"))
+                        .returns(zod_1.z.string()),
+                };
+                result.current.addToolAssociation("TestComponent", newTool);
+            });
+            expect(result.current.componentToolAssociations.TestComponent).toContain("new-tool");
+        });
+        it("should throw error when adding tool association to non-existent component", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            const newTool = {
+                name: "new-tool",
+                description: "New tool",
+                tool: jest.fn().mockResolvedValue("new-tool-result"),
+                toolSchema: zod_1.z
+                    .function()
+                    .args(zod_1.z.string().describe("input"))
+                    .returns(zod_1.z.string()),
+            };
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.addToolAssociation("NonExistentComponent", newTool);
+                });
+            }).toThrow("Component NonExistentComponent not found in registry");
+        });
+        it("should validate tool schemas and throw error for invalid schemas", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            const invalidTool = {
+                name: "invalid-tool",
+                description: "Invalid tool",
+                tool: jest.fn().mockResolvedValue("result"),
+                toolSchema: zod_1.z.record(zod_1.z.string()), // This should cause validation to fail
+            };
+            // This should throw during registration due to invalid schema
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.registerTool(invalidTool);
+                });
+            }).toThrow('z.record() is not supported in toolSchema of tool "invalid-tool"');
+        });
+        it("should validate component schemas and throw error for invalid schemas", () => {
+            const invalidComponent = {
+                name: "InvalidComponent",
+                component: () => react_2.default.createElement("div", null, "Invalid"),
+                description: "Invalid component",
+                propsSchema: zod_1.z.record(zod_1.z.string()), // This should cause validation to fail
+            };
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // This should throw during registration due to invalid schema
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.registerComponent(invalidComponent);
+                });
+            }).toThrow('z.record() is not supported in propsSchema of component "InvalidComponent"');
+        });
+        it("should throw error when component has both propsSchema and propsDefinition", () => {
+            const invalidComponent = {
+                name: "InvalidComponent",
+                component: () => react_2.default.createElement("div", null, "Invalid"),
+                description: "Invalid component",
+                propsSchema: zod_1.z.object({ test: zod_1.z.string() }),
+                propsDefinition: { test: "string" }, // Both defined - should throw
+            };
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.registerComponent(invalidComponent);
+                });
+            }).toThrow("Component InvalidComponent cannot have both propsSchema and propsDefinition defined");
+        });
+        it("should throw error when component has neither propsSchema nor propsDefinition", () => {
+            const invalidComponent = {
+                name: "InvalidComponent",
+                component: () => react_2.default.createElement("div", null, "Invalid"),
+                description: "Invalid component",
+                // Neither propsSchema nor propsDefinition defined - should throw
+            };
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.registerComponent(invalidComponent);
+                });
+            }).toThrow("Component InvalidComponent must have either propsSchema (recommended) or propsDefinition defined");
+        });
+    });
+    describe("Tool call handling", () => {
+        it("should provide onCallUnregisteredTool callback for handling unknown tools", async () => {
+            const mockonCallUnregisteredTool = jest
+                .fn()
+                .mockResolvedValue("unknown-tool-result");
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockComponents, onCallUnregisteredTool: mockonCallUnregisteredTool }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // Verify the callback is available
+            expect(result.current.onCallUnregisteredTool).toBeDefined();
+            expect(typeof result.current.onCallUnregisteredTool).toBe("function");
+            // Simulate calling the unknown tool handler
+            const toolName = "unknown-tool";
+            const args = [{ parameterName: "input", parameterValue: "test-input" }];
+            await (0, react_1.act)(async () => {
+                if (result.current.onCallUnregisteredTool) {
+                    await result.current.onCallUnregisteredTool(toolName, args);
+                }
+            });
+            expect(mockonCallUnregisteredTool).toHaveBeenCalledWith(toolName, args);
+        });
+        it("should handle onCallUnregisteredTool being undefined", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockComponents }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            expect(result.current.onCallUnregisteredTool).toBeUndefined();
+        });
+    });
+});
+//# sourceMappingURL=tambo-registry-provider.test.js.map

package/dist/providers/__tests__/tambo-registry-provider.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-registry-provider.test.js","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-registry-provider.test.tsx"],"names":[],"mappings":";;;;;AAAA,kDAAyD;AACzD,kDAA0B;AAC1B,6BAAwB;AAExB,wEAGoC;AAEpC,qCAAqC;AACrC,MAAM,eAAe,GAAG,GAAgB,EAAE,CAAC;IACzC;QACE,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,iBAAiB;QAC9B,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,oBAAoB,CAAC;QACvD,UAAU,EAAE,OAAC;aACV,QAAQ,EAAE;aACV,IAAI,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;aAC5C,OAAO,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;KACvB;IACD;QACE,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,kBAAkB;QAC/B,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,oBAAoB,CAAC;QACvD,UAAU,EAAE,OAAC;aACV,QAAQ,EAAE;aACV,IAAI,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;aAC7C,OAAO,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;KACvB;CACF,CAAC;AAEF,MAAM,oBAAoB,GAAG,CAAC,KAAkB,EAAoB,EAAE,CAAC;IACrE;QACE,IAAI,EAAE,eAAe;QACrB,SAAS,EAAE,GAAG,EAAE,CAAC,2DAAwB;QACzC,WAAW,EAAE,gBAAgB;QAC7B,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YACpB,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;SACjB,CAAC;QACF,eAAe,EAAE,KAAK;KACvB;CACF,CAAC;AAEF,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,MAAM,SAAS,GAAG,eAAe,EAAE,CAAC;IACpC,MAAM,cAAc,GAAG,oBAAoB,CAAC,SAAS,CAAC,CAAC;IAEvD,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IAAC,UAAU,EAAE,cAAc,IAC9C,QAAQ,CACa,CACzB,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC,cAAc,CAC7D,eAAe,CAChB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC;gBACrE,aAAa;gBACb,aAAa;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,0BAA0B,GAAG,IAAI;iBACpC,EAAE,EAAE;iBACJ,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAEpC,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IACpB,UAAU,EAAE,cAAc,EAC1B,sBAAsB,EAAE,0BAA0B,IAEjD,QAAQ,CACa,CACzB,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,gEAAgE;YAChE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5D,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,sBAAsB;YACtB,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5C,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAElE,2BAA2B;YAC3B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;YACtD,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC,cAAc,CAC7D,eAAe,CAChB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,0BAA0B;YAC1B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAC1C,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IAAC,UAAU,EAAE,cAAc,IAC9C,QAAQ,CACa,CACzB,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,6BAA6B;YAC7B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,OAAO,GAAc;oBACzB,IAAI,EAAE,UAAU;oBAChB,WAAW,EAAE,UAAU;oBACvB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,iBAAiB,CAAC;oBACpD,UAAU,EAAE,OAAC;yBACV,QAAQ,EAAE;yBACV,IAAI,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;yBAClC,OAAO,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;iBACvB,CAAC;gBACF,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;YAC9D,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,aAAa,CAAC,CAAC,SAAS,CACtE,UAAU,CACX,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;YACnF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,OAAO,GAAc;gBACzB,IAAI,EAAE,UAAU;gBAChB,WAAW,EAAE,UAAU;gBACvB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,iBAAiB,CAAC;gBACpD,UAAU,EAAE,OAAC;qBACV,QAAQ,EAAE;qBACV,IAAI,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;qBAClC,OAAO,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;aACvB,CAAC;YAEF,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC;gBACrE,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CAAC,sDAAsD,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,WAAW,GAAc;gBAC7B,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,cAAc;gBAC3B,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC;gBAC3C,UAAU,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,EAAE,uCAAuC;aAC1E,CAAC;YAEF,8DAA8D;YAC9D,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;gBAC3C,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CACR,kEAAkE,CACnE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;YAC/E,MAAM,gBAAgB,GAAmB;gBACvC,IAAI,EAAE,kBAAkB;gBACxB,SAAS,EAAE,GAAG,EAAE,CAAC,qDAAkB;gBACnC,WAAW,EAAE,mBAAmB;gBAChC,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,EAAE,uCAAuC;aAC3E,CAAC;YAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,8DAA8D;YAC9D,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;gBACrD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CACR,4EAA4E,CAC7E,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,MAAM,gBAAgB,GAAmB;gBACvC,IAAI,EAAE,kBAAkB;gBACxB,SAAS,EAAE,GAAG,EAAE,CAAC,qDAAkB;gBACnC,WAAW,EAAE,mBAAmB;gBAChC,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC3C,eAAe,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,8BAA8B;aACpE,CAAC;YAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;gBACrD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CACR,qFAAqF,CACtF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,GAAG,EAAE;YACvF,MAAM,gBAAgB,GAAmB;gBACvC,IAAI,EAAE,kBAAkB;gBACxB,SAAS,EAAE,GAAG,EAAE,CAAC,qDAAkB;gBACnC,WAAW,EAAE,mBAAmB;gBAChC,iEAAiE;aAClE,CAAC;YAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;gBACrD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CACR,kGAAkG,CACnG,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACzF,MAAM,0BAA0B,GAAG,IAAI;iBACpC,EAAE,EAAE;iBACJ,iBAAiB,CAAC,qBAAqB,CAAC,CAAC;YAE5C,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IACpB,UAAU,EAAE,cAAc,EAC1B,sBAAsB,EAAE,0BAA0B,IAEjD,QAAQ,CACa,CACzB,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,mCAAmC;YACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5D,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAEtE,4CAA4C;YAC5C,MAAM,QAAQ,GAAG,cAAc,CAAC;YAChC,MAAM,IAAI,GAAG,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC,CAAC;YAExE,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;gBACnB,IAAI,MAAM,CAAC,OAAO,CAAC,sBAAsB,EAAE,CAAC;oBAC1C,MAAM,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,0BAA0B,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IAAC,UAAU,EAAE,cAAc,IAC9C,QAAQ,CACa,CACzB,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,aAAa,EAAE,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { act, renderHook } from \"@testing-library/react\";\nimport React from \"react\";\nimport { z } from \"zod\";\nimport { TamboComponent, TamboTool } from \"../../model/component-metadata\";\nimport {\n  TamboRegistryProvider,\n  useTamboRegistry,\n} from \"../tambo-registry-provider\";\n\n// Shared tool registry for all tests\nconst createMockTools = (): TamboTool[] => [\n  {\n    name: \"test-tool-1\",\n    description: \"First test tool\",\n    tool: jest.fn().mockResolvedValue(\"test-tool-1-result\"),\n    toolSchema: z\n      .function()\n      .args(z.string().describe(\"input parameter\"))\n      .returns(z.string()),\n  },\n  {\n    name: \"test-tool-2\",\n    description: \"Second test tool\",\n    tool: jest.fn().mockResolvedValue(\"test-tool-2-result\"),\n    toolSchema: z\n      .function()\n      .args(z.number().describe(\"number parameter\"))\n      .returns(z.string()),\n  },\n];\n\nconst createMockComponents = (tools: TamboTool[]): TamboComponent[] => [\n  {\n    name: \"TestComponent\",\n    component: () => <div>TestComponent</div>,\n    description: \"Test component\",\n    propsSchema: z.object({\n      test: z.string(),\n    }),\n    associatedTools: tools,\n  },\n];\n\ndescribe(\"TamboRegistryProvider\", () => {\n  const mockTools = createMockTools();\n  const mockComponents = createMockComponents(mockTools);\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n  });\n\n  describe(\"Component and tool registration\", () => {\n    it(\"should register components and tools correctly\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider components={mockComponents}>\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      expect(result.current.componentList).toHaveProperty(\"TestComponent\");\n      expect(result.current.toolRegistry).toHaveProperty(\"test-tool-1\");\n      expect(result.current.toolRegistry).toHaveProperty(\"test-tool-2\");\n      expect(result.current.componentToolAssociations).toHaveProperty(\n        \"TestComponent\",\n      );\n      expect(result.current.componentToolAssociations.TestComponent).toEqual([\n        \"test-tool-1\",\n        \"test-tool-2\",\n      ]);\n    });\n\n    it(\"should provide onCallUnregisteredTool in context\", () => {\n      const mockonCallUnregisteredTool = jest\n        .fn()\n        .mockResolvedValue(\"test-result\");\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider\n          components={mockComponents}\n          onCallUnregisteredTool={mockonCallUnregisteredTool}\n        >\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // The onCallUnregisteredTool should be available in the context\n      expect(result.current.onCallUnregisteredTool).toBeDefined();\n      expect(typeof result.current.onCallUnregisteredTool).toBe(\"function\");\n    });\n\n    it(\"should handle tool registration and association\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // Register a new tool\n      act(() => {\n        result.current.registerTool(mockTools[0]);\n      });\n\n      expect(result.current.toolRegistry).toHaveProperty(\"test-tool-1\");\n\n      // Register a new component\n      act(() => {\n        result.current.registerComponent(mockComponents[0]);\n      });\n\n      expect(result.current.componentList).toHaveProperty(\"TestComponent\");\n      expect(result.current.componentToolAssociations).toHaveProperty(\n        \"TestComponent\",\n      );\n    });\n\n    it(\"should handle multiple tool registration\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // Register multiple tools\n      act(() => {\n        result.current.registerTools(mockTools);\n      });\n\n      expect(result.current.toolRegistry).toHaveProperty(\"test-tool-1\");\n      expect(result.current.toolRegistry).toHaveProperty(\"test-tool-2\");\n      expect(Object.keys(result.current.toolRegistry)).toHaveLength(2);\n    });\n\n    it(\"should handle tool association with components\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider components={mockComponents}>\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // Add a new tool association\n      act(() => {\n        const newTool: TamboTool = {\n          name: \"new-tool\",\n          description: \"New tool\",\n          tool: jest.fn().mockResolvedValue(\"new-tool-result\"),\n          toolSchema: z\n            .function()\n            .args(z.string().describe(\"input\"))\n            .returns(z.string()),\n        };\n        result.current.addToolAssociation(\"TestComponent\", newTool);\n      });\n\n      expect(result.current.componentToolAssociations.TestComponent).toContain(\n        \"new-tool\",\n      );\n    });\n\n    it(\"should throw error when adding tool association to non-existent component\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      const newTool: TamboTool = {\n        name: \"new-tool\",\n        description: \"New tool\",\n        tool: jest.fn().mockResolvedValue(\"new-tool-result\"),\n        toolSchema: z\n          .function()\n          .args(z.string().describe(\"input\"))\n          .returns(z.string()),\n      };\n\n      expect(() => {\n        act(() => {\n          result.current.addToolAssociation(\"NonExistentComponent\", newTool);\n        });\n      }).toThrow(\"Component NonExistentComponent not found in registry\");\n    });\n\n    it(\"should validate tool schemas and throw error for invalid schemas\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      const invalidTool: TamboTool = {\n        name: \"invalid-tool\",\n        description: \"Invalid tool\",\n        tool: jest.fn().mockResolvedValue(\"result\"),\n        toolSchema: z.record(z.string()), // This should cause validation to fail\n      };\n\n      // This should throw during registration due to invalid schema\n      expect(() => {\n        act(() => {\n          result.current.registerTool(invalidTool);\n        });\n      }).toThrow(\n        'z.record() is not supported in toolSchema of tool \"invalid-tool\"',\n      );\n    });\n\n    it(\"should validate component schemas and throw error for invalid schemas\", () => {\n      const invalidComponent: TamboComponent = {\n        name: \"InvalidComponent\",\n        component: () => <div>Invalid</div>,\n        description: \"Invalid component\",\n        propsSchema: z.record(z.string()), // This should cause validation to fail\n      };\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // This should throw during registration due to invalid schema\n      expect(() => {\n        act(() => {\n          result.current.registerComponent(invalidComponent);\n        });\n      }).toThrow(\n        'z.record() is not supported in propsSchema of component \"InvalidComponent\"',\n      );\n    });\n\n    it(\"should throw error when component has both propsSchema and propsDefinition\", () => {\n      const invalidComponent: TamboComponent = {\n        name: \"InvalidComponent\",\n        component: () => <div>Invalid</div>,\n        description: \"Invalid component\",\n        propsSchema: z.object({ test: z.string() }),\n        propsDefinition: { test: \"string\" }, // Both defined - should throw\n      };\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      expect(() => {\n        act(() => {\n          result.current.registerComponent(invalidComponent);\n        });\n      }).toThrow(\n        \"Component InvalidComponent cannot have both propsSchema and propsDefinition defined\",\n      );\n    });\n\n    it(\"should throw error when component has neither propsSchema nor propsDefinition\", () => {\n      const invalidComponent: TamboComponent = {\n        name: \"InvalidComponent\",\n        component: () => <div>Invalid</div>,\n        description: \"Invalid component\",\n        // Neither propsSchema nor propsDefinition defined - should throw\n      };\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      expect(() => {\n        act(() => {\n          result.current.registerComponent(invalidComponent);\n        });\n      }).toThrow(\n        \"Component InvalidComponent must have either propsSchema (recommended) or propsDefinition defined\",\n      );\n    });\n  });\n\n  describe(\"Tool call handling\", () => {\n    it(\"should provide onCallUnregisteredTool callback for handling unknown tools\", async () => {\n      const mockonCallUnregisteredTool = jest\n        .fn()\n        .mockResolvedValue(\"unknown-tool-result\");\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider\n          components={mockComponents}\n          onCallUnregisteredTool={mockonCallUnregisteredTool}\n        >\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // Verify the callback is available\n      expect(result.current.onCallUnregisteredTool).toBeDefined();\n      expect(typeof result.current.onCallUnregisteredTool).toBe(\"function\");\n\n      // Simulate calling the unknown tool handler\n      const toolName = \"unknown-tool\";\n      const args = [{ parameterName: \"input\", parameterValue: \"test-input\" }];\n\n      await act(async () => {\n        if (result.current.onCallUnregisteredTool) {\n          await result.current.onCallUnregisteredTool(toolName, args);\n        }\n      });\n\n      expect(mockonCallUnregisteredTool).toHaveBeenCalledWith(toolName, args);\n    });\n\n    it(\"should handle onCallUnregisteredTool being undefined\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider components={mockComponents}>\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      expect(result.current.onCallUnregisteredTool).toBeUndefined();\n    });\n  });\n});\n"]}