@inkeep/agents-run-api 0.1.3 → 0.1.7

package/dist/utils/stream-helpers.d.ts

@@ -1,174 +0,0 @@
-import type { OperationEvent } from './agent-operations';
-export interface StreamHelper {
-    writeRole(role?: string): Promise<void>;
-    writeContent(content: string): Promise<void>;
-    streamData(data: any): Promise<void>;
-    streamText(text: string, delayMs?: number): Promise<void>;
-    writeError(errorMessage: string): Promise<void>;
-    complete(): Promise<void>;
-    writeData(type: string, data: any): Promise<void>;
-    writeOperation(operation: OperationEvent): Promise<void>;
-}
-export interface HonoSSEStream {
-    writeSSE(message: {
-        data: string;
-        event?: string;
-        id?: string;
-    }): Promise<void>;
-    sleep(ms: number): Promise<unknown>;
-}
-export interface ChatCompletionChunk {
-    id: string;
-    object: string;
-    created: number;
-    model: string;
-    choices: Array<{
-        index: number;
-        delta: {
-            role?: string;
-            content?: string;
-        };
-        finish_reason: string | null;
-    }>;
-}
-export declare class SSEStreamHelper implements StreamHelper {
-    private stream;
-    private requestId;
-    private timestamp;
-    private isTextStreaming;
-    private queuedOperations;
-    constructor(stream: HonoSSEStream, requestId: string, timestamp: number);
-    private get sessionId();
-    /**
-     * Write the initial role message
-     */
-    writeRole(role?: string): Promise<void>;
-    /**
-     * Write content chunk
-     */
-    writeContent(content: string): Promise<void>;
-    /**
-     * Stream text word by word with optional delay
-     */
-    streamText(text: string, delayMs?: number): Promise<void>;
-    streamData(data: any): Promise<void>;
-    /**
-     * Write error message
-     */
-    writeError(errorMessage: string): Promise<void>;
-    /**
-     * Write the final completion message
-     */
-    writeCompletion(finishReason?: string): Promise<void>;
-    /**
-     * Write the final [DONE] message
-     */
-    writeDone(): Promise<void>;
-    /**
-     * Complete the stream with finish reason and done message
-     */
-    complete(finishReason?: string): Promise<void>;
-    writeData(type: string, data: any): Promise<void>;
-    writeOperation(operation: OperationEvent): Promise<void>;
-    /**
-     * Flush all queued operations in order after text streaming completes
-     */
-    private flushQueuedOperations;
-}
-/**
- * Factory function to create SSE stream helper
- */
-export declare function createSSEStreamHelper(stream: HonoSSEStream, requestId: string, timestamp: number): SSEStreamHelper;
-export interface VercelUIWriter {
-    write(chunk: any): void;
-    merge(stream: any): void;
-    onError?: (error: Error) => void;
-}
-export declare class VercelDataStreamHelper implements StreamHelper {
-    private writer;
-    private textId;
-    private jsonBuffer;
-    private sentItems;
-    private completedItems;
-    private sessionId;
-    private static readonly MAX_BUFFER_SIZE;
-    private isCompleted;
-    private isTextStreaming;
-    private queuedOperations;
-    constructor(writer: VercelUIWriter);
-    setSessionId(sessionId: string): void;
-    writeRole(_?: string): Promise<void>;
-    writeContent(content: string): Promise<void>;
-    streamText(text: string, delayMs?: number): Promise<void>;
-    writeData(type: 'operation', data: {
-        type: string;
-        ctx: any;
-    }): Promise<void>;
-    writeError(errorMessage: string): Promise<void>;
-    streamData(data: any): Promise<void>;
-    mergeStream(stream: any): Promise<void>;
-    writeCompletion(_finishReason?: string): Promise<void>;
-    writeDone(): Promise<void>;
-    /**
-     * Complete the stream and clean up all memory
-     * This is the primary cleanup point to prevent memory leaks between requests
-     */
-    complete(): Promise<void>;
-    /**
-     * Clean up all memory allocations
-     * Should be called when the stream helper is no longer needed
-     */
-    cleanup(): void;
-    /**
-     * Check if the stream has been completed and cleaned up
-     */
-    isStreamCompleted(): boolean;
-    /**
-     * Get current memory usage stats (for debugging/monitoring)
-     */
-    getMemoryStats(): {
-        bufferSize: number;
-        sentItemsCount: number;
-        completedItemsCount: number;
-        isCompleted: boolean;
-    };
-    writeOperation(operation: OperationEvent): Promise<void>;
-    /**
-     * Flush all queued operations in order after text streaming completes
-     */
-    private flushQueuedOperations;
-}
-export declare function createVercelStreamHelper(writer: VercelUIWriter): VercelDataStreamHelper;
-/**
- * MCP Stream Helper that captures content instead of streaming
- * Used for MCP tool responses which require a single response message
- */
-export declare class MCPStreamHelper implements StreamHelper {
-    private capturedText;
-    private capturedData;
-    private capturedOperations;
-    private hasError;
-    private errorMessage;
-    private sessionId;
-    setSessionId(sessionId: string): void;
-    writeRole(_role?: string): Promise<void>;
-    writeContent(content: string): Promise<void>;
-    streamText(text: string, _delayMs?: number): Promise<void>;
-    streamData(data: any): Promise<void>;
-    writeData(_type: string, data: any): Promise<void>;
-    writeError(errorMessage: string): Promise<void>;
-    complete(): Promise<void>;
-    writeOperation(operation: OperationEvent): Promise<void>;
-    /**
-     * Get the captured response for MCP tool result
-     */
-    getCapturedResponse(): {
-        text: string;
-        data: any[];
-        operations: OperationEvent[];
-        hasError: boolean;
-        errorMessage: string;
-    };
-}
-export declare function createMCPStreamHelper(): MCPStreamHelper;
-//# sourceMappingURL=stream-helpers.d.ts.map

