@ariaflowagents/core 0.1.0

package/dist/flows/AgentFlowManager.js

@@ -0,0 +1,272 @@
+import { streamText, generateText } from 'ai';
+import { createFlowTransition, isFlowTransition } from './transitions.js';
+export var ContextStrategy;
+(function (ContextStrategy) {
+    ContextStrategy["APPEND"] = "append";
+    ContextStrategy["RESET"] = "reset";
+    ContextStrategy["RESET_WITH_SUMMARY"] = "reset_with_summary";
+})(ContextStrategy || (ContextStrategy = {}));
+export function createCollectorTool(description, schema, nextNode) {
+    return {
+        description,
+        parameters: schema,
+        execute: async (data) => {
+            return createFlowTransition(nextNode, data);
+        },
+    };
+}
+export class AgentFlowManager {
+    nodes = new Map();
+    currentNode = null;
+    context;
+    config;
+    initialized = false;
+    constructor(config, sessionId, userId) {
+        this.config = config;
+        for (const node of config.nodes) {
+            this.nodes.set(node.name, node);
+        }
+        this.context = {
+            state: {},
+            currentNode: config.initialNode,
+            messages: [],
+            collectedData: {},
+            nodeHistory: [],
+            nodeTurnCounts: {},
+            sessionId,
+            userId,
+            ended: false,
+        };
+    }
+    async initialize() {
+        const parts = [];
+        await this.config.hooks?.onFlowStart?.(this.context);
+        const transitionParts = await this.transitionTo(this.config.initialNode);
+        parts.push(...transitionParts);
+        this.initialized = true;
+        if (this.currentNode?.autoRespond !== false) {
+            const responseParts = [];
+            for await (const part of this.generateNodeResponse()) {
+                responseParts.push(part);
+            }
+            parts.push(...responseParts);
+        }
+        return parts;
+    }
+    async *process(userInput) {
+        if (!this.initialized) {
+            throw new Error('Flow not initialized. Call initialize() first.');
+        }
+        if (this.context.ended) {
+            yield { type: 'error', error: 'Flow has ended' };
+            return;
+        }
+        if (!this.currentNode) {
+            yield { type: 'error', error: 'No current node' };
+            return;
+        }
+        try {
+            this.context.messages.push({ role: 'user', content: userInput });
+            const nodeName = this.currentNode.name;
+            this.context.nodeTurnCounts[nodeName] = (this.context.nodeTurnCounts[nodeName] ?? 0) + 1;
+            if (this.currentNode.maxTurns && this.context.nodeTurnCounts[nodeName] > this.currentNode.maxTurns) {
+                yield { type: 'error', error: `Max turns exceeded for node "${nodeName}"` };
+                return;
+            }
+            yield* this.runInference();
+            yield { type: 'done' };
+        }
+        catch (error) {
+            await this.config.hooks?.onError?.(this.context, error);
+            yield { type: 'error', error: error.message };
+        }
+    }
+    async forceTransition(nodeName, data) {
+        return this.transitionTo(nodeName, data);
+    }
+    async endFlow(reason = 'completed') {
+        if (this.context.ended) {
+            return;
+        }
+        this.context.ended = true;
+        this.context.endReason = reason;
+        if (this.currentNode?.postActions) {
+            await this.runActions(this.currentNode.postActions);
+        }
+        await this.config.hooks?.onFlowEnd?.(this.context, reason);
+    }
+    get state() { return this.context.state; }
+    get collectedData() { return this.context.collectedData; }
+    get currentNodeName() { return this.context.currentNode; }
+    get nodeHistory() { return this.context.nodeHistory; }
+    get hasEnded() { return this.context.ended; }
+    get fullContext() { return { ...this.context }; }
+    async transitionTo(nodeName, data) {
+        const parts = [];
+        const node = this.nodes.get(nodeName);
+        if (!node) {
+            throw new Error(`Node "${nodeName}" not found`);
+        }
+        const previousNode = this.currentNode;
+        const previousNodeName = previousNode?.name;
+        if (previousNode) {
+            await this.runActions(previousNode.postActions ?? []);
+            await this.config.hooks?.onNodeExit?.(this.context, previousNode);
+            parts.push({ type: 'node-exit', nodeName: previousNode.name });
+        }
+        this.currentNode = node;
+        this.context.currentNode = nodeName;
+        this.context.nodeHistory.push(nodeName);
+        if (data) {
+            Object.assign(this.context.collectedData, data);
+        }
+        await this.applyContextStrategy(node);
+        await this.config.hooks?.onNodeEnter?.(this.context, node);
+        parts.push({ type: 'node-enter', nodeName: node.name });
+        await this.runActions(node.preActions ?? []);
+        if (previousNodeName) {
+            await this.config.hooks?.onTransition?.(this.context, previousNodeName, nodeName, data);
+            parts.push({ type: 'transition', from: previousNodeName, to: nodeName, data });
+        }
+        return parts;
+    }
+    async *generateNodeResponse() {
+        if (!this.currentNode)
+            return;
+        yield* this.runInference();
+    }
+    async *runInference() {
+        if (!this.currentNode)
+            return;
+        const node = this.currentNode;
+        const systemPrompt = this.buildSystemPrompt(node);
+        const tools = {
+            ...this.config.globalTools,
+            ...node.tools,
+        };
+        await this.config.hooks?.onBeforeInference?.(this.context, node);
+        const streamConfig = {
+            model: this.config.model,
+            system: systemPrompt,
+            tools,
+        };
+        if (this.context.messages.length === 0) {
+            streamConfig.prompt = 'Generate the greeting for this conversation.';
+        }
+        else {
+            streamConfig.messages = this.context.messages;
+        }
+        const result = streamText(streamConfig);
+        let responseText = '';
+        let pendingTransition = null;
+        for await (const chunk of result.fullStream) {
+            if (chunk.type === 'text-delta') {
+                responseText += chunk.text;
+                yield { type: 'text-delta', text: chunk.text };
+            }
+            if (chunk.type === 'tool-call') {
+                const args = 'args' in chunk ? chunk.args : chunk.input;
+                yield { type: 'tool-call', toolName: chunk.toolName, args };
+            }
+            if (chunk.type === 'tool-result') {
+                const toolResult = 'result' in chunk ? chunk.result : chunk.output;
+                if (isFlowTransition(toolResult)) {
+                    pendingTransition = toolResult;
+                }
+                yield { type: 'tool-result', toolName: chunk.toolName, result: toolResult };
+            }
+        }
+        const response = await result.response;
+        this.context.messages.push(...response.messages);
+        await this.config.hooks?.onAfterInference?.(this.context, node, responseText);
+        if (pendingTransition) {
+            const transitionParts = await this.transitionTo(pendingTransition.targetNode, pendingTransition.data);
+            for (const part of transitionParts) {
+                yield part;
+            }
+            if (this.currentNode && this.currentNode.autoRespond !== false) {
+                yield* this.generateNodeResponse();
+            }
+        }
+        if (node.postActions?.some(action => action.type === 'end')) {
+            await this.endFlow('post-action');
+            yield { type: 'flow-end', reason: 'post-action' };
+        }
+    }
+    buildSystemPrompt(node) {
+        const rolePrompt = node.rolePrompt ?? this.config.defaultRolePrompt ?? '';
+        const dataKeys = Object.keys(this.context.collectedData);
+        const stateContext = dataKeys.length > 0
+            ? `\n\n## Collected Information\n${JSON.stringify(this.context.collectedData, null, 2)}`
+            : '';
+        const historyContext = this.context.nodeHistory.length > 1
+            ? `\n\n## Progress\n${this.context.nodeHistory.slice(0, -1).join(' -> ')} -> **${node.name}**`
+            : '';
+        return `${rolePrompt}\n\n## Current Task\n${node.taskPrompt}${stateContext}${historyContext}\n\n## Instructions\n- Focus ONLY on the current task\n- Use the available tools to progress the conversation\n- Do not attempt tasks outside your current scope`;
+    }
+    async applyContextStrategy(node) {
+        const strategyConfig = node.contextStrategy
+            ?? this.config.defaultContextStrategy
+            ?? { strategy: ContextStrategy.APPEND };
+        switch (strategyConfig.strategy) {
+            case ContextStrategy.RESET:
+                this.context.messages = [];
+                break;
+            case ContextStrategy.RESET_WITH_SUMMARY:
+                if (this.context.messages.length > 0) {
+                    const summary = await this.generateSummary(strategyConfig);
+                    this.context.messages = [
+                        { role: 'system', content: `Previous conversation summary: ${summary}` },
+                    ];
+                }
+                break;
+            case ContextStrategy.APPEND:
+            default:
+                break;
+        }
+    }
+    async generateSummary(config) {
+        const prompt = config.summaryPrompt
+            ?? 'Summarize the key points from this conversation in 2-3 sentences.';
+        const result = await generateText({
+            model: this.config.model,
+            system: prompt,
+            messages: this.context.messages,
+        });
+        return result.text;
+    }
+    async runActions(actions) {
+        for (const action of actions) {
+            switch (action.type) {
+                case 'say':
+                    console.log(`[FLOW SAY] ${action.text}`);
+                    break;
+                case 'end':
+                    await this.endFlow(action.message ?? 'action-triggered');
+                    break;
+                case 'function':
+                    await action.handler(this.context);
+                    break;
+                case 'delay':
+                    await new Promise(resolve => setTimeout(resolve, action.ms));
+                    break;
+                case 'emit':
+                    console.log(`[FLOW EVENT] ${action.event}`, action.data);
+                    break;
+                default:
+                    break;
+            }
+        }
+    }
+    static create(config, sessionId, userId) {
+        return new AgentFlowManager(config, sessionId, userId);
+    }
+    static restore(config, savedContext) {
+        const manager = new AgentFlowManager(config, savedContext.sessionId, savedContext.userId);
+        manager.context = { ...savedContext };
+        manager.currentNode = manager.nodes.get(savedContext.currentNode) ?? null;
+        manager.initialized = true;
+        return manager;
+    }
+}
+//# sourceMappingURL=AgentFlowManager.js.map

