@tambo-ai/client 0.0.1

package/esm/tambo-stream.d.ts

@@ -0,0 +1,112 @@
+/**
+ * TamboStream - Streaming response handler for Tambo AI
+ *
+ * Provides two consumption modes:
+ * 1. AsyncIterable: `for await (const { event, snapshot } of stream) { ... }`
+ * 2. Promise: `const thread = await stream.thread`
+ *
+ * The internal processing loop always runs (fire-and-forget in constructor).
+ * The `.thread` promise resolves when the loop completes.
+ */
+import { type BaseEvent } from "@ag-ui/core";
+import type TamboAI from "@tambo-ai/typescript-sdk";
+import type { TamboThread } from "./types/thread.js";
+import type { ToolChoice } from "./types/tool-choice.js";
+import type { InputMessage } from "./types/message.js";
+import type { ComponentRegistry, TamboToolRegistry } from "./model/component-metadata.js";
+import type { StreamAction } from "./utils/event-accumulator.js";
+/**
+ * Event yielded by the TamboStream async iterator.
+ */
+export interface StreamEvent {
+    /** The raw AG-UI event. */
+    event: BaseEvent;
+    /** A snapshot of the thread state after this event was processed. */
+    snapshot: TamboThread;
+}
+/**
+ * Options for creating a TamboStream.
+ */
+export interface TamboStreamOptions {
+    /** The Tambo API client. */
+    client: TamboAI;
+    /** The message to send. */
+    message: InputMessage;
+    /** Existing thread ID (undefined = create new). */
+    threadId: string | undefined;
+    /** User message text for optimistic display. */
+    userMessageText?: string;
+    /** Registered components. */
+    componentList: ComponentRegistry;
+    /** Registered tools. */
+    toolRegistry: TamboToolRegistry;
+    /** User key for auth. */
+    userKey: string | undefined;
+    /** Previous run ID for continuations. */
+    previousRunId: string | undefined;
+    /** Additional context merged into message. */
+    additionalContext?: Record<string, unknown>;
+    /** Tool choice mode. */
+    toolChoice?: ToolChoice;
+    /** Whether to auto-execute tools. */
+    autoExecuteTools: boolean;
+    /** Max tool execution rounds. */
+    maxSteps: number;
+    /** Enable debug logging. */
+    debug: boolean;
+    /** AbortSignal for cancellation. */
+    signal?: AbortSignal;
+    /** Dispatch function to update client state. */
+    dispatch: (action: StreamAction) => void;
+    /** Accessor for current thread state snapshot. */
+    getThreadSnapshot: (threadId: string) => TamboThread | undefined;
+}
+/**
+ * TamboStream handles streaming AI responses with tool execution.
+ *
+ * Supports two consumption modes:
+ * - Async iteration: `for await (const { event, snapshot } of stream) { ... }`
+ * - Promise-based: `const thread = await stream.thread`
+ *
+ * The processing loop runs automatically. The async iterator can only be
+ * iterated once (like ReadableStream).
+ */
+export declare class TamboStream implements AsyncIterable<StreamEvent> {
+    private readonly options;
+    /** Promise that resolves to the final thread state when the stream completes. */
+    readonly thread: Promise<TamboThread>;
+    private eventQueue;
+    private resolveThread;
+    private rejectThread;
+    private streamError;
+    private abortController;
+    private consumed;
+    /**
+     * Create a new TamboStream.
+     * @param options - Configuration for the stream.
+     */
+    constructor(options: TamboStreamOptions);
+    /**
+     * Abort the stream. The `.thread` promise rejects with an AbortError.
+     * The async iterator ends cleanly.
+     */
+    abort(): void;
+    /**
+     * Returns an async iterator over stream events.
+     * Can only be iterated once.
+     * @returns An async iterator yielding StreamEvent pairs.
+     */
+    [Symbol.asyncIterator](): AsyncIterableIterator<StreamEvent>;
+    /**
+     * Main processing loop. Runs automatically in the constructor.
+     * Handles event streaming, tool execution, and stream stitching.
+     */
+    private processLoop;
+    /**
+     * Handle abort by transitioning thread to idle and rejecting the thread promise.
+     * @param threadId - The thread to clean up.
+     * @param dispatch - The dispatch function for state updates.
+     */
+    private handleAbort;
+}
+//# sourceMappingURL=tambo-stream.d.ts.map

package/esm/tambo-stream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-stream.d.ts","sourceRoot":"","sources":["../src/tambo-stream.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAa,KAAK,SAAS,EAAsB,MAAM,aAAa,CAAC;AAC5E,OAAO,KAAK,OAAO,MAAM,0BAA0B,CAAC;AAEpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAElD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,KAAK,EACV,iBAAiB,EACjB,iBAAiB,EAClB,MAAM,4BAA4B,CAAC;AACpC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAY9D;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,2BAA2B;IAC3B,KAAK,EAAE,SAAS,CAAC;IACjB,qEAAqE;IACrE,QAAQ,EAAE,WAAW,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,4BAA4B;IAC5B,MAAM,EAAE,OAAO,CAAC;IAChB,2BAA2B;IAC3B,OAAO,EAAE,YAAY,CAAC;IACtB,mDAAmD;IACnD,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,gDAAgD;IAChD,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,6BAA6B;IAC7B,aAAa,EAAE,iBAAiB,CAAC;IACjC,wBAAwB;IACxB,YAAY,EAAE,iBAAiB,CAAC;IAChC,yBAAyB;IACzB,OAAO,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,yCAAyC;IACzC,aAAa,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,8CAA8C;IAC9C,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC5C,wBAAwB;IACxB,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,qCAAqC;IACrC,gBAAgB,EAAE,OAAO,CAAC;IAC1B,iCAAiC;IACjC,QAAQ,EAAE,MAAM,CAAC;IACjB,4BAA4B;IAC5B,KAAK,EAAE,OAAO,CAAC;IACf,oCAAoC;IACpC,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,gDAAgD;IAChD,QAAQ,EAAE,CAAC,MAAM,EAAE,YAAY,KAAK,IAAI,CAAC;IACzC,kDAAkD;IAClD,iBAAiB,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,WAAW,GAAG,SAAS,CAAC;CAClE;AA0ED;;;;;;;;;GASG;AACH,qBAAa,WAAY,YAAW,aAAa,CAAC,WAAW,CAAC;IAehD,OAAO,CAAC,QAAQ,CAAC,OAAO;IAdpC,iFAAiF;IACjF,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAEtC,OAAO,CAAC,UAAU,CAAiC;IACnD,OAAO,CAAC,aAAa,CAAiC;IACtD,OAAO,CAAC,YAAY,CAA0B;IAC9C,OAAO,CAAC,WAAW,CAAsB;IACzC,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,QAAQ,CAAS;IAEzB;;;OAGG;gBAC0B,OAAO,EAAE,kBAAkB;IA4BxD;;;OAGG;IACH,KAAK,IAAI,IAAI;IAIb;;;;OAIG;IACH,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,qBAAqB,CAAC,WAAW,CAAC;IAkB5D;;;OAGG;YACW,WAAW;IAuNzB;;;;OAIG;IACH,OAAO,CAAC,WAAW;CAsBpB"}

