@ariaflowagents/core 0.1.0

package/dist/flows/FlowManager.js

@@ -0,0 +1,256 @@
+import { streamText, generateText } from 'ai';
+import { isFlowTransition } from './transitions.js';
+export class FlowManager {
+    config;
+    nodes = new Map();
+    transitions = new Map();
+    context;
+    currentNodeConfig = null;
+    initialized = false;
+    pendingEvents = [];
+    flowEnded = false;
+    sessionMessages;
+    constructor(config) {
+        this.config = config;
+        for (const node of config.flow.nodes) {
+            this.nodes.set(node.id, node);
+        }
+        for (const transition of config.flow.transitions ?? []) {
+            const existing = this.transitions.get(transition.from) ?? [];
+            existing.push(transition);
+            this.transitions.set(transition.from, existing);
+        }
+        this.context = {
+            currentNode: config.initialNode,
+            collectedData: {},
+            nodeHistory: [],
+            nodeTurnCounts: {},
+            messages: config.contextMessages ?? [],
+        };
+        this.sessionMessages = config.sessionMessages;
+    }
+    async *initialize() {
+        if (this.initialized) {
+            return;
+        }
+        yield* this.transitionToGenerator(this.config.initialNode);
+        this.initialized = true;
+        if (this.currentNodeConfig?.autoRespond !== false) {
+            yield* this.runInference();
+        }
+    }
+    async *process(userInput) {
+        if (!this.initialized) {
+            throw new Error('Flow not initialized. Call initialize() first.');
+        }
+        if (this.flowEnded) {
+            yield { type: 'error', error: 'Flow has ended' };
+            return;
+        }
+        if (!this.currentNodeConfig) {
+            yield { type: 'error', error: 'No current node' };
+            return;
+        }
+        this.appendMessage({ role: 'user', content: userInput });
+        const nodeId = this.currentNodeConfig.id;
+        this.context.nodeTurnCounts[nodeId] = (this.context.nodeTurnCounts[nodeId] ?? 0) + 1;
+        if (this.currentNodeConfig.maxTurns && this.context.nodeTurnCounts[nodeId] > this.currentNodeConfig.maxTurns) {
+            yield { type: 'error', error: `Max turns exceeded for node "${nodeId}"` };
+            return;
+        }
+        yield* this.runInference();
+        yield { type: 'turn-end' };
+    }
+    async transitionTo(nodeId, data) {
+        for await (const _part of this.transitionToGenerator(nodeId, data)) {
+            // consume parts
+        }
+    }
+    async *transitionToGenerator(nodeId, data) {
+        const nextNode = this.nodes.get(nodeId);
+        if (!nextNode) {
+            throw new Error(`Node "${nodeId}" not found`);
+        }
+        const previousNode = this.currentNodeConfig;
+        if (previousNode) {
+            yield* this.runActions(previousNode.postActions ?? []);
+            yield { type: 'node-exit', nodeName: previousNode.name ?? previousNode.id };
+        }
+        this.currentNodeConfig = nextNode;
+        this.context.currentNode = nodeId;
+        this.context.nodeHistory.push(nodeId);
+        if (data) {
+            Object.assign(this.context.collectedData, data);
+        }
+        await this.applyContextStrategy(nextNode);
+        yield { type: 'node-enter', nodeName: nextNode.name ?? nextNode.id };
+        yield* this.runActions(nextNode.preActions ?? []);
+        if (previousNode) {
+            yield { type: 'flow-transition', from: previousNode.id, to: nextNode.id };
+        }
+    }
+    get collectedData() {
+        return this.context.collectedData;
+    }
+    get currentNode() {
+        return this.currentNodeConfig?.id;
+    }
+    async *runInference() {
+        if (!this.currentNodeConfig) {
+            return;
+        }
+        const node = this.currentNodeConfig;
+        const systemPrompt = this.buildSystemPrompt(node);
+        const tools = {
+            ...(node.tools ?? {}),
+        };
+        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 pendingTransition = null;
+        for await (const chunk of result.fullStream) {
+            if (chunk.type === 'text-delta') {
+                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;
+                yield { type: 'tool-result', toolName: chunk.toolName, result: toolResult };
+                if (isFlowTransition(toolResult)) {
+                    pendingTransition = {
+                        targetNode: toolResult.targetNode,
+                        data: toolResult.data,
+                    };
+                }
+            }
+        }
+        const response = await result.response;
+        for (const message of response.messages) {
+            this.appendMessage(message);
+        }
+        const nextTransition = pendingTransition ?? (await this.evaluateTransitions());
+        if (nextTransition) {
+            yield* this.transitionToGenerator(nextTransition.targetNode, nextTransition.data);
+            if (this.currentNodeConfig?.autoRespond !== false) {
+                yield* this.runInference();
+            }
+        }
+        if (node.postActions?.some(action => action.type === 'end')) {
+            this.flowEnded = true;
+            yield { type: 'flow-end', reason: 'post-action' };
+        }
+    }
+    buildSystemPrompt(node) {
+        const rolePrompt = this.config.defaultRolePrompt ?? this.config.flow.defaultRolePrompt ?? '';
+        const dataKeys = Object.keys(this.context.collectedData);
+        const dataContext = 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 ?? node.id}**`
+            : '';
+        return `${rolePrompt}
+
+## Current Task
+${node.prompt}${dataContext}${historyContext}
+
+## Instructions
+- Focus only on the current task
+- Use the available tools to progress the conversation
+- Do not attempt tasks outside your current scope`;
+    }
+    async applyContextStrategy(node) {
+        const strategy = node.contextStrategy ?? this.config.flow.contextStrategy ?? 'append';
+        switch (strategy) {
+            case 'reset':
+                this.context.messages = [];
+                break;
+            case 'reset_with_summary':
+                if (this.context.messages.length > 0) {
+                    const summary = await this.generateSummary();
+                    this.context.messages = [
+                        { role: 'system', content: `Previous conversation summary: ${summary}` },
+                    ];
+                }
+                break;
+            case 'append':
+            default:
+                break;
+        }
+    }
+    async generateSummary() {
+        const prompt = '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;
+    }
+    appendMessage(message) {
+        this.context.messages.push(message);
+        if (this.sessionMessages && this.sessionMessages !== this.context.messages) {
+            this.sessionMessages.push(message);
+        }
+    }
+    async *runActions(actions) {
+        for (const action of actions) {
+            switch (action.type) {
+                case 'say':
+                    yield { type: 'text-delta', text: action.text };
+                    break;
+                case 'end':
+                    this.flowEnded = true;
+                    yield { type: 'flow-end', reason: action.reason ?? 'action' };
+                    break;
+                case 'function':
+                    await action.handler(this.context);
+                    break;
+                case 'emit':
+                    this.pendingEvents.push({ name: action.event, data: action.data });
+                    break;
+                default:
+                    break;
+            }
+        }
+    }
+    async evaluateTransitions() {
+        const currentNode = this.currentNodeConfig?.id;
+        if (!currentNode) {
+            return null;
+        }
+        const transitions = this.transitions.get(currentNode) ?? [];
+        if (this.pendingEvents.length > 0) {
+            for (const [index, event] of this.pendingEvents.entries()) {
+                const match = transitions.find(transition => transition.on === event.name);
+                if (match) {
+                    this.pendingEvents.splice(index, 1);
+                    return { targetNode: match.to, data: event.data };
+                }
+            }
+        }
+        for (const transition of transitions) {
+            if (!transition.condition) {
+                continue;
+            }
+            const shouldTransition = await transition.condition(this.context);
+            if (shouldTransition) {
+                return { targetNode: transition.to };
+            }
+        }
+        return null;
+    }
+}
+//# sourceMappingURL=FlowManager.js.map

package/dist/flows/FlowManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FlowManager.js","sourceRoot":"","sources":["../../src/flows/FlowManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAuD,MAAM,IAAI,CAAC;AAUnG,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAapD,MAAM,OAAO,WAAW;IAUF;IATZ,KAAK,GAAG,IAAI,GAAG,EAA0B,CAAC;IAC1C,WAAW,GAAG,IAAI,GAAG,EAAkC,CAAC;IACxD,OAAO,CAAc;IACrB,iBAAiB,GAA0B,IAAI,CAAC;IAChD,WAAW,GAAG,KAAK,CAAC;IACpB,aAAa,GAAmB,EAAE,CAAC;IACnC,SAAS,GAAG,KAAK,CAAC;IAClB,eAAe,CAAkB;IAEzC,YAAoB,MAAyB;QAAzB,WAAM,GAAN,MAAM,CAAmB;QAC3C,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YACrC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAChC,CAAC;QAED,KAAK,MAAM,UAAU,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,EAAE,CAAC;YACvD,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YAC7D,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC1B,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAClD,CAAC;QAED,IAAI,CAAC,OAAO,GAAG;YACb,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,aAAa,EAAE,EAAE;YACjB,WAAW,EAAE,EAAE;YACf,cAAc,EAAE,EAAE;YAClB,QAAQ,EAAE,MAAM,CAAC,eAAe,IAAI,EAAE;SACvC,CAAC;QAEF,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,CAAC,UAAU;QACf,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,KAAK,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAC3D,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAExB,IAAI,IAAI,CAAC,iBAAiB,EAAE,WAAW,KAAK,KAAK,EAAE,CAAC;YAClD,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;QAC7B,CAAC;IACH,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,SAAS,EAAE,CAAC;YACnB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC;YACjD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC;YAClD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QAEzD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACzC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QAErF,IAAI,IAAI,CAAC,iBAAiB,CAAC,QAAQ,IAAI,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC;YAC7G,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,gCAAgC,MAAM,GAAG,EAAE,CAAC;YAC1E,OAAO;QACT,CAAC;QAED,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;QAC3B,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,IAA8B;QAC/D,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC;YACnE,gBAAgB;QAClB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,qBAAqB,CAC1B,MAAc,EACd,IAA8B;QAE9B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,SAAS,MAAM,aAAa,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAC5C,IAAI,YAAY,EAAE,CAAC;YACjB,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;YACvD,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,YAAY,CAAC,IAAI,IAAI,YAAY,CAAC,EAAE,EAAE,CAAC;QAC9E,CAAC;QAED,IAAI,CAAC,iBAAiB,GAAG,QAAQ,CAAC;QAClC,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,MAAM,CAAC;QAClC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEtC,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,QAAQ,CAAC,CAAC;QAE1C,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC;QACrE,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;QAElD,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC;QAC5E,CAAC;IACH,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;IACpC,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC;IACpC,CAAC;IAEO,KAAK,CAAC,CAAC,YAAY;QACzB,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC;QACpC,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAClD,MAAM,KAAK,GAAY;YACrB,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;SACtB,CAAC;QAEF,MAAM,YAAY,GAA4B;YAC5C,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,YAAmB,CAAC,CAAC;QAE/C,IAAI,iBAAiB,GAAkE,IAAI,CAAC;QAE5F,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YAC5C,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAChC,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;gBAC5E,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;gBAE5E,IAAI,gBAAgB,CAAC,UAAU,CAAC,EAAE,CAAC;oBACjC,iBAAiB,GAAG;wBAClB,UAAU,EAAE,UAAU,CAAC,UAAU;wBACjC,IAAI,EAAE,UAAU,CAAC,IAAI;qBACtB,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;QACvC,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACxC,IAAI,CAAC,aAAa,CAAC,OAAuB,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,cAAc,GAAG,iBAAiB,IAAI,CAAC,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAC/E,IAAI,cAAc,EAAE,CAAC;YACnB,KAAK,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,cAAc,CAAC,UAAU,EAAE,cAAc,CAAC,IAAI,CAAC,CAAC;YAElF,IAAI,IAAI,CAAC,iBAAiB,EAAE,WAAW,KAAK,KAAK,EAAE,CAAC;gBAClD,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YAC7B,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YAC5D,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,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,MAAM,CAAC,iBAAiB,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,IAAI,EAAE,CAAC;QAE7F,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC;YACrC,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,IAAI,CAAC,EAAE,IAAI;YACzG,CAAC,CAAC,EAAE,CAAC;QAEP,OAAO,GAAG,UAAU;;;EAGtB,IAAI,CAAC,MAAM,GAAG,WAAW,GAAG,cAAc;;;;;kDAKM,CAAC;IACjD,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAAC,IAAoB;QACrD,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,IAAI,QAAQ,CAAC;QAEtF,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,OAAO;gBACV,IAAI,CAAC,OAAO,CAAC,QAAQ,GAAG,EAAE,CAAC;gBAC3B,MAAM;YACR,KAAK,oBAAoB;gBACvB,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACrC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;oBAC7C,IAAI,CAAC,OAAO,CAAC,QAAQ,GAAG;wBACtB,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,kCAAkC,OAAO,EAAE,EAAkB;qBACzF,CAAC;gBACJ,CAAC;gBACD,MAAM;YACR,KAAK,QAAQ,CAAC;YACd;gBACE,MAAM;QACV,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,MAAM,MAAM,GAAG,mEAAmE,CAAC;QAEnF,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,aAAa,CAAC,OAAqB;QACzC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpC,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,KAAK,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC3E,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,UAAU,CAAC,OAAqB;QAC7C,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;gBACpB,KAAK,KAAK;oBACR,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;oBAChD,MAAM;gBACR,KAAK,KAAK;oBACR,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;oBACtB,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,QAAQ,EAAE,CAAC;oBAC9D,MAAM;gBACR,KAAK,UAAU;oBACb,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACnC,MAAM;gBACR,KAAK,MAAM;oBACT,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;oBACnE,MAAM;gBACR;oBACE,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB;QAC/B,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC;QAC/C,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC;QAE5D,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClC,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,EAAE,CAAC;gBAC1D,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE,KAAK,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC3E,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;oBACpC,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAA2C,EAAE,CAAC;gBAC3F,CAAC;YACH,CAAC;QACH,CAAC;QAED,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;YACD,MAAM,gBAAgB,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAClE,IAAI,gBAAgB,EAAE,CAAC;gBACrB,OAAO,EAAE,UAAU,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC;YACvC,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}

package/dist/flows/FlowNode.d.ts

@@ -0,0 +1,8 @@
+import type { FlowNodeConfig } from '../types/index.js';
+export declare class FlowNode {
+    readonly id: string;
+    readonly config: FlowNodeConfig;
+    constructor(config: FlowNodeConfig);
+    toConfig(): FlowNodeConfig;
+}
+//# sourceMappingURL=FlowNode.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"FlowNode.d.ts","sourceRoot":"","sources":["../../src/flows/FlowNode.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAExD,qBAAa,QAAQ;IACnB,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAC;gBAEpB,MAAM,EAAE,cAAc;IAKlC,QAAQ,IAAI,cAAc;CAG3B"}

package/dist/flows/FlowNode.js

@@ -0,0 +1,12 @@
+export class FlowNode {
+    id;
+    config;
+    constructor(config) {
+        this.id = config.id;
+        this.config = config;
+    }
+    toConfig() {
+        return { ...this.config };
+    }
+}
+//# sourceMappingURL=FlowNode.js.map

package/dist/flows/FlowNode.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FlowNode.js","sourceRoot":"","sources":["../../src/flows/FlowNode.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,QAAQ;IACV,EAAE,CAAS;IACX,MAAM,CAAiB;IAEhC,YAAY,MAAsB;QAChC,IAAI,CAAC,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,QAAQ;QACN,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;CACF"}

package/dist/flows/index.d.ts

@@ -0,0 +1,8 @@
+export { FlowManager } from './FlowManager.js';
+export { FlowGraph, FlowGraphBuilder } from './FlowGraph.js';
+export { FlowNode } from './FlowNode.js';
+export { createFlowTransition, isFlowTransition } from './transitions.js';
+export type { FlowManagerConfig } from './FlowManager.js';
+export { AgentFlowManager, ContextStrategy, createCollectorTool, } from './AgentFlowManager.js';
+export type { FlowManagerConfig as AgentFlowManagerConfig, FlowNodeConfig, FlowContext, FlowHooks, FlowAction, FlowStreamPart, ContextStrategyConfig, } from './AgentFlowManager.js';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/flows/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAC7D,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,oBAAoB,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAC1E,YAAY,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,mBAAmB,GACpB,MAAM,uBAAuB,CAAC;AAC/B,YAAY,EACV,iBAAiB,IAAI,sBAAsB,EAC3C,cAAc,EACd,WAAW,EACX,SAAS,EACT,UAAU,EACV,cAAc,EACd,qBAAqB,GACtB,MAAM,uBAAuB,CAAC"}

package/dist/flows/index.js

@@ -0,0 +1,6 @@
+export { FlowManager } from './FlowManager.js';
+export { FlowGraph, FlowGraphBuilder } from './FlowGraph.js';
+export { FlowNode } from './FlowNode.js';
+export { createFlowTransition, isFlowTransition } from './transitions.js';
+export { AgentFlowManager, ContextStrategy, createCollectorTool, } from './AgentFlowManager.js';
+//# sourceMappingURL=index.js.map

package/dist/flows/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/flows/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAC7D,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,oBAAoB,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAE1E,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,mBAAmB,GACpB,MAAM,uBAAuB,CAAC"}

package/dist/flows/transitions.d.ts

@@ -0,0 +1,9 @@
+export interface FlowTransitionSignal {
+    __flow_transition: true;
+    targetNode: string;
+    data?: Record<string, unknown>;
+    message?: string;
+}
+export declare function createFlowTransition(targetNode: string, data?: Record<string, unknown>, message?: string): FlowTransitionSignal;
+export declare function isFlowTransition(result: unknown): result is FlowTransitionSignal;
+//# sourceMappingURL=transitions.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"transitions.d.ts","sourceRoot":"","sources":["../../src/flows/transitions.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,oBAAoB;IACnC,iBAAiB,EAAE,IAAI,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC/B,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,wBAAgB,oBAAoB,CAClC,UAAU,EAAE,MAAM,EAClB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC9B,OAAO,CAAC,EAAE,MAAM,GACf,oBAAoB,CAOtB;AAED,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,IAAI,oBAAoB,CAOhF"}

package/dist/flows/transitions.js

@@ -0,0 +1,15 @@
+export function createFlowTransition(targetNode, data, message) {
+    return {
+        __flow_transition: true,
+        targetNode,
+        data,
+        message,
+    };
+}
+export function isFlowTransition(result) {
+    return (typeof result === 'object' &&
+        result !== null &&
+        '__flow_transition' in result &&
+        result.__flow_transition === true);
+}
+//# sourceMappingURL=transitions.js.map

package/dist/flows/transitions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transitions.js","sourceRoot":"","sources":["../../src/flows/transitions.ts"],"names":[],"mappings":"AAOA,MAAM,UAAU,oBAAoB,CAClC,UAAkB,EAClB,IAA8B,EAC9B,OAAgB;IAEhB,OAAO;QACL,iBAAiB,EAAE,IAAI;QACvB,UAAU;QACV,IAAI;QACJ,OAAO;KACR,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAe;IAC9C,OAAO,CACL,OAAO,MAAM,KAAK,QAAQ;QAC1B,MAAM,KAAK,IAAI;QACf,mBAAmB,IAAI,MAAM;QAC5B,MAAyC,CAAC,iBAAiB,KAAK,IAAI,CACtE,CAAC;AACJ,CAAC"}

package/dist/guards/StopConditions.d.ts

@@ -0,0 +1,14 @@
+import type { RunContext, StopCondition, StopConditionResult } from '../types/index.js';
+export declare function maxSteps(n: number): StopCondition;
+export declare function tokenBudget(budget: number): StopCondition;
+export declare function timeout(ms: number): StopCondition;
+export declare function consecutiveErrors(n: number): StopCondition;
+export declare function loopDetection(windowSize?: number): StopCondition;
+export declare function sameToolRepetition(maxRepeats?: number): StopCondition;
+export declare function maxHandoffs(n: number): StopCondition;
+export declare function taskComplete(phrases: string[]): StopCondition;
+export declare function anyOf(...conditions: StopCondition[]): StopCondition;
+export declare function allOf(...conditions: StopCondition[]): StopCondition;
+export declare const defaultStopConditions: StopCondition[];
+export declare function checkStopConditions(context: RunContext, conditions: StopCondition[]): StopConditionResult;
+//# sourceMappingURL=StopConditions.d.ts.map

package/dist/guards/StopConditions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"StopConditions.d.ts","sourceRoot":"","sources":["../../src/guards/StopConditions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAExF,wBAAgB,QAAQ,CAAC,CAAC,EAAE,MAAM,GAAG,aAAa,CAQjD;AAED,wBAAgB,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa,CAUzD;AAED,wBAAgB,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,aAAa,CAWjD;AAED,wBAAgB,iBAAiB,CAAC,CAAC,EAAE,MAAM,GAAG,aAAa,CAU1D;AAED,wBAAgB,aAAa,CAAC,UAAU,GAAE,MAAU,GAAG,aAAa,CAuBnE;AAED,wBAAgB,kBAAkB,CAAC,UAAU,GAAE,MAAU,GAAG,aAAa,CAoBxE;AAED,wBAAgB,WAAW,CAAC,CAAC,EAAE,MAAM,GAAG,aAAa,CAUpD;AAED,wBAAgB,YAAY,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,aAAa,CAsB7D;AAED,wBAAgB,KAAK,CAAC,GAAG,UAAU,EAAE,aAAa,EAAE,GAAG,aAAa,CAanE;AAED,wBAAgB,KAAK,CAAC,GAAG,UAAU,EAAE,aAAa,EAAE,GAAG,aAAa,CAiBnE;AAED,eAAO,MAAM,qBAAqB,EAAE,aAAa,EAOhD,CAAC;AAEF,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,UAAU,EACnB,UAAU,EAAE,aAAa,EAAE,GAC1B,mBAAmB,CAQrB"}

package/dist/guards/StopConditions.js

@@ -0,0 +1,161 @@
+export function maxSteps(n) {
+    return {
+        name: `maxSteps(${n})`,
+        check: (context) => ({
+            shouldStop: context.stepCount >= n,
+            reason: context.stepCount >= n ? `Reached maximum steps (${n})` : undefined,
+        }),
+    };
+}
+export function tokenBudget(budget) {
+    return {
+        name: `tokenBudget(${budget})`,
+        check: (context) => ({
+            shouldStop: context.totalTokens >= budget,
+            reason: context.totalTokens >= budget
+                ? `Token budget exceeded (${context.totalTokens}/${budget})`
+                : undefined,
+        }),
+    };
+}
+export function timeout(ms) {
+    return {
+        name: `timeout(${ms}ms)`,
+        check: (context) => {
+            const elapsed = Date.now() - context.startTime;
+            return {
+                shouldStop: elapsed >= ms,
+                reason: elapsed >= ms ? `Timeout after ${ms}ms` : undefined,
+            };
+        },
+    };
+}
+export function consecutiveErrors(n) {
+    return {
+        name: `consecutiveErrors(${n})`,
+        check: (context) => ({
+            shouldStop: context.consecutiveErrors >= n,
+            reason: context.consecutiveErrors >= n
+                ? `Too many consecutive errors (${n})`
+                : undefined,
+        }),
+    };
+}
+export function loopDetection(windowSize = 5) {
+    return {
+        name: `loopDetection(${windowSize})`,
+        check: (context) => {
+            const history = context.toolCallHistory;
+            if (history.length < windowSize * 2) {
+                return { shouldStop: false };
+            }
+            const recent = history.slice(-windowSize);
+            const previous = history.slice(-windowSize * 2, -windowSize);
+            const recentPattern = recent.map(call => `${call.toolName}:${JSON.stringify(call.args)}`).join('|');
+            const previousPattern = previous.map(call => `${call.toolName}:${JSON.stringify(call.args)}`).join('|');
+            const isLoop = recentPattern === previousPattern;
+            return {
+                shouldStop: isLoop,
+                reason: isLoop ? 'Detected repetitive loop pattern' : undefined,
+            };
+        },
+    };
+}
+export function sameToolRepetition(maxRepeats = 3) {
+    return {
+        name: `sameToolRepetition(${maxRepeats})`,
+        check: (context) => {
+            const history = context.toolCallHistory;
+            if (history.length < maxRepeats) {
+                return { shouldStop: false };
+            }
+            const recent = history.slice(-maxRepeats);
+            const allSameTool = recent.every(call => call.toolName === recent[0].toolName);
+            return {
+                shouldStop: allSameTool,
+                reason: allSameTool
+                    ? `Tool "${recent[0].toolName}" called ${maxRepeats} times in a row`
+                    : undefined,
+            };
+        },
+    };
+}
+export function maxHandoffs(n) {
+    return {
+        name: `maxHandoffs(${n})`,
+        check: (context) => ({
+            shouldStop: context.handoffStack.length >= n,
+            reason: context.handoffStack.length >= n
+                ? `Maximum handoffs exceeded (${n})`
+                : undefined,
+        }),
+    };
+}
+export function taskComplete(phrases) {
+    return {
+        name: 'taskComplete',
+        check: (context) => {
+            const lastMessage = context.session.messages[context.session.messages.length - 1];
+            if (!lastMessage || lastMessage.role !== 'assistant') {
+                return { shouldStop: false };
+            }
+            const content = typeof lastMessage.content === 'string'
+                ? lastMessage.content
+                : JSON.stringify(lastMessage.content);
+            const lower = content.toLowerCase();
+            const found = phrases.some(phrase => lower.includes(phrase.toLowerCase()));
+            return {
+                shouldStop: found,
+                reason: found ? 'Task completed successfully' : undefined,
+            };
+        },
+    };
+}
+export function anyOf(...conditions) {
+    return {
+        name: `anyOf(${conditions.map(condition => condition.name).join(', ')})`,
+        check: (context) => {
+            for (const condition of conditions) {
+                const result = condition.check(context);
+                if (result.shouldStop) {
+                    return result;
+                }
+            }
+            return { shouldStop: false };
+        },
+    };
+}
+export function allOf(...conditions) {
+    return {
+        name: `allOf(${conditions.map(condition => condition.name).join(', ')})`,
+        check: (context) => {
+            const results = conditions.map(condition => condition.check(context));
+            const allMet = results.every(result => result.shouldStop);
+            if (allMet) {
+                return {
+                    shouldStop: true,
+                    reason: results.map(result => result.reason).filter(Boolean).join('; '),
+                };
+            }
+            return { shouldStop: false };
+        },
+    };
+}
+export const defaultStopConditions = [
+    maxSteps(20),
+    tokenBudget(100000),
+    timeout(120000),
+    consecutiveErrors(3),
+    loopDetection(5),
+    maxHandoffs(10),
+];
+export function checkStopConditions(context, conditions) {
+    for (const condition of conditions) {
+        const result = condition.check(context);
+        if (result.shouldStop) {
+            return result;
+        }
+    }
+    return { shouldStop: false };
+}
+//# sourceMappingURL=StopConditions.js.map

package/dist/guards/StopConditions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"StopConditions.js","sourceRoot":"","sources":["../../src/guards/StopConditions.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,QAAQ,CAAC,CAAS;IAChC,OAAO;QACL,IAAI,EAAE,YAAY,CAAC,GAAG;QACtB,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACnB,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,CAAC;YAClC,MAAM,EAAE,OAAO,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,CAAC,0BAA0B,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS;SAC5E,CAAC;KACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,MAAc;IACxC,OAAO;QACL,IAAI,EAAE,eAAe,MAAM,GAAG;QAC9B,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACnB,UAAU,EAAE,OAAO,CAAC,WAAW,IAAI,MAAM;YACzC,MAAM,EAAE,OAAO,CAAC,WAAW,IAAI,MAAM;gBACnC,CAAC,CAAC,0BAA0B,OAAO,CAAC,WAAW,IAAI,MAAM,GAAG;gBAC5D,CAAC,CAAC,SAAS;SACd,CAAC;KACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,OAAO,CAAC,EAAU;IAChC,OAAO;QACL,IAAI,EAAE,WAAW,EAAE,KAAK;QACxB,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE;YACjB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC;YAC/C,OAAO;gBACL,UAAU,EAAE,OAAO,IAAI,EAAE;gBACzB,MAAM,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC,CAAC,SAAS;aAC5D,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,CAAS;IACzC,OAAO;QACL,IAAI,EAAE,qBAAqB,CAAC,GAAG;QAC/B,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACnB,UAAU,EAAE,OAAO,CAAC,iBAAiB,IAAI,CAAC;YAC1C,MAAM,EAAE,OAAO,CAAC,iBAAiB,IAAI,CAAC;gBACpC,CAAC,CAAC,gCAAgC,CAAC,GAAG;gBACtC,CAAC,CAAC,SAAS;SACd,CAAC;KACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,aAAqB,CAAC;IAClD,OAAO;QACL,IAAI,EAAE,iBAAiB,UAAU,GAAG;QACpC,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE;YACjB,MAAM,OAAO,GAAG,OAAO,CAAC,eAAe,CAAC;YACxC,IAAI,OAAO,CAAC,MAAM,GAAG,UAAU,GAAG,CAAC,EAAE,CAAC;gBACpC,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;YAC/B,CAAC;YAED,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,UAAU,CAAC,CAAC;YAC1C,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,UAAU,GAAG,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;YAE7D,MAAM,aAAa,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACpG,MAAM,eAAe,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAExG,MAAM,MAAM,GAAG,aAAa,KAAK,eAAe,CAAC;YAEjD,OAAO;gBACL,UAAU,EAAE,MAAM;gBAClB,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,kCAAkC,CAAC,CAAC,CAAC,SAAS;aAChE,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,aAAqB,CAAC;IACvD,OAAO;QACL,IAAI,EAAE,sBAAsB,UAAU,GAAG;QACzC,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE;YACjB,MAAM,OAAO,GAAG,OAAO,CAAC,eAAe,CAAC;YACxC,IAAI,OAAO,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;gBAChC,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;YAC/B,CAAC;YAED,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,UAAU,CAAC,CAAC;YAC1C,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YAE/E,OAAO;gBACL,UAAU,EAAE,WAAW;gBACvB,MAAM,EAAE,WAAW;oBACjB,CAAC,CAAC,SAAS,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,YAAY,UAAU,iBAAiB;oBACpE,CAAC,CAAC,SAAS;aACd,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,CAAS;IACnC,OAAO;QACL,IAAI,EAAE,eAAe,CAAC,GAAG;QACzB,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACnB,UAAU,EAAE,OAAO,CAAC,YAAY,CAAC,MAAM,IAAI,CAAC;YAC5C,MAAM,EAAE,OAAO,CAAC,YAAY,CAAC,MAAM,IAAI,CAAC;gBACtC,CAAC,CAAC,8BAA8B,CAAC,GAAG;gBACpC,CAAC,CAAC,SAAS;SACd,CAAC;KACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,OAAiB;IAC5C,OAAO;QACL,IAAI,EAAE,cAAc;QACpB,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE;YACjB,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAClF,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACrD,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;YAC/B,CAAC;YAED,MAAM,OAAO,GAAG,OAAO,WAAW,CAAC,OAAO,KAAK,QAAQ;gBACrD,CAAC,CAAC,WAAW,CAAC,OAAO;gBACrB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAExC,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;YAE3E,OAAO;gBACL,UAAU,EAAE,KAAK;gBACjB,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,6BAA6B,CAAC,CAAC,CAAC,SAAS;aAC1D,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,KAAK,CAAC,GAAG,UAA2B;IAClD,OAAO;QACL,IAAI,EAAE,SAAS,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG;QACxE,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE;YACjB,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;gBACnC,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBACxC,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;oBACtB,OAAO,MAAM,CAAC;gBAChB,CAAC;YACH,CAAC;YACD,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;QAC/B,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,KAAK,CAAC,GAAG,UAA2B;IAClD,OAAO;QACL,IAAI,EAAE,SAAS,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG;QACxE,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE;YACjB,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;YACtE,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAE1D,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO;oBACL,UAAU,EAAE,IAAI;oBAChB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;iBACxE,CAAC;YACJ,CAAC;YAED,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;QAC/B,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAoB;IACpD,QAAQ,CAAC,EAAE,CAAC;IACZ,WAAW,CAAC,MAAM,CAAC;IACnB,OAAO,CAAC,MAAM,CAAC;IACf,iBAAiB,CAAC,CAAC,CAAC;IACpB,aAAa,CAAC,CAAC,CAAC;IAChB,WAAW,CAAC,EAAE,CAAC;CAChB,CAAC;AAEF,MAAM,UAAU,mBAAmB,CACjC,OAAmB,EACnB,UAA2B;IAE3B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACnC,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACxC,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YACtB,OAAO,MAAM,CAAC;QAChB,CAAC;IACH,CAAC;IACD,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;AAC/B,CAAC"}

package/dist/guards/ToolEnforcer.d.ts

@@ -0,0 +1,11 @@
+import type { EnforcementContext, EnforcementResult, EnforcementRule, ToolCallRecord } from '../types/index.js';
+export declare class ToolEnforcer {
+    private rules;
+    constructor(rules?: EnforcementRule[]);
+    addRule(rule: EnforcementRule): void;
+    removeRule(name: string): boolean;
+    check(call: ToolCallRecord, context: EnforcementContext): Promise<EnforcementResult>;
+    getRules(): EnforcementRule[];
+}
+export declare function createToolEnforcer(rules?: EnforcementRule[]): ToolEnforcer;
+//# sourceMappingURL=ToolEnforcer.d.ts.map

package/dist/guards/ToolEnforcer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolEnforcer.d.ts","sourceRoot":"","sources":["../../src/guards/ToolEnforcer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,kBAAkB,EAClB,iBAAiB,EACjB,eAAe,EACf,cAAc,EACf,MAAM,mBAAmB,CAAC;AAE3B,qBAAa,YAAY;IACvB,OAAO,CAAC,KAAK,CAAyB;gBAE1B,KAAK,GAAE,eAAe,EAAO;IAIzC,OAAO,CAAC,IAAI,EAAE,eAAe,GAAG,IAAI;IAIpC,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAS3B,KAAK,CAAC,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAqB1F,QAAQ,IAAI,eAAe,EAAE;CAG9B;AAED,wBAAgB,kBAAkB,CAAC,KAAK,CAAC,EAAE,eAAe,EAAE,GAAG,YAAY,CAE1E"}

package/dist/guards/ToolEnforcer.js

@@ -0,0 +1,40 @@
+export class ToolEnforcer {
+    rules = [];
+    constructor(rules = []) {
+        this.rules = rules;
+    }
+    addRule(rule) {
+        this.rules.push(rule);
+    }
+    removeRule(name) {
+        const index = this.rules.findIndex(rule => rule.name === name);
+        if (index >= 0) {
+            this.rules.splice(index, 1);
+            return true;
+        }
+        return false;
+    }
+    async check(call, context) {
+        for (const rule of this.rules) {
+            if (rule.appliesTo !== '*' && !rule.appliesTo.includes(call.toolName)) {
+                continue;
+            }
+            const result = await Promise.resolve(rule.check(call, context));
+            if (!result.allowed) {
+                console.log(`[Enforcement] Rule "${rule.name}" blocked ${call.toolName}: ${result.reason}`);
+                return result;
+            }
+            if (result.reminder) {
+                console.log(`[Enforcement] Reminder for ${call.toolName}: ${result.reminder}`);
+            }
+        }
+        return { allowed: true };
+    }
+    getRules() {
+        return [...this.rules];
+    }
+}
+export function createToolEnforcer(rules) {
+    return new ToolEnforcer(rules);
+}
+//# sourceMappingURL=ToolEnforcer.js.map

package/dist/guards/ToolEnforcer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolEnforcer.js","sourceRoot":"","sources":["../../src/guards/ToolEnforcer.ts"],"names":[],"mappings":"AAOA,MAAM,OAAO,YAAY;IACf,KAAK,GAAsB,EAAE,CAAC;IAEtC,YAAY,QAA2B,EAAE;QACvC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,OAAO,CAAC,IAAqB;QAC3B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAED,UAAU,CAAC,IAAY;QACrB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QAC/D,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACf,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAC5B,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAoB,EAAE,OAA2B;QAC3D,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,IAAI,CAAC,SAAS,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACtE,SAAS;YACX,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC;YAEhE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,CAAC,IAAI,aAAa,IAAI,CAAC,QAAQ,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC5F,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CAAC,8BAA8B,IAAI,CAAC,QAAQ,KAAK,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;YACjF,CAAC;QACH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,QAAQ;QACN,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;CACF;AAED,MAAM,UAAU,kBAAkB,CAAC,KAAyB;IAC1D,OAAO,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC;AACjC,CAAC"}

package/dist/guards/index.d.ts

@@ -0,0 +1,4 @@
+export { ToolEnforcer, createToolEnforcer } from './ToolEnforcer.js';
+export { maxSteps, tokenBudget, timeout, consecutiveErrors, loopDetection, sameToolRepetition, maxHandoffs, taskComplete, anyOf, allOf, defaultStopConditions, checkStopConditions, } from './StopConditions.js';
+export { readBeforeEdit, createRateLimitRule, createDependencyRule, contentValidation, createSequentialLimitRule, defaultEnforcementRules, } from './rules.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/guards/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/guards/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AACrE,OAAO,EACL,QAAQ,EACR,WAAW,EACX,OAAO,EACP,iBAAiB,EACjB,aAAa,EACb,kBAAkB,EAClB,WAAW,EACX,YAAY,EACZ,KAAK,EACL,KAAK,EACL,qBAAqB,EACrB,mBAAmB,GACpB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,cAAc,EACd,mBAAmB,EACnB,oBAAoB,EACpB,iBAAiB,EACjB,yBAAyB,EACzB,uBAAuB,GACxB,MAAM,YAAY,CAAC"}

package/dist/guards/index.js

@@ -0,0 +1,4 @@
+export { ToolEnforcer, createToolEnforcer } from './ToolEnforcer.js';
+export { maxSteps, tokenBudget, timeout, consecutiveErrors, loopDetection, sameToolRepetition, maxHandoffs, taskComplete, anyOf, allOf, defaultStopConditions, checkStopConditions, } from './StopConditions.js';
+export { readBeforeEdit, createRateLimitRule, createDependencyRule, contentValidation, createSequentialLimitRule, defaultEnforcementRules, } from './rules.js';
+//# sourceMappingURL=index.js.map

package/dist/guards/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/guards/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AACrE,OAAO,EACL,QAAQ,EACR,WAAW,EACX,OAAO,EACP,iBAAiB,EACjB,aAAa,EACb,kBAAkB,EAClB,WAAW,EACX,YAAY,EACZ,KAAK,EACL,KAAK,EACL,qBAAqB,EACrB,mBAAmB,GACpB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,cAAc,EACd,mBAAmB,EACnB,oBAAoB,EACpB,iBAAiB,EACjB,yBAAyB,EACzB,uBAAuB,GACxB,MAAM,YAAY,CAAC"}