@tambo-ai/react 0.45.0 → 0.46.1

package/dist/providers/hooks/__tests__/use-tambo-session-token.test.js

@@ -1,5 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+const react_query_1 = require("@tanstack/react-query");
 const react_1 = require("@testing-library/react");
 const use_tambo_session_token_1 = require("../use-tambo-session-token");
 describe("useTamboSessionToken", () => {
@@ -19,40 +20,59 @@ describe("useTamboSessionToken", () => {
         beta: mockBeta,
         bearer: "",
     };
+    const queryClient = new react_query_1.QueryClient({
+        defaultOptions: {
+            queries: {
+                retry: false,
+                refetchOnWindowFocus: false,
+            },
+        },
+    });
     beforeEach(() => {
         jest.clearAllMocks();
         jest.clearAllTimers();
         jest.useFakeTimers();
         mockTamboAI.bearer = "";
+        queryClient.clear();
     });
-    afterEach(() => {
-        jest.runOnlyPendingTimers();
+    afterEach(async () => {
+        await (0, react_1.act)(async () => {
+            await jest.runOnlyPendingTimersAsync();
+        });
         jest.useRealTimers();
+        jest.resetAllMocks();
     });
     it("should return null initially when no userToken is provided", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, undefined));
-        expect(result.current).toBeNull();
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, undefined));
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: false,
+        });
         expect(mockAuthApi.getToken).not.toHaveBeenCalled();
     });
     it("should fetch and return session token when userToken is provided", async () => {
-        jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, "user-token"));
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, "user-token"));
+        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);
         await (0, react_1.act)(async () => {
             await jest.runOnlyPendingTimersAsync();
         });
-        expect(result.current).toBe("test-access-token");
-        expect(mockTamboAI.bearer).toBe("test-access-token");
+        expect(result.current).toMatchObject({
+            data: mockTokenResponse,
+            isFetching: false,
+        });
+        expect(mockTamboAI.bearer).toBe(mockTokenResponse.access_token);
         // Verify the hook was called with correct parameters
-        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);
+        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(2);
         expect(mockAuthApi.getToken).toHaveBeenCalledWith(expect.any(Object));
     });
     it("should call getToken with correct token exchange parameters", async () => {
-        jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);
-        (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, "user-token"));
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, "user-token"));
         await (0, react_1.act)(async () => {
             await jest.runOnlyPendingTimersAsync();
         });
-        const callArgs = jest.mocked(mockAuthApi.getToken).mock.calls[0][0];
+        const callArgs = mockAuthApi.getToken.mock.calls[0][0];
         const tokenRequestString = new TextDecoder().decode(callArgs);
         const tokenRequest = new URLSearchParams(tokenRequestString);
         expect(tokenRequest.get("grant_type")).toBe("urn:ietf:params:oauth:grant-type:token-exchange");
@@ -60,12 +80,14 @@ describe("useTamboSessionToken", () => {
         expect(tokenRequest.get("subject_token_type")).toBe("urn:ietf:params:oauth:token-type:access_token");
     });
     it("should set bearer token on client", async () => {
-        jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, "user-token"));
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, "user-token"));
         await (0, react_1.act)(async () => {
             await jest.runOnlyPendingTimersAsync();
         });
-        expect(result.current).toBe("test-access-token");
+        expect(result.current).toMatchObject({
+            data: mockTokenResponse,
+        });
         expect(mockTamboAI.bearer).toBe("test-access-token");
     });
     it("should handle different token responses", async () => {
@@ -74,24 +96,35 @@ describe("useTamboSessionToken", () => {
             expires_in: 7200, // 2 hours
             token_type: "Bearer",
         };
-        jest.mocked(mockAuthApi.getToken).mockResolvedValue(customTokenResponse);
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, "user-token"));
+        mockAuthApi.getToken.mockResolvedValue(customTokenResponse);
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, "user-token"));
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: true,
+        });
         await (0, react_1.act)(async () => {
             await jest.runOnlyPendingTimersAsync();
         });
-        expect(result.current).toBe("custom-access-token");
+        expect(result.current).toMatchObject({
+            data: customTokenResponse,
+            isFetching: false,
+        });
         expect(mockTamboAI.bearer).toBe("custom-access-token");
     });
     it("should not fetch token when userToken changes to undefined", async () => {
-        jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);
-        const { result, rerender } = (0, react_1.renderHook)(({ userToken }) => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, userToken), {
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        const { result, rerender } = (0, react_1.renderHook)(({ userToken }) => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, userToken), {
             initialProps: { userToken: "user-token" },
         });
+        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);
         await (0, react_1.act)(async () => {
             await jest.runOnlyPendingTimersAsync();
         });
-        expect(result.current).toBe("test-access-token");
-        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);
+        expect(result.current).toMatchObject({
+            data: mockTokenResponse,
+            isFetching: false,
+        });
+        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(2);
         // Clear mock and change userToken to undefined
         jest.clearAllMocks();
         (0, react_1.act)(() => {
@@ -100,17 +133,21 @@ describe("useTamboSessionToken", () => {
         expect(mockAuthApi.getToken).not.toHaveBeenCalled();
     });
     it("should refetch token when userToken changes", async () => {
-        jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);
-        const { result, rerender } = (0, react_1.renderHook)(({ userToken }) => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, userToken), {
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        const { result, rerender } = (0, react_1.renderHook)(({ userToken }) => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, userToken), {
             initialProps: { userToken: "user-token-1" },
         });
+        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);
         await (0, react_1.act)(async () => {
             await jest.runOnlyPendingTimersAsync();
         });
-        expect(result.current).toBe("test-access-token");
-        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);
+        expect(result.current).toMatchObject({
+            data: mockTokenResponse,
+            isFetching: false,
+        });
+        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(2);
         // Mock response for new token
-        jest.mocked(mockAuthApi.getToken).mockResolvedValue({
+        mockAuthApi.getToken.mockResolvedValue({
             ...mockTokenResponse,
             access_token: "new-access-token",
         });
@@ -121,33 +158,46 @@ describe("useTamboSessionToken", () => {
         await (0, react_1.act)(async () => {
             await jest.runOnlyPendingTimersAsync();
         });
-        expect(result.current).toBe("new-access-token");
-        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(2);
+        expect(result.current).toMatchObject({
+            data: { access_token: "new-access-token" },
+            isFetching: false,
+        });
+        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(4);
     });
     it("should reset token when userToken becomes null", async () => {
-        jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);
-        const { result, rerender } = (0, react_1.renderHook)(({ userToken }) => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, userToken), {
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        const { result, rerender } = (0, react_1.renderHook)(({ userToken }) => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, userToken), {
             initialProps: { userToken: "user-token" },
         });
         await (0, react_1.act)(async () => {
             await jest.runOnlyPendingTimersAsync();
         });
-        expect(result.current).toBe("test-access-token");
+        expect(result.current).toMatchObject({
+            data: mockTokenResponse,
+            isFetching: false,
+        });
         // Change userToken to undefined
-        (0, react_1.act)(() => {
-            rerender({ userToken: undefined });
+        rerender({ userToken: undefined });
+        await (0, react_1.act)(async () => {
+            await jest.runAllTimersAsync();
+        });
+        // Token should reset to null (hook doesn't reset it to null when userToken is undefined)
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: false,
         });
-        // Token should remain the same (hook doesn't reset it to null when userToken is undefined)
-        expect(result.current).toBe("test-access-token");
     });
     it("should not update state if component is unmounted during token fetch", async () => {
         let resolvePromise;
         const promise = new Promise((resolve) => {
             resolvePromise = resolve;
         });
-        jest.mocked(mockAuthApi.getToken).mockReturnValue(promise);
-        const { result, unmount } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, "user-token"));
-        expect(result.current).toBeNull();
+        mockAuthApi.getToken.mockReturnValue(promise);
+        const { result, unmount } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, "user-token"));
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: true,
+        });
         // Unmount before the promise resolves
         unmount();
         // Now resolve the promise
