@tambo-ai/react 0.45.0 → 0.46.1

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

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

@@ -1,9 +1,10 @@
-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, } from "./tambo-context-helpers-provider";
 export { Pending, Streaming, Success, TamboPropStreamProvider, useTamboStream, } from "./tambo-prop-stream-provider";
 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,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,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 {\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/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

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

@@ -1 +1 @@
-{"version":3,"file":"tambo-client-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-client-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,OAA0B,MAAM,0BAA0B,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,KAAK,EAAE,EAAE,aAAa,EAAqB,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC1E,OAAO,WAAW,MAAM,oBAAoB,CAAC;AAC7C,OAAO,EAAE,oBAAoB,EAAE,MAAM,iCAAiC,CAAC;AAiCvE,MAAM,CAAC,MAAM,kBAAkB,GAAG,aAAa,CAE7C,SAAS,CAAC,CAAC;AAEb;;;;;;;;;;GAUG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAE5B,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,EAAE,EAAE;IAC7D,MAAM,WAAW,GAAkB;QACjC,MAAM;QACN,cAAc,EAAE;YACd,uBAAuB,EAAE,WAAW,CAAC,OAAO;SAC7C;KACF,CAAC;IACF,IAAI,QAAQ,EAAE,CAAC;QACb,WAAW,CAAC,OAAO,GAAG,QAAQ,CAAC;IACjC,CAAC;IACD,IAAI,WAAW,EAAE,CAAC;QAChB,WAAW,CAAC,WAAW,GAAG,WAAW,CAAC;IACxC,CAAC;IACD,MAAM,CAAC,MAAM,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,IAAI,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;IAC1D,MAAM,CAAC,WAAW,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC,CAAC;IAExD,iCAAiC;IACjC,oBAAoB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAExC,OAAO,CACL,oBAAC,kBAAkB,CAAC,QAAQ,IAAC,KAAK,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,IACxD,QAAQ,CACmB,CAC/B,CAAC;AACJ,CAAC,CAAC;AAEF;;;;GAIG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,GAAG,EAAE;IACjC,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAC9E,CAAC;IACD,OAAO,OAAO,CAAC,MAAM,CAAC;AACxB,CAAC,CAAC;AAEF;;;;;GAKG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,GAAG,EAAE;IACtC,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CACb,+DAA+D,CAChE,CAAC;IACJ,CAAC;IACD,OAAO,OAAO,CAAC,WAAW,CAAC;AAC7B,CAAC,CAAC","sourcesContent":["\"use client\";\nimport TamboAI, { ClientOptions } from \"@tambo-ai/typescript-sdk\";\nimport { QueryClient } from \"@tanstack/react-query\";\nimport React, { createContext, PropsWithChildren, useState } from \"react\";\nimport packageJson from \"../../package.json\";\nimport { useTamboSessionToken } from \"./hooks/use-tambo-session-token\";\n\nexport interface TamboClientProviderProps {\n  /**\n   * The URL of the Tambo API (only used for local development and debugging)\n   */\n  tamboUrl?: string;\n  /**\n   * The API key for the Tambo API. This typically comes from a variable like\n   * `process.env.NEXT_PUBLIC_TAMBO_API_KEY`\n   */\n  apiKey: string;\n  /**\n   * The environment to use for the Tambo API\n   */\n  environment?: \"production\" | \"staging\";\n\n  /**\n   * The user token to use to identify the user in the Tambo API. This token is\n   * a 3rd party token like a Google or GitHub access token, exchanged with the\n   * Tambo API to get a session token. This is used to securely identify the\n   * user when calling the Tambo API.\n   */\n  userToken?: string;\n}\n\nexport interface TamboClientContextProps {\n  /** The TamboAI client */\n  client: TamboAI;\n  /** The tambo-specific query client */\n  queryClient: QueryClient;\n}\n\nexport const TamboClientContext = createContext<\n  TamboClientContextProps | undefined\n>(undefined);\n\n/**\n * The TamboClientProvider is a React provider that provides a TamboAI client\n * and a query client to the descendants of the provider.\n * @param props - The props for the TamboClientProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.environment - The environment to use for the Tambo API\n * @param props.userToken - The oauth access token to use to identify the user in the Tambo API\n * @returns The TamboClientProvider component\n */\nexport const TamboClientProvider: React.FC<\n  PropsWithChildren<TamboClientProviderProps>\n> = ({ children, tamboUrl, apiKey, environment, userToken }) => {\n  const tamboConfig: ClientOptions = {\n    apiKey,\n    defaultHeaders: {\n      \"X-Tambo-React-Version\": packageJson.version,\n    },\n  };\n  if (tamboUrl) {\n    tamboConfig.baseURL = tamboUrl;\n  }\n  if (environment) {\n    tamboConfig.environment = environment;\n  }\n  const [client] = useState(() => new TamboAI(tamboConfig));\n  const [queryClient] = useState(() => new QueryClient());\n\n  // Keep the session token updated\n  useTamboSessionToken(client, userToken);\n\n  return (\n    <TamboClientContext.Provider value={{ client, queryClient }}>\n      {children}\n    </TamboClientContext.Provider>\n  );\n};\n\n/**\n * The useTamboClient hook provides access to the TamboAI client\n * to the descendants of the TamboClientProvider.\n * @returns The TamboAI client\n */\nexport const useTamboClient = () => {\n  const context = React.useContext(TamboClientContext);\n  if (context === undefined) {\n    throw new Error(\"useTamboClient must be used within a TamboClientProvider\");\n  }\n  return context.client;\n};\n\n/**\n * The useTamboQueryClient hook provides access to the tambo-specific query client\n * to the descendants of the TamboClientProvider.\n * @returns The tambo-specific query client\n * @private\n */\nexport const useTamboQueryClient = () => {\n  const context = React.useContext(TamboClientContext);\n  if (context === undefined) {\n    throw new Error(\n      \"useTamboQueryClient must be used within a TamboClientProvider\",\n    );\n  }\n  return context.queryClient;\n};\n"]}
+{"version":3,"file":"tambo-client-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-client-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,OAA0B,MAAM,0BAA0B,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,KAAK,EAAE,EAAE,aAAa,EAAqB,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC1E,OAAO,WAAW,MAAM,oBAAoB,CAAC;AAC7C,OAAO,EAAE,oBAAoB,EAAE,MAAM,iCAAiC,CAAC;AAmCvE,MAAM,CAAC,MAAM,kBAAkB,GAAG,aAAa,CAE7C,SAAS,CAAC,CAAC;AAEb;;;;;;;;;;GAUG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAE5B,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,EAAE,EAAE;IAC7D,MAAM,WAAW,GAAkB;QACjC,MAAM;QACN,cAAc,EAAE;YACd,uBAAuB,EAAE,WAAW,CAAC,OAAO;SAC7C;KACF,CAAC;IACF,IAAI,QAAQ,EAAE,CAAC;QACb,WAAW,CAAC,OAAO,GAAG,QAAQ,CAAC;IACjC,CAAC;IACD,IAAI,WAAW,EAAE,CAAC;QAChB,WAAW,CAAC,WAAW,GAAG,WAAW,CAAC;IACxC,CAAC;IACD,MAAM,CAAC,MAAM,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,IAAI,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;IAC1D,MAAM,CAAC,WAAW,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC,CAAC;IAExD,4DAA4D;IAC5D,MAAM,EAAE,UAAU,EAAE,eAAe,EAAE,GAAG,oBAAoB,CAC1D,MAAM,EACN,WAAW,EACX,SAAS,CACV,CAAC;IAEF,OAAO,CACL,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;YACL,MAAM;YACN,WAAW;YACX,eAAe;SAChB,IAEA,QAAQ,CACmB,CAC/B,CAAC;AACJ,CAAC,CAAC;AAEF;;;;GAIG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,GAAG,EAAE;IACjC,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAC9E,CAAC;IACD,OAAO,OAAO,CAAC,MAAM,CAAC;AACxB,CAAC,CAAC;AAEF;;;;;GAKG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,GAAG,EAAE;IACtC,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CACb,+DAA+D,CAChE,CAAC;IACJ,CAAC;IACD,OAAO,OAAO,CAAC,WAAW,CAAC;AAC7B,CAAC,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,GAAG,EAAE;IAC1C,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;IACrD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CACb,mEAAmE,CACpE,CAAC;IACJ,CAAC;IACD,OAAO,OAAO,CAAC,eAAe,CAAC;AACjC,CAAC,CAAC","sourcesContent":["\"use client\";\nimport TamboAI, { ClientOptions } from \"@tambo-ai/typescript-sdk\";\nimport { QueryClient } from \"@tanstack/react-query\";\nimport React, { createContext, PropsWithChildren, useState } from \"react\";\nimport packageJson from \"../../package.json\";\nimport { useTamboSessionToken } from \"./hooks/use-tambo-session-token\";\n\nexport interface TamboClientProviderProps {\n  /**\n   * The URL of the Tambo API (only used for local development and debugging)\n   */\n  tamboUrl?: string;\n  /**\n   * The API key for the Tambo API. This typically comes from a variable like\n   * `process.env.NEXT_PUBLIC_TAMBO_API_KEY`\n   */\n  apiKey: string;\n  /**\n   * The environment to use for the Tambo API\n   */\n  environment?: \"production\" | \"staging\";\n\n  /**\n   * The user token to use to identify the user in the Tambo API. This token is\n   * a 3rd party token like a Google or GitHub access token, exchanged with the\n   * Tambo API to get a session token. This is used to securely identify the\n   * user when calling the Tambo API.\n   */\n  userToken?: string;\n}\n\nexport interface TamboClientContextProps {\n  /** The TamboAI client */\n  client: TamboAI;\n  /** The tambo-specific query client */\n  queryClient: QueryClient;\n  /** Whether the session token is currently being updated */\n  isUpdatingToken: boolean;\n}\n\nexport const TamboClientContext = createContext<\n  TamboClientContextProps | undefined\n>(undefined);\n\n/**\n * The TamboClientProvider is a React provider that provides a TamboAI client\n * and a query client to the descendants of the provider.\n * @param props - The props for the TamboClientProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.environment - The environment to use for the Tambo API\n * @param props.userToken - The oauth access token to use to identify the user in the Tambo API\n * @returns The TamboClientProvider component\n */\nexport const TamboClientProvider: React.FC<\n  PropsWithChildren<TamboClientProviderProps>\n> = ({ children, tamboUrl, apiKey, environment, userToken }) => {\n  const tamboConfig: ClientOptions = {\n    apiKey,\n    defaultHeaders: {\n      \"X-Tambo-React-Version\": packageJson.version,\n    },\n  };\n  if (tamboUrl) {\n    tamboConfig.baseURL = tamboUrl;\n  }\n  if (environment) {\n    tamboConfig.environment = environment;\n  }\n  const [client] = useState(() => new TamboAI(tamboConfig));\n  const [queryClient] = useState(() => new QueryClient());\n\n  // Keep the session token updated and get the updating state\n  const { isFetching: isUpdatingToken } = useTamboSessionToken(\n    client,\n    queryClient,\n    userToken,\n  );\n\n  return (\n    <TamboClientContext.Provider\n      value={{\n        client,\n        queryClient,\n        isUpdatingToken,\n      }}\n    >\n      {children}\n    </TamboClientContext.Provider>\n  );\n};\n\n/**\n * The useTamboClient hook provides access to the TamboAI client\n * to the descendants of the TamboClientProvider.\n * @returns The TamboAI client\n */\nexport const useTamboClient = () => {\n  const context = React.useContext(TamboClientContext);\n  if (context === undefined) {\n    throw new Error(\"useTamboClient must be used within a TamboClientProvider\");\n  }\n  return context.client;\n};\n\n/**\n * The useTamboQueryClient hook provides access to the tambo-specific query client\n * to the descendants of the TamboClientProvider.\n * @returns The tambo-specific query client\n * @private\n */\nexport const useTamboQueryClient = () => {\n  const context = React.useContext(TamboClientContext);\n  if (context === undefined) {\n    throw new Error(\n      \"useTamboQueryClient must be used within a TamboClientProvider\",\n    );\n  }\n  return context.queryClient;\n};\n\n/**\n * Hook to check if the session token is currently being updated\n * @returns true if the token is being refreshed, false otherwise\n */\nexport const useIsTamboTokenUpdating = () => {\n  const context = React.useContext(TamboClientContext);\n  if (context === undefined) {\n    throw new Error(\n      \"useIsTamboTokenUpdating must be used within a TamboClientProvider\",\n    );\n  }\n  return context.isUpdatingToken;\n};\n"]}

package/esm/providers/tambo-provider.d.ts

@@ -4,6 +4,7 @@ import { TamboClientContextProps, TamboClientProviderProps } from "./tambo-clien
 import { TamboComponentContextProps } from "./tambo-component-provider";
 import { TamboContextHelpersContextProps, TamboContextHelpersProviderProps } from "./tambo-context-helpers-provider";
 import { TamboRegistryProviderProps } from "./tambo-registry-provider";
+import { TamboThreadInputProviderProps } from "./tambo-thread-input-provider";
 import { TamboGenerationStageContextProps, TamboThreadContextProps, TamboThreadProviderProps } from "./tambo-thread-provider";
 /**
  * The TamboProvider gives full access to the whole Tambo API. This includes the
@@ -17,9 +18,11 @@ import { TamboGenerationStageContextProps, TamboThreadContextProps, TamboThreadP
  * @param props.tools - The tools to register
  * @param props.streaming - Whether to stream the response by default. Defaults to true.
  * @param props.contextHelpers - Configuration for which context helpers are enabled/disabled
+ * @param props.userToken - The JWT id token to use to identify the user in the Tambo API. (preferred over contextKey)
+ * @param props.contextKey - Optional context key to be used in the thread input provider
  * @returns The TamboProvider component
  */
-export declare const TamboProvider: React.FC<PropsWithChildren<TamboClientProviderProps & TamboRegistryProviderProps & TamboThreadProviderProps & TamboContextHelpersProviderProps>>;
+export declare const TamboProvider: React.FC<PropsWithChildren<TamboClientProviderProps & TamboRegistryProviderProps & TamboThreadProviderProps & TamboContextHelpersProviderProps & TamboThreadInputProviderProps>>;
 export type TamboContextProps = TamboClientContextProps & TamboThreadContextProps & TamboGenerationStageContextProps & TamboComponentContextProps & TamboInteractableContext & TamboContextHelpersContextProps;
 export declare const TamboContext: React.Context<TamboContextProps>;
 /**

package/esm/providers/tambo-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAA6B,MAAM,OAAO,CAAC;AAC5E,OAAO,EAAE,wBAAwB,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EACL,uBAAuB,EAEvB,wBAAwB,EAGzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,0BAA0B,EAG3B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,+BAA+B,EAE/B,gCAAgC,EAEjC,MAAM,kCAAkC,CAAC;AAK1C,OAAO,EAEL,0BAA0B,EAC3B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,gCAAgC,EAChC,uBAAuB,EAEvB,wBAAwB,EAEzB,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;;;;GAaG;AACH,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAClC,iBAAiB,CACf,wBAAwB,GACtB,0BAA0B,GAC1B,wBAAwB,GACxB,gCAAgC,CACnC,CAqCF,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG,uBAAuB,GACrD,uBAAuB,GACvB,gCAAgC,GAChC,0BAA0B,GAC1B,wBAAwB,GACxB,+BAA+B,CAAC;AAElC,eAAO,MAAM,YAAY,kCAExB,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,sBAAsB,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAwB9D,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,QAAQ,yBAEpB,CAAC"}
+{"version":3,"file":"tambo-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAA6B,MAAM,OAAO,CAAC;AAC5E,OAAO,EAAE,wBAAwB,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EACL,uBAAuB,EAEvB,wBAAwB,EAIzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,0BAA0B,EAG3B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,+BAA+B,EAE/B,gCAAgC,EAEjC,MAAM,kCAAkC,CAAC;AAK1C,OAAO,EAEL,0BAA0B,EAC3B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAEL,6BAA6B,EAC9B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,gCAAgC,EAChC,uBAAuB,EAEvB,wBAAwB,EAEzB,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;;;;;;GAeG;AACH,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAClC,iBAAiB,CACf,wBAAwB,GACtB,0BAA0B,GAC1B,wBAAwB,GACxB,gCAAgC,GAChC,6BAA6B,CAChC,CAwCF,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG,uBAAuB,GACrD,uBAAuB,GACvB,gCAAgC,GAChC,0BAA0B,GAC1B,wBAAwB,GACxB,+BAA+B,CAAC;AAElC,eAAO,MAAM,YAAY,kCAExB,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,sBAAsB,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CA0B9D,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,QAAQ,yBAEpB,CAAC"}

package/esm/providers/tambo-provider.js

@@ -1,10 +1,11 @@
 "use client";
 import React, { createContext, useContext } from "react";
-import { TamboClientProvider, useTamboClient, useTamboQueryClient, } from "./tambo-client-provider";
+import { TamboClientProvider, useIsTamboTokenUpdating, useTamboClient, useTamboQueryClient, } from "./tambo-client-provider";
 import { TamboComponentProvider, useTamboComponent, } from "./tambo-component-provider";
 import { TamboContextHelpersProvider, useTamboContextHelpers, } from "./tambo-context-helpers-provider";
 import { TamboInteractableProvider, useTamboInteractable, } from "./tambo-interactable-provider";
 import { TamboRegistryProvider, } from "./tambo-registry-provider";
+import { TamboThreadInputProvider, } from "./tambo-thread-input-provider";
 import { TamboThreadProvider, useTamboThread, } from "./tambo-thread-provider";
 /**
  * The TamboProvider gives full access to the whole Tambo API. This includes the
@@ -18,9 +19,11 @@ import { TamboThreadProvider, useTamboThread, } from "./tambo-thread-provider";
  * @param props.tools - The tools to register
  * @param props.streaming - Whether to stream the response by default. Defaults to true.
  * @param props.contextHelpers - Configuration for which context helpers are enabled/disabled
+ * @param props.userToken - The JWT id token to use to identify the user in the Tambo API. (preferred over contextKey)
+ * @param props.contextKey - Optional context key to be used in the thread input provider
  * @returns The TamboProvider component
  */
-export const TamboProvider = ({ children, tamboUrl, apiKey, userToken, components, environment, tools, streaming, contextHelpers, }) => {
+export const TamboProvider = ({ children, tamboUrl, apiKey, userToken, components, environment, tools, streaming, contextHelpers, contextKey, }) => {
     // Should only be used in browser
     if (typeof window === "undefined") {
         console.error("TamboProvider must be used within a browser");
@@ -29,9 +32,10 @@ export const TamboProvider = ({ children, tamboUrl, apiKey, userToken, component
         React.createElement(TamboRegistryProvider, { components: components, tools: tools },
             React.createElement(TamboContextHelpersProvider, { contextHelpers: contextHelpers },
                 React.createElement(TamboThreadProvider, { streaming: streaming },
-                    React.createElement(TamboComponentProvider, null,
-                        React.createElement(TamboInteractableProvider, null,
-                            React.createElement(TamboCompositeProvider, null, children))))))));
+                    React.createElement(TamboThreadInputProvider, { contextKey: contextKey },
+                        React.createElement(TamboComponentProvider, null,
+                            React.createElement(TamboInteractableProvider, null,
+                                React.createElement(TamboCompositeProvider, null, children)))))))));
 };
 export const TamboContext = createContext({});
 /**
@@ -45,12 +49,14 @@ export const TamboCompositeProvider = ({ children, }) => {
     const threads = useTamboThread();
     const client = useTamboClient();
     const queryClient = useTamboQueryClient();
+    const isUpdatingToken = useIsTamboTokenUpdating();
     const componentRegistry = useTamboComponent();
     const interactableComponents = useTamboInteractable();
     const contextHelpers = useTamboContextHelpers();
     return (React.createElement(TamboContext.Provider, { value: {
             client,
             queryClient,
+            isUpdatingToken,
             ...componentRegistry,
             ...threads,
             ...interactableComponents,

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

@@ -1 +1 @@
-{"version":3,"file":"tambo-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,KAAK,EAAE,EAAqB,aAAa,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC;AAE5E,OAAO,EAEL,mBAAmB,EAEnB,cAAc,EACd,mBAAmB,GACpB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAEL,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAEL,2BAA2B,EAE3B,sBAAsB,GACvB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,yBAAyB,EACzB,oBAAoB,GACrB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,qBAAqB,GAEtB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAGL,mBAAmB,EAEnB,cAAc,GACf,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;;;;GAaG;AACH,MAAM,CAAC,MAAM,aAAa,GAOtB,CAAC,EACH,QAAQ,EACR,QAAQ,EACR,MAAM,EACN,SAAS,EACT,UAAU,EACV,WAAW,EACX,KAAK,EACL,SAAS,EACT,cAAc,GACf,EAAE,EAAE;IACH,iCAAiC;IACjC,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,OAAO,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,CACL,oBAAC,mBAAmB,IAClB,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW,EACxB,SAAS,EAAE,SAAS;QAEpB,oBAAC,qBAAqB,IAAC,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK;YACzD,oBAAC,2BAA2B,IAAC,cAAc,EAAE,cAAc;gBACzD,oBAAC,mBAAmB,IAAC,SAAS,EAAE,SAAS;oBACvC,oBAAC,sBAAsB;wBACrB,oBAAC,yBAAyB;4BACxB,oBAAC,sBAAsB,QAAE,QAAQ,CAA0B,CACjC,CACL,CACL,CACM,CACR,CACJ,CACvB,CAAC;AACJ,CAAC,CAAC;AASF,MAAM,CAAC,MAAM,YAAY,GAAG,aAAa,CACvC,EAAuB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAgC,CAAC,EAClE,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;IACjC,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;IAChC,MAAM,WAAW,GAAG,mBAAmB,EAAE,CAAC;IAC1C,MAAM,iBAAiB,GAAG,iBAAiB,EAAE,CAAC;IAC9C,MAAM,sBAAsB,GAAG,oBAAoB,EAAE,CAAC;IACtD,MAAM,cAAc,GAAG,sBAAsB,EAAE,CAAC;IAEhD,OAAO,CACL,oBAAC,YAAY,CAAC,QAAQ,IACpB,KAAK,EAAE;YACL,MAAM;YACN,WAAW;YACX,GAAG,iBAAiB;YACpB,GAAG,OAAO;YACV,GAAG,sBAAsB;YACzB,GAAG,cAAc;SAClB,IAEA,QAAQ,CACa,CACzB,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,GAAG,EAAE;IAC3B,OAAO,UAAU,CAAC,YAAY,CAAC,CAAC;AAClC,CAAC,CAAC","sourcesContent":["\"use client\";\nimport React, { PropsWithChildren, createContext, useContext } from \"react\";\nimport { TamboInteractableContext } from \"../model/tambo-interactable\";\nimport {\n  TamboClientContextProps,\n  TamboClientProvider,\n  TamboClientProviderProps,\n  useTamboClient,\n  useTamboQueryClient,\n} from \"./tambo-client-provider\";\nimport {\n  TamboComponentContextProps,\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nimport {\n  TamboContextHelpersContextProps,\n  TamboContextHelpersProvider,\n  TamboContextHelpersProviderProps,\n  useTamboContextHelpers,\n} from \"./tambo-context-helpers-provider\";\nimport {\n  TamboInteractableProvider,\n  useTamboInteractable,\n} from \"./tambo-interactable-provider\";\nimport {\n  TamboRegistryProvider,\n  TamboRegistryProviderProps,\n} from \"./tambo-registry-provider\";\nimport {\n  TamboGenerationStageContextProps,\n  TamboThreadContextProps,\n  TamboThreadProvider,\n  TamboThreadProviderProps,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n\n/**\n * The TamboProvider gives full access to the whole Tambo API. This includes the\n * TamboAI client, the component registry, the current thread context, and interactable components.\n * @param props - The props for the TamboProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.components - The components to register\n * @param props.environment - The environment to use for the Tambo API\n * @param props.tools - The tools to register\n * @param props.streaming - Whether to stream the response by default. Defaults to true.\n * @param props.contextHelpers - Configuration for which context helpers are enabled/disabled\n * @returns The TamboProvider component\n */\nexport const TamboProvider: React.FC<\n  PropsWithChildren<\n    TamboClientProviderProps &\n      TamboRegistryProviderProps &\n      TamboThreadProviderProps &\n      TamboContextHelpersProviderProps\n  >\n> = ({\n  children,\n  tamboUrl,\n  apiKey,\n  userToken,\n  components,\n  environment,\n  tools,\n  streaming,\n  contextHelpers,\n}) => {\n  // Should only be used in browser\n  if (typeof window === \"undefined\") {\n    console.error(\"TamboProvider must be used within a browser\");\n  }\n\n  return (\n    <TamboClientProvider\n      tamboUrl={tamboUrl}\n      apiKey={apiKey}\n      environment={environment}\n      userToken={userToken}\n    >\n      <TamboRegistryProvider components={components} tools={tools}>\n        <TamboContextHelpersProvider contextHelpers={contextHelpers}>\n          <TamboThreadProvider streaming={streaming}>\n            <TamboComponentProvider>\n              <TamboInteractableProvider>\n                <TamboCompositeProvider>{children}</TamboCompositeProvider>\n              </TamboInteractableProvider>\n            </TamboComponentProvider>\n          </TamboThreadProvider>\n        </TamboContextHelpersProvider>\n      </TamboRegistryProvider>\n    </TamboClientProvider>\n  );\n};\n\nexport type TamboContextProps = TamboClientContextProps &\n  TamboThreadContextProps &\n  TamboGenerationStageContextProps &\n  TamboComponentContextProps &\n  TamboInteractableContext &\n  TamboContextHelpersContextProps;\n\nexport const TamboContext = createContext<TamboContextProps>(\n  {} as TamboContextProps,\n);\n\n/**\n * TamboCompositeProvider is a provider that combines the TamboClient,\n * TamboThread, TamboComponent, and TamboInteractable providers\n * @param props - The props for the TamboCompositeProvider\n * @param props.children - The children to wrap\n * @returns The wrapped component\n */\nexport const TamboCompositeProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const threads = useTamboThread();\n  const client = useTamboClient();\n  const queryClient = useTamboQueryClient();\n  const componentRegistry = useTamboComponent();\n  const interactableComponents = useTamboInteractable();\n  const contextHelpers = useTamboContextHelpers();\n\n  return (\n    <TamboContext.Provider\n      value={{\n        client,\n        queryClient,\n        ...componentRegistry,\n        ...threads,\n        ...interactableComponents,\n        ...contextHelpers,\n      }}\n    >\n      {children}\n    </TamboContext.Provider>\n  );\n};\n\n/**\n * The useTambo hook provides access to the Tambo API. This is the primary entrypoint\n * for the Tambo React SDK.\n *\n * This includes the TamboAI client, the component registry, the current thread context,\n * and interactable component management.\n * @returns The Tambo API\n */\nexport const useTambo = () => {\n  return useContext(TamboContext);\n};\n"]}
+{"version":3,"file":"tambo-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAO,KAAK,EAAE,EAAqB,aAAa,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC;AAE5E,OAAO,EAEL,mBAAmB,EAEnB,uBAAuB,EACvB,cAAc,EACd,mBAAmB,GACpB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAEL,sBAAsB,EACtB,iBAAiB,GAClB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAEL,2BAA2B,EAE3B,sBAAsB,GACvB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EACL,yBAAyB,EACzB,oBAAoB,GACrB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,qBAAqB,GAEtB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,wBAAwB,GAEzB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAGL,mBAAmB,EAEnB,cAAc,GACf,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;;;;;;GAeG;AACH,MAAM,CAAC,MAAM,aAAa,GAQtB,CAAC,EACH,QAAQ,EACR,QAAQ,EACR,MAAM,EACN,SAAS,EACT,UAAU,EACV,WAAW,EACX,KAAK,EACL,SAAS,EACT,cAAc,EACd,UAAU,GACX,EAAE,EAAE;IACH,iCAAiC;IACjC,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,OAAO,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,CACL,oBAAC,mBAAmB,IAClB,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW,EACxB,SAAS,EAAE,SAAS;QAEpB,oBAAC,qBAAqB,IAAC,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK;YACzD,oBAAC,2BAA2B,IAAC,cAAc,EAAE,cAAc;gBACzD,oBAAC,mBAAmB,IAAC,SAAS,EAAE,SAAS;oBACvC,oBAAC,wBAAwB,IAAC,UAAU,EAAE,UAAU;wBAC9C,oBAAC,sBAAsB;4BACrB,oBAAC,yBAAyB;gCACxB,oBAAC,sBAAsB,QAAE,QAAQ,CAA0B,CACjC,CACL,CACA,CACP,CACM,CACR,CACJ,CACvB,CAAC;AACJ,CAAC,CAAC;AASF,MAAM,CAAC,MAAM,YAAY,GAAG,aAAa,CACvC,EAAuB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAgC,CAAC,EAClE,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;IACjC,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;IAChC,MAAM,WAAW,GAAG,mBAAmB,EAAE,CAAC;IAC1C,MAAM,eAAe,GAAG,uBAAuB,EAAE,CAAC;IAClD,MAAM,iBAAiB,GAAG,iBAAiB,EAAE,CAAC;IAC9C,MAAM,sBAAsB,GAAG,oBAAoB,EAAE,CAAC;IACtD,MAAM,cAAc,GAAG,sBAAsB,EAAE,CAAC;IAEhD,OAAO,CACL,oBAAC,YAAY,CAAC,QAAQ,IACpB,KAAK,EAAE;YACL,MAAM;YACN,WAAW;YACX,eAAe;YACf,GAAG,iBAAiB;YACpB,GAAG,OAAO;YACV,GAAG,sBAAsB;YACzB,GAAG,cAAc;SAClB,IAEA,QAAQ,CACa,CACzB,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,GAAG,EAAE;IAC3B,OAAO,UAAU,CAAC,YAAY,CAAC,CAAC;AAClC,CAAC,CAAC","sourcesContent":["\"use client\";\nimport React, { PropsWithChildren, createContext, useContext } from \"react\";\nimport { TamboInteractableContext } from \"../model/tambo-interactable\";\nimport {\n  TamboClientContextProps,\n  TamboClientProvider,\n  TamboClientProviderProps,\n  useIsTamboTokenUpdating,\n  useTamboClient,\n  useTamboQueryClient,\n} from \"./tambo-client-provider\";\nimport {\n  TamboComponentContextProps,\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nimport {\n  TamboContextHelpersContextProps,\n  TamboContextHelpersProvider,\n  TamboContextHelpersProviderProps,\n  useTamboContextHelpers,\n} from \"./tambo-context-helpers-provider\";\nimport {\n  TamboInteractableProvider,\n  useTamboInteractable,\n} from \"./tambo-interactable-provider\";\nimport {\n  TamboRegistryProvider,\n  TamboRegistryProviderProps,\n} from \"./tambo-registry-provider\";\nimport {\n  TamboThreadInputProvider,\n  TamboThreadInputProviderProps,\n} from \"./tambo-thread-input-provider\";\nimport {\n  TamboGenerationStageContextProps,\n  TamboThreadContextProps,\n  TamboThreadProvider,\n  TamboThreadProviderProps,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n\n/**\n * The TamboProvider gives full access to the whole Tambo API. This includes the\n * TamboAI client, the component registry, the current thread context, and interactable components.\n * @param props - The props for the TamboProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.components - The components to register\n * @param props.environment - The environment to use for the Tambo API\n * @param props.tools - The tools to register\n * @param props.streaming - Whether to stream the response by default. Defaults to true.\n * @param props.contextHelpers - Configuration for which context helpers are enabled/disabled\n * @param props.userToken - The JWT id token to use to identify the user in the Tambo API. (preferred over contextKey)\n * @param props.contextKey - Optional context key to be used in the thread input provider\n * @returns The TamboProvider component\n */\nexport const TamboProvider: React.FC<\n  PropsWithChildren<\n    TamboClientProviderProps &\n      TamboRegistryProviderProps &\n      TamboThreadProviderProps &\n      TamboContextHelpersProviderProps &\n      TamboThreadInputProviderProps\n  >\n> = ({\n  children,\n  tamboUrl,\n  apiKey,\n  userToken,\n  components,\n  environment,\n  tools,\n  streaming,\n  contextHelpers,\n  contextKey,\n}) => {\n  // Should only be used in browser\n  if (typeof window === \"undefined\") {\n    console.error(\"TamboProvider must be used within a browser\");\n  }\n\n  return (\n    <TamboClientProvider\n      tamboUrl={tamboUrl}\n      apiKey={apiKey}\n      environment={environment}\n      userToken={userToken}\n    >\n      <TamboRegistryProvider components={components} tools={tools}>\n        <TamboContextHelpersProvider contextHelpers={contextHelpers}>\n          <TamboThreadProvider streaming={streaming}>\n            <TamboThreadInputProvider contextKey={contextKey}>\n              <TamboComponentProvider>\n                <TamboInteractableProvider>\n                  <TamboCompositeProvider>{children}</TamboCompositeProvider>\n                </TamboInteractableProvider>\n              </TamboComponentProvider>\n            </TamboThreadInputProvider>\n          </TamboThreadProvider>\n        </TamboContextHelpersProvider>\n      </TamboRegistryProvider>\n    </TamboClientProvider>\n  );\n};\n\nexport type TamboContextProps = TamboClientContextProps &\n  TamboThreadContextProps &\n  TamboGenerationStageContextProps &\n  TamboComponentContextProps &\n  TamboInteractableContext &\n  TamboContextHelpersContextProps;\n\nexport const TamboContext = createContext<TamboContextProps>(\n  {} as TamboContextProps,\n);\n\n/**\n * TamboCompositeProvider is a provider that combines the TamboClient,\n * TamboThread, TamboComponent, and TamboInteractable providers\n * @param props - The props for the TamboCompositeProvider\n * @param props.children - The children to wrap\n * @returns The wrapped component\n */\nexport const TamboCompositeProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const threads = useTamboThread();\n  const client = useTamboClient();\n  const queryClient = useTamboQueryClient();\n  const isUpdatingToken = useIsTamboTokenUpdating();\n  const componentRegistry = useTamboComponent();\n  const interactableComponents = useTamboInteractable();\n  const contextHelpers = useTamboContextHelpers();\n\n  return (\n    <TamboContext.Provider\n      value={{\n        client,\n        queryClient,\n        isUpdatingToken,\n        ...componentRegistry,\n        ...threads,\n        ...interactableComponents,\n        ...contextHelpers,\n      }}\n    >\n      {children}\n    </TamboContext.Provider>\n  );\n};\n\n/**\n * The useTambo hook provides access to the Tambo API. This is the primary entrypoint\n * for the Tambo React SDK.\n *\n * This includes the TamboAI client, the component registry, the current thread context,\n * and interactable component management.\n * @returns The Tambo API\n */\nexport const useTambo = () => {\n  return useContext(TamboContext);\n};\n"]}

package/esm/providers/tambo-stubs.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-stubs.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-stubs.tsx"],"names":[],"mappings":"AACA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAE/C,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAAa,MAAM,OAAO,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAExE,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAMpD,OAAO,EAEL,gCAAgC,EACjC,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAEL,KAAK,iBAAiB,EACvB,MAAM,kBAAkB,CAAC;AAQ1B;;GAEG;AACH,MAAM,WAAW,sBAAuB,SAAQ,OAAO,CAAC,iBAAiB,CAAC;IACxE,2CAA2C;IAC3C,MAAM,EAAE,WAAW,CAAC;IACpB,wDAAwD;IACxD,UAAU,CAAC,EAAE,cAAc,EAAE,CAAC;IAC9B,mDAAmD;IACnD,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;IACpB,sFAAsF;IACtF,OAAO,CAAC,EAAE,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;IAC9D,iFAAiF;IACjF,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,oDAAoD;IACpD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,6EAA6E;IAC7E,cAAc,CAAC,EAAE,gCAAgC,CAAC,gBAAgB,CAAC,CAAC;CACrE;AAiJD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6DG;AACH,eAAO,MAAM,iBAAiB,EAAE,KAAK,CAAC,EAAE,CACtC,iBAAiB,CAAC,sBAAsB,CAAC,CAiH1C,CAAC"}
+{"version":3,"file":"tambo-stubs.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-stubs.tsx"],"names":[],"mappings":"AACA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAE/C,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAAa,MAAM,OAAO,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAExE,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAMpD,OAAO,EAEL,gCAAgC,EACjC,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAEL,KAAK,iBAAiB,EACvB,MAAM,kBAAkB,CAAC;AAQ1B;;GAEG;AACH,MAAM,WAAW,sBAAuB,SAAQ,OAAO,CAAC,iBAAiB,CAAC;IACxE,2CAA2C;IAC3C,MAAM,EAAE,WAAW,CAAC;IACpB,wDAAwD;IACxD,UAAU,CAAC,EAAE,cAAc,EAAE,CAAC;IAC9B,mDAAmD;IACnD,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;IACpB,sFAAsF;IACtF,OAAO,CAAC,EAAE,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;IAC9D,iFAAiF;IACjF,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,oDAAoD;IACpD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,6EAA6E;IAC7E,cAAc,CAAC,EAAE,gCAAgC,CAAC,gBAAgB,CAAC,CAAC;CACrE;AA4JD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6DG;AACH,eAAO,MAAM,iBAAiB,EAAE,KAAK,CAAC,EAAE,CACtC,iBAAiB,CAAC,sBAAsB,CAAC,CAkH1C,CAAC"}