package/esm/tambo-stream.js

@@ -0,0 +1,345 @@
+"use strict";
+/**
+ * TamboStream - Streaming response handler for Tambo AI
+ *
+ * Provides two consumption modes:
+ * 1. AsyncIterable: `for await (const { event, snapshot } of stream) { ... }`
+ * 2. Promise: `const thread = await stream.thread`
+ *
+ * The internal processing loop always runs (fire-and-forget in constructor).
+ * The `.thread` promise resolves when the loop completes.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TamboStream = void 0;
+const core_1 = require("@ag-ui/core");
+const event_1 = require("./types/event.js");
+const stream_handler_1 = require("./utils/stream-handler.js");
+const tool_call_tracker_1 = require("./utils/tool-call-tracker.js");
+const tool_executor_1 = require("./utils/tool-executor.js");
+const send_message_1 = require("./utils/send-message.js");
+/**
+ * Internal event queue for bridging the processing loop and async iterator.
+ * Events are pushed by the processing loop and pulled by the iterator.
+ */
+class EventQueue {
+    queue = [];
+    waiters = [];
+    done = false;
+    error = null;
+    /**
+     * Push an event into the queue.
+     * @param value - The event to push.
+     */
+    push(value) {
+        const waiter = this.waiters.shift();
+        if (waiter) {
+            waiter({ value, done: false });
+        }
+        else {
+            this.queue.push(value);
+        }
+    }
+    /**
+     * Close the queue normally.
+     */
+    close() {
+        this.done = true;
+        for (const waiter of this.waiters) {
+            waiter({ value: undefined, done: true });
+        }
+        this.waiters = [];
+    }
+    /**
+     * Close the queue with an error.
+     * @param err - The error to propagate to waiting consumers.
+     */
+    closeWithError(err) {
+        this.error = err;
+        this.done = true;
+        // Reject all waiters... but async iterator protocol uses throw, not reject.
+        // We'll store the error and throw it on next pull.
+        for (const waiter of this.waiters) {
+            waiter({ value: undefined, done: true });
+        }
+        this.waiters = [];
+    }
+    /**
+     * Pull the next event from the queue.
+     * @returns An iterator result with the next event or done signal.
+     */
+    async pull() {
+        if (this.error) {
+            const err = this.error;
+            this.error = null;
+            throw err;
+        }
+        const item = this.queue.shift();
+        if (item !== undefined) {
+            return { value: item, done: false };
+        }
+        if (this.done) {
+            return { value: undefined, done: true };
+        }
+        return await new Promise((resolve) => {
+            this.waiters.push(resolve);
+        });
+    }
+}
+/**
+ * TamboStream handles streaming AI responses with tool execution.
+ *
+ * Supports two consumption modes:
+ * - Async iteration: `for await (const { event, snapshot } of stream) { ... }`
+ * - Promise-based: `const thread = await stream.thread`
+ *
+ * The processing loop runs automatically. The async iterator can only be
+ * iterated once (like ReadableStream).
+ */
+class TamboStream {
+    options;
+    /** Promise that resolves to the final thread state when the stream completes. */
+    thread;
+    eventQueue = new EventQueue();
+    resolveThread;
+    rejectThread;
+    streamError = null;
+    abortController;
+    consumed = false;
+    /**
+     * Create a new TamboStream.
+     * @param options - Configuration for the stream.
+     */
+    constructor(options) {
+        this.options = options;
+        this.abortController = new AbortController();
+        // Wire external signal to our internal abort controller
+        if (options.signal) {
+            if (options.signal.aborted) {
+                this.abortController.abort(options.signal.reason);
+            }
+            else {
+                options.signal.addEventListener("abort", () => {
+                    this.abortController.abort(options.signal?.reason);
+                });
+            }
+        }
+        this.thread = new Promise((resolve, reject) => {
+            this.resolveThread = resolve;
+            this.rejectThread = reject;
+        });
+        // Fire-and-forget the processing loop
+        void this.processLoop().catch((err) => {
+            const error = err instanceof Error ? err : new Error(String(err));
+            this.streamError = error;
+            this.rejectThread(error);
+            this.eventQueue.closeWithError(error);
+        });
+    }
+    /**
+     * Abort the stream. The `.thread` promise rejects with an AbortError.
+     * The async iterator ends cleanly.
+     */
+    abort() {
+        this.abortController.abort(new Error("Stream aborted"));
+    }
+    /**
+     * Returns an async iterator over stream events.
+     * Can only be iterated once.
+     * @returns An async iterator yielding StreamEvent pairs.
+     */
+    [Symbol.asyncIterator]() {
+        if (this.consumed) {
+            throw new Error("TamboStream can only be iterated once");
+        }
+        this.consumed = true;
+        return {
+            next: async () => await this.eventQueue.pull(),
+            return: async () => {
+                this.eventQueue.close();
+                return { value: undefined, done: true };
+            },
+            [Symbol.asyncIterator]() {
+                return this;
+            },
+        };
+    }
+    /**
+     * Main processing loop. Runs automatically in the constructor.
+     * Handles event streaming, tool execution, and stream stitching.
+     */
+    async processLoop() {
+        const { client, message, threadId: initialThreadId, userMessageText, componentList, toolRegistry, userKey, previousRunId, additionalContext, toolChoice, autoExecuteTools, maxSteps, debug, dispatch, getThreadSnapshot, } = this.options;
+        // Generate user message ID if we have user text
+        const userMessageId = userMessageText
+            ? `user_msg_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`
+            : undefined;
+        // Dispatch user message immediately for optimistic display on existing threads
+        if (initialThreadId && userMessageText && userMessageId) {
+            (0, send_message_1.dispatchUserMessage)(dispatch, initialThreadId, userMessageId, userMessageText);
+        }
+        // Create the initial run stream
+        const { stream: initialStream, initialThreadId: streamThreadId } = await (0, send_message_1.createRunStream)({
+            client,
+            threadId: initialThreadId,
+            message,
+            componentList,
+            toolRegistry,
+            userKey,
+            previousRunId,
+            additionalContext,
+            toolChoice,
+        });
+        const toolTracker = new tool_call_tracker_1.ToolCallTracker(toolRegistry);
+        const throttledStreamable = (0, tool_executor_1.createThrottledStreamableExecutor)(toolTracker, toolRegistry);
+        let actualThreadId = streamThreadId;
+        let runId;
+        let currentStream = initialStream;
+        let stepCount = 0;
+        try {
+            while (true) {
+                // Check abort before processing each stream segment
+                if (this.abortController.signal.aborted) {
+                    this.handleAbort(actualThreadId, dispatch);
+                    return;
+                }
+                let pendingAwaitingInput;
+                for await (const event of (0, stream_handler_1.handleEventStream)(currentStream, { debug })) {
+                    // Check abort on each event
+                    if (this.abortController.signal.aborted) {
+                        this.handleAbort(actualThreadId, dispatch);
+                        return;
+                    }
+                    // Extract threadId and runId from RUN_STARTED
+                    if (event.type === core_1.EventType.RUN_STARTED) {
+                        runId = event.runId;
+                        actualThreadId ??= event.threadId;
+                        // For new threads: dispatch user message now that we have the real ID
+                        if (!initialThreadId && userMessageText && userMessageId) {
+                            (0, send_message_1.dispatchUserMessage)(dispatch, actualThreadId, userMessageId, userMessageText);
+                        }
+                    }
+                    else if (!actualThreadId) {
+                        throw new Error(`Expected first event to be RUN_STARTED with threadId, got: ${event.type}`);
+                    }
+                    toolTracker.handleEvent(event);
+                    // Parse partial JSON for TOOL_CALL_ARGS
+                    const parsedToolArgs = event.type === core_1.EventType.TOOL_CALL_ARGS
+                        ? toolTracker.parsePartialArgs(event.toolCallId)
+                        : undefined;
+                    dispatch({
+                        type: "EVENT",
+                        event,
+                        threadId: actualThreadId,
+                        parsedToolArgs,
+                        toolSchemas: toolTracker.toolSchemas,
+                    });
+                    // Schedule debounced streamable tool execution
+                    if (parsedToolArgs && event.type === core_1.EventType.TOOL_CALL_ARGS) {
+                        throttledStreamable.schedule(event.toolCallId, parsedToolArgs);
+                    }
+                    // Emit event to async iterator
+                    const snapshot = getThreadSnapshot(actualThreadId);
+                    if (snapshot) {
+                        this.eventQueue.push({ event, snapshot });
+                    }
+                    // Handle awaiting_input for tool execution
+                    if (event.type === core_1.EventType.CUSTOM) {
+                        const customEvent = (0, event_1.asTamboCustomEvent)(event);
+                        if (customEvent?.name === "tambo.run.awaiting_input") {
+                            pendingAwaitingInput = customEvent;
+                            break;
+                        }
+                    }
+                }
+                throttledStreamable.flush();
+                // Stream finished without awaiting_input - we're done
+                if (!pendingAwaitingInput) {
+                    break;
+                }
+                // Don't execute tools if auto-execute is off
+                if (!autoExecuteTools) {
+                    break;
+                }
+                // Check step limit
+                stepCount++;
+                if (stepCount >= maxSteps) {
+                    console.warn(`[TamboStream] maxSteps (${maxSteps}) reached. Stream resolving with pending tool calls.`);
+                    break;
+                }
+                // Execute tools and get continuation stream
+                if (!runId || !actualThreadId) {
+                    throw new Error("Cannot continue run after awaiting_input: missing runId or threadId");
+                }
+                const { stream: continuationStream, toolResults } = await (0, send_message_1.executeToolsAndContinue)({
+                    event: pendingAwaitingInput,
+                    toolTracker,
+                    toolRegistry,
+                    componentList,
+                    client,
+                    threadId: actualThreadId,
+                    runId,
+                    userKey,
+                    additionalContext,
+                    toolChoice,
+                });
+                (0, send_message_1.dispatchToolResults)(dispatch, actualThreadId, toolResults);
+                currentStream = continuationStream;
+            }
+            // Stream completed successfully
+            const finalSnapshot = actualThreadId
+                ? getThreadSnapshot(actualThreadId)
+                : undefined;
+            if (finalSnapshot) {
+                this.resolveThread(finalSnapshot);
+            }
+            else {
+                this.rejectThread(new Error("Stream completed but no thread state found"));
+            }
+            this.eventQueue.close();
+        }
+        catch (error) {
+            // Dispatch synthetic RUN_ERROR to clean up thread state
+            if (actualThreadId) {
+                const errorMessage = error instanceof Error ? error.message : "Unknown streaming error";
+                const errorEvent = {
+                    type: core_1.EventType.RUN_ERROR,
+                    message: errorMessage,
+                };
+                dispatch({
+                    type: "EVENT",
+                    event: errorEvent,
+                    threadId: actualThreadId,
+                });
+            }
+            const err = error instanceof Error ? error : new Error(String(error));
+            this.streamError = err;
+            this.rejectThread(err);
+            this.eventQueue.closeWithError(err);
+        }
+    }
+    /**
+     * Handle abort by transitioning thread to idle and rejecting the thread promise.
+     * @param threadId - The thread to clean up.
+     * @param dispatch - The dispatch function for state updates.
+     */
+    handleAbort(threadId, dispatch) {
+        if (threadId) {
+            const errorEvent = {
+                type: core_1.EventType.RUN_ERROR,
+                message: "Stream aborted",
+            };
+            dispatch({
+                type: "EVENT",
+                event: errorEvent,
+                threadId,
+            });
+        }
+        const abortError = new Error("Stream aborted");
+        abortError.name = "AbortError";
+        this.streamError = abortError;
+        this.rejectThread(abortError);
+        this.eventQueue.close(); // Iterator ends cleanly on abort
+    }
+}
+exports.TamboStream = TamboStream;
+//# sourceMappingURL=tambo-stream.js.map