@@ -155,7 +205,56 @@ describe("useTamboSessionToken", () => {
             resolvePromise(mockTokenResponse);
         });
         // Token should still be null since component was unmounted
-        expect(result.current).toBeNull();
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: true,
+        });
+    });
+    it("should set isUpdating to true while fetching token", () => {
+        mockAuthApi.getToken.mockImplementation(async () => {
+            return await new Promise(() => { }); // Never resolves
+        });
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, "user-token"));
+        // Should be updating immediately when userToken is provided
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: true,
+        });
+    });
+    it("should set isUpdating to false after token fetch completes", async () => {
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, "user-token"));
+        // Should be updating initially
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: true,
+        });
+        await (0, react_1.act)(async () => {
+            await jest.runOnlyPendingTimersAsync();
+        });
+        // Should not be updating after completion
+        expect(result.current).toMatchObject({
+            data: mockTokenResponse,
+            isFetching: false,
+        });
+    });
+    it("should set isUpdating to false after token fetch fails", async () => {
+        mockAuthApi.getToken.mockRejectedValue(new Error("Token fetch failed"));
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_session_token_1.useTamboSessionToken)(mockTamboAI, queryClient, "user-token"));
+        // Should be updating initially
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: true,
+        });
+        await (0, react_1.act)(async () => {
+            await jest.runAllTimersAsync();
+        });
+        // Should not be updating after failure
+        expect(result.current).toMatchObject({
+            data: undefined,
+            error: expect.any(Error),
+            isFetching: false,
+        });
     });
 });
 //# sourceMappingURL=use-tambo-session-token.test.js.map