package/esm/providers/tambo-stubs.js

@@ -12,7 +12,7 @@ import { TamboGenerationStageProvider, TamboThreadContext, } from "./tambo-threa
  * Stub client provider that accepts a client and queryClient as props
  * @returns The TamboStubClientProvider component
  */
-const TamboStubClientProvider = ({ children, client, queryClient, threads, projectId, contextKey }) => {
+const TamboStubClientProvider = ({ children, client, queryClient, threads, projectId, contextKey, isUpdatingToken, }) => {
     // Prepopulate the query cache with threads data if provided
     useEffect(() => {
         if (threads) {
@@ -22,7 +22,7 @@ const TamboStubClientProvider = ({ children, client, queryClient, threads, proje
             queryClient.setQueryData(["threads", projectId, contextKey], threads);
         }
     }, [threads, projectId, contextKey, queryClient]);
-    return (React.createElement(TamboClientContext.Provider, { value: { client, queryClient } }, children));
+    return (React.createElement(TamboClientContext.Provider, { value: { client, queryClient, isUpdatingToken } }, children));
 };
 /**
  * Stub registry provider that accepts componentList, toolRegistry, and componentToolAssociations as props
@@ -208,7 +208,7 @@ export const TamboStubProvider = ({ children, thread, components = [], tools = [
         registerTools: overrides.registerTools ?? defaults.registerTools,
         addToolAssociation: overrides.addToolAssociation ?? defaults.addToolAssociation,
     };
-    return (React.createElement(TamboStubClientProvider, { client: stubClient, queryClient: stubQueryClient, threads: threads, projectId: resolvedProjectId, contextKey: contextKey },
+    return (React.createElement(TamboStubClientProvider, { client: stubClient, queryClient: stubQueryClient, threads: threads, projectId: resolvedProjectId, contextKey: contextKey, isUpdatingToken: false },
         React.createElement(TamboStubRegistryProvider, { componentList: componentList, toolRegistry: toolRegistry, componentToolAssociations: componentToolAssociations, registerComponent: componentContextProps.registerComponent, registerTool: componentContextProps.registerTool, registerTools: componentContextProps.registerTools, addToolAssociation: componentContextProps.addToolAssociation },
             React.createElement(TamboStubThreadProvider, { ...threadContextProps },
                 React.createElement(TamboContextHelpersProvider, { contextHelpers: contextHelpers },