package/dist/utils/stream-helpers.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"stream-helpers.d.ts","sourceRoot":"","sources":["../../src/utils/stream-helpers.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAGzD,MAAM,WAAW,YAAY;IAC3B,SAAS,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7C,UAAU,CAAC,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACrC,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1D,UAAU,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1B,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAElD,cAAc,CAAC,SAAS,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1D;AAGD,MAAM,WAAW,aAAa;IAC5B,QAAQ,CAAC,OAAO,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAChF,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,mBAAmB;IAClC,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,KAAK,CAAC;QACb,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,EAAE;YACL,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,OAAO,CAAC,EAAE,MAAM,CAAC;SAClB,CAAC;QACF,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;KAC9B,CAAC,CAAC;CACJ;AAED,qBAAa,eAAgB,YAAW,YAAY;IAMhD,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,SAAS;IANnB,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,gBAAgB,CAAwB;gBAGtC,MAAM,EAAE,aAAa,EACrB,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM;IAG3B,OAAO,KAAK,SAAS,GAEpB;IAED;;OAEG;IACG,SAAS,CAAC,IAAI,SAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBlD;;OAEG;IACG,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBlD;;OAEG;IACG,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,SAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAsBtD,UAAU,CAAC,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;IAI1C;;OAEG;IACG,UAAU,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIrD;;OAEG;IACG,eAAe,CAAC,YAAY,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;IAiB3D;;OAEG;IACG,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC;IAMhC;;OAEG;IACG,QAAQ,CAAC,YAAY,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ9C,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBjD,cAAc,CAAC,SAAS,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAoB9D;;OAEG;YACW,qBAAqB;CAYpC;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CACnC,MAAM,EAAE,aAAa,EACrB,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,GAChB,eAAe,CAEjB;AAGD,MAAM,WAAW,cAAc;IAC7B,KAAK,CAAC,KAAK,EAAE,GAAG,GAAG,IAAI,CAAC;IACxB,KAAK,CAAC,MAAM,EAAE,GAAG,GAAG,IAAI,CAAC;IACzB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CAClC;AAED,qBAAa,sBAAuB,YAAW,YAAY;IAe7C,OAAO,CAAC,MAAM;IAd1B,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,UAAU,CAAM;IACxB,OAAO,CAAC,SAAS,CAA6B;IAC9C,OAAO,CAAC,cAAc,CAAqB;IAC3C,OAAO,CAAC,SAAS,CAAuB;IAGxC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAmB;IAC1D,OAAO,CAAC,WAAW,CAAS;IAG5B,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,gBAAgB,CAAwB;gBAE5B,MAAM,EAAE,cAAc;IAE1C,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAK/B,SAAS,CAAC,CAAC,SAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAIzC,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA2C5C,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,SAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA0DtD,SAAS,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,GAAG,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAY7E,UAAU,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAY/C,UAAU,CAAC,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpC,WAAW,CAAC,MAAM,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;IASvC,eAAe,CAAC,aAAa,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;IAItD,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC;IAIhC;;;OAGG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAa/B;;;OAGG;IACI,OAAO,IAAI,IAAI;IAStB;;OAEG;IACI,iBAAiB,IAAI,OAAO;IAInC;;OAEG;IACI,cAAc;;;;;;IASf,cAAc,CAAC,SAAS,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IA6B9D;;OAEG;YACW,qBAAqB;CAgBpC;AAED,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,cAAc,0BAE9D;AAED;;;GAGG;AACH,qBAAa,eAAgB,YAAW,YAAY;IAClD,OAAO,CAAC,YAAY,CAAM;IAC1B,OAAO,CAAC,YAAY,CAAa;IACjC,OAAO,CAAC,kBAAkB,CAAwB;IAClD,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,YAAY,CAAM;IAC1B,OAAO,CAAC,SAAS,CAAuB;IAExC,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAI/B,SAAS,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIxC,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI5C,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK1D,UAAU,CAAC,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpC,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;IAIlD,UAAU,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK/C,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAIzB,cAAc,CAAC,SAAS,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAI9D;;OAEG;IACH,mBAAmB,IAAI;QACrB,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,GAAG,EAAE,CAAC;QACZ,UAAU,EAAE,cAAc,EAAE,CAAC;QAC7B,QAAQ,EAAE,OAAO,CAAC;QAClB,YAAY,EAAE,MAAM,CAAC;KACtB;CASF;AAED,wBAAgB,qBAAqB,IAAI,eAAe,CAEvD"}