package/dist/flows/AgentFlowManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AgentFlowManager.js","sourceRoot":"","sources":["../../src/flows/AgentFlowManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAmC,MAAM,IAAI,CAAC;AAG/E,OAAO,EAAE,oBAAoB,EAAE,gBAAgB,EAA6B,MAAM,kBAAkB,CAAC;AAErG,MAAM,CAAN,IAAY,eAIX;AAJD,WAAY,eAAe;IACzB,oCAAiB,CAAA;IACjB,kCAAe,CAAA;IACf,4DAAyC,CAAA;AAC3C,CAAC,EAJW,eAAe,KAAf,eAAe,QAI1B;AA4ED,MAAM,UAAU,mBAAmB,CACjC,WAAmB,EACnB,MAAsB,EACtB,QAAgB;IAEhB,OAAO;QACL,WAAW;QACX,UAAU,EAAE,MAAM;QAClB,OAAO,EAAE,KAAK,EAAE,IAA6B,EAAE,EAAE;YAC/C,OAAO,oBAAoB,CAAC,QAAQ,EAAE,IAA+B,CAAC,CAAC;QACzE,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,gBAAgB;IACnB,KAAK,GAAG,IAAI,GAAG,EAA0B,CAAC;IAC1C,WAAW,GAA0B,IAAI,CAAC;IAC1C,OAAO,CAAc;IACrB,MAAM,CAAoB;IAC1B,WAAW,GAAG,KAAK,CAAC;IAE5B,YAAY,MAAyB,EAAE,SAAiB,EAAE,MAAe;QACvE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YAChC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAClC,CAAC;QAED,IAAI,CAAC,OAAO,GAAG;YACb,KAAK,EAAE,EAAE;YACT,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,QAAQ,EAAE,EAAE;YACZ,aAAa,EAAE,EAAE;YACjB,WAAW,EAAE,EAAE;YACf,cAAc,EAAE,EAAE;YAClB,SAAS;YACT,MAAM;YACN,KAAK,EAAE,KAAK;SACb,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,KAAK,GAAqB,EAAE,CAAC;QAEnC,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,WAAW,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAErD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACzE,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,CAAC;QAE/B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAExB,IAAI,IAAI,CAAC,WAAW,EAAE,WAAW,KAAK,KAAK,EAAE,CAAC;YAC5C,MAAM,aAAa,GAAqB,EAAE,CAAC;YAC3C,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBACrD,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3B,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAC/B,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,CAAC,OAAO,CAAC,SAAiB;QAC9B,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACpE,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACvB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC;YACjD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC;YAClD,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YAEjE,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;YACvC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;YAEzF,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,IAAI,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;gBACnG,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,gCAAgC,QAAQ,GAAG,EAAE,CAAC;gBAC5E,OAAO;YACT,CAAC;YAED,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YAE3B,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,KAAc,CAAC,CAAC;YACjE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CAAC;QAC3D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,QAAgB,EAAE,IAA8B;QACpE,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,SAAiB,WAAW;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC;QAC1B,IAAI,CAAC,OAAO,CAAC,SAAS,GAAG,MAAM,CAAC;QAEhC,IAAI,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,CAAC;YAClC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;QACtD,CAAC;QAED,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC7D,CAAC;IAED,IAAI,KAAK,KAAK,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;IAC1C,IAAI,aAAa,KAAK,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC;IAC1D,IAAI,eAAe,KAAK,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;IAC1D,IAAI,WAAW,KAAK,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;IACtD,IAAI,QAAQ,KAAK,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;IAC7C,IAAI,WAAW,KAAK,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEzC,KAAK,CAAC,YAAY,CACxB,QAAgB,EAChB,IAA8B;QAE9B,MAAM,KAAK,GAAqB,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAEtC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,SAAS,QAAQ,aAAa,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC;QACtC,MAAM,gBAAgB,GAAG,YAAY,EAAE,IAAI,CAAC;QAE5C,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;YACtD,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YAClE,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;QACjE,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,QAAQ,CAAC;QACpC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAExC,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAEtC,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,WAAW,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAC3D,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAExD,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;QAE7C,IAAI,gBAAgB,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,gBAAgB,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;YACxF,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QACjF,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QACjC,IAAI,CAAC,IAAI,CAAC,WAAW;YAAE,OAAO;QAC9B,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;IAC7B,CAAC;IAEO,KAAK,CAAC,CAAC,YAAY;QACzB,IAAI,CAAC,IAAI,CAAC,WAAW;YAAE,OAAO;QAE9B,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC;QAC9B,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAClD,MAAM,KAAK,GAAY;YACrB,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;YAC1B,GAAG,IAAI,CAAC,KAAK;SACd,CAAC;QAEF,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,iBAAiB,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAEjE,MAAM,YAAY,GAAQ;YACxB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAY;YAC/B,MAAM,EAAE,YAAY;YACpB,KAAK;SACN,CAAC;QAEF,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvC,YAAY,CAAC,MAAM,GAAG,8CAA8C,CAAC;QACvE,CAAC;aAAM,CAAC;YACN,YAAY,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;QAChD,CAAC;QAED,MAAM,MAAM,GAAG,UAAU,CAAC,YAAY,CAAC,CAAC;QAExC,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,IAAI,iBAAiB,GAAgC,IAAI,CAAC;QAE1D,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YAC5C,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAChC,YAAY,IAAI,KAAK,CAAC,IAAI,CAAC;gBAC3B,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;YACjD,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC/B,MAAM,IAAI,GAAG,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAE,KAAa,CAAC,KAAK,CAAC;gBACjE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;YAC9D,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;gBACjC,MAAM,UAAU,GAAG,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAE,KAAa,CAAC,MAAM,CAAC;gBAE5E,IAAI,gBAAgB,CAAC,UAAU,CAAC,EAAE,CAAC;oBACjC,iBAAiB,GAAG,UAAU,CAAC;gBACjC,CAAC;gBAED,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;YAC9E,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;QACvC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAEjD,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,gBAAgB,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,YAAY,CAAC,CAAC;QAE9E,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAC7C,iBAAiB,CAAC,UAAU,EAC5B,iBAAiB,CAAC,IAAI,CACvB,CAAC;YAEF,KAAK,MAAM,IAAI,IAAI,eAAe,EAAE,CAAC;gBACnC,MAAM,IAAI,CAAC;YACb,CAAC;YAED,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC,WAAW,KAAK,KAAK,EAAE,CAAC;gBAC/D,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YACrC,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YAC5D,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAClC,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;QACpD,CAAC;IACH,CAAC;IAEO,iBAAiB,CAAC,IAAoB;QAC5C,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,EAAE,CAAC;QAE1E,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC;YACtC,CAAC,CAAC,iCAAiC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE;YACxF,CAAC,CAAC,EAAE,CAAC;QAEP,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC;YACxD,CAAC,CAAC,oBAAoB,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,CAAC,IAAI,IAAI;YAC9F,CAAC,CAAC,EAAE,CAAC;QAEP,OAAO,GAAG,UAAU,wBAAwB,IAAI,CAAC,UAAU,GAAG,YAAY,GAAG,cAAc,kKAAkK,CAAC;IAChQ,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAAC,IAAoB;QACrD,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe;eACtC,IAAI,CAAC,MAAM,CAAC,sBAAsB;eAClC,EAAE,QAAQ,EAAE,eAAe,CAAC,MAAM,EAAE,CAAC;QAE1C,QAAQ,cAAc,CAAC,QAAQ,EAAE,CAAC;YAChC,KAAK,eAAe,CAAC,KAAK;gBACxB,IAAI,CAAC,OAAO,CAAC,QAAQ,GAAG,EAAE,CAAC;gBAC3B,MAAM;YAER,KAAK,eAAe,CAAC,kBAAkB;gBACrC,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACrC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;oBAC3D,IAAI,CAAC,OAAO,CAAC,QAAQ,GAAG;wBACtB,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,kCAAkC,OAAO,EAAE,EAAkB;qBACzF,CAAC;gBACJ,CAAC;gBACD,MAAM;YAER,KAAK,eAAe,CAAC,MAAM,CAAC;YAC5B;gBACE,MAAM;QACV,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,MAA6B;QACzD,MAAM,MAAM,GAAG,MAAM,CAAC,aAAa;eAC9B,mEAAmE,CAAC;QAEzE,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;YAChC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAY;YAC/B,MAAM,EAAE,MAAM;YACd,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ;SAChC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,OAAqB;QAC5C,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;gBACpB,KAAK,KAAK;oBACR,OAAO,CAAC,GAAG,CAAC,cAAc,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;oBACzC,MAAM;gBACR,KAAK,KAAK;oBACR,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,IAAI,kBAAkB,CAAC,CAAC;oBACzD,MAAM;gBACR,KAAK,UAAU;oBACb,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACnC,MAAM;gBACR,KAAK,OAAO;oBACV,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;oBAC7D,MAAM;gBACR,KAAK,MAAM;oBACT,OAAO,CAAC,GAAG,CAAC,gBAAgB,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;oBACzD,MAAM;gBACR;oBACE,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,MAAyB,EAAE,SAAiB,EAAE,MAAe;QACzE,OAAO,IAAI,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IACzD,CAAC;IAED,MAAM,CAAC,OAAO,CAAC,MAAyB,EAAE,YAAyB;QACjE,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,MAAM,EAAE,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,MAAM,CAAC,CAAC;QAC1F,OAAO,CAAC,OAAO,GAAG,EAAE,GAAG,YAAY,EAAE,CAAC;QACtC,OAAO,CAAC,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC;QAC1E,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;QAC3B,OAAO,OAAO,CAAC;IACjB,CAAC;CACF"}

