@tambo-ai/react 0.75.0 → 1.0.0-rc.4

package/dist/hooks/use-suggestions.js

@@ -1,118 +0,0 @@
-"use strict";
-"use client";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.useTamboSuggestions = useTamboSuggestions;
-const react_1 = require("react");
-const generate_component_response_1 = require("../model/generate-component-response");
-const validate_input_1 = require("../model/validate-input");
-const tambo_client_provider_1 = require("../providers/tambo-client-provider");
-const tambo_provider_1 = require("../providers/tambo-provider");
-const tambo_registry_provider_1 = require("../providers/tambo-registry-provider");
-const tambo_thread_input_provider_1 = require("../providers/tambo-thread-input-provider");
-const tambo_thread_provider_1 = require("../providers/tambo-thread-provider");
-const query_utils_1 = require("../util/query-utils");
-const registry_1 = require("../util/registry");
-const react_query_hooks_1 = require("./react-query-hooks");
-/**
- * Hook for managing Tambo AI suggestions in a thread
- * @param options - Configuration options for suggestion generation
- * @returns Object containing suggestions state and control functions
- */
-function useTamboSuggestions(options = {}) {
-    const { maxSuggestions = 3 } = options;
-    const { thread, generationStage } = (0, tambo_thread_provider_1.useTamboThread)();
-    const { sendThreadMessage } = (0, tambo_provider_1.useTambo)();
-    const tamboClient = (0, tambo_client_provider_1.useTamboClient)();
-    const { componentList, toolRegistry, componentToolAssociations } = (0, tambo_registry_provider_1.useTamboRegistry)();
-    const [selectedSuggestionId, setSelectedSuggestionId] = (0, react_1.useState)(null);
-    const { setValue: setInputValue } = (0, tambo_thread_input_provider_1.useTamboThreadInput)();
-    const latestMessage = thread.messages[thread.messages.length - 1];
-    const isLatestFromTambo = latestMessage?.role === "assistant";
-    const latestMessageId = latestMessage?.id;
-    // Reset selected suggestion when the message changes
-    (0, react_1.useEffect)(() => {
-        setSelectedSuggestionId(null);
-    }, [latestMessageId]);
-    const shouldGenerateSuggestions = latestMessageId && isLatestFromTambo && (0, generate_component_response_1.isIdleStage)(generationStage);
-    // Use React Query to fetch suggestions when a new hydra message is received
-    const suggestionsResult = (0, react_query_hooks_1.useTamboQuery)({
-        // Make sure the query key changes when the message changes, so that we are
-        // always generating suggestions when there is a new message
-        queryKey: [
-            "suggestions",
-            shouldGenerateSuggestions ? latestMessageId : null,
-        ],
-        queryFn: async () => {
-            if (!shouldGenerateSuggestions) {
-                return [];
-            }
-            // Get registered components from the registry
-            const components = (0, registry_1.getAvailableComponents)(componentList, toolRegistry, componentToolAssociations);
-            return await tamboClient.beta.threads.suggestions.generate(latestMessageId, {
-                id: thread.id,
-                maxSuggestions,
-                availableComponents: components,
-            });
-        },
-        // Only run the query if we have a valid message from hydra
-        enabled: Boolean(latestMessageId && isLatestFromTambo),
-        // Don't refetch on window focus or reconnect
-        refetchOnWindowFocus: false,
-        refetchOnReconnect: false,
-        // Don't retry on failure
-        retry: false,
-    });
-    // Accept suggestion mutation
-    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) {
-                if (validation.error) {
-                    throw validation.error;
-                }
-                throw new Error(tambo_thread_input_provider_1.INPUT_ERROR_MESSAGES.VALIDATION);
-            }
-            if (shouldSubmit) {
-                await sendThreadMessage(validation.sanitizedInput, {
-                    threadId: thread.id,
-                });
-            }
-            else {
-                setInputValue(validation.sanitizedInput);
-            }
-            setSelectedSuggestionId(suggestion.id);
-        },
-    });
-    // Generate suggestions mutation
-    const generateMutationState = (0, react_query_hooks_1.useTamboMutation)({
-        mutationFn: async (abortController) => {
-            if (!shouldGenerateSuggestions) {
-                return undefined;
-            }
-            // Get registered components from the registry
-            const components = (0, registry_1.getAvailableComponents)(componentList, toolRegistry, componentToolAssociations);
-            return await tamboClient.beta.threads.suggestions.generate(latestMessageId, {
-                id: thread.id,
-                maxSuggestions,
-                availableComponents: components,
-            }, { signal: abortController.signal });
-        },
-        // Don't retry on failure
-        retry: false,
-    });
-    // Use the query data if available, otherwise use the mutation data
-    // Only return suggestions if the latest message is from hydra
-    const suggestions = isLatestFromTambo
-        ? (suggestionsResult.data ?? generateMutationState.data ?? [])
-        : [];
-    return {
-        suggestions,
-        accept: acceptMutationState.mutateAsync,
-        selectedSuggestionId,
-        acceptResult: acceptMutationState,
-        generateResult: generateMutationState,
-        suggestionsResult,
-        ...(0, query_utils_1.combineMutationResults)(acceptMutationState, generateMutationState),
-    };
-}
-//# sourceMappingURL=use-suggestions.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"use-suggestions.js","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AA+Eb,kDAwIC;AArND,iCAA4C;AAC5C,sFAAmE;AACnE,4DAAwD;AACxD,8EAAoE;AACpE,gEAAuD;AACvD,kFAAwE;AACxE,0FAGkD;AAClD,8EAAoE;AACpE,qDAG6B;AAC7B,+CAA0D;AAC1D,2DAK6B;AAmD7B;;;;GAIG;AACH,SAAgB,mBAAmB,CACjC,UAAsC,EAAE;IAExC,MAAM,EAAE,cAAc,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IACvC,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,IAAA,sCAAc,GAAE,CAAC;IACrD,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,iDAAmB,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,MAAM,yBAAyB,GAC7B,eAAe,IAAI,iBAAiB,IAAI,IAAA,yCAAW,EAAC,eAAe,CAAC,CAAC;IACvE,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,IAAA,iCAAa,EAAC;QACtC,2EAA2E;QAC3E,4DAA4D;QAC5D,QAAQ,EAAE;YACR,aAAa;YACb,yBAAyB,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI;SACnD;QACD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,yBAAyB,EAAE,CAAC;gBAC/B,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,eAAe,EACf;gBACE,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,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,kDAAoB,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,yBAAyB,EAAE,CAAC;gBAC/B,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YACF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,eAAe,EACf;gBACE,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,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":["\"use client\";\nimport TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { useEffect, useState } from \"react\";\nimport { isIdleStage } from \"../model/generate-component-response\";\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 {\n  INPUT_ERROR_MESSAGES,\n  useTamboThreadInput,\n} from \"../providers/tambo-thread-input-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\";\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  >;\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, generationStage } = 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  const shouldGenerateSuggestions =\n    latestMessageId && isLatestFromTambo && isIdleStage(generationStage);\n  // Use React Query to fetch suggestions when a new hydra message is received\n  const suggestionsResult = useTamboQuery({\n    // Make sure the query key changes when the message changes, so that we are\n    // always generating suggestions when there is a new message\n    queryKey: [\n      \"suggestions\",\n      shouldGenerateSuggestions ? latestMessageId : null,\n    ],\n    queryFn: async () => {\n      if (!shouldGenerateSuggestions) {\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        latestMessageId,\n        {\n          id: thread.id,\n          maxSuggestions,\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 (!shouldGenerateSuggestions) {\n        return undefined;\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n      return await tamboClient.beta.threads.suggestions.generate(\n        latestMessageId,\n        {\n          id: thread.id,\n          maxSuggestions,\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"]}