package/esm/tambo-stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-stream.js","sourceRoot":"","sources":["../src/tambo-stream.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAEH,sCAA4E;AAI5E,yCAA+E;AAQ/E,2DAA2D;AAC3D,iEAA4D;AAC5D,yDAA0E;AAC1E,uDAM8B;AAkD9B;;;GAGG;AACH,MAAM,UAAU;IACN,KAAK,GAAQ,EAAE,CAAC;IAChB,OAAO,GAA2C,EAAE,CAAC;IACrD,IAAI,GAAG,KAAK,CAAC;IACb,KAAK,GAAiB,IAAI,CAAC;IAEnC;;;OAGG;IACH,IAAI,CAAC,KAAQ;QACX,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACpC,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,CAAC,EAAE,KAAK,EAAE,SAAyB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACpB,CAAC;IAED;;;OAGG;IACH,cAAc,CAAC,GAAU;QACvB,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,4EAA4E;QAC5E,mDAAmD;QACnD,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,CAAC,EAAE,KAAK,EAAE,SAAyB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACpB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,IAAI;QACR,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC;YACvB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;YAClB,MAAM,GAAG,CAAC;QACZ,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QAChC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QACtC,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,OAAO,EAAE,KAAK,EAAE,SAAyB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC1D,CAAC;QACD,OAAO,MAAM,IAAI,OAAO,CAAoB,CAAC,OAAO,EAAE,EAAE;YACtD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED;;;;;;;;;GASG;AACH,MAAa,WAAW;IAeO;IAd7B,iFAAiF;IACxE,MAAM,CAAuB;IAE9B,UAAU,GAAG,IAAI,UAAU,EAAe,CAAC;IAC3C,aAAa,CAAiC;IAC9C,YAAY,CAA0B;IACtC,WAAW,GAAiB,IAAI,CAAC;IACjC,eAAe,CAAkB;IACjC,QAAQ,GAAG,KAAK,CAAC;IAEzB;;;OAGG;IACH,YAA6B,OAA2B;QAA3B,YAAO,GAAP,OAAO,CAAoB;QACtD,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAE7C,wDAAwD;QACxD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,IAAI,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAC3B,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACpD,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;oBAC5C,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACrD,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,OAAO,CAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACzD,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;YAC7B,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,sCAAsC;QACtC,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACpC,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAClE,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;YACzB,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACH,KAAK;QACH,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC;IAC1D,CAAC;IAED;;;;OAIG;IACH,CAAC,MAAM,CAAC,aAAa,CAAC;QACpB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QAErB,OAAO;YACL,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE;YAC9C,MAAM,EAAE,KAAK,IAAI,EAAE;gBACjB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;gBACxB,OAAO,EAAE,KAAK,EAAE,SAAmC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;YACpE,CAAC;YACD,CAAC,MAAM,CAAC,aAAa,CAAC;gBACpB,OAAO,IAAI,CAAC;YACd,CAAC;SACF,CAAC;IACJ,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,WAAW;QACvB,MAAM,EACJ,MAAM,EACN,OAAO,EACP,QAAQ,EAAE,eAAe,EACzB,eAAe,EACf,aAAa,EACb,YAAY,EACZ,OAAO,EACP,aAAa,EACb,iBAAiB,EACjB,UAAU,EACV,gBAAgB,EAChB,QAAQ,EACR,KAAK,EACL,QAAQ,EACR,iBAAiB,GAClB,GAAG,IAAI,CAAC,OAAO,CAAC;QAEjB,gDAAgD;QAChD,MAAM,aAAa,GAAG,eAAe;YACnC,CAAC,CAAC,YAAY,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;YACpE,CAAC,CAAC,SAAS,CAAC;QAEd,+EAA+E;QAC/E,IAAI,eAAe,IAAI,eAAe,IAAI,aAAa,EAAE,CAAC;YACxD,IAAA,kCAAmB,EACjB,QAAQ,EACR,eAAe,EACf,aAAa,EACb,eAAe,CAChB,CAAC;QACJ,CAAC;QAED,gCAAgC;QAChC,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,eAAe,EAAE,cAAc,EAAE,GAC9D,MAAM,IAAA,8BAAe,EAAC;YACpB,MAAM;YACN,QAAQ,EAAE,eAAe;YACzB,OAAO;YACP,aAAa;YACb,YAAY;YACZ,OAAO;YACP,aAAa;YACb,iBAAiB;YACjB,UAAU;SACX,CAAC,CAAC;QAEL,MAAM,WAAW,GAAG,IAAI,mCAAe,CAAC,YAAY,CAAC,CAAC;QACtD,MAAM,mBAAmB,GAAG,IAAA,iDAAiC,EAC3D,WAAW,EACX,YAAY,CACb,CAAC;QAEF,IAAI,cAAc,GAAG,cAAc,CAAC;QACpC,IAAI,KAAyB,CAAC;QAC9B,IAAI,aAAa,GAAoC,aAAa,CAAC;QACnE,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,IAAI,CAAC;YACH,OAAO,IAAI,EAAE,CAAC;gBACZ,oDAAoD;gBACpD,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACxC,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;oBAC3C,OAAO;gBACT,CAAC;gBAED,IAAI,oBAAuD,CAAC;gBAE5D,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAA,kCAAiB,EAAC,aAAa,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;oBACtE,4BAA4B;oBAC5B,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;wBACxC,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;wBAC3C,OAAO;oBACT,CAAC;oBAED,8CAA8C;oBAC9C,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAS,CAAC,WAAW,EAAE,CAAC;wBACzC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;wBACpB,cAAc,KAAK,KAAK,CAAC,QAAQ,CAAC;wBAElC,sEAAsE;wBACtE,IAAI,CAAC,eAAe,IAAI,eAAe,IAAI,aAAa,EAAE,CAAC;4BACzD,IAAA,kCAAmB,EACjB,QAAQ,EACR,cAAc,EACd,aAAa,EACb,eAAe,CAChB,CAAC;wBACJ,CAAC;oBACH,CAAC;yBAAM,IAAI,CAAC,cAAc,EAAE,CAAC;wBAC3B,MAAM,IAAI,KAAK,CACb,8DAA8D,KAAK,CAAC,IAAI,EAAE,CAC3E,CAAC;oBACJ,CAAC;oBAED,WAAW,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;oBAE/B,wCAAwC;oBACxC,MAAM,cAAc,GAClB,KAAK,CAAC,IAAI,KAAK,gBAAS,CAAC,cAAc;wBACrC,CAAC,CAAC,WAAW,CAAC,gBAAgB,CAAC,KAAK,CAAC,UAAU,CAAC;wBAChD,CAAC,CAAC,SAAS,CAAC;oBAEhB,QAAQ,CAAC;wBACP,IAAI,EAAE,OAAO;wBACb,KAAK;wBACL,QAAQ,EAAE,cAAc;wBACxB,cAAc;wBACd,WAAW,EAAE,WAAW,CAAC,WAAW;qBACrC,CAAC,CAAC;oBAEH,+CAA+C;oBAC/C,IAAI,cAAc,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAS,CAAC,cAAc,EAAE,CAAC;wBAC9D,mBAAmB,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;oBACjE,CAAC;oBAED,+BAA+B;oBAC/B,MAAM,QAAQ,GAAG,iBAAiB,CAAC,cAAc,CAAC,CAAC;oBACnD,IAAI,QAAQ,EAAE,CAAC;wBACb,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;oBAC5C,CAAC;oBAED,2CAA2C;oBAC3C,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAS,CAAC,MAAM,EAAE,CAAC;wBACpC,MAAM,WAAW,GAAG,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC;wBAC9C,IAAI,WAAW,EAAE,IAAI,KAAK,0BAA0B,EAAE,CAAC;4BACrD,oBAAoB,GAAG,WAAW,CAAC;4BACnC,MAAM;wBACR,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,mBAAmB,CAAC,KAAK,EAAE,CAAC;gBAE5B,sDAAsD;gBACtD,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBAC1B,MAAM;gBACR,CAAC;gBAED,6CAA6C;gBAC7C,IAAI,CAAC,gBAAgB,EAAE,CAAC;oBACtB,MAAM;gBACR,CAAC;gBAED,mBAAmB;gBACnB,SAAS,EAAE,CAAC;gBACZ,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;oBAC1B,OAAO,CAAC,IAAI,CACV,2BAA2B,QAAQ,sDAAsD,CAC1F,CAAC;oBACF,MAAM;gBACR,CAAC;gBAED,4CAA4C;gBAC5C,IAAI,CAAC,KAAK,IAAI,CAAC,cAAc,EAAE,CAAC;oBAC9B,MAAM,IAAI,KAAK,CACb,qEAAqE,CACtE,CAAC;gBACJ,CAAC;gBAED,MAAM,EAAE,MAAM,EAAE,kBAAkB,EAAE,WAAW,EAAE,GAC/C,MAAM,IAAA,sCAAuB,EAAC;oBAC5B,KAAK,EAAE,oBAAoB;oBAC3B,WAAW;oBACX,YAAY;oBACZ,aAAa;oBACb,MAAM;oBACN,QAAQ,EAAE,cAAc;oBACxB,KAAK;oBACL,OAAO;oBACP,iBAAiB;oBACjB,UAAU;iBACX,CAAC,CAAC;gBAEL,IAAA,kCAAmB,EAAC,QAAQ,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;gBAC3D,aAAa,GAAG,kBAAkB,CAAC;YACrC,CAAC;YAED,gCAAgC;YAChC,MAAM,aAAa,GAAG,cAAc;gBAClC,CAAC,CAAC,iBAAiB,CAAC,cAAc,CAAC;gBACnC,CAAC,CAAC,SAAS,CAAC;YAEd,IAAI,aAAa,EAAE,CAAC;gBAClB,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,YAAY,CACf,IAAI,KAAK,CAAC,4CAA4C,CAAC,CACxD,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,wDAAwD;YACxD,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,yBAAyB,CAAC;gBACrE,MAAM,UAAU,GAAkB;oBAChC,IAAI,EAAE,gBAAS,CAAC,SAAS;oBACzB,OAAO,EAAE,YAAY;iBACtB,CAAC;gBACF,QAAQ,CAAC;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,UAAU;oBACjB,QAAQ,EAAE,cAAc;iBACzB,CAAC,CAAC;YACL,CAAC;YAED,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YACtE,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC;YACvB,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YACvB,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,WAAW,CACjB,QAA4B,EAC5B,QAAwC;QAExC,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,UAAU,GAAkB;gBAChC,IAAI,EAAE,gBAAS,CAAC,SAAS;gBACzB,OAAO,EAAE,gBAAgB;aAC1B,CAAC;YACF,QAAQ,CAAC;gBACP,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,UAAU;gBACjB,QAAQ;aACT,CAAC,CAAC;QACL,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAC/C,UAAU,CAAC,IAAI,GAAG,YAAY,CAAC;QAC/B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QAC9B,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC,iCAAiC;IAC5D,CAAC;CACF;AAhUD,kCAgUC","sourcesContent":["/**\n * TamboStream - Streaming response handler for Tambo AI\n *\n * Provides two consumption modes:\n * 1. AsyncIterable: `for await (const { event, snapshot } of stream) { ... }`\n * 2. Promise: `const thread = await stream.thread`\n *\n * The internal processing loop always runs (fire-and-forget in constructor).\n * The `.thread` promise resolves when the loop completes.\n */\n\nimport { EventType, type BaseEvent, type RunErrorEvent } from \"@ag-ui/core\";\nimport type TamboAI from \"@tambo-ai/typescript-sdk\";\n\nimport type { TamboThread } from \"./types/thread\";\nimport { asTamboCustomEvent, type RunAwaitingInputEvent } from \"./types/event\";\nimport type { ToolChoice } from \"./types/tool-choice\";\nimport type { InputMessage } from \"./types/message\";\nimport type {\n  ComponentRegistry,\n  TamboToolRegistry,\n} from \"./model/component-metadata\";\nimport type { StreamAction } from \"./utils/event-accumulator\";\nimport { handleEventStream } from \"./utils/stream-handler\";\nimport { ToolCallTracker } from \"./utils/tool-call-tracker\";\nimport { createThrottledStreamableExecutor } from \"./utils/tool-executor\";\nimport {\n  createRunStream,\n  dispatchUserMessage,\n  dispatchToolResults,\n  executeToolsAndContinue,\n  type CreateRunStreamResult,\n} from \"./utils/send-message\";\n\n/**\n * Event yielded by the TamboStream async iterator.\n */\nexport interface StreamEvent {\n  /** The raw AG-UI event. */\n  event: BaseEvent;\n  /** A snapshot of the thread state after this event was processed. */\n  snapshot: TamboThread;\n}\n\n/**\n * Options for creating a TamboStream.\n */\nexport interface TamboStreamOptions {\n  /** The Tambo API client. */\n  client: TamboAI;\n  /** The message to send. */\n  message: InputMessage;\n  /** Existing thread ID (undefined = create new). */\n  threadId: string | undefined;\n  /** User message text for optimistic display. */\n  userMessageText?: string;\n  /** Registered components. */\n  componentList: ComponentRegistry;\n  /** Registered tools. */\n  toolRegistry: TamboToolRegistry;\n  /** User key for auth. */\n  userKey: string | undefined;\n  /** Previous run ID for continuations. */\n  previousRunId: string | undefined;\n  /** Additional context merged into message. */\n  additionalContext?: Record<string, unknown>;\n  /** Tool choice mode. */\n  toolChoice?: ToolChoice;\n  /** Whether to auto-execute tools. */\n  autoExecuteTools: boolean;\n  /** Max tool execution rounds. */\n  maxSteps: number;\n  /** Enable debug logging. */\n  debug: boolean;\n  /** AbortSignal for cancellation. */\n  signal?: AbortSignal;\n  /** Dispatch function to update client state. */\n  dispatch: (action: StreamAction) => void;\n  /** Accessor for current thread state snapshot. */\n  getThreadSnapshot: (threadId: string) => TamboThread | undefined;\n}\n\n/**\n * Internal event queue for bridging the processing loop and async iterator.\n * Events are pushed by the processing loop and pulled by the iterator.\n */\nclass EventQueue<T> {\n  private queue: T[] = [];\n  private waiters: ((value: IteratorResult<T>) => void)[] = [];\n  private done = false;\n  private error: Error | null = null;\n\n  /**\n   * Push an event into the queue.\n   * @param value - The event to push.\n   */\n  push(value: T): void {\n    const waiter = this.waiters.shift();\n    if (waiter) {\n      waiter({ value, done: false });\n    } else {\n      this.queue.push(value);\n    }\n  }\n\n  /**\n   * Close the queue normally.\n   */\n  close(): void {\n    this.done = true;\n    for (const waiter of this.waiters) {\n      waiter({ value: undefined as unknown as T, done: true });\n    }\n    this.waiters = [];\n  }\n\n  /**\n   * Close the queue with an error.\n   * @param err - The error to propagate to waiting consumers.\n   */\n  closeWithError(err: Error): void {\n    this.error = err;\n    this.done = true;\n    // Reject all waiters... but async iterator protocol uses throw, not reject.\n    // We'll store the error and throw it on next pull.\n    for (const waiter of this.waiters) {\n      waiter({ value: undefined as unknown as T, done: true });\n    }\n    this.waiters = [];\n  }\n\n  /**\n   * Pull the next event from the queue.\n   * @returns An iterator result with the next event or done signal.\n   */\n  async pull(): Promise<IteratorResult<T>> {\n    if (this.error) {\n      const err = this.error;\n      this.error = null;\n      throw err;\n    }\n    const item = this.queue.shift();\n    if (item !== undefined) {\n      return { value: item, done: false };\n    }\n    if (this.done) {\n      return { value: undefined as unknown as T, done: true };\n    }\n    return await new Promise<IteratorResult<T>>((resolve) => {\n      this.waiters.push(resolve);\n    });\n  }\n}\n\n/**\n * TamboStream handles streaming AI responses with tool execution.\n *\n * Supports two consumption modes:\n * - Async iteration: `for await (const { event, snapshot } of stream) { ... }`\n * - Promise-based: `const thread = await stream.thread`\n *\n * The processing loop runs automatically. The async iterator can only be\n * iterated once (like ReadableStream).\n */\nexport class TamboStream implements AsyncIterable<StreamEvent> {\n  /** Promise that resolves to the final thread state when the stream completes. */\n  readonly thread: Promise<TamboThread>;\n\n  private eventQueue = new EventQueue<StreamEvent>();\n  private resolveThread!: (thread: TamboThread) => void;\n  private rejectThread!: (error: Error) => void;\n  private streamError: Error | null = null;\n  private abortController: AbortController;\n  private consumed = false;\n\n  /**\n   * Create a new TamboStream.\n   * @param options - Configuration for the stream.\n   */\n  constructor(private readonly options: TamboStreamOptions) {\n    this.abortController = new AbortController();\n\n    // Wire external signal to our internal abort controller\n    if (options.signal) {\n      if (options.signal.aborted) {\n        this.abortController.abort(options.signal.reason);\n      } else {\n        options.signal.addEventListener(\"abort\", () => {\n          this.abortController.abort(options.signal?.reason);\n        });\n      }\n    }\n\n    this.thread = new Promise<TamboThread>((resolve, reject) => {\n      this.resolveThread = resolve;\n      this.rejectThread = reject;\n    });\n\n    // Fire-and-forget the processing loop\n    void this.processLoop().catch((err) => {\n      const error = err instanceof Error ? err : new Error(String(err));\n      this.streamError = error;\n      this.rejectThread(error);\n      this.eventQueue.closeWithError(error);\n    });\n  }\n\n  /**\n   * Abort the stream. The `.thread` promise rejects with an AbortError.\n   * The async iterator ends cleanly.\n   */\n  abort(): void {\n    this.abortController.abort(new Error(\"Stream aborted\"));\n  }\n\n  /**\n   * Returns an async iterator over stream events.\n   * Can only be iterated once.\n   * @returns An async iterator yielding StreamEvent pairs.\n   */\n  [Symbol.asyncIterator](): AsyncIterableIterator<StreamEvent> {\n    if (this.consumed) {\n      throw new Error(\"TamboStream can only be iterated once\");\n    }\n    this.consumed = true;\n\n    return {\n      next: async () => await this.eventQueue.pull(),\n      return: async () => {\n        this.eventQueue.close();\n        return { value: undefined as unknown as StreamEvent, done: true };\n      },\n      [Symbol.asyncIterator]() {\n        return this;\n      },\n    };\n  }\n\n  /**\n   * Main processing loop. Runs automatically in the constructor.\n   * Handles event streaming, tool execution, and stream stitching.\n   */\n  private async processLoop(): Promise<void> {\n    const {\n      client,\n      message,\n      threadId: initialThreadId,\n      userMessageText,\n      componentList,\n      toolRegistry,\n      userKey,\n      previousRunId,\n      additionalContext,\n      toolChoice,\n      autoExecuteTools,\n      maxSteps,\n      debug,\n      dispatch,\n      getThreadSnapshot,\n    } = this.options;\n\n    // Generate user message ID if we have user text\n    const userMessageId = userMessageText\n      ? `user_msg_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`\n      : undefined;\n\n    // Dispatch user message immediately for optimistic display on existing threads\n    if (initialThreadId && userMessageText && userMessageId) {\n      dispatchUserMessage(\n        dispatch,\n        initialThreadId,\n        userMessageId,\n        userMessageText,\n      );\n    }\n\n    // Create the initial run stream\n    const { stream: initialStream, initialThreadId: streamThreadId } =\n      await createRunStream({\n        client,\n        threadId: initialThreadId,\n        message,\n        componentList,\n        toolRegistry,\n        userKey,\n        previousRunId,\n        additionalContext,\n        toolChoice,\n      });\n\n    const toolTracker = new ToolCallTracker(toolRegistry);\n    const throttledStreamable = createThrottledStreamableExecutor(\n      toolTracker,\n      toolRegistry,\n    );\n\n    let actualThreadId = streamThreadId;\n    let runId: string | undefined;\n    let currentStream: CreateRunStreamResult[\"stream\"] = initialStream;\n    let stepCount = 0;\n\n    try {\n      while (true) {\n        // Check abort before processing each stream segment\n        if (this.abortController.signal.aborted) {\n          this.handleAbort(actualThreadId, dispatch);\n          return;\n        }\n\n        let pendingAwaitingInput: RunAwaitingInputEvent | undefined;\n\n        for await (const event of handleEventStream(currentStream, { debug })) {\n          // Check abort on each event\n          if (this.abortController.signal.aborted) {\n            this.handleAbort(actualThreadId, dispatch);\n            return;\n          }\n\n          // Extract threadId and runId from RUN_STARTED\n          if (event.type === EventType.RUN_STARTED) {\n            runId = event.runId;\n            actualThreadId ??= event.threadId;\n\n            // For new threads: dispatch user message now that we have the real ID\n            if (!initialThreadId && userMessageText && userMessageId) {\n              dispatchUserMessage(\n                dispatch,\n                actualThreadId,\n                userMessageId,\n                userMessageText,\n              );\n            }\n          } else if (!actualThreadId) {\n            throw new Error(\n              `Expected first event to be RUN_STARTED with threadId, got: ${event.type}`,\n            );\n          }\n\n          toolTracker.handleEvent(event);\n\n          // Parse partial JSON for TOOL_CALL_ARGS\n          const parsedToolArgs =\n            event.type === EventType.TOOL_CALL_ARGS\n              ? toolTracker.parsePartialArgs(event.toolCallId)\n              : undefined;\n\n          dispatch({\n            type: \"EVENT\",\n            event,\n            threadId: actualThreadId,\n            parsedToolArgs,\n            toolSchemas: toolTracker.toolSchemas,\n          });\n\n          // Schedule debounced streamable tool execution\n          if (parsedToolArgs && event.type === EventType.TOOL_CALL_ARGS) {\n            throttledStreamable.schedule(event.toolCallId, parsedToolArgs);\n          }\n\n          // Emit event to async iterator\n          const snapshot = getThreadSnapshot(actualThreadId);\n          if (snapshot) {\n            this.eventQueue.push({ event, snapshot });\n          }\n\n          // Handle awaiting_input for tool execution\n          if (event.type === EventType.CUSTOM) {\n            const customEvent = asTamboCustomEvent(event);\n            if (customEvent?.name === \"tambo.run.awaiting_input\") {\n              pendingAwaitingInput = customEvent;\n              break;\n            }\n          }\n        }\n\n        throttledStreamable.flush();\n\n        // Stream finished without awaiting_input - we're done\n        if (!pendingAwaitingInput) {\n          break;\n        }\n\n        // Don't execute tools if auto-execute is off\n        if (!autoExecuteTools) {\n          break;\n        }\n\n        // Check step limit\n        stepCount++;\n        if (stepCount >= maxSteps) {\n          console.warn(\n            `[TamboStream] maxSteps (${maxSteps}) reached. Stream resolving with pending tool calls.`,\n          );\n          break;\n        }\n\n        // Execute tools and get continuation stream\n        if (!runId || !actualThreadId) {\n          throw new Error(\n            \"Cannot continue run after awaiting_input: missing runId or threadId\",\n          );\n        }\n\n        const { stream: continuationStream, toolResults } =\n          await executeToolsAndContinue({\n            event: pendingAwaitingInput,\n            toolTracker,\n            toolRegistry,\n            componentList,\n            client,\n            threadId: actualThreadId,\n            runId,\n            userKey,\n            additionalContext,\n            toolChoice,\n          });\n\n        dispatchToolResults(dispatch, actualThreadId, toolResults);\n        currentStream = continuationStream;\n      }\n\n      // Stream completed successfully\n      const finalSnapshot = actualThreadId\n        ? getThreadSnapshot(actualThreadId)\n        : undefined;\n\n      if (finalSnapshot) {\n        this.resolveThread(finalSnapshot);\n      } else {\n        this.rejectThread(\n          new Error(\"Stream completed but no thread state found\"),\n        );\n      }\n      this.eventQueue.close();\n    } catch (error) {\n      // Dispatch synthetic RUN_ERROR to clean up thread state\n      if (actualThreadId) {\n        const errorMessage =\n          error instanceof Error ? error.message : \"Unknown streaming error\";\n        const errorEvent: RunErrorEvent = {\n          type: EventType.RUN_ERROR,\n          message: errorMessage,\n        };\n        dispatch({\n          type: \"EVENT\",\n          event: errorEvent,\n          threadId: actualThreadId,\n        });\n      }\n\n      const err = error instanceof Error ? error : new Error(String(error));\n      this.streamError = err;\n      this.rejectThread(err);\n      this.eventQueue.closeWithError(err);\n    }\n  }\n\n  /**\n   * Handle abort by transitioning thread to idle and rejecting the thread promise.\n   * @param threadId - The thread to clean up.\n   * @param dispatch - The dispatch function for state updates.\n   */\n  private handleAbort(\n    threadId: string | undefined,\n    dispatch: (action: StreamAction) => void,\n  ): void {\n    if (threadId) {\n      const errorEvent: RunErrorEvent = {\n        type: EventType.RUN_ERROR,\n        message: \"Stream aborted\",\n      };\n      dispatch({\n        type: \"EVENT\",\n        event: errorEvent,\n        threadId,\n      });\n    }\n\n    const abortError = new Error(\"Stream aborted\");\n    abortError.name = \"AbortError\";\n    this.streamError = abortError;\n    this.rejectThread(abortError);\n    this.eventQueue.close(); // Iterator ends cleanly on abort\n  }\n}\n"]}

