@tambo-ai/react 0.12.0

package/dist/util/generate-component.js

@@ -0,0 +1,209 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateAndHydrate = generateAndHydrate;
+exports.renderComponentIntoMessage = renderComponentIntoMessage;
+const typescript_sdk_1 = require("@tambo-ai/typescript-sdk");
+const partial_json_1 = require("partial-json");
+const react_1 = __importDefault(require("react"));
+const zod_1 = require("zod");
+const use_current_message_1 = require("../hooks/use-current-message");
+const generate_component_response_1 = require("../model/generate-component-response");
+const registry_1 = require("../util/registry");
+const tool_caller_1 = require("../util/tool-caller");
+async function generateAndHydrate(client, componentList, toolRegistry, toolAssociations, currentThreadId, content, params, onUpdateState, stream = false, options) {
+    const availableComponents = (0, registry_1.getAvailableComponents)(componentList, toolRegistry, toolAssociations);
+    let threadId = currentThreadId;
+    onUpdateState(generate_component_response_1.GenerationStage.CHOOSING_COMPONENT);
+    try {
+        if (stream) {
+            const streamResponse = await (0, typescript_sdk_1.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(generate_component_response_1.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(generate_component_response_1.GenerationStage.COMPLETE);
+        return componentResponse;
+    }
+    if (message.toolCallRequest) {
+        onUpdateState(generate_component_response_1.GenerationStage.FETCHING_CONTEXT);
+        const toolResult = await (0, tool_caller_1.handleToolCall)(message, toolRegistry);
+        onUpdateState(generate_component_response_1.GenerationStage.HYDRATING_COMPONENT);
+        try {
+            const hydrationResponse = await client.beta.components.hydrate({
+                component: (0, registry_1.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(generate_component_response_1.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_1.default.createElement((0, registry_1.getComponentFromRegistry)(message.component.componentName, componentList)
+        .component, message.component.props);
+    const componentResponse = {
+        ...message,
+        renderedComponent: (0, use_current_message_1.wrapWithTamboMessageProvider)(renderedComponent, message.threadId, message.id),
+    };
+    onUpdateState(generate_component_response_1.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(generate_component_response_1.GenerationStage.STREAMING_RESPONSE);
+            accumulatedComponentResponse = {
+                ...chunk,
+                renderedComponent: null,
+            };
+            yield accumulatedComponentResponse;
+        }
+        if (chunk.toolCallRequest) {
+            if (!chunk.component?.componentName) {
+                continue;
+            }
+            onUpdateState(generate_component_response_1.GenerationStage.FETCHING_CONTEXT);
+            const toolResult = await (0, tool_caller_1.handleToolCall)(chunk, toolRegistry);
+            onUpdateState(generate_component_response_1.GenerationStage.HYDRATING_COMPONENT);
+            const streamResponse = await (0, typescript_sdk_1.hydrateStream)(client, {
+                component: (0, registry_1.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(generate_component_response_1.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(generate_component_response_1.GenerationStage.COMPLETE);
+}
+/**
+ * Generate a message that has a component rendered into it, if the message came with one
+ */
+function renderComponentIntoMessage(message, componentList) {
+    if (!message.component?.componentName) {
+        throw new Error("Component not found");
+    }
+    const parsedProps = (0, partial_json_1.parse)(JSON.stringify(message.component.props));
+    const registeredComponent = (0, registry_1.getComponentFromRegistry)(message.component.componentName, componentList);
+    const validatedProps = registeredComponent.props instanceof zod_1.z.ZodType
+        ? registeredComponent.props.parse(parsedProps)
+        : parsedProps;
+    const renderedComponent = react_1.default.createElement(registeredComponent.component, validatedProps);
+    const wrappedComponent = (0, use_current_message_1.wrapWithTamboMessageProvider)(renderedComponent, message.threadId, message.id);
+    return {
+        ...message,
+        component: {
+            ...message.component,
+            props: validatedProps,
+        },
+        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/dist/util/generate-component.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-component.js","sourceRoot":"","sources":["../../src/util/generate-component.ts"],"names":[],"mappings":";;;;;AAuBA,gDA+FC;AAiKD,gEAoCC;AA3TD,6DAGkC;AAClC,+CAAqC;AACrC,kDAA0B;AAC1B,6BAAwB;AACxB,sEAA4E;AAM5E,sFAG8C;AAC9C,+CAG0B;AAC1B,qDAAqD;AAE9C,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,IAAA,iCAAsB,EAChD,aAAa,EACb,YAAY,EACZ,gBAAgB,CACjB,CAAC;IACF,IAAI,QAAQ,GAAG,eAAe,CAAC;IAE/B,aAAa,CAAC,6CAAe,CAAC,kBAAkB,CAAC,CAAC;IAClD,IAAI,CAAC;QACH,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,cAAc,GAAG,MAAM,IAAA,+BAAc,EACzC,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,6CAAe,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,6CAAe,CAAC,QAAQ,CAAC,CAAC;QACxC,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;QAC5B,aAAa,CAAC,6CAAe,CAAC,gBAAgB,CAAC,CAAC;QAChD,MAAM,UAAU,GAAG,MAAM,IAAA,4BAAc,EAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAC/D,aAAa,CAAC,6CAAe,CAAC,mBAAmB,CAAC,CAAC;QACnD,IAAI,CAAC;YACH,MAAM,iBAAiB,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAC5D;gBACE,SAAS,EAAE,IAAA,mCAAwB,EACjC,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,6CAAe,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,eAAK,CAAC,aAAa,CAC3C,IAAA,mCAAwB,EAAC,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,IAAA,kDAA4B,EAC7C,iBAAiB,EACjB,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,EAAE,CACX;KACF,CAAC;IAEF,aAAa,CAAC,6CAAe,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,6CAAe,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,6CAAe,CAAC,gBAAgB,CAAC,CAAC;YAChD,MAAM,UAAU,GAAG,MAAM,IAAA,4BAAc,EAAC,KAAK,EAAE,YAAY,CAAC,CAAC;YAC7D,aAAa,CAAC,6CAAe,CAAC,mBAAmB,CAAC,CAAC;YACnD,MAAM,cAAc,GAAG,MAAM,IAAA,8BAAa,EACxC,MAAM,EACN;gBACE,SAAS,EAAE,IAAA,mCAAwB,EACjC,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,6CAAe,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,6CAAe,CAAC,QAAQ,CAAC,CAAC;AAC1C,CAAC;AAED;;GAEG;AACH,SAAgB,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,IAAA,oBAAK,EAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;IACnE,MAAM,mBAAmB,GAAG,IAAA,mCAAwB,EAClD,OAAO,CAAC,SAAS,CAAC,aAAa,EAC/B,aAAa,CACd,CAAC;IAEF,MAAM,cAAc,GAClB,mBAAmB,CAAC,KAAK,YAAY,OAAC,CAAC,OAAO;QAC5C,CAAC,CAAC,mBAAmB,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;QAC9C,CAAC,CAAC,WAAW,CAAC;IAElB,MAAM,iBAAiB,GAAG,eAAK,CAAC,aAAa,CAC3C,mBAAmB,CAAC,SAAS,EAC7B,cAAc,CACf,CAAC;IACF,MAAM,gBAAgB,GAAG,IAAA,kDAA4B,EACnD,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"]}

package/dist/util/messages.d.ts

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

package/dist/util/messages.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.threadMessagesToLegacyMessages = threadMessagesToLegacyMessages;
+function threadMessagesToLegacyMessages(messages) {
+    return messages
+        .map((m) => ({
+        sender: m.role,
+        message: m.content[0].text ?? "",
+    }))
+        .filter((m) => !!m.message);
+}
+//# sourceMappingURL=messages.js.map

package/dist/util/messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.js","sourceRoot":"","sources":["../../src/util/messages.ts"],"names":[],"mappings":";;AAEA,wEAWC;AAXD,SAAgB,8BAA8B,CAC5C,QAA8C;IAE9C,OAAO,QAAQ;SACZ,GAAG,CACF,CAAC,CAAC,EAAwC,EAAE,CAAC,CAAC;QAC5C,MAAM,EAAE,CAAC,CAAC,IAAwB;QAClC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE;KACjC,CAAC,CACH;SACA,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;AAChC,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\n\nexport function threadMessagesToLegacyMessages(\n  messages: TamboAI.Beta.Threads.ThreadMessage[],\n) {\n  return messages\n    .map(\n      (m): TamboAI.Components.LegacyChatMessage => ({\n        sender: m.role as \"user\" | \"hydra\",\n        message: m.content[0].text ?? \"\",\n      }),\n    )\n    .filter((m) => !!m.message);\n}\n"]}

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

@@ -0,0 +1,5 @@
+import { UseMutationResult, UseQueryResult } from "@tanstack/react-query";
+export type CombinedMutationResult<TData = unknown, TError = unknown, TVariables = unknown, TContext = unknown> = Omit<UseMutationResult<TData, TError, TVariables, TContext>, "mutate" | "mutateAsync" | "reset" | "data" | "variables" | "context">;
+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>;

package/dist/util/query-utils.js

@@ -0,0 +1,63 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.combineMutationResults = combineMutationResults;
+exports.combineQueryResults = combineQueryResults;
+function combineMutationResults(resultA, resultB) {
+    {
+        return {
+            isPending: resultA.isPending || resultB.isPending,
+            isSuccess: resultA.isSuccess && resultB.isSuccess,
+            isError: resultA.isError || resultB.isError,
+            isIdle: resultA.isIdle && resultB.isIdle,
+            isPaused: resultA.isPaused || resultB.isPaused,
+            submittedAt: resultA.submittedAt || resultB.submittedAt,
+            status: resultA.isPending || resultB.isPending
+                ? "pending"
+                : resultA.isError || resultB.isError
+                    ? "error"
+                    : resultA.isSuccess && resultB.isSuccess
+                        ? "success"
+                        : "idle",
+            error: resultA.error ?? resultB.error,
+            failureCount: resultA.failureCount + resultB.failureCount,
+            failureReason: resultA.failureReason ?? resultB.failureReason,
+        };
+    }
+}
+function combineQueryResults(resultA, resultB) {
+    return {
+        isPending: resultA.isPending || resultB.isPending,
+        isSuccess: resultA.isSuccess && resultB.isSuccess,
+        isError: resultA.isError || resultB.isError,
+        isLoading: resultA.isLoading || resultB.isLoading,
+        isFetched: resultA.isFetched && resultB.isFetched,
+        isFetchedAfterMount: resultA.isFetchedAfterMount && resultB.isFetchedAfterMount,
+        isInitialLoading: resultA.isInitialLoading || resultB.isInitialLoading,
+        isPaused: resultA.isPaused || resultB.isPaused,
+        isLoadingError: resultA.isLoadingError || resultB.isLoadingError,
+        isRefetchError: resultA.isRefetchError || resultB.isRefetchError,
+        isPlaceholderData: resultA.isPlaceholderData || resultB.isPlaceholderData,
+        isStale: resultA.isStale || resultB.isStale,
+        isRefetching: resultA.isRefetching || resultB.isRefetching,
+        isFetching: resultA.isFetching || resultB.isFetching,
+        status: resultA.isPending || resultB.isPending
+            ? "pending"
+            : resultA.isError || resultB.isError
+                ? "error"
+                : resultA.isSuccess && resultB.isSuccess
+                    ? "success"
+                    : "pending",
+        error: resultA.error ?? resultB.error,
+        failureCount: resultA.failureCount + resultB.failureCount,
+        failureReason: resultA.failureReason ?? resultB.failureReason,
+        errorUpdateCount: resultA.errorUpdateCount + resultB.errorUpdateCount,
+        fetchStatus: resultA.isFetching || resultB.isFetching
+            ? "fetching"
+            : resultA.isPaused || resultB.isPaused
+                ? "paused"
+                : "idle",
+        dataUpdatedAt: Math.max(resultA.dataUpdatedAt, resultB.dataUpdatedAt),
+        errorUpdatedAt: Math.max(resultA.errorUpdatedAt, resultB.errorUpdatedAt),
+    };
+}
+//# sourceMappingURL=query-utils.js.map

package/dist/util/query-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-utils.js","sourceRoot":"","sources":["../../src/util/query-utils.ts"],"names":[],"mappings":";;AAYA,wDAyBC;AAMD,kDAyCC;AAxED,SAAgB,sBAAsB,CACpC,OAAqD,EACrD,OAAqD;IAErD,CAAC;QACC,OAAO;YACL,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS;YACjD,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS;YACjD,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO;YAC3C,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM;YACxC,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ;YAC9C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,WAAW;YACvD,MAAM,EACJ,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS;gBACpC,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO;oBAClC,CAAC,CAAC,OAAO;oBACT,CAAC,CAAC,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS;wBACtC,CAAC,CAAC,SAAS;wBACX,CAAC,CAAC,MAAM;YAChB,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK;YACrC,YAAY,EAAE,OAAO,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY;YACzD,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,OAAO,CAAC,aAAa;SAC9D,CAAC;IACJ,CAAC;AACH,CAAC;AAMD,SAAgB,mBAAmB,CACjC,OAAwC,EACxC,OAAwC;IAExC,OAAO;QACL,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS;QACjD,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS;QACjD,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO;QAC3C,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS;QACjD,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS;QACjD,mBAAmB,EACjB,OAAO,CAAC,mBAAmB,IAAI,OAAO,CAAC,mBAAmB;QAC5D,gBAAgB,EAAE,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,gBAAgB;QACtE,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ;QAC9C,cAAc,EAAE,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,cAAc;QAChE,cAAc,EAAE,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,cAAc;QAChE,iBAAiB,EAAE,OAAO,CAAC,iBAAiB,IAAI,OAAO,CAAC,iBAAiB;QACzE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO;QAC3C,YAAY,EAAE,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,YAAY;QAC1D,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU;QACpD,MAAM,EACJ,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS;YACpC,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO;gBAClC,CAAC,CAAC,OAAO;gBACT,CAAC,CAAC,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS;oBACtC,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,SAAS;QACnB,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK;QACrC,YAAY,EAAE,OAAO,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY;QACzD,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,OAAO,CAAC,aAAa;QAC7D,gBAAgB,EAAE,OAAO,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB;QACrE,WAAW,EACT,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU;YACtC,CAAC,CAAC,UAAU;YACZ,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ;gBACpC,CAAC,CAAC,QAAQ;gBACV,CAAC,CAAC,MAAM;QACd,aAAa,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,aAAa,CAAC;QACrE,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,EAAE,OAAO,CAAC,cAAc,CAAC;KACzE,CAAC;AACJ,CAAC","sourcesContent":["import { UseMutationResult, UseQueryResult } from \"@tanstack/react-query\";\n\nexport type CombinedMutationResult<\n  TData = unknown,\n  TError = unknown,\n  TVariables = unknown,\n  TContext = unknown,\n> = Omit<\n  UseMutationResult<TData, TError, TVariables, TContext>,\n  \"mutate\" | \"mutateAsync\" | \"reset\" | \"data\" | \"variables\" | \"context\"\n>;\n\nexport function combineMutationResults<TData1, TData2, TError1, TError2>(\n  resultA: UseMutationResult<TData1, TError1, any, any>,\n  resultB: UseMutationResult<TData2, TError2, any, any>,\n): CombinedMutationResult<TData1 | TData2, TError1 | TError2> {\n  {\n    return {\n      isPending: resultA.isPending || resultB.isPending,\n      isSuccess: resultA.isSuccess && resultB.isSuccess,\n      isError: resultA.isError || resultB.isError,\n      isIdle: resultA.isIdle && resultB.isIdle,\n      isPaused: resultA.isPaused || resultB.isPaused,\n      submittedAt: resultA.submittedAt || resultB.submittedAt,\n      status:\n        resultA.isPending || resultB.isPending\n          ? \"pending\"\n          : resultA.isError || resultB.isError\n            ? \"error\"\n            : resultA.isSuccess && resultB.isSuccess\n              ? \"success\"\n              : \"idle\",\n      error: resultA.error ?? resultB.error,\n      failureCount: resultA.failureCount + resultB.failureCount,\n      failureReason: resultA.failureReason ?? resultB.failureReason,\n    };\n  }\n}\n\nexport type CombinedQueryResult<TData1, TData2, TError1, TError2> = Omit<\n  UseQueryResult<TData1 | TData2, TError1 | TError2>,\n  \"data\" | \"refetch\" | \"promise\"\n>;\nexport function combineQueryResults<TData1, TData2, TError1, TError2>(\n  resultA: UseQueryResult<TData1, TError1>,\n  resultB: UseQueryResult<TData2, TError2>,\n): CombinedQueryResult<void, void, TError1, TError2> {\n  return {\n    isPending: resultA.isPending || resultB.isPending,\n    isSuccess: resultA.isSuccess && resultB.isSuccess,\n    isError: resultA.isError || resultB.isError,\n    isLoading: resultA.isLoading || resultB.isLoading,\n    isFetched: resultA.isFetched && resultB.isFetched,\n    isFetchedAfterMount:\n      resultA.isFetchedAfterMount && resultB.isFetchedAfterMount,\n    isInitialLoading: resultA.isInitialLoading || resultB.isInitialLoading,\n    isPaused: resultA.isPaused || resultB.isPaused,\n    isLoadingError: resultA.isLoadingError || resultB.isLoadingError,\n    isRefetchError: resultA.isRefetchError || resultB.isRefetchError,\n    isPlaceholderData: resultA.isPlaceholderData || resultB.isPlaceholderData,\n    isStale: resultA.isStale || resultB.isStale,\n    isRefetching: resultA.isRefetching || resultB.isRefetching,\n    isFetching: resultA.isFetching || resultB.isFetching,\n    status:\n      resultA.isPending || resultB.isPending\n        ? \"pending\"\n        : resultA.isError || resultB.isError\n          ? \"error\"\n          : resultA.isSuccess && resultB.isSuccess\n            ? \"success\"\n            : \"pending\",\n    error: resultA.error ?? resultB.error,\n    failureCount: resultA.failureCount + resultB.failureCount,\n    failureReason: resultA.failureReason ?? resultB.failureReason,\n    errorUpdateCount: resultA.errorUpdateCount + resultB.errorUpdateCount,\n    fetchStatus:\n      resultA.isFetching || resultB.isFetching\n        ? \"fetching\"\n        : resultA.isPaused || resultB.isPaused\n          ? \"paused\"\n          : \"idle\",\n    dataUpdatedAt: Math.max(resultA.dataUpdatedAt, resultB.dataUpdatedAt),\n    errorUpdatedAt: Math.max(resultA.errorUpdatedAt, resultB.errorUpdatedAt),\n  };\n}\n"]}

package/dist/util/registry.d.ts

@@ -0,0 +1,7 @@
+import TamboAI from "@tambo-ai/typescript-sdk";
+import { ComponentContextToolMetadata, ComponentRegistry, RegisteredComponent, TamboTool, TamboToolAssociations, TamboToolRegistry } from "../model/component-metadata";
+export declare const getAvailableComponents: (componentRegistry: ComponentRegistry, toolRegistry: TamboToolRegistry, toolAssociations: TamboToolAssociations) => TamboAI.AvailableComponent[];
+export declare const getComponentFromRegistry: (componentName: string, componentRegistry: ComponentRegistry) => RegisteredComponent;
+export declare const getDefaultContextAdditions: () => string[];
+export declare const getClientContext: () => string;
+export declare const mapTamboToolToContextTool: (tool: TamboTool) => ComponentContextToolMetadata;

package/dist/util/registry.js

@@ -0,0 +1,99 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mapTamboToolToContextTool = exports.getClientContext = exports.getDefaultContextAdditions = exports.getComponentFromRegistry = exports.getAvailableComponents = void 0;
+const zod_to_json_schema_1 = __importDefault(require("zod-to-json-schema"));
+const getAvailableComponents = (componentRegistry, toolRegistry, toolAssociations) => {
+    const availableComponents = [];
+    for (const [name, componentEntry] of Object.entries(componentRegistry)) {
+        const associatedToolNames = toolAssociations[name] || [];
+        const contextTools = [
+            ...associatedToolNames.map((toolName) => {
+                const tool = toolRegistry[toolName];
+                if (!tool)
+                    return null;
+                return (0, exports.mapTamboToolToContextTool)(tool);
+            }),
+        ].filter((tool) => tool !== null);
+        availableComponents.push({
+            name: componentEntry.name,
+            description: componentEntry.description,
+            props: componentEntry.props,
+            contextTools,
+        });
+    }
+    return availableComponents;
+};
+exports.getAvailableComponents = getAvailableComponents;
+const getComponentFromRegistry = (componentName, componentRegistry) => {
+    const componentEntry = componentRegistry[componentName];
+    if (!componentEntry) {
+        throw new Error(`Tambo tried to use Component ${componentName}, but it was not found.`);
+    }
+    return componentEntry;
+};
+exports.getComponentFromRegistry = getComponentFromRegistry;
+const getDefaultContextAdditions = () => {
+    const utcOffsetHours = new Date().getTimezoneOffset() / 60;
+    const utcOffset = `(UTC${utcOffsetHours > 0 ? "+" : ""}${utcOffsetHours})`;
+    return [
+        `The current time in user's timezone (${utcOffset}) is: ${new Date().toLocaleString()}`,
+    ];
+};
+exports.getDefaultContextAdditions = getDefaultContextAdditions;
+const getClientContext = () => {
+    const contextAdditions = (0, exports.getDefaultContextAdditions)();
+    return contextAdditions.join("\n");
+};
+exports.getClientContext = getClientContext;
+const mapTamboToolToContextTool = (tool) => {
+    const parameters = getParametersFromZodFunction(tool.toolSchema);
+    return {
+        name: tool.name,
+        description: tool.description,
+        parameters,
+    };
+};
+exports.mapTamboToolToContextTool = mapTamboToolToContextTool;
+const getParametersFromZodFunction = (schema) => {
+    const parameters = schema.parameters();
+    return parameters.items.map((param, index) => {
+        const name = `param${index + 1}`;
+        const type = getZodBaseType(param);
+        const description = param.description ?? "";
+        const isRequired = !param.isOptional();
+        const schema = (0, zod_to_json_schema_1.default)(param);
+        return {
+            name,
+            type,
+            description,
+            isRequired,
+            schema,
+        };
+    });
+};
+const getZodBaseType = (schema) => {
+    const typeName = schema._def.typeName;
+    switch (typeName) {
+        case "ZodString":
+            return "string";
+        case "ZodNumber":
+            return "number";
+        case "ZodBoolean":
+            return "boolean";
+        case "ZodArray":
+            return "array";
+        case "ZodEnum":
+            return "enum";
+        case "ZodDate":
+            return "date";
+        case "ZodObject":
+            return "object";
+        default:
+            console.warn("falling back to string for", typeName);
+            return "string";
+    }
+};
+//# sourceMappingURL=registry.js.map

package/dist/util/registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../src/util/registry.ts"],"names":[],"mappings":";;;;;;AAEA,4EAAiD;AAW1C,MAAM,sBAAsB,GAAG,CACpC,iBAAoC,EACpC,YAA+B,EAC/B,gBAAuC,EACT,EAAE;IAChC,MAAM,mBAAmB,GAAiC,EAAE,CAAC;IAE7D,KAAK,MAAM,CAAC,IAAI,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACvE,MAAM,mBAAmB,GAAG,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAEzD,MAAM,YAAY,GAAG;YACnB,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;gBACtC,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;gBACpC,IAAI,CAAC,IAAI;oBAAE,OAAO,IAAI,CAAC;gBACvB,OAAO,IAAA,iCAAyB,EAAC,IAAI,CAAC,CAAC;YACzC,CAAC,CAAC;SACH,CAAC,MAAM,CAAC,CAAC,IAAI,EAAwC,EAAE,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QAExE,mBAAmB,CAAC,IAAI,CAAC;YACvB,IAAI,EAAE,cAAc,CAAC,IAAI;YACzB,WAAW,EAAE,cAAc,CAAC,WAAW;YACvC,KAAK,EAAE,cAAc,CAAC,KAAK;YAC3B,YAAY;SACb,CAAC,CAAC;IACL,CAAC;IAED,OAAO,mBAAmB,CAAC;AAC7B,CAAC,CAAC;AA3BW,QAAA,sBAAsB,0BA2BjC;AAEK,MAAM,wBAAwB,GAAG,CACtC,aAAqB,EACrB,iBAAoC,EACf,EAAE;IACvB,MAAM,cAAc,GAAG,iBAAiB,CAAC,aAAa,CAAC,CAAC;IAExD,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,MAAM,IAAI,KAAK,CACb,gCAAgC,aAAa,yBAAyB,CACvE,CAAC;IACJ,CAAC;IAED,OAAO,cAAc,CAAC;AACxB,CAAC,CAAC;AAbW,QAAA,wBAAwB,4BAanC;AAEK,MAAM,0BAA0B,GAAG,GAAa,EAAE;IACvD,MAAM,cAAc,GAAG,IAAI,IAAI,EAAE,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC;IAC3D,MAAM,SAAS,GAAG,OAAO,cAAc,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,cAAc,GAAG,CAAC;IAC3E,OAAO;QACL,wCAAwC,SAAS,SAAS,IAAI,IAAI,EAAE,CAAC,cAAc,EAAE,EAAE;KACxF,CAAC;AACJ,CAAC,CAAC;AANW,QAAA,0BAA0B,8BAMrC;AAEK,MAAM,gBAAgB,GAAG,GAAW,EAAE;IAC3C,MAAM,gBAAgB,GAAG,IAAA,kCAA0B,GAAE,CAAC;IACtD,OAAO,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACrC,CAAC,CAAC;AAHW,QAAA,gBAAgB,oBAG3B;AAEK,MAAM,yBAAyB,GAAG,CACvC,IAAe,EACe,EAAE;IAChC,MAAM,UAAU,GAAG,4BAA4B,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAEjE,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,UAAU;KACX,CAAC;AACJ,CAAC,CAAC;AAVW,QAAA,yBAAyB,6BAUpC;AAEF,MAAM,4BAA4B,GAAG,CACnC,MAA+B,EACd,EAAE;IACnB,MAAM,UAAU,GAAe,MAAM,CAAC,UAAU,EAAE,CAAC;IACnD,OAAO,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAiB,EAAE;QAC1D,MAAM,IAAI,GAAG,QAAQ,KAAK,GAAG,CAAC,EAAE,CAAC;QACjC,MAAM,IAAI,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,EAAE,CAAC;QAC5C,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,IAAA,4BAAe,EAAC,KAAK,CAAC,CAAC;QAEtC,OAAO;YACL,IAAI;YACJ,IAAI;YACJ,WAAW;YACX,UAAU;YACV,MAAM;SACP,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,MAAM,cAAc,GAAG,CAAC,MAAoB,EAAU,EAAE;IACtD,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC;IACtC,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,WAAW;YACd,OAAO,QAAQ,CAAC;QAClB,KAAK,WAAW;YACd,OAAO,QAAQ,CAAC;QAClB,KAAK,YAAY;YACf,OAAO,SAAS,CAAC;QACnB,KAAK,UAAU;YACb,OAAO,OAAO,CAAC;QACjB,KAAK,SAAS;YACZ,OAAO,MAAM,CAAC;QAChB,KAAK,SAAS;YACZ,OAAO,MAAM,CAAC;QAChB,KAAK,WAAW;YACd,OAAO,QAAQ,CAAC;QAClB;YACE,OAAO,CAAC,IAAI,CAAC,4BAA4B,EAAE,QAAQ,CAAC,CAAC;YACrD,OAAO,QAAQ,CAAC;IACpB,CAAC;AACH,CAAC,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { z } from \"zod\";\nimport zodToJsonSchema from \"zod-to-json-schema\";\nimport {\n  ComponentContextToolMetadata,\n  ComponentRegistry,\n  ParameterSpec,\n  RegisteredComponent,\n  TamboTool,\n  TamboToolAssociations,\n  TamboToolRegistry,\n} from \"../model/component-metadata\";\n\nexport const getAvailableComponents = (\n  componentRegistry: ComponentRegistry,\n  toolRegistry: TamboToolRegistry,\n  toolAssociations: TamboToolAssociations,\n): TamboAI.AvailableComponent[] => {\n  const availableComponents: TamboAI.AvailableComponent[] = [];\n\n  for (const [name, componentEntry] of Object.entries(componentRegistry)) {\n    const associatedToolNames = toolAssociations[name] || [];\n\n    const contextTools = [\n      ...associatedToolNames.map((toolName) => {\n        const tool = toolRegistry[toolName];\n        if (!tool) return null;\n        return mapTamboToolToContextTool(tool);\n      }),\n    ].filter((tool): tool is ComponentContextToolMetadata => tool !== null);\n\n    availableComponents.push({\n      name: componentEntry.name,\n      description: componentEntry.description,\n      props: componentEntry.props,\n      contextTools,\n    });\n  }\n\n  return availableComponents;\n};\n\nexport const getComponentFromRegistry = (\n  componentName: string,\n  componentRegistry: ComponentRegistry,\n): RegisteredComponent => {\n  const componentEntry = componentRegistry[componentName];\n\n  if (!componentEntry) {\n    throw new Error(\n      `Tambo tried to use Component ${componentName}, but it was not found.`,\n    );\n  }\n\n  return componentEntry;\n};\n\nexport const getDefaultContextAdditions = (): string[] => {\n  const utcOffsetHours = new Date().getTimezoneOffset() / 60;\n  const utcOffset = `(UTC${utcOffsetHours > 0 ? \"+\" : \"\"}${utcOffsetHours})`;\n  return [\n    `The current time in user's timezone (${utcOffset}) is: ${new Date().toLocaleString()}`,\n  ];\n};\n\nexport const getClientContext = (): string => {\n  const contextAdditions = getDefaultContextAdditions();\n  return contextAdditions.join(\"\\n\");\n};\n\nexport const mapTamboToolToContextTool = (\n  tool: TamboTool,\n): ComponentContextToolMetadata => {\n  const parameters = getParametersFromZodFunction(tool.toolSchema);\n\n  return {\n    name: tool.name,\n    description: tool.description,\n    parameters,\n  };\n};\n\nconst getParametersFromZodFunction = (\n  schema: z.ZodFunction<any, any>,\n): ParameterSpec[] => {\n  const parameters: z.ZodTuple = schema.parameters();\n  return parameters.items.map((param, index): ParameterSpec => {\n    const name = `param${index + 1}`;\n    const type = getZodBaseType(param);\n    const description = param.description ?? \"\";\n    const isRequired = !param.isOptional();\n    const schema = zodToJsonSchema(param);\n\n    return {\n      name,\n      type,\n      description,\n      isRequired,\n      schema,\n    };\n  });\n};\n\nconst getZodBaseType = (schema: z.ZodTypeAny): string => {\n  const typeName = schema._def.typeName;\n  switch (typeName) {\n    case \"ZodString\":\n      return \"string\";\n    case \"ZodNumber\":\n      return \"number\";\n    case \"ZodBoolean\":\n      return \"boolean\";\n    case \"ZodArray\":\n      return \"array\";\n    case \"ZodEnum\":\n      return \"enum\";\n    case \"ZodDate\":\n      return \"date\";\n    case \"ZodObject\":\n      return \"object\";\n    default:\n      console.warn(\"falling back to string for\", typeName);\n      return \"string\";\n  }\n};\n"]}

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

@@ -0,0 +1,3 @@
+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>;

package/dist/util/tool-caller.js

@@ -0,0 +1,30 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.handleToolCall = void 0;
+const registry_1 = require("./registry");
+const handleToolCall = async (message, toolRegistry) => {
+    if (!message?.toolCallRequest?.toolName) {
+        throw new Error("Tool name is required");
+    }
+    const tool = findTool(message.toolCallRequest.toolName, toolRegistry);
+    const toolResult = await runToolChoice(message.toolCallRequest, tool);
+    return toolResult;
+};
+exports.handleToolCall = handleToolCall;
+const findTool = (toolName, toolRegistry) => {
+    const registryTool = toolRegistry[toolName];
+    if (!registryTool) {
+        throw new Error(`Tool ${toolName} not found in registry`);
+    }
+    const contextTool = (0, registry_1.mapTamboToolToContextTool)(registryTool);
+    return {
+        getComponentContext: registryTool.tool,
+        definition: contextTool,
+    };
+};
+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);
+};
+//# sourceMappingURL=tool-caller.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tool-caller.js","sourceRoot":"","sources":["../../src/util/tool-caller.ts"],"names":[],"mappings":";;;AAKA,yCAAuD;AAChD,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;AAZW,QAAA,cAAc,kBAYzB;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,IAAA,oCAAyB,EAAC,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"]}

package/esm/hooks/__tests__/use-suggestions.test.d.ts

@@ -0,0 +1 @@
+export {};