@tambo-ai/react 0.12.0

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

@@ -0,0 +1,379 @@
+"use client";
+import { advanceStream } from "@tambo-ai/typescript-sdk";
+import React, { createContext, useCallback, useContext, useEffect, useMemo, useState, } from "react";
+import { GenerationStage, } from "../model/generate-component-response";
+import { renderComponentIntoMessage } from "../util/generate-component";
+import { getAvailableComponents, getClientContext } from "../util/registry";
+import { handleToolCall } from "../util/tool-caller";
+import { useTamboClient } from "./tambo-client-provider";
+import { useTamboRegistry } from "./tambo-registry-provider";
+/** 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.
+ */
+export const PLACEHOLDER_THREAD = {
+    id: "placeholder",
+    messages: [],
+    createdAt: "",
+    projectId: "",
+    updatedAt: "",
+    metadata: {},
+};
+export const TamboThreadContext = createContext({
+    thread: PLACEHOLDER_THREAD,
+    switchCurrentThread: () => {
+        throw new Error("switchCurrentThread not implemented");
+    },
+    addThreadMessage: () => {
+        throw new Error("updateThreadMessageHistory not implemented");
+    },
+    setLastThreadStatus: () => {
+        throw new Error("setLastThreadStatus not implemented");
+    },
+    inputValue: "",
+    setInputValue: () => {
+        throw new Error("setInputValue not implemented");
+    },
+    updateThreadMessage: () => {
+        throw new Error("updateThreadMessage not implemented");
+    },
+    sendThreadMessage: () => {
+        throw new Error("advance not implemented");
+    },
+});
+export const TamboThreadProvider = ({ children, }) => {
+    const [threadMap, setThreadMap] = useState({
+        [PLACEHOLDER_THREAD.id]: PLACEHOLDER_THREAD,
+    });
+    const client = useTamboClient();
+    const { componentList, toolRegistry, componentToolAssociations } = useTamboRegistry();
+    const [inputValue, setInputValue] = useState("");
+    const [currentThreadId, setCurrentThreadId] = useState(PLACEHOLDER_THREAD.id);
+    const [unresolvedThreadId, setUnresolvedThreadId] = useState(PLACEHOLDER_THREAD.id);
+    const currentThread = threadMap[currentThreadId];
+    // Use existing messages from the current thread to avoid re-generating any components
+    const currentMessageCache = useMemo(() => {
+        const messageCache = new Map();
+        if (currentThread) {
+            for (const message of currentThread.messages) {
+                messageCache.set(message.id, message);
+            }
+        }
+        return messageCache;
+    }, [currentThread]);
+    const fetchThread = 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 = renderComponentIntoMessage(message, componentList);
+                    return messageWithComponent;
+                }
+                return message;
+            }),
+        };
+        setThreadMap((prevMap) => ({
+            ...prevMap,
+            [threadId]: threadWithRenderedComponents,
+        }));
+    }, [client.beta.threads, componentList, currentMessageCache]);
+    useEffect(() => {
+        if (currentThreadId &&
+            currentThreadId !== PLACEHOLDER_THREAD.id &&
+            !threadMap[currentThreadId]) {
+            fetchThread(currentThreadId);
+        }
+    }, [currentThreadId, fetchThread, threadMap]);
+    const addThreadMessage = 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,
+            additionalContext: message.role === "user" ? getClientContext() : undefined,
+            createdAt,
+        };
+        const threadId = message.threadId;
+        // optimistically update the thread in the local state
+        setThreadMap((prevMap) => {
+            if (!threadId) {
+                return prevMap;
+            }
+            const prevMessages = prevMap[threadId]?.messages || [];
+            return {
+                ...prevMap,
+                [threadId]: {
+                    ...prevMap[threadId],
+                    messages: [...prevMessages, chatMessage],
+                },
+            };
+        });
+        if (sendToServer) {
+            // TODO: if this fails, we need to revert the local state update
+            await client.beta.threads.messages.create(currentThreadId, {
+                content: message.content,
+                role: message.role,
+                // additionalContext: chatMessage.additionalContext,
+            });
+        }
+        const updatedMessageHistory = [...currentThread.messages, chatMessage];
+        return updatedMessageHistory;
+    }, [client.beta.threads.messages, currentThread, currentThreadId]);
+    const updateThreadMessage = useCallback(async (id, message, sendToServer = true, createdAt = new Date().toISOString()) => {
+        const chatMessage = {
+            ...message,
+            createdAt,
+        };
+        setThreadMap((prevMap) => {
+            if (!message.threadId) {
+                return prevMap;
+            }
+            const prevMessages = prevMap[message.threadId]?.messages || [];
+            const updatedMessages = prevMessages.map((msg) => {
+                if (msg.id === id) {
+                    return chatMessage;
+                }
+                return msg;
+            });
+            return {
+                ...prevMap,
+                [message.threadId]: {
+                    ...prevMap[message.threadId],
+                    messages: updatedMessages,
+                },
+            };
+        });
+        if (sendToServer) {
+            // TODO: if this fails, we need to revert the local state update
+            await client.beta.threads.messages.create(currentThreadId, {
+                content: message.content,
+                role: message.role,
+                // additionalContext: chatMessage.additionalContext,
+            });
+        }
+    }, [client.beta.threads.messages, currentThreadId]);
+    useEffect(() => {
+        if (unresolvedThreadId && currentThreadId !== unresolvedThreadId) {
+            setThreadMap((prevMap) => {
+                const unresolvedThread = prevMap[unresolvedThreadId];
+                const currentThread = prevMap[currentThreadId];
+                return {
+                    ...prevMap,
+                    [unresolvedThreadId]: PLACEHOLDER_THREAD,
+                    [currentThreadId]: {
+                        ...currentThread,
+                        id: currentThreadId,
+                        messages: [...unresolvedThread.messages, ...currentThread.messages],
+                    },
+                };
+            });
+            setUnresolvedThreadId(undefined);
+        }
+    }, [currentThreadId, unresolvedThreadId]);
+    const switchCurrentThread = useCallback(async (threadId) => {
+        if (threadId === PLACEHOLDER_THREAD.id) {
+            console.warn("Switching to placeholder thread, may be a bug");
+            return;
+        }
+        setCurrentThreadId(threadId);
+        if (!threadMap[threadId]) {
+            setThreadMap((prevMap) => ({
+                ...prevMap,
+                [threadId]: {
+                    ...PLACEHOLDER_THREAD,
+                    id: threadId,
+                },
+            }));
+        }
+        await fetchThread(threadId);
+    }, [fetchThread, threadMap]);
+    const setLastThreadStatus = (status) => {
+        setThreadMap((prevMap) => {
+            if (!currentThreadId) {
+                return prevMap;
+            }
+            const headMessages = prevMap[currentThreadId].messages.slice(0, -1);
+            const lastMessage = prevMap[currentThreadId].messages[prevMap[currentThreadId].messages.length - 1];
+            const updatedLastMessage = {
+                ...lastMessage,
+                status,
+            };
+            return {
+                ...prevMap,
+                [currentThreadId]: {
+                    ...prevMap[currentThreadId],
+                    messages: [...headMessages, updatedLastMessage],
+                },
+            };
+        });
+    };
+    const updateThreadStatus = useCallback((stage) => {
+        setThreadMap((prevMap) => {
+            return {
+                ...prevMap,
+                [currentThreadId]: {
+                    ...prevMap[currentThreadId],
+                    generationStage: stage,
+                },
+            };
+        });
+    }, [currentThreadId]);
+    const handleAdvanceStream = useCallback(async (stream, params) => {
+        let finalMessage;
+        let hasSetThreadId = false;
+        updateThreadStatus(GenerationStage.STREAMING_RESPONSE);
+        for await (const chunk of stream) {
+            if (chunk.responseMessageDto.toolCallRequest) {
+                updateThreadStatus(GenerationStage.FETCHING_CONTEXT);
+                const toolCallResponse = await handleToolCall(chunk.responseMessageDto, toolRegistry);
+                const toolCallResponseParams = {
+                    ...params,
+                    messageToAppend: {
+                        content: [{ type: "text", text: "tool response" }],
+                        role: "tool",
+                        actionType: "tool_response",
+                        toolResponse: toolCallResponse,
+                        component: chunk.responseMessageDto.component,
+                    },
+                };
+                updateThreadStatus(GenerationStage.STREAMING_RESPONSE);
+                const toolCallResponseStream = await advanceStream(client, toolCallResponseParams, chunk.responseMessageDto.threadId);
+                return handleAdvanceStream(toolCallResponseStream, toolCallResponseParams);
+            }
+            else {
+                if (!hasSetThreadId &&
+                    chunk.responseMessageDto.threadId &&
+                    chunk.responseMessageDto.threadId !== currentThread?.id) {
+                    hasSetThreadId = true;
+                    switchCurrentThread(chunk.responseMessageDto.threadId);
+                }
+                if (!finalMessage) {
+                    finalMessage = chunk.responseMessageDto.component?.componentName
+                        ? renderComponentIntoMessage(chunk.responseMessageDto, componentList)
+                        : chunk.responseMessageDto;
+                    addThreadMessage(finalMessage, false);
+                }
+                else {
+                    const previousId = finalMessage.id;
+                    finalMessage = chunk.responseMessageDto.component?.componentName
+                        ? renderComponentIntoMessage(chunk.responseMessageDto, componentList)
+                        : chunk.responseMessageDto;
+                    updateThreadMessage(previousId, finalMessage, false);
+                }
+            }
+        }
+        updateThreadStatus(GenerationStage.COMPLETE);
+        return (finalMessage ?? {
+            threadId: "",
+            content: [{ type: "text", text: `Error processing stream` }],
+            role: "hydra",
+            createdAt: new Date().toISOString(),
+            id: crypto.randomUUID(),
+        });
+    }, [
+        toolRegistry,
+        client,
+        currentThread?.id,
+        switchCurrentThread,
+        componentList,
+        addThreadMessage,
+        updateThreadMessage,
+    ]);
+    const sendThreadMessage = useCallback(async (message, options = {}) => {
+        const { threadId, streamResponse } = options;
+        const currentThreadId = threadId ?? currentThread?.id;
+        if (currentThreadId !== PLACEHOLDER_THREAD.id) {
+            await switchCurrentThread(currentThreadId);
+        }
+        updateThreadStatus(GenerationStage.CHOOSING_COMPONENT);
+        addThreadMessage({
+            content: [{ type: "text", text: message }],
+            renderedComponent: null,
+            role: "user",
+            threadId: currentThread.id,
+            id: crypto.randomUUID(),
+            createdAt: new Date().toISOString(),
+        }, false);
+        const availableComponents = getAvailableComponents(componentList, toolRegistry, componentToolAssociations);
+        const params = {
+            messageToAppend: {
+                content: [{ type: "text", text: message }],
+                role: "user",
+            },
+            contextKey: options.contextKey,
+            availableComponents: availableComponents,
+        };
+        if (streamResponse) {
+            const advanceStreamResponse = await advanceStream(client, params, currentThreadId === PLACEHOLDER_THREAD.id
+                ? undefined
+                : currentThreadId);
+            return handleAdvanceStream(advanceStreamResponse, params);
+        }
+        let advanceResponse = await (currentThreadId === PLACEHOLDER_THREAD.id
+            ? client.beta.threads.advance(params)
+            : client.beta.threads.advanceById(currentThreadId, params));
+        //handle tool calls
+        while (advanceResponse.responseMessageDto.toolCallRequest) {
+            updateThreadStatus(GenerationStage.FETCHING_CONTEXT);
+            const toolCallResponse = await handleToolCall(advanceResponse.responseMessageDto, toolRegistry);
+            const toolCallResponseParams = {
+                ...params,
+                messageToAppend: {
+                    ...params.messageToAppend,
+                    content: [{ type: "text", text: "tool response" }],
+                    role: "tool",
+                    actionType: "tool_response",
+                    toolResponse: toolCallResponse,
+                    component: advanceResponse.responseMessageDto.component,
+                },
+            };
+            updateThreadStatus(GenerationStage.HYDRATING_COMPONENT);
+            advanceResponse = await client.beta.threads.advanceById(advanceResponse.responseMessageDto.threadId, toolCallResponseParams);
+        }
+        const finalMessage = advanceResponse.responseMessageDto.component
+            ?.componentName
+            ? renderComponentIntoMessage(advanceResponse.responseMessageDto, componentList)
+            : advanceResponse.responseMessageDto;
+        await switchCurrentThread(advanceResponse.responseMessageDto.threadId);
+        updateThreadStatus(GenerationStage.COMPLETE);
+        return finalMessage;
+    }, [
+        componentList,
+        toolRegistry,
+        componentToolAssociations,
+        currentThread,
+        switchCurrentThread,
+        addThreadMessage,
+        client,
+        handleAdvanceStream,
+    ]);
+    return (React.createElement(TamboThreadContext.Provider, { value: {
+            thread: currentThread,
+            switchCurrentThread,
+            addThreadMessage,
+            updateThreadMessage,
+            setLastThreadStatus,
+            inputValue,
+            setInputValue,
+            sendThreadMessage,
+        } }, children));
+};
+export const useTamboThread = () => {
+    const context = useContext(TamboThreadContext);
+    if (context === undefined) {
+        throw new Error("useTamboThread must be used within a TamboThreadProvider");
+    }
+    return context;
+};
+//# sourceMappingURL=tambo-thread-provider.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-thread-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-thread-provider.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;AACb,OAAgB,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,KAAK,EAAE,EACZ,aAAa,EAEb,WAAW,EACX,UAAU,EACV,SAAS,EACT,OAAO,EACP,QAAQ,GACT,MAAM,OAAO,CAAC;AACf,OAAO,EACL,eAAe,GAEhB,MAAM,sCAAsC,CAAC;AAE9C,OAAO,EAAE,0BAA0B,EAAE,MAAM,4BAA4B,CAAC;AACxE,OAAO,EAAE,sBAAsB,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAC5E,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AA2B7D;;;;;GAKG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAgB;IAC7C,EAAE,EAAE,aAAa;IACjB,QAAQ,EAAE,EAAE;IACZ,SAAS,EAAE,EAAE;IACb,SAAS,EAAE,EAAE;IACb,SAAS,EAAE,EAAE;IACb,QAAQ,EAAE,EAAE;CACb,CAAC;AAEF,MAAM,CAAC,MAAM,kBAAkB,GAAG,aAAa,CAA0B;IACvE,MAAM,EAAE,kBAAkB;IAC1B,mBAAmB,EAAE,GAAG,EAAE;QACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD,gBAAgB,EAAE,GAAG,EAAE;QACrB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAChE,CAAC;IACD,mBAAmB,EAAE,GAAG,EAAE;QACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD,UAAU,EAAE,EAAE;IACd,aAAa,EAAE,GAAG,EAAE;QAClB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IACD,mBAAmB,EAAE,GAAG,EAAE;QACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD,iBAAiB,EAAE,GAAG,EAAE;QACtB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;CACF,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,mBAAmB,GAAgC,CAAC,EAC/D,QAAQ,GACT,EAAE,EAAE;IACH,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAA8B;QACtE,CAAC,kBAAkB,CAAC,EAAE,CAAC,EAAE,kBAAkB;KAC5C,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;IAChC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,gBAAgB,EAAE,CAAC;IACrB,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEjD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CACpD,kBAAkB,CAAC,EAAE,CACtB,CAAC;IACF,MAAM,CAAC,kBAAkB,EAAE,qBAAqB,CAAC,GAAG,QAAQ,CAE1D,kBAAkB,CAAC,EAAE,CAAC,CAAC;IACzB,MAAM,aAAa,GAA4B,SAAS,CAAC,eAAe,CAAC,CAAC;IAC1E,sFAAsF;IACtF,MAAM,mBAAmB,GAAG,OAAO,CAAC,GAAG,EAAE;QACvC,MAAM,YAAY,GAAG,IAAI,GAAG,EAA8B,CAAC;QAC3D,IAAI,aAAa,EAAE,CAAC;YAClB,KAAK,MAAM,OAAO,IAAI,aAAa,CAAC,QAAQ,EAAE,CAAC;gBAC7C,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC;IAEpB,MAAM,WAAW,GAAG,WAAW,CAC7B,KAAK,EAAE,QAAgB,EAAE,EAAE;QACzB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC5D,MAAM,4BAA4B,GAAG;YACnC,GAAG,MAAM;YACT,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;gBACxC,IAAI,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;oBACxC,MAAM,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBAC5D,OAAO;wBACL,GAAG,eAAe;wBAClB,GAAG,OAAO;qBACX,CAAC;gBACJ,CAAC;gBACD,IAAI,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC;oBACrC,MAAM,oBAAoB,GAAG,0BAA0B,CACrD,OAAO,EACP,aAAa,CACd,CAAC;oBACF,OAAO,oBAAoB,CAAC;gBAC9B,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC;SACH,CAAC;QAEF,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACzB,GAAG,OAAO;YACV,CAAC,QAAQ,CAAC,EAAE,4BAA4B;SACzC,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAC1D,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,IACE,eAAe;YACf,eAAe,KAAK,kBAAkB,CAAC,EAAE;YACzC,CAAC,SAAS,CAAC,eAAe,CAAC,EAC3B,CAAC;YACD,WAAW,CAAC,eAAe,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC,EAAE,CAAC,eAAe,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC;IAE9C,MAAM,gBAAgB,GAAG,WAAW,CAClC,KAAK,EACH,OAA2B,EAC3B,YAAY,GAAG,IAAI,EACnB,YAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAC5C,EAAE;QACF,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;YACvE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,WAAW,GAEb;YACF,GAAG,OAAO;YACV,iBAAiB,EACf,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,SAAS;YAC1D,SAAS;SACV,CAAC;QACF,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAClC,sDAAsD;QACtD,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;YACvD,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,OAAO,CAAC,QAAQ,CAAC;oBACpB,QAAQ,EAAE,CAAC,GAAG,YAAY,EAAE,WAAW,CAAC;iBACzC;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,YAAY,EAAE,CAAC;YACjB,gEAAgE;YAChE,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,eAAe,EAAE;gBACzD,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,oDAAoD;aACrD,CAAC,CAAC;QACL,CAAC;QACD,MAAM,qBAAqB,GAAG,CAAC,GAAG,aAAa,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAEvE,OAAO,qBAAqB,CAAC;IAC/B,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,aAAa,EAAE,eAAe,CAAC,CAC/D,CAAC;IAEF,MAAM,mBAAmB,GAAG,WAAW,CACrC,KAAK,EACH,EAAU,EACV,OAA2B,EAC3B,YAAY,GAAG,IAAI,EACnB,YAAoB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAC5C,EAAE;QACF,MAAM,WAAW,GAAuB;YACtC,GAAG,OAAO;YACV,SAAS;SACV,CAAC;QAEF,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACtB,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;YAC/D,MAAM,eAAe,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC/C,IAAI,GAAG,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;oBAClB,OAAO,WAAW,CAAC;gBACrB,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;YACH,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;oBAClB,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;oBAC5B,QAAQ,EAAE,eAAe;iBAC1B;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,YAAY,EAAE,CAAC;YACjB,gEAAgE;YAChE,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,eAAe,EAAE;gBACzD,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,oDAAoD;aACrD,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EACD,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,eAAe,CAAC,CAChD,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,kBAAkB,IAAI,eAAe,KAAK,kBAAkB,EAAE,CAAC;YACjE,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;gBACvB,MAAM,gBAAgB,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;gBACrD,MAAM,aAAa,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC;gBAC/C,OAAO;oBACL,GAAG,OAAO;oBACV,CAAC,kBAAkB,CAAC,EAAE,kBAAkB;oBACxC,CAAC,eAAe,CAAC,EAAE;wBACjB,GAAG,aAAa;wBAChB,EAAE,EAAE,eAAe;wBACnB,QAAQ,EAAE,CAAC,GAAG,gBAAgB,CAAC,QAAQ,EAAE,GAAG,aAAa,CAAC,QAAQ,CAAC;qBACpE;iBACF,CAAC;YACJ,CAAC,CAAC,CAAC;YACH,qBAAqB,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;IACH,CAAC,EAAE,CAAC,eAAe,EAAE,kBAAkB,CAAC,CAAC,CAAC;IAE1C,MAAM,mBAAmB,GAAG,WAAW,CACrC,KAAK,EAAE,QAAgB,EAAE,EAAE;QACzB,IAAI,QAAQ,KAAK,kBAAkB,CAAC,EAAE,EAAE,CAAC;YACvC,OAAO,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;YAC9D,OAAO;QACT,CAAC;QACD,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAC7B,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzB,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBACzB,GAAG,OAAO;gBACV,CAAC,QAAQ,CAAC,EAAE;oBACV,GAAG,kBAAkB;oBACrB,EAAE,EAAE,QAAQ;iBACb;aACF,CAAC,CAAC,CAAC;QACN,CAAC;QACD,MAAM,WAAW,CAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC,EACD,CAAC,WAAW,EAAE,SAAS,CAAC,CACzB,CAAC;IAEF,MAAM,mBAAmB,GAAG,CAAC,MAAuB,EAAE,EAAE;QACtD,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACpE,MAAM,WAAW,GACf,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,CAC/B,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAC7C,CAAC;YACJ,MAAM,kBAAkB,GAAG;gBACzB,GAAG,WAAW;gBACd,MAAM;aACP,CAAC;YACF,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,eAAe,CAAC,EAAE;oBACjB,GAAG,OAAO,CAAC,eAAe,CAAC;oBAC3B,QAAQ,EAAE,CAAC,GAAG,YAAY,EAAE,kBAAkB,CAAC;iBAChD;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,kBAAkB,GAAG,WAAW,CACpC,CAAC,KAAsB,EAAE,EAAE;QACzB,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;YACvB,OAAO;gBACL,GAAG,OAAO;gBACV,CAAC,eAAe,CAAC,EAAE;oBACjB,GAAG,OAAO,CAAC,eAAe,CAAC;oBAC3B,eAAe,EAAE,KAAK;iBACvB;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EACD,CAAC,eAAe,CAAC,CAClB,CAAC;IAEF,MAAM,mBAAmB,GAAG,WAAW,CACrC,KAAK,EACH,MAAiE,EACjE,MAAgD,EACnB,EAAE;QAC/B,IAAI,YAA4C,CAAC;QACjD,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,kBAAkB,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;QACvD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,KAAK,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;gBAC7C,kBAAkB,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;gBACrD,MAAM,gBAAgB,GAAG,MAAM,cAAc,CAC3C,KAAK,CAAC,kBAAkB,EACxB,YAAY,CACb,CAAC;gBACF,MAAM,sBAAsB,GAC1B;oBACE,GAAG,MAAM;oBACT,eAAe,EAAE;wBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;wBAClD,IAAI,EAAE,MAAM;wBACZ,UAAU,EAAE,eAAe;wBAC3B,YAAY,EAAE,gBAAgB;wBAC9B,SAAS,EAAE,KAAK,CAAC,kBAAkB,CAAC,SAAS;qBAC9C;iBACF,CAAC;gBACJ,kBAAkB,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;gBACvD,MAAM,sBAAsB,GAAG,MAAM,aAAa,CAChD,MAAM,EACN,sBAAsB,EACtB,KAAK,CAAC,kBAAkB,CAAC,QAAQ,CAClC,CAAC;gBACF,OAAO,mBAAmB,CACxB,sBAAsB,EACtB,sBAAsB,CACvB,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,IACE,CAAC,cAAc;oBACf,KAAK,CAAC,kBAAkB,CAAC,QAAQ;oBACjC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,KAAK,aAAa,EAAE,EAAE,EACvD,CAAC;oBACD,cAAc,GAAG,IAAI,CAAC;oBACtB,mBAAmB,CAAC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;gBACzD,CAAC;gBAED,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,aAAa;wBAC9D,CAAC,CAAC,0BAA0B,CACxB,KAAK,CAAC,kBAAkB,EACxB,aAAa,CACd;wBACH,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC;oBAC7B,gBAAgB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;gBACxC,CAAC;qBAAM,CAAC;oBACN,MAAM,UAAU,GAAG,YAAY,CAAC,EAAE,CAAC;oBACnC,YAAY,GAAG,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,aAAa;wBAC9D,CAAC,CAAC,0BAA0B,CACxB,KAAK,CAAC,kBAAkB,EACxB,aAAa,CACd;wBACH,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC;oBAC7B,mBAAmB,CAAC,UAAU,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;QACH,CAAC;QAED,kBAAkB,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC7C,OAAO,CACL,YAAY,IAAI;YACd,QAAQ,EAAE,EAAE;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,yBAAyB,EAAE,CAAC;YAC5D,IAAI,EAAE,OAAO;YACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;SACxB,CACF,CAAC;IACJ,CAAC,EACD;QACE,YAAY;QACZ,MAAM;QACN,aAAa,EAAE,EAAE;QACjB,mBAAmB;QACnB,aAAa;QACb,gBAAgB;QAChB,mBAAmB;KACpB,CACF,CAAC;IAEF,MAAM,iBAAiB,GAAG,WAAW,CACnC,KAAK,EACH,OAAe,EACf,UAII,EAAE,EACuB,EAAE;QAC/B,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC;QAC7C,MAAM,eAAe,GAAG,QAAQ,IAAI,aAAa,EAAE,EAAE,CAAC;QAEtD,IAAI,eAAe,KAAK,kBAAkB,CAAC,EAAE,EAAE,CAAC;YAC9C,MAAM,mBAAmB,CAAC,eAAe,CAAC,CAAC;QAC7C,CAAC;QAED,kBAAkB,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAEvD,gBAAgB,CACd;YACE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;YAC1C,iBAAiB,EAAE,IAAI;YACvB,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE,aAAa,CAAC,EAAE;YAC1B,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,EACD,KAAK,CACN,CAAC;QAEF,MAAM,mBAAmB,GAAG,sBAAsB,CAChD,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;QACF,MAAM,MAAM,GAA6C;YACvD,eAAe,EAAE;gBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,IAAI,EAAE,MAAM;aACb;YACD,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,mBAAmB,EAAE,mBAAmB;SACzC,CAAC;QAEF,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,qBAAqB,GAAG,MAAM,aAAa,CAC/C,MAAM,EACN,MAAM,EACN,eAAe,KAAK,kBAAkB,CAAC,EAAE;gBACvC,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,eAAe,CACpB,CAAC;YACF,OAAO,mBAAmB,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;QAC5D,CAAC;QACD,IAAI,eAAe,GAAG,MAAM,CAAC,eAAe,KAAK,kBAAkB,CAAC,EAAE;YACpE,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YACrC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC,CAAC;QAE9D,mBAAmB;QACnB,OAAO,eAAe,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;YAC1D,kBAAkB,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;YACrD,MAAM,gBAAgB,GAAG,MAAM,cAAc,CAC3C,eAAe,CAAC,kBAAkB,EAClC,YAAY,CACb,CAAC;YACF,MAAM,sBAAsB,GAC1B;gBACE,GAAG,MAAM;gBACT,eAAe,EAAE;oBACf,GAAG,MAAM,CAAC,eAAe;oBACzB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;oBAClD,IAAI,EAAE,MAAM;oBACZ,UAAU,EAAE,eAAe;oBAC3B,YAAY,EAAE,gBAAgB;oBAC9B,SAAS,EAAE,eAAe,CAAC,kBAAkB,CAAC,SAAS;iBACxD;aACF,CAAC;YACJ,kBAAkB,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;YACxD,eAAe,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CACrD,eAAe,CAAC,kBAAkB,CAAC,QAAQ,EAC3C,sBAAsB,CACvB,CAAC;QACJ,CAAC;QAED,MAAM,YAAY,GAAG,eAAe,CAAC,kBAAkB,CAAC,SAAS;YAC/D,EAAE,aAAa;YACf,CAAC,CAAC,0BAA0B,CACxB,eAAe,CAAC,kBAAkB,EAClC,aAAa,CACd;YACH,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC;QACvC,MAAM,mBAAmB,CAAC,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACvE,kBAAkB,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC7C,OAAO,YAAY,CAAC;IACtB,CAAC,EACD;QACE,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,aAAa;QACb,mBAAmB;QACnB,gBAAgB;QAChB,MAAM;QACN,mBAAmB;KACpB,CACF,CAAC;IAEF,OAAO,CACL,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;YACL,MAAM,EAAE,aAAa;YACrB,mBAAmB;YACnB,gBAAgB;YAChB,mBAAmB;YACnB,mBAAmB;YACnB,UAAU;YACV,aAAa;YACb,iBAAiB;SAClB,IAEA,QAAQ,CACmB,CAC/B,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,GAAG,EAAE;IACjC,MAAM,OAAO,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;IAC/C,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAC9E,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC","sourcesContent":["\"use client\";\nimport TamboAI, { advanceStream } from \"@tambo-ai/typescript-sdk\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useMemo,\n  useState,\n} from \"react\";\nimport {\n  GenerationStage,\n  TamboThreadMessage,\n} from \"../model/generate-component-response\";\nimport { TamboThread } from \"../model/tambo-thread\";\nimport { renderComponentIntoMessage } from \"../util/generate-component\";\nimport { getAvailableComponents, getClientContext } from \"../util/registry\";\nimport { handleToolCall } from \"../util/tool-caller\";\nimport { useTamboClient } from \"./tambo-client-provider\";\nimport { useTamboRegistry } from \"./tambo-registry-provider\";\n\nexport interface TamboThreadContextProps {\n  thread: TamboThread;\n  switchCurrentThread: (threadId: string) => void;\n  addThreadMessage: (\n    message: TamboThreadMessage,\n    sendToServer: boolean,\n  ) => Promise<TamboAI.Beta.Threads.ThreadMessage[]>;\n  updateThreadMessage: (\n    id: string,\n    message: TamboThreadMessage,\n    sendToServer: boolean,\n  ) => Promise<void>;\n  setLastThreadStatus: (status: GenerationStage) => void;\n  inputValue: string;\n  setInputValue: (value: string) => void;\n  sendThreadMessage: (\n    message: string,\n    options?: {\n      threadId?: string;\n      streamResponse?: boolean;\n      contextKey?: string;\n    },\n  ) => Promise<TamboThreadMessage>;\n}\n\n/** This is a stub entry for when the thread is not yet created, the first time\n * the user sends a message\n *\n * Note that the consumer needs to be careful never to send `PLACEHOLDER_THREAD.id` to the server,\n * as this doesn't really exist on the server side.\n */\nexport const PLACEHOLDER_THREAD: TamboThread = {\n  id: \"placeholder\",\n  messages: [],\n  createdAt: \"\",\n  projectId: \"\",\n  updatedAt: \"\",\n  metadata: {},\n};\n\nexport const TamboThreadContext = createContext<TamboThreadContextProps>({\n  thread: PLACEHOLDER_THREAD,\n  switchCurrentThread: () => {\n    throw new Error(\"switchCurrentThread not implemented\");\n  },\n  addThreadMessage: () => {\n    throw new Error(\"updateThreadMessageHistory not implemented\");\n  },\n  setLastThreadStatus: () => {\n    throw new Error(\"setLastThreadStatus not implemented\");\n  },\n  inputValue: \"\",\n  setInputValue: () => {\n    throw new Error(\"setInputValue not implemented\");\n  },\n  updateThreadMessage: () => {\n    throw new Error(\"updateThreadMessage not implemented\");\n  },\n  sendThreadMessage: () => {\n    throw new Error(\"advance not implemented\");\n  },\n});\n\nexport const TamboThreadProvider: React.FC<PropsWithChildren> = ({\n  children,\n}) => {\n  const [threadMap, setThreadMap] = useState<Record<string, TamboThread>>({\n    [PLACEHOLDER_THREAD.id]: PLACEHOLDER_THREAD,\n  });\n  const client = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n  const [inputValue, setInputValue] = useState(\"\");\n\n  const [currentThreadId, setCurrentThreadId] = useState<string>(\n    PLACEHOLDER_THREAD.id,\n  );\n  const [unresolvedThreadId, setUnresolvedThreadId] = useState<\n    string | undefined\n  >(PLACEHOLDER_THREAD.id);\n  const currentThread: TamboThread | undefined = threadMap[currentThreadId];\n  // Use existing messages from the current thread to avoid re-generating any components\n  const currentMessageCache = useMemo(() => {\n    const messageCache = new Map<string, TamboThreadMessage>();\n    if (currentThread) {\n      for (const message of currentThread.messages) {\n        messageCache.set(message.id, message);\n      }\n    }\n    return messageCache;\n  }, [currentThread]);\n\n  const fetchThread = useCallback(\n    async (threadId: string) => {\n      const thread = await client.beta.threads.retrieve(threadId);\n      const threadWithRenderedComponents = {\n        ...thread,\n        messages: thread.messages.map((message) => {\n          if (currentMessageCache.has(message.id)) {\n            const renderedMessage = currentMessageCache.get(message.id);\n            return {\n              ...renderedMessage,\n              ...message,\n            };\n          }\n          if (message.component?.componentName) {\n            const messageWithComponent = renderComponentIntoMessage(\n              message,\n              componentList,\n            );\n            return messageWithComponent;\n          }\n          return message;\n        }),\n      };\n\n      setThreadMap((prevMap) => ({\n        ...prevMap,\n        [threadId]: threadWithRenderedComponents,\n      }));\n    },\n    [client.beta.threads, componentList, currentMessageCache],\n  );\n\n  useEffect(() => {\n    if (\n      currentThreadId &&\n      currentThreadId !== PLACEHOLDER_THREAD.id &&\n      !threadMap[currentThreadId]\n    ) {\n      fetchThread(currentThreadId);\n    }\n  }, [currentThreadId, fetchThread, threadMap]);\n\n  const addThreadMessage = useCallback(\n    async (\n      message: TamboThreadMessage,\n      sendToServer = true,\n      createdAt: string = new Date().toISOString(),\n    ) => {\n      if (!currentThread) {\n        console.warn(\"Cannot add messages if we do not have a current thread\");\n        return [];\n      }\n      const chatMessage: TamboThreadMessage & {\n        additionalContext?: string;\n      } = {\n        ...message,\n        additionalContext:\n          message.role === \"user\" ? getClientContext() : undefined,\n        createdAt,\n      };\n      const threadId = message.threadId;\n      // optimistically update the thread in the local state\n      setThreadMap((prevMap) => {\n        if (!threadId) {\n          return prevMap;\n        }\n        const prevMessages = prevMap[threadId]?.messages || [];\n        return {\n          ...prevMap,\n          [threadId]: {\n            ...prevMap[threadId],\n            messages: [...prevMessages, chatMessage],\n          },\n        };\n      });\n      if (sendToServer) {\n        // TODO: if this fails, we need to revert the local state update\n        await client.beta.threads.messages.create(currentThreadId, {\n          content: message.content,\n          role: message.role,\n          // additionalContext: chatMessage.additionalContext,\n        });\n      }\n      const updatedMessageHistory = [...currentThread.messages, chatMessage];\n\n      return updatedMessageHistory;\n    },\n    [client.beta.threads.messages, currentThread, currentThreadId],\n  );\n\n  const updateThreadMessage = useCallback(\n    async (\n      id: string,\n      message: TamboThreadMessage,\n      sendToServer = true,\n      createdAt: string = new Date().toISOString(),\n    ) => {\n      const chatMessage: TamboThreadMessage = {\n        ...message,\n        createdAt,\n      };\n\n      setThreadMap((prevMap) => {\n        if (!message.threadId) {\n          return prevMap;\n        }\n        const prevMessages = prevMap[message.threadId]?.messages || [];\n        const updatedMessages = prevMessages.map((msg) => {\n          if (msg.id === id) {\n            return chatMessage;\n          }\n          return msg;\n        });\n        return {\n          ...prevMap,\n          [message.threadId]: {\n            ...prevMap[message.threadId],\n            messages: updatedMessages,\n          },\n        };\n      });\n      if (sendToServer) {\n        // TODO: if this fails, we need to revert the local state update\n        await client.beta.threads.messages.create(currentThreadId, {\n          content: message.content,\n          role: message.role,\n          // additionalContext: chatMessage.additionalContext,\n        });\n      }\n    },\n    [client.beta.threads.messages, currentThreadId],\n  );\n\n  useEffect(() => {\n    if (unresolvedThreadId && currentThreadId !== unresolvedThreadId) {\n      setThreadMap((prevMap) => {\n        const unresolvedThread = prevMap[unresolvedThreadId];\n        const currentThread = prevMap[currentThreadId];\n        return {\n          ...prevMap,\n          [unresolvedThreadId]: PLACEHOLDER_THREAD,\n          [currentThreadId]: {\n            ...currentThread,\n            id: currentThreadId,\n            messages: [...unresolvedThread.messages, ...currentThread.messages],\n          },\n        };\n      });\n      setUnresolvedThreadId(undefined);\n    }\n  }, [currentThreadId, unresolvedThreadId]);\n\n  const switchCurrentThread = useCallback(\n    async (threadId: string) => {\n      if (threadId === PLACEHOLDER_THREAD.id) {\n        console.warn(\"Switching to placeholder thread, may be a bug\");\n        return;\n      }\n      setCurrentThreadId(threadId);\n      if (!threadMap[threadId]) {\n        setThreadMap((prevMap) => ({\n          ...prevMap,\n          [threadId]: {\n            ...PLACEHOLDER_THREAD,\n            id: threadId,\n          },\n        }));\n      }\n      await fetchThread(threadId);\n    },\n    [fetchThread, threadMap],\n  );\n\n  const setLastThreadStatus = (status: GenerationStage) => {\n    setThreadMap((prevMap) => {\n      if (!currentThreadId) {\n        return prevMap;\n      }\n      const headMessages = prevMap[currentThreadId].messages.slice(0, -1);\n      const lastMessage =\n        prevMap[currentThreadId].messages[\n          prevMap[currentThreadId].messages.length - 1\n        ];\n      const updatedLastMessage = {\n        ...lastMessage,\n        status,\n      };\n      return {\n        ...prevMap,\n        [currentThreadId]: {\n          ...prevMap[currentThreadId],\n          messages: [...headMessages, updatedLastMessage],\n        },\n      };\n    });\n  };\n\n  const updateThreadStatus = useCallback(\n    (stage: GenerationStage) => {\n      setThreadMap((prevMap) => {\n        return {\n          ...prevMap,\n          [currentThreadId]: {\n            ...prevMap[currentThreadId],\n            generationStage: stage,\n          },\n        };\n      });\n    },\n    [currentThreadId],\n  );\n\n  const handleAdvanceStream = useCallback(\n    async (\n      stream: AsyncIterable<TamboAI.Beta.Threads.ThreadAdvanceResponse>,\n      params: TamboAI.Beta.Threads.ThreadAdvanceParams,\n    ): Promise<TamboThreadMessage> => {\n      let finalMessage: TamboThreadMessage | undefined;\n      let hasSetThreadId = false;\n      updateThreadStatus(GenerationStage.STREAMING_RESPONSE);\n      for await (const chunk of stream) {\n        if (chunk.responseMessageDto.toolCallRequest) {\n          updateThreadStatus(GenerationStage.FETCHING_CONTEXT);\n          const toolCallResponse = await handleToolCall(\n            chunk.responseMessageDto,\n            toolRegistry,\n          );\n          const toolCallResponseParams: TamboAI.Beta.Threads.ThreadAdvanceParams =\n            {\n              ...params,\n              messageToAppend: {\n                content: [{ type: \"text\", text: \"tool response\" }],\n                role: \"tool\",\n                actionType: \"tool_response\",\n                toolResponse: toolCallResponse,\n                component: chunk.responseMessageDto.component,\n              },\n            };\n          updateThreadStatus(GenerationStage.STREAMING_RESPONSE);\n          const toolCallResponseStream = await advanceStream(\n            client,\n            toolCallResponseParams,\n            chunk.responseMessageDto.threadId,\n          );\n          return handleAdvanceStream(\n            toolCallResponseStream,\n            toolCallResponseParams,\n          );\n        } else {\n          if (\n            !hasSetThreadId &&\n            chunk.responseMessageDto.threadId &&\n            chunk.responseMessageDto.threadId !== currentThread?.id\n          ) {\n            hasSetThreadId = true;\n            switchCurrentThread(chunk.responseMessageDto.threadId);\n          }\n\n          if (!finalMessage) {\n            finalMessage = chunk.responseMessageDto.component?.componentName\n              ? renderComponentIntoMessage(\n                  chunk.responseMessageDto,\n                  componentList,\n                )\n              : chunk.responseMessageDto;\n            addThreadMessage(finalMessage, false);\n          } else {\n            const previousId = finalMessage.id;\n            finalMessage = chunk.responseMessageDto.component?.componentName\n              ? renderComponentIntoMessage(\n                  chunk.responseMessageDto,\n                  componentList,\n                )\n              : chunk.responseMessageDto;\n            updateThreadMessage(previousId, finalMessage, false);\n          }\n        }\n      }\n\n      updateThreadStatus(GenerationStage.COMPLETE);\n      return (\n        finalMessage ?? {\n          threadId: \"\",\n          content: [{ type: \"text\", text: `Error processing stream` }],\n          role: \"hydra\",\n          createdAt: new Date().toISOString(),\n          id: crypto.randomUUID(),\n        }\n      );\n    },\n    [\n      toolRegistry,\n      client,\n      currentThread?.id,\n      switchCurrentThread,\n      componentList,\n      addThreadMessage,\n      updateThreadMessage,\n    ],\n  );\n\n  const sendThreadMessage = useCallback(\n    async (\n      message: string,\n      options: {\n        threadId?: string;\n        streamResponse?: boolean;\n        contextKey?: string;\n      } = {},\n    ): Promise<TamboThreadMessage> => {\n      const { threadId, streamResponse } = options;\n      const currentThreadId = threadId ?? currentThread?.id;\n\n      if (currentThreadId !== PLACEHOLDER_THREAD.id) {\n        await switchCurrentThread(currentThreadId);\n      }\n\n      updateThreadStatus(GenerationStage.CHOOSING_COMPONENT);\n\n      addThreadMessage(\n        {\n          content: [{ type: \"text\", text: message }],\n          renderedComponent: null,\n          role: \"user\",\n          threadId: currentThread.id,\n          id: crypto.randomUUID(),\n          createdAt: new Date().toISOString(),\n        },\n        false,\n      );\n\n      const availableComponents = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n      const params: TamboAI.Beta.Threads.ThreadAdvanceParams = {\n        messageToAppend: {\n          content: [{ type: \"text\", text: message }],\n          role: \"user\",\n        },\n        contextKey: options.contextKey,\n        availableComponents: availableComponents,\n      };\n\n      if (streamResponse) {\n        const advanceStreamResponse = await advanceStream(\n          client,\n          params,\n          currentThreadId === PLACEHOLDER_THREAD.id\n            ? undefined\n            : currentThreadId,\n        );\n        return handleAdvanceStream(advanceStreamResponse, params);\n      }\n      let advanceResponse = await (currentThreadId === PLACEHOLDER_THREAD.id\n        ? client.beta.threads.advance(params)\n        : client.beta.threads.advanceById(currentThreadId, params));\n\n      //handle tool calls\n      while (advanceResponse.responseMessageDto.toolCallRequest) {\n        updateThreadStatus(GenerationStage.FETCHING_CONTEXT);\n        const toolCallResponse = await handleToolCall(\n          advanceResponse.responseMessageDto,\n          toolRegistry,\n        );\n        const toolCallResponseParams: TamboAI.Beta.Threads.ThreadAdvanceParams =\n          {\n            ...params,\n            messageToAppend: {\n              ...params.messageToAppend,\n              content: [{ type: \"text\", text: \"tool response\" }],\n              role: \"tool\",\n              actionType: \"tool_response\",\n              toolResponse: toolCallResponse,\n              component: advanceResponse.responseMessageDto.component,\n            },\n          };\n        updateThreadStatus(GenerationStage.HYDRATING_COMPONENT);\n        advanceResponse = await client.beta.threads.advanceById(\n          advanceResponse.responseMessageDto.threadId,\n          toolCallResponseParams,\n        );\n      }\n\n      const finalMessage = advanceResponse.responseMessageDto.component\n        ?.componentName\n        ? renderComponentIntoMessage(\n            advanceResponse.responseMessageDto,\n            componentList,\n          )\n        : advanceResponse.responseMessageDto;\n      await switchCurrentThread(advanceResponse.responseMessageDto.threadId);\n      updateThreadStatus(GenerationStage.COMPLETE);\n      return finalMessage;\n    },\n    [\n      componentList,\n      toolRegistry,\n      componentToolAssociations,\n      currentThread,\n      switchCurrentThread,\n      addThreadMessage,\n      client,\n      handleAdvanceStream,\n    ],\n  );\n\n  return (\n    <TamboThreadContext.Provider\n      value={{\n        thread: currentThread,\n        switchCurrentThread,\n        addThreadMessage,\n        updateThreadMessage,\n        setLastThreadStatus,\n        inputValue,\n        setInputValue,\n        sendThreadMessage,\n      }}\n    >\n      {children}\n    </TamboThreadContext.Provider>\n  );\n};\n\nexport const useTamboThread = () => {\n  const context = useContext(TamboThreadContext);\n  if (context === undefined) {\n    throw new Error(\"useTamboThread must be used within a TamboThreadProvider\");\n  }\n  return context;\n};\n"]}