package/dist/hooks/use-suggestions.test.d.ts

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

package/dist/hooks/use-suggestions.test.d.ts.map

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

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

@@ -1,247 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const react_query_1 = require("@tanstack/react-query");
-const react_1 = require("@testing-library/react");
-const generate_component_response_1 = require("../model/generate-component-response");
-const tambo_client_provider_1 = require("../providers/tambo-client-provider");
-const tambo_provider_1 = require("../providers/tambo-provider");
-const tambo_thread_input_provider_1 = require("../providers/tambo-thread-input-provider");
-const tambo_thread_provider_1 = require("../providers/tambo-thread-provider");
-const react_query_hooks_1 = require("./react-query-hooks");
-const use_suggestions_1 = require("./use-suggestions");
-// Mock the required providers
-jest.mock("../providers/tambo-client-provider", () => ({
-    useTamboClient: jest.fn(),
-    useTamboQueryClient: jest.fn(),
-}));
-jest.mock("../providers/tambo-provider", () => ({ useTambo: jest.fn() }));
-jest.mock("../providers/tambo-thread-provider", () => ({
-    useTamboThread: jest.fn(),
-}));
-jest.mock("../providers/tambo-thread-input-provider", () => ({
-    useTamboThreadInput: jest.fn(),
-}));
-// Mock the react-query-hooks
-jest.mock("./react-query-hooks", () => ({
-    useTamboQuery: jest.fn(),
-    useTamboMutation: jest.fn(),
-}));
-describe("useTamboSuggestions", () => {
-    const mockSuggestions = [
-        {
-            id: "suggestion-1",
-            messageId: "test-message-id",
-            title: "Test Suggestion 1",
-            detailedSuggestion: "Test suggestion 1",
-        },
-        {
-            id: "suggestion-2",
-            messageId: "test-message-id",
-            title: "Test Suggestion 2",
-            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 = {}) => {
-        const mockThread = createMockThread();
-        return {
-            thread: mockThread,
-            currentThreadId: mockThread.id,
-            currentThread: mockThread,
-            threadMap: { [mockThread.id]: mockThread },
-            setThreadMap: jest.fn(),
-            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: generate_component_response_1.GenerationStage.IDLE,
-            generationStatusMessage: "",
-            isIdle: true,
-            ...overrides,
-        };
-    };
-    beforeEach(() => {
-        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
-        jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue({
-            beta: { threads: { suggestions: { generate: jest.fn() } } },
-        });
-        jest.mocked(tambo_provider_1.useTambo).mockReturnValue({
-            sendThreadMessage: jest.fn(),
-        });
-        jest.mocked(tambo_thread_provider_1.useTamboThread).mockReturnValue(createMockThreadContext({
-            thread: createMockThread({
-                messages: [
-                    createMockMessage({
-                        role: "assistant",
-                    }),
-                ],
-            }),
-        }));
-        jest.mocked(tambo_thread_input_provider_1.useTamboThreadInput).mockReturnValue({
-            setValue: jest.fn(),
-            value: "",
-            submit: jest.fn(),
-        });
-        // Default query mock returns empty array
-        jest.mocked(react_query_hooks_1.useTamboQuery).mockReturnValue({
-            data: [],
-            isLoading: false,
-            isError: false,
-            error: null,
-        });
-    });
-    it("should initialize with empty suggestions and no selected suggestion", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_suggestions_1.useTamboSuggestions)());
-        expect(result.current.suggestions).toEqual([]);
-        expect(result.current.selectedSuggestionId).toBeNull();
-    });
-    it("should generate suggestions when latest message is from Tambo", async () => {
-        const mockGenerate = jest.fn().mockResolvedValue(mockSuggestions);
-        jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue({
-            beta: {
-                threads: {
-                    suggestions: {
-                        generate: mockGenerate,
-                        list: jest.fn(),
-                    },
-                },
-            },
-        });
-        jest.mocked(tambo_thread_provider_1.useTamboThread).mockReturnValue(createMockThreadContext({
-            thread: createMockThread({
-                messages: [
-                    createMockMessage({
-                        role: "assistant",
-                    }),
-                ],
-            }),
-        }));
-        // Mock the query result to return the mock suggestions
-        jest.mocked(react_query_hooks_1.useTamboQuery).mockReturnValue({
-            data: mockSuggestions,
-            isLoading: false,
-            isError: false,
-            error: null,
-        });
-        const { result } = (0, react_1.renderHook)(() => (0, use_suggestions_1.useTamboSuggestions)());
-        // Wait for the effect to run
-        await (0, react_1.act)(async () => {
-            await new Promise((resolve) => setTimeout(resolve, 0));
-        });
-        // Since we're mocking useTamboQuery to return the suggestions directly,
-        // the generate function won't be called, so we don't need to check that
-        expect(result.current.suggestions).toEqual(mockSuggestions);
-    });
-    it("should not generate suggestions when latest message is not from Tambo", async () => {
-        const mockGenerate = jest.fn();
-        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
-        jest.mocked(tambo_thread_provider_1.useTamboThread).mockReturnValue(createMockThreadContext({
-            thread: createMockThread({
-                messages: [
-                    createMockMessage({
-                        role: "user",
-                    }),
-                ],
-            }),
-        }));
-        const { result } = (0, react_1.renderHook)(() => (0, use_suggestions_1.useTamboSuggestions)());
-        // Wait for the effect to run
-        await (0, react_1.act)(async () => {
-            await new Promise((resolve) => setTimeout(resolve, 0));
-        });
-        expect(mockGenerate).not.toHaveBeenCalled();
-        expect(result.current.suggestions).toEqual([]);
-    });
-    it("should accept a suggestion and update input value", async () => {
-        const mockSetValue = jest.fn();
-        jest.mocked(tambo_thread_input_provider_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 () => {
-            await result.current.accept({
-                suggestion: mockSuggestions[0],
-                shouldSubmit: false,
-            });
-        });
-        expect(mockSetValue).toHaveBeenCalledWith("Test suggestion 1");
-        expect(result.current.selectedSuggestionId).toBe("suggestion-1");
-    });
-    it("should accept a suggestion and submit it", async () => {
-        const mockSendThreadMessage = jest.fn();
-        jest.mocked(tambo_provider_1.useTambo).mockReturnValue({
-            sendThreadMessage: mockSendThreadMessage,
-        });
-        const { result } = (0, react_1.renderHook)(() => (0, use_suggestions_1.useTamboSuggestions)());
-        await (0, react_1.act)(async () => {
-            await result.current.accept({
-                suggestion: mockSuggestions[0],
-                shouldSubmit: true,
-            });
-        });
-        expect(mockSendThreadMessage).toHaveBeenCalledWith("Test suggestion 1", {
-            threadId: "test-thread-id",
-        });
-        expect(result.current.selectedSuggestionId).toBe("suggestion-1");
-    });
-    it("should throw error when accepting invalid suggestion", async () => {
-        const invalidSuggestion = {
-            id: "invalid-suggestion",
-            messageId: "test-message-id",
-            title: "Invalid Suggestion",
-            detailedSuggestion: "", // Empty suggestion should fail validation
-        };
-        const { result } = (0, react_1.renderHook)(() => (0, use_suggestions_1.useTamboSuggestions)());
-        await (0, react_1.act)(async () => {
-            await expect(result.current.accept({
-                suggestion: invalidSuggestion,
-                shouldSubmit: false,
-            })).rejects.toThrow("Message cannot be empty");
-        });
-        expect(result.current.selectedSuggestionId).toBeNull();
-    });
-});
-//# sourceMappingURL=use-suggestions.test.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"use-suggestions.test.js","sourceRoot":"","sources":["../../src/hooks/use-suggestions.test.tsx"],"names":[],"mappings":";;AACA,uDAAoE;AACpE,kDAAyD;AACzD,sFAG8C;AAE9C,8EAG4C;AAC5C,gEAA0E;AAC1E,0FAGkD;AAClD,8EAG4C;AAE5C,2DAAsE;AACtE,uDAAwD;AAExD,8BAA8B;AAC9B,IAAI,CAAC,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE,CAAC,CAAC;IACrD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC/B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAE1E,IAAI,CAAC,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE,CAAC,CAAC;IACrD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,0CAA0C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3D,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC/B,CAAC,CAAC,CAAC;AAEJ,6BAA6B;AAC7B,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,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;QACnC,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;QACtC,OAAO;YACL,MAAM,EAAE,UAAU;YAClB,eAAe,EAAE,UAAU,CAAC,EAAE;YAC9B,aAAa,EAAE,UAAU;YACzB,SAAS,EAAE,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,UAAU,EAAE;YAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC9B,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;YACzB,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC3B,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC7B,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC3B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC9B,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;YACjB,SAAS,EAAE,KAAK;YAChB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC5B,eAAe,EAAE,6CAAe,CAAC,IAAI;YACrC,uBAAuB,EAAE,EAAE;YAC3B,MAAM,EAAE,IAAI;YACZ,GAAG,SAAS;SACb,CAAC;IACJ,CAAC,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,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;QACF,IAAI,CAAC,MAAM,CAAC,iDAAmB,CAAC,CAAC,eAAe,CAAC;YAC/C,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;SACqC,CAAC,CAAC;QAC1D,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,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,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,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,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,iDAAmB,CAAC,CAAC,eAAe,CAAC;YAC/C,QAAQ,EAAE,YAAY;YACtB,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;SACqC,CAAC,CAAC;QAE1D,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 {\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  TamboThreadInputContextProps,\n  useTamboThreadInput,\n} from \"../providers/tambo-thread-input-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\";\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(\"../providers/tambo-thread-input-provider\", () => ({\n  useTamboThreadInput: jest.fn(),\n}));\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    const mockThread = createMockThread();\n    return {\n      thread: mockThread,\n      currentThreadId: mockThread.id,\n      currentThread: mockThread,\n      threadMap: { [mockThread.id]: mockThread },\n      setThreadMap: jest.fn(),\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\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: \"assistant\",\n            }),\n          ],\n        }),\n      }),\n    );\n    jest.mocked(useTamboThreadInput).mockReturnValue({\n      setValue: jest.fn(),\n      value: \"\",\n      submit: jest.fn(),\n    } satisfies Partial<TamboThreadInputContextProps> 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<TamboThreadInputContextProps> 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/use-tambo-stream-status.d.ts