package/dist/providers/hooks/__tests__/use-tambo-session-token.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-session-token.test.js","sourceRoot":"","sources":["../../../../src/providers/hooks/__tests__/use-tambo-session-token.test.tsx"],"names":[],"mappings":";;AACA,kDAAyD;AAEzD,wEAAkE;AAIlE,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,MAAM,iBAAiB,GAAG;QACxB,YAAY,EAAE,mBAAmB;QACjC,UAAU,EAAE,IAAI,EAAE,SAAS;QAC3B,UAAU,EAAE,QAAQ;KACrB,CAAC;IAEF,MAAM,WAAW,GAAG;QAClB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;KAC2B,CAAC;IAEjD,MAAM,QAAQ,GAAG;QACf,IAAI,EAAE,WAAW;KACe,CAAC;IAEnC,MAAM,WAAW,GAAG;QAClB,MAAM,EAAE,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,MAAM,EAAE,EAAE;KACoC,CAAC;IAEjD,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,aAAa,EAAE,CAAC;QACpB,WAAmB,CAAC,MAAM,GAAG,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC5B,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,EAAC,WAAW,EAAE,SAAS,CAAC,CAC7C,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;QAClC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,EAAC,WAAW,EAAE,YAAY,CAAC,CAChD,CAAC;QAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACrD,qDAAqD;QACrD,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACtD,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAEvE,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,8CAAoB,EAAC,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;QAElE,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACpE,MAAM,kBAAkB,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9D,MAAM,YAAY,GAAG,IAAI,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAE7D,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CACzC,iDAAiD,CAClD,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC7D,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CACjD,+CAA+C,CAChD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,EAAC,WAAW,EAAE,YAAY,CAAC,CAChD,CAAC;QAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,mBAAmB,GAAG;YAC1B,YAAY,EAAE,qBAAqB;YACnC,UAAU,EAAE,IAAI,EAAE,UAAU;YAC5B,UAAU,EAAE,QAAQ;SACrB,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;QAEzE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,EAAC,WAAW,EAAE,YAAY,CAAC,CAChD,CAAC;QAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACnD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAEvE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,IAAA,8CAAoB,EAAC,WAAW,EAAE,SAAS,CAAC,EAC/D;YACE,YAAY,EAAE,EAAE,SAAS,EAAE,YAAkC,EAAE;SAChE,CACF,CAAC;QAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEtD,+CAA+C;QAC/C,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,QAAQ,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAEvE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,IAAA,8CAAoB,EAAC,WAAW,EAAE,SAAS,CAAC,EAC/D;YACE,YAAY,EAAE,EAAE,SAAS,EAAE,cAAc,EAAE;SAC5C,CACF,CAAC;QAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEtD,8BAA8B;QAC9B,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC;YAClD,GAAG,iBAAiB;YACpB,YAAY,EAAE,kBAAkB;SACjC,CAAC,CAAC;QAEH,mBAAmB;QACnB,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,QAAQ,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAChD,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAEvE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,IAAA,8CAAoB,EAAC,WAAW,EAAE,SAAS,CAAC,EAC/D;YACE,YAAY,EAAE,EAAE,SAAS,EAAE,YAAkC,EAAE;SAChE,CACF,CAAC;QAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAEjD,gCAAgC;QAChC,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,QAAQ,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,2FAA2F;QAC3F,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,IAAI,cAAoC,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACtC,cAAc,GAAG,OAAO,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAE3D,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAC1C,IAAA,8CAAoB,EAAC,WAAW,EAAE,YAAY,CAAC,CAChD,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;QAElC,sCAAsC;QACtC,OAAO,EAAE,CAAC;QAEV,0BAA0B;QAC1B,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,cAAe,CAAC,iBAAiB,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,2DAA2D;QAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;IACpC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { act, renderHook } from \"@testing-library/react\";\nimport { DeepPartial } from \"ts-essentials\";\nimport { useTamboSessionToken } from \"../use-tambo-session-token\";\n\ntype PartialTamboAI = DeepPartial<TamboAI>;\n\ndescribe(\"useTamboSessionToken\", () => {\n  const mockTokenResponse = {\n    access_token: \"test-access-token\",\n    expires_in: 3600, // 1 hour\n    token_type: \"Bearer\",\n  };\n\n  const mockAuthApi = {\n    getToken: jest.fn(),\n  } satisfies DeepPartial<TamboAI[\"beta\"][\"auth\"]>;\n\n  const mockBeta = {\n    auth: mockAuthApi,\n  } satisfies PartialTamboAI[\"beta\"];\n\n  const mockTamboAI = {\n    apiKey: \"\",\n    beta: mockBeta,\n    bearer: \"\",\n  } satisfies PartialTamboAI as unknown as TamboAI;\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n    jest.clearAllTimers();\n    jest.useFakeTimers();\n    (mockTamboAI as any).bearer = \"\";\n  });\n\n  afterEach(() => {\n    jest.runOnlyPendingTimers();\n    jest.useRealTimers();\n  });\n\n  it(\"should return null initially when no userToken is provided\", () => {\n    const { result } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, undefined),\n    );\n\n    expect(result.current).toBeNull();\n    expect(mockAuthApi.getToken).not.toHaveBeenCalled();\n  });\n\n  it(\"should fetch and return session token when userToken is provided\", async () => {\n    jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);\n\n    const { result } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, \"user-token\"),\n    );\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toBe(\"test-access-token\");\n    expect(mockTamboAI.bearer).toBe(\"test-access-token\");\n    // Verify the hook was called with correct parameters\n    expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);\n    expect(mockAuthApi.getToken).toHaveBeenCalledWith(expect.any(Object));\n  });\n\n  it(\"should call getToken with correct token exchange parameters\", async () => {\n    jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);\n\n    renderHook(() => useTamboSessionToken(mockTamboAI, \"user-token\"));\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    const callArgs = jest.mocked(mockAuthApi.getToken).mock.calls[0][0];\n    const tokenRequestString = new TextDecoder().decode(callArgs);\n    const tokenRequest = new URLSearchParams(tokenRequestString);\n\n    expect(tokenRequest.get(\"grant_type\")).toBe(\n      \"urn:ietf:params:oauth:grant-type:token-exchange\",\n    );\n    expect(tokenRequest.get(\"subject_token\")).toBe(\"user-token\");\n    expect(tokenRequest.get(\"subject_token_type\")).toBe(\n      \"urn:ietf:params:oauth:token-type:access_token\",\n    );\n  });\n\n  it(\"should set bearer token on client\", async () => {\n    jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);\n\n    const { result } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, \"user-token\"),\n    );\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toBe(\"test-access-token\");\n    expect(mockTamboAI.bearer).toBe(\"test-access-token\");\n  });\n\n  it(\"should handle different token responses\", async () => {\n    const customTokenResponse = {\n      access_token: \"custom-access-token\",\n      expires_in: 7200, // 2 hours\n      token_type: \"Bearer\",\n    };\n\n    jest.mocked(mockAuthApi.getToken).mockResolvedValue(customTokenResponse);\n\n    const { result } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, \"user-token\"),\n    );\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toBe(\"custom-access-token\");\n    expect(mockTamboAI.bearer).toBe(\"custom-access-token\");\n  });\n\n  it(\"should not fetch token when userToken changes to undefined\", async () => {\n    jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);\n\n    const { result, rerender } = renderHook(\n      ({ userToken }) => useTamboSessionToken(mockTamboAI, userToken),\n      {\n        initialProps: { userToken: \"user-token\" as string | undefined },\n      },\n    );\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toBe(\"test-access-token\");\n    expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);\n\n    // Clear mock and change userToken to undefined\n    jest.clearAllMocks();\n\n    act(() => {\n      rerender({ userToken: undefined });\n    });\n\n    expect(mockAuthApi.getToken).not.toHaveBeenCalled();\n  });\n\n  it(\"should refetch token when userToken changes\", async () => {\n    jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);\n\n    const { result, rerender } = renderHook(\n      ({ userToken }) => useTamboSessionToken(mockTamboAI, userToken),\n      {\n        initialProps: { userToken: \"user-token-1\" },\n      },\n    );\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toBe(\"test-access-token\");\n    expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);\n\n    // Mock response for new token\n    jest.mocked(mockAuthApi.getToken).mockResolvedValue({\n      ...mockTokenResponse,\n      access_token: \"new-access-token\",\n    });\n\n    // Change userToken\n    act(() => {\n      rerender({ userToken: \"user-token-2\" });\n    });\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toBe(\"new-access-token\");\n    expect(mockAuthApi.getToken).toHaveBeenCalledTimes(2);\n  });\n\n  it(\"should reset token when userToken becomes null\", async () => {\n    jest.mocked(mockAuthApi.getToken).mockResolvedValue(mockTokenResponse);\n\n    const { result, rerender } = renderHook(\n      ({ userToken }) => useTamboSessionToken(mockTamboAI, userToken),\n      {\n        initialProps: { userToken: \"user-token\" as string | undefined },\n      },\n    );\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toBe(\"test-access-token\");\n\n    // Change userToken to undefined\n    act(() => {\n      rerender({ userToken: undefined });\n    });\n\n    // Token should remain the same (hook doesn't reset it to null when userToken is undefined)\n    expect(result.current).toBe(\"test-access-token\");\n  });\n\n  it(\"should not update state if component is unmounted during token fetch\", async () => {\n    let resolvePromise: (value: any) => void;\n    const promise = new Promise((resolve) => {\n      resolvePromise = resolve;\n    });\n\n    jest.mocked(mockAuthApi.getToken).mockReturnValue(promise);\n\n    const { result, unmount } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, \"user-token\"),\n    );\n\n    expect(result.current).toBeNull();\n\n    // Unmount before the promise resolves\n    unmount();\n\n    // Now resolve the promise\n    act(() => {\n      resolvePromise!(mockTokenResponse);\n    });\n\n    // Token should still be null since component was unmounted\n    expect(result.current).toBeNull();\n  });\n});\n"]}
+{"version":3,"file":"use-tambo-session-token.test.js","sourceRoot":"","sources":["../../../../src/providers/hooks/__tests__/use-tambo-session-token.test.tsx"],"names":[],"mappings":";;AACA,uDAAoD;AACpD,kDAAyD;AAEzD,wEAAkE;AAIlE,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,MAAM,iBAAiB,GAAG;QACxB,YAAY,EAAE,mBAAmB;QACjC,UAAU,EAAE,IAAI,EAAE,SAAS;QAC3B,UAAU,EAAE,QAAQ;KACrB,CAAC;IAEF,MAAM,WAAW,GAAG;QAClB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;KAC2B,CAAC;IAEjD,MAAM,QAAQ,GAAG;QACf,IAAI,EAAE,WAAW;KACe,CAAC;IAEnC,MAAM,WAAW,GAAG;QAClB,MAAM,EAAE,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,MAAM,EAAE,EAAE;KACoC,CAAC;IACjD,MAAM,WAAW,GAAG,IAAI,yBAAW,CAAC;QAClC,cAAc,EAAE;YACd,OAAO,EAAE;gBACP,KAAK,EAAE,KAAK;gBACZ,oBAAoB,EAAE,KAAK;aAC5B;SACF;KACF,CAAC,CAAC;IAEH,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,aAAa,EAAE,CAAC;QACpB,WAAmB,CAAC,MAAM,GAAG,EAAE,CAAC;QACjC,WAAW,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,SAAS,CAAC,CAC1D,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAE1D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEtD,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,iBAAiB;YACvB,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAChE,qDAAqD;QACrD,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACtD,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAE1D,IAAA,kBAAU,EAAC,GAAG,EAAE,CACd,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;QAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACvD,MAAM,kBAAkB,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9D,MAAM,YAAY,GAAG,IAAI,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAE7D,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CACzC,iDAAiD,CAClD,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC7D,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CACjD,+CAA+C,CAChD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAE1D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;QAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,iBAAiB;SACxB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,mBAAmB,GAAG;YAC1B,YAAY,EAAE,qBAAqB;YACnC,UAAU,EAAE,IAAI,EAAE,UAAU;YAC5B,UAAU,EAAE,QAAQ;SACrB,CAAC;QAEF,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;QAE5D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QAEH,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,mBAAmB;YACzB,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAE1D,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAChB,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,SAAS,CAAC,EAC3D;YACE,YAAY,EAAE,EAAE,SAAS,EAAE,YAAkC,EAAE;SAChE,CACF,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEtD,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,iBAAiB;YACvB,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEtD,+CAA+C;QAC/C,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,QAAQ,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAE1D,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAChB,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,SAAS,CAAC,EAC3D;YACE,YAAY,EAAE,EAAE,SAAS,EAAE,cAAc,EAAE;SAC5C,CACF,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEtD,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,iBAAiB;YACvB,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEtD,8BAA8B;QAC9B,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC;YACrC,GAAG,iBAAiB;YACpB,YAAY,EAAE,kBAAkB;SACjC,CAAC,CAAC;QAEH,mBAAmB;QACnB,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,QAAQ,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,EAAE,YAAY,EAAE,kBAAkB,EAAE;YAC1C,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAE1D,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAChB,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,SAAS,CAAC,EAC3D;YACE,YAAY,EAAE,EAAE,SAAS,EAAE,YAAkC,EAAE;SAChE,CACF,CAAC;QAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,iBAAiB;YACvB,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;QAEH,gCAAgC;QAChC,QAAQ,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;QACnC,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,yFAAyF;QACzF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,IAAI,cAAoC,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACtC,cAAc,GAAG,OAAO,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,WAAW,CAAC,QAAQ,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAE9C,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAC1C,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QAEH,sCAAsC;QACtC,OAAO,EAAE,CAAC;QAEV,0BAA0B;QAC1B,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,cAAe,CAAC,iBAAiB,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,2DAA2D;QAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,WAAW,CAAC,QAAQ,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;YACjD,OAAO,MAAM,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC,iBAAiB;QACvD,CAAC,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;QAEF,4DAA4D;QAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAE1D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;QAEF,+BAA+B;QAC/B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QAEH,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,0CAA0C;QAC1C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,iBAAiB;YACvB,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,WAAW,CAAC,QAAQ,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC;QAExE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,EAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;QAEF,+BAA+B;QAC/B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QAEH,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,uCAAuC;QACvC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,SAAS;YACf,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;YACxB,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { QueryClient } from \"@tanstack/react-query\";\nimport { act, renderHook } from \"@testing-library/react\";\nimport { DeepPartial } from \"ts-essentials\";\nimport { useTamboSessionToken } from \"../use-tambo-session-token\";\n\ntype PartialTamboAI = DeepPartial<TamboAI>;\n\ndescribe(\"useTamboSessionToken\", () => {\n  const mockTokenResponse = {\n    access_token: \"test-access-token\",\n    expires_in: 3600, // 1 hour\n    token_type: \"Bearer\",\n  };\n\n  const mockAuthApi = {\n    getToken: jest.fn(),\n  } satisfies DeepPartial<TamboAI[\"beta\"][\"auth\"]>;\n\n  const mockBeta = {\n    auth: mockAuthApi,\n  } satisfies PartialTamboAI[\"beta\"];\n\n  const mockTamboAI = {\n    apiKey: \"\",\n    beta: mockBeta,\n    bearer: \"\",\n  } satisfies PartialTamboAI as unknown as TamboAI;\n  const queryClient = new QueryClient({\n    defaultOptions: {\n      queries: {\n        retry: false,\n        refetchOnWindowFocus: false,\n      },\n    },\n  });\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n    jest.clearAllTimers();\n    jest.useFakeTimers();\n    (mockTamboAI as any).bearer = \"\";\n    queryClient.clear();\n  });\n\n  afterEach(async () => {\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n    jest.useRealTimers();\n    jest.resetAllMocks();\n  });\n\n  it(\"should return null initially when no userToken is provided\", () => {\n    const { result } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, queryClient, undefined),\n    );\n\n    expect(result.current).toMatchObject({\n      data: undefined,\n      isFetching: false,\n    });\n    expect(mockAuthApi.getToken).not.toHaveBeenCalled();\n  });\n\n  it(\"should fetch and return session token when userToken is provided\", async () => {\n    mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);\n\n    const { result } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, queryClient, \"user-token\"),\n    );\n    expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toMatchObject({\n      data: mockTokenResponse,\n      isFetching: false,\n    });\n    expect(mockTamboAI.bearer).toBe(mockTokenResponse.access_token);\n    // Verify the hook was called with correct parameters\n    expect(mockAuthApi.getToken).toHaveBeenCalledTimes(2);\n    expect(mockAuthApi.getToken).toHaveBeenCalledWith(expect.any(Object));\n  });\n\n  it(\"should call getToken with correct token exchange parameters\", async () => {\n    mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);\n\n    renderHook(() =>\n      useTamboSessionToken(mockTamboAI, queryClient, \"user-token\"),\n    );\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    const callArgs = mockAuthApi.getToken.mock.calls[0][0];\n    const tokenRequestString = new TextDecoder().decode(callArgs);\n    const tokenRequest = new URLSearchParams(tokenRequestString);\n\n    expect(tokenRequest.get(\"grant_type\")).toBe(\n      \"urn:ietf:params:oauth:grant-type:token-exchange\",\n    );\n    expect(tokenRequest.get(\"subject_token\")).toBe(\"user-token\");\n    expect(tokenRequest.get(\"subject_token_type\")).toBe(\n      \"urn:ietf:params:oauth:token-type:access_token\",\n    );\n  });\n\n  it(\"should set bearer token on client\", async () => {\n    mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);\n\n    const { result } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, queryClient, \"user-token\"),\n    );\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toMatchObject({\n      data: mockTokenResponse,\n    });\n    expect(mockTamboAI.bearer).toBe(\"test-access-token\");\n  });\n\n  it(\"should handle different token responses\", async () => {\n    const customTokenResponse = {\n      access_token: \"custom-access-token\",\n      expires_in: 7200, // 2 hours\n      token_type: \"Bearer\",\n    };\n\n    mockAuthApi.getToken.mockResolvedValue(customTokenResponse);\n\n    const { result } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, queryClient, \"user-token\"),\n    );\n    expect(result.current).toMatchObject({\n      data: undefined,\n      isFetching: true,\n    });\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toMatchObject({\n      data: customTokenResponse,\n      isFetching: false,\n    });\n    expect(mockTamboAI.bearer).toBe(\"custom-access-token\");\n  });\n\n  it(\"should not fetch token when userToken changes to undefined\", async () => {\n    mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);\n\n    const { result, rerender } = renderHook(\n      ({ userToken }) =>\n        useTamboSessionToken(mockTamboAI, queryClient, userToken),\n      {\n        initialProps: { userToken: \"user-token\" as string | undefined },\n      },\n    );\n    expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toMatchObject({\n      data: mockTokenResponse,\n      isFetching: false,\n    });\n    expect(mockAuthApi.getToken).toHaveBeenCalledTimes(2);\n\n    // Clear mock and change userToken to undefined\n    jest.clearAllMocks();\n\n    act(() => {\n      rerender({ userToken: undefined });\n    });\n\n    expect(mockAuthApi.getToken).not.toHaveBeenCalled();\n  });\n\n  it(\"should refetch token when userToken changes\", async () => {\n    mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);\n\n    const { result, rerender } = renderHook(\n      ({ userToken }) =>\n        useTamboSessionToken(mockTamboAI, queryClient, userToken),\n      {\n        initialProps: { userToken: \"user-token-1\" },\n      },\n    );\n    expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toMatchObject({\n      data: mockTokenResponse,\n      isFetching: false,\n    });\n    expect(mockAuthApi.getToken).toHaveBeenCalledTimes(2);\n\n    // Mock response for new token\n    mockAuthApi.getToken.mockResolvedValue({\n      ...mockTokenResponse,\n      access_token: \"new-access-token\",\n    });\n\n    // Change userToken\n    act(() => {\n      rerender({ userToken: \"user-token-2\" });\n    });\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toMatchObject({\n      data: { access_token: \"new-access-token\" },\n      isFetching: false,\n    });\n    expect(mockAuthApi.getToken).toHaveBeenCalledTimes(4);\n  });\n\n  it(\"should reset token when userToken becomes null\", async () => {\n    mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);\n\n    const { result, rerender } = renderHook(\n      ({ userToken }) =>\n        useTamboSessionToken(mockTamboAI, queryClient, userToken),\n      {\n        initialProps: { userToken: \"user-token\" as string | undefined },\n      },\n    );\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    expect(result.current).toMatchObject({\n      data: mockTokenResponse,\n      isFetching: false,\n    });\n\n    // Change userToken to undefined\n    rerender({ userToken: undefined });\n    await act(async () => {\n      await jest.runAllTimersAsync();\n    });\n\n    // Token should reset to null (hook doesn't reset it to null when userToken is undefined)\n    expect(result.current).toMatchObject({\n      data: undefined,\n      isFetching: false,\n    });\n  });\n\n  it(\"should not update state if component is unmounted during token fetch\", async () => {\n    let resolvePromise: (value: any) => void;\n    const promise = new Promise((resolve) => {\n      resolvePromise = resolve;\n    });\n\n    mockAuthApi.getToken.mockReturnValue(promise);\n\n    const { result, unmount } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, queryClient, \"user-token\"),\n    );\n\n    expect(result.current).toMatchObject({\n      data: undefined,\n      isFetching: true,\n    });\n\n    // Unmount before the promise resolves\n    unmount();\n\n    // Now resolve the promise\n    act(() => {\n      resolvePromise!(mockTokenResponse);\n    });\n\n    // Token should still be null since component was unmounted\n    expect(result.current).toMatchObject({\n      data: undefined,\n      isFetching: true,\n    });\n  });\n\n  it(\"should set isUpdating to true while fetching token\", () => {\n    mockAuthApi.getToken.mockImplementation(async () => {\n      return await new Promise(() => {}); // Never resolves\n    });\n\n    const { result } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, queryClient, \"user-token\"),\n    );\n\n    // Should be updating immediately when userToken is provided\n    expect(result.current).toMatchObject({\n      data: undefined,\n      isFetching: true,\n    });\n  });\n\n  it(\"should set isUpdating to false after token fetch completes\", async () => {\n    mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);\n\n    const { result } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, queryClient, \"user-token\"),\n    );\n\n    // Should be updating initially\n    expect(result.current).toMatchObject({\n      data: undefined,\n      isFetching: true,\n    });\n\n    await act(async () => {\n      await jest.runOnlyPendingTimersAsync();\n    });\n\n    // Should not be updating after completion\n    expect(result.current).toMatchObject({\n      data: mockTokenResponse,\n      isFetching: false,\n    });\n  });\n\n  it(\"should set isUpdating to false after token fetch fails\", async () => {\n    mockAuthApi.getToken.mockRejectedValue(new Error(\"Token fetch failed\"));\n\n    const { result } = renderHook(() =>\n      useTamboSessionToken(mockTamboAI, queryClient, \"user-token\"),\n    );\n\n    // Should be updating initially\n    expect(result.current).toMatchObject({\n      data: undefined,\n      isFetching: true,\n    });\n\n    await act(async () => {\n      await jest.runAllTimersAsync();\n    });\n\n    // Should not be updating after failure\n    expect(result.current).toMatchObject({\n      data: undefined,\n      error: expect.any(Error),\n      isFetching: false,\n    });\n  });\n});\n"]}