package/esm/types/auth.d.ts

@@ -0,0 +1,24 @@
+/**
+ * Authentication state for the SDK.
+ *
+ * Discriminated union tracking the current auth lifecycle:
+ * - `identified`: ready to make API calls (via userKey or successful token exchange)
+ * - `exchanging`: userToken provided, exchange in-flight
+ * - `error`: token exchange failed
+ * - `invalid`: both userKey AND userToken provided (must choose one)
+ * - `unauthenticated`: neither userKey nor userToken provided
+ */
+export type TamboAuthState = {
+    status: "identified";
+    source: "userKey" | "tokenExchange";
+} | {
+    status: "exchanging";
+} | {
+    status: "error";
+    error: Error;
+} | {
+    status: "invalid";
+} | {
+    status: "unauthenticated";
+};
+//# sourceMappingURL=auth.d.ts.map

package/esm/types/auth.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.d.ts","sourceRoot":"","sources":["../../src/types/auth.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AACH,MAAM,MAAM,cAAc,GACtB;IAAE,MAAM,EAAE,YAAY,CAAC;IAAC,MAAM,EAAE,SAAS,GAAG,eAAe,CAAA;CAAE,GAC7D;IAAE,MAAM,EAAE,YAAY,CAAA;CAAE,GACxB;IAAE,MAAM,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,KAAK,CAAA;CAAE,GACjC;IAAE,MAAM,EAAE,SAAS,CAAA;CAAE,GACrB;IAAE,MAAM,EAAE,iBAAiB,CAAA;CAAE,CAAC"}

