@ariaflowagents/core 0.6.3 → 0.7.0

package/dist/callbacks/streamCallback.js

@@ -0,0 +1,281 @@
+import { appendFile, mkdir } from 'fs/promises';
+import { dirname, resolve } from 'path';
+const BASE_MESSAGE_EVENT_ALLOW_LIST = ['input', 'done', 'error', 'tripwire'];
+const TOOL_EVENT_ALLOW_LIST = ['tool-call', 'tool-result', 'tool-error'];
+const TRANSITION_EVENT_ALLOW_LIST = ['flow-transition', 'handoff'];
+function resolveEnvVars(value) {
+    return value.replace(/\$(?:ENV\.)?(\{)?([A-Z_][A-Z0-9_]*)\}?/g, (match, _brace, varName) => {
+        const envVar = process.env[varName];
+        return envVar !== undefined ? envVar : match;
+    });
+}
+function shouldSendEvent(eventType, config) {
+    if (config.denyList?.includes(eventType)) {
+        return false;
+    }
+    if (config.allowList && config.allowList.length > 0) {
+        return config.allowList.includes(eventType);
+    }
+    return true;
+}
+function isTerminalEvent(part) {
+    return part.type === 'turn-end' || part.type === 'error' || part.type === 'done';
+}
+function shouldResetAccumulator(part) {
+    return part.type === 'error' || part.type === 'done';
+}
+class AsyncSinkDispatcher {
+    sinks;
+    maxQueueSize;
+    dropPolicy;
+    logDroppedEvents;
+    queue = [];
+    processing = false;
+    dropped = 0;
+    flushWaiters = [];
+    constructor(sinks, maxQueueSize, dropPolicy, logDroppedEvents) {
+        this.sinks = sinks;
+        this.maxQueueSize = maxQueueSize;
+        this.dropPolicy = dropPolicy;
+        this.logDroppedEvents = logDroppedEvents;
+    }
+    enqueue(payload) {
+        if (this.maxQueueSize > 0 && this.queue.length >= this.maxQueueSize) {
+            if (this.dropPolicy === 'drop_oldest') {
+                this.queue.shift();
+                this.queue.push(payload);
+            }
+            else {
+                this.dropped += 1;
+                if (this.logDroppedEvents && this.dropped % 100 === 1) {
+                    console.warn(`[StreamCallback] Dropped ${this.dropped} events due to full queue`);
+                }
+                return;
+            }
+        }
+        else {
+            this.queue.push(payload);
+        }
+        this.startProcessing();
+    }
+    startProcessing() {
+        if (this.processing)
+            return;
+        this.processing = true;
+        void this.processLoop();
+    }
+    async processLoop() {
+        try {
+            while (this.queue.length > 0) {
+                const payload = this.queue.shift();
+                if (!payload)
+                    continue;
+                for (const sink of this.sinks) {
+                    try {
+                        await sink.write(payload);
+                    }
+                    catch (error) {
+                        console.error(`[StreamCallback] Sink ${sink.name ?? 'unnamed'} failed: ${error.message}`);
+                    }
+                }
+            }
+        }
+        finally {
+            this.processing = false;
+            if (this.queue.length > 0) {
+                this.startProcessing();
+            }
+            else {
+                const waiters = this.flushWaiters.splice(0, this.flushWaiters.length);
+                for (const resolve of waiters) {
+                    resolve();
+                }
+            }
+        }
+    }
+    async flush(timeoutMs = 2000) {
+        if (!this.processing && this.queue.length === 0)
+            return;
+        await Promise.race([
+            new Promise(resolve => {
+                this.flushWaiters.push(resolve);
+            }),
+            new Promise(resolve => {
+                setTimeout(resolve, Math.max(0, timeoutMs));
+            }),
+        ]);
+    }
+    async close(timeoutMs = 2000) {
+        await this.flush(timeoutMs);
+        for (const sink of this.sinks) {
+            try {
+                await sink.close?.();
+            }
+            catch (error) {
+                console.error(`[StreamCallback] Sink ${sink.name ?? 'unnamed'} close failed: ${error.message}`);
+            }
+        }
+    }
+}
+export function createConsoleStreamSink(options = {}) {
+    const logger = options.logger ?? console;
+    const prefix = options.prefix ?? '[StreamCallback]';
+    return {
+        name: 'console',
+        write: async (payload) => {
+            logger.log(`${prefix} ${JSON.stringify(payload)}`);
+        },
+    };
+}
+export function createFileStreamSink(config) {
+    if (!config.path && !config.directory) {
+        throw new Error('createFileStreamSink requires either "path" or "directory"');
+    }
+    let targetPath = config.path ? resolve(config.path) : null;
+    const mkdirp = config.mkdirp ?? true;
+    const format = config.format ?? 'jsonl';
+    let initialized = false;
+    function buildFilename(payload) {
+        if (typeof config.filename === 'function') {
+            return config.filename(payload);
+        }
+        if (typeof config.filename === 'string') {
+            return config.filename;
+        }
+        const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
+        return `${timestamp}-${payload.sessionId}.${format}`;
+    }
+    function resolveTargetPath(payload) {
+        if (targetPath)
+            return targetPath;
+        const directory = resolve(config.directory);
+        targetPath = resolve(directory, buildFilename(payload));
+        return targetPath;
+    }
+    async function ensurePath(pathToEnsure) {
+        if (initialized || !mkdirp)
+            return;
+        await mkdir(dirname(pathToEnsure), { recursive: true });
+        initialized = true;
+    }
+    return {
+        name: config.path ? `file:${resolve(config.path)}` : `file:${resolve(config.directory)}`,
+        write: async (payload) => {
+            const filePath = resolveTargetPath(payload);
+            await ensurePath(filePath);
+            const line = format === 'jsonl'
+                ? `${JSON.stringify(payload)}\n`
+                : `[${payload.timestamp}] ${payload.part.type} ${JSON.stringify(payload.part)}\n`;
+            await appendFile(filePath, line, 'utf8');
+        },
+    };
+}
+export function createFunctionStreamSink(handler, name = 'function') {
+    return {
+        name,
+        write: async (payload) => {
+            await handler(payload);
+        },
+    };
+}
+export function createHttpStreamSink(config) {
+    const resolvedHeaders = config.headers
+        ? Object.fromEntries(Object.entries(config.headers).map(([key, value]) => [key, resolveEnvVars(value)]))
+        : undefined;
+    return {
+        name: `http:${config.url}`,
+        write: async (payload) => {
+            const controller = new AbortController();
+            const timeout = setTimeout(() => controller.abort('HTTP stream sink timeout'), config.timeoutMs ?? 5000);
+            try {
+                const response = await fetch(resolveEnvVars(config.url), {
+                    method: config.method ?? 'POST',
+                    headers: {
+                        'Content-Type': 'application/json',
+                        ...resolvedHeaders,
+                    },
+                    body: JSON.stringify(payload),
+                    signal: controller.signal,
+                });
+                if (!response.ok) {
+                    console.error(`[StreamCallback] HTTP sink returned ${response.status}`);
+                }
+            }
+            catch (error) {
+                console.error(`[StreamCallback] HTTP sink failed: ${error.message}`);
+            }
+            finally {
+                clearTimeout(timeout);
+            }
+        },
+    };
+}
+export function createStreamCallbackAdapter(config = {}) {
+    const sinks = config.sinks ?? [];
+    if (sinks.length === 0) {
+        return {
+            onStreamPart: async () => { },
+            flush: async () => { },
+            close: async () => { },
+        };
+    }
+    const eventMode = config.eventMode ?? 'message';
+    const emitTextDeltas = config.emitTextDeltas ?? false;
+    const emitToolEvents = config.emitToolEvents ?? true;
+    const emitTransitionEvents = config.emitTransitionEvents ?? true;
+    const emitFinalText = config.emitFinalText ?? true;
+    const allowList = config.allowList ?? (eventMode === 'message'
+        ? [
+            ...BASE_MESSAGE_EVENT_ALLOW_LIST,
+            ...(emitToolEvents ? TOOL_EVENT_ALLOW_LIST : []),
+            ...(emitTransitionEvents ? TRANSITION_EVENT_ALLOW_LIST : []),
+        ]
+        : undefined);
+    const effectiveConfig = {
+        ...config,
+        allowList,
+    };
+    const dispatcher = new AsyncSinkDispatcher(sinks, config.maxQueueSize ?? 1000, config.dropPolicy ?? 'drop_oldest', config.logDroppedEvents ?? true);
+    const textAccumulator = new Map();
+    return {
+        onStreamPart: async (context, part) => {
+            const terminal = isTerminalEvent(part);
+            if (part.type === 'text-delta') {
+                const current = textAccumulator.get(context.session.id) ?? '';
+                textAccumulator.set(context.session.id, current + part.text);
+            }
+            const shouldEmitTextDelta = part.type === 'text-delta' ? emitTextDeltas : true;
+            if (!shouldEmitTextDelta || !shouldSendEvent(part.type, effectiveConfig)) {
+                if (shouldResetAccumulator(part)) {
+                    textAccumulator.delete(context.session.id);
+                }
+                return;
+            }
+            const payload = {
+                sessionId: context.session.id,
+                agentId: context.agentId,
+                timestamp: new Date().toISOString(),
+                part,
+            };
+            const shouldAttachFullText = config.includeFullText || (emitFinalText && terminal);
+            if (shouldAttachFullText) {
+                const fullText = textAccumulator.get(context.session.id);
+                if (fullText !== undefined) {
+                    payload.fullText = fullText;
+                }
+            }
+            dispatcher.enqueue(payload);
+            if (shouldResetAccumulator(part)) {
+                textAccumulator.delete(context.session.id);
+            }
+        },
+        flush: async (timeoutMs) => {
+            await dispatcher.flush(timeoutMs ?? config.flushTimeoutMs ?? 2000);
+        },
+        close: async (timeoutMs) => {
+            await dispatcher.close(timeoutMs ?? config.flushTimeoutMs ?? 2000);
+            textAccumulator.clear();
+        },
+    };
+}
+//# sourceMappingURL=streamCallback.js.map