package/dist/providers/hooks/use-tambo-session-token.d.ts

@@ -1,4 +1,5 @@
 import TamboAI from "@tambo-ai/typescript-sdk";
+import { QueryClient } from "@tanstack/react-query";
 /**
  * This internal hook is used to get the Tambo session token and keep it
  * refreshed.
@@ -8,8 +9,9 @@ import TamboAI from "@tambo-ai/typescript-sdk";
  *
  * This hook is used by the TamboClientProvider.
  * @param client - The Tambo client.
+ * @param queryClient - The query client.
  * @param userToken - The user token.
  * @returns The Tambo session token.
  */
-export declare function useTamboSessionToken(client: TamboAI, userToken: string | undefined): string | null;
+export declare function useTamboSessionToken(client: TamboAI, queryClient: QueryClient, userToken: string | undefined): import("@tanstack/react-query").UseQueryResult<TamboAI.Beta.Auth.AuthGetTokenResponse, Error>;
 //# sourceMappingURL=use-tambo-session-token.d.ts.map

package/dist/providers/hooks/use-tambo-session-token.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-session-token.d.ts","sourceRoot":"","sources":["../../../src/providers/hooks/use-tambo-session-token.tsx"],"names":[],"mappings":"AACA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAG/C;;;;;;;;;;;GAWG;AACH,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,OAAO,EACf,SAAS,EAAE,MAAM,GAAG,SAAS,iBAiE9B"}
+{"version":3,"file":"use-tambo-session-token.d.ts","sourceRoot":"","sources":["../../../src/providers/hooks/use-tambo-session-token.tsx"],"names":[],"mappings":"AACA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAY,MAAM,uBAAuB,CAAC;AAG9D;;;;;;;;;;;;GAYG;AACH,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,OAAO,EACf,WAAW,EAAE,WAAW,EACxB,SAAS,EAAE,MAAM,GAAG,SAAS,iGAmC9B"}

