@tambo-ai/react 0.44.1 → 0.46.0

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

@@ -1,3 +1,4 @@
+import { QueryClient } from "@tanstack/react-query";
 import { act, renderHook } from "@testing-library/react";
 import { useTamboSessionToken } from "../use-tambo-session-token";
 describe("useTamboSessionToken", () => {
@@ -17,40 +18,59 @@ describe("useTamboSessionToken", () => {
         beta: mockBeta,
         bearer: "",
     };
+    const queryClient = new QueryClient({
+        defaultOptions: {
+            queries: {
+                retry: false,
+                refetchOnWindowFocus: false,
+            },
+        },
+    });
     beforeEach(() => {
         jest.clearAllMocks();
         jest.clearAllTimers();
         jest.useFakeTimers();
         mockTamboAI.bearer = "";
+        queryClient.clear();
     });
-    afterEach(() => {
-        jest.runOnlyPendingTimers();
+    afterEach(async () => {
+        await act(async () => {
+            await jest.runOnlyPendingTimersAsync();
+        });
         jest.useRealTimers();
+        jest.resetAllMocks();
     });
     it("should return null initially when no userToken is provided", () => {
-        const { result } = renderHook(() => useTamboSessionToken(mockTamboAI, undefined));
-        expect(result.current).toBeNull();
+        const { result } = renderHook(() => 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 } = renderHook(() => useTamboSessionToken(mockTamboAI, "user-token"));
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        const { result } = renderHook(() => useTamboSessionToken(mockTamboAI, queryClient, "user-token"));
+        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);
         await 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);
-        renderHook(() => useTamboSessionToken(mockTamboAI, "user-token"));
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        renderHook(() => useTamboSessionToken(mockTamboAI, queryClient, "user-token"));
         await 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");
@@ -58,12 +78,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 } = renderHook(() => useTamboSessionToken(mockTamboAI, "user-token"));
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        const { result } = renderHook(() => useTamboSessionToken(mockTamboAI, queryClient, "user-token"));
         await 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 () => {
@@ -72,24 +94,35 @@ describe("useTamboSessionToken", () => {
             expires_in: 7200, // 2 hours
             token_type: "Bearer",
         };
-        jest.mocked(mockAuthApi.getToken).mockResolvedValue(customTokenResponse);
-        const { result } = renderHook(() => useTamboSessionToken(mockTamboAI, "user-token"));
+        mockAuthApi.getToken.mockResolvedValue(customTokenResponse);
+        const { result } = renderHook(() => useTamboSessionToken(mockTamboAI, queryClient, "user-token"));
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: true,
+        });
         await 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 } = renderHook(({ userToken }) => useTamboSessionToken(mockTamboAI, userToken), {
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        const { result, rerender } = renderHook(({ userToken }) => useTamboSessionToken(mockTamboAI, queryClient, userToken), {
             initialProps: { userToken: "user-token" },
         });
+        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);
         await 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();
         act(() => {
@@ -98,17 +131,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 } = renderHook(({ userToken }) => useTamboSessionToken(mockTamboAI, userToken), {
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        const { result, rerender } = renderHook(({ userToken }) => useTamboSessionToken(mockTamboAI, queryClient, userToken), {
             initialProps: { userToken: "user-token-1" },
         });
+        expect(mockAuthApi.getToken).toHaveBeenCalledTimes(1);
         await 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",
         });
@@ -119,33 +156,46 @@ describe("useTamboSessionToken", () => {
         await 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 } = renderHook(({ userToken }) => useTamboSessionToken(mockTamboAI, userToken), {
+        mockAuthApi.getToken.mockResolvedValue(mockTokenResponse);
+        const { result, rerender } = renderHook(({ userToken }) => useTamboSessionToken(mockTamboAI, queryClient, userToken), {
             initialProps: { userToken: "user-token" },
         });
         await act(async () => {
             await jest.runOnlyPendingTimersAsync();
         });
-        expect(result.current).toBe("test-access-token");
+        expect(result.current).toMatchObject({
+            data: mockTokenResponse,
+            isFetching: false,
+        });
         // Change userToken to undefined
-        act(() => {
-            rerender({ userToken: undefined });
+        rerender({ userToken: undefined });
+        await 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 } = renderHook(() => useTamboSessionToken(mockTamboAI, "user-token"));
-        expect(result.current).toBeNull();
+        mockAuthApi.getToken.mockReturnValue(promise);
+        const { result, unmount } = renderHook(() => useTamboSessionToken(mockTamboAI, queryClient, "user-token"));
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: true,
+        });
         // Unmount before the promise resolves
         unmount();
         // Now resolve the promise
@@ -153,7 +203,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 } = renderHook(() => 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 } = renderHook(() => useTamboSessionToken(mockTamboAI, queryClient, "user-token"));
+        // Should be updating initially
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: true,
+        });
+        await 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 } = renderHook(() => useTamboSessionToken(mockTamboAI, queryClient, "user-token"));
+        // Should be updating initially
+        expect(result.current).toMatchObject({
+            data: undefined,
+            isFetching: true,
+        });
+        await 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/esm/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,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAEzD,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;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,UAAU,CAAC,GAAG,EAAE,CACjC,oBAAoB,CAAC,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,UAAU,CAAC,GAAG,EAAE,CACjC,oBAAoB,CAAC,WAAW,EAAE,YAAY,CAAC,CAChD,CAAC;QAEF,MAAM,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CAAC,oBAAoB,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;QAElE,MAAM,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CACjC,oBAAoB,CAAC,WAAW,EAAE,YAAY,CAAC,CAChD,CAAC;QAEF,MAAM,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CACjC,oBAAoB,CAAC,WAAW,EAAE,YAAY,CAAC,CAChD,CAAC;QAEF,MAAM,GAAG,CAAC,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,UAAU,CACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,oBAAoB,CAAC,WAAW,EAAE,SAAS,CAAC,EAC/D;YACE,YAAY,EAAE,EAAE,SAAS,EAAE,YAAkC,EAAE;SAChE,CACF,CAAC;QAEF,MAAM,GAAG,CAAC,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,GAAG,CAAC,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,UAAU,CACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,oBAAoB,CAAC,WAAW,EAAE,SAAS,CAAC,EAC/D;YACE,YAAY,EAAE,EAAE,SAAS,EAAE,cAAc,EAAE;SAC5C,CACF,CAAC;QAEF,MAAM,GAAG,CAAC,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,GAAG,CAAC,GAAG,EAAE;YACP,QAAQ,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,MAAM,GAAG,CAAC,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,UAAU,CACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,oBAAoB,CAAC,WAAW,EAAE,SAAS,CAAC,EAC/D;YACE,YAAY,EAAE,EAAE,SAAS,EAAE,YAAkC,EAAE;SAChE,CACF,CAAC;QAEF,MAAM,GAAG,CAAC,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,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CAC1C,oBAAoB,CAAC,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,GAAG,CAAC,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,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAEzD,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;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,WAAW,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,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CACjC,oBAAoB,CAAC,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,UAAU,CAAC,GAAG,EAAE,CACjC,oBAAoB,CAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEtD,MAAM,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CACd,oBAAoB,CAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;QAEF,MAAM,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CACjC,oBAAoB,CAAC,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAC7D,CAAC;QAEF,MAAM,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CACjC,oBAAoB,CAAC,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,GAAG,CAAC,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,UAAU,CACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAChB,oBAAoB,CAAC,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,GAAG,CAAC,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,GAAG,CAAC,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,UAAU,CACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAChB,oBAAoB,CAAC,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,GAAG,CAAC,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,GAAG,CAAC,GAAG,EAAE;YACP,QAAQ,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,MAAM,GAAG,CAAC,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,UAAU,CACrC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAChB,oBAAoB,CAAC,WAAW,EAAE,WAAW,EAAE,SAAS,CAAC,EAC3D;YACE,YAAY,EAAE,EAAE,SAAS,EAAE,YAAkC,EAAE;SAChE,CACF,CAAC;QAEF,MAAM,GAAG,CAAC,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,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CAC1C,oBAAoB,CAAC,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,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CACjC,oBAAoB,CAAC,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,UAAU,CAAC,GAAG,EAAE,CACjC,oBAAoB,CAAC,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,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CACjC,oBAAoB,CAAC,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,GAAG,CAAC,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/esm/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/esm/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/esm/providers/hooks/use-tambo-session-token.js

@@ -1,5 +1,6 @@
 "use client";
-import { useEffect, useState } from "react";
+import { useQuery } from "@tanstack/react-query";
+import { useEffect } from "react";
 /**
  * This internal hook is used to get the Tambo session token and keep it
  * refreshed.
@@ -9,54 +10,36 @@ import { useEffect, useState } from "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.
  */
-export function useTamboSessionToken(client, userToken) {
-    const [tamboSessionToken, setTamboSessionToken] = 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] = useState(true);
-    useEffect(() => {
-        let expireTimer = null;
-        async function updateToken(subjectToken, abortController) {
-            if (abortController.signal.aborted || !userToken) {
-                return;
-            }
+export function useTamboSessionToken(client, queryClient, userToken) {
+    const result = 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;
+    useEffect(() => {
+        client.bearer = accessToken;
+    }, [accessToken, client]);
+    return result;
 }
 //# sourceMappingURL=use-tambo-session-token.js.map

package/esm/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;AAEb,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAE5C;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,oBAAoB,CAClC,MAAe,EACf,SAA6B;IAE7B,MAAM,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,GAAG,QAAQ,CACxD,IAAI,CACL,CAAC;IACF,6EAA6E;IAC7E,yDAAyD;IACzD,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IACjD,SAAS,CAAC,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;AAEb,OAAO,EAAe,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AAElC;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,oBAAoB,CAClC,MAAe,EACf,WAAwB,EACxB,SAA6B;IAE7B,MAAM,MAAM,GAAG,QAAQ,CACrB;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,SAAS,CAAC,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/esm/providers/index.d.ts

@@ -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/esm/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,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,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/esm/providers/index.js

@@ -5,5 +5,6 @@ export { Pending, Streaming, Success, TamboPropStreamProvider, useTamboStream, }
 export { TamboContext, TamboProvider, useTambo } from "./tambo-provider";
 export { TamboRegistryProvider, useTamboRegistry, } from "./tambo-registry-provider";
 export { TamboStubProvider } from "./tambo-stubs";
+export { TamboThreadInputProvider, useTamboThreadInput, } from "./tambo-thread-input-provider";
 export { TamboThreadContext, TamboThreadProvider, useTamboGenerationStage, useTamboThread, } from "./tambo-thread-provider";
 //# sourceMappingURL=index.js.map

package/esm/providers/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AACA,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,GAGvB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,OAAO,EACP,SAAS,EACT,OAAO,EACP,uBAAuB,EACvB,cAAc,GACf,MAAM,8BAA8B,CAAC;AAMtC,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AACzE,OAAO,EACL,qBAAqB,EACrB,gBAAgB,GAEjB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,iBAAiB,EAA+B,MAAM,eAAe,CAAC;AAC/E,OAAO,EACL,kBAAkB,EAClB,mBAAmB,EACnB,uBAAuB,EACvB,cAAc,GACf,MAAM,yBAAyB,CAAC","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,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,GAGvB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,OAAO,EACP,SAAS,EACT,OAAO,EACP,uBAAuB,EACvB,cAAc,GACf,MAAM,8BAA8B,CAAC;AAMtC,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AACzE,OAAO,EACL,qBAAqB,EACrB,gBAAgB,GAEjB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,iBAAiB,EAA+B,MAAM,eAAe,CAAC;AAC/E,OAAO,EACL,wBAAwB,EACxB,mBAAmB,GAEpB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,kBAAkB,EAClB,mBAAmB,EACnB,uBAAuB,EACvB,cAAc,GACf,MAAM,yBAAyB,CAAC","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  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/esm/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/esm/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"}

package/esm/providers/tambo-client-provider.js

@@ -31,9 +31,13 @@ export const TamboClientProvider = ({ children, tamboUrl, apiKey, environment, u
     }
     const [client] = useState(() => new TamboAI(tamboConfig));
     const [queryClient] = useState(() => new QueryClient());
-    // Keep the session token updated
-    useTamboSessionToken(client, userToken);
-    return (React.createElement(TamboClientContext.Provider, { value: { client, queryClient } }, children));
+    // Keep the session token updated and get the updating state
+    const { isFetching: isUpdatingToken } = useTamboSessionToken(client, queryClient, userToken);
+    return (React.createElement(TamboClientContext.Provider, { value: {
+            client,
+            queryClient,
+            isUpdatingToken,
+        } }, children));
 };
 /**
  * The useTamboClient hook provides access to the TamboAI client
@@ -60,4 +64,15 @@ export const useTamboQueryClient = () => {
     }
     return context.queryClient;
 };
+/**
+ * Hook to check if the session token is currently being updated
+ * @returns true if the token is being refreshed, false otherwise
+ */
+export const useIsTamboTokenUpdating = () => {
+    const context = React.useContext(TamboClientContext);
+    if (context === undefined) {
+        throw new Error("useIsTamboTokenUpdating must be used within a TamboClientProvider");
+    }
+    return context.isUpdatingToken;
+};
 //# sourceMappingURL=tambo-client-provider.js.map