package/dist/callbacks/streamCallback.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamCallback.js","sourceRoot":"","sources":["../../src/callbacks/streamCallback.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAGxC,MAAM,6BAA6B,GAAa,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;AACvF,MAAM,qBAAqB,GAAa,CAAC,WAAW,EAAE,aAAa,EAAE,YAAY,CAAC,CAAC;AACnF,MAAM,2BAA2B,GAAa,CAAC,iBAAiB,EAAE,SAAS,CAAC,CAAC;AAE7E,SAAS,cAAc,CAAC,KAAa;IACnC,OAAO,KAAK,CAAC,OAAO,CAAC,yCAAyC,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE;QACzF,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACpC,OAAO,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,SAAiB,EAAE,MAA4B;IACtE,IAAI,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACzC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpD,OAAO,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,eAAe,CAAC,IAAuB;IAC9C,OAAO,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;AACnF,CAAC;AAED,SAAS,sBAAsB,CAAC,IAAuB;IACrD,OAAO,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;AACvD,CAAC;AAED,MAAM,mBAAmB;IAOJ;IACA;IACA;IACA;IATX,KAAK,GAA4B,EAAE,CAAC;IACpC,UAAU,GAAG,KAAK,CAAC;IACnB,OAAO,GAAG,CAAC,CAAC;IACZ,YAAY,GAAsB,EAAE,CAAC;IAE7C,YACmB,KAA2B,EAC3B,YAAoB,EACpB,UAAyC,EACzC,gBAAyB;QAHzB,UAAK,GAAL,KAAK,CAAsB;QAC3B,iBAAY,GAAZ,YAAY,CAAQ;QACpB,eAAU,GAAV,UAAU,CAA+B;QACzC,qBAAgB,GAAhB,gBAAgB,CAAS;IACzC,CAAC;IAEJ,OAAO,CAAC,OAA8B;QACpC,IAAI,IAAI,CAAC,YAAY,GAAG,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACpE,IAAI,IAAI,CAAC,UAAU,KAAK,aAAa,EAAE,CAAC;gBACtC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;gBACnB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC3B,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,OAAO,IAAI,CAAC,CAAC;gBAClB,IAAI,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,OAAO,GAAG,GAAG,KAAK,CAAC,EAAE,CAAC;oBACtD,OAAO,CAAC,IAAI,CAAC,4BAA4B,IAAI,CAAC,OAAO,2BAA2B,CAAC,CAAC;gBACpF,CAAC;gBACD,OAAO;YACT,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;QAED,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC;IAEO,eAAe;QACrB,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO;QAC5B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,WAAW;QACvB,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;gBACnC,IAAI,CAAC,OAAO;oBAAE,SAAS;gBAEvB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBAC9B,IAAI,CAAC;wBACH,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBAC5B,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,OAAO,CAAC,KAAK,CACX,yBAAyB,IAAI,CAAC,IAAI,IAAI,SAAS,YAAa,KAAe,CAAC,OAAO,EAAE,CACtF,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;YACxB,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,CAAC;iBAAM,CAAC;gBACN,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;gBACtE,KAAK,MAAM,OAAO,IAAI,OAAO,EAAE,CAAC;oBAC9B,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,YAAoB,IAAI;QAClC,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAExD,MAAM,OAAO,CAAC,IAAI,CAAC;YACjB,IAAI,OAAO,CAAO,OAAO,CAAC,EAAE;gBAC1B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAClC,CAAC,CAAC;YACF,IAAI,OAAO,CAAO,OAAO,CAAC,EAAE;gBAC1B,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC;YAC9C,CAAC,CAAC;SACH,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,YAAoB,IAAI;QAClC,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAE5B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC;YACvB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CACX,yBAAyB,IAAI,CAAC,IAAI,IAAI,SAAS,kBAAmB,KAAe,CAAC,OAAO,EAAE,CAC5F,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAQD,MAAM,UAAU,uBAAuB,CAAC,UAGpC,EAAE;IACJ,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC;IACzC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,kBAAkB,CAAC;IAEpD,OAAO;QACL,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,KAAK,EAAC,OAAO,EAAC,EAAE;YACrB,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACrD,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAMpC;IACC,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;IAChF,CAAC;IAED,IAAI,UAAU,GAAkB,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAC1E,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC;IACrC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC;IACxC,IAAI,WAAW,GAAG,KAAK,CAAC;IAExB,SAAS,aAAa,CAAC,OAA8B;QACnD,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;YAC1C,OAAO,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAClC,CAAC;QACD,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACxC,OAAO,MAAM,CAAC,QAAQ,CAAC;QACzB,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QACjE,OAAO,GAAG,SAAS,IAAI,OAAO,CAAC,SAAS,IAAI,MAAM,EAAE,CAAC;IACvD,CAAC;IAED,SAAS,iBAAiB,CAAC,OAA8B;QACvD,IAAI,UAAU;YAAE,OAAO,UAAU,CAAC;QAClC,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC,SAAU,CAAC,CAAC;QAC7C,UAAU,GAAG,OAAO,CAAC,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;QACxD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,UAAU,UAAU,CAAC,YAAoB;QAC5C,IAAI,WAAW,IAAI,CAAC,MAAM;YAAE,OAAO;QACnC,MAAM,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,WAAW,GAAG,IAAI,CAAC;IACrB,CAAC;IAED,OAAO;QACL,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,OAAO,CAAC,MAAM,CAAC,SAAU,CAAC,EAAE;QACzF,KAAK,EAAE,KAAK,EAAC,OAAO,EAAC,EAAE;YACrB,MAAM,QAAQ,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAC5C,MAAM,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC3B,MAAM,IAAI,GACR,MAAM,KAAK,OAAO;gBAChB,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI;gBAChC,CAAC,CAAC,IAAI,OAAO,CAAC,SAAS,KAAK,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;YACtF,MAAM,UAAU,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QAC3C,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,OAAiE,EACjE,OAAe,UAAU;IAEzB,OAAO;QACL,IAAI;QACJ,KAAK,EAAE,KAAK,EAAC,OAAO,EAAC,EAAE;YACrB,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC;QACzB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAKpC;IACC,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO;QACpC,CAAC,CAAC,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CACnF;QACH,CAAC,CAAC,SAAS,CAAC;IAEd,OAAO;QACL,IAAI,EAAE,QAAQ,MAAM,CAAC,GAAG,EAAE;QAC1B,KAAK,EAAE,KAAK,EAAC,OAAO,EAAC,EAAE;YACrB,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,UAAU,CACxB,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,0BAA0B,CAAC,EAClD,MAAM,CAAC,SAAS,IAAI,IAAI,CACzB,CAAC;YAEF,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE;oBACvD,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,MAAM;oBAC/B,OAAO,EAAE;wBACP,cAAc,EAAE,kBAAkB;wBAClC,GAAG,eAAe;qBACnB;oBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;oBAC7B,MAAM,EAAE,UAAU,CAAC,MAAM;iBAC1B,CAAC,CAAC;gBACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,OAAO,CAAC,KAAK,CAAC,uCAAuC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC1E,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,sCAAuC,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;YAClF,CAAC;oBAAS,CAAC;gBACT,YAAY,CAAC,OAAO,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,SAA+B,EAAE;IAC3E,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;IACjC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO;YACL,YAAY,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;YAC5B,KAAK,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;YACrB,KAAK,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;SACtB,CAAC;IACJ,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,SAAS,CAAC;IAChD,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,KAAK,CAAC;IACtD,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,IAAI,CAAC;IACrD,MAAM,oBAAoB,GAAG,MAAM,CAAC,oBAAoB,IAAI,IAAI,CAAC;IACjE,MAAM,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC;IACnD,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,CACpC,SAAS,KAAK,SAAS;QACrB,CAAC,CAAC;YACA,GAAG,6BAA6B;YAChC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,EAAE,CAAC;YAChD,GAAG,CAAC,oBAAoB,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,EAAE,CAAC;SAC7D;QACD,CAAC,CAAC,SAAS,CACd,CAAC;IACF,MAAM,eAAe,GAAyB;QAC5C,GAAG,MAAM;QACT,SAAS;KACV,CAAC;IAEF,MAAM,UAAU,GAAG,IAAI,mBAAmB,CACxC,KAAK,EACL,MAAM,CAAC,YAAY,IAAI,IAAI,EAC3B,MAAM,CAAC,UAAU,IAAI,aAAa,EAClC,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAChC,CAAC;IAEF,MAAM,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;IAElD,OAAO;QACL,YAAY,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;YACpC,MAAM,QAAQ,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;YAEvC,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC/B,MAAM,OAAO,GAAG,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;gBAC9D,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/D,CAAC;YAED,MAAM,mBAAmB,GAAG,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC;YAC/E,IAAI,CAAC,mBAAmB,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,eAAe,CAAC,EAAE,CAAC;gBACzE,IAAI,sBAAsB,CAAC,IAAI,CAAC,EAAE,CAAC;oBACjC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC7C,CAAC;gBACD,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAA0B;gBACrC,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;gBAC7B,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,IAAI;aACL,CAAC;YAEF,MAAM,oBAAoB,GAAG,MAAM,CAAC,eAAe,IAAI,CAAC,aAAa,IAAI,QAAQ,CAAC,CAAC;YACnF,IAAI,oBAAoB,EAAE,CAAC;gBACzB,MAAM,QAAQ,GAAG,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACzD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;oBAC3B,OAAO,CAAC,QAAQ,GAAG,QAAQ,CAAC;gBAC9B,CAAC;YACH,CAAC;YAED,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE5B,IAAI,sBAAsB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACjC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QAED,KAAK,EAAE,KAAK,EAAE,SAAkB,EAAE,EAAE;YAClC,MAAM,UAAU,CAAC,KAAK,CAAC,SAAS,IAAI,MAAM,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC;QACrE,CAAC;QAED,KAAK,EAAE,KAAK,EAAE,SAAkB,EAAE,EAAE;YAClC,MAAM,UAAU,CAAC,KAAK,CAAC,SAAS,IAAI,MAAM,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC;YACnE,eAAe,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/flows/FlowManager.d.ts

