@inkeep/agents-run-api 0.1.0

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

@@ -0,0 +1,162 @@
+import type { OperationEvent } from './agent-operations.js';
+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;
+    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>;
+}
+/**
+ * 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;
+    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>;
+}
+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

@@ -0,0 +1 @@
+{"version":3,"file":"stream-helpers.d.ts","sourceRoot":"","sources":["../../src/utils/stream-helpers.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAG5D,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;IAEhD,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,SAAS;gBAFT,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;IAWtD,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;IAK9C,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;CAG/D;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;IAW7C,OAAO,CAAC,MAAM;IAV1B,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;gBAER,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;IA4DtD,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;IAU/B;;;OAGG;IACI,OAAO,IAAI,IAAI;IAOtB;;OAEG;IACI,iBAAiB,IAAI,OAAO;IAInC;;OAEG;IACI,cAAc;;;;;;IASf,cAAc,CAAC,SAAS,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;CAO/D;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

@@ -0,0 +1,385 @@
+import { parsePartialJson } from 'ai';
+export class SSEStreamHelper {
+    stream;
+    requestId;
+    timestamp;
+    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(' ');
+        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);
+        }
+    }
+    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') {
+        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) {
+        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;
+    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;
+        // Start - notify GraphSession that text streaming is starting
+        // Import is at the top of the file, we need to call it here
+        if (this.sessionId) {
+            const { graphSessionManager } = await import('./graph-session.js');
+            graphSessionManager.setTextStreaming(this.sessionId, true);
+        }
+        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,
+        });
+        // Notify GraphSession that text streaming has finished
+        if (this.sessionId) {
+            const { graphSessionManager } = await import('./graph-session.js');
+            graphSessionManager.setTextStreaming(this.sessionId, false);
+        }
+    }
+    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;
+        // 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;
+    }
+    /**
+     * 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) {
+        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

@@ -0,0 +1,18 @@
+import type { StreamHelper } from './stream-helpers.js';
+/**
+ * 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

@@ -0,0 +1 @@
+{"version":3,"file":"stream-registry.d.ts","sourceRoot":"","sources":["../../src/utils/stream-registry.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAQxD;;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"}

package/dist/utils/stream-registry.js

@@ -0,0 +1,33 @@
+/**
+ * Global registry for StreamHelper instances
+ * Allows agents to access streamHelper via requestId across A2A boundaries
+ */
+const streamHelperRegistry = new Map();
+/**
+ * Register a StreamHelper for a specific request ID
+ */
+export function registerStreamHelper(requestId, streamHelper) {
+    streamHelperRegistry.set(requestId, streamHelper);
+    // Set sessionId for stream helpers that support it
+    if ('setSessionId' in streamHelper && typeof streamHelper.setSessionId === 'function') {
+        streamHelper.setSessionId(requestId);
+    }
+}
+/**
+ * Get a StreamHelper by request ID
+ */
+export function getStreamHelper(requestId) {
+    return streamHelperRegistry.get(requestId);
+}
+/**
+ * Unregister a StreamHelper for a specific request ID
+ */
+export function unregisterStreamHelper(requestId) {
+    streamHelperRegistry.delete(requestId);
+}
+/**
+ * Get registry size (for debugging)
+ */
+export function getRegistrySize() {
+    return streamHelperRegistry.size;
+}