@tambo-ai/react 0.43.1 → 0.44.1

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

@@ -1,5 +1,6 @@
 import { QueryClient } from "@tanstack/react-query";
 import { act, renderHook } from "@testing-library/react";
+import { GenerationStage, } from "../../model/generate-component-response";
 import { useTamboClient, useTamboQueryClient, } from "../../providers/tambo-client-provider";
 import { useTambo } from "../../providers/tambo-provider";
 import { useTamboThread, } from "../../providers/tambo-thread-provider";
@@ -36,6 +37,42 @@ describe("useTamboSuggestions", () => {
             detailedSuggestion: "Test suggestion 2",
         },
     ];
+    // Helper function to create mock TamboThreadMessage
+    const createMockMessage = (overrides = {}) => ({
+        id: "test-message-id",
+        componentState: {},
+        content: [{ type: "text", text: "Test message" }],
+        createdAt: new Date().toISOString(),
+        role: "assistant",
+        threadId: "test-thread-id",
+        ...overrides,
+    });
+    // Helper function to create mock TamboThread
+    const createMockThread = (overrides = {}) => ({
+        id: "test-thread-id",
+        projectId: "test-project-id",
+        createdAt: new Date().toISOString(),
+        updatedAt: new Date().toISOString(),
+        messages: [],
+        ...overrides,
+    });
+    // Helper function to create mock CombinedTamboThreadContextProps
+    const createMockThreadContext = (overrides = {}) => ({
+        thread: createMockThread(),
+        switchCurrentThread: jest.fn(),
+        startNewThread: jest.fn(),
+        updateThreadName: jest.fn(),
+        generateThreadName: jest.fn(),
+        addThreadMessage: jest.fn(),
+        updateThreadMessage: jest.fn(),
+        cancel: jest.fn(),
+        streaming: false,
+        sendThreadMessage: jest.fn(),
+        generationStage: GenerationStage.IDLE,
+        generationStatusMessage: "",
+        isIdle: true,
+        ...overrides,
+    });
     beforeEach(() => {
         jest.mocked(useTamboQueryClient).mockReturnValue(new QueryClient());
         jest.mocked(useTamboMutation).mockImplementation(({ mutationFn }) => ({
@@ -51,18 +88,15 @@ describe("useTamboSuggestions", () => {
         jest.mocked(useTambo).mockReturnValue({
             sendThreadMessage: jest.fn(),
         });
-        jest.mocked(useTamboThread).mockReturnValue({
-            thread: {
-                id: "test-thread-id",
+        jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext({
+            thread: createMockThread({
                 messages: [
-                    {
-                        id: "test-message-id",
+                    createMockMessage({
                         role: "hydra",
-                        content: [{ type: "text", text: "Test message" }],
-                    },
+                    }),
                 ],
-            },
-        });
+            }),
+        }));
         jest.mocked(useTamboThreadInput).mockReturnValue({
             setValue: jest.fn(),
             value: "",
@@ -93,18 +127,15 @@ describe("useTamboSuggestions", () => {
                 },
             },
         });
-        jest.mocked(useTamboThread).mockReturnValue({
-            thread: {
-                id: "test-thread-id",
+        jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext({
+            thread: createMockThread({
                 messages: [
-                    {
-                        id: "test-message-id",
+                    createMockMessage({
                         role: "assistant",
-                        content: [{ type: "text", text: "Test message" }],
-                    },
+                    }),
                 ],
-            },
-        });
+            }),
+        }));
         // Mock the query result to return the mock suggestions
         jest.mocked(useTamboQuery).mockReturnValue({
             data: mockSuggestions,
@@ -134,18 +165,15 @@ describe("useTamboSuggestions", () => {
             },
         });
         // Mock the thread to have a non-Tambo message
-        jest.mocked(useTamboThread).mockReturnValue({
-            thread: {
-                id: "test-thread-id",
+        jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext({
+            thread: createMockThread({
                 messages: [
-                    {
-                        id: "test-message-id",
+                    createMockMessage({
                         role: "user",
-                        content: [{ type: "text", text: "Test message" }],
-                    },
+                    }),
                 ],
-            },
-        });
+            }),
+        }));
         const { result } = renderHook(() => useTamboSuggestions());
         // Wait for the effect to run
         await act(async () => {

package/esm/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,OAAO,EAAE,WAAW,EAAkB,MAAM,uBAAuB,CAAC;AACpE,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAEzD,OAAO,EACL,cAAc,EACd,mBAAmB,GACpB,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EAAqB,QAAQ,EAAE,MAAM,gCAAgC,CAAC;AAC7E,OAAO,EAEL,cAAc,GACf,MAAM,uCAAuC,CAAC;AAE/C,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAkB,MAAM,qBAAqB,CAAC;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,mBAAmB,CAAC,CAAC,eAAe,CAAC,IAAI,WAAW,EAAE,CAAC,CAAC;QACpE,IAAI,CAAC,MAAM,CAAC,gBAAgB,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,cAAc,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,QAAQ,CAAC,CAAC,eAAe,CAAC;YACpC,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;SACe,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM,CAAC,cAAc,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,mBAAmB,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,aAAa,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,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,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,cAAc,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,cAAc,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,aAAa,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,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,6BAA6B;QAC7B,MAAM,GAAG,CAAC,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,cAAc,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,cAAc,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,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,6BAA6B;QAC7B,MAAM,GAAG,CAAC,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,mBAAmB,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,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,MAAM,GAAG,CAAC,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,QAAQ,CAAC,CAAC,eAAe,CAAC;YACpC,iBAAiB,EAAE,qBAAqB;SACG,CAAC,CAAC;QAE/C,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,MAAM,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,MAAM,GAAG,CAAC,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"]}
+{"version":3,"file":"use-suggestions.test.js","sourceRoot":"","sources":["../../../src/hooks/__tests__/use-suggestions.test.tsx"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAkB,MAAM,uBAAuB,CAAC;AACpE,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EACL,eAAe,GAEhB,MAAM,yCAAyC,CAAC;AAEjD,OAAO,EACL,cAAc,EACd,mBAAmB,GACpB,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EAAqB,QAAQ,EAAE,MAAM,gCAAgC,CAAC;AAC7E,OAAO,EAEL,cAAc,GACf,MAAM,uCAAuC,CAAC;AAE/C,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAkB,MAAM,qBAAqB,CAAC;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,oDAAoD;IACpD,MAAM,iBAAiB,GAAG,CACxB,YAAyC,EAAE,EACvB,EAAE,CAAC,CAAC;QACxB,EAAE,EAAE,iBAAiB;QACrB,cAAc,EAAE,EAAE;QAClB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;QACjD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,IAAI,EAAE,WAAW;QACjB,QAAQ,EAAE,gBAAgB;QAC1B,GAAG,SAAS;KACb,CAAC,CAAC;IAEH,6CAA6C;IAC7C,MAAM,gBAAgB,GAAG,CACvB,YAAkC,EAAE,EACvB,EAAE,CAAC,CAAC;QACjB,EAAE,EAAE,gBAAgB;QACpB,SAAS,EAAE,iBAAiB;QAC5B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,QAAQ,EAAE,EAAE;QACZ,GAAG,SAAS;KACb,CAAC,CAAC;IAEH,iEAAiE;IACjE,MAAM,uBAAuB,GAAG,CAC9B,YAAsD,EAAE,EACvB,EAAE,CAAC,CAAC;QACrC,MAAM,EAAE,gBAAgB,EAAE;QAC1B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC9B,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;QACzB,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC3B,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC7B,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC3B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC9B,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,SAAS,EAAE,KAAK;QAChB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC5B,eAAe,EAAE,eAAe,CAAC,IAAI;QACrC,uBAAuB,EAAE,EAAE;QAC3B,MAAM,EAAE,IAAI;QACZ,GAAG,SAAS;KACb,CAAC,CAAC;IAEH,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC,IAAI,WAAW,EAAE,CAAC,CAAC;QACpE,IAAI,CAAC,MAAM,CAAC,gBAAgB,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,cAAc,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,QAAQ,CAAC,CAAC,eAAe,CAAC;YACpC,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;SACe,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;YACtB,MAAM,EAAE,gBAAgB,CAAC;gBACvB,QAAQ,EAAE;oBACR,iBAAiB,CAAC;wBAChB,IAAI,EAAE,OAAO;qBACd,CAAC;iBACH;aACF,CAAC;SACH,CAAC,CACH,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,mBAAmB,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,aAAa,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,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,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,cAAc,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,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;YACtB,MAAM,EAAE,gBAAgB,CAAC;gBACvB,QAAQ,EAAE;oBACR,iBAAiB,CAAC;wBAChB,IAAI,EAAE,WAAW;qBAClB,CAAC;iBACH;aACF,CAAC;SACH,CAAC,CACH,CAAC;QAEF,uDAAuD;QACvD,IAAI,CAAC,MAAM,CAAC,aAAa,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,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,6BAA6B;QAC7B,MAAM,GAAG,CAAC,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,cAAc,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,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;YACtB,MAAM,EAAE,gBAAgB,CAAC;gBACvB,QAAQ,EAAE;oBACR,iBAAiB,CAAC;wBAChB,IAAI,EAAE,MAAM;qBACb,CAAC;iBACH;aACF,CAAC;SACH,CAAC,CACH,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,6BAA6B;QAC7B,MAAM,GAAG,CAAC,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,mBAAmB,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,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,MAAM,GAAG,CAAC,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,QAAQ,CAAC,CAAC,eAAe,CAAC;YACpC,iBAAiB,EAAE,qBAAqB;SACG,CAAC,CAAC;QAE/C,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,MAAM,GAAG,CAAC,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,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,MAAM,GAAG,CAAC,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 {\n  GenerationStage,\n  TamboThreadMessage,\n} from \"../../model/generate-component-response\";\nimport { TamboThread } from \"../../model/tambo-thread\";\nimport {\n  useTamboClient,\n  useTamboQueryClient,\n} from \"../../providers/tambo-client-provider\";\nimport { TamboContextProps, useTambo } from \"../../providers/tambo-provider\";\nimport {\n  CombinedTamboThreadContextProps,\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  // Helper function to create mock TamboThreadMessage\n  const createMockMessage = (\n    overrides: Partial<TamboThreadMessage> = {},\n  ): TamboThreadMessage => ({\n    id: \"test-message-id\",\n    componentState: {},\n    content: [{ type: \"text\", text: \"Test message\" }],\n    createdAt: new Date().toISOString(),\n    role: \"assistant\",\n    threadId: \"test-thread-id\",\n    ...overrides,\n  });\n\n  // Helper function to create mock TamboThread\n  const createMockThread = (\n    overrides: Partial<TamboThread> = {},\n  ): TamboThread => ({\n    id: \"test-thread-id\",\n    projectId: \"test-project-id\",\n    createdAt: new Date().toISOString(),\n    updatedAt: new Date().toISOString(),\n    messages: [],\n    ...overrides,\n  });\n\n  // Helper function to create mock CombinedTamboThreadContextProps\n  const createMockThreadContext = (\n    overrides: Partial<CombinedTamboThreadContextProps> = {},\n  ): CombinedTamboThreadContextProps => ({\n    thread: createMockThread(),\n    switchCurrentThread: jest.fn(),\n    startNewThread: jest.fn(),\n    updateThreadName: jest.fn(),\n    generateThreadName: jest.fn(),\n    addThreadMessage: jest.fn(),\n    updateThreadMessage: jest.fn(),\n    cancel: jest.fn(),\n    streaming: false,\n    sendThreadMessage: jest.fn(),\n    generationStage: GenerationStage.IDLE,\n    generationStatusMessage: \"\",\n    isIdle: true,\n    ...overrides,\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      createMockThreadContext({\n        thread: createMockThread({\n          messages: [\n            createMockMessage({\n              role: \"hydra\",\n            }),\n          ],\n        }),\n      }),\n    );\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      createMockThreadContext({\n        thread: createMockThread({\n          messages: [\n            createMockMessage({\n              role: \"assistant\",\n            }),\n          ],\n        }),\n      }),\n    );\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      createMockThreadContext({\n        thread: createMockThread({\n          messages: [\n            createMockMessage({\n              role: \"user\",\n            }),\n          ],\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    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/esm/hooks/__tests__/use-tambo-stream-status.test.js

@@ -1,15 +1,16 @@
 import { renderHook } from "@testing-library/react";
 import { GenerationStage, } from "../../model/generate-component-response";
-import { useTamboThread, } from "../../providers/tambo-thread-provider";
-import { useTamboCurrentMessage } from "../use-current-message";
 import { useTamboStreamStatus } from "../use-tambo-stream-status";
 // Mock the required providers
 jest.mock("../../providers/tambo-thread-provider", () => ({
-    useTamboThread: jest.fn(),
+    useTamboGenerationStage: jest.fn(),
 }));
 jest.mock("../use-current-message", () => ({
     useTamboCurrentMessage: jest.fn(),
 }));
+// Import the mocked functions
+import { useTamboGenerationStage } from "../../providers/tambo-thread-provider";
+import { useTamboCurrentMessage } from "../use-current-message";
 // Mock window for SSR tests
 const originalWindow = global.window;
 // Helper function to create mock ComponentDecisionV2
@@ -30,15 +31,20 @@ const createMockMessage = (overrides = {}) => ({
     threadId: "test-thread",
     ...overrides,
 });
+// Get the mocked functions
+const mockUseTamboGenerationStage = jest.mocked(useTamboGenerationStage);
+const mockUseTamboCurrentMessage = jest.mocked(useTamboCurrentMessage);
 describe("useTamboStreamStatus", () => {
     beforeEach(() => {
         // Restore window for client-side tests
         global.window = originalWindow;
         // Default mock implementations
-        jest.mocked(useTamboThread).mockReturnValue({
+        mockUseTamboGenerationStage.mockReturnValue({
             generationStage: GenerationStage.IDLE,
+            generationStatusMessage: "",
+            isIdle: true,
         });
-        jest.mocked(useTamboCurrentMessage).mockReturnValue({
+        mockUseTamboCurrentMessage.mockReturnValue({
             id: "test-message",
             component: {
                 props: {},
@@ -50,10 +56,12 @@ describe("useTamboStreamStatus", () => {
     });
     describe("Initial State", () => {
         it("should start with all flags as pending when idle and no props", () => {
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.IDLE,
+                generationStatusMessage: "",
+                isIdle: true,
             });
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({ title: "", body: "" }),
             }));
             const { result } = renderHook(() => useTamboStreamStatus());
@@ -80,10 +88,12 @@ describe("useTamboStreamStatus", () => {
     });
     describe("Generation vs Props Streaming", () => {
         it("should show generation streaming but props still pending when STREAMING_RESPONSE with no prop content", () => {
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.STREAMING_RESPONSE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({ title: "", body: "" }),
             }));
             const { result } = renderHook(() => useTamboStreamStatus());
@@ -97,10 +107,12 @@ describe("useTamboStreamStatus", () => {
             expect(result.current.propStatus.body.isStreaming).toBe(false);
         });
         it("should show prop streaming when props receive content during STREAMING_RESPONSE", () => {
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.STREAMING_RESPONSE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({ title: "Hello", body: "" }),
             }));
             const { result } = renderHook(() => useTamboStreamStatus());
@@ -117,10 +129,12 @@ describe("useTamboStreamStatus", () => {
     describe("Boolean Lifecycle", () => {
         it("should transition through Init -> Streaming -> Success lifecycle", () => {
             // Start with IDLE (Init phase)
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.IDLE,
+                generationStatusMessage: "",
+                isIdle: true,
             });
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({ title: "", body: "" }),
             }));
             const { result, rerender } = renderHook(() => useTamboStreamStatus());
@@ -129,10 +143,12 @@ describe("useTamboStreamStatus", () => {
             expect(result.current.streamStatus.isStreaming).toBe(false);
             expect(result.current.streamStatus.isSuccess).toBe(false);
             // Phase 2: Streaming - move to STREAMING_RESPONSE with content
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.STREAMING_RESPONSE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({
                     title: "Hello World",
                     body: "Some content",
@@ -143,8 +159,10 @@ describe("useTamboStreamStatus", () => {
             expect(result.current.streamStatus.isStreaming).toBe(true);
             expect(result.current.streamStatus.isSuccess).toBe(false);
             // Phase 3: Complete - move to COMPLETE
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.COMPLETE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
             rerender();
             expect(result.current.streamStatus.isPending).toBe(false);
@@ -152,11 +170,13 @@ describe("useTamboStreamStatus", () => {
             expect(result.current.streamStatus.isSuccess).toBe(true);
         });
         it("should handle error state correctly", () => {
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.ERROR,
+                generationStatusMessage: "",
+                isIdle: false,
             });
             const errorMessage = "Generation failed";
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({ title: "", body: "" }),
                 error: errorMessage,
             }));
@@ -170,10 +190,12 @@ describe("useTamboStreamStatus", () => {
     });
     describe("Derivation Rules", () => {
         it("should derive isPending correctly (no generation activity AND all props pending)", () => {
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.IDLE,
+                generationStatusMessage: "",
+                isIdle: true,
             });
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({ title: "", body: "", footer: "" }),
             }));
             const { result } = renderHook(() => useTamboStreamStatus());
@@ -184,11 +206,13 @@ describe("useTamboStreamStatus", () => {
             expect(result.current.propStatus.footer.isPending).toBe(true);
         });
         it("should derive isStreaming correctly (generation streaming OR any prop streaming)", () => {
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.COMPLETE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
             // One prop still streaming
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({
                     title: "Complete Title",
                     body: "Still streaming...",
@@ -200,10 +224,12 @@ describe("useTamboStreamStatus", () => {
         });
         it("should derive isSuccess correctly (generation complete AND all props successful)", () => {
             // Step 1: Start with streaming, props empty
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.STREAMING_RESPONSE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({
                     title: "",
                     body: "",
@@ -211,7 +237,7 @@ describe("useTamboStreamStatus", () => {
             }));
             const { result, rerender } = renderHook(() => useTamboStreamStatus());
             // Step 2: Simulate streaming in title
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({
                     title: "Complete Title",
                     body: "",
@@ -219,7 +245,7 @@ describe("useTamboStreamStatus", () => {
             }));
             rerender();
             // Step 3: Simulate streaming in body
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({
                     title: "Complete Title",
                     body: "Complete Body",
@@ -227,8 +253,10 @@ describe("useTamboStreamStatus", () => {
             }));
             rerender();
             // Step 4: Generation complete
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.COMPLETE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
             rerender();
             // Now both props should be successful
@@ -237,10 +265,12 @@ describe("useTamboStreamStatus", () => {
             expect(result.current.streamStatus.isSuccess).toBe(true);
         });
         it("should derive isError correctly (generation error OR any prop error)", () => {
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.COMPLETE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({
                     title: "Good Title",
                     body: "Good Body",
@@ -254,7 +284,7 @@ describe("useTamboStreamStatus", () => {
     });
     describe("Type Safety", () => {
         it("should provide strongly typed prop status based on generic", () => {
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({
                     title: "Test",
                     description: "Test desc",
@@ -270,7 +300,7 @@ describe("useTamboStreamStatus", () => {
             // expect(result.current.propStatus.nonExistentProp).toBeDefined();
         });
         it("should work without generic type parameter", () => {
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({ dynamicProp: "value" }),
             }));
             const { result } = renderHook(() => useTamboStreamStatus());
@@ -280,13 +310,20 @@ describe("useTamboStreamStatus", () => {
     });
     describe("Edge Cases", () => {
         it("should handle missing message gracefully", () => {
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(undefined);
+            mockUseTamboCurrentMessage.mockReturnValue({
+                id: "test-message",
+                role: "user",
+                content: [],
+                componentState: {},
+                createdAt: "",
+                threadId: "",
+            });
             const { result } = renderHook(() => useTamboStreamStatus());
             expect(result.current.streamStatus.isPending).toBe(true);
             expect(result.current.propStatus).toEqual({});
         });
         it("should handle missing component props gracefully", () => {
-            jest.mocked(useTamboCurrentMessage).mockReturnValue({
+            mockUseTamboCurrentMessage.mockReturnValue({
                 id: "test-message",
                 // Missing component property
             });
@@ -296,30 +333,36 @@ describe("useTamboStreamStatus", () => {
         });
         it("should reset prop tracking when generation restarts", () => {
             // Step 1: Complete a message
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.STREAMING_RESPONSE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({ title: "" }),
             }));
             const { result, rerender } = renderHook(() => useTamboStreamStatus());
             // Simulate streaming in title
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 component: createMockComponent({ title: "First Title" }),
             }));
             rerender();
             // Complete generation
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.COMPLETE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
             rerender();
             // Should be successful initially
             expect(result.current.propStatus.title.isSuccess).toBe(true);
             // Step 2: Start new generation with a new message ID to trigger reset
-            jest.mocked(useTamboThread).mockReturnValue({
+            mockUseTamboGenerationStage.mockReturnValue({
                 generationStage: GenerationStage.CHOOSING_COMPONENT,
+                generationStatusMessage: "",
+                isIdle: false,
             });
-            jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
+            mockUseTamboCurrentMessage.mockReturnValue(createMockMessage({
                 id: "new-message-id", // Different message ID to trigger reset
                 component: createMockComponent({ title: "" }),
             }));