package/esm/setupTests.d.ts

@@ -0,0 +1 @@
+import "@testing-library/jest-dom";

package/esm/setupTests.js

@@ -0,0 +1,18 @@
+import "@testing-library/jest-dom";
+// Mock AbortController
+const mockSignal = {
+    aborted: false,
+    reason: undefined,
+    onabort: null,
+    addEventListener: jest.fn(),
+    removeEventListener: jest.fn(),
+    dispatchEvent: jest.fn(),
+    throwIfAborted: jest.fn(),
+    any: jest.fn(),
+};
+const MockAbortController = class {
+    signal = mockSignal;
+    abort = jest.fn();
+};
+global.AbortController = MockAbortController;
+//# sourceMappingURL=setupTests.js.map

package/esm/setupTests.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setupTests.js","sourceRoot":"","sources":["../src/setupTests.ts"],"names":[],"mappings":"AAAA,OAAO,2BAA2B,CAAC;AAEnC,uBAAuB;AACvB,MAAM,UAAU,GAAG;IACjB,OAAO,EAAE,KAAK;IACd,MAAM,EAAE,SAAS;IACjB,OAAO,EAAE,IAAI;IACb,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC3B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC9B,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;IACxB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;CACW,CAAC;AAE5B,MAAM,mBAAmB,GAAG;IAC1B,MAAM,GAAG,UAAU,CAAC;IACpB,KAAK,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;CACkB,CAAC;AAEvC,MAAM,CAAC,eAAe,GAAG,mBAAmB,CAAC","sourcesContent":["import \"@testing-library/jest-dom\";\n\n// Mock AbortController\nconst mockSignal = {\n  aborted: false,\n  reason: undefined,\n  onabort: null,\n  addEventListener: jest.fn(),\n  removeEventListener: jest.fn(),\n  dispatchEvent: jest.fn(),\n  throwIfAborted: jest.fn(),\n  any: jest.fn(),\n} as unknown as AbortSignal;\n\nconst MockAbortController = class {\n  signal = mockSignal;\n  abort = jest.fn();\n} as unknown as typeof AbortController;\n\nglobal.AbortController = MockAbortController;\n"]}

package/esm/util/generate-component.d.ts

@@ -0,0 +1,8 @@
+import TamboAI from "@tambo-ai/typescript-sdk";
+import { ComponentRegistry, TamboToolAssociations, TamboToolRegistry } from "../model/component-metadata";
+import { GenerationStage, TamboThreadMessage } from "../model/generate-component-response";
+export declare function generateAndHydrate(client: TamboAI, componentList: ComponentRegistry, toolRegistry: TamboToolRegistry, toolAssociations: TamboToolAssociations, currentThreadId: string | undefined, content: TamboAI.Beta.Threads.ChatCompletionContentPart[], params: TamboAI.Beta.Components.ComponentGenerateParams, onUpdateState: (state: GenerationStage) => void, stream?: boolean, options?: TamboAI.RequestOptions): Promise<TamboThreadMessage | AsyncIterable<TamboThreadMessage>>;
+/**
+ * Generate a message that has a component rendered into it, if the message came with one
+ */
+export declare function renderComponentIntoMessage(message: TamboAI.Beta.Threads.ThreadMessage, componentList: ComponentRegistry): TamboThreadMessage;