package/dist/flows/FlowGraph.d.ts

@@ -0,0 +1,23 @@
+import type { ContextStrategy, FlowConfig, FlowNodeConfig, FlowTransitionConfig } from '../types/index.js';
+import { FlowNode } from './FlowNode.js';
+export declare class FlowGraph {
+    private nodes;
+    private transitions;
+    private defaultRolePrompt?;
+    private contextStrategy?;
+    addNode(node: FlowNode): this;
+    addTransition(from: string, to: string, config?: Omit<FlowTransitionConfig, 'from' | 'to'>): this;
+    setDefaultRolePrompt(prompt: string): this;
+    setContextStrategy(strategy: ContextStrategy): this;
+    toConfig(): FlowConfig;
+    static create(): FlowGraphBuilder;
+}
+export declare class FlowGraphBuilder {
+    private graph;
+    node(id: string, config: Omit<FlowNodeConfig, 'id'>): this;
+    transition(from: string, to: string, config?: Omit<FlowTransitionConfig, 'from' | 'to'>): this;
+    defaultRolePrompt(prompt: string): this;
+    contextStrategy(strategy: ContextStrategy): this;
+    build(): FlowConfig;
+}
+//# sourceMappingURL=FlowGraph.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"FlowGraph.d.ts","sourceRoot":"","sources":["../../src/flows/FlowGraph.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAC3G,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAEzC,qBAAa,SAAS;IACpB,OAAO,CAAC,KAAK,CAA+B;IAC5C,OAAO,CAAC,WAAW,CAA8B;IACjD,OAAO,CAAC,iBAAiB,CAAC,CAAS;IACnC,OAAO,CAAC,eAAe,CAAC,CAAkB;IAE1C,OAAO,CAAC,IAAI,EAAE,QAAQ,GAAG,IAAI;IAK7B,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,GAAE,IAAI,CAAC,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAM,GAAG,IAAI;IAKrG,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAK1C,kBAAkB,CAAC,QAAQ,EAAE,eAAe,GAAG,IAAI;IAKnD,QAAQ,IAAI,UAAU;IAStB,MAAM,CAAC,MAAM,IAAI,gBAAgB;CAGlC;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,KAAK,CAAmB;IAEhC,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,GAAG,IAAI;IAK1D,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,GAAE,IAAI,CAAC,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAM,GAAG,IAAI;IAKlG,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAKvC,eAAe,CAAC,QAAQ,EAAE,eAAe,GAAG,IAAI;IAKhD,KAAK,IAAI,UAAU;CAGpB"}