package/esm/types/auth.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=auth.js.map

package/esm/types/auth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.js","sourceRoot":"","sources":["../../src/types/auth.ts"],"names":[],"mappings":"","sourcesContent":["/**\n * Authentication state for the SDK.\n *\n * Discriminated union tracking the current auth lifecycle:\n * - `identified`: ready to make API calls (via userKey or successful token exchange)\n * - `exchanging`: userToken provided, exchange in-flight\n * - `error`: token exchange failed\n * - `invalid`: both userKey AND userToken provided (must choose one)\n * - `unauthenticated`: neither userKey nor userToken provided\n */\nexport type TamboAuthState =\n  | { status: \"identified\"; source: \"userKey\" | \"tokenExchange\" }\n  | { status: \"exchanging\" }\n  | { status: \"error\"; error: Error }\n  | { status: \"invalid\" }\n  | { status: \"unauthenticated\" };\n"]}

package/esm/types/event.d.ts

@@ -0,0 +1,89 @@
+/**
+ * Tambo-specific Custom Event Types for Streaming API
+ *
+ * Defines custom events specific to Tambo functionality.
+ * For standard AG-UI events, import directly from `@ag-ui/core`.
+ */
+import type { CustomEvent } from "@ag-ui/core";
+import type { Operation } from "fast-json-patch";
+type TamboCustomEventEnvelope<TName extends string, TValue> = Omit<CustomEvent, "name" | "value"> & {
+    name: TName;
+    value: TValue;
+};
+/**
+ * Component start event (custom: tambo.component.start)
+ */
+export type ComponentStartEvent = TamboCustomEventEnvelope<"tambo.component.start", {
+    messageId: string;
+    componentId: string;
+    componentName: string;
+}>;
+/**
+ * Component props delta event (custom: tambo.component.props_delta)
+ * Uses JSON Patch (RFC 6902) to update component props
+ */
+export type ComponentPropsDeltaEvent = TamboCustomEventEnvelope<"tambo.component.props_delta", {
+    componentId: string;
+    operations: Operation[];
+}>;
+/**
+ * Component state delta event (custom: tambo.component.state_delta)
+ * Uses JSON Patch (RFC 6902) to update component state
+ */
+export type ComponentStateDeltaEvent = TamboCustomEventEnvelope<"tambo.component.state_delta", {
+    componentId: string;
+    operations: Operation[];
+}>;
+/**
+ * Component end event (custom: tambo.component.end)
+ */
+export type ComponentEndEvent = TamboCustomEventEnvelope<"tambo.component.end", {
+    componentId: string;
+}>;
+/**
+ * Pending tool call information from awaiting_input event
+ */
+export interface PendingToolCall {
+    toolCallId: string;
+    toolName: string;
+    arguments: string;
+}
+/**
+ * Run awaiting input event (custom: tambo.run.awaiting_input)
+ * Signals that the run is paused waiting for client-side tool execution
+ */
+export type RunAwaitingInputEvent = TamboCustomEventEnvelope<"tambo.run.awaiting_input", {
+    pendingToolCalls: PendingToolCall[];
+}>;
+/**
+ * Message parent event (custom: tambo.message.parent)
+ * Emitted when a message was created during the generation of another message
+ * (e.g., MCP sampling or elicitation).
+ */
+export type MessageParentEvent = TamboCustomEventEnvelope<"tambo.message.parent", {
+    messageId: string;
+    parentMessageId: string;
+}>;
+/**
+ * Union type of Tambo-specific custom events
+ */
+export type TamboCustomEvent = ComponentStartEvent | ComponentPropsDeltaEvent | ComponentStateDeltaEvent | ComponentEndEvent | RunAwaitingInputEvent | MessageParentEvent;
+/**
+ * Type guard to check if an event is a Tambo custom event.
+ * Validates that the event has a name matching known Tambo custom event types.
+ * @param event - Event object to check
+ * @param event.name - Event name to match against known Tambo event types
+ * @returns True if event is a TamboCustomEvent
+ */
+export declare function isTamboCustomEvent(event: {
+    name?: string;
+}): event is TamboCustomEvent;
+/**
+ * Casts a CustomEvent to the specific TamboCustomEvent type based on its name.
+ * Uses exhaustive type checking to ensure all event types are handled.
+ * @param event - The CustomEvent to cast
+ * @returns The properly typed TamboCustomEvent, or undefined if not a known Tambo event
+ */
+export declare function asTamboCustomEvent(event: CustomEvent): TamboCustomEvent | undefined;
+export {};
+//# sourceMappingURL=event.d.ts.map

