@tambo-ai/react 0.20.1 → 0.20.3

package/dist/hooks/__tests__/use-suggestions.test.js

@@ -11,7 +11,7 @@ const use_thread_input_1 = require("../use-thread-input");
 // Mock the required providers
 jest.mock("../../providers/tambo-client-provider", () => ({
     useTamboClient: jest.fn(),
-    useTamboQueryClient: jest.fn().mockReturnValue(new react_query_1.QueryClient()),
+    useTamboQueryClient: jest.fn(),
 }));
 jest.mock("../../providers/tambo-provider", () => ({ useTambo: jest.fn() }));
 jest.mock("../../providers/tambo-thread-provider", () => ({
@@ -21,12 +21,7 @@ jest.mock("../use-thread-input", () => ({ useTamboThreadInput: jest.fn() }));
 // Mock the react-query-hooks
 jest.mock("../react-query-hooks", () => ({
     useTamboQuery: jest.fn(),
-    useTamboMutationResult: jest.fn().mockImplementation(({ mutationFn }) => ({
-        mutateAsync: mutationFn,
-        isLoading: false,
-        isError: false,
-        error: null,
-    })),
+    useTamboMutation: jest.fn(),
 }));
 describe("useTamboSuggestions", () => {
     const mockSuggestions = [
@@ -44,23 +39,39 @@ describe("useTamboSuggestions", () => {
         },
     ];
     beforeEach(() => {
-        jest.clearAllMocks();
+        jest.mocked(tambo_client_provider_1.useTamboQueryClient).mockReturnValue(new react_query_1.QueryClient());
+        jest.mocked(react_query_hooks_1.useTamboMutation).mockImplementation(({ mutationFn }) => ({
+            mutateAsync: mutationFn,
+            isLoading: false,
+            isError: false,
+            error: null,
+        }));
         // Setup default mock implementations
-        tambo_client_provider_1.useTamboClient.mockReturnValue({
+        jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue({
             beta: { threads: { suggestions: { generate: jest.fn() } } },
         });
-        tambo_provider_1.useTambo.mockReturnValue({ sendThreadMessage: jest.fn() });
-        tambo_thread_provider_1.useTamboThread.mockReturnValue({
+        jest.mocked(tambo_provider_1.useTambo).mockReturnValue({
+            sendThreadMessage: jest.fn(),
+        });
+        jest.mocked(tambo_thread_provider_1.useTamboThread).mockReturnValue({
             thread: {
                 id: "test-thread-id",
                 messages: [
-                    { id: "test-message-id", role: "hydra", content: "Test message" },
+                    {
+                        id: "test-message-id",
+                        role: "hydra",
+                        content: [{ type: "text", text: "Test message" }],
+                    },
                 ],
             },
         });
-        use_thread_input_1.useTamboThreadInput.mockReturnValue({ setValue: jest.fn() });
+        jest.mocked(use_thread_input_1.useTamboThreadInput).mockReturnValue({
+            setValue: jest.fn(),
+            value: "",
+            submit: jest.fn(),
+        });
         // Default query mock returns empty array
-        react_query_hooks_1.useTamboQuery.mockReturnValue({
+        jest.mocked(react_query_hooks_1.useTamboQuery).mockReturnValue({
             data: [],
             isLoading: false,
             isError: false,
@@ -74,11 +85,30 @@ describe("useTamboSuggestions", () => {
     });
     it("should generate suggestions when latest message is from Tambo", async () => {
         const mockGenerate = jest.fn().mockResolvedValue(mockSuggestions);
-        tambo_client_provider_1.useTamboClient.mockReturnValue({
-            beta: { threads: { suggestions: { generate: mockGenerate } } },
+        jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue({
+            beta: {
+                threads: {
+                    suggestions: {
+                        generate: mockGenerate,
+                        list: jest.fn(),
+                    },
+                },
+            },
+        });
+        jest.mocked(tambo_thread_provider_1.useTamboThread).mockReturnValue({
+            thread: {
+                id: "test-thread-id",
+                messages: [
+                    {
+                        id: "test-message-id",
+                        role: "assistant",
+                        content: [{ type: "text", text: "Test message" }],
+                    },
+                ],
+            },
         });
         // Mock the query result to return the mock suggestions
-        react_query_hooks_1.useTamboQuery.mockReturnValue({
+        jest.mocked(react_query_hooks_1.useTamboQuery).mockReturnValue({
             data: mockSuggestions,
             isLoading: false,
             isError: false,
@@ -95,15 +125,26 @@ describe("useTamboSuggestions", () => {
     });
     it("should not generate suggestions when latest message is not from Tambo", async () => {
         const mockGenerate = jest.fn();
-        tambo_client_provider_1.useTamboClient.mockReturnValue({
-            beta: { threads: { suggestions: { generate: mockGenerate } } },
+        jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue({
+            beta: {
+                threads: {
+                    suggestions: {
+                        generate: mockGenerate,
+                        list: jest.fn(),
+                    },
+                },
+            },
         });
         // Mock the thread to have a non-Tambo message
-        tambo_thread_provider_1.useTamboThread.mockReturnValue({
+        jest.mocked(tambo_thread_provider_1.useTamboThread).mockReturnValue({
             thread: {
                 id: "test-thread-id",
                 messages: [
-                    { id: "test-message-id", role: "user", content: "Test message" },
+                    {
+                        id: "test-message-id",
+                        role: "user",
+                        content: [{ type: "text", text: "Test message" }],
+                    },
                 ],
             },
         });
@@ -117,8 +158,10 @@ describe("useTamboSuggestions", () => {
     });
     it("should accept a suggestion and update input value", async () => {
         const mockSetValue = jest.fn();
-        use_thread_input_1.useTamboThreadInput.mockReturnValue({
+        jest.mocked(use_thread_input_1.useTamboThreadInput).mockReturnValue({
             setValue: mockSetValue,
+            value: "",
+            submit: jest.fn(),
         });
         const { result } = (0, react_1.renderHook)(() => (0, use_suggestions_1.useTamboSuggestions)());
         await (0, react_1.act)(async () => {
@@ -132,7 +175,7 @@ describe("useTamboSuggestions", () => {
     });
     it("should accept a suggestion and submit it", async () => {
         const mockSendThreadMessage = jest.fn();
-        tambo_provider_1.useTambo.mockReturnValue({
+        jest.mocked(tambo_provider_1.useTambo).mockReturnValue({
             sendThreadMessage: mockSendThreadMessage,
         });
         const { result } = (0, react_1.renderHook)(() => (0, use_suggestions_1.useTamboSuggestions)());

package/dist/hooks/__tests__/use-suggestions.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-suggestions.test.js","sourceRoot":"","sources":["../../../src/hooks/__tests__/use-suggestions.test.tsx"],"names":[],"mappings":";;AACA,uDAAoD;AACpD,kDAAyD;AACzD,iFAAuE;AACvE,mEAA0D;AAC1D,iFAAuE;AACvE,4DAAqD;AACrD,wDAAyD;AACzD,0DAA0D;AAE1D,8BAA8B;AAC9B,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,yBAAW,EAAE,CAAC;CAClE,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,gCAAgC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAE7E,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAE7E,6BAA6B;AAC7B,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;IACxB,sBAAsB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;QACxE,WAAW,EAAE,UAAU;QACvB,SAAS,EAAE,KAAK;QAChB,OAAO,EAAE,KAAK;QACd,KAAK,EAAE,IAAI;KACZ,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,MAAM,eAAe,GAAsC;QACzD;YACE,EAAE,EAAE,cAAc;YAClB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,mBAAmB;YAC1B,kBAAkB,EAAE,mBAAmB;SACxC;QACD;YACE,EAAE,EAAE,cAAc;YAClB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,mBAAmB;YAC1B,kBAAkB,EAAE,mBAAmB;SACxC;KACF,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,qCAAqC;QACpC,sCAA4B,CAAC,eAAe,CAAC;YAC5C,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE;SAC5D,CAAC,CAAC;QACF,yBAAsB,CAAC,eAAe,CAAC,EAAE,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACzE,sCAA4B,CAAC,eAAe,CAAC;YAC5C,MAAM,EAAE;gBACN,EAAE,EAAE,gBAAgB;gBACpB,QAAQ,EAAE;oBACR,EAAE,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE;iBAClE;aACF;SACF,CAAC,CAAC;QACF,sCAAiC,CAAC,eAAe,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAC5E,yCAAyC;QACxC,iCAA2B,CAAC,eAAe,CAAC;YAC3C,IAAI,EAAE,EAAE;YACR,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,QAAQ,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QACjE,sCAA4B,CAAC,eAAe,CAAC;YAC5C,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE;SAC/D,CAAC,CAAC;QAEH,uDAAuD;QACtD,iCAA2B,CAAC,eAAe,CAAC;YAC3C,IAAI,EAAE,eAAe;YACrB,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,6BAA6B;QAC7B,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,wEAAwE;QACxE,wEAAwE;QACxE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC9B,sCAA4B,CAAC,eAAe,CAAC;YAC5C,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE;SAC/D,CAAC,CAAC;QAEH,8CAA8C;QAC7C,sCAA4B,CAAC,eAAe,CAAC;YAC5C,MAAM,EAAE;gBACN,EAAE,EAAE,gBAAgB;gBACpB,QAAQ,EAAE;oBACR,EAAE,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE;iBACjE;aACF;SACF,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,6BAA6B;QAC7B,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC9B,sCAAiC,CAAC,eAAe,CAAC;YACjD,QAAQ,EAAE,YAAY;SACvB,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;gBAC1B,UAAU,EAAE,eAAe,CAAC,CAAC,CAAC;gBAC9B,YAAY,EAAE,KAAK;aACpB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,qBAAqB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACvC,yBAAsB,CAAC,eAAe,CAAC;YACtC,iBAAiB,EAAE,qBAAqB;SACzC,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;gBAC1B,UAAU,EAAE,eAAe,CAAC,CAAC,CAAC;gBAC9B,YAAY,EAAE,IAAI;aACnB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,EAAE;YACtE,QAAQ,EAAE,gBAAgB;SAC3B,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,iBAAiB,GAAG;YACxB,EAAE,EAAE,oBAAoB;YACxB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,oBAAoB;YAC3B,kBAAkB,EAAE,EAAE,EAAE,0CAA0C;SACnE,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CACV,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;gBACpB,UAAU,EAAE,iBAAiB;gBAC7B,YAAY,EAAE,KAAK;aACpB,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,QAAQ,EAAE,CAAC;IACzD,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 { useTamboClient } from \"../../providers/tambo-client-provider\";\nimport { useTambo } from \"../../providers/tambo-provider\";\nimport { useTamboThread } from \"../../providers/tambo-thread-provider\";\nimport { useTamboQuery } from \"../react-query-hooks\";\nimport { useTamboSuggestions } from \"../use-suggestions\";\nimport { useTamboThreadInput } from \"../use-thread-input\";\n\n// Mock the required providers\njest.mock(\"../../providers/tambo-client-provider\", () => ({\n  useTamboClient: jest.fn(),\n  useTamboQueryClient: jest.fn().mockReturnValue(new QueryClient()),\n}));\n\njest.mock(\"../../providers/tambo-provider\", () => ({ useTambo: jest.fn() }));\n\njest.mock(\"../../providers/tambo-thread-provider\", () => ({\n  useTamboThread: jest.fn(),\n}));\n\njest.mock(\"../use-thread-input\", () => ({ useTamboThreadInput: jest.fn() }));\n\n// Mock the react-query-hooks\njest.mock(\"../react-query-hooks\", () => ({\n  useTamboQuery: jest.fn(),\n  useTamboMutationResult: jest.fn().mockImplementation(({ mutationFn }) => ({\n    mutateAsync: mutationFn,\n    isLoading: false,\n    isError: false,\n    error: null,\n  })),\n}));\n\ndescribe(\"useTamboSuggestions\", () => {\n  const mockSuggestions: TamboAI.Beta.Threads.Suggestion[] = [\n    {\n      id: \"suggestion-1\",\n      messageId: \"test-message-id\",\n      title: \"Test Suggestion 1\",\n      detailedSuggestion: \"Test suggestion 1\",\n    },\n    {\n      id: \"suggestion-2\",\n      messageId: \"test-message-id\",\n      title: \"Test Suggestion 2\",\n      detailedSuggestion: \"Test suggestion 2\",\n    },\n  ];\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n    // Setup default mock implementations\n    (useTamboClient as jest.Mock).mockReturnValue({\n      beta: { threads: { suggestions: { generate: jest.fn() } } },\n    });\n    (useTambo as jest.Mock).mockReturnValue({ sendThreadMessage: jest.fn() });\n    (useTamboThread as jest.Mock).mockReturnValue({\n      thread: {\n        id: \"test-thread-id\",\n        messages: [\n          { id: \"test-message-id\", role: \"hydra\", content: \"Test message\" },\n        ],\n      },\n    });\n    (useTamboThreadInput as jest.Mock).mockReturnValue({ setValue: jest.fn() });\n    // Default query mock returns empty array\n    (useTamboQuery as jest.Mock).mockReturnValue({\n      data: [],\n      isLoading: false,\n      isError: false,\n      error: null,\n    });\n  });\n\n  it(\"should initialize with empty suggestions and no selected suggestion\", () => {\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    expect(result.current.suggestions).toEqual([]);\n    expect(result.current.selectedSuggestionId).toBeNull();\n  });\n\n  it(\"should generate suggestions when latest message is from Tambo\", async () => {\n    const mockGenerate = jest.fn().mockResolvedValue(mockSuggestions);\n    (useTamboClient as jest.Mock).mockReturnValue({\n      beta: { threads: { suggestions: { generate: mockGenerate } } },\n    });\n\n    // Mock the query result to return the mock suggestions\n    (useTamboQuery as jest.Mock).mockReturnValue({\n      data: mockSuggestions,\n      isLoading: false,\n      isError: false,\n      error: null,\n    });\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    // Wait for the effect to run\n    await act(async () => {\n      await new Promise((resolve) => setTimeout(resolve, 0));\n    });\n\n    // Since we're mocking useTamboQuery to return the suggestions directly,\n    // the generate function won't be called, so we don't need to check that\n    expect(result.current.suggestions).toEqual(mockSuggestions);\n  });\n\n  it(\"should not generate suggestions when latest message is not from Tambo\", async () => {\n    const mockGenerate = jest.fn();\n    (useTamboClient as jest.Mock).mockReturnValue({\n      beta: { threads: { suggestions: { generate: mockGenerate } } },\n    });\n\n    // Mock the thread to have a non-Tambo message\n    (useTamboThread as jest.Mock).mockReturnValue({\n      thread: {\n        id: \"test-thread-id\",\n        messages: [\n          { id: \"test-message-id\", role: \"user\", content: \"Test message\" },\n        ],\n      },\n    });\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    // Wait for the effect to run\n    await act(async () => {\n      await new Promise((resolve) => setTimeout(resolve, 0));\n    });\n\n    expect(mockGenerate).not.toHaveBeenCalled();\n    expect(result.current.suggestions).toEqual([]);\n  });\n\n  it(\"should accept a suggestion and update input value\", async () => {\n    const mockSetValue = jest.fn();\n    (useTamboThreadInput as jest.Mock).mockReturnValue({\n      setValue: mockSetValue,\n    });\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    await act(async () => {\n      await result.current.accept({\n        suggestion: mockSuggestions[0],\n        shouldSubmit: false,\n      });\n    });\n\n    expect(mockSetValue).toHaveBeenCalledWith(\"Test suggestion 1\");\n    expect(result.current.selectedSuggestionId).toBe(\"suggestion-1\");\n  });\n\n  it(\"should accept a suggestion and submit it\", async () => {\n    const mockSendThreadMessage = jest.fn();\n    (useTambo as jest.Mock).mockReturnValue({\n      sendThreadMessage: mockSendThreadMessage,\n    });\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    await act(async () => {\n      await result.current.accept({\n        suggestion: mockSuggestions[0],\n        shouldSubmit: true,\n      });\n    });\n\n    expect(mockSendThreadMessage).toHaveBeenCalledWith(\"Test suggestion 1\", {\n      threadId: \"test-thread-id\",\n    });\n    expect(result.current.selectedSuggestionId).toBe(\"suggestion-1\");\n  });\n\n  it(\"should throw error when accepting invalid suggestion\", async () => {\n    const invalidSuggestion = {\n      id: \"invalid-suggestion\",\n      messageId: \"test-message-id\",\n      title: \"Invalid Suggestion\",\n      detailedSuggestion: \"\", // Empty suggestion should fail validation\n    };\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    await act(async () => {\n      await expect(\n        result.current.accept({\n          suggestion: invalidSuggestion,\n          shouldSubmit: false,\n        }),\n      ).rejects.toThrow(\"Message cannot be empty\");\n    });\n\n    expect(result.current.selectedSuggestionId).toBeNull();\n  });\n});\n"]}
+{"version":3,"file":"use-suggestions.test.js","sourceRoot":"","sources":["../../../src/hooks/__tests__/use-suggestions.test.tsx"],"names":[],"mappings":";;AACA,uDAAoE;AACpE,kDAAyD;AAEzD,iFAG+C;AAC/C,mEAA6E;AAC7E,iFAG+C;AAE/C,4DAAuE;AACvE,wDAAyD;AACzD,0DAA0E;AAE1E,8BAA8B;AAC9B,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC/B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,gCAAgC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAE7E,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAE7E,6BAA6B;AAC7B,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;IACxB,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC5B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,MAAM,eAAe,GAAsC;QACzD;YACE,EAAE,EAAE,cAAc;YAClB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,mBAAmB;YAC1B,kBAAkB,EAAE,mBAAmB;SACxC;QACD;YACE,EAAE,EAAE,cAAc;YAClB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,mBAAmB;YAC1B,kBAAkB,EAAE,mBAAmB;SACxC;KACF,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,MAAM,CAAC,2CAAmB,CAAC,CAAC,eAAe,CAAC,IAAI,yBAAW,EAAE,CAAC,CAAC;QACpE,IAAI,CAAC,MAAM,CAAC,oCAAgB,CAAC,CAAC,kBAAkB,CAC9C,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CACjB,CAAC;YACC,WAAW,EAAE,UAAU;YACvB,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,IAAI;SACZ,CAAQ,CACZ,CAAC;QACF,qCAAqC;QACrC,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE;SAC5B,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM,CAAC,yBAAQ,CAAC,CAAC,eAAe,CAAC;YACpC,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;SACe,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,MAAM,EAAE;gBACN,EAAE,EAAE,gBAAgB;gBACpB,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,iBAAiB;wBACrB,IAAI,EAAE,OAAO;wBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;qBAClD;iBACF;aACF;SACyB,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,CAAC,sCAAmB,CAAC,CAAC,eAAe,CAAC;YAC/C,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;SACuB,CAAC,CAAC;QAC5C,yCAAyC;QACzC,IAAI,CAAC,MAAM,CAAC,iCAAa,CAAC,CAAC,eAAe,CAAC;YACzC,IAAI,EAAE,EAAE;YACR,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,IAAI;SAC+C,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,QAAQ,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAClE,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,IAAI,EAAE;gBACJ,OAAO,EAAE;oBACP,WAAW,EAAE;wBACX,QAAQ,EAAE,YAAY;wBACtB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;qBAChB;iBACF;aACF;SAC8B,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,MAAM,EAAE;gBACN,EAAE,EAAE,gBAAgB;gBACpB,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,iBAAiB;wBACrB,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;qBAC5B;iBACxB;aACF;SACyB,CAAC,CAAC;QAE9B,uDAAuD;QACvD,IAAI,CAAC,MAAM,CAAC,iCAAa,CAAC,CAAC,eAAe,CAAC;YACzC,IAAI,EAAE,eAAe;YACrB,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,IAAI;SACwB,CAAC,CAAC;QAEvC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,6BAA6B;QAC7B,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,wEAAwE;QACxE,wEAAwE;QACxE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,IAAI,EAAE;gBACJ,OAAO,EAAE;oBACP,WAAW,EAAE;wBACX,QAAQ,EAAE,YAAY;wBACtB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;qBAChB;iBACF;aACF;SAC8B,CAAC,CAAC;QACnC,8CAA8C;QAC9C,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,MAAM,EAAE;gBACN,EAAE,EAAE,gBAAgB;gBACpB,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,iBAAiB;wBACrB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;qBAC5B;iBACxB;aACF;SACyB,CAAC,CAAC;QAE9B,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,6BAA6B;QAC7B,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,sCAAmB,CAAC,CAAC,eAAe,CAAC;YAC/C,QAAQ,EAAE,YAAY;YACtB,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;SACuB,CAAC,CAAC;QAE5C,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;gBAC1B,UAAU,EAAE,eAAe,CAAC,CAAC,CAAC;gBAC9B,YAAY,EAAE,KAAK;aACpB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,qBAAqB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACxC,IAAI,CAAC,MAAM,CAAC,yBAAQ,CAAC,CAAC,eAAe,CAAC;YACpC,iBAAiB,EAAE,qBAAqB;SACG,CAAC,CAAC;QAE/C,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;gBAC1B,UAAU,EAAE,eAAe,CAAC,CAAC,CAAC;gBAC9B,YAAY,EAAE,IAAI;aACnB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,EAAE;YACtE,QAAQ,EAAE,gBAAgB;SAC3B,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,iBAAiB,GAAG;YACxB,EAAE,EAAE,oBAAoB;YACxB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,oBAAoB;YAC3B,kBAAkB,EAAE,EAAE,EAAE,0CAA0C;SACnE,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CACV,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;gBACpB,UAAU,EAAE,iBAAiB;gBAC7B,YAAY,EAAE,KAAK;aACpB,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,QAAQ,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import type TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { QueryClient, UseQueryResult } from \"@tanstack/react-query\";\nimport { act, renderHook } from \"@testing-library/react\";\nimport { TamboThreadMessage } from \"../../model/generate-component-response\";\nimport {\n  useTamboClient,\n  useTamboQueryClient,\n} from \"../../providers/tambo-client-provider\";\nimport { TamboContextProps, useTambo } from \"../../providers/tambo-provider\";\nimport {\n  TamboThreadContextProps,\n  useTamboThread,\n} from \"../../providers/tambo-thread-provider\";\nimport { PartialTamboAI } from \"../../testing/types\";\nimport { useTamboMutation, useTamboQuery } from \"../react-query-hooks\";\nimport { useTamboSuggestions } from \"../use-suggestions\";\nimport { useTamboThreadInput, UseThreadInput } from \"../use-thread-input\";\n\n// Mock the required providers\njest.mock(\"../../providers/tambo-client-provider\", () => ({\n  useTamboClient: jest.fn(),\n  useTamboQueryClient: jest.fn(),\n}));\n\njest.mock(\"../../providers/tambo-provider\", () => ({ useTambo: jest.fn() }));\n\njest.mock(\"../../providers/tambo-thread-provider\", () => ({\n  useTamboThread: jest.fn(),\n}));\n\njest.mock(\"../use-thread-input\", () => ({ useTamboThreadInput: jest.fn() }));\n\n// Mock the react-query-hooks\njest.mock(\"../react-query-hooks\", () => ({\n  useTamboQuery: jest.fn(),\n  useTamboMutation: jest.fn(),\n}));\n\ndescribe(\"useTamboSuggestions\", () => {\n  const mockSuggestions: TamboAI.Beta.Threads.Suggestion[] = [\n    {\n      id: \"suggestion-1\",\n      messageId: \"test-message-id\",\n      title: \"Test Suggestion 1\",\n      detailedSuggestion: \"Test suggestion 1\",\n    },\n    {\n      id: \"suggestion-2\",\n      messageId: \"test-message-id\",\n      title: \"Test Suggestion 2\",\n      detailedSuggestion: \"Test suggestion 2\",\n    },\n  ];\n\n  beforeEach(() => {\n    jest.mocked(useTamboQueryClient).mockReturnValue(new QueryClient());\n    jest.mocked(useTamboMutation).mockImplementation(\n      ({ mutationFn }) =>\n        ({\n          mutateAsync: mutationFn,\n          isLoading: false,\n          isError: false,\n          error: null,\n        }) as any,\n    );\n    // Setup default mock implementations\n    jest.mocked(useTamboClient).mockReturnValue({\n      beta: { threads: { suggestions: { generate: jest.fn() } } },\n    } satisfies PartialTamboAI as any);\n    jest.mocked(useTambo).mockReturnValue({\n      sendThreadMessage: jest.fn(),\n    } satisfies Partial<TamboContextProps> as any);\n    jest.mocked(useTamboThread).mockReturnValue({\n      thread: {\n        id: \"test-thread-id\",\n        messages: [\n          {\n            id: \"test-message-id\",\n            role: \"hydra\",\n            content: [{ type: \"text\", text: \"Test message\" }],\n          },\n        ],\n      },\n    } as TamboThreadContextProps);\n    jest.mocked(useTamboThreadInput).mockReturnValue({\n      setValue: jest.fn(),\n      value: \"\",\n      submit: jest.fn(),\n    } satisfies Partial<UseThreadInput> as any);\n    // Default query mock returns empty array\n    jest.mocked(useTamboQuery).mockReturnValue({\n      data: [],\n      isLoading: false,\n      isError: false,\n      error: null,\n    } satisfies Partial<UseQueryResult<unknown, unknown>> as any);\n  });\n\n  it(\"should initialize with empty suggestions and no selected suggestion\", () => {\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    expect(result.current.suggestions).toEqual([]);\n    expect(result.current.selectedSuggestionId).toBeNull();\n  });\n\n  it(\"should generate suggestions when latest message is from Tambo\", async () => {\n    const mockGenerate = jest.fn().mockResolvedValue(mockSuggestions);\n    jest.mocked(useTamboClient).mockReturnValue({\n      beta: {\n        threads: {\n          suggestions: {\n            generate: mockGenerate,\n            list: jest.fn(),\n          },\n        },\n      },\n    } satisfies PartialTamboAI as any);\n    jest.mocked(useTamboThread).mockReturnValue({\n      thread: {\n        id: \"test-thread-id\",\n        messages: [\n          {\n            id: \"test-message-id\",\n            role: \"assistant\",\n            content: [{ type: \"text\", text: \"Test message\" }],\n          } as TamboThreadMessage,\n        ],\n      },\n    } as TamboThreadContextProps);\n\n    // Mock the query result to return the mock suggestions\n    jest.mocked(useTamboQuery).mockReturnValue({\n      data: mockSuggestions,\n      isLoading: false,\n      isError: false,\n      error: null,\n    } as UseQueryResult<unknown, unknown>);\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    // Wait for the effect to run\n    await act(async () => {\n      await new Promise((resolve) => setTimeout(resolve, 0));\n    });\n\n    // Since we're mocking useTamboQuery to return the suggestions directly,\n    // the generate function won't be called, so we don't need to check that\n    expect(result.current.suggestions).toEqual(mockSuggestions);\n  });\n\n  it(\"should not generate suggestions when latest message is not from Tambo\", async () => {\n    const mockGenerate = jest.fn();\n    jest.mocked(useTamboClient).mockReturnValue({\n      beta: {\n        threads: {\n          suggestions: {\n            generate: mockGenerate,\n            list: jest.fn(),\n          },\n        },\n      },\n    } satisfies PartialTamboAI as any);\n    // Mock the thread to have a non-Tambo message\n    jest.mocked(useTamboThread).mockReturnValue({\n      thread: {\n        id: \"test-thread-id\",\n        messages: [\n          {\n            id: \"test-message-id\",\n            role: \"user\",\n            content: [{ type: \"text\", text: \"Test message\" }],\n          } as TamboThreadMessage,\n        ],\n      },\n    } as TamboThreadContextProps);\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    // Wait for the effect to run\n    await act(async () => {\n      await new Promise((resolve) => setTimeout(resolve, 0));\n    });\n\n    expect(mockGenerate).not.toHaveBeenCalled();\n    expect(result.current.suggestions).toEqual([]);\n  });\n\n  it(\"should accept a suggestion and update input value\", async () => {\n    const mockSetValue = jest.fn();\n    jest.mocked(useTamboThreadInput).mockReturnValue({\n      setValue: mockSetValue,\n      value: \"\",\n      submit: jest.fn(),\n    } satisfies Partial<UseThreadInput> as any);\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    await act(async () => {\n      await result.current.accept({\n        suggestion: mockSuggestions[0],\n        shouldSubmit: false,\n      });\n    });\n\n    expect(mockSetValue).toHaveBeenCalledWith(\"Test suggestion 1\");\n    expect(result.current.selectedSuggestionId).toBe(\"suggestion-1\");\n  });\n\n  it(\"should accept a suggestion and submit it\", async () => {\n    const mockSendThreadMessage = jest.fn();\n    jest.mocked(useTambo).mockReturnValue({\n      sendThreadMessage: mockSendThreadMessage,\n    } satisfies Partial<TamboContextProps> as any);\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    await act(async () => {\n      await result.current.accept({\n        suggestion: mockSuggestions[0],\n        shouldSubmit: true,\n      });\n    });\n\n    expect(mockSendThreadMessage).toHaveBeenCalledWith(\"Test suggestion 1\", {\n      threadId: \"test-thread-id\",\n    });\n    expect(result.current.selectedSuggestionId).toBe(\"suggestion-1\");\n  });\n\n  it(\"should throw error when accepting invalid suggestion\", async () => {\n    const invalidSuggestion = {\n      id: \"invalid-suggestion\",\n      messageId: \"test-message-id\",\n      title: \"Invalid Suggestion\",\n      detailedSuggestion: \"\", // Empty suggestion should fail validation\n    };\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    await act(async () => {\n      await expect(\n        result.current.accept({\n          suggestion: invalidSuggestion,\n          shouldSubmit: false,\n        }),\n      ).rejects.toThrow(\"Message cannot be empty\");\n    });\n\n    expect(result.current.selectedSuggestionId).toBeNull();\n  });\n});\n"]}

package/dist/hooks/__tests__/use-tambo-threads.test.d.ts

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

package/dist/hooks/__tests__/use-tambo-threads.test.d.ts.map

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

package/dist/hooks/__tests__/use-tambo-threads.test.js

@@ -0,0 +1,212 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const react_query_1 = require("@tanstack/react-query");
+const react_1 = require("@testing-library/react");
+const tambo_client_provider_1 = require("../../providers/tambo-client-provider");
+const use_tambo_threads_1 = require("../use-tambo-threads");
+jest.mock("../../providers/tambo-client-provider", () => ({
+    useTamboClient: jest.fn(),
+    useTamboQueryClient: jest.fn(),
+}));
+describe("useTamboThreadList", () => {
+    const mockThreads = [
+        { id: "thread-1", title: "Thread 1" },
+        { id: "thread-2", title: "Thread 2" },
+    ];
+    const mockProjects = {
+        getCurrent: jest.fn(),
+        retrieve: jest.fn(),
+        delete: jest.fn(),
+    };
+    const mockThreadsApi = {
+        list: jest.fn(),
+        messages: {
+            list: jest.fn(),
+            create: jest.fn(),
+            delete: jest.fn(),
+            updateComponentState: jest.fn(),
+        },
+        suggestions: {
+            list: jest.fn(),
+            generate: jest.fn(),
+        },
+        create: jest.fn(),
+        retrieve: jest.fn(),
+        update: jest.fn(),
+        delete: jest.fn(),
+        advance: jest.fn(),
+        advanceById: jest.fn(),
+    };
+    const mockBeta = {
+        projects: mockProjects,
+        threads: mockThreadsApi,
+        registry: {
+            retrieve: jest.fn(),
+        },
+    };
+    const mockTamboAI = {
+        apiKey: "",
+        components: {},
+        beta: mockBeta,
+    };
+    beforeEach(() => {
+        jest.mocked(tambo_client_provider_1.useTamboQueryClient).mockReturnValue(new react_query_1.QueryClient());
+    });
+    it("should fetch threads for current project when no projectId is provided", async () => {
+        const mockClient = jest.mocked(tambo_client_provider_1.useTamboClient);
+        mockClient.mockReturnValue({
+            ...mockTamboAI,
+            beta: {
+                ...mockBeta,
+                projects: {
+                    ...mockProjects,
+                    getCurrent: jest.fn().mockResolvedValue({ id: "current-project" }),
+                },
+                threads: {
+                    ...mockThreadsApi,
+                    list: jest.fn().mockResolvedValue(mockThreads),
+                },
+            },
+        });
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_threads_1.useTamboThreadList)());
+        await (0, react_1.waitFor)(() => {
+            expect(result.current.data).toEqual(mockThreads);
+        });
+    });
+    it("should fetch threads for specified projectId", async () => {
+        const mockList = jest.fn().mockResolvedValue(mockThreads);
+        const mockClient = jest.mocked(tambo_client_provider_1.useTamboClient);
+        mockClient.mockReturnValue({
+            ...mockTamboAI,
+            beta: {
+                ...mockBeta,
+                threads: {
+                    ...mockThreadsApi,
+                    list: mockList,
+                },
+            },
+        });
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_threads_1.useTamboThreadList)({ projectId: "custom-project" }));
+        await (0, react_1.waitFor)(() => {
+            expect(result.current.data).toEqual(mockThreads);
+        });
+        expect(mockList).toHaveBeenCalledWith("custom-project", {});
+    });
+    it("should fetch threads with contextKey when provided", async () => {
+        const mockList = jest.fn().mockResolvedValue(mockThreads);
+        const mockClient = jest.mocked(tambo_client_provider_1.useTamboClient);
+        mockClient.mockReturnValue({
+            ...mockTamboAI,
+            beta: {
+                ...mockBeta,
+                projects: {
+                    ...mockProjects,
+                    getCurrent: jest.fn().mockResolvedValue({ id: "current-project" }),
+                },
+                threads: {
+                    ...mockThreadsApi,
+                    list: mockList,
+                },
+            },
+        });
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_threads_1.useTamboThreadList)({ contextKey: "test-context" }));
+        await (0, react_1.waitFor)(() => {
+            expect(result.current.data).toEqual(mockThreads);
+        });
+        expect(mockList).toHaveBeenCalledWith("current-project", {
+            contextKey: "test-context",
+        });
+    });
+    it("should handle loading state", async () => {
+        let resolvePromise;
+        const promise = new Promise((resolve) => {
+            resolvePromise = resolve;
+        });
+        const mockClient = jest.mocked(tambo_client_provider_1.useTamboClient);
+        mockClient.mockReturnValue({
+            ...mockTamboAI,
+            beta: {
+                ...mockBeta,
+                projects: {
+                    ...mockProjects,
+                    getCurrent: jest.fn().mockResolvedValue({ id: "current-project" }),
+                },
+                threads: {
+                    ...mockThreadsApi,
+                    list: jest.fn().mockReturnValue(promise),
+                },
+            },
+        });
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_threads_1.useTamboThreadList)({}, { retry: false }));
+        expect(result.current).toMatchInlineSnapshot(`
+      {
+        "data": null,
+        "dataUpdatedAt": 0,
+        "error": null,
+        "errorUpdateCount": 0,
+        "errorUpdatedAt": 0,
+        "failureCount": 0,
+        "failureReason": null,
+        "fetchStatus": "fetching",
+        "isError": false,
+        "isFetched": false,
+        "isFetchedAfterMount": false,
+        "isFetching": true,
+        "isInitialLoading": true,
+        "isLoading": true,
+        "isLoadingError": false,
+        "isPaused": false,
+        "isPending": true,
+        "isPlaceholderData": false,
+        "isRefetchError": false,
+        "isRefetching": false,
+        "isStale": true,
+        "isSuccess": false,
+        "promise": Promise {
+          "reason": [Error: experimental_prefetchInRender feature flag is not enabled],
+          "status": "rejected",
+        },
+        "refetch": [Function],
+        "status": "pending",
+      }
+    `);
+        expect(result.current.isLoading).toBe(true);
+        expect(result.current.data).toBeNull();
+        resolvePromise(mockThreads);
+        await (0, react_1.waitFor)(() => {
+            expect(result.current.isLoading).toBe(false);
+        });
+        await (0, react_1.waitFor)(() => {
+            expect(result.current.isLoading).toBe(false);
+        });
+    });
+    it("should handle error state", async () => {
+        const mockError = new Error("Failed to fetch threads");
+        const mockClient = jest.mocked(tambo_client_provider_1.useTamboClient);
+        mockClient.mockReturnValue({
+            ...mockTamboAI,
+            beta: {
+                ...mockBeta,
+                projects: {
+                    ...mockProjects,
+                    getCurrent: jest.fn().mockResolvedValue({ id: "current-project" }),
+                },
+                threads: {
+                    ...mockThreadsApi,
+                    list: jest.fn().mockImplementation(async () => {
+                        // console.log("Mocking error", mockCount++);
+                        throw mockError;
+                    }),
+                },
+            },
+        });
+        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_threads_1.useTamboThreadList)({}, { retry: false }));
+        await (0, react_1.waitFor)(() => {
+            const { isLoading, error, isError } = result.current;
+            expect(isLoading).toBe(false);
+            expect(isError).toBe(true);
+            expect(error).toBe(mockError);
+        });
+    });
+});
+//# sourceMappingURL=use-tambo-threads.test.js.map

package/dist/hooks/__tests__/use-tambo-threads.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-threads.test.js","sourceRoot":"","sources":["../../../src/hooks/__tests__/use-tambo-threads.test.tsx"],"names":[],"mappings":";;AACA,uDAAoD;AACpD,kDAA6D;AAE7D,iFAG+C;AAE/C,4DAA0D;AAE1D,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC/B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,MAAM,WAAW,GAAG;QAClB,EAAE,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE;QACrC,EAAE,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE;KACtC,CAAC;IAEF,MAAM,YAAY,GAAG;QACnB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;QACrB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;QACnB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;KAC6B,CAAC;IAEjD,MAAM,cAAc,GAAG;QACrB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;QACf,QAAQ,EAAE;YACR,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;YACjB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;YACjB,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;SAChC;QACD,WAAW,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;SACpB;QACD,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;QACnB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;QAClB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;KAC2B,CAAC;IAEpD,MAAM,QAAQ,GAAG;QACf,QAAQ,EAAE,YAAY;QACtB,OAAO,EAAE,cAAc;QACvB,QAAQ,EAAE;YACR,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;SACpB;KAC+B,CAAC;IAEnC,MAAM,WAAW,GAAG;QAClB,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,EAAE;QACd,IAAI,EAAE,QAAQ;KACgC,CAAC;IAEjD,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,MAAM,CAAC,2CAAmB,CAAC,CAAC,eAAe,CAAC,IAAI,yBAAW,EAAE,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC;QAC/C,UAAU,CAAC,eAAe,CAAC;YACzB,GAAG,WAAW;YACd,IAAI,EAAE;gBACJ,GAAG,QAAQ;gBACX,QAAQ,EAAE;oBACR,GAAG,YAAY;oBACf,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC;iBACnE;gBACD,OAAO,EAAE;oBACP,GAAG,cAAc;oBACjB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC;iBAC/C;aACF;SAC8B,CAAC,CAAC;QAEnC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAkB,GAAE,CAAC,CAAC;QAE1D,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAC1D,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC;QAC/C,UAAU,CAAC,eAAe,CAAC;YACzB,GAAG,WAAW;YACd,IAAI,EAAE;gBACJ,GAAG,QAAQ;gBACX,OAAO,EAAE;oBACP,GAAG,cAAc;oBACjB,IAAI,EAAE,QAAQ;iBACf;aACF;SAC8B,CAAC,CAAC;QAEnC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,sCAAkB,EAAC,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC,CACpD,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAC1D,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC;QAC/C,UAAU,CAAC,eAAe,CAAC;YACzB,GAAG,WAAW;YACd,IAAI,EAAE;gBACJ,GAAG,QAAQ;gBACX,QAAQ,EAAE;oBACR,GAAG,YAAY;oBACf,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC;iBACnE;gBACD,OAAO,EAAE;oBACP,GAAG,cAAc;oBACjB,IAAI,EAAE,QAAQ;iBACf;aACF;SAC8B,CAAC,CAAC;QAEnC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,sCAAkB,EAAC,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC,CACnD,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE;YACvD,UAAU,EAAE,cAAc;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,IAAI,cAAoC,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACtC,cAAc,GAAG,OAAO,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC;QAC/C,UAAU,CAAC,eAAe,CAAC;YACzB,GAAG,WAAW;YACd,IAAI,EAAE;gBACJ,GAAG,QAAQ;gBACX,QAAQ,EAAE;oBACR,GAAG,YAAY;oBACf,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC;iBACnE;gBACD,OAAO,EAAE;oBACP,GAAG,cAAc;oBACjB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,OAAO,CAAC;iBACzC;aACF;SAC8B,CAAC,CAAC;QAEnC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,sCAAkB,EAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CACzC,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA+B5C,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;QAEvC,cAAe,CAAC,WAAW,CAAC,CAAC;QAC7B,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QACH,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACvD,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC;QAC/C,UAAU,CAAC,eAAe,CAAC;YACzB,GAAG,WAAW;YACd,IAAI,EAAE;gBACJ,GAAG,QAAQ;gBACX,QAAQ,EAAE;oBACR,GAAG,YAAY;oBACf,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC;iBACnE;gBACD,OAAO,EAAE;oBACP,GAAG,cAAc;oBACjB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;wBAC5C,6CAA6C;wBAC7C,MAAM,SAAS,CAAC;oBAClB,CAAC,CAAC;iBACH;aACF;SAC8B,CAAC,CAAC;QAEnC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,sCAAkB,EAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CACzC,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC,OAAO,CAAC;YACrD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3B,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChC,CAAC,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 { renderHook, waitFor } from \"@testing-library/react\";\nimport { DeepPartial } from \"ts-essentials\";\nimport {\n  useTamboClient,\n  useTamboQueryClient,\n} from \"../../providers/tambo-client-provider\";\nimport { PartialTamboAI } from \"../../testing/types\";\nimport { useTamboThreadList } from \"../use-tambo-threads\";\n\njest.mock(\"../../providers/tambo-client-provider\", () => ({\n  useTamboClient: jest.fn(),\n  useTamboQueryClient: jest.fn(),\n}));\n\ndescribe(\"useTamboThreadList\", () => {\n  const mockThreads = [\n    { id: \"thread-1\", title: \"Thread 1\" },\n    { id: \"thread-2\", title: \"Thread 2\" },\n  ];\n\n  const mockProjects = {\n    getCurrent: jest.fn(),\n    retrieve: jest.fn(),\n    delete: jest.fn(),\n  } satisfies Partial<TamboAI[\"beta\"][\"projects\"]>;\n\n  const mockThreadsApi = {\n    list: jest.fn(),\n    messages: {\n      list: jest.fn(),\n      create: jest.fn(),\n      delete: jest.fn(),\n      updateComponentState: jest.fn(),\n    },\n    suggestions: {\n      list: jest.fn(),\n      generate: jest.fn(),\n    },\n    create: jest.fn(),\n    retrieve: jest.fn(),\n    update: jest.fn(),\n    delete: jest.fn(),\n    advance: jest.fn(),\n    advanceById: jest.fn(),\n  } satisfies DeepPartial<TamboAI[\"beta\"][\"threads\"]>;\n\n  const mockBeta = {\n    projects: mockProjects,\n    threads: mockThreadsApi,\n    registry: {\n      retrieve: jest.fn(),\n    },\n  } satisfies PartialTamboAI[\"beta\"];\n\n  const mockTamboAI = {\n    apiKey: \"\",\n    components: {},\n    beta: mockBeta,\n  } satisfies PartialTamboAI as unknown as TamboAI;\n\n  beforeEach(() => {\n    jest.mocked(useTamboQueryClient).mockReturnValue(new QueryClient());\n  });\n\n  it(\"should fetch threads for current project when no projectId is provided\", async () => {\n    const mockClient = jest.mocked(useTamboClient);\n    mockClient.mockReturnValue({\n      ...mockTamboAI,\n      beta: {\n        ...mockBeta,\n        projects: {\n          ...mockProjects,\n          getCurrent: jest.fn().mockResolvedValue({ id: \"current-project\" }),\n        },\n        threads: {\n          ...mockThreadsApi,\n          list: jest.fn().mockResolvedValue(mockThreads),\n        },\n      },\n    } satisfies PartialTamboAI as any);\n\n    const { result } = renderHook(() => useTamboThreadList());\n\n    await waitFor(() => {\n      expect(result.current.data).toEqual(mockThreads);\n    });\n  });\n\n  it(\"should fetch threads for specified projectId\", async () => {\n    const mockList = jest.fn().mockResolvedValue(mockThreads);\n    const mockClient = jest.mocked(useTamboClient);\n    mockClient.mockReturnValue({\n      ...mockTamboAI,\n      beta: {\n        ...mockBeta,\n        threads: {\n          ...mockThreadsApi,\n          list: mockList,\n        },\n      },\n    } satisfies PartialTamboAI as any);\n\n    const { result } = renderHook(() =>\n      useTamboThreadList({ projectId: \"custom-project\" }),\n    );\n\n    await waitFor(() => {\n      expect(result.current.data).toEqual(mockThreads);\n    });\n\n    expect(mockList).toHaveBeenCalledWith(\"custom-project\", {});\n  });\n\n  it(\"should fetch threads with contextKey when provided\", async () => {\n    const mockList = jest.fn().mockResolvedValue(mockThreads);\n    const mockClient = jest.mocked(useTamboClient);\n    mockClient.mockReturnValue({\n      ...mockTamboAI,\n      beta: {\n        ...mockBeta,\n        projects: {\n          ...mockProjects,\n          getCurrent: jest.fn().mockResolvedValue({ id: \"current-project\" }),\n        },\n        threads: {\n          ...mockThreadsApi,\n          list: mockList,\n        },\n      },\n    } satisfies PartialTamboAI as any);\n\n    const { result } = renderHook(() =>\n      useTamboThreadList({ contextKey: \"test-context\" }),\n    );\n\n    await waitFor(() => {\n      expect(result.current.data).toEqual(mockThreads);\n    });\n\n    expect(mockList).toHaveBeenCalledWith(\"current-project\", {\n      contextKey: \"test-context\",\n    });\n  });\n\n  it(\"should handle loading state\", async () => {\n    let resolvePromise: (value: any) => void;\n    const promise = new Promise((resolve) => {\n      resolvePromise = resolve;\n    });\n\n    const mockClient = jest.mocked(useTamboClient);\n    mockClient.mockReturnValue({\n      ...mockTamboAI,\n      beta: {\n        ...mockBeta,\n        projects: {\n          ...mockProjects,\n          getCurrent: jest.fn().mockResolvedValue({ id: \"current-project\" }),\n        },\n        threads: {\n          ...mockThreadsApi,\n          list: jest.fn().mockReturnValue(promise),\n        },\n      },\n    } satisfies PartialTamboAI as any);\n\n    const { result } = renderHook(() =>\n      useTamboThreadList({}, { retry: false }),\n    );\n\n    expect(result.current).toMatchInlineSnapshot(`\n      {\n        \"data\": null,\n        \"dataUpdatedAt\": 0,\n        \"error\": null,\n        \"errorUpdateCount\": 0,\n        \"errorUpdatedAt\": 0,\n        \"failureCount\": 0,\n        \"failureReason\": null,\n        \"fetchStatus\": \"fetching\",\n        \"isError\": false,\n        \"isFetched\": false,\n        \"isFetchedAfterMount\": false,\n        \"isFetching\": true,\n        \"isInitialLoading\": true,\n        \"isLoading\": true,\n        \"isLoadingError\": false,\n        \"isPaused\": false,\n        \"isPending\": true,\n        \"isPlaceholderData\": false,\n        \"isRefetchError\": false,\n        \"isRefetching\": false,\n        \"isStale\": true,\n        \"isSuccess\": false,\n        \"promise\": Promise {\n          \"reason\": [Error: experimental_prefetchInRender feature flag is not enabled],\n          \"status\": \"rejected\",\n        },\n        \"refetch\": [Function],\n        \"status\": \"pending\",\n      }\n    `);\n    expect(result.current.isLoading).toBe(true);\n    expect(result.current.data).toBeNull();\n\n    resolvePromise!(mockThreads);\n    await waitFor(() => {\n      expect(result.current.isLoading).toBe(false);\n    });\n    await waitFor(() => {\n      expect(result.current.isLoading).toBe(false);\n    });\n  });\n\n  it(\"should handle error state\", async () => {\n    const mockError = new Error(\"Failed to fetch threads\");\n    const mockClient = jest.mocked(useTamboClient);\n    mockClient.mockReturnValue({\n      ...mockTamboAI,\n      beta: {\n        ...mockBeta,\n        projects: {\n          ...mockProjects,\n          getCurrent: jest.fn().mockResolvedValue({ id: \"current-project\" }),\n        },\n        threads: {\n          ...mockThreadsApi,\n          list: jest.fn().mockImplementation(async () => {\n            // console.log(\"Mocking error\", mockCount++);\n            throw mockError;\n          }),\n        },\n      },\n    } satisfies PartialTamboAI as any);\n\n    const { result } = renderHook(() =>\n      useTamboThreadList({}, { retry: false }),\n    );\n\n    await waitFor(() => {\n      const { isLoading, error, isError } = result.current;\n      expect(isLoading).toBe(false);\n      expect(isError).toBe(true);\n      expect(error).toBe(mockError);\n    });\n  });\n});\n"]}

package/dist/hooks/react-query-hooks.d.ts

@@ -19,14 +19,6 @@ export declare function useTamboMutation<TData = unknown, TError = Error, TVaria
  * Type alias for the result of a mutation.
  */
 export type UseTamboMutationResult<TData = unknown, TError = Error, TVariables = void, TContext = unknown> = UseMutationResult<TData, TError, TVariables, TContext>;
-/**
- * Hook for creating a mutation with the tambo query client.
- *
- * Use this instead of useMutation from @tanstack/react-query
- * @param options - The options for the mutation, same as useMutation from @tanstack/react-query
- * @returns The mutation result
- */
-export declare function useTamboMutationResult<TData = unknown, TError = Error, TVariables = void, TContext = unknown>(options: UseMutationOptions<TData, TError, TVariables, TContext>): UseMutationResult<TData, TError, TVariables, TContext>;
 /**
  * Type alias for the result of a query.
  */

package/dist/hooks/react-query-hooks.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"react-query-hooks.d.ts","sourceRoot":"","sources":["../../src/hooks/react-query-hooks.ts"],"names":[],"mappings":"AACA,OAAO,EACL,QAAQ,EAER,kBAAkB,EAClB,iBAAiB,EAEjB,eAAe,EACf,cAAc,EACf,MAAM,uBAAuB,CAAC;AAG/B;;;;;;GAMG;AACH,wBAAgB,aAAa,CAC3B,YAAY,GAAG,OAAO,EACtB,MAAM,GAAG,KAAK,EACd,KAAK,GAAG,YAAY,EACpB,SAAS,SAAS,QAAQ,GAAG,QAAQ,EACrC,OAAO,EAAE,eAAe,CAAC,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,CAAC,iCAGjE;AAED;;;;;;GAMG;AACH,wBAAgB,gBAAgB,CAC9B,KAAK,GAAG,OAAO,EACf,MAAM,GAAG,KAAK,EACd,UAAU,GAAG,IAAI,EACjB,QAAQ,GAAG,OAAO,EAClB,OAAO,EAAE,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,CAAC,0DAGjE;AAED;;GAEG;AACH,MAAM,MAAM,sBAAsB,CAChC,KAAK,GAAG,OAAO,EACf,MAAM,GAAG,KAAK,EACd,UAAU,GAAG,IAAI,EACjB,QAAQ,GAAG,OAAO,IAChB,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;AAE3D;;;;;;GAMG;AACH,wBAAgB,sBAAsB,CACpC,KAAK,GAAG,OAAO,EACf,MAAM,GAAG,KAAK,EACd,UAAU,GAAG,IAAI,EACjB,QAAQ,GAAG,OAAO,EAClB,OAAO,EAAE,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,CAAC,0DAEjE;AAED;;GAEG;AACH,MAAM,MAAM,mBAAmB,CAC7B,KAAK,GAAG,OAAO,EACf,MAAM,GAAG,KAAK,IACZ,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC"}
+{"version":3,"file":"react-query-hooks.d.ts","sourceRoot":"","sources":["../../src/hooks/react-query-hooks.ts"],"names":[],"mappings":"AACA,OAAO,EACL,QAAQ,EAER,kBAAkB,EAClB,iBAAiB,EAEjB,eAAe,EACf,cAAc,EACf,MAAM,uBAAuB,CAAC;AAG/B;;;;;;GAMG;AACH,wBAAgB,aAAa,CAC3B,YAAY,GAAG,OAAO,EACtB,MAAM,GAAG,KAAK,EACd,KAAK,GAAG,YAAY,EACpB,SAAS,SAAS,QAAQ,GAAG,QAAQ,EACrC,OAAO,EAAE,eAAe,CAAC,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,CAAC,iCAGjE;AAED;;;;;;GAMG;AACH,wBAAgB,gBAAgB,CAC9B,KAAK,GAAG,OAAO,EACf,MAAM,GAAG,KAAK,EACd,UAAU,GAAG,IAAI,EACjB,QAAQ,GAAG,OAAO,EAClB,OAAO,EAAE,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,CAAC,0DAGjE;AAED;;GAEG;AACH,MAAM,MAAM,sBAAsB,CAChC,KAAK,GAAG,OAAO,EACf,MAAM,GAAG,KAAK,EACd,UAAU,GAAG,IAAI,EACjB,QAAQ,GAAG,OAAO,IAChB,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;AAE3D;;GAEG;AACH,MAAM,MAAM,mBAAmB,CAC7B,KAAK,GAAG,OAAO,EACf,MAAM,GAAG,KAAK,IACZ,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC"}

package/dist/hooks/react-query-hooks.js

@@ -2,7 +2,6 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.useTamboQuery = useTamboQuery;
 exports.useTamboMutation = useTamboMutation;
-exports.useTamboMutationResult = useTamboMutationResult;
 // tamboHooks.ts
 const react_query_1 = require("@tanstack/react-query");
 const tambo_client_provider_1 = require("../providers/tambo-client-provider");
@@ -28,14 +27,4 @@ function useTamboMutation(options) {
     const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
     return (0, react_query_1.useMutation)(options, queryClient);
 }
-/**
- * Hook for creating a mutation with the tambo query client.
- *
- * Use this instead of useMutation from @tanstack/react-query
- * @param options - The options for the mutation, same as useMutation from @tanstack/react-query
- * @returns The mutation result
- */
-function useTamboMutationResult(options) {
-    return useTamboMutation(options);
-}
 //# sourceMappingURL=react-query-hooks.js.map

package/dist/hooks/react-query-hooks.js.map

@@ -1 +1 @@
-{"version":3,"file":"react-query-hooks.js","sourceRoot":"","sources":["../../src/hooks/react-query-hooks.ts"],"names":[],"mappings":";;AAmBA,sCAQC;AASD,4CAQC;AAmBD,wDAOC;AAtED,gBAAgB;AAChB,uDAQ+B;AAC/B,8EAAyE;AAEzE;;;;;;GAMG;AACH,SAAgB,aAAa,CAK3B,OAAgE;IAChE,MAAM,WAAW,GAAG,IAAA,2CAAmB,GAAE,CAAC;IAC1C,OAAO,IAAA,sBAAQ,EAAC,OAAO,EAAE,WAAW,CAAC,CAAC;AACxC,CAAC;AAED;;;;;;GAMG;AACH,SAAgB,gBAAgB,CAK9B,OAAgE;IAChE,MAAM,WAAW,GAAG,IAAA,2CAAmB,GAAE,CAAC;IAC1C,OAAO,IAAA,yBAAW,EAAC,OAAO,EAAE,WAAW,CAAC,CAAC;AAC3C,CAAC;AAYD;;;;;;GAMG;AACH,SAAgB,sBAAsB,CAKpC,OAAgE;IAChE,OAAO,gBAAgB,CAAsC,OAAO,CAAC,CAAC;AACxE,CAAC","sourcesContent":["// tamboHooks.ts\nimport {\n  QueryKey,\n  useMutation,\n  UseMutationOptions,\n  UseMutationResult,\n  useQuery,\n  UseQueryOptions,\n  UseQueryResult,\n} from \"@tanstack/react-query\";\nimport { useTamboQueryClient } from \"../providers/tambo-client-provider\";\n\n/**\n * Wrapper around useQuery that uses the internal tambo query client.\n *\n * Use this instead of useQuery from @tanstack/react-query\n * @param options - The options for the query, same as useQuery from @tanstack/react-query\n * @returns The query result\n */\nexport function useTamboQuery<\n  TQueryFnData = unknown,\n  TError = Error,\n  TData = TQueryFnData,\n  TQueryKey extends QueryKey = QueryKey,\n>(options: UseQueryOptions<TQueryFnData, TError, TData, TQueryKey>) {\n  const queryClient = useTamboQueryClient();\n  return useQuery(options, queryClient);\n}\n\n/**\n * Wrapper around useMutation that uses the internal tambo query client.\n *\n * Use this instead of useMutation from @tanstack/react-query\n * @param options - The options for the mutation, same as useMutation from @tanstack/react-query\n * @returns The mutation result\n */\nexport function useTamboMutation<\n  TData = unknown,\n  TError = Error,\n  TVariables = void,\n  TContext = unknown,\n>(options: UseMutationOptions<TData, TError, TVariables, TContext>) {\n  const queryClient = useTamboQueryClient();\n  return useMutation(options, queryClient);\n}\n\n/**\n * Type alias for the result of a mutation.\n */\nexport type UseTamboMutationResult<\n  TData = unknown,\n  TError = Error,\n  TVariables = void,\n  TContext = unknown,\n> = UseMutationResult<TData, TError, TVariables, TContext>;\n\n/**\n * Hook for creating a mutation with the tambo query client.\n *\n * Use this instead of useMutation from @tanstack/react-query\n * @param options - The options for the mutation, same as useMutation from @tanstack/react-query\n * @returns The mutation result\n */\nexport function useTamboMutationResult<\n  TData = unknown,\n  TError = Error,\n  TVariables = void,\n  TContext = unknown,\n>(options: UseMutationOptions<TData, TError, TVariables, TContext>) {\n  return useTamboMutation<TData, TError, TVariables, TContext>(options);\n}\n\n/**\n * Type alias for the result of a query.\n */\nexport type UseTamboQueryResult<\n  TData = unknown,\n  TError = Error,\n> = UseQueryResult<TData, TError>;\n"]}
+{"version":3,"file":"react-query-hooks.js","sourceRoot":"","sources":["../../src/hooks/react-query-hooks.ts"],"names":[],"mappings":";;AAmBA,sCAQC;AASD,4CAQC;AA5CD,gBAAgB;AAChB,uDAQ+B;AAC/B,8EAAyE;AAEzE;;;;;;GAMG;AACH,SAAgB,aAAa,CAK3B,OAAgE;IAChE,MAAM,WAAW,GAAG,IAAA,2CAAmB,GAAE,CAAC;IAC1C,OAAO,IAAA,sBAAQ,EAAC,OAAO,EAAE,WAAW,CAAC,CAAC;AACxC,CAAC;AAED;;;;;;GAMG;AACH,SAAgB,gBAAgB,CAK9B,OAAgE;IAChE,MAAM,WAAW,GAAG,IAAA,2CAAmB,GAAE,CAAC;IAC1C,OAAO,IAAA,yBAAW,EAAC,OAAO,EAAE,WAAW,CAAC,CAAC;AAC3C,CAAC","sourcesContent":["// tamboHooks.ts\nimport {\n  QueryKey,\n  useMutation,\n  UseMutationOptions,\n  UseMutationResult,\n  useQuery,\n  UseQueryOptions,\n  UseQueryResult,\n} from \"@tanstack/react-query\";\nimport { useTamboQueryClient } from \"../providers/tambo-client-provider\";\n\n/**\n * Wrapper around useQuery that uses the internal tambo query client.\n *\n * Use this instead of useQuery from @tanstack/react-query\n * @param options - The options for the query, same as useQuery from @tanstack/react-query\n * @returns The query result\n */\nexport function useTamboQuery<\n  TQueryFnData = unknown,\n  TError = Error,\n  TData = TQueryFnData,\n  TQueryKey extends QueryKey = QueryKey,\n>(options: UseQueryOptions<TQueryFnData, TError, TData, TQueryKey>) {\n  const queryClient = useTamboQueryClient();\n  return useQuery(options, queryClient);\n}\n\n/**\n * Wrapper around useMutation that uses the internal tambo query client.\n *\n * Use this instead of useMutation from @tanstack/react-query\n * @param options - The options for the mutation, same as useMutation from @tanstack/react-query\n * @returns The mutation result\n */\nexport function useTamboMutation<\n  TData = unknown,\n  TError = Error,\n  TVariables = void,\n  TContext = unknown,\n>(options: UseMutationOptions<TData, TError, TVariables, TContext>) {\n  const queryClient = useTamboQueryClient();\n  return useMutation(options, queryClient);\n}\n\n/**\n * Type alias for the result of a mutation.\n */\nexport type UseTamboMutationResult<\n  TData = unknown,\n  TError = Error,\n  TVariables = void,\n  TContext = unknown,\n> = UseMutationResult<TData, TError, TVariables, TContext>;\n\n/**\n * Type alias for the result of a query.\n */\nexport type UseTamboQueryResult<\n  TData = unknown,\n  TError = Error,\n> = UseQueryResult<TData, TError>;\n"]}

package/dist/hooks/use-suggestions.js

@@ -56,7 +56,7 @@ function useTamboSuggestions(options = {}) {
         retry: false,
     });
     // Accept suggestion mutation
-    const acceptMutationState = (0, react_query_hooks_1.useTamboMutationResult)({
+    const acceptMutationState = (0, react_query_hooks_1.useTamboMutation)({
         mutationFn: async ({ suggestion, shouldSubmit = false }) => {
             const validation = (0, validate_input_1.validateInput)(suggestion.detailedSuggestion);
             if (!validation.isValid) {
@@ -77,7 +77,7 @@ function useTamboSuggestions(options = {}) {
         },
     });
     // Generate suggestions mutation
-    const generateMutationState = (0, react_query_hooks_1.useTamboMutationResult)({
+    const generateMutationState = (0, react_query_hooks_1.useTamboMutation)({
         mutationFn: async (abortController) => {
             if (!latestMessageId || !isLatestFromTambo) {
                 return undefined;

package/dist/hooks/use-suggestions.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-suggestions.js","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":";;AA2EA,kDAqIC;AA/MD,iCAA4C;AAC5C,4DAAwD;AACxD,8EAAoE;AACpE,gEAAuD;AACvD,kFAAwE;AACxE,8EAAoE;AACpE,qDAG6B;AAC7B,+CAA0D;AAC1D,2DAK6B;AAC7B,yDAA+E;AAoD/E;;;;GAIG;AACH,SAAgB,mBAAmB,CACjC,UAAsC,EAAE;IAExC,MAAM,EAAE,cAAc,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IACvC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,sCAAc,GAAE,CAAC;IACpC,MAAM,EAAE,iBAAiB,EAAE,GAAG,IAAA,yBAAQ,GAAE,CAAC;IACzC,MAAM,WAAW,GAAG,IAAA,sCAAc,GAAE,CAAC;IACrC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,IAAA,0CAAgB,GAAE,CAAC;IAErB,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,IAAA,gBAAQ,EAE9D,IAAI,CAAC,CAAC;IACR,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,IAAA,sCAAmB,GAAE,CAAC;IAE1D,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,iBAAiB,GAAG,aAAa,EAAE,IAAI,KAAK,WAAW,CAAC;IAC9D,MAAM,eAAe,GAAG,aAAa,EAAE,EAAE,CAAC;IAE1C,qDAAqD;IACrD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;IAEtB,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,IAAA,iCAAa,EAAC;QACtC,4EAA4E;QAC5E,QAAQ,EAAE,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC;QACrE,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,eAAe,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3C,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,6DAA6D;gBAC7D,mBAAmB,EAAE,CAAC,UAAU,CAAC;aAClC,CACF,CAAC;QACJ,CAAC;QACD,2DAA2D;QAC3D,OAAO,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB,CAAC;QACtD,6CAA6C;QAC7C,oBAAoB,EAAE,KAAK;QAC3B,kBAAkB,EAAE,KAAK;QACzB,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,mBAAmB,GAAG,IAAA,0CAAsB,EAIhD;QACA,UAAU,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,YAAY,GAAG,KAAK,EAAE,EAAE,EAAE;YACzD,MAAM,UAAU,GAAG,IAAA,8BAAa,EAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;YAChE,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBACxB,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,MAAM,UAAU,CAAC,KAAK,CAAC;gBACzB,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,uCAAoB,CAAC,UAAU,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,iBAAiB,CAAC,UAAU,CAAC,cAAc,EAAE;oBACjD,QAAQ,EAAE,MAAM,CAAC,EAAE;iBACpB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YAC3C,CAAC;YACD,uBAAuB,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACzC,CAAC;KACF,CAAC,CAAC;IAEH,gCAAgC;IAChC,MAAM,qBAAqB,GAAG,IAAA,0CAAsB,EAIlD;QACA,UAAU,EAAE,KAAK,EAAE,eAAgC,EAAE,EAAE;YACrD,IAAI,CAAC,eAAe,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3C,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,6DAA6D;gBAC7D,mBAAmB,EAAE,CAAC,UAAU,CAAC;aAClC,EACD,EAAE,MAAM,EAAE,eAAe,CAAC,MAAM,EAAE,CACnC,CAAC;QACJ,CAAC;QACD,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,mEAAmE;IACnE,8DAA8D;IAC9D,MAAM,WAAW,GAAG,iBAAiB;QACnC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,IAAI,qBAAqB,CAAC,IAAI,IAAI,EAAE,CAAC;QAC9D,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO;QACL,WAAW;QACX,MAAM,EAAE,mBAAmB,CAAC,WAAW;QACvC,oBAAoB;QACpB,YAAY,EAAE,mBAAmB;QACjC,cAAc,EAAE,qBAAqB;QACrC,iBAAiB;QACjB,GAAG,IAAA,oCAAsB,EAAC,mBAAmB,EAAE,qBAAqB,CAAC;KACtE,CAAC;AACJ,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { useEffect, useState } from \"react\";\nimport { validateInput } from \"../model/validate-input\";\nimport { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTambo } from \"../providers/tambo-provider\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport { useTamboThread } from \"../providers/tambo-thread-provider\";\nimport {\n  CombinedMutationResult,\n  combineMutationResults,\n} from \"../util/query-utils\";\nimport { getAvailableComponents } from \"../util/registry\";\nimport {\n  UseTamboMutationResult,\n  UseTamboQueryResult,\n  useTamboMutationResult,\n  useTamboQuery,\n} from \"./react-query-hooks\";\nimport { INPUT_ERROR_MESSAGES, useTamboThreadInput } from \"./use-thread-input\";\n\n/**\n * Configuration options for the useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsOptions {\n  /** Maximum number of suggestions to generate (1-10, default 3) */\n  maxSuggestions?: number;\n}\n\n/**\n * Return value interface for useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsResultInternal {\n  /** List of available suggestions (also available in generateResult.data) */\n  suggestions: TamboAI.Beta.Threads.Suggestion[];\n  /** ID of the currently selected suggestion */\n  selectedSuggestionId: string | null;\n  /**\n   * Accept and apply a suggestion (also available in acceptResult.mutateAsync)\n   * @param suggestion - The suggestion to accept\n   * @param shouldSubmit - Whether to automatically submit after accepting (default: false)\n   */\n  accept: (acceptOptions: {\n    suggestion: TamboAI.Beta.Threads.Suggestion;\n    shouldSubmit?: boolean;\n  }) => Promise<void>;\n\n  /** Result and network state for accepting a suggestion */\n  acceptResult: UseTamboMutationResult<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >;\n\n  /** Result and network state for generating suggestions */\n  generateResult: UseTamboMutationResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >;\n\n  /** The full suggestions query object from React Query */\n  suggestionsResult: UseTamboQueryResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error\n  >;\n}\n\ntype useTamboSuggestionsResult = CombinedMutationResult<any, Error> &\n  useTamboSuggestionsResultInternal;\n\n/**\n * Hook for managing Tambo AI suggestions in a thread\n * @param options - Configuration options for suggestion generation\n * @returns Object containing suggestions state and control functions\n */\nexport function useTamboSuggestions(\n  options: useTamboSuggestionsOptions = {},\n): useTamboSuggestionsResult {\n  const { maxSuggestions = 3 } = options;\n  const { thread } = useTamboThread();\n  const { sendThreadMessage } = useTambo();\n  const tamboClient = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n\n  const [selectedSuggestionId, setSelectedSuggestionId] = useState<\n    string | null\n  >(null);\n  const { setValue: setInputValue } = useTamboThreadInput();\n\n  const latestMessage = thread.messages[thread.messages.length - 1];\n  const isLatestFromTambo = latestMessage?.role === \"assistant\";\n  const latestMessageId = latestMessage?.id;\n\n  // Reset selected suggestion when the message changes\n  useEffect(() => {\n    setSelectedSuggestionId(null);\n  }, [latestMessageId]);\n\n  // Use React Query to fetch suggestions when a new hydra message is received\n  const suggestionsResult = useTamboQuery({\n    // Only include latestMessageId in the queryKey if the message is from hydra\n    queryKey: [\"suggestions\", isLatestFromTambo ? latestMessageId : null],\n    queryFn: async () => {\n      if (!latestMessageId || !isLatestFromTambo) {\n        return [];\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          // The API expects an array of arrays for availableComponents\n          availableComponents: [components],\n        },\n      );\n    },\n    // Only run the query if we have a valid message from hydra\n    enabled: Boolean(latestMessageId && isLatestFromTambo),\n    // Don't refetch on window focus or reconnect\n    refetchOnWindowFocus: false,\n    refetchOnReconnect: false,\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Accept suggestion mutation\n  const acceptMutationState = useTamboMutationResult<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >({\n    mutationFn: async ({ suggestion, shouldSubmit = false }) => {\n      const validation = validateInput(suggestion.detailedSuggestion);\n      if (!validation.isValid) {\n        if (validation.error) {\n          throw validation.error;\n        }\n        throw new Error(INPUT_ERROR_MESSAGES.VALIDATION);\n      }\n\n      if (shouldSubmit) {\n        await sendThreadMessage(validation.sanitizedInput, {\n          threadId: thread.id,\n        });\n      } else {\n        setInputValue(validation.sanitizedInput);\n      }\n      setSelectedSuggestionId(suggestion.id);\n    },\n  });\n\n  // Generate suggestions mutation\n  const generateMutationState = useTamboMutationResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >({\n    mutationFn: async (abortController: AbortController) => {\n      if (!latestMessageId || !isLatestFromTambo) {\n        return undefined;\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          // The API expects an array of arrays for availableComponents\n          availableComponents: [components],\n        },\n        { signal: abortController.signal },\n      );\n    },\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Use the query data if available, otherwise use the mutation data\n  // Only return suggestions if the latest message is from hydra\n  const suggestions = isLatestFromTambo\n    ? (suggestionsResult.data ?? generateMutationState.data ?? [])\n    : [];\n\n  return {\n    suggestions,\n    accept: acceptMutationState.mutateAsync,\n    selectedSuggestionId,\n    acceptResult: acceptMutationState,\n    generateResult: generateMutationState,\n    suggestionsResult,\n    ...combineMutationResults(acceptMutationState, generateMutationState),\n  };\n}\n"]}
+{"version":3,"file":"use-suggestions.js","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":";;AA2EA,kDAqIC;AA/MD,iCAA4C;AAC5C,4DAAwD;AACxD,8EAAoE;AACpE,gEAAuD;AACvD,kFAAwE;AACxE,8EAAoE;AACpE,qDAG6B;AAC7B,+CAA0D;AAC1D,2DAK6B;AAC7B,yDAA+E;AAoD/E;;;;GAIG;AACH,SAAgB,mBAAmB,CACjC,UAAsC,EAAE;IAExC,MAAM,EAAE,cAAc,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IACvC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,sCAAc,GAAE,CAAC;IACpC,MAAM,EAAE,iBAAiB,EAAE,GAAG,IAAA,yBAAQ,GAAE,CAAC;IACzC,MAAM,WAAW,GAAG,IAAA,sCAAc,GAAE,CAAC;IACrC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,IAAA,0CAAgB,GAAE,CAAC;IAErB,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,IAAA,gBAAQ,EAE9D,IAAI,CAAC,CAAC;IACR,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,IAAA,sCAAmB,GAAE,CAAC;IAE1D,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,iBAAiB,GAAG,aAAa,EAAE,IAAI,KAAK,WAAW,CAAC;IAC9D,MAAM,eAAe,GAAG,aAAa,EAAE,EAAE,CAAC;IAE1C,qDAAqD;IACrD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;IAEtB,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,IAAA,iCAAa,EAAC;QACtC,4EAA4E;QAC5E,QAAQ,EAAE,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC;QACrE,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,eAAe,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3C,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,6DAA6D;gBAC7D,mBAAmB,EAAE,CAAC,UAAU,CAAC;aAClC,CACF,CAAC;QACJ,CAAC;QACD,2DAA2D;QAC3D,OAAO,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB,CAAC;QACtD,6CAA6C;QAC7C,oBAAoB,EAAE,KAAK;QAC3B,kBAAkB,EAAE,KAAK;QACzB,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,mBAAmB,GAAG,IAAA,oCAAgB,EAI1C;QACA,UAAU,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,YAAY,GAAG,KAAK,EAAE,EAAE,EAAE;YACzD,MAAM,UAAU,GAAG,IAAA,8BAAa,EAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;YAChE,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBACxB,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,MAAM,UAAU,CAAC,KAAK,CAAC;gBACzB,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,uCAAoB,CAAC,UAAU,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,iBAAiB,CAAC,UAAU,CAAC,cAAc,EAAE;oBACjD,QAAQ,EAAE,MAAM,CAAC,EAAE;iBACpB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YAC3C,CAAC;YACD,uBAAuB,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACzC,CAAC;KACF,CAAC,CAAC;IAEH,gCAAgC;IAChC,MAAM,qBAAqB,GAAG,IAAA,oCAAgB,EAI5C;QACA,UAAU,EAAE,KAAK,EAAE,eAAgC,EAAE,EAAE;YACrD,IAAI,CAAC,eAAe,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3C,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,MAAM,CAAC,EAAE,EACT,eAAe,EACf;gBACE,cAAc;gBACd,6DAA6D;gBAC7D,mBAAmB,EAAE,CAAC,UAAU,CAAC;aAClC,EACD,EAAE,MAAM,EAAE,eAAe,CAAC,MAAM,EAAE,CACnC,CAAC;QACJ,CAAC;QACD,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,mEAAmE;IACnE,8DAA8D;IAC9D,MAAM,WAAW,GAAG,iBAAiB;QACnC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,IAAI,qBAAqB,CAAC,IAAI,IAAI,EAAE,CAAC;QAC9D,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO;QACL,WAAW;QACX,MAAM,EAAE,mBAAmB,CAAC,WAAW;QACvC,oBAAoB;QACpB,YAAY,EAAE,mBAAmB;QACjC,cAAc,EAAE,qBAAqB;QACrC,iBAAiB;QACjB,GAAG,IAAA,oCAAsB,EAAC,mBAAmB,EAAE,qBAAqB,CAAC;KACtE,CAAC;AACJ,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { useEffect, useState } from \"react\";\nimport { validateInput } from \"../model/validate-input\";\nimport { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTambo } from \"../providers/tambo-provider\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport { useTamboThread } from \"../providers/tambo-thread-provider\";\nimport {\n  CombinedMutationResult,\n  combineMutationResults,\n} from \"../util/query-utils\";\nimport { getAvailableComponents } from \"../util/registry\";\nimport {\n  UseTamboMutationResult,\n  UseTamboQueryResult,\n  useTamboMutation,\n  useTamboQuery,\n} from \"./react-query-hooks\";\nimport { INPUT_ERROR_MESSAGES, useTamboThreadInput } from \"./use-thread-input\";\n\n/**\n * Configuration options for the useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsOptions {\n  /** Maximum number of suggestions to generate (1-10, default 3) */\n  maxSuggestions?: number;\n}\n\n/**\n * Return value interface for useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsResultInternal {\n  /** List of available suggestions (also available in generateResult.data) */\n  suggestions: TamboAI.Beta.Threads.Suggestion[];\n  /** ID of the currently selected suggestion */\n  selectedSuggestionId: string | null;\n  /**\n   * Accept and apply a suggestion (also available in acceptResult.mutateAsync)\n   * @param suggestion - The suggestion to accept\n   * @param shouldSubmit - Whether to automatically submit after accepting (default: false)\n   */\n  accept: (acceptOptions: {\n    suggestion: TamboAI.Beta.Threads.Suggestion;\n    shouldSubmit?: boolean;\n  }) => Promise<void>;\n\n  /** Result and network state for accepting a suggestion */\n  acceptResult: UseTamboMutationResult<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >;\n\n  /** Result and network state for generating suggestions */\n  generateResult: UseTamboMutationResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >;\n\n  /** The full suggestions query object from React Query */\n  suggestionsResult: UseTamboQueryResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error\n  >;\n}\n\ntype useTamboSuggestionsResult = CombinedMutationResult<any, Error> &\n  useTamboSuggestionsResultInternal;\n\n/**\n * Hook for managing Tambo AI suggestions in a thread\n * @param options - Configuration options for suggestion generation\n * @returns Object containing suggestions state and control functions\n */\nexport function useTamboSuggestions(\n  options: useTamboSuggestionsOptions = {},\n): useTamboSuggestionsResult {\n  const { maxSuggestions = 3 } = options;\n  const { thread } = useTamboThread();\n  const { sendThreadMessage } = useTambo();\n  const tamboClient = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n\n  const [selectedSuggestionId, setSelectedSuggestionId] = useState<\n    string | null\n  >(null);\n  const { setValue: setInputValue } = useTamboThreadInput();\n\n  const latestMessage = thread.messages[thread.messages.length - 1];\n  const isLatestFromTambo = latestMessage?.role === \"assistant\";\n  const latestMessageId = latestMessage?.id;\n\n  // Reset selected suggestion when the message changes\n  useEffect(() => {\n    setSelectedSuggestionId(null);\n  }, [latestMessageId]);\n\n  // Use React Query to fetch suggestions when a new hydra message is received\n  const suggestionsResult = useTamboQuery({\n    // Only include latestMessageId in the queryKey if the message is from hydra\n    queryKey: [\"suggestions\", isLatestFromTambo ? latestMessageId : null],\n    queryFn: async () => {\n      if (!latestMessageId || !isLatestFromTambo) {\n        return [];\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          // The API expects an array of arrays for availableComponents\n          availableComponents: [components],\n        },\n      );\n    },\n    // Only run the query if we have a valid message from hydra\n    enabled: Boolean(latestMessageId && isLatestFromTambo),\n    // Don't refetch on window focus or reconnect\n    refetchOnWindowFocus: false,\n    refetchOnReconnect: false,\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Accept suggestion mutation\n  const acceptMutationState = useTamboMutation<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >({\n    mutationFn: async ({ suggestion, shouldSubmit = false }) => {\n      const validation = validateInput(suggestion.detailedSuggestion);\n      if (!validation.isValid) {\n        if (validation.error) {\n          throw validation.error;\n        }\n        throw new Error(INPUT_ERROR_MESSAGES.VALIDATION);\n      }\n\n      if (shouldSubmit) {\n        await sendThreadMessage(validation.sanitizedInput, {\n          threadId: thread.id,\n        });\n      } else {\n        setInputValue(validation.sanitizedInput);\n      }\n      setSelectedSuggestionId(suggestion.id);\n    },\n  });\n\n  // Generate suggestions mutation\n  const generateMutationState = useTamboMutation<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >({\n    mutationFn: async (abortController: AbortController) => {\n      if (!latestMessageId || !isLatestFromTambo) {\n        return undefined;\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        thread.id,\n        latestMessageId,\n        {\n          maxSuggestions,\n          // The API expects an array of arrays for availableComponents\n          availableComponents: [components],\n        },\n        { signal: abortController.signal },\n      );\n    },\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Use the query data if available, otherwise use the mutation data\n  // Only return suggestions if the latest message is from hydra\n  const suggestions = isLatestFromTambo\n    ? (suggestionsResult.data ?? generateMutationState.data ?? [])\n    : [];\n\n  return {\n    suggestions,\n    accept: acceptMutationState.mutateAsync,\n    selectedSuggestionId,\n    acceptResult: acceptMutationState,\n    generateResult: generateMutationState,\n    suggestionsResult,\n    ...combineMutationResults(acceptMutationState, generateMutationState),\n  };\n}\n"]}