package/dist/flows/FlowGraph.js

@@ -0,0 +1,57 @@
+import { FlowNode } from './FlowNode.js';
+export class FlowGraph {
+    nodes = new Map();
+    transitions = [];
+    defaultRolePrompt;
+    contextStrategy;
+    addNode(node) {
+        this.nodes.set(node.id, node);
+        return this;
+    }
+    addTransition(from, to, config = {}) {
+        this.transitions.push({ from, to, ...config });
+        return this;
+    }
+    setDefaultRolePrompt(prompt) {
+        this.defaultRolePrompt = prompt;
+        return this;
+    }
+    setContextStrategy(strategy) {
+        this.contextStrategy = strategy;
+        return this;
+    }
+    toConfig() {
+        return {
+            nodes: Array.from(this.nodes.values()).map(node => node.toConfig()),
+            transitions: this.transitions,
+            defaultRolePrompt: this.defaultRolePrompt,
+            contextStrategy: this.contextStrategy,
+        };
+    }
+    static create() {
+        return new FlowGraphBuilder();
+    }
+}
+export class FlowGraphBuilder {
+    graph = new FlowGraph();
+    node(id, config) {
+        this.graph.addNode(new FlowNode({ id, ...config }));
+        return this;
+    }
+    transition(from, to, config = {}) {
+        this.graph.addTransition(from, to, config);
+        return this;
+    }
+    defaultRolePrompt(prompt) {
+        this.graph.setDefaultRolePrompt(prompt);
+        return this;
+    }
+    contextStrategy(strategy) {
+        this.graph.setContextStrategy(strategy);
+        return this;
+    }
+    build() {
+        return this.graph.toConfig();
+    }
+}
+//# sourceMappingURL=FlowGraph.js.map