package/dist/providers/hooks/use-tambo-session-token.js

@@ -2,6 +2,7 @@
 "use client";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.useTamboSessionToken = useTamboSessionToken;
+const react_query_1 = require("@tanstack/react-query");
 const react_1 = require("react");
 /**
  * This internal hook is used to get the Tambo session token and keep it
@@ -12,54 +13,36 @@ const react_1 = require("react");
  *
  * This hook is used by the TamboClientProvider.
  * @param client - The Tambo client.
+ * @param queryClient - The query client.
  * @param userToken - The user token.
  * @returns The Tambo session token.
  */
-function useTamboSessionToken(client, userToken) {
-    const [tamboSessionToken, setTamboSessionToken] = (0, react_1.useState)(null);
-    // we need to set this to true when the token is expired, this is effectively
-    // like a dirty bit, which will trigger a new useEffect()
-    const [isExpired, setIsExpired] = (0, react_1.useState)(true);
-    (0, react_1.useEffect)(() => {
-        let expireTimer = null;
-        async function updateToken(subjectToken, abortController) {
-            if (abortController.signal.aborted || !userToken) {
-                return;
-            }
+function useTamboSessionToken(client, queryClient, userToken) {
+    const result = (0, react_query_1.useQuery)({
+        queryKey: ["tambo-session-token", userToken],
+        queryFn: async () => {
             const tokenRequest = {
                 grant_type: "urn:ietf:params:oauth:grant-type:token-exchange",
-                subject_token: subjectToken,
+                // will only be undefined if the userToken is not provided, in which case the query will be disabled.
+                subject_token: userToken,
                 subject_token_type: "urn:ietf:params:oauth:token-type:access_token",
             };
             const tokenRequestFormEncoded = new URLSearchParams(tokenRequest).toString();
             const tokenAsArrayBuffer = new TextEncoder().encode(tokenRequestFormEncoded);
-            const tamboToken = await client.beta.auth.getToken(tokenAsArrayBuffer);
-            if (abortController.signal.aborted) {
-                return;
+            return await client.beta.auth.getToken(tokenAsArrayBuffer);
+        },
+        enabled: !!userToken,
+        refetchInterval: (result) => {
+            if (result.state.data?.expires_in) {
+                return result.state.data.expires_in * 1000;
             }
-            setTamboSessionToken(tamboToken.access_token);
-            client.bearer = tamboToken.access_token;
-            // we need to set a timer to refresh the token when it expires
-            const refreshTime = Math.max(tamboToken.expires_in - 60, 0);
-            // careful with the assignment here: since this is an async function, this
-            // code is executed outside the of the scope of the useEffect() hook, so
-            // we need to use a let variable to store the timer
-            expireTimer = setTimeout(() => {
-                setIsExpired(true);
-            }, refreshTime * 1000);
-        }
-        const abortController = new AbortController();
-        if (userToken && isExpired) {
-            updateToken(userToken, abortController);
-        }
-        return () => {
-            // This fires when the component unmounts or the userToken changes
-            abortController.abort();
-            if (expireTimer) {
-                clearTimeout(expireTimer);
-            }
-        };
-    }, [client, isExpired, userToken]);
-    return tamboSessionToken;
+            return false;
+        },
+    }, queryClient);
+    const accessToken = result.data?.access_token ?? null;
+    (0, react_1.useEffect)(() => {
+        client.bearer = accessToken;
+    }, [accessToken, client]);
+    return result;
 }
 //# sourceMappingURL=use-tambo-session-token.js.map

package/dist/providers/hooks/use-tambo-session-token.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-session-token.js","sourceRoot":"","sources":["../../../src/providers/hooks/use-tambo-session-token.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AAgBb,oDAmEC;AAjFD,iCAA4C;AAE5C;;;;;;;;;;;GAWG;AACH,SAAgB,oBAAoB,CAClC,MAAe,EACf,SAA6B;IAE7B,MAAM,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,GAAG,IAAA,gBAAQ,EACxD,IAAI,CACL,CAAC;IACF,6EAA6E;IAC7E,yDAAyD;IACzD,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,IAAA,gBAAQ,EAAC,IAAI,CAAC,CAAC;IACjD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,WAAW,GAA0B,IAAI,CAAC;QAE9C,KAAK,UAAU,WAAW,CACxB,YAAoB,EACpB,eAAgC;YAEhC,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC;gBACjD,OAAO;YACT,CAAC;YACD,MAAM,YAAY,GAAG;gBACnB,UAAU,EAAE,iDAAiD;gBAC7D,aAAa,EAAE,YAAY;gBAC3B,kBAAkB,EAAE,+CAA+C;aACpE,CAAC;YACF,MAAM,uBAAuB,GAAG,IAAI,eAAe,CACjD,YAAY,CACb,CAAC,QAAQ,EAAE,CAAC;YACb,MAAM,kBAAkB,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CACjD,uBAAuB,CACxB,CAAC;YACF,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAChD,kBAAyB,CAC1B,CAAC;YAEF,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnC,OAAO;YACT,CAAC;YACD,oBAAoB,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,GAAG,UAAU,CAAC,YAAY,CAAC;YAExC,8DAA8D;YAC9D,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,UAAU,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;YAE5D,0EAA0E;YAC1E,wEAAwE;YACxE,mDAAmD;YACnD,WAAW,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,YAAY,CAAC,IAAI,CAAC,CAAC;YACrB,CAAC,EAAE,WAAW,GAAG,IAAI,CAAC,CAAC;QACzB,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC9C,IAAI,SAAS,IAAI,SAAS,EAAE,CAAC;YAC3B,WAAW,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;QAC1C,CAAC;QAED,OAAO,GAAG,EAAE;YACV,kEAAkE;YAClE,eAAe,CAAC,KAAK,EAAE,CAAC;YACxB,IAAI,WAAW,EAAE,CAAC;gBAChB,YAAY,CAAC,WAAW,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;IAEnC,OAAO,iBAAiB,CAAC;AAC3B,CAAC","sourcesContent":["\"use client\";\nimport TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { useEffect, useState } from \"react\";\n\n/**\n * This internal hook is used to get the Tambo session token and keep it\n * refreshed.\n *\n * It will refresh the token when it expires.\n * It will also set the bearer token on the client.\n *\n * This hook is used by the TamboClientProvider.\n * @param client - The Tambo client.\n * @param userToken - The user token.\n * @returns The Tambo session token.\n */\nexport function useTamboSessionToken(\n  client: TamboAI,\n  userToken: string | undefined,\n) {\n  const [tamboSessionToken, setTamboSessionToken] = useState<string | null>(\n    null,\n  );\n  // we need to set this to true when the token is expired, this is effectively\n  // like a dirty bit, which will trigger a new useEffect()\n  const [isExpired, setIsExpired] = useState(true);\n  useEffect(() => {\n    let expireTimer: NodeJS.Timeout | null = null;\n\n    async function updateToken(\n      subjectToken: string,\n      abortController: AbortController,\n    ) {\n      if (abortController.signal.aborted || !userToken) {\n        return;\n      }\n      const tokenRequest = {\n        grant_type: \"urn:ietf:params:oauth:grant-type:token-exchange\",\n        subject_token: subjectToken,\n        subject_token_type: \"urn:ietf:params:oauth:token-type:access_token\",\n      };\n      const tokenRequestFormEncoded = new URLSearchParams(\n        tokenRequest,\n      ).toString();\n      const tokenAsArrayBuffer = new TextEncoder().encode(\n        tokenRequestFormEncoded,\n      );\n      const tamboToken = await client.beta.auth.getToken(\n        tokenAsArrayBuffer as any,\n      );\n\n      if (abortController.signal.aborted) {\n        return;\n      }\n      setTamboSessionToken(tamboToken.access_token);\n      client.bearer = tamboToken.access_token;\n\n      // we need to set a timer to refresh the token when it expires\n      const refreshTime = Math.max(tamboToken.expires_in - 60, 0);\n\n      // careful with the assignment here: since this is an async function, this\n      // code is executed outside the of the scope of the useEffect() hook, so\n      // we need to use a let variable to store the timer\n      expireTimer = setTimeout(() => {\n        setIsExpired(true);\n      }, refreshTime * 1000);\n    }\n\n    const abortController = new AbortController();\n    if (userToken && isExpired) {\n      updateToken(userToken, abortController);\n    }\n\n    return () => {\n      // This fires when the component unmounts or the userToken changes\n      abortController.abort();\n      if (expireTimer) {\n        clearTimeout(expireTimer);\n      }\n    };\n  }, [client, isExpired, userToken]);\n\n  return tamboSessionToken;\n}\n"]}
+{"version":3,"file":"use-tambo-session-token.js","sourceRoot":"","sources":["../../../src/providers/hooks/use-tambo-session-token.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AAkBb,oDAsCC;AAtDD,uDAA8D;AAC9D,iCAAkC;AAElC;;;;;;;;;;;;GAYG;AACH,SAAgB,oBAAoB,CAClC,MAAe,EACf,WAAwB,EACxB,SAA6B;IAE7B,MAAM,MAAM,GAAG,IAAA,sBAAQ,EACrB;QACE,QAAQ,EAAE,CAAC,qBAAqB,EAAE,SAAS,CAAC;QAC5C,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,MAAM,YAAY,GAAG;gBACnB,UAAU,EAAE,iDAAiD;gBAC7D,qGAAqG;gBACrG,aAAa,EAAE,SAAU;gBACzB,kBAAkB,EAAE,+CAA+C;aACpE,CAAC;YACF,MAAM,uBAAuB,GAAG,IAAI,eAAe,CACjD,YAAY,CACb,CAAC,QAAQ,EAAE,CAAC;YACb,MAAM,kBAAkB,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CACjD,uBAAuB,CACxB,CAAC;YACF,OAAO,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,kBAAyB,CAAC,CAAC;QACpE,CAAC;QACD,OAAO,EAAE,CAAC,CAAC,SAAS;QACpB,eAAe,EAAE,CAAC,MAAM,EAAE,EAAE;YAC1B,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,UAAU,EAAE,CAAC;gBAClC,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YAC7C,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;KACF,EACD,WAAW,CACZ,CAAC;IACF,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,EAAE,YAAY,IAAI,IAAI,CAAC;IACtD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,MAAM,CAAC,MAAM,GAAG,WAAW,CAAC;IAC9B,CAAC,EAAE,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC;IAC1B,OAAO,MAAM,CAAC;AAChB,CAAC","sourcesContent":["\"use client\";\nimport TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { QueryClient, useQuery } from \"@tanstack/react-query\";\nimport { useEffect } from \"react\";\n\n/**\n * This internal hook is used to get the Tambo session token and keep it\n * refreshed.\n *\n * It will refresh the token when it expires.\n * It will also set the bearer token on the client.\n *\n * This hook is used by the TamboClientProvider.\n * @param client - The Tambo client.\n * @param queryClient - The query client.\n * @param userToken - The user token.\n * @returns The Tambo session token.\n */\nexport function useTamboSessionToken(\n  client: TamboAI,\n  queryClient: QueryClient,\n  userToken: string | undefined,\n) {\n  const result = useQuery(\n    {\n      queryKey: [\"tambo-session-token\", userToken],\n      queryFn: async () => {\n        const tokenRequest = {\n          grant_type: \"urn:ietf:params:oauth:grant-type:token-exchange\",\n          // will only be undefined if the userToken is not provided, in which case the query will be disabled.\n          subject_token: userToken!,\n          subject_token_type: \"urn:ietf:params:oauth:token-type:access_token\",\n        };\n        const tokenRequestFormEncoded = new URLSearchParams(\n          tokenRequest,\n        ).toString();\n        const tokenAsArrayBuffer = new TextEncoder().encode(\n          tokenRequestFormEncoded,\n        );\n        return await client.beta.auth.getToken(tokenAsArrayBuffer as any);\n      },\n      enabled: !!userToken,\n      refetchInterval: (result) => {\n        if (result.state.data?.expires_in) {\n          return result.state.data.expires_in * 1000;\n        }\n        return false;\n      },\n    },\n    queryClient,\n  );\n  const accessToken = result.data?.access_token ?? null;\n  useEffect(() => {\n    client.bearer = accessToken;\n  }, [accessToken, client]);\n  return result;\n}\n"]}

package/dist/providers/index.d.ts

@@ -1,5 +1,5 @@
 export type { TamboComponent, TamboTool } from "../model/component-metadata";
-export { TamboClientProvider, useTamboClient } from "./tambo-client-provider";
+export { TamboClientProvider, useIsTamboTokenUpdating, useTamboClient, } from "./tambo-client-provider";
 export { TamboComponentProvider, useTamboComponent, } from "./tambo-component-provider";
 export { TamboContextHelpersProvider, useTamboContextHelpers, type TamboContextHelpersContextProps, type TamboContextHelpersProviderProps, } from "./tambo-context-helpers-provider";
 export { Pending, Streaming, Success, TamboPropStreamProvider, useTamboStream, } from "./tambo-prop-stream-provider";
@@ -7,5 +7,6 @@ export type { StreamStateComponentProps, StreamStatus, TamboPropStreamContextVal
 export { TamboContext, TamboProvider, useTambo } from "./tambo-provider";
 export { TamboRegistryProvider, useTamboRegistry, type TamboRegistryContext, } from "./tambo-registry-provider";
 export { TamboStubProvider, type TamboStubProviderProps } from "./tambo-stubs";
+export { TamboThreadInputProvider, useTamboThreadInput, type TamboThreadInputContextProps, } from "./tambo-thread-input-provider";
 export { TamboThreadContext, TamboThreadProvider, useTamboGenerationStage, useTamboThread, } from "./tambo-thread-provider";
 //# sourceMappingURL=index.d.ts.map

package/dist/providers/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAC7E,OAAO,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAC9E,OAAO,EACL,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,2BAA2B,EAC3B,sBAAsB,EACtB,KAAK,+BAA+B,EACpC,KAAK,gCAAgC,GACtC,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,OAAO,EACP,SAAS,EACT,OAAO,EACP,uBAAuB,EACvB,cAAc,GACf,MAAM,8BAA8B,CAAC;AACtC,YAAY,EACV,yBAAyB,EACzB,YAAY,EACZ,2BAA2B,GAC5B,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AACzE,OAAO,EACL,qBAAqB,EACrB,gBAAgB,EAChB,KAAK,oBAAoB,GAC1B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,iBAAiB,EAAE,KAAK,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAC/E,OAAO,EACL,kBAAkB,EAClB,mBAAmB,EACnB,uBAAuB,EACvB,cAAc,GACf,MAAM,yBAAyB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAC7E,OAAO,EACL,mBAAmB,EACnB,uBAAuB,EACvB,cAAc,GACf,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,2BAA2B,EAC3B,sBAAsB,EACtB,KAAK,+BAA+B,EACpC,KAAK,gCAAgC,GACtC,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,OAAO,EACP,SAAS,EACT,OAAO,EACP,uBAAuB,EACvB,cAAc,GACf,MAAM,8BAA8B,CAAC;AACtC,YAAY,EACV,yBAAyB,EACzB,YAAY,EACZ,2BAA2B,GAC5B,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AACzE,OAAO,EACL,qBAAqB,EACrB,gBAAgB,EAChB,KAAK,oBAAoB,GAC1B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,iBAAiB,EAAE,KAAK,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAC/E,OAAO,EACL,wBAAwB,EACxB,mBAAmB,EACnB,KAAK,4BAA4B,GAClC,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,kBAAkB,EAClB,mBAAmB,EACnB,uBAAuB,EACvB,cAAc,GACf,MAAM,yBAAyB,CAAC"}

package/dist/providers/index.js

@@ -1,8 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.useTamboThread = exports.useTamboGenerationStage = exports.TamboThreadProvider = exports.TamboThreadContext = exports.TamboStubProvider = exports.useTamboRegistry = exports.TamboRegistryProvider = exports.useTambo = exports.TamboProvider = exports.TamboContext = exports.useTamboStream = exports.TamboPropStreamProvider = exports.Success = exports.Streaming = exports.Pending = exports.useTamboContextHelpers = exports.TamboContextHelpersProvider = exports.useTamboComponent = exports.TamboComponentProvider = exports.useTamboClient = exports.TamboClientProvider = void 0;
+exports.useTamboThread = exports.useTamboGenerationStage = exports.TamboThreadProvider = exports.TamboThreadContext = exports.useTamboThreadInput = exports.TamboThreadInputProvider = exports.TamboStubProvider = exports.useTamboRegistry = exports.TamboRegistryProvider = exports.useTambo = exports.TamboProvider = exports.TamboContext = exports.useTamboStream = exports.TamboPropStreamProvider = exports.Success = exports.Streaming = exports.Pending = exports.useTamboContextHelpers = exports.TamboContextHelpersProvider = exports.useTamboComponent = exports.TamboComponentProvider = exports.useTamboClient = exports.useIsTamboTokenUpdating = exports.TamboClientProvider = void 0;
 var tambo_client_provider_1 = require("./tambo-client-provider");
 Object.defineProperty(exports, "TamboClientProvider", { enumerable: true, get: function () { return tambo_client_provider_1.TamboClientProvider; } });
+Object.defineProperty(exports, "useIsTamboTokenUpdating", { enumerable: true, get: function () { return tambo_client_provider_1.useIsTamboTokenUpdating; } });
 Object.defineProperty(exports, "useTamboClient", { enumerable: true, get: function () { return tambo_client_provider_1.useTamboClient; } });
 var tambo_component_provider_1 = require("./tambo-component-provider");
 Object.defineProperty(exports, "TamboComponentProvider", { enumerable: true, get: function () { return tambo_component_provider_1.TamboComponentProvider; } });
@@ -25,6 +26,9 @@ Object.defineProperty(exports, "TamboRegistryProvider", { enumerable: true, get:
 Object.defineProperty(exports, "useTamboRegistry", { enumerable: true, get: function () { return tambo_registry_provider_1.useTamboRegistry; } });
 var tambo_stubs_1 = require("./tambo-stubs");
 Object.defineProperty(exports, "TamboStubProvider", { enumerable: true, get: function () { return tambo_stubs_1.TamboStubProvider; } });
+var tambo_thread_input_provider_1 = require("./tambo-thread-input-provider");
+Object.defineProperty(exports, "TamboThreadInputProvider", { enumerable: true, get: function () { return tambo_thread_input_provider_1.TamboThreadInputProvider; } });
+Object.defineProperty(exports, "useTamboThreadInput", { enumerable: true, get: function () { return tambo_thread_input_provider_1.useTamboThreadInput; } });
 var tambo_thread_provider_1 = require("./tambo-thread-provider");
 Object.defineProperty(exports, "TamboThreadContext", { enumerable: true, get: function () { return tambo_thread_provider_1.TamboThreadContext; } });
 Object.defineProperty(exports, "TamboThreadProvider", { enumerable: true, get: function () { return tambo_thread_provider_1.TamboThreadProvider; } });

package/dist/providers/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":";;;AACA,iEAA8E;AAArE,4HAAA,mBAAmB,OAAA;AAAE,uHAAA,cAAc,OAAA;AAC5C,uEAGoC;AAFlC,kIAAA,sBAAsB,OAAA;AACtB,6HAAA,iBAAiB,OAAA;AAEnB,mFAK0C;AAJxC,6IAAA,2BAA2B,OAAA;AAC3B,wIAAA,sBAAsB,OAAA;AAIxB,2EAMsC;AALpC,qHAAA,OAAO,OAAA;AACP,uHAAA,SAAS,OAAA;AACT,qHAAA,OAAO,OAAA;AACP,qIAAA,uBAAuB,OAAA;AACvB,4HAAA,cAAc,OAAA;AAOhB,mDAAyE;AAAhE,8GAAA,YAAY,OAAA;AAAE,+GAAA,aAAa,OAAA;AAAE,0GAAA,QAAQ,OAAA;AAC9C,qEAImC;AAHjC,gIAAA,qBAAqB,OAAA;AACrB,2HAAA,gBAAgB,OAAA;AAGlB,6CAA+E;AAAtE,gHAAA,iBAAiB,OAAA;AAC1B,iEAKiC;AAJ/B,2HAAA,kBAAkB,OAAA;AAClB,4HAAA,mBAAmB,OAAA;AACnB,gIAAA,uBAAuB,OAAA;AACvB,uHAAA,cAAc,OAAA","sourcesContent":["export type { TamboComponent, TamboTool } from \"../model/component-metadata\";\nexport { TamboClientProvider, useTamboClient } from \"./tambo-client-provider\";\nexport {\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nexport {\n  TamboContextHelpersProvider,\n  useTamboContextHelpers,\n  type TamboContextHelpersContextProps,\n  type TamboContextHelpersProviderProps,\n} from \"./tambo-context-helpers-provider\";\nexport {\n  Pending,\n  Streaming,\n  Success,\n  TamboPropStreamProvider,\n  useTamboStream,\n} from \"./tambo-prop-stream-provider\";\nexport type {\n  StreamStateComponentProps,\n  StreamStatus,\n  TamboPropStreamContextValue,\n} from \"./tambo-prop-stream-provider\";\nexport { TamboContext, TamboProvider, useTambo } from \"./tambo-provider\";\nexport {\n  TamboRegistryProvider,\n  useTamboRegistry,\n  type TamboRegistryContext,\n} from \"./tambo-registry-provider\";\nexport { TamboStubProvider, type TamboStubProviderProps } from \"./tambo-stubs\";\nexport {\n  TamboThreadContext,\n  TamboThreadProvider,\n  useTamboGenerationStage,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":";;;AACA,iEAIiC;AAH/B,4HAAA,mBAAmB,OAAA;AACnB,gIAAA,uBAAuB,OAAA;AACvB,uHAAA,cAAc,OAAA;AAEhB,uEAGoC;AAFlC,kIAAA,sBAAsB,OAAA;AACtB,6HAAA,iBAAiB,OAAA;AAEnB,mFAK0C;AAJxC,6IAAA,2BAA2B,OAAA;AAC3B,wIAAA,sBAAsB,OAAA;AAIxB,2EAMsC;AALpC,qHAAA,OAAO,OAAA;AACP,uHAAA,SAAS,OAAA;AACT,qHAAA,OAAO,OAAA;AACP,qIAAA,uBAAuB,OAAA;AACvB,4HAAA,cAAc,OAAA;AAOhB,mDAAyE;AAAhE,8GAAA,YAAY,OAAA;AAAE,+GAAA,aAAa,OAAA;AAAE,0GAAA,QAAQ,OAAA;AAC9C,qEAImC;AAHjC,gIAAA,qBAAqB,OAAA;AACrB,2HAAA,gBAAgB,OAAA;AAGlB,6CAA+E;AAAtE,gHAAA,iBAAiB,OAAA;AAC1B,6EAIuC;AAHrC,uIAAA,wBAAwB,OAAA;AACxB,kIAAA,mBAAmB,OAAA;AAGrB,iEAKiC;AAJ/B,2HAAA,kBAAkB,OAAA;AAClB,4HAAA,mBAAmB,OAAA;AACnB,gIAAA,uBAAuB,OAAA;AACvB,uHAAA,cAAc,OAAA","sourcesContent":["export type { TamboComponent, TamboTool } from \"../model/component-metadata\";\nexport {\n  TamboClientProvider,\n  useIsTamboTokenUpdating,\n  useTamboClient,\n} from \"./tambo-client-provider\";\nexport {\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nexport {\n  TamboContextHelpersProvider,\n  useTamboContextHelpers,\n  type TamboContextHelpersContextProps,\n  type TamboContextHelpersProviderProps,\n} from \"./tambo-context-helpers-provider\";\nexport {\n  Pending,\n  Streaming,\n  Success,\n  TamboPropStreamProvider,\n  useTamboStream,\n} from \"./tambo-prop-stream-provider\";\nexport type {\n  StreamStateComponentProps,\n  StreamStatus,\n  TamboPropStreamContextValue,\n} from \"./tambo-prop-stream-provider\";\nexport { TamboContext, TamboProvider, useTambo } from \"./tambo-provider\";\nexport {\n  TamboRegistryProvider,\n  useTamboRegistry,\n  type TamboRegistryContext,\n} from \"./tambo-registry-provider\";\nexport { TamboStubProvider, type TamboStubProviderProps } from \"./tambo-stubs\";\nexport {\n  TamboThreadInputProvider,\n  useTamboThreadInput,\n  type TamboThreadInputContextProps,\n} from \"./tambo-thread-input-provider\";\nexport {\n  TamboThreadContext,\n  TamboThreadProvider,\n  useTamboGenerationStage,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n"]}

package/dist/providers/tambo-client-provider.d.ts

@@ -28,6 +28,8 @@ export interface TamboClientContextProps {
     client: TamboAI;
     /** The tambo-specific query client */
     queryClient: QueryClient;
+    /** Whether the session token is currently being updated */
+    isUpdatingToken: boolean;
 }
 export declare const TamboClientContext: React.Context<TamboClientContextProps | undefined>;
 /**
@@ -55,4 +57,9 @@ export declare const useTamboClient: () => TamboAI;
  * @private
  */
 export declare const useTamboQueryClient: () => QueryClient;
+/**
+ * Hook to check if the session token is currently being updated
+ * @returns true if the token is being refreshed, false otherwise
+ */
+export declare const useIsTamboTokenUpdating: () => boolean;
 //# sourceMappingURL=tambo-client-provider.d.ts.map

package/dist/providers/tambo-client-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-client-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-client-provider.tsx"],"names":[],"mappings":"AACA,OAAO,OAA0B,MAAM,0BAA0B,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,KAAK,EAAE,EAAiB,iBAAiB,EAAY,MAAM,OAAO,CAAC;AAI1E,MAAM,WAAW,wBAAwB;IACvC;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;OAGG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,WAAW,CAAC,EAAE,YAAY,GAAG,SAAS,CAAC;IAEvC;;;;;OAKG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,uBAAuB;IACtC,yBAAyB;IACzB,MAAM,EAAE,OAAO,CAAC;IAChB,sCAAsC;IACtC,WAAW,EAAE,WAAW,CAAC;CAC1B;AAED,eAAO,MAAM,kBAAkB,oDAEnB,CAAC;AAEb;;;;;;;;;;GAUG;AACH,eAAO,MAAM,mBAAmB,EAAE,KAAK,CAAC,EAAE,CACxC,iBAAiB,CAAC,wBAAwB,CAAC,CAyB5C,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,cAAc,eAM1B,CAAC;AAEF;;;;;GAKG;AACH,eAAO,MAAM,mBAAmB,mBAQ/B,CAAC"}
+{"version":3,"file":"tambo-client-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-client-provider.tsx"],"names":[],"mappings":"AACA,OAAO,OAA0B,MAAM,0BAA0B,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,KAAK,EAAE,EAAiB,iBAAiB,EAAY,MAAM,OAAO,CAAC;AAI1E,MAAM,WAAW,wBAAwB;IACvC;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;OAGG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,WAAW,CAAC,EAAE,YAAY,GAAG,SAAS,CAAC;IAEvC;;;;;OAKG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,uBAAuB;IACtC,yBAAyB;IACzB,MAAM,EAAE,OAAO,CAAC;IAChB,sCAAsC;IACtC,WAAW,EAAE,WAAW,CAAC;IACzB,2DAA2D;IAC3D,eAAe,EAAE,OAAO,CAAC;CAC1B;AAED,eAAO,MAAM,kBAAkB,oDAEnB,CAAC;AAEb;;;;;;;;;;GAUG;AACH,eAAO,MAAM,mBAAmB,EAAE,KAAK,CAAC,EAAE,CACxC,iBAAiB,CAAC,wBAAwB,CAAC,CAmC5C,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,cAAc,eAM1B,CAAC;AAEF;;;;;GAKG;AACH,eAAO,MAAM,mBAAmB,mBAQ/B,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,uBAAuB,eAQnC,CAAC"}