@@ -6,9 +6,9 @@ export interface FlowManagerConfig {
     model: LanguageModel;
     defaultRolePrompt?: string;
     contextMessages?: ModelMessage[];
-    sessionMessages?: ModelMessage[];
     state?: FlowManagerState;
     telemetry?: TelemetrySettings;
+    abortSignal?: AbortSignal;
     /** Default maxSteps for streamText tool loop. */
     maxSteps?: number;
     /**
@@ -19,12 +19,22 @@ export interface FlowManagerConfig {
     toolCallGuard?: (call: {
         toolName: string;
         args: unknown;
+        toolCallId?: string;
     }) => Promise<{
         allowed: true;
     } | {
         allowed: false;
         reason: string;
     }>;
+    /**
+     * Optional hook to enrich tool execute options (for example to inject idempotency metadata).
+     * Returned fields are shallow-merged into `options.experimental_context`.
+     */
+    toolExecutionOptionsFactory?: (call: {
+        toolName: string;
+        args: unknown;
+        toolCallId?: string;
+    }) => Record<string, unknown> | undefined;
 }
 export declare class FlowManager {
     private config;
@@ -34,13 +44,12 @@ export declare class FlowManager {
     private currentNodeConfig;
     private initialized;
     private pendingEvents;
+    private deferredActions;
     private flowEnded;
     private sessionMessages?;
     constructor(config: FlowManagerConfig);
     initialize(): AsyncGenerator<AgentStreamPart>;
-    process(userInput: string, options?: {
-        appendUserToSession?: boolean;
-    }): AsyncGenerator<AgentStreamPart>;
+    process(userInput: string): AsyncGenerator<AgentStreamPart>;
     transitionTo(nodeId: string, data?: Record<string, unknown>): Promise<void>;
     transitionToNode(node: FlowNodeConfig, data?: Record<string, unknown>): Promise<void>;
     transitionToGenerator(nodeId: string, data?: Record<string, unknown>, dynamicNode?: FlowNodeConfig): AsyncGenerator<AgentStreamPart>;
@@ -52,11 +61,17 @@ export declare class FlowManager {
     private buildSystemPrompt;
     private applyContextStrategy;
     private generateSummary;
+    private normalizeForComparison;
+    private isEquivalentPayload;
+    private resolvePendingSignals;
     private resolveTools;
+    private buildImplicitTransitionTools;
     private wrapTools;
     private appendMessage;
     private normalizeMessage;
     private runActions;
+    private flushDeferredActions;
+    private checkTransitionPolicy;
     private evaluateTransitions;
 }
 //# sourceMappingURL=FlowManager.d.ts.map

package/dist/flows/FlowManager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"FlowManager.d.ts","sourceRoot":"","sources":["../../src/flows/FlowManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAA4B,KAAK,aAAa,EAAE,KAAK,YAAY,EAAgB,KAAK,iBAAiB,EAAE,MAAM,IAAI,CAAC;AAC3H,OAAO,KAAK,EACV,eAAe,EAGf,UAAU,EAEV,gBAAgB,EAChB,cAAc,EAEf,MAAM,mBAAmB,CAAC;AAM3B,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,UAAU,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,aAAa,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,eAAe,CAAC,EAAE,YAAY,EAAE,CAAC;IACjC,eAAe,CAAC,EAAE,YAAY,EAAE,CAAC;IACjC,KAAK,CAAC,EAAE,gBAAgB,CAAC;IACzB,SAAS,CAAC,EAAE,iBAAiB,CAAC;IAC9B,iDAAiD;IACjD,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;;OAIG;IACH,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,OAAO,CAAA;KAAE,KAAK,OAAO,CAAC;QAAE,OAAO,EAAE,IAAI,CAAA;KAAE,GAAG;QAAE,OAAO,EAAE,KAAK,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAChI;AAID,qBAAa,WAAW;IAUV,OAAO,CAAC,MAAM;IAT1B,OAAO,CAAC,KAAK,CAAqC;IAClD,OAAO,CAAC,WAAW,CAA6C;IAChE,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,iBAAiB,CAA+B;IACxD,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,aAAa,CAAsB;IAC3C,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,eAAe,CAAC,CAAiB;gBAErB,MAAM,EAAE,iBAAiB;IA+BtC,UAAU,IAAI,cAAc,CAAC,eAAe,CAAC;IAa7C,OAAO,CACZ,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,mBAAmB,CAAC,EAAE,OAAO,CAAA;KAAO,GAC9C,cAAc,CAAC,eAAe,CAAC;IAgC5B,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAM3E,gBAAgB,CAAC,IAAI,EAAE,cAAc,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAMpF,qBAAqB,CAC1B,MAAM,EAAE,MAAM,EACd,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAE9B,WAAW,CAAC,EAAE,cAAc,GAC3B,cAAc,CAAC,eAAe,CAAC;IAkClC,IAAI,aAAa,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAE3C;IAED,IAAI,WAAW,IAAI,MAAM,GAAG,SAAS,CAEpC;IAED,IAAI,QAAQ,IAAI,OAAO,CAEtB;IAED,QAAQ,IAAI,gBAAgB;YAQb,YAAY;IAsP3B,OAAO,CAAC,iBAAiB;YAwBX,oBAAoB;YAqBpB,eAAe;IAiB7B,OAAO,CAAC,YAAY;IAQpB,OAAO,CAAC,SAAS;IA0BjB,OAAO,CAAC,aAAa;IAWrB,OAAO,CAAC,gBAAgB;YA2BT,UAAU;YAsBX,mBAAmB;CA8BlC"}
+{"version":3,"file":"FlowManager.d.ts","sourceRoot":"","sources":["../../src/flows/FlowManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkC,KAAK,aAAa,EAAE,KAAK,YAAY,EAAgB,KAAK,iBAAiB,EAAE,MAAM,IAAI,CAAC;AAEjI,OAAO,KAAK,EACV,eAAe,EAGf,UAAU,EAEV,gBAAgB,EAChB,cAAc,EAEf,MAAM,mBAAmB,CAAC;AAU3B,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,UAAU,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,aAAa,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,eAAe,CAAC,EAAE,YAAY,EAAE,CAAC;IACjC,KAAK,CAAC,EAAE,gBAAgB,CAAC;IACzB,SAAS,CAAC,EAAE,iBAAiB,CAAC;IAC9B,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,iDAAiD;IACjD,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;;OAIG;IACH,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE;QACrB,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,OAAO,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,KAAK,OAAO,CAAC;QAAE,OAAO,EAAE,IAAI,CAAA;KAAE,GAAG;QAAE,OAAO,EAAE,KAAK,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACtE;;;OAGG;IACH,2BAA2B,CAAC,EAAE,CAAC,IAAI,EAAE;QACnC,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,OAAO,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;CAC3C;AAwBD,qBAAa,WAAW;IAWV,OAAO,CAAC,MAAM;IAV1B,OAAO,CAAC,KAAK,CAAqC;IAClD,OAAO,CAAC,WAAW,CAA6C;IAChE,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,iBAAiB,CAA+B;IACxD,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,aAAa,CAAsB;IAC3C,OAAO,CAAC,eAAe,CAAoB;IAC3C,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,eAAe,CAAC,CAAiB;gBAErB,MAAM,EAAE,iBAAiB;IAgCtC,UAAU,IAAI,cAAc,CAAC,eAAe,CAAC;IAc7C,OAAO,CACZ,SAAS,EAAE,MAAM,GAChB,cAAc,CAAC,eAAe,CAAC;IA6B5B,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAM3E,gBAAgB,CAAC,IAAI,EAAE,cAAc,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAMpF,qBAAqB,CAC1B,MAAM,EAAE,MAAM,EACd,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAE9B,WAAW,CAAC,EAAE,cAAc,GAC3B,cAAc,CAAC,eAAe,CAAC;IAwClC,IAAI,aAAa,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAE3C;IAED,IAAI,WAAW,IAAI,MAAM,GAAG,SAAS,CAEpC;IAED,IAAI,QAAQ,IAAI,OAAO,CAEtB;IAED,QAAQ,IAAI,gBAAgB;YAQb,YAAY;IA6O3B,OAAO,CAAC,iBAAiB;YAiCX,oBAAoB;YAuBpB,eAAe;IAwC7B,OAAO,CAAC,sBAAsB;IAyB9B,OAAO,CAAC,mBAAmB;IAI3B,OAAO,CAAC,qBAAqB;IAoE7B,OAAO,CAAC,YAAY;IAkBpB,OAAO,CAAC,4BAA4B;IAoCpC,OAAO,CAAC,SAAS;IAgCjB,OAAO,CAAC,aAAa;IAQrB,OAAO,CAAC,gBAAgB;YAIR,UAAU;YAgCV,oBAAoB;IA2CpC,OAAO,CAAC,qBAAqB;YAwBf,mBAAmB;CA2ClC"}