package/dist/flows/FlowGraph.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FlowGraph.js","sourceRoot":"","sources":["../../src/flows/FlowGraph.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAEzC,MAAM,OAAO,SAAS;IACZ,KAAK,GAAG,IAAI,GAAG,EAAoB,CAAC;IACpC,WAAW,GAA2B,EAAE,CAAC;IACzC,iBAAiB,CAAU;IAC3B,eAAe,CAAmB;IAE1C,OAAO,CAAC,IAAc;QACpB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAC9B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,aAAa,CAAC,IAAY,EAAE,EAAU,EAAE,SAAoD,EAAE;QAC5F,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,oBAAoB,CAAC,MAAc;QACjC,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC;QAChC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,kBAAkB,CAAC,QAAyB;QAC1C,IAAI,CAAC,eAAe,GAAG,QAAQ,CAAC;QAChC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,QAAQ;QACN,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnE,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;YACzC,eAAe,EAAE,IAAI,CAAC,eAAe;SACtC,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,MAAM;QACX,OAAO,IAAI,gBAAgB,EAAE,CAAC;IAChC,CAAC;CACF;AAED,MAAM,OAAO,gBAAgB;IACnB,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;IAEhC,IAAI,CAAC,EAAU,EAAE,MAAkC;QACjD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,QAAQ,CAAC,EAAE,EAAE,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC;QACpD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,UAAU,CAAC,IAAY,EAAE,EAAU,EAAE,SAAoD,EAAE;QACzF,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,iBAAiB,CAAC,MAAc;QAC9B,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,eAAe,CAAC,QAAyB;QACvC,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;IAC/B,CAAC;CACF"}