package/esm/types/event.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"event.d.ts","sourceRoot":"","sources":["../../src/types/event.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAEjD,KAAK,wBAAwB,CAAC,KAAK,SAAS,MAAM,EAAE,MAAM,IAAI,IAAI,CAChE,WAAW,EACX,MAAM,GAAG,OAAO,CACjB,GAAG;IACF,IAAI,EAAE,KAAK,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;CACf,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG,wBAAwB,CACxD,uBAAuB,EACvB;IACE,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;CACvB,CACF,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,wBAAwB,GAAG,wBAAwB,CAC7D,6BAA6B,EAC7B;IACE,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,SAAS,EAAE,CAAC;CACzB,CACF,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,wBAAwB,GAAG,wBAAwB,CAC7D,6BAA6B,EAC7B;IACE,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,SAAS,EAAE,CAAC;CACzB,CACF,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG,wBAAwB,CACtD,qBAAqB,EACrB;IACE,WAAW,EAAE,MAAM,CAAC;CACrB,CACF,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;;GAGG;AACH,MAAM,MAAM,qBAAqB,GAAG,wBAAwB,CAC1D,0BAA0B,EAC1B;IACE,gBAAgB,EAAE,eAAe,EAAE,CAAC;CACrC,CACF,CAAC;AAEF;;;;GAIG;AACH,MAAM,MAAM,kBAAkB,GAAG,wBAAwB,CACvD,sBAAsB,EACtB;IACE,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,EAAE,MAAM,CAAC;CACzB,CACF,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,gBAAgB,GACxB,mBAAmB,GACnB,wBAAwB,GACxB,wBAAwB,GACxB,iBAAiB,GACjB,qBAAqB,GACrB,kBAAkB,CAAC;AAcvB;;;;;;GAMG;AACH,wBAAgB,kBAAkB,CAAC,KAAK,EAAE;IACxC,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,GAAG,KAAK,IAAI,gBAAgB,CAK5B;AAED;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAChC,KAAK,EAAE,WAAW,GACjB,gBAAgB,GAAG,SAAS,CAiB9B"}