@@ -1,90 +0,0 @@
-/**
- * Global stream status flags for a specific component in a message.
- * Represents the aggregate state across all props for this component only.
- * Once a component completes, its status remains stable regardless of other generations.
- */
-export interface StreamStatus {
-    /**
-     * Indicates no tokens have been received for any prop and generation is not active.
-     * Useful for showing initial loading states before any data arrives.
-     */
-    isPending: boolean;
-    /**
-     * Indicates active streaming - either generation is streaming OR at least one prop is still streaming.
-     * Use this to show loading animations or skeleton states during data transmission.
-     */
-    isStreaming: boolean;
-    /**
-     * Indicates successful completion - generation is complete AND every prop finished without error.
-     * Safe to render the final component when this is true.
-     */
-    isSuccess: boolean;
-    /**
-     * Indicates a fatal error occurred in any prop or the stream itself.
-     * Check streamError for details about what went wrong.
-     */
-    isError: boolean;
-    /**
-     * The first fatal error encountered during streaming (if any).
-     * Will be undefined if no errors occurred.
-     */
-    streamError?: Error;
-}
-/**
- * Streaming status flags for individual component props.
- * Tracks the state of each prop as it streams from the LLM.
- */
-export interface PropStatus {
-    /**
-     * Indicates no tokens have been received for this specific prop yet.
-     * The prop value is still undefined, null, or empty string.
-     */
-    isPending: boolean;
-    /**
-     * Indicates at least one token has been received but streaming is not complete.
-     * The prop has partial content that may still be updating.
-     */
-    isStreaming: boolean;
-    /**
-     * Indicates this prop has finished streaming successfully.
-     * The prop value is complete and stable.
-     */
-    isSuccess: boolean;
-    /**
-     * The error that occurred during streaming (if any).
-     * Will be undefined if no error occurred for this prop.
-     */
-    error?: Error;
-}
-/**
- * Track streaming status for Tambo component props.
- *
- * **Important**: Props update repeatedly during streaming and may be partial.
- * Use `propStatus.<field>?.isSuccess` before treating a prop as complete.
- *
- * Pair with `useTamboComponentState` to disable inputs while streaming.
- * @see {@link https://docs.tambo.co/concepts/generative-interfaces/component-state}
- * @template Props - Component props type
- * @returns `streamStatus` (overall) and `propStatus` (per-prop) flags
- * @throws {Error} When used during SSR/SSG
- * @example
- * ```tsx
- * // Wait for entire stream
- * const { streamStatus } = useTamboStreamStatus();
- * if (!streamStatus.isSuccess) return <Spinner />;
- * return <Card {...props} />;
- * ```
- * @example
- * ```tsx
- * // Highlight in-flight props
- * const { propStatus } = useTamboStreamStatus<Props>();
- * <h2 className={propStatus.title.isStreaming ? "animate-pulse" : ""}>
- *   {title}
- * </h2>
- * ```
- */
-export declare function useTamboStreamStatus<Props extends Record<string, any> = Record<string, any>>(): {
-    streamStatus: StreamStatus;
-    propStatus: Record<keyof Props, PropStatus>;
-};
-//# sourceMappingURL=use-tambo-stream-status.d.ts.map

package/dist/hooks/use-tambo-stream-status.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"use-tambo-stream-status.d.ts","sourceRoot":"","sources":["../../src/hooks/use-tambo-stream-status.tsx"],"names":[],"mappings":"AAMA;;;;GAIG;AACH,MAAM,WAAW,YAAY;IAC3B;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,WAAW,EAAE,OAAO,CAAC;IAErB;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,OAAO,EAAE,OAAO,CAAC;IAEjB;;;OAGG;IACH,WAAW,CAAC,EAAE,KAAK,CAAC;CACrB;AAED;;;GAGG;AACH,MAAM,WAAW,UAAU;IACzB;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,WAAW,EAAE,OAAO,CAAC;IAErB;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,KAAK,CAAC,EAAE,KAAK,CAAC;CACf;AA8ND;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,wBAAgB,oBAAoB,CAClC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KACpD;IACH,YAAY,EAAE,YAAY,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC,MAAM,KAAK,EAAE,UAAU,CAAC,CAAC;CAC7C,CAmCA"}