package/dist/flows/FlowManager.d.ts

@@ -0,0 +1,36 @@
+import { type LanguageModel, type ModelMessage } from 'ai';
+import type { AgentStreamPart, FlowConfig } from '../types/index.js';
+export interface FlowManagerConfig {
+    flow: FlowConfig;
+    initialNode: string;
+    model: LanguageModel;
+    defaultRolePrompt?: string;
+    contextMessages?: ModelMessage[];
+    sessionMessages?: ModelMessage[];
+}
+export declare class FlowManager {
+    private config;
+    private nodes;
+    private transitions;
+    private context;
+    private currentNodeConfig;
+    private initialized;
+    private pendingEvents;
+    private flowEnded;
+    private sessionMessages?;
+    constructor(config: FlowManagerConfig);
+    initialize(): AsyncGenerator<AgentStreamPart>;
+    process(userInput: string): AsyncGenerator<AgentStreamPart>;
+    transitionTo(nodeId: string, data?: Record<string, unknown>): Promise<void>;
+    transitionToGenerator(nodeId: string, data?: Record<string, unknown>): AsyncGenerator<AgentStreamPart>;
+    get collectedData(): Record<string, unknown>;
+    get currentNode(): string | undefined;
+    private runInference;
+    private buildSystemPrompt;
+    private applyContextStrategy;
+    private generateSummary;
+    private appendMessage;
+    private runActions;
+    private evaluateTransitions;
+}
+//# sourceMappingURL=FlowManager.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"FlowManager.d.ts","sourceRoot":"","sources":["../../src/flows/FlowManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAA4B,KAAK,aAAa,EAAE,KAAK,YAAY,EAAgB,MAAM,IAAI,CAAC;AACnG,OAAO,KAAK,EACV,eAAe,EAGf,UAAU,EAIX,MAAM,mBAAmB,CAAC;AAG3B,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;CAClC;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;IAsBtC,UAAU,IAAI,cAAc,CAAC,eAAe,CAAC;IAa7C,OAAO,CAAC,SAAS,EAAE,MAAM,GAAG,cAAc,CAAC,eAAe,CAAC;IA6B5D,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAM1E,qBAAqB,CAC1B,MAAM,EAAE,MAAM,EACd,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC7B,cAAc,CAAC,eAAe,CAAC;IA8BlC,IAAI,aAAa,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAE3C;IAED,IAAI,WAAW,IAAI,MAAM,GAAG,SAAS,CAEpC;YAEc,YAAY;IAsE3B,OAAO,CAAC,iBAAiB;YAuBX,oBAAoB;YAqBpB,eAAe;IAY7B,OAAO,CAAC,aAAa;YAON,UAAU;YAsBX,mBAAmB;CA8BlC"}