package/dist/utils/stream-helpers.js

@@ -1,466 +0,0 @@
-import { parsePartialJson } from 'ai';
-export class SSEStreamHelper {
-    stream;
-    requestId;
-    timestamp;
-    // Stream queuing for proper event ordering
-    isTextStreaming = false;
-    queuedOperations = [];
-    constructor(stream, requestId, timestamp) {
-        this.stream = stream;
-        this.requestId = requestId;
-        this.timestamp = timestamp;
-    }
-    get sessionId() {
-        return this.requestId;
-    }
-    /**
-     * Write the initial role message
-     */
-    async writeRole(role = 'assistant') {
-        await this.stream.writeSSE({
-            data: JSON.stringify({
-                id: this.requestId,
-                object: 'chat.completion.chunk',
-                created: this.timestamp,
-                choices: [
-                    {
-                        index: 0,
-                        delta: {
-                            role,
-                        },
-                        finish_reason: null,
-                    },
-                ],
-            }),
-        });
-    }
-    /**
-     * Write content chunk
-     */
-    async writeContent(content) {
-        await this.stream.writeSSE({
-            data: JSON.stringify({
-                id: this.requestId,
-                object: 'chat.completion.chunk',
-                created: this.timestamp,
-                choices: [
-                    {
-                        index: 0,
-                        delta: {
-                            content,
-                        },
-                        finish_reason: null,
-                    },
-                ],
-            }),
-        });
-    }
-    /**
-     * Stream text word by word with optional delay
-     */
-    async streamText(text, delayMs = 100) {
-        const words = text.split(' ');
-        // Mark that text streaming is starting
-        this.isTextStreaming = true;
-        try {
-            for (let i = 0; i < words.length; i++) {
-                await this.stream.sleep(delayMs);
-                const content = i === 0 ? words[i] : ` ${words[i]}`;
-                await this.writeContent(content);
-            }
-        }
-        finally {
-            // Mark that text streaming has finished
-            this.isTextStreaming = false;
-            // Flush any queued operations now that text sequence is complete
-            await this.flushQueuedOperations();
-        }
-    }
-    async streamData(data) {
-        await this.writeContent(JSON.stringify(data));
-    }
-    /**
-     * Write error message
-     */
-    async writeError(errorMessage) {
-        await this.writeContent(`\n\n${errorMessage}`);
-    }
-    /**
-     * Write the final completion message
-     */
-    async writeCompletion(finishReason = 'stop') {
-        await this.stream.writeSSE({
-            data: JSON.stringify({
-                id: this.requestId,
-                object: 'chat.completion.chunk',
-                created: this.timestamp,
-                choices: [
-                    {
-                        index: 0,
-                        delta: {},
-                        finish_reason: finishReason,
-                    },
-                ],
-            }),
-        });
-    }
-    /**
-     * Write the final [DONE] message
-     */
-    async writeDone() {
-        await this.stream.writeSSE({
-            data: '[DONE]',
-        });
-    }
-    /**
-     * Complete the stream with finish reason and done message
-     */
-    async complete(finishReason = 'stop') {
-        // Flush any remaining queued operations before completing
-        await this.flushQueuedOperations();
-        await this.writeCompletion(finishReason);
-        await this.writeDone();
-    }
-    async writeData(type, data) {
-        await this.stream.writeSSE({
-            data: JSON.stringify({
-                id: this.requestId,
-                object: 'chat.completion.chunk',
-                created: this.timestamp,
-                choices: [
-                    {
-                        index: 0,
-                        delta: {
-                            content: JSON.stringify({ type, data }),
-                        },
-                        finish_reason: null,
-                    },
-                ],
-            }),
-        });
-    }
-    async writeOperation(operation) {
-        if (operation.type === 'status_update' && operation.ctx.operationType) {
-            operation = {
-                type: operation.ctx.operationType,
-                ctx: operation.ctx.data,
-            };
-        }
-        // Queue operation if text is currently streaming
-        if (this.isTextStreaming) {
-            this.queuedOperations.push(operation);
-            return;
-        }
-        // If not streaming, flush any queued operations first, then send this one
-        await this.flushQueuedOperations();
-        await this.writeData('data-operation', operation);
-    }
-    /**
-     * Flush all queued operations in order after text streaming completes
-     */
-    async flushQueuedOperations() {
-        if (this.queuedOperations.length === 0) {
-            return;
-        }
-        const operationsToFlush = [...this.queuedOperations];
-        this.queuedOperations = []; // Clear the queue
-        for (const operation of operationsToFlush) {
-            await this.writeData('data-operation', operation);
-        }
-    }
-}
-/**
- * Factory function to create SSE stream helper
- */
-export function createSSEStreamHelper(stream, requestId, timestamp) {
-    return new SSEStreamHelper(stream, requestId, timestamp);
-}
-export class VercelDataStreamHelper {
-    writer;
-    textId = null;
-    jsonBuffer = '';
-    sentItems = new Map(); // Track what we've sent for each index
-    completedItems = new Set(); // Track completed items
-    sessionId = null;
-    // Memory management - focused on connection completion cleanup
-    static MAX_BUFFER_SIZE = 5 * 1024 * 1024; // 5MB limit (more generous during request)
-    isCompleted = false;
-    // Stream queuing for proper event ordering
-    isTextStreaming = false;
-    queuedOperations = [];
-    constructor(writer) {
-        this.writer = writer;
-    }
-    setSessionId(sessionId) {
-        this.sessionId = sessionId;
-    }
-    // This mirrors SSEStreamHelper API but outputs using Vercel AI SDK writer
-    async writeRole(_ = 'assistant') {
-        // noop
-    }
-    async writeContent(content) {
-        if (this.isCompleted) {
-            console.warn('Attempted to write content to completed stream');
-            return;
-        }
-        if (!this.textId)
-            this.textId = `${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
-        // Only prevent catastrophic buffer growth during request
-        if (this.jsonBuffer.length + content.length > VercelDataStreamHelper.MAX_BUFFER_SIZE) {
-            // Keep more context since we're not cleaning up during request
-            const keepSize = Math.floor(VercelDataStreamHelper.MAX_BUFFER_SIZE * 0.8);
-            this.jsonBuffer = this.jsonBuffer.slice(-keepSize);
-        }
-        this.jsonBuffer += content;
-        const { value, state } = await parsePartialJson(this.jsonBuffer);
-        if (!['repaired-parse', 'successful-parse'].includes(state))
-            return;
-        if (!Array.isArray(value))
-            return;
-        for (let i = 0; i < value.length; i++) {
-            const { type, ...data } = value[i];
-            // TODO: Check for kind data and JSON.stringify
-            // Create a content hash to check if this item has changed
-            const currentContent = JSON.stringify(data);
-            const lastSentContent = this.sentItems.get(i);
-            // Only send if content has changed or is new
-            if (currentContent !== lastSentContent) {
-                const chunk = {
-                    type: 'data-component',
-                    id: `${this.textId}-${i}`,
-                    data: { type, ...data },
-                };
-                this.writer.write(chunk);
-                this.sentItems.set(i, currentContent);
-            }
-        }
-    }
-    async streamText(text, delayMs = 100) {
-        if (this.isCompleted) {
-            console.warn('Attempted to stream text to completed stream');
-            return;
-        }
-        // For plain text, write directly to the stream as text chunks
-        if (!this.textId)
-            this.textId = `${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
-        const words = text.split(' ');
-        // ------------------------------
-        // New Vercel data-stream v2 format
-        // ------------------------------
-        // Emit "text-start" once at the beginning, followed by "text-delta" chunks
-        // for each word (with preceding space when necessary) and finish with
-        // a single "text-end".
-        const id = this.textId;
-        // Mark that text streaming is starting
-        this.isTextStreaming = true;
-        try {
-            this.writer.write({
-                type: 'text-start',
-                id,
-            });
-            // Deltas (optionally throttled)
-            for (let i = 0; i < words.length; i++) {
-                if (delayMs > 0) {
-                    await new Promise((r) => setTimeout(r, delayMs));
-                }
-                const delta = i === 0 ? words[i] : ` ${words[i]}`;
-                this.writer.write({
-                    type: 'text-delta',
-                    id,
-                    delta,
-                });
-            }
-            // End
-            this.writer.write({
-                type: 'text-end',
-                id,
-            });
-        }
-        finally {
-            // Mark that text streaming has finished
-            this.isTextStreaming = false;
-            // Flush any queued operations now that text sequence is complete
-            await this.flushQueuedOperations();
-        }
-    }
-    async writeData(type, data) {
-        if (this.isCompleted) {
-            console.warn('Attempted to write data to completed stream');
-            return;
-        }
-        this.writer.write({
-            type: `${type}`,
-            data,
-        });
-    }
-    async writeError(errorMessage) {
-        if (this.isCompleted) {
-            console.warn('Attempted to write error to completed stream');
-            return;
-        }
-        this.writer.write({
-            type: 'error',
-            errorText: errorMessage,
-        });
-    }
-    async streamData(data) {
-        await this.writeContent(JSON.stringify(data));
-    }
-    async mergeStream(stream) {
-        if (this.isCompleted) {
-            console.warn('Attempted to merge stream to completed stream');
-            return;
-        }
-        this.writer.merge(stream);
-    }
-    async writeCompletion(_finishReason = 'stop') {
-        // Completion is handled automatically by Vercel's writer
-    }
-    async writeDone() {
-        // Done is handled automatically by Vercel's writer
-    }
-    /**
-     * Complete the stream and clean up all memory
-     * This is the primary cleanup point to prevent memory leaks between requests
-     */
-    async complete() {
-        if (this.isCompleted)
-            return;
-        // Flush any remaining queued operations before completing
-        await this.flushQueuedOperations();
-        // Mark as completed to prevent further writes
-        this.isCompleted = true;
-        // Clean up all buffers and references
-        this.cleanup();
-    }
-    /**
-     * Clean up all memory allocations
-     * Should be called when the stream helper is no longer needed
-     */
-    cleanup() {
-        this.jsonBuffer = '';
-        this.sentItems.clear();
-        this.completedItems.clear();
-        this.textId = null;
-        this.queuedOperations = [];
-        this.isTextStreaming = false;
-    }
-    /**
-     * Check if the stream has been completed and cleaned up
-     */
-    isStreamCompleted() {
-        return this.isCompleted;
-    }
-    /**
-     * Get current memory usage stats (for debugging/monitoring)
-     */
-    getMemoryStats() {
-        return {
-            bufferSize: this.jsonBuffer.length,
-            sentItemsCount: this.sentItems.size,
-            completedItemsCount: this.completedItems.size,
-            isCompleted: this.isCompleted,
-        };
-    }
-    async writeOperation(operation) {
-        if (this.isCompleted) {
-            console.warn('Attempted to write operation to completed stream');
-            return;
-        }
-        if (operation.type === 'status_update' && operation.ctx.operationType) {
-            operation = {
-                type: operation.ctx.operationType,
-                ctx: operation.ctx.data,
-            };
-        }
-        // Queue operation if text is currently streaming
-        if (this.isTextStreaming) {
-            this.queuedOperations.push(operation);
-            return;
-        }
-        // If not streaming, flush any queued operations first, then send this one
-        await this.flushQueuedOperations();
-        this.writer.write({
-            id: 'id' in operation ? operation.id : undefined,
-            type: 'data-operation',
-            data: operation,
-        });
-    }
-    /**
-     * Flush all queued operations in order after text streaming completes
-     */
-    async flushQueuedOperations() {
-        if (this.queuedOperations.length === 0) {
-            return;
-        }
-        const operationsToFlush = [...this.queuedOperations];
-        this.queuedOperations = []; // Clear the queue
-        for (const operation of operationsToFlush) {
-            this.writer.write({
-                id: 'id' in operation ? operation.id : undefined,
-                type: 'data-operation',
-                data: operation,
-            });
-        }
-    }
-}
-export function createVercelStreamHelper(writer) {
-    return new VercelDataStreamHelper(writer);
-}
-/**
- * MCP Stream Helper that captures content instead of streaming
- * Used for MCP tool responses which require a single response message
- */
-export class MCPStreamHelper {
-    capturedText = '';
-    capturedData = [];
-    capturedOperations = [];
-    hasError = false;
-    errorMessage = '';
-    sessionId = null;
-    setSessionId(sessionId) {
-        this.sessionId = sessionId;
-    }
-    async writeRole(_role) {
-        // No-op for MCP
-    }
-    async writeContent(content) {
-        this.capturedText += content;
-    }
-    async streamText(text, _delayMs) {
-        // Capture text without streaming delay
-        this.capturedText += text;
-    }
-    async streamData(data) {
-        this.capturedData.push(data);
-    }
-    async writeData(_type, data) {
-        this.capturedData.push(data);
-    }
-    async writeError(errorMessage) {
-        this.hasError = true;
-        this.errorMessage = errorMessage;
-    }
-    async complete() {
-        // No-op for MCP
-    }
-    async writeOperation(operation) {
-        this.capturedOperations.push(operation);
-    }
-    /**
-     * Get the captured response for MCP tool result
-     */
-    getCapturedResponse() {
-        return {
-            text: this.capturedText,
-            data: this.capturedData,
-            operations: this.capturedOperations,
-            hasError: this.hasError,
-            errorMessage: this.errorMessage,
-        };
-    }
-}
-export function createMCPStreamHelper() {
-    return new MCPStreamHelper();
-}

package/dist/utils/stream-registry.d.ts

@@ -1,18 +0,0 @@
-import type { StreamHelper } from './stream-helpers';
-/**
- * Register a StreamHelper for a specific request ID
- */
-export declare function registerStreamHelper(requestId: string, streamHelper: StreamHelper): void;
-/**
- * Get a StreamHelper by request ID
- */
-export declare function getStreamHelper(requestId: string): StreamHelper | undefined;
-/**
- * Unregister a StreamHelper for a specific request ID
- */
-export declare function unregisterStreamHelper(requestId: string): void;
-/**
- * Get registry size (for debugging)
- */
-export declare function getRegistrySize(): number;
-//# sourceMappingURL=stream-registry.d.ts.map

package/dist/utils/stream-registry.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"stream-registry.d.ts","sourceRoot":"","sources":["../../src/utils/stream-registry.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAQrD;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,YAAY,EAAE,YAAY,GAAG,IAAI,CAOxF;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS,CAE3E;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAE9D;AAED;;GAEG;AACH,wBAAgB,eAAe,IAAI,MAAM,CAExC"}