@tambo-ai/react 0.12.0 → 0.13.0

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

@@ -1,7 +1,7 @@
 "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 { GenerationStage, isIdleStage, } from "../model/generate-component-response";
 import { renderComponentIntoMessage } from "../util/generate-component";
 import { getAvailableComponents, getClientContext } from "../util/registry";
 import { handleToolCall } from "../util/tool-caller";
@@ -42,6 +42,9 @@ export const TamboThreadContext = createContext({
     sendThreadMessage: () => {
         throw new Error("advance not implemented");
     },
+    generationStage: GenerationStage.IDLE,
+    generationStatusMessage: "",
+    isIdle: true,
 });
 export const TamboThreadProvider = ({ children, }) => {
     const [threadMap, setThreadMap] = useState({
@@ -51,7 +54,6 @@ export const TamboThreadProvider = ({ children, }) => {
     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(() => {
@@ -163,24 +165,17 @@ export const TamboThreadProvider = ({ children, }) => {
             });
         }
     }, [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 deleteThreadMessage = useCallback((messageId) => {
+        if (!currentThread)
+            return;
+        setThreadMap((prevMap) => ({
+            ...prevMap,
+            [currentThread.id]: {
+                ...prevMap[currentThread.id],
+                messages: prevMap[currentThread.id].messages.filter((msg) => msg.id !== messageId),
+            },
+        }));
+    }, [currentThread]);
     const switchCurrentThread = useCallback(async (threadId) => {
         if (threadId === PLACEHOLDER_THREAD.id) {
             console.warn("Switching to placeholder thread, may be a bug");
@@ -188,13 +183,15 @@ export const TamboThreadProvider = ({ children, }) => {
         }
         setCurrentThreadId(threadId);
         if (!threadMap[threadId]) {
-            setThreadMap((prevMap) => ({
-                ...prevMap,
-                [threadId]: {
-                    ...PLACEHOLDER_THREAD,
-                    id: threadId,
-                },
-            }));
+            setThreadMap((prevMap) => {
+                return {
+                    ...prevMap,
+                    [threadId]: {
+                        ...prevMap[PLACEHOLDER_THREAD.id],
+                        id: threadId,
+                    },
+                };
+            });
         }
         await fetchThread(threadId);
     }, [fetchThread, threadMap]);
@@ -218,22 +215,28 @@ export const TamboThreadProvider = ({ children, }) => {
             };
         });
     };
-    const updateThreadStatus = useCallback((stage) => {
+    const updateThreadStatus = useCallback((stage, statusMessage) => {
         setThreadMap((prevMap) => {
             return {
                 ...prevMap,
                 [currentThreadId]: {
                     ...prevMap[currentThreadId],
                     generationStage: stage,
+                    statusMessage: statusMessage,
                 },
             };
         });
     }, [currentThreadId]);
-    const handleAdvanceStream = useCallback(async (stream, params) => {
+    const handleAdvanceStream = useCallback(async (stream, params, messageIdToRemove) => {
         let finalMessage;
         let hasSetThreadId = false;
+        let isFirstChunk = true;
         updateThreadStatus(GenerationStage.STREAMING_RESPONSE);
         for await (const chunk of stream) {
+            if (isFirstChunk && messageIdToRemove) {
+                deleteThreadMessage(messageIdToRemove);
+            }
+            isFirstChunk = false;
             if (chunk.responseMessageDto.toolCallRequest) {
                 updateThreadStatus(GenerationStage.FETCHING_CONTEXT);
                 const toolCallResponse = await handleToolCall(chunk.responseMessageDto, toolRegistry);
@@ -249,7 +252,8 @@ export const TamboThreadProvider = ({ children, }) => {
                 };
                 updateThreadStatus(GenerationStage.STREAMING_RESPONSE);
                 const toolCallResponseStream = await advanceStream(client, toolCallResponseParams, chunk.responseMessageDto.threadId);
-                return handleAdvanceStream(toolCallResponseStream, toolCallResponseParams);
+                // Pass the current message's ID to be removed when the new stream starts, since we now know it is a tool call request message
+                return await handleAdvanceStream(toolCallResponseStream, toolCallResponseParams, finalMessage?.id);
             }
             else {
                 if (!hasSetThreadId &&
@@ -289,6 +293,8 @@ export const TamboThreadProvider = ({ children, }) => {
         componentList,
         addThreadMessage,
         updateThreadMessage,
+        updateThreadStatus,
+        deleteThreadMessage,
     ]);
     const sendThreadMessage = useCallback(async (message, options = {}) => {
         const { threadId, streamResponse } = options;
@@ -318,7 +324,7 @@ export const TamboThreadProvider = ({ children, }) => {
             const advanceStreamResponse = await advanceStream(client, params, currentThreadId === PLACEHOLDER_THREAD.id
                 ? undefined
                 : currentThreadId);
-            return handleAdvanceStream(advanceStreamResponse, params);
+            return await handleAdvanceStream(advanceStreamResponse, params);
         }
         let advanceResponse = await (currentThreadId === PLACEHOLDER_THREAD.id
             ? client.beta.threads.advance(params)
@@ -356,6 +362,7 @@ export const TamboThreadProvider = ({ children, }) => {
         switchCurrentThread,
         addThreadMessage,
         client,
+        updateThreadStatus,
         handleAdvanceStream,
     ]);
     return (React.createElement(TamboThreadContext.Provider, { value: {
@@ -367,6 +374,11 @@ export const TamboThreadProvider = ({ children, }) => {
             inputValue,
             setInputValue,
             sendThreadMessage,
+            generationStage: (currentThread?.generationStage ??
+                GenerationStage.IDLE),
+            generationStatusMessage: currentThread?.statusMessage ?? "",
+            isIdle: isIdleStage((currentThread?.generationStage ??
+                GenerationStage.IDLE)),
         } }, children));
 };
 export const useTamboThread = () => {

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

@@ -1 +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"]}
+{"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,EACf,WAAW,GAEZ,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;AA8B7D;;;;;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;IACD,eAAe,EAAE,eAAe,CAAC,IAAI;IACrC,uBAAuB,EAAE,EAAE;IAC3B,MAAM,EAAE,IAAI;CACb,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,aAAa,GAA4B,SAAS,CAAC,eAAe,CAAC,CAAC;IAE1E,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,MAAM,mBAAmB,GAAG,WAAW,CACrC,CAAC,SAAiB,EAAE,EAAE;QACpB,IAAI,CAAC,aAAa;YAAE,OAAO;QAE3B,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACzB,GAAG,OAAO;YACV,CAAC,aAAa,CAAC,EAAE,CAAC,EAAE;gBAClB,GAAG,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;gBAC5B,QAAQ,EAAE,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,MAAM,CACjD,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,SAAS,CAC9B;aACF;SACF,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IAEF,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;QAED,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAC7B,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzB,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE;gBACvB,OAAO;oBACL,GAAG,OAAO;oBACV,CAAC,QAAQ,CAAC,EAAE;wBACV,GAAG,OAAO,CAAC,kBAAkB,CAAC,EAAE,CAAC;wBACjC,EAAE,EAAE,QAAQ;qBACb;iBACF,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,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,aAAsB,EAAE,EAAE;QACjD,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;oBACtB,aAAa,EAAE,aAAa;iBAC7B;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,EAChD,iBAA0B,EACG,EAAE;QAC/B,IAAI,YAA4C,CAAC;QACjD,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,IAAI,YAAY,GAAG,IAAI,CAAC;QACxB,kBAAkB,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAEvD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,YAAY,IAAI,iBAAiB,EAAE,CAAC;gBACtC,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;YACzC,CAAC;YACD,YAAY,GAAG,KAAK,CAAC;YAErB,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;gBAEF,8HAA8H;gBAC9H,OAAO,MAAM,mBAAmB,CAC9B,sBAAsB,EACtB,sBAAsB,EACtB,YAAY,EAAE,EAAE,CACjB,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;QACnB,kBAAkB;QAClB,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,MAAM,mBAAmB,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;QAClE,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,kBAAkB;QAClB,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;YACjB,eAAe,EAAE,CAAC,aAAa,EAAE,eAAe;gBAC9C,eAAe,CAAC,IAAI,CAAoB;YAC1C,uBAAuB,EAAE,aAAa,EAAE,aAAa,IAAI,EAAE;YAC3D,MAAM,EAAE,WAAW,CACjB,CAAC,aAAa,EAAE,eAAe;gBAC7B,eAAe,CAAC,IAAI,CAAoB,CAC3C;SACF,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  isIdleStage,\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  generationStage: GenerationStage;\n  generationStatusMessage: string;\n  isIdle: boolean;\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  generationStage: GenerationStage.IDLE,\n  generationStatusMessage: \"\",\n  isIdle: true,\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 currentThread: TamboThread | undefined = threadMap[currentThreadId];\n\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  const deleteThreadMessage = useCallback(\n    (messageId: string) => {\n      if (!currentThread) return;\n\n      setThreadMap((prevMap) => ({\n        ...prevMap,\n        [currentThread.id]: {\n          ...prevMap[currentThread.id],\n          messages: prevMap[currentThread.id].messages.filter(\n            (msg) => msg.id !== messageId,\n          ),\n        },\n      }));\n    },\n    [currentThread],\n  );\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\n      setCurrentThreadId(threadId);\n      if (!threadMap[threadId]) {\n        setThreadMap((prevMap) => {\n          return {\n            ...prevMap,\n            [threadId]: {\n              ...prevMap[PLACEHOLDER_THREAD.id],\n              id: threadId,\n            },\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, statusMessage?: string) => {\n      setThreadMap((prevMap) => {\n        return {\n          ...prevMap,\n          [currentThreadId]: {\n            ...prevMap[currentThreadId],\n            generationStage: stage,\n            statusMessage: statusMessage,\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      messageIdToRemove?: string,\n    ): Promise<TamboThreadMessage> => {\n      let finalMessage: TamboThreadMessage | undefined;\n      let hasSetThreadId = false;\n      let isFirstChunk = true;\n      updateThreadStatus(GenerationStage.STREAMING_RESPONSE);\n\n      for await (const chunk of stream) {\n        if (isFirstChunk && messageIdToRemove) {\n          deleteThreadMessage(messageIdToRemove);\n        }\n        isFirstChunk = false;\n\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\n          // Pass the current message's ID to be removed when the new stream starts, since we now know it is a tool call request message\n          return await handleAdvanceStream(\n            toolCallResponseStream,\n            toolCallResponseParams,\n            finalMessage?.id,\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      updateThreadStatus,\n      deleteThreadMessage,\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 await 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      updateThreadStatus,\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        generationStage: (currentThread?.generationStage ??\n          GenerationStage.IDLE) as GenerationStage,\n        generationStatusMessage: currentThread?.statusMessage ?? \"\",\n        isIdle: isIdleStage(\n          (currentThread?.generationStage ??\n            GenerationStage.IDLE) as GenerationStage,\n        ),\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/util/generate-component.d.ts

@@ -1,8 +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>>;
+import { ComponentRegistry } from "../model/component-metadata";
+import { TamboThreadMessage } from "../model/generate-component-response";
 /**
  * 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;
+//# sourceMappingURL=generate-component.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"generate-component.d.ts","sourceRoot":"","sources":["../../src/util/generate-component.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAK/C,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,MAAM,sCAAsC,CAAC;AAG1E;;GAEG;AACH,wBAAgB,0BAA0B,CACxC,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAC3C,aAAa,EAAE,iBAAiB,GAC/B,kBAAkB,CAiCpB"}

package/esm/util/generate-component.js

@@ -1,153 +1,8 @@
-import { generateStream, hydrateStream, } from "@tambo-ai/typescript-sdk";
 import { parse } from "partial-json";
 import React from "react";
 import { z } from "zod";
 import { wrapWithTamboMessageProvider } from "../hooks/use-current-message";
-import { GenerationStage, } from "../model/generate-component-response";
-import { getAvailableComponents, getComponentFromRegistry, } from "../util/registry";
-import { handleToolCall } from "../util/tool-caller";
-export async function generateAndHydrate(client, componentList, toolRegistry, toolAssociations, currentThreadId, content, params, onUpdateState, stream = false, options) {
-    const availableComponents = getAvailableComponents(componentList, toolRegistry, toolAssociations);
-    let threadId = currentThreadId;
-    onUpdateState(GenerationStage.CHOOSING_COMPONENT);
-    try {
-        if (stream) {
-            const streamResponse = await generateStream(client, {
-                availableComponents,
-                contextKey: params.contextKey,
-                content,
-                threadId: params.threadId === "placeholder" ? undefined : params.threadId,
-            }, options);
-            return processComponentDecisionStream(handleStream(streamResponse), componentList, toolRegistry, onUpdateState, client, options);
-        }
-        const response = await client.beta.components.generate({
-            availableComponents,
-            contextKey: params.contextKey,
-            content,
-            threadId: params.threadId === "placeholder" ? undefined : params.threadId,
-        }, options);
-        // Capture the current threadId in case the following processComponentDecision fails
-        threadId = response.message.threadId;
-        return await processComponentDecision(client, componentList, toolRegistry, response.message.threadId, content, response.message, onUpdateState, options);
-    }
-    catch (error) {
-        console.error(error);
-        onUpdateState(GenerationStage.ERROR);
-        // manually create a message with the error
-        const errorMessage = `Error generating component: ${error}`;
-        try {
-            const errorMessageResponse = await client.beta.threads.messages.create(threadId ?? currentThreadId ?? "", {
-                content: [{ type: "text", text: errorMessage }],
-                role: "hydra",
-            });
-            return errorMessageResponse;
-        }
-        catch (errorError) {
-            // Log to the console, there's nothing more we can do!
-            console.error(`Error creating error message: ${errorError}`);
-            console.error("Original error:", error);
-        }
-        return {
-            renderedComponent: null,
-            content: [{ type: "text", text: errorMessage }],
-            role: "hydra",
-            createdAt: new Date().toISOString(),
-            id: crypto.randomUUID(),
-            threadId: currentThreadId ?? "",
-        };
-    }
-}
-async function processComponentDecision(client, componentList, toolRegistry, currentThreadId, content, message, onUpdateState, options) {
-    if (!message.component?.componentName) {
-        const componentResponse = {
-            ...message,
-            renderedComponent: null,
-        };
-        // no component to hydrate, just return the message
-        onUpdateState(GenerationStage.COMPLETE);
-        return componentResponse;
-    }
-    if (message.toolCallRequest) {
-        onUpdateState(GenerationStage.FETCHING_CONTEXT);
-        const toolResult = await handleToolCall(message, toolRegistry);
-        onUpdateState(GenerationStage.HYDRATING_COMPONENT);
-        try {
-            const hydrationResponse = await client.beta.components.hydrate({
-                component: getComponentFromRegistry(message.component.componentName, componentList),
-                threadId: message.threadId,
-                toolResponse: toolResult,
-            }, options);
-            return await processComponentDecision(client, componentList, toolRegistry, currentThreadId, content, hydrationResponse.message, onUpdateState, options);
-        }
-        catch (error) {
-            console.error(error);
-            onUpdateState(GenerationStage.ERROR);
-            return {
-                threadId: message.threadId,
-                renderedComponent: null,
-                content: [
-                    { type: "text", text: `Error hydrating component: ${error}` },
-                ],
-                role: "hydra",
-                createdAt: new Date().toISOString(),
-                id: crypto.randomUUID(),
-            };
-        }
-    }
-    const renderedComponent = React.createElement(getComponentFromRegistry(message.component.componentName, componentList)
-        .component, message.component.props);
-    const componentResponse = {
-        ...message,
-        renderedComponent: wrapWithTamboMessageProvider(renderedComponent, message.threadId, message.id),
-    };
-    onUpdateState(GenerationStage.COMPLETE);
-    return componentResponse;
-}
-async function* processComponentDecisionStream(stream, componentList, toolRegistry, onUpdateState, client, options) {
-    let accumulatedComponentResponse = null;
-    for await (const chunk of stream) {
-        if (!chunk.component?.componentName) {
-            onUpdateState(GenerationStage.STREAMING_RESPONSE);
-            accumulatedComponentResponse = {
-                ...chunk,
-                renderedComponent: null,
-            };
-            yield accumulatedComponentResponse;
-        }
-        if (chunk.toolCallRequest) {
-            if (!chunk.component?.componentName) {
-                continue;
-            }
-            onUpdateState(GenerationStage.FETCHING_CONTEXT);
-            const toolResult = await handleToolCall(chunk, toolRegistry);
-            onUpdateState(GenerationStage.HYDRATING_COMPONENT);
-            const streamResponse = await hydrateStream(client, {
-                component: getComponentFromRegistry(chunk.component?.componentName ?? "", componentList),
-                threadId: chunk.threadId,
-                toolResponse: toolResult,
-            }, options);
-            yield* processComponentDecisionStream(handleStream(streamResponse), componentList, toolRegistry, onUpdateState, client, options);
-            return;
-        }
-        if (chunk.component?.componentName) {
-            try {
-                onUpdateState(GenerationStage.STREAMING_RESPONSE);
-                // Try to parse and validate accumulated component choice props:
-                const messageWithComponent = renderComponentIntoMessage(chunk, componentList);
-                // TODO: do we have to do this now
-                accumulatedComponentResponse = {
-                    ...accumulatedComponentResponse,
-                    ...messageWithComponent,
-                };
-                yield messageWithComponent;
-            }
-            catch (error) {
-                console.error(error);
-            }
-        }
-    }
-    onUpdateState(GenerationStage.COMPLETE);
-}
+import { getComponentFromRegistry } from "../util/registry";
 /**
  * Generate a message that has a component rendered into it, if the message came with one
  */
@@ -171,32 +26,4 @@ export function renderComponentIntoMessage(message, componentList) {
         renderedComponent: wrappedComponent,
     };
 }
-async function* handleStream(stream) {
-    let threadMessage = {
-        threadId: "",
-        content: [],
-        role: "hydra",
-        createdAt: new Date().toISOString(),
-        id: crypto.randomUUID(),
-    };
-    try {
-        for await (const chunk of stream) {
-            threadMessage = {
-                ...threadMessage,
-                ...chunk,
-            };
-            yield threadMessage;
-        }
-    }
-    catch (error) {
-        console.error("Error processing stream:", error);
-        yield {
-            threadId: "",
-            content: [{ type: "text", text: `Error processing stream: ${error}` }],
-            role: "hydra",
-            createdAt: new Date().toISOString(),
-            id: crypto.randomUUID(),
-        };
-    }
-}
 //# sourceMappingURL=generate-component.js.map

package/esm/util/generate-component.js.map

@@ -1 +1 @@
-{"version":3,"file":"generate-component.js","sourceRoot":"","sources":["../../src/util/generate-component.ts"],"names":[],"mappings":"AAAA,OAAgB,EACd,cAAc,EACd,aAAa,GACd,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,KAAK,EAAE,MAAM,cAAc,CAAC;AACrC,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,4BAA4B,EAAE,MAAM,8BAA8B,CAAC;AAM5E,OAAO,EACL,eAAe,GAEhB,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EACL,sBAAsB,EACtB,wBAAwB,GACzB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAErD,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,MAAe,EACf,aAAgC,EAChC,YAA+B,EAC/B,gBAAuC,EACvC,eAAmC,EACnC,OAAyD,EACzD,MAAuD,EACvD,aAA+C,EAC/C,MAAM,GAAG,KAAK,EACd,OAAgC;IAEhC,MAAM,mBAAmB,GAAG,sBAAsB,CAChD,aAAa,EACb,YAAY,EACZ,gBAAgB,CACjB,CAAC;IACF,IAAI,QAAQ,GAAG,eAAe,CAAC;IAE/B,aAAa,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;IAClD,IAAI,CAAC;QACH,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,cAAc,GAAG,MAAM,cAAc,CACzC,MAAM,EACN;gBACE,mBAAmB;gBACnB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO;gBACP,QAAQ,EACN,MAAM,CAAC,QAAQ,KAAK,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ;aAClE,EACD,OAAO,CACR,CAAC;YAEF,OAAO,8BAA8B,CACnC,YAAY,CACV,cAAkE,CACnE,EACD,aAAa,EACb,YAAY,EACZ,aAAa,EACb,MAAM,EACN,OAAO,CACR,CAAC;QACJ,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CACpD;YACE,mBAAmB;YACnB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO;YACP,QAAQ,EACN,MAAM,CAAC,QAAQ,KAAK,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ;SAClE,EACD,OAAO,CACR,CAAC;QACF,oFAAoF;QACpF,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC;QACrC,OAAO,MAAM,wBAAwB,CACnC,MAAM,EACN,aAAa,EACb,YAAY,EACZ,QAAQ,CAAC,OAAO,CAAC,QAAQ,EACzB,OAAO,EACP,QAAQ,CAAC,OAAO,EAChB,aAAa,EACb,OAAO,CACR,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACrB,aAAa,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACrC,2CAA2C;QAC3C,MAAM,YAAY,GAAG,+BAA+B,KAAK,EAAE,CAAC;QAC5D,IAAI,CAAC;YACH,MAAM,oBAAoB,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CACpE,QAAQ,IAAI,eAAe,IAAI,EAAE,EACjC;gBACE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;gBAC/C,IAAI,EAAE,OAAO;aACd,CACF,CAAC;YACF,OAAO,oBAAoB,CAAC;QAC9B,CAAC;QAAC,OAAO,UAAU,EAAE,CAAC;YACpB,sDAAsD;YACtD,OAAO,CAAC,KAAK,CAAC,iCAAiC,UAAU,EAAE,CAAC,CAAC;YAC7D,OAAO,CAAC,KAAK,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAC;QAC1C,CAAC;QACD,OAAO;YACL,iBAAiB,EAAE,IAAI;YACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;YAC/C,IAAI,EAAE,OAAO;YACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,QAAQ,EAAE,eAAe,IAAI,EAAE;SAChC,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,wBAAwB,CACrC,MAAe,EACf,aAAgC,EAChC,YAA+B,EAC/B,eAAuB,EACvB,OAAmE,EACnE,OAA2C,EAC3C,aAA+C,EAC/C,OAAgC;IAEhC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC;QACtC,MAAM,iBAAiB,GAAuB;YAC5C,GAAG,OAAO;YACV,iBAAiB,EAAE,IAAI;SACxB,CAAC;QACF,mDAAmD;QACnD,aAAa,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QACxC,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;QAC5B,aAAa,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;QAChD,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAC/D,aAAa,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACnD,IAAI,CAAC;YACH,MAAM,iBAAiB,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAC5D;gBACE,SAAS,EAAE,wBAAwB,CACjC,OAAO,CAAC,SAAS,CAAC,aAAa,EAC/B,aAAa,CACd;gBACD,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAE1B,YAAY,EAAE,UAAU;aACzB,EACD,OAAO,CACR,CAAC;YACF,OAAO,MAAM,wBAAwB,CACnC,MAAM,EACN,aAAa,EACb,YAAY,EACZ,eAAe,EACf,OAAO,EACP,iBAAiB,CAAC,OAAO,EACzB,aAAa,EACb,OAAO,CACR,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACrB,aAAa,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACrC,OAAO;gBACL,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,iBAAiB,EAAE,IAAI;gBACvB,OAAO,EAAE;oBACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8BAA8B,KAAK,EAAE,EAAE;iBAC9D;gBACD,IAAI,EAAE,OAAO;gBACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;aACxB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,iBAAiB,GAAG,KAAK,CAAC,aAAa,CAC3C,wBAAwB,CAAC,OAAO,CAAC,SAAS,CAAC,aAAa,EAAE,aAAa,CAAC;SACrE,SAAS,EACZ,OAAO,CAAC,SAAS,CAAC,KAAK,CACxB,CAAC;IACF,MAAM,iBAAiB,GAAuB;QAC5C,GAAG,OAAO;QACV,iBAAiB,EAAE,4BAA4B,CAC7C,iBAAiB,EACjB,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,EAAE,CACX;KACF,CAAC;IAEF,aAAa,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;IACxC,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,8BAA8B,CAC5C,MAAyC,EACzC,aAAgC,EAChC,YAA+B,EAC/B,aAA+C,EAC/C,MAAe,EACf,OAAgC;IAEhC,IAAI,4BAA4B,GAA8B,IAAI,CAAC;IAEnE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QACjC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC;YACpC,aAAa,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YAClD,4BAA4B,GAAG;gBAC7B,GAAG,KAAK;gBACR,iBAAiB,EAAE,IAAI;aACxB,CAAC;YACF,MAAM,4BAA4B,CAAC;QACrC,CAAC;QAED,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;YAC1B,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC;gBACpC,SAAS;YACX,CAAC;YACD,aAAa,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;YAChD,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;YAC7D,aAAa,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;YACnD,MAAM,cAAc,GAAG,MAAM,aAAa,CACxC,MAAM,EACN;gBACE,SAAS,EAAE,wBAAwB,CACjC,KAAK,CAAC,SAAS,EAAE,aAAa,IAAI,EAAE,EACpC,aAAa,CACd;gBACD,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,YAAY,EAAE,UAAU;aACzB,EACD,OAAO,CACR,CAAC;YAEF,KAAK,CAAC,CAAC,8BAA8B,CACnC,YAAY,CACV,cAAiE,CAClE,EACD,aAAa,EACb,YAAY,EACZ,aAAa,EACb,MAAM,EACN,OAAO,CACR,CAAC;YACF,OAAO;QACT,CAAC;QAED,IAAI,KAAK,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC;YACnC,IAAI,CAAC;gBACH,aAAa,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;gBAClD,gEAAgE;gBAChE,MAAM,oBAAoB,GAAG,0BAA0B,CACrD,KAAK,EACL,aAAa,CACd,CAAC;gBACF,kCAAkC;gBAClC,4BAA4B,GAAG;oBAC7B,GAAG,4BAA4B;oBAC/B,GAAG,oBAAoB;iBACxB,CAAC;gBACF,MAAM,oBAAoB,CAAC;YAC7B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;IACH,CAAC;IACD,aAAa,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;AAC1C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,0BAA0B,CACxC,OAA2C,EAC3C,aAAgC;IAEhC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACzC,CAAC;IACD,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;IACnE,MAAM,mBAAmB,GAAG,wBAAwB,CAClD,OAAO,CAAC,SAAS,CAAC,aAAa,EAC/B,aAAa,CACd,CAAC;IAEF,MAAM,cAAc,GAClB,mBAAmB,CAAC,KAAK,YAAY,CAAC,CAAC,OAAO;QAC5C,CAAC,CAAC,mBAAmB,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;QAC9C,CAAC,CAAC,WAAW,CAAC;IAElB,MAAM,iBAAiB,GAAG,KAAK,CAAC,aAAa,CAC3C,mBAAmB,CAAC,SAAS,EAC7B,cAAc,CACf,CAAC;IACF,MAAM,gBAAgB,GAAG,4BAA4B,CACnD,iBAAiB,EACjB,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,EAAE,CACX,CAAC;IAEF,OAAO;QACL,GAAG,OAAO;QACV,SAAS,EAAE;YACT,GAAG,OAAO,CAAC,SAAS;YACpB,KAAK,EAAE,cAAc;SACtB;QACD,iBAAiB,EAAE,gBAAgB;KACpC,CAAC;AACJ,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,YAAY,CAC1B,MAEC;IAED,IAAI,aAAa,GAAuC;QACtD,QAAQ,EAAE,EAAE;QACZ,OAAO,EAAE,EAAE;QACX,IAAI,EAAE,OAAO;QACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;KACxB,CAAC;IAEF,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,aAAa,GAAG;gBACd,GAAG,aAAa;gBAChB,GAAG,KAAK;aACT,CAAC;YACF,MAAM,aAAa,CAAC;QACtB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM;YACJ,QAAQ,EAAE,EAAE;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,4BAA4B,KAAK,EAAE,EAAE,CAAC;YACtE,IAAI,EAAE,OAAO;YACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;SACxB,CAAC;IACJ,CAAC;AACH,CAAC","sourcesContent":["import TamboAI, {\n  generateStream,\n  hydrateStream,\n} from \"@tambo-ai/typescript-sdk\";\nimport { parse } from \"partial-json\";\nimport React from \"react\";\nimport { z } from \"zod\";\nimport { wrapWithTamboMessageProvider } from \"../hooks/use-current-message\";\nimport {\n  ComponentRegistry,\n  TamboToolAssociations,\n  TamboToolRegistry,\n} from \"../model/component-metadata\";\nimport {\n  GenerationStage,\n  TamboThreadMessage,\n} from \"../model/generate-component-response\";\nimport {\n  getAvailableComponents,\n  getComponentFromRegistry,\n} from \"../util/registry\";\nimport { handleToolCall } from \"../util/tool-caller\";\n\nexport async function generateAndHydrate(\n  client: TamboAI,\n  componentList: ComponentRegistry,\n  toolRegistry: TamboToolRegistry,\n  toolAssociations: TamboToolAssociations,\n  currentThreadId: string | undefined,\n  content: TamboAI.Beta.Threads.ChatCompletionContentPart[],\n  params: TamboAI.Beta.Components.ComponentGenerateParams,\n  onUpdateState: (state: GenerationStage) => void,\n  stream = false,\n  options?: TamboAI.RequestOptions,\n): Promise<TamboThreadMessage | AsyncIterable<TamboThreadMessage>> {\n  const availableComponents = getAvailableComponents(\n    componentList,\n    toolRegistry,\n    toolAssociations,\n  );\n  let threadId = currentThreadId;\n\n  onUpdateState(GenerationStage.CHOOSING_COMPONENT);\n  try {\n    if (stream) {\n      const streamResponse = await generateStream(\n        client,\n        {\n          availableComponents,\n          contextKey: params.contextKey,\n          content,\n          threadId:\n            params.threadId === \"placeholder\" ? undefined : params.threadId,\n        },\n        options,\n      );\n\n      return processComponentDecisionStream(\n        handleStream(\n          streamResponse as AsyncIterable<TamboAI.ComponentGenerateResponse>,\n        ),\n        componentList,\n        toolRegistry,\n        onUpdateState,\n        client,\n        options,\n      );\n    }\n    const response = await client.beta.components.generate(\n      {\n        availableComponents,\n        contextKey: params.contextKey,\n        content,\n        threadId:\n          params.threadId === \"placeholder\" ? undefined : params.threadId,\n      },\n      options,\n    );\n    // Capture the current threadId in case the following processComponentDecision fails\n    threadId = response.message.threadId;\n    return await processComponentDecision(\n      client,\n      componentList,\n      toolRegistry,\n      response.message.threadId,\n      content,\n      response.message,\n      onUpdateState,\n      options,\n    );\n  } catch (error) {\n    console.error(error);\n    onUpdateState(GenerationStage.ERROR);\n    // manually create a message with the error\n    const errorMessage = `Error generating component: ${error}`;\n    try {\n      const errorMessageResponse = await client.beta.threads.messages.create(\n        threadId ?? currentThreadId ?? \"\",\n        {\n          content: [{ type: \"text\", text: errorMessage }],\n          role: \"hydra\",\n        },\n      );\n      return errorMessageResponse;\n    } catch (errorError) {\n      // Log to the console, there's nothing more we can do!\n      console.error(`Error creating error message: ${errorError}`);\n      console.error(\"Original error:\", error);\n    }\n    return {\n      renderedComponent: null,\n      content: [{ type: \"text\", text: errorMessage }],\n      role: \"hydra\",\n      createdAt: new Date().toISOString(),\n      id: crypto.randomUUID(),\n      threadId: currentThreadId ?? \"\",\n    };\n  }\n}\n\nasync function processComponentDecision(\n  client: TamboAI,\n  componentList: ComponentRegistry,\n  toolRegistry: TamboToolRegistry,\n  currentThreadId: string,\n  content: TamboAI.Beta.Components.ComponentGenerateParams[\"content\"],\n  message: TamboAI.Beta.Threads.ThreadMessage,\n  onUpdateState: (state: GenerationStage) => void,\n  options?: TamboAI.RequestOptions,\n): Promise<TamboThreadMessage> {\n  if (!message.component?.componentName) {\n    const componentResponse: TamboThreadMessage = {\n      ...message,\n      renderedComponent: null,\n    };\n    // no component to hydrate, just return the message\n    onUpdateState(GenerationStage.COMPLETE);\n    return componentResponse;\n  }\n\n  if (message.toolCallRequest) {\n    onUpdateState(GenerationStage.FETCHING_CONTEXT);\n    const toolResult = await handleToolCall(message, toolRegistry);\n    onUpdateState(GenerationStage.HYDRATING_COMPONENT);\n    try {\n      const hydrationResponse = await client.beta.components.hydrate(\n        {\n          component: getComponentFromRegistry(\n            message.component.componentName,\n            componentList,\n          ),\n          threadId: message.threadId,\n\n          toolResponse: toolResult,\n        },\n        options,\n      );\n      return await processComponentDecision(\n        client,\n        componentList,\n        toolRegistry,\n        currentThreadId,\n        content,\n        hydrationResponse.message,\n        onUpdateState,\n        options,\n      );\n    } catch (error) {\n      console.error(error);\n      onUpdateState(GenerationStage.ERROR);\n      return {\n        threadId: message.threadId,\n        renderedComponent: null,\n        content: [\n          { type: \"text\", text: `Error hydrating component: ${error}` },\n        ],\n        role: \"hydra\",\n        createdAt: new Date().toISOString(),\n        id: crypto.randomUUID(),\n      };\n    }\n  }\n\n  const renderedComponent = React.createElement(\n    getComponentFromRegistry(message.component.componentName, componentList)\n      .component,\n    message.component.props,\n  );\n  const componentResponse: TamboThreadMessage = {\n    ...message,\n    renderedComponent: wrapWithTamboMessageProvider(\n      renderedComponent,\n      message.threadId,\n      message.id,\n    ),\n  };\n\n  onUpdateState(GenerationStage.COMPLETE);\n  return componentResponse;\n}\n\nasync function* processComponentDecisionStream(\n  stream: AsyncIterable<TamboThreadMessage>,\n  componentList: ComponentRegistry,\n  toolRegistry: TamboToolRegistry,\n  onUpdateState: (state: GenerationStage) => void,\n  client: TamboAI,\n  options?: TamboAI.RequestOptions,\n): AsyncGenerator<TamboThreadMessage> {\n  let accumulatedComponentResponse: TamboThreadMessage | null = null;\n\n  for await (const chunk of stream) {\n    if (!chunk.component?.componentName) {\n      onUpdateState(GenerationStage.STREAMING_RESPONSE);\n      accumulatedComponentResponse = {\n        ...chunk,\n        renderedComponent: null,\n      };\n      yield accumulatedComponentResponse;\n    }\n\n    if (chunk.toolCallRequest) {\n      if (!chunk.component?.componentName) {\n        continue;\n      }\n      onUpdateState(GenerationStage.FETCHING_CONTEXT);\n      const toolResult = await handleToolCall(chunk, toolRegistry);\n      onUpdateState(GenerationStage.HYDRATING_COMPONENT);\n      const streamResponse = await hydrateStream(\n        client,\n        {\n          component: getComponentFromRegistry(\n            chunk.component?.componentName ?? \"\",\n            componentList,\n          ),\n          threadId: chunk.threadId,\n          toolResponse: toolResult,\n        },\n        options,\n      );\n\n      yield* processComponentDecisionStream(\n        handleStream(\n          streamResponse as AsyncIterable<TamboAI.ComponentHydrateResponse>,\n        ),\n        componentList,\n        toolRegistry,\n        onUpdateState,\n        client,\n        options,\n      );\n      return;\n    }\n\n    if (chunk.component?.componentName) {\n      try {\n        onUpdateState(GenerationStage.STREAMING_RESPONSE);\n        // Try to parse and validate accumulated component choice props:\n        const messageWithComponent = renderComponentIntoMessage(\n          chunk,\n          componentList,\n        );\n        // TODO: do we have to do this now\n        accumulatedComponentResponse = {\n          ...accumulatedComponentResponse,\n          ...messageWithComponent,\n        };\n        yield messageWithComponent;\n      } catch (error) {\n        console.error(error);\n      }\n    }\n  }\n  onUpdateState(GenerationStage.COMPLETE);\n}\n\n/**\n * Generate a message that has a component rendered into it, if the message came with one\n */\nexport function renderComponentIntoMessage(\n  message: TamboAI.Beta.Threads.ThreadMessage,\n  componentList: ComponentRegistry,\n): TamboThreadMessage {\n  if (!message.component?.componentName) {\n    throw new Error(\"Component not found\");\n  }\n  const parsedProps = parse(JSON.stringify(message.component.props));\n  const registeredComponent = getComponentFromRegistry(\n    message.component.componentName,\n    componentList,\n  );\n\n  const validatedProps =\n    registeredComponent.props instanceof z.ZodType\n      ? registeredComponent.props.parse(parsedProps)\n      : parsedProps;\n\n  const renderedComponent = React.createElement(\n    registeredComponent.component,\n    validatedProps,\n  );\n  const wrappedComponent = wrapWithTamboMessageProvider(\n    renderedComponent,\n    message.threadId,\n    message.id,\n  );\n\n  return {\n    ...message,\n    component: {\n      ...message.component,\n      props: validatedProps,\n    },\n    renderedComponent: wrappedComponent,\n  };\n}\n\nasync function* handleStream(\n  stream: AsyncIterable<\n    TamboAI.ComponentGenerateResponse | TamboAI.ComponentHydrateResponse\n  >,\n): AsyncGenerator<TamboAI.Beta.Threads.ThreadMessage> {\n  let threadMessage: TamboAI.Beta.Threads.ThreadMessage = {\n    threadId: \"\",\n    content: [],\n    role: \"hydra\",\n    createdAt: new Date().toISOString(),\n    id: crypto.randomUUID(),\n  };\n\n  try {\n    for await (const chunk of stream) {\n      threadMessage = {\n        ...threadMessage,\n        ...chunk,\n      };\n      yield threadMessage;\n    }\n  } catch (error) {\n    console.error(\"Error processing stream:\", error);\n    yield {\n      threadId: \"\",\n      content: [{ type: \"text\", text: `Error processing stream: ${error}` }],\n      role: \"hydra\",\n      createdAt: new Date().toISOString(),\n      id: crypto.randomUUID(),\n    };\n  }\n}\n"]}
+{"version":3,"file":"generate-component.js","sourceRoot":"","sources":["../../src/util/generate-component.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,MAAM,cAAc,CAAC;AACrC,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,4BAA4B,EAAE,MAAM,8BAA8B,CAAC;AAG5E,OAAO,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AAE5D;;GAEG;AACH,MAAM,UAAU,0BAA0B,CACxC,OAA2C,EAC3C,aAAgC;IAEhC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACzC,CAAC;IACD,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;IACnE,MAAM,mBAAmB,GAAG,wBAAwB,CAClD,OAAO,CAAC,SAAS,CAAC,aAAa,EAC/B,aAAa,CACd,CAAC;IAEF,MAAM,cAAc,GAClB,mBAAmB,CAAC,KAAK,YAAY,CAAC,CAAC,OAAO;QAC5C,CAAC,CAAC,mBAAmB,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;QAC9C,CAAC,CAAC,WAAW,CAAC;IAElB,MAAM,iBAAiB,GAAG,KAAK,CAAC,aAAa,CAC3C,mBAAmB,CAAC,SAAS,EAC7B,cAAc,CACf,CAAC;IACF,MAAM,gBAAgB,GAAG,4BAA4B,CACnD,iBAAiB,EACjB,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,EAAE,CACX,CAAC;IAEF,OAAO;QACL,GAAG,OAAO;QACV,SAAS,EAAE;YACT,GAAG,OAAO,CAAC,SAAS;YACpB,KAAK,EAAE,cAAc;SACtB;QACD,iBAAiB,EAAE,gBAAgB;KACpC,CAAC;AACJ,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { parse } from \"partial-json\";\nimport React from \"react\";\nimport { z } from \"zod\";\nimport { wrapWithTamboMessageProvider } from \"../hooks/use-current-message\";\nimport { ComponentRegistry } from \"../model/component-metadata\";\nimport { TamboThreadMessage } from \"../model/generate-component-response\";\nimport { getComponentFromRegistry } from \"../util/registry\";\n\n/**\n * Generate a message that has a component rendered into it, if the message came with one\n */\nexport function renderComponentIntoMessage(\n  message: TamboAI.Beta.Threads.ThreadMessage,\n  componentList: ComponentRegistry,\n): TamboThreadMessage {\n  if (!message.component?.componentName) {\n    throw new Error(\"Component not found\");\n  }\n  const parsedProps = parse(JSON.stringify(message.component.props));\n  const registeredComponent = getComponentFromRegistry(\n    message.component.componentName,\n    componentList,\n  );\n\n  const validatedProps =\n    registeredComponent.props instanceof z.ZodType\n      ? registeredComponent.props.parse(parsedProps)\n      : parsedProps;\n\n  const renderedComponent = React.createElement(\n    registeredComponent.component,\n    validatedProps,\n  );\n  const wrappedComponent = wrapWithTamboMessageProvider(\n    renderedComponent,\n    message.threadId,\n    message.id,\n  );\n\n  return {\n    ...message,\n    component: {\n      ...message.component,\n      props: validatedProps,\n    },\n    renderedComponent: wrappedComponent,\n  };\n}\n"]}

package/esm/util/messages.d.ts

@@ -1,2 +1,3 @@
 import TamboAI from "@tambo-ai/typescript-sdk";
 export declare function threadMessagesToLegacyMessages(messages: TamboAI.Beta.Threads.ThreadMessage[]): TamboAI.Components.LegacyChatMessage[];
+//# sourceMappingURL=messages.d.ts.map

package/esm/util/messages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../../src/util/messages.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAE/C,wBAAgB,8BAA8B,CAC5C,QAAQ,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,0CAU/C"}

package/esm/util/query-utils.d.ts

@@ -3,3 +3,4 @@ export type CombinedMutationResult<TData = unknown, TError = unknown, TVariables
 export declare function combineMutationResults<TData1, TData2, TError1, TError2>(resultA: UseMutationResult<TData1, TError1, any, any>, resultB: UseMutationResult<TData2, TError2, any, any>): CombinedMutationResult<TData1 | TData2, TError1 | TError2>;
 export type CombinedQueryResult<TData1, TData2, TError1, TError2> = Omit<UseQueryResult<TData1 | TData2, TError1 | TError2>, "data" | "refetch" | "promise">;
 export declare function combineQueryResults<TData1, TData2, TError1, TError2>(resultA: UseQueryResult<TData1, TError1>, resultB: UseQueryResult<TData2, TError2>): CombinedQueryResult<void, void, TError1, TError2>;
+//# sourceMappingURL=query-utils.d.ts.map

package/esm/util/query-utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-utils.d.ts","sourceRoot":"","sources":["../../src/util/query-utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAE1E,MAAM,MAAM,sBAAsB,CAChC,KAAK,GAAG,OAAO,EACf,MAAM,GAAG,OAAO,EAChB,UAAU,GAAG,OAAO,EACpB,QAAQ,GAAG,OAAO,IAChB,IAAI,CACN,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,CAAC,EACtD,QAAQ,GAAG,aAAa,GAAG,OAAO,GAAG,MAAM,GAAG,WAAW,GAAG,SAAS,CACtE,CAAC;AAEF,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EACrE,OAAO,EAAE,iBAAiB,CAAC,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,CAAC,EACrD,OAAO,EAAE,iBAAiB,CAAC,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,CAAC,GACpD,sBAAsB,CAAC,MAAM,GAAG,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,CAsB5D;AAED,MAAM,MAAM,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,IAAI,IAAI,CACtE,cAAc,CAAC,MAAM,GAAG,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,EAClD,MAAM,GAAG,SAAS,GAAG,SAAS,CAC/B,CAAC;AACF,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAClE,OAAO,EAAE,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,EACxC,OAAO,EAAE,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,GACvC,mBAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAsCnD"}

package/esm/util/registry.d.ts

@@ -5,3 +5,4 @@ export declare const getComponentFromRegistry: (componentName: string, component
 export declare const getDefaultContextAdditions: () => string[];
 export declare const getClientContext: () => string;
 export declare const mapTamboToolToContextTool: (tool: TamboTool) => ComponentContextToolMetadata;
+//# sourceMappingURL=registry.d.ts.map

package/esm/util/registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../src/util/registry.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAG/C,OAAO,EACL,4BAA4B,EAC5B,iBAAiB,EAEjB,mBAAmB,EACnB,SAAS,EACT,qBAAqB,EACrB,iBAAiB,EAClB,MAAM,6BAA6B,CAAC;AAErC,eAAO,MAAM,sBAAsB,GACjC,mBAAmB,iBAAiB,EACpC,cAAc,iBAAiB,EAC/B,kBAAkB,qBAAqB,KACtC,OAAO,CAAC,kBAAkB,EAuB5B,CAAC;AAEF,eAAO,MAAM,wBAAwB,GACnC,eAAe,MAAM,EACrB,mBAAmB,iBAAiB,KACnC,mBAUF,CAAC;AAEF,eAAO,MAAM,0BAA0B,QAAO,MAAM,EAMnD,CAAC;AAEF,eAAO,MAAM,gBAAgB,QAAO,MAGnC,CAAC;AAEF,eAAO,MAAM,yBAAyB,GACpC,MAAM,SAAS,KACd,4BAQF,CAAC"}

package/esm/util/tool-caller.d.ts

@@ -1,3 +1,4 @@
 import TamboAI from "@tambo-ai/typescript-sdk";
 import { TamboToolRegistry } from "../model/component-metadata";
 export declare const handleToolCall: (message: TamboAI.Beta.ThreadMessage, toolRegistry: TamboToolRegistry) => Promise<any>;
+//# sourceMappingURL=tool-caller.d.ts.map

package/esm/util/tool-caller.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-caller.d.ts","sourceRoot":"","sources":["../../src/util/tool-caller.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAC/C,OAAO,EAEL,iBAAiB,EAClB,MAAM,6BAA6B,CAAC;AAErC,eAAO,MAAM,cAAc,GACzB,SAAS,OAAO,CAAC,IAAI,CAAC,aAAa,EACnC,cAAc,iBAAiB,iBAUhC,CAAC"}

package/esm/util/tool-caller.js

@@ -21,6 +21,6 @@ const findTool = (toolName, toolRegistry) => {
 const runToolChoice = async (toolCallRequest, tool) => {
     // Assumes parameters are in the order they are defined in the tool
     const parameterValues = toolCallRequest.parameters?.map((param) => param.parameterValue) ?? [];
-    return tool.getComponentContext(...parameterValues);
+    return await tool.getComponentContext(...parameterValues);
 };
 //# sourceMappingURL=tool-caller.js.map

package/esm/util/tool-caller.js.map

@@ -1 +1 @@
-{"version":3,"file":"tool-caller.js","sourceRoot":"","sources":["../../src/util/tool-caller.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,yBAAyB,EAAE,MAAM,YAAY,CAAC;AACvD,MAAM,CAAC,MAAM,cAAc,GAAG,KAAK,EACjC,OAAmC,EACnC,YAA+B,EAC/B,EAAE;IACF,IAAI,CAAC,OAAO,EAAE,eAAe,EAAE,QAAQ,EAAE,CAAC;QACxC,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,eAAe,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IACtE,MAAM,UAAU,GAAG,MAAM,aAAa,CAAC,OAAO,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;IAEtE,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAEF,MAAM,QAAQ,GAAG,CAAC,QAAgB,EAAE,YAA+B,EAAE,EAAE;IACrE,MAAM,YAAY,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAE5C,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,wBAAwB,CAAC,CAAC;IAC5D,CAAC;IAED,MAAM,WAAW,GAAG,yBAAyB,CAAC,YAAY,CAAC,CAAC;IAC5D,OAAO;QACL,mBAAmB,EAAE,YAAY,CAAC,IAAI;QACtC,UAAU,EAAE,WAAW;KACxB,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,aAAa,GAAG,KAAK,EACzB,eAAwC,EACxC,IAA0B,EACZ,EAAE;IAChB,mEAAmE;IACnE,MAAM,eAAe,GACnB,eAAe,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAEzE,OAAO,IAAI,CAAC,mBAAmB,CAAC,GAAG,eAAe,CAAC,CAAC;AACtD,CAAC,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport {\n  ComponentContextTool,\n  TamboToolRegistry,\n} from \"../model/component-metadata\";\nimport { mapTamboToolToContextTool } from \"./registry\";\nexport const handleToolCall = async (\n  message: TamboAI.Beta.ThreadMessage,\n  toolRegistry: TamboToolRegistry,\n) => {\n  if (!message?.toolCallRequest?.toolName) {\n    throw new Error(\"Tool name is required\");\n  }\n\n  const tool = findTool(message.toolCallRequest.toolName, toolRegistry);\n  const toolResult = await runToolChoice(message.toolCallRequest, tool);\n\n  return toolResult;\n};\n\nconst findTool = (toolName: string, toolRegistry: TamboToolRegistry) => {\n  const registryTool = toolRegistry[toolName];\n\n  if (!registryTool) {\n    throw new Error(`Tool ${toolName} not found in registry`);\n  }\n\n  const contextTool = mapTamboToolToContextTool(registryTool);\n  return {\n    getComponentContext: registryTool.tool,\n    definition: contextTool,\n  };\n};\n\nconst runToolChoice = async (\n  toolCallRequest: TamboAI.ToolCallRequest,\n  tool: ComponentContextTool,\n): Promise<any> => {\n  // Assumes parameters are in the order they are defined in the tool\n  const parameterValues =\n    toolCallRequest.parameters?.map((param) => param.parameterValue) ?? [];\n\n  return tool.getComponentContext(...parameterValues);\n};\n"]}
+{"version":3,"file":"tool-caller.js","sourceRoot":"","sources":["../../src/util/tool-caller.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,yBAAyB,EAAE,MAAM,YAAY,CAAC;AACvD,MAAM,CAAC,MAAM,cAAc,GAAG,KAAK,EACjC,OAAmC,EACnC,YAA+B,EAC/B,EAAE;IACF,IAAI,CAAC,OAAO,EAAE,eAAe,EAAE,QAAQ,EAAE,CAAC;QACxC,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,eAAe,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IACtE,MAAM,UAAU,GAAG,MAAM,aAAa,CAAC,OAAO,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;IAEtE,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAEF,MAAM,QAAQ,GAAG,CAAC,QAAgB,EAAE,YAA+B,EAAE,EAAE;IACrE,MAAM,YAAY,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAE5C,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,wBAAwB,CAAC,CAAC;IAC5D,CAAC;IAED,MAAM,WAAW,GAAG,yBAAyB,CAAC,YAAY,CAAC,CAAC;IAC5D,OAAO;QACL,mBAAmB,EAAE,YAAY,CAAC,IAAI;QACtC,UAAU,EAAE,WAAW;KACxB,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,aAAa,GAAG,KAAK,EACzB,eAAwC,EACxC,IAA0B,EACZ,EAAE;IAChB,mEAAmE;IACnE,MAAM,eAAe,GACnB,eAAe,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAEzE,OAAO,MAAM,IAAI,CAAC,mBAAmB,CAAC,GAAG,eAAe,CAAC,CAAC;AAC5D,CAAC,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport {\n  ComponentContextTool,\n  TamboToolRegistry,\n} from \"../model/component-metadata\";\nimport { mapTamboToolToContextTool } from \"./registry\";\nexport const handleToolCall = async (\n  message: TamboAI.Beta.ThreadMessage,\n  toolRegistry: TamboToolRegistry,\n) => {\n  if (!message?.toolCallRequest?.toolName) {\n    throw new Error(\"Tool name is required\");\n  }\n\n  const tool = findTool(message.toolCallRequest.toolName, toolRegistry);\n  const toolResult = await runToolChoice(message.toolCallRequest, tool);\n\n  return toolResult;\n};\n\nconst findTool = (toolName: string, toolRegistry: TamboToolRegistry) => {\n  const registryTool = toolRegistry[toolName];\n\n  if (!registryTool) {\n    throw new Error(`Tool ${toolName} not found in registry`);\n  }\n\n  const contextTool = mapTamboToolToContextTool(registryTool);\n  return {\n    getComponentContext: registryTool.tool,\n    definition: contextTool,\n  };\n};\n\nconst runToolChoice = async (\n  toolCallRequest: TamboAI.ToolCallRequest,\n  tool: ComponentContextTool,\n): Promise<any> => {\n  // Assumes parameters are in the order they are defined in the tool\n  const parameterValues =\n    toolCallRequest.parameters?.map((param) => param.parameterValue) ?? [];\n\n  return await tool.getComponentContext(...parameterValues);\n};\n"]}