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

package/dist/providers/tambo-thread-provider.js

@@ -1,931 +0,0 @@
-"use strict";
-"use client";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.useTamboThread = exports.useTamboGenerationStage = exports.TamboThreadProvider = exports.TamboThreadContext = exports.PLACEHOLDER_THREAD = exports.TamboGenerationStageProvider = void 0;
-const typescript_sdk_1 = require("@tambo-ai/typescript-sdk");
-const react_1 = __importStar(require("react"));
-const generate_component_response_1 = require("../model/generate-component-response");
-const content_parts_1 = require("../util/content-parts");
-const generate_component_1 = require("../util/generate-component");
-const registry_1 = require("../util/registry");
-const tool_caller_1 = require("../util/tool-caller");
-const tambo_client_provider_1 = require("./tambo-client-provider");
-const tambo_context_helpers_provider_1 = require("./tambo-context-helpers-provider");
-const tambo_registry_provider_1 = require("./tambo-registry-provider");
-const TamboGenerationStageContext = (0, react_1.createContext)(undefined);
-/**
- *
- * This provider is used to provide the generation stage context to the descendants of the provider.
- * @param props - The props for the GenerationStageProvider
- * @param props.children - The children to wrap
- * @param props.generationStage - The generation stage to provide
- * @param props.statusMessage - The status message to provide
- * @returns The GenerationStageProvider component
- */
-const TamboGenerationStageProvider = ({ children, generationStage, statusMessage }) => {
-    const isIdle = (0, generate_component_response_1.isIdleStage)(generationStage);
-    const contextValue = (0, react_1.useMemo)(() => {
-        return {
-            generationStage,
-            generationStatusMessage: statusMessage,
-            isIdle,
-        };
-    }, [generationStage, statusMessage, isIdle]);
-    return (react_1.default.createElement(TamboGenerationStageContext.Provider, { value: contextValue }, children));
-};
-exports.TamboGenerationStageProvider = TamboGenerationStageProvider;
-/**
- * This is a stub entry for when the thread is not yet created, the first time
- * the user sends a message
- *
- * Note that the consumer needs to be careful never to send `PLACEHOLDER_THREAD.id` to the server,
- * as this doesn't really exist on the server side.
- */
-exports.PLACEHOLDER_THREAD = {
-    id: "placeholder",
-    messages: [],
-    createdAt: "",
-    projectId: "",
-    updatedAt: "",
-    metadata: {},
-};
-exports.TamboThreadContext = (0, react_1.createContext)({
-    thread: exports.PLACEHOLDER_THREAD,
-    currentThreadId: exports.PLACEHOLDER_THREAD.id,
-    currentThread: exports.PLACEHOLDER_THREAD,
-    threadMap: { [exports.PLACEHOLDER_THREAD.id]: exports.PLACEHOLDER_THREAD },
-    setThreadMap: () => {
-        throw new Error("setThreadMap not implemented");
-    },
-    /**
-     *
-     */
-    switchCurrentThread: () => {
-        throw new Error("switchCurrentThread not implemented");
-    },
-    /**
-     *
-     */
-    startNewThread: () => {
-        throw new Error("startNewThread not implemented");
-    },
-    /**
-     *
-     */
-    updateThreadName: () => {
-        throw new Error("updateThreadName not implemented");
-    },
-    /**
-     *
-     */
-    generateThreadName: () => {
-        throw new Error("generateThreadName not implemented");
-    },
-    /**
-     *
-     */
-    addThreadMessage: () => {
-        throw new Error("updateThreadMessageHistory not implemented");
-    },
-    streaming: true,
-    /**
-     *
-     */
-    updateThreadMessage: () => {
-        throw new Error("updateThreadMessage not implemented");
-    },
-    /**
-     *
-     */
-    sendThreadMessage: () => {
-        throw new Error("sendThreadMessage not implemented");
-    },
-    /**
-     *
-     */
-    cancel: () => {
-        throw new Error("cancel not implemented");
-    },
-});
-/**
- * The TamboThreadProvider is a React provider that provides a thread context
- * to the descendants of the provider.
- * @param props - The props for the TamboThreadProvider
- * @param props.children - The children to wrap
- * @param props.streaming - Whether to stream the response by default. Defaults to true.
- * @param props.initialMessages - Initial messages to be included in new threads
- * @param props.autoGenerateThreadName - Whether to automatically generate thread names. Defaults to true.
- * @param props.autoGenerateNameThreshold - The message count threshold at which the thread name will be auto-generated. Defaults to 3.
- * @param props.contextKey - The context key to use for thread isolation
- * @returns The TamboThreadProvider component
- */
-const TamboThreadProvider = ({ children, streaming = true, initialMessages = [], autoGenerateThreadName = true, autoGenerateNameThreshold = 3, contextKey, }) => {
-    // Create placeholder thread with initial messages
-    const placeholderThread = (0, react_1.useMemo)(() => ({
-        id: "placeholder",
-        messages: initialMessages.map((msg) => ({
-            ...msg,
-            id: msg.id ?? crypto.randomUUID(),
-            threadId: "placeholder",
-            createdAt: msg.createdAt ?? new Date().toISOString(),
-            componentState: msg.componentState ?? {},
-        })),
-        createdAt: "",
-        projectId: "",
-        updatedAt: "",
-        metadata: {},
-        contextKey,
-    }), [contextKey, initialMessages]);
-    const [threadMap, setThreadMap] = (0, react_1.useState)({
-        [placeholderThread.id]: placeholderThread,
-    });
-    const client = (0, tambo_client_provider_1.useTamboClient)();
-    const queryClient = (0, tambo_client_provider_1.useTamboQueryClient)();
-    const { componentList, toolRegistry, componentToolAssociations, onCallUnregisteredTool, } = (0, tambo_registry_provider_1.useTamboRegistry)();
-    const { getAdditionalContext } = (0, tambo_context_helpers_provider_1.useTamboContextHelpers)();
-    const [ignoreResponse, setIgnoreResponse] = (0, react_1.useState)(false);
-    const ignoreResponseRef = (0, react_1.useRef)(ignoreResponse);
-    const [currentThreadId, setCurrentThreadId] = (0, react_1.useState)(placeholderThread.id);
-    const currentThread = threadMap[currentThreadId];
-    // Generation stage props for GenerationStageProvider
-    const currentGenerationStage = currentThread?.generationStage ?? generate_component_response_1.GenerationStage.IDLE;
-    const currentStatusMessage = currentThread?.statusMessage ?? "";
-    // Use existing messages from the current thread to avoid re-generating any components
-    const currentMessageCache = (0, react_1.useMemo)(() => {
-        const messageCache = new Map();
-        if (currentThread) {
-            for (const message of currentThread.messages) {
-                messageCache.set(message.id, message);
-            }
-        }
-        return messageCache;
-    }, [currentThread]);
-    (0, react_1.useEffect)(() => {
-        ignoreResponseRef.current = ignoreResponse;
-    }, [ignoreResponse]);
-    const updateThreadsCache = (0, react_1.useCallback)(async (updateFn, contextKey) => {
-        try {
-            const currentProject = await client.beta.projects.getCurrent();
-            queryClient.setQueryData(["threads", currentProject.id, contextKey], updateFn);
-            await queryClient.invalidateQueries({
-                queryKey: ["threads"],
-            });
-        }
-        catch (error) {
-            console.warn("Failed to update threads cache:", error);
-        }
-    }, [client.beta.projects, queryClient]);
-    const addThreadToCache = (0, react_1.useCallback)(async (threadId, contextKey) => {
-        const optimisticThread = {
-            ...exports.PLACEHOLDER_THREAD,
-            id: threadId,
-            createdAt: new Date().toISOString(),
-            updatedAt: new Date().toISOString(),
-        };
-        await updateThreadsCache((old) => {
-            if (!old)
-                return old;
-            return {
-                ...old,
-                items: [optimisticThread, ...(old.items ?? [])],
-                total: (old.total ?? 0) + 1,
-                count: (old.count ?? 0) + 1,
-            };
-        }, contextKey);
-    }, [updateThreadsCache]);
-    const updateThreadInCache = (0, react_1.useCallback)(async (threadId, updateFn, contextKey) => {
-        await updateThreadsCache((old) => {
-            if (!old?.items)
-                return old;
-            return {
-                ...old,
-                items: old.items.map((thread) => thread.id === threadId ? updateFn(thread) : thread),
-                total: old.total,
-                count: old.count,
-            };
-        }, contextKey);
-    }, [updateThreadsCache]);
-    const fetchThread = (0, react_1.useCallback)(async (threadId) => {
-        const thread = await client.beta.threads.retrieve(threadId);
-        const threadWithRenderedComponents = {
-            ...thread,
-            messages: thread.messages.map((message) => {
-                if (currentMessageCache.has(message.id)) {
-                    const renderedMessage = currentMessageCache.get(message.id);
-                    return {
-                        ...renderedMessage,
-                        ...message,
-                    };
-                }
-                if (message.component?.componentName) {
-                    const messageWithComponent = (0, generate_component_1.renderComponentIntoMessage)(message, componentList);
-                    return messageWithComponent;
-                }
-                return message;
-            }),
-        };
-        setThreadMap((prevMap) => {
-            const updatedThreadMap = {
-                ...prevMap,
-                [threadId]: threadWithRenderedComponents,
-            };
-            return updatedThreadMap;
-        });
-    }, [client.beta.threads, componentList, currentMessageCache]);
-    (0, react_1.useEffect)(() => {
-        async function loadThread() {
-            try {
-                await fetchThread(currentThreadId);
-            }
-            catch (error) {
-                console.error("Failed to fetch thread:", error);
-            }
-        }
-        if (currentThreadId &&
-            currentThreadId !== placeholderThread.id &&
-            !threadMap[currentThreadId]) {
-            loadThread().catch((err) => {
-                console.error("Unexpected error in loadThread:", err);
-            });
-        }
-    }, [currentThreadId, fetchThread, threadMap, placeholderThread.id]);
-    const addThreadMessage = (0, react_1.useCallback)(async (message, sendToServer = true, createdAt = new Date().toISOString()) => {
-        if (!currentThread) {
-            console.warn("Cannot add messages if we do not have a current thread");
-            return [];
-        }
-        const chatMessage = {
-            ...message,
-            createdAt,
-        };
-        const threadId = message.threadId;
-        const messageId = chatMessage.id;
-        // optimistically update the thread in the local state
-        setThreadMap((prevMap) => {
-            if (!threadId) {
-                return prevMap;
-            }
-            const prevMessages = prevMap[threadId]?.messages || [];
-            const haveMessage = prevMessages.find((msg) => msg.id === messageId);
-            // Update in place if the message already exists
-            const updatedMessages = haveMessage
-                ? prevMessages.map((msg) => {
-                    if (msg.id === messageId) {
-                        return chatMessage;
-                    }
-                    return msg;
-                })
-                : [...prevMessages, chatMessage];
-            const updatedThreadMap = {
-                ...prevMap,
-                [threadId]: {
-                    ...prevMap[threadId],
-                    messages: updatedMessages,
-                },
-            };
-            return updatedThreadMap;
-        });
-        if (sendToServer) {
-            try {
-                await client.beta.threads.messages.create(message.threadId, {
-                    content: message.content,
-                    role: message.role,
-                    additionalContext: chatMessage.additionalContext,
-                });
-            }
-            catch (error) {
-                console.error("Failed to add message to server", error);
-                // Revert optimistic local state update
-                setThreadMap((prev) => {
-                    const prevMessages = prev[threadId]?.messages || [];
-                    // addThreadMessage -> for new messages or simple updates
-                    // On Fail -> remove the added/updated message (no prev state)
-                    const updatedMessages = prevMessages.filter((msg) => msg.id !== messageId);
-                    return {
-                        ...prev,
-                        [threadId]: {
-                            ...prev[threadId],
-                            messages: updatedMessages,
-                        },
-                    };
-                });
-                throw error;
-            }
-        }
-        return threadMap[threadId]?.messages || [];
-    }, [client.beta.threads.messages, currentThread, threadMap]);
-    const updateThreadMessage = (0, react_1.useCallback)(async (id, message, sendToServer = true, createdAt = new Date().toISOString()) => {
-        setThreadMap((prevMap) => {
-            const prevMessages = prevMap[message.threadId]?.messages || [];
-            const updatedMessages = prevMessages.map((msg) => {
-                if (msg.id === id) {
-                    // Merge the partial update with the existing message
-                    const updatedMessage = {
-                        ...msg,
-                        ...message,
-                        id: msg.id,
-                        createdAt: message.createdAt ?? msg.createdAt ?? createdAt,
-                    };
-                    return updatedMessage;
-                }
-                return msg;
-            });
-            return {
-                ...prevMap,
-                [message.threadId]: {
-                    ...prevMap[message.threadId],
-                    messages: updatedMessages,
-                },
-            };
-        });
-        if (sendToServer && message.content && message.role) {
-            try {
-                await client.beta.threads.messages.create(message.threadId, {
-                    content: message.content,
-                    role: message.role,
-                    additionalContext: message.additionalContext,
-                });
-            }
-            catch (error) {
-                console.error("Failed to update message on server", error);
-                // Revert local state update by removing the optimistic message
-                setThreadMap((prev) => {
-                    const updatedMessages = prev[message.threadId]?.messages?.filter((msg) => msg.id !== id);
-                    return {
-                        ...prev,
-                        [message.threadId]: {
-                            ...prev[message.threadId],
-                            messages: updatedMessages,
-                        },
-                    };
-                });
-                throw error;
-            }
-        }
-    }, [client.beta.threads.messages]);
-    const generateThreadName = (0, react_1.useCallback)(async (threadId, contextKey) => {
-        threadId ??= currentThreadId;
-        if (threadId === placeholderThread.id) {
-            console.warn("Cannot generate name for empty thread");
-            return threadMap[threadId];
-        }
-        const threadWithGeneratedName = await client.beta.threads.generateName(threadId);
-        // Update local thread state
-        setThreadMap((prevMap) => {
-            if (!prevMap[threadId]) {
-                return prevMap;
-            }
-            return {
-                ...prevMap,
-                [threadId]: {
-                    ...prevMap[threadId],
-                    name: threadWithGeneratedName.name,
-                },
-            };
-        });
-        // Update threads cache to reflect the new name in the UI
-        await updateThreadInCache(threadId, (thread) => ({
-            ...thread,
-            name: threadWithGeneratedName.name,
-        }), contextKey);
-        return threadWithGeneratedName;
-    }, [
-        client.beta.threads,
-        currentThreadId,
-        threadMap,
-        placeholderThread.id,
-        updateThreadInCache,
-    ]);
-    const maybeAutoGenerateThreadName = (0, react_1.useCallback)((threadId, contextKey) => {
-        // Use setThreadMap to access the latest state
-        setThreadMap((map) => {
-            const thread = map[threadId];
-            if (!thread || !autoGenerateThreadName) {
-                return map;
-            }
-            if (autoGenerateNameThreshold == null ||
-                thread.id === placeholderThread.id) {
-                return map;
-            }
-            const messageCount = thread.messages.length;
-            // Only auto-generate if thread has no name and threshold is met
-            if (!thread.name && messageCount >= autoGenerateNameThreshold) {
-                // Generating a thread name is not critical, so we can fire-and-forget
-                void generateThreadName(threadId, contextKey).catch((error) => {
-                    console.error("Failed to generate thread name:", error);
-                });
-            }
-            return map;
-        });
-    }, [
-        autoGenerateThreadName,
-        autoGenerateNameThreshold,
-        placeholderThread.id,
-        generateThreadName,
-    ]);
-    const startNewThread = (0, react_1.useCallback)(() => {
-        setCurrentThreadId(placeholderThread.id);
-        setThreadMap((prevMap) => {
-            return {
-                ...prevMap,
-                [placeholderThread.id]: placeholderThread,
-            };
-        });
-    }, [placeholderThread]);
-    const updateThreadName = (0, react_1.useCallback)(async (name, threadId) => {
-        threadId ??= currentThreadId;
-        const previousName = threadMap[threadId]?.name;
-        setThreadMap((prevMap) => {
-            if (!prevMap[threadId]) {
-                return prevMap;
-            }
-            return { ...prevMap, [threadId]: { ...prevMap[threadId], name } };
-        });
-        if (threadId !== placeholderThread.id) {
-            try {
-                const currentProject = await client.beta.projects.getCurrent();
-                await client.beta.threads.update(threadId, {
-                    name,
-                    projectId: currentProject.id,
-                });
-            }
-            catch (error) {
-                console.error("Failed to update thread name on server", error);
-                // Revert local state update
-                setThreadMap((prevMap) => {
-                    if (!prevMap[threadId]) {
-                        return prevMap;
-                    }
-                    return {
-                        ...prevMap,
-                        [threadId]: { ...prevMap[threadId], name: previousName },
-                    };
-                });
-                throw error;
-            }
-        }
-    }, [
-        currentThreadId,
-        client.beta.projects,
-        client.beta.threads,
-        placeholderThread.id,
-        threadMap,
-    ]);
-    const switchCurrentThread = (0, react_1.useCallback)(async (threadId, fetch = true) => {
-        if (threadId === placeholderThread.id) {
-            console.warn("Switching to placeholder thread, may be a bug.");
-            return;
-        }
-        setCurrentThreadId(threadId);
-        setThreadMap((prevMap) => {
-            if (prevMap[threadId]) {
-                return prevMap;
-            }
-            // If this is a new thread, add placeholder thread messages to the thread
-            const updatedThreadMap = {
-                ...prevMap,
-                [threadId]: {
-                    ...prevMap[placeholderThread.id],
-                    id: threadId,
-                },
-            };
-            return updatedThreadMap;
-        });
-        if (fetch) {
-            await fetchThread(threadId);
-        }
-    }, [fetchThread, placeholderThread]);
-    const updateThreadStatus = (0, react_1.useCallback)((threadId, stage, statusMessage) => {
-        setThreadMap((prevMap) => {
-            const updatedThreadMap = {
-                ...prevMap,
-                [threadId]: {
-                    ...prevMap[threadId],
-                    generationStage: stage,
-                    statusMessage: statusMessage,
-                },
-            };
-            return updatedThreadMap;
-        });
-    }, []);
-    const cancel = (0, react_1.useCallback)(async (threadId) => {
-        threadId ??= currentThreadId;
-        const currentGenerationStage = currentThread?.generationStage ?? generate_component_response_1.GenerationStage.IDLE;
-        if ((0, generate_component_response_1.isIdleStage)(currentGenerationStage)) {
-            return;
-        }
-        setIgnoreResponse(true);
-        setThreadMap((prevMap) => {
-            if (!prevMap[threadId]) {
-                return prevMap;
-            }
-            return {
-                ...prevMap,
-                [threadId]: {
-                    ...prevMap[threadId],
-                    generationStage: generate_component_response_1.GenerationStage.CANCELLED,
-                    messages: prevMap[threadId].messages.map((message) => {
-                        if (message.id ===
-                            prevMap[threadId].messages[prevMap[threadId].messages.length - 1].id) {
-                            return {
-                                ...message,
-                                isCancelled: true,
-                            };
-                        }
-                        return message;
-                    }),
-                },
-            };
-        });
-        await client.beta.threads.cancel(threadId);
-    }, [client.beta.threads, currentThreadId, currentThread?.generationStage]);
-    const handleAdvanceStream = (0, react_1.useCallback)(async (stream, params, threadId, contextKey) => {
-        if (ignoreResponseRef.current) {
-            setIgnoreResponse(false);
-            return createEmptyMessage(threadId);
-        }
-        let finalMessage;
-        let hasSetThreadId = false;
-        updateThreadStatus(threadId, generate_component_response_1.GenerationStage.STREAMING_RESPONSE);
-        for await (const chunk of stream) {
-            // Store MCP access token in thread data
-            if (chunk.mcpAccessToken) {
-                // note that we're only setting it positively during streaming, because it might
-                // not have been set yet in the chunk (i.e. we're not unsetting it in the chunk)
-                setThreadMap((prev) => {
-                    const thread = prev[threadId];
-                    if (thread) {
-                        return {
-                            ...prev,
-                            [threadId]: {
-                                ...thread,
-                                mcpAccessToken: chunk.mcpAccessToken,
-                            },
-                        };
-                    }
-                    return prev;
-                });
-            }
-            if (chunk.responseMessageDto.toolCallRequest) {
-                const toolName = chunk.responseMessageDto.toolCallRequest.toolName;
-                updateThreadStatus(chunk.responseMessageDto.threadId, generate_component_response_1.GenerationStage.FETCHING_CONTEXT);
-                await updateThreadMessage(chunk.responseMessageDto.id, {
-                    ...chunk.responseMessageDto,
-                }, false);
-                const toolCallResponse = await (0, tool_caller_1.handleToolCall)(chunk.responseMessageDto.toolCallRequest, toolRegistry, onCallUnregisteredTool);
-                if (ignoreResponseRef.current) {
-                    setIgnoreResponse(false);
-                    return createEmptyMessage(threadId);
-                }
-                const contentParts = await convertToolResponse(toolCallResponse);
-                const toolCallResponseParams = {
-                    ...params,
-                    // Exclude initialMessages from tool response since thread already exists
-                    initialMessages: undefined,
-                    ...(toolName
-                        ? {
-                            toolCallCounts: {
-                                ...(params.toolCallCounts ?? {}),
-                                [toolName]: (params.toolCallCounts?.[toolName] ?? 0) + 1,
-                            },
-                        }
-                        : {}),
-                    messageToAppend: {
-                        content: contentParts,
-                        role: "tool",
-                        actionType: "tool_response",
-                        component: chunk.responseMessageDto.component,
-                        tool_call_id: chunk.responseMessageDto.tool_call_id,
-                        error: toolCallResponse.error,
-                    },
-                };
-                await updateThreadMessage(chunk.responseMessageDto.id, {
-                    ...chunk.responseMessageDto,
-                    error: toolCallResponse.error,
-                }, false);
-                await addThreadMessage({
-                    threadId: chunk.responseMessageDto.threadId,
-                    content: contentParts,
-                    role: "tool",
-                    id: crypto.randomUUID(),
-                    createdAt: new Date().toISOString(),
-                    componentState: {},
-                    actionType: "tool_response",
-                    tool_call_id: chunk.responseMessageDto.tool_call_id,
-                    error: toolCallResponse.error,
-                }, false);
-                updateThreadStatus(chunk.responseMessageDto.threadId, generate_component_response_1.GenerationStage.STREAMING_RESPONSE);
-                const toolCallResponseStream = await (0, typescript_sdk_1.advanceStream)(client, toolCallResponseParams, chunk.responseMessageDto.threadId);
-                return await handleAdvanceStream(toolCallResponseStream, toolCallResponseParams, chunk.responseMessageDto.threadId, contextKey);
-            }
-            else {
-                if (ignoreResponseRef.current) {
-                    setIgnoreResponse(false);
-                    return finalMessage ?? createEmptyMessage(threadId);
-                }
-                if (!hasSetThreadId &&
-                    chunk.responseMessageDto.threadId &&
-                    chunk.responseMessageDto.threadId !== currentThread?.id) {
-                    hasSetThreadId = true;
-                    const wasPlaceholderThread = currentThreadId === exports.PLACEHOLDER_THREAD.id;
-                    await switchCurrentThread(chunk.responseMessageDto.threadId, false);
-                    // If we're switching from placeholder to a real thread
-                    // this means a new thread was created, so add it to cache
-                    if (wasPlaceholderThread) {
-                        await addThreadToCache(chunk.responseMessageDto.threadId, contextKey);
-                        // Check if we should auto-generate name for the newly created thread
-                        maybeAutoGenerateThreadName(chunk.responseMessageDto.threadId, contextKey);
-                    }
-                }
-                if (!finalMessage) {
-                    finalMessage = chunk.responseMessageDto.component?.componentName
-                        ? (0, generate_component_1.renderComponentIntoMessage)(chunk.responseMessageDto, componentList)
-                        : chunk.responseMessageDto;
-                    await addThreadMessage(finalMessage, false);
-                }
-                else {
-                    // Handle streaming tool calls
-                    if (chunk.responseMessageDto.component?.toolCallRequest) {
-                        const { toolName } = chunk.responseMessageDto.component.toolCallRequest;
-                        const tool = toolName ? toolRegistry[toolName] : undefined;
-                        // if there is a tool call request on a component in a streaming chunk, and the tool
-                        // is marked as streamable by the correct annotation, handle it
-                        const isStreamable = tool?.annotations?.tamboStreamableHint ?? false;
-                        if (isStreamable) {
-                            // We're not paying attention to the tool call response here - we only want the
-                            // final tool call which is handled by the presence of a toolCallRequest on the
-                            // top-level responseMessageDto
-                            // see above: `if (chunk.responseMessageDto.toolCallRequest) { ... }`
-                            void (await (0, tool_caller_1.handleToolCall)(chunk.responseMessageDto.component.toolCallRequest, toolRegistry, onCallUnregisteredTool));
-                        }
-                    }
-                    // Capture previous message ID before updating finalMessage
-                    const previousMessageId = finalMessage.id;
-                    if (chunk.responseMessageDto.component?.componentName) {
-                        finalMessage = (0, generate_component_1.renderComponentIntoMessage)(chunk.responseMessageDto, componentList);
-                    }
-                    else {
-                        finalMessage = chunk.responseMessageDto;
-                    }
-                    // if we start getting a new message mid-stream, put the previous one on screen
-                    const isNewMessage = chunk.responseMessageDto.id !== previousMessageId;
-                    if (isNewMessage) {
-                        await addThreadMessage(finalMessage, false);
-                    }
-                    else {
-                        await updateThreadMessage(finalMessage.id, finalMessage, false);
-                    }
-                }
-            }
-        }
-        const completedThreadId = finalMessage?.threadId ?? threadId;
-        updateThreadStatus(completedThreadId, generate_component_response_1.GenerationStage.COMPLETE);
-        maybeAutoGenerateThreadName(completedThreadId, contextKey);
-        return (finalMessage ?? {
-            ...createEmptyMessage(completedThreadId),
-            content: [{ type: "text", text: "Error processing stream" }],
-        });
-    }, [
-        addThreadMessage,
-        addThreadToCache,
-        maybeAutoGenerateThreadName,
-        client,
-        componentList,
-        currentThread?.id,
-        currentThreadId,
-        onCallUnregisteredTool,
-        switchCurrentThread,
-        toolRegistry,
-        updateThreadMessage,
-        updateThreadStatus,
-    ]);
-    const sendThreadMessage = (0, react_1.useCallback)(async (message, options = {}) => {
-        setIgnoreResponse(false);
-        const { threadId = currentThreadId ?? placeholderThread.id, streamResponse = streaming, forceToolChoice, contextKey, additionalContext, content, } = options;
-        if (!streamResponse) {
-            throw new Error("Non-streaming mode is deprecated. Use streaming mode instead.");
-        }
-        updateThreadStatus(threadId, generate_component_response_1.GenerationStage.FETCHING_CONTEXT);
-        // Get additional context from enabled helpers
-        const helperContexts = await getAdditionalContext();
-        // Combine all contexts
-        const combinedContext = {
-            ...(additionalContext ?? {}),
-        };
-        // Add helper contexts to combinedContext
-        for (const helperContext of helperContexts) {
-            combinedContext[helperContext.name] = helperContext.context;
-        }
-        // Build and optimistically add user message (for revert on fail)
-        const optimiticMessageId = crypto.randomUUID();
-        const messageContent = content ?? [
-            { type: "text", text: message },
-        ];
-        await addThreadMessage({
-            content: messageContent,
-            renderedComponent: null,
-            role: "user",
-            threadId: threadId,
-            id: optimiticMessageId,
-            createdAt: new Date().toISOString(),
-            componentState: {},
-            additionalContext: combinedContext,
-        }, false);
-        maybeAutoGenerateThreadName(threadId, contextKey);
-        const availableComponents = (0, registry_1.getAvailableComponents)(componentList, toolRegistry, componentToolAssociations);
-        const unassociatedTools = (0, registry_1.getUnassociatedTools)(toolRegistry, componentToolAssociations);
-        // Track tool call counts for this message processing
-        const toolCallCounts = {};
-        const params = {
-            messageToAppend: {
-                content: messageContent,
-                role: "user",
-                additionalContext: combinedContext,
-            },
-            contextKey,
-            availableComponents: availableComponents,
-            clientTools: unassociatedTools.map((tool) => (0, registry_1.mapTamboToolToContextTool)(tool)),
-            forceToolChoice: forceToolChoice,
-            toolCallCounts,
-            ...(threadId === placeholderThread.id &&
-                initialMessages.length > 0 && {
-                initialMessages: initialMessages.map((msg) => ({
-                    content: msg.content,
-                    role: msg.role,
-                    additionalContext: msg.additionalContext,
-                })),
-            }),
-        };
-        let advanceStreamResponse;
-        try {
-            advanceStreamResponse = await (0, typescript_sdk_1.advanceStream)(client, params, threadId === placeholderThread.id ? undefined : threadId);
-        }
-        catch (error) {
-            updateThreadStatus(threadId, generate_component_response_1.GenerationStage.ERROR);
-            // Rollback the optimistic user message
-            setThreadMap((prev) => {
-                const thread = prev[threadId];
-                if (!thread)
-                    return prev;
-                return {
-                    ...prev,
-                    [threadId]: {
-                        ...thread,
-                        messages: thread.messages.filter((msg) => msg.id !== optimiticMessageId),
-                    },
-                };
-            });
-            throw error;
-        }
-        try {
-            const result = await handleAdvanceStream(advanceStreamResponse, params, threadId, contextKey);
-            return result;
-        }
-        catch (error) {
-            updateThreadStatus(threadId, generate_component_response_1.GenerationStage.ERROR);
-            // Rollback the optimistic user message
-            setThreadMap((prev) => {
-                const thread = prev[threadId];
-                if (!thread)
-                    return prev;
-                return {
-                    ...prev,
-                    [threadId]: {
-                        ...thread,
-                        messages: thread.messages.filter((msg) => msg.id !== optimiticMessageId),
-                    },
-                };
-            });
-            throw error;
-        }
-    }, [
-        componentList,
-        toolRegistry,
-        componentToolAssociations,
-        currentThreadId,
-        addThreadMessage,
-        client,
-        updateThreadStatus,
-        handleAdvanceStream,
-        streaming,
-        getAdditionalContext,
-        placeholderThread.id,
-        initialMessages,
-        maybeAutoGenerateThreadName,
-    ]);
-    return (react_1.default.createElement(exports.TamboThreadContext.Provider, { value: {
-            thread: currentThread,
-            currentThreadId,
-            currentThread,
-            threadMap,
-            setThreadMap,
-            switchCurrentThread,
-            startNewThread,
-            updateThreadName,
-            generateThreadName,
-            addThreadMessage,
-            updateThreadMessage,
-            streaming,
-            cancel,
-            sendThreadMessage,
-            contextKey,
-        } },
-        react_1.default.createElement(exports.TamboGenerationStageProvider, { generationStage: currentGenerationStage, statusMessage: currentStatusMessage }, children)));
-};
-exports.TamboThreadProvider = TamboThreadProvider;
-/**
- * The useTamboGenerationStage hook provides access to the generation stage context
- * to the descendants of the TamboThreadProvider.
- * @returns The generation stage context
- */
-const useTamboGenerationStage = () => {
-    const generationStageContext = (0, react_1.useContext)(TamboGenerationStageContext);
-    if (generationStageContext === undefined) {
-        throw new Error("useTamboGenerationStage must be used within a TamboThreadProvider");
-    }
-    return generationStageContext;
-};
-exports.useTamboGenerationStage = useTamboGenerationStage;
-/**
- * The useTamboThread hook provides access to the current thread context
- * to the descendants of the TamboThreadProvider.
- * @returns All state and actions for the current thread
- */
-const useTamboThread = () => {
-    const threadContext = (0, react_1.useContext)(exports.TamboThreadContext);
-    const generationStageContext = (0, react_1.useContext)(TamboGenerationStageContext);
-    if (threadContext === undefined) {
-        throw new Error("useTamboThread must be used within a TamboThreadProvider");
-    }
-    if (generationStageContext === undefined) {
-        throw new Error("useTamboThread must be used within a TamboThreadProvider");
-    }
-    return {
-        ...threadContext,
-        ...generationStageContext,
-    };
-};
-exports.useTamboThread = useTamboThread;
-async function convertToolResponse(toolCallResponse) {
-    // If the tool call errored, surface that as text so the model reliably sees the error
-    if (toolCallResponse.error) {
-        return [{ type: "text", text: (0, content_parts_1.toText)(toolCallResponse.result) }];
-    }
-    // Use custom transform when available
-    if (toolCallResponse.tamboTool?.transformToContent) {
-        return await toolCallResponse.tamboTool.transformToContent(
-        // result shape is user-defined; let the transform decide how to handle it
-        toolCallResponse.result);
-    }
-    // Default fallback to stringified text
-    return [{ type: "text", text: (0, content_parts_1.toText)(toolCallResponse.result) }];
-}
-/**
- * Create a placeholder/cancelled message with default values
- * @param threadId - ID of the thread the message belongs to
- * @returns An empty TamboThreadMessage
- */
-function createEmptyMessage(threadId) {
-    return {
-        threadId,
-        content: [{ type: "text", text: "" }],
-        role: "assistant",
-        createdAt: new Date().toISOString(),
-        id: crypto.randomUUID(),
-        componentState: {},
-    };
-}
-//# sourceMappingURL=tambo-thread-provider.js.map