@just-every/ensemble 0.1.25 → 0.1.26

package/dist/unified_request.js

@@ -0,0 +1,222 @@
+import { createRequestContext, ToolCallAction } from './types/tool_types.js';
+import { getModelProvider } from './model_providers/model_provider.js';
+import { MessageHistory } from './utils/message_history.js';
+import { EnsembleErrorHandler } from './utils/error_handler.js';
+class RequestAgent {
+    agent_id;
+    modelSettings;
+    modelClass;
+    tools;
+    constructor(options) {
+        this.agent_id = options.agentId || 'ensemble';
+        this.modelSettings = options.modelSettings;
+        this.modelClass = options.modelClass;
+        this.tools = options.tools || [];
+    }
+    async getTools() {
+        return this.tools;
+    }
+}
+export async function* unifiedRequest(model, messages, options = {}) {
+    const useEnhanced = options.useEnhancedMode ||
+        options.toolHandler ||
+        options.loop ||
+        options.toolCategories ||
+        options.toolFilter ||
+        options.toolResultTransformer;
+    const history = options.messageHistory || new MessageHistory(messages, {
+        compactToolCalls: true,
+        preserveSystemMessages: true
+    });
+    const context = useEnhanced ? createRequestContext({
+        messages: history.getMessages()
+    }) : undefined;
+    try {
+        let iteration = 0;
+        const maxIterations = getMaxIterations(options);
+        while (true) {
+            if (!(await shouldContinue(context, iteration, maxIterations, options))) {
+                break;
+            }
+            const currentMessages = history.getMessages();
+            const result = await executeRound(model, currentMessages, options, context, history, iteration);
+            for (const event of result.events) {
+                yield event;
+            }
+            if (result.done || (!options.loop && !result.hasToolCalls)) {
+                break;
+            }
+            iteration++;
+        }
+        if (options.debug) {
+            yield createMetricsEvent(context, history);
+        }
+    }
+    catch (error) {
+        yield EnsembleErrorHandler.toStreamEvent(error);
+    }
+    finally {
+        yield {
+            type: 'stream_end',
+            timestamp: new Date().toISOString()
+        };
+    }
+}
+async function executeRound(model, messages, options, context, history, iteration) {
+    const events = [];
+    const collectedToolCalls = [];
+    const toolResults = [];
+    let messageContent = '';
+    let hasMessage = false;
+    const provider = getModelProvider(model);
+    const roundOptions = { ...options };
+    if (iteration > 0 && !options.toolChoiceStrategy) {
+        if (roundOptions.modelSettings?.tool_choice) {
+            roundOptions.modelSettings = { ...roundOptions.modelSettings };
+            delete roundOptions.modelSettings.tool_choice;
+        }
+    }
+    const agent = new RequestAgent(roundOptions);
+    if (options.toolChoiceStrategy && context) {
+        const toolChoice = options.toolChoiceStrategy(context.toolCallCount, context.turnCount, context);
+        agent.modelSettings = {
+            ...agent.modelSettings,
+            tool_choice: toolChoice
+        };
+    }
+    const stream = provider.createResponseStream(model, messages, agent);
+    for await (const event of stream) {
+        if (options.allowedEvents && !options.allowedEvents.includes(event.type)) {
+            continue;
+        }
+        if (options.eventEmitter) {
+            await options.eventEmitter(event, context);
+        }
+        events.push(event);
+        switch (event.type) {
+            case 'message_complete':
+                if ('content' in event) {
+                    messageContent = event.content;
+                    hasMessage = true;
+                }
+                break;
+            case 'tool_start':
+                if ('tool_calls' in event && event.tool_calls) {
+                    const processedCalls = await processToolCalls(event.tool_calls, options, context);
+                    collectedToolCalls.push(...processedCalls.calls);
+                    toolResults.push(...processedCalls.results);
+                }
+                break;
+            case 'error':
+                if (context) {
+                    context.halt();
+                }
+                break;
+        }
+    }
+    if (hasMessage || collectedToolCalls.length > 0) {
+        history.addAssistantResponse(messageContent, collectedToolCalls.map((call, i) => ({
+            ...call,
+            result: toolResults[i]?.output
+        })));
+    }
+    if (context) {
+        context.messages = history.getMessages();
+        context.toolCallCount += collectedToolCalls.length;
+    }
+    return {
+        events,
+        done: context?.isHalted || false,
+        hasToolCalls: collectedToolCalls.length > 0
+    };
+}
+async function processToolCalls(toolCalls, options, context) {
+    const processedCalls = [];
+    const results = [];
+    for (const toolCall of toolCalls) {
+        if (options.toolHandler?.onToolCall) {
+            const action = await options.toolHandler.onToolCall(toolCall, options.toolHandler.context || context);
+            if (action === ToolCallAction.SKIP) {
+                continue;
+            }
+            if (action === ToolCallAction.HALT && context) {
+                context.halt();
+                break;
+            }
+        }
+        processedCalls.push(toolCall);
+        try {
+            let result;
+            if (options.processToolCall) {
+                result = await options.processToolCall([toolCall]);
+            }
+            else if (options.toolHandler?.executor) {
+                const tool = options.tools?.find(t => t.definition.function.name === toolCall.function.name);
+                if (tool) {
+                    result = await options.toolHandler.executor(tool, JSON.parse(toolCall.function.arguments || '{}'), options.toolHandler.context || context);
+                }
+            }
+            else if (options.tools) {
+                const tool = options.tools.find(t => t.definition.function.name === toolCall.function.name);
+                if (tool && 'function' in tool) {
+                    result = await tool.function(JSON.parse(toolCall.function.arguments || '{}'));
+                }
+            }
+            if (result !== undefined && options.toolResultTransformer?.transform) {
+                result = options.toolResultTransformer.transform(toolCall.function.name, result, options.toolHandler?.context || context);
+            }
+            results.push({
+                id: toolCall.id,
+                call_id: toolCall.call_id || toolCall.id,
+                output: typeof result === 'string' ? result : JSON.stringify(result)
+            });
+            if (options.toolHandler?.onToolComplete) {
+                await options.toolHandler.onToolComplete(toolCall, result, options.toolHandler.context || context);
+            }
+        }
+        catch (error) {
+            let errorResult = `Tool execution failed: ${error}`;
+            if (options.toolHandler?.onToolError) {
+                errorResult = await options.toolHandler.onToolError(toolCall, error, options.toolHandler.context || context) || errorResult;
+            }
+            results.push({
+                id: toolCall.id,
+                call_id: toolCall.call_id || toolCall.id,
+                output: errorResult
+            });
+        }
+    }
+    return { calls: processedCalls, results };
+}
+async function shouldContinue(context, iteration, maxIterations, options) {
+    if (context?.isHalted)
+        return false;
+    if (iteration >= maxIterations)
+        return false;
+    if (options.loop && typeof options.loop === 'object' && options.loop.continueCondition) {
+        return await Promise.resolve(options.loop.continueCondition(context));
+    }
+    return true;
+}
+function getMaxIterations(options) {
+    if (!options.loop)
+        return 1;
+    if (typeof options.loop === 'boolean')
+        return options.loop ? Infinity : 1;
+    return options.loop.maxIterations || Infinity;
+}
+function createMetricsEvent(context, history) {
+    return {
+        type: 'system_update',
+        data: {
+            type: 'metrics',
+            messageCount: history.count(),
+            toolCallCount: context?.toolCallCount || 0,
+            duration: context ? Date.now() - context.startTime : 0,
+            summary: history.getSummary()
+        },
+        timestamp: new Date().toISOString()
+    };
+}
+export { RequestAgent };
+//# sourceMappingURL=unified_request.js.map

package/dist/unified_request.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"unified_request.js","sourceRoot":"","sources":["../unified_request.ts"],"names":[],"mappings":"AAcA,OAAO,EAGH,oBAAoB,EACpB,cAAc,EACjB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,gBAAgB,EAAE,MAAM,qCAAqC,CAAC;AACvE,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,oBAAoB,EAAa,MAAM,0BAA0B,CAAC;AAG3E,MAAM,YAAY;IACd,QAAQ,CAAS;IACjB,aAAa,CAAiB;IAC9B,UAAU,CAAgB;IAClB,KAAK,CAAiB;IAE9B,YAAY,OAA2B;QACnC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,OAAO,IAAI,UAAU,CAAC;QAC9C,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;QAC3C,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QACrC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,IAAI,CAAC,KAAK,CAAC;IACtB,CAAC;CACJ;AAYD,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,cAAc,CACjC,KAAa,EACb,QAAuB,EACvB,UAAiC,EAAE;IAGnC,MAAM,WAAW,GAAG,OAAO,CAAC,eAAe;QACvC,OAAO,CAAC,WAAW;QACnB,OAAO,CAAC,IAAI;QACZ,OAAO,CAAC,cAAc;QACtB,OAAO,CAAC,UAAU;QAClB,OAAO,CAAC,qBAAqB,CAAC;IAGlC,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,IAAI,IAAI,cAAc,CAAC,QAAQ,EAAE;QACnE,gBAAgB,EAAE,IAAI;QACtB,sBAAsB,EAAE,IAAI;KAC/B,CAAC,CAAC;IAGH,MAAM,OAAO,GAAG,WAAW,CAAC,CAAC,CAAC,oBAAoB,CAAC;QAC/C,QAAQ,EAAE,OAAO,CAAC,WAAW,EAAE;KAClC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAEf,IAAI,CAAC;QAED,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,aAAa,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAEhD,OAAO,IAAI,EAAE,CAAC;YAEV,IAAI,CAAC,CAAC,MAAM,cAAc,CAAC,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC,EAAE,CAAC;gBACtE,MAAM;YACV,CAAC;YAGD,MAAM,eAAe,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;YAG9C,MAAM,MAAM,GAAG,MAAM,YAAY,CAC7B,KAAK,EACL,eAAe,EACf,OAAO,EACP,OAAO,EACP,OAAO,EACP,SAAS,CACZ,CAAC;YAGF,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAChC,MAAM,KAAK,CAAC;YAChB,CAAC;YAGD,IAAI,MAAM,CAAC,IAAI,IAAI,CAAC,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;gBACzD,MAAM;YACV,CAAC;YAED,SAAS,EAAE,CAAC;QAChB,CAAC;QAGD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC/C,CAAC;IAEL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAEb,MAAM,oBAAoB,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;IACpD,CAAC;YAAS,CAAC;QAEP,MAAM;YACF,IAAI,EAAE,YAAY;YAClB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACf,CAAC;IAC7B,CAAC;AACL,CAAC;AAKD,KAAK,UAAU,YAAY,CACvB,KAAa,EACb,QAAuB,EACvB,OAA8B,EAC9B,OAAmC,EACnC,OAAuB,EACvB,SAAiB;IAEjB,MAAM,MAAM,GAA0B,EAAE,CAAC;IACzC,MAAM,kBAAkB,GAAe,EAAE,CAAC;IAC1C,MAAM,WAAW,GAA2D,EAAE,CAAC;IAC/E,IAAI,cAAc,GAAG,EAAE,CAAC;IACxB,IAAI,UAAU,GAAG,KAAK,CAAC;IAGvB,MAAM,QAAQ,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;IAGzC,MAAM,YAAY,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC;IAIpC,IAAI,SAAS,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAC/C,IAAI,YAAY,CAAC,aAAa,EAAE,WAAW,EAAE,CAAC;YAE1C,YAAY,CAAC,aAAa,GAAG,EAAE,GAAG,YAAY,CAAC,aAAa,EAAE,CAAC;YAC/D,OAAO,YAAY,CAAC,aAAa,CAAC,WAAW,CAAC;QAClD,CAAC;IACL,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,YAAkC,CAAC,CAAC;IAGnE,IAAI,OAAO,CAAC,kBAAkB,IAAI,OAAO,EAAE,CAAC;QACxC,MAAM,UAAU,GAAG,OAAO,CAAC,kBAAkB,CACzC,OAAO,CAAC,aAAa,EACrB,OAAO,CAAC,SAAS,EACjB,OAAO,CACV,CAAC;QACF,KAAK,CAAC,aAAa,GAAG;YAClB,GAAG,KAAK,CAAC,aAAa;YACtB,WAAW,EAAE,UAAU;SAC1B,CAAC;IACN,CAAC;IAGD,MAAM,MAAM,GAAG,QAAQ,CAAC,oBAAoB,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAY,CAAC,CAAC;IAE5E,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAE/B,IAAI,OAAO,CAAC,aAAa,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YACvE,SAAS;QACb,CAAC;QAGD,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/C,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAGnB,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACjB,KAAK,kBAAkB;gBACnB,IAAI,SAAS,IAAI,KAAK,EAAE,CAAC;oBACrB,cAAc,GAAG,KAAK,CAAC,OAAO,CAAC;oBAC/B,UAAU,GAAG,IAAI,CAAC;gBACtB,CAAC;gBACD,MAAM;YAEV,KAAK,YAAY;gBACb,IAAI,YAAY,IAAI,KAAK,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBAE5C,MAAM,cAAc,GAAG,MAAM,gBAAgB,CACzC,KAAK,CAAC,UAAU,EAChB,OAAO,EACP,OAAO,CACV,CAAC;oBAEF,kBAAkB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;oBACjD,WAAW,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;gBAChD,CAAC;gBACD,MAAM;YAEV,KAAK,OAAO;gBACR,IAAI,OAAO,EAAE,CAAC;oBACV,OAAO,CAAC,IAAI,EAAE,CAAC;gBACnB,CAAC;gBACD,MAAM;QACd,CAAC;IACL,CAAC;IAGD,IAAI,UAAU,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9C,OAAO,CAAC,oBAAoB,CACxB,cAAc,EACd,kBAAkB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACjC,GAAG,IAAI;YACP,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,EAAE,MAAM;SACjC,CAAC,CAAC,CACN,CAAC;IACN,CAAC;IAGD,IAAI,OAAO,EAAE,CAAC;QACV,OAAO,CAAC,QAAQ,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACzC,OAAO,CAAC,aAAa,IAAI,kBAAkB,CAAC,MAAM,CAAC;IACvD,CAAC;IAED,OAAO;QACH,MAAM;QACN,IAAI,EAAE,OAAO,EAAE,QAAQ,IAAI,KAAK;QAChC,YAAY,EAAE,kBAAkB,CAAC,MAAM,GAAG,CAAC;KAC9C,CAAC;AACN,CAAC;AAKD,KAAK,UAAU,gBAAgB,CAC3B,SAAqB,EACrB,OAA8B,EAC9B,OAAwB;IAExB,MAAM,cAAc,GAAe,EAAE,CAAC;IACtC,MAAM,OAAO,GAAU,EAAE,CAAC;IAE1B,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QAE/B,IAAI,OAAO,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,UAAU,CAC/C,QAAQ,EACR,OAAO,CAAC,WAAW,CAAC,OAAO,IAAI,OAAO,CACzC,CAAC;YAEF,IAAI,MAAM,KAAK,cAAc,CAAC,IAAI,EAAE,CAAC;gBACjC,SAAS;YACb,CAAC;YAED,IAAI,MAAM,KAAK,cAAc,CAAC,IAAI,IAAI,OAAO,EAAE,CAAC;gBAC5C,OAAO,CAAC,IAAI,EAAE,CAAC;gBACf,MAAM;YACV,CAAC;QACL,CAAC;QAED,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAG9B,IAAI,CAAC;YACD,IAAI,MAAW,CAAC;YAEhB,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;gBAE1B,MAAM,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;YACvD,CAAC;iBAAM,IAAI,OAAO,CAAC,WAAW,EAAE,QAAQ,EAAE,CAAC;gBAEvC,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,EAAE,IAAI,CAC5B,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,KAAK,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAC7D,CAAC;gBACF,IAAI,IAAI,EAAE,CAAC;oBACP,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,QAAQ,CACvC,IAAI,EACJ,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC,EAC/C,OAAO,CAAC,WAAW,CAAC,OAAO,IAAI,OAAO,CACzC,CAAC;gBACN,CAAC;YACL,CAAC;iBAAM,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAEvB,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAC3B,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,KAAK,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAC7D,CAAC;gBACF,IAAI,IAAI,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;oBAC7B,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CACxB,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC,CAClD,CAAC;gBACN,CAAC;YACL,CAAC;YAGD,IAAI,MAAM,KAAK,SAAS,IAAI,OAAO,CAAC,qBAAqB,EAAE,SAAS,EAAE,CAAC;gBACnE,MAAM,GAAG,OAAO,CAAC,qBAAqB,CAAC,SAAS,CAC5C,QAAQ,CAAC,QAAQ,CAAC,IAAI,EACtB,MAAM,EACN,OAAO,CAAC,WAAW,EAAE,OAAO,IAAI,OAAO,CAC1C,CAAC;YACN,CAAC;YAED,OAAO,CAAC,IAAI,CAAC;gBACT,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE;gBACxC,MAAM,EAAE,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aACvE,CAAC,CAAC;YAGH,IAAI,OAAO,CAAC,WAAW,EAAE,cAAc,EAAE,CAAC;gBACtC,MAAM,OAAO,CAAC,WAAW,CAAC,cAAc,CACpC,QAAQ,EACR,MAAM,EACN,OAAO,CAAC,WAAW,CAAC,OAAO,IAAI,OAAO,CACzC,CAAC;YACN,CAAC;QAEL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEb,IAAI,WAAW,GAAG,0BAA0B,KAAK,EAAE,CAAC;YAEpD,IAAI,OAAO,CAAC,WAAW,EAAE,WAAW,EAAE,CAAC;gBACnC,WAAW,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,WAAW,CAC/C,QAAQ,EACR,KAAc,EACd,OAAO,CAAC,WAAW,CAAC,OAAO,IAAI,OAAO,CACzC,IAAI,WAAW,CAAC;YACrB,CAAC;YAED,OAAO,CAAC,IAAI,CAAC;gBACT,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE;gBACxC,MAAM,EAAE,WAAW;aACtB,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE,CAAC;AAC9C,CAAC;AAKD,KAAK,UAAU,cAAc,CACzB,OAAmC,EACnC,SAAiB,EACjB,aAAqB,EACrB,OAA8B;IAE9B,IAAI,OAAO,EAAE,QAAQ;QAAE,OAAO,KAAK,CAAC;IACpC,IAAI,SAAS,IAAI,aAAa;QAAE,OAAO,KAAK,CAAC;IAE7C,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACrF,OAAO,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAQ,CAAC,CAAC,CAAC;IAC3E,CAAC;IAED,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,gBAAgB,CAAC,OAA8B;IACpD,IAAI,CAAC,OAAO,CAAC,IAAI;QAAE,OAAO,CAAC,CAAC;IAC5B,IAAI,OAAO,OAAO,CAAC,IAAI,KAAK,SAAS;QAAE,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1E,OAAO,OAAO,CAAC,IAAI,CAAC,aAAa,IAAI,QAAQ,CAAC;AAClD,CAAC;AAED,SAAS,kBAAkB,CACvB,OAAmC,EACnC,OAAuB;IAEvB,OAAO;QACH,IAAI,EAAE,eAAe;QACrB,IAAI,EAAE;YACF,IAAI,EAAE,SAAS;YACf,YAAY,EAAE,OAAO,CAAC,KAAK,EAAE;YAC7B,aAAa,EAAE,OAAO,EAAE,aAAa,IAAI,CAAC;YAC1C,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACtD,OAAO,EAAE,OAAO,CAAC,UAAU,EAAE;SAChC;QACD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACf,CAAC;AAC7B,CAAC;AAGD,OAAO,EAAE,YAAY,EAAyB,CAAC"}

package/dist/utils/error_handler.d.ts

@@ -0,0 +1,33 @@
+import { EnsembleStreamEvent } from '../types.js';
+export declare enum ErrorCode {
+    PROVIDER_NOT_FOUND = "PROVIDER_NOT_FOUND",
+    PROVIDER_AUTH_FAILED = "PROVIDER_AUTH_FAILED",
+    PROVIDER_RATE_LIMIT = "PROVIDER_RATE_LIMIT",
+    PROVIDER_QUOTA_EXCEEDED = "PROVIDER_QUOTA_EXCEEDED",
+    TOOL_NOT_FOUND = "TOOL_NOT_FOUND",
+    TOOL_EXECUTION_FAILED = "TOOL_EXECUTION_FAILED",
+    TOOL_TIMEOUT = "TOOL_TIMEOUT",
+    TOOL_VALIDATION_FAILED = "TOOL_VALIDATION_FAILED",
+    MAX_TOOL_CALLS_EXCEEDED = "MAX_TOOL_CALLS_EXCEEDED",
+    LOOP_TIMEOUT = "LOOP_TIMEOUT",
+    INVALID_REQUEST = "INVALID_REQUEST",
+    UNKNOWN_ERROR = "UNKNOWN_ERROR"
+}
+export interface EnsembleError extends Error {
+    code: ErrorCode;
+    details?: any;
+    recoverable?: boolean;
+    retry?: {
+        attempts: number;
+        delay: number;
+    };
+}
+export declare class EnsembleErrorHandler {
+    private static retryDelays;
+    static createError(code: ErrorCode, message: string, details?: any, recoverable?: boolean): EnsembleError;
+    static handleWithRetry<T>(operation: () => Promise<T>, maxRetries?: number, isRetryable?: (error: any) => boolean): Promise<T>;
+    static isRetryableError(error: any): boolean;
+    static toStreamEvent(error: any): EnsembleStreamEvent;
+    static getUserMessage(error: any): string;
+}
+//# sourceMappingURL=error_handler.d.ts.map

package/dist/utils/error_handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"error_handler.d.ts","sourceRoot":"","sources":["../../utils/error_handler.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAElD,oBAAY,SAAS;IAEjB,kBAAkB,uBAAuB;IACzC,oBAAoB,yBAAyB;IAC7C,mBAAmB,wBAAwB;IAC3C,uBAAuB,4BAA4B;IAGnD,cAAc,mBAAmB;IACjC,qBAAqB,0BAA0B;IAC/C,YAAY,iBAAiB;IAC7B,sBAAsB,2BAA2B;IAGjD,uBAAuB,4BAA4B;IACnD,YAAY,iBAAiB;IAC7B,eAAe,oBAAoB;IAGnC,aAAa,kBAAkB;CAClC;AAED,MAAM,WAAW,aAAc,SAAQ,KAAK;IACxC,IAAI,EAAE,SAAS,CAAC;IAChB,OAAO,CAAC,EAAE,GAAG,CAAC;IACd,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,KAAK,CAAC,EAAE;QACJ,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,CAAC;KACjB,CAAC;CACL;AAED,qBAAa,oBAAoB;IAC7B,OAAO,CAAC,MAAM,CAAC,WAAW,CAA4B;IAKtD,MAAM,CAAC,WAAW,CACd,IAAI,EAAE,SAAS,EACf,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,GAAG,EACb,WAAW,UAAQ,GACpB,aAAa;WAWH,eAAe,CAAC,CAAC,EAC1B,SAAS,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAC3B,UAAU,SAAI,EACd,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,OAAO,GACtC,OAAO,CAAC,CAAC,CAAC;IA6Bb,MAAM,CAAC,gBAAgB,CAAC,KAAK,EAAE,GAAG,GAAG,OAAO;IAoB5C,MAAM,CAAC,aAAa,CAAC,KAAK,EAAE,GAAG,GAAG,mBAAmB;IAgBrD,MAAM,CAAC,cAAc,CAAC,KAAK,EAAE,GAAG,GAAG,MAAM;CAgB5C"}

package/dist/utils/error_handler.js

@@ -0,0 +1,84 @@
+export var ErrorCode;
+(function (ErrorCode) {
+    ErrorCode["PROVIDER_NOT_FOUND"] = "PROVIDER_NOT_FOUND";
+    ErrorCode["PROVIDER_AUTH_FAILED"] = "PROVIDER_AUTH_FAILED";
+    ErrorCode["PROVIDER_RATE_LIMIT"] = "PROVIDER_RATE_LIMIT";
+    ErrorCode["PROVIDER_QUOTA_EXCEEDED"] = "PROVIDER_QUOTA_EXCEEDED";
+    ErrorCode["TOOL_NOT_FOUND"] = "TOOL_NOT_FOUND";
+    ErrorCode["TOOL_EXECUTION_FAILED"] = "TOOL_EXECUTION_FAILED";
+    ErrorCode["TOOL_TIMEOUT"] = "TOOL_TIMEOUT";
+    ErrorCode["TOOL_VALIDATION_FAILED"] = "TOOL_VALIDATION_FAILED";
+    ErrorCode["MAX_TOOL_CALLS_EXCEEDED"] = "MAX_TOOL_CALLS_EXCEEDED";
+    ErrorCode["LOOP_TIMEOUT"] = "LOOP_TIMEOUT";
+    ErrorCode["INVALID_REQUEST"] = "INVALID_REQUEST";
+    ErrorCode["UNKNOWN_ERROR"] = "UNKNOWN_ERROR";
+})(ErrorCode || (ErrorCode = {}));
+export class EnsembleErrorHandler {
+    static retryDelays = [1000, 2000, 4000, 8000];
+    static createError(code, message, details, recoverable = false) {
+        const error = new Error(message);
+        error.code = code;
+        error.details = details;
+        error.recoverable = recoverable;
+        return error;
+    }
+    static async handleWithRetry(operation, maxRetries = 3, isRetryable) {
+        let lastError;
+        for (let attempt = 0; attempt <= maxRetries; attempt++) {
+            try {
+                return await operation();
+            }
+            catch (error) {
+                lastError = error;
+                const shouldRetry = isRetryable ? isRetryable(error) :
+                    this.isRetryableError(error);
+                if (!shouldRetry || attempt === maxRetries) {
+                    throw error;
+                }
+                const delay = this.retryDelays[Math.min(attempt, this.retryDelays.length - 1)];
+                await new Promise(resolve => setTimeout(resolve, delay));
+            }
+        }
+        throw lastError;
+    }
+    static isRetryableError(error) {
+        if (error?.code) {
+            return [
+                ErrorCode.PROVIDER_RATE_LIMIT,
+                ErrorCode.TOOL_TIMEOUT,
+                ErrorCode.PROVIDER_QUOTA_EXCEEDED
+            ].includes(error.code);
+        }
+        if (error?.status) {
+            return [429, 502, 503, 504].includes(error.status);
+        }
+        return false;
+    }
+    static toStreamEvent(error) {
+        const ensembleError = error;
+        return {
+            type: 'error',
+            error: error.message || 'Unknown error',
+            code: ensembleError.code,
+            details: ensembleError.details,
+            recoverable: ensembleError.recoverable,
+            timestamp: new Date().toISOString()
+        };
+    }
+    static getUserMessage(error) {
+        const ensembleError = error;
+        switch (ensembleError.code) {
+            case ErrorCode.PROVIDER_AUTH_FAILED:
+                return 'Authentication failed. Please check your API key.';
+            case ErrorCode.PROVIDER_RATE_LIMIT:
+                return 'Rate limit exceeded. Please try again later.';
+            case ErrorCode.TOOL_NOT_FOUND:
+                return `Tool "${ensembleError.details?.toolName}" not found.`;
+            case ErrorCode.MAX_TOOL_CALLS_EXCEEDED:
+                return 'Maximum number of tool calls exceeded.';
+            default:
+                return error.message || 'An unexpected error occurred.';
+        }
+    }
+}
+//# sourceMappingURL=error_handler.js.map

package/dist/utils/error_handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"error_handler.js","sourceRoot":"","sources":["../../utils/error_handler.ts"],"names":[],"mappings":"AAMA,MAAM,CAAN,IAAY,SAoBX;AApBD,WAAY,SAAS;IAEjB,sDAAyC,CAAA;IACzC,0DAA6C,CAAA;IAC7C,wDAA2C,CAAA;IAC3C,gEAAmD,CAAA;IAGnD,8CAAiC,CAAA;IACjC,4DAA+C,CAAA;IAC/C,0CAA6B,CAAA;IAC7B,8DAAiD,CAAA;IAGjD,gEAAmD,CAAA;IACnD,0CAA6B,CAAA;IAC7B,gDAAmC,CAAA;IAGnC,4CAA+B,CAAA;AACnC,CAAC,EApBW,SAAS,KAAT,SAAS,QAoBpB;AAYD,MAAM,OAAO,oBAAoB;IACrB,MAAM,CAAC,WAAW,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IAKtD,MAAM,CAAC,WAAW,CACd,IAAe,EACf,OAAe,EACf,OAAa,EACb,WAAW,GAAG,KAAK;QAEnB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,OAAO,CAAkB,CAAC;QAClD,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC;QAClB,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;QACxB,KAAK,CAAC,WAAW,GAAG,WAAW,CAAC;QAChC,OAAO,KAAK,CAAC;IACjB,CAAC;IAKD,MAAM,CAAC,KAAK,CAAC,eAAe,CACxB,SAA2B,EAC3B,UAAU,GAAG,CAAC,EACd,WAAqC;QAErC,IAAI,SAAc,CAAC;QAEnB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;YACrD,IAAI,CAAC;gBACD,OAAO,MAAM,SAAS,EAAE,CAAC;YAC7B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,SAAS,GAAG,KAAK,CAAC;gBAGlB,MAAM,WAAW,GAAG,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;oBAClD,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;gBAEjC,IAAI,CAAC,WAAW,IAAI,OAAO,KAAK,UAAU,EAAE,CAAC;oBACzC,MAAM,KAAK,CAAC;gBAChB,CAAC;gBAGD,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC/E,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;YAC7D,CAAC;QACL,CAAC;QAED,MAAM,SAAS,CAAC;IACpB,CAAC;IAKD,MAAM,CAAC,gBAAgB,CAAC,KAAU;QAC9B,IAAI,KAAK,EAAE,IAAI,EAAE,CAAC;YACd,OAAO;gBACH,SAAS,CAAC,mBAAmB;gBAC7B,SAAS,CAAC,YAAY;gBACtB,SAAS,CAAC,uBAAuB;aACpC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;QAGD,IAAI,KAAK,EAAE,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACvD,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAKD,MAAM,CAAC,aAAa,CAAC,KAAU;QAC3B,MAAM,aAAa,GAAG,KAAsB,CAAC;QAE7C,OAAO;YACH,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,eAAe;YACvC,IAAI,EAAE,aAAa,CAAC,IAAI;YACxB,OAAO,EAAE,aAAa,CAAC,OAAO;YAC9B,WAAW,EAAE,aAAa,CAAC,WAAW;YACtC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACf,CAAC;IAC7B,CAAC;IAKD,MAAM,CAAC,cAAc,CAAC,KAAU;QAC5B,MAAM,aAAa,GAAG,KAAsB,CAAC;QAE7C,QAAQ,aAAa,CAAC,IAAI,EAAE,CAAC;YACzB,KAAK,SAAS,CAAC,oBAAoB;gBAC/B,OAAO,mDAAmD,CAAC;YAC/D,KAAK,SAAS,CAAC,mBAAmB;gBAC9B,OAAO,8CAA8C,CAAC;YAC1D,KAAK,SAAS,CAAC,cAAc;gBACzB,OAAO,SAAS,aAAa,CAAC,OAAO,EAAE,QAAQ,cAAc,CAAC;YAClE,KAAK,SAAS,CAAC,uBAAuB;gBAClC,OAAO,wCAAwC,CAAC;YACpD;gBACI,OAAO,KAAK,CAAC,OAAO,IAAI,+BAA+B,CAAC;QAChE,CAAC;IACL,CAAC"}

package/dist/utils/message_history.d.ts

@@ -0,0 +1,24 @@
+import { ResponseInput } from '../types.js';
+export interface MessageHistoryOptions {
+    maxMessages?: number;
+    maxTokens?: number;
+    preserveSystemMessages?: boolean;
+    compactToolCalls?: boolean;
+}
+export declare class MessageHistory {
+    private messages;
+    private options;
+    constructor(initialMessages?: ResponseInput, options?: MessageHistoryOptions);
+    add(message: any): void;
+    addMany(messages: any[]): void;
+    addAssistantResponse(content: string, toolCalls?: any[]): void;
+    getMessages(): ResponseInput;
+    count(): number;
+    clear(): void;
+    private trim;
+    private compactToolCalls;
+    getSummary(): string;
+    findLast(predicate: (msg: any) => boolean): any | undefined;
+    lastAssistantHadToolCalls(): boolean;
+}
+//# sourceMappingURL=message_history.d.ts.map

package/dist/utils/message_history.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"message_history.d.ts","sourceRoot":"","sources":["../../utils/message_history.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,MAAM,WAAW,qBAAqB;IAClC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,sBAAsB,CAAC,EAAE,OAAO,CAAC;IACjC,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC9B;AAED,qBAAa,cAAc;IACvB,OAAO,CAAC,QAAQ,CAAgB;IAChC,OAAO,CAAC,OAAO,CAAwB;gBAE3B,eAAe,GAAE,aAAkB,EAAE,OAAO,GAAE,qBAA0B;IAapF,GAAG,CAAC,OAAO,EAAE,GAAG,GAAG,IAAI;IAQvB,OAAO,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,IAAI;IAQ9B,oBAAoB,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,GAAG,EAAE,GAAG,IAAI;IAsC9D,WAAW,IAAI,aAAa;IAO5B,KAAK,IAAI,MAAM;IAOf,KAAK,IAAI,IAAI;IAUb,OAAO,CAAC,IAAI;IAqBZ,OAAO,CAAC,gBAAgB;IAmCxB,UAAU,IAAI,MAAM;IA4BpB,QAAQ,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,KAAK,OAAO,GAAG,GAAG,GAAG,SAAS;IAY3D,yBAAyB,IAAI,OAAO;CA2BvC"}

package/dist/utils/message_history.js

@@ -0,0 +1,154 @@
+export class MessageHistory {
+    messages;
+    options;
+    constructor(initialMessages = [], options = {}) {
+        this.messages = [...initialMessages];
+        this.options = {
+            maxMessages: options.maxMessages,
+            maxTokens: options.maxTokens,
+            preserveSystemMessages: options.preserveSystemMessages ?? true,
+            compactToolCalls: options.compactToolCalls ?? true
+        };
+    }
+    add(message) {
+        this.messages.push(message);
+        this.trim();
+    }
+    addMany(messages) {
+        this.messages.push(...messages);
+        this.trim();
+    }
+    addAssistantResponse(content, toolCalls) {
+        if (content || toolCalls?.length) {
+            this.add({
+                type: 'message',
+                role: 'assistant',
+                content: content || '',
+                status: 'completed'
+            });
+        }
+        if (toolCalls) {
+            for (const toolCall of toolCalls) {
+                this.add({
+                    type: 'function_call',
+                    id: toolCall.id,
+                    call_id: toolCall.call_id || toolCall.id,
+                    name: toolCall.function.name,
+                    arguments: toolCall.function.arguments
+                });
+                if (toolCall.result) {
+                    this.add({
+                        type: 'function_call_output',
+                        id: toolCall.id,
+                        call_id: toolCall.call_id || toolCall.id,
+                        name: toolCall.function.name,
+                        output: toolCall.result
+                    });
+                }
+            }
+        }
+    }
+    getMessages() {
+        return [...this.messages];
+    }
+    count() {
+        return this.messages.length;
+    }
+    clear() {
+        const systemMessages = this.options.preserveSystemMessages
+            ? this.messages.filter(m => m.type === 'message' && m.role === 'system')
+            : [];
+        this.messages = systemMessages;
+    }
+    trim() {
+        if (this.options.maxMessages && this.messages.length > this.options.maxMessages) {
+            const systemMessages = this.options.preserveSystemMessages
+                ? this.messages.filter(m => m.type === 'message' && m.role === 'system')
+                : [];
+            const nonSystemMessages = this.messages.filter(m => !(m.type === 'message' && m.role === 'system'));
+            const trimmedMessages = nonSystemMessages.slice(-this.options.maxMessages);
+            this.messages = [...systemMessages, ...trimmedMessages];
+        }
+        if (this.options.compactToolCalls) {
+            this.compactToolCalls();
+        }
+    }
+    compactToolCalls() {
+        const compacted = [];
+        let i = 0;
+        while (i < this.messages.length) {
+            const msg = this.messages[i];
+            if (msg.type === 'message' && msg.role === 'assistant') {
+                compacted.push(msg);
+                i++;
+                const toolCalls = [];
+                while (i < this.messages.length &&
+                    (this.messages[i].type === 'function_call' ||
+                        this.messages[i].type === 'function_call_output')) {
+                    toolCalls.push(this.messages[i]);
+                    i++;
+                }
+                compacted.push(...toolCalls);
+            }
+            else {
+                compacted.push(msg);
+                i++;
+            }
+        }
+        this.messages = compacted;
+    }
+    getSummary() {
+        const counts = {
+            user: 0,
+            assistant: 0,
+            system: 0,
+            toolCalls: 0,
+            toolOutputs: 0
+        };
+        for (const msg of this.messages) {
+            if (msg.type === 'message' && msg.role) {
+                const role = msg.role;
+                if (role in counts) {
+                    counts[role]++;
+                }
+            }
+            else if (msg.type === 'function_call') {
+                counts.toolCalls++;
+            }
+            else if (msg.type === 'function_call_output') {
+                counts.toolOutputs++;
+            }
+        }
+        return `Messages: ${this.messages.length} (User: ${counts.user}, Assistant: ${counts.assistant}, System: ${counts.system}, Tools: ${counts.toolCalls}/${counts.toolOutputs})`;
+    }
+    findLast(predicate) {
+        for (let i = this.messages.length - 1; i >= 0; i--) {
+            if (predicate(this.messages[i])) {
+                return this.messages[i];
+            }
+        }
+        return undefined;
+    }
+    lastAssistantHadToolCalls() {
+        let foundAssistant = false;
+        for (let i = this.messages.length - 1; i >= 0; i--) {
+            const msg = this.messages[i];
+            if (msg.type === 'function_call' && !foundAssistant) {
+                for (let j = i - 1; j >= 0; j--) {
+                    const msg = this.messages[j];
+                    if (msg.type === 'message' && msg.role === 'assistant') {
+                        return true;
+                    }
+                    if (msg.type === 'message' && msg.role === 'user') {
+                        break;
+                    }
+                }
+            }
+            if (msg.type === 'message' && msg.role === 'assistant') {
+                foundAssistant = true;
+            }
+        }
+        return false;
+    }
+}
+//# sourceMappingURL=message_history.js.map

package/dist/utils/message_history.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message_history.js","sourceRoot":"","sources":["../../utils/message_history.ts"],"names":[],"mappings":"AAaA,MAAM,OAAO,cAAc;IACf,QAAQ,CAAgB;IACxB,OAAO,CAAwB;IAEvC,YAAY,kBAAiC,EAAE,EAAE,UAAiC,EAAE;QAChF,IAAI,CAAC,QAAQ,GAAG,CAAC,GAAG,eAAe,CAAC,CAAC;QACrC,IAAI,CAAC,OAAO,GAAG;YACX,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,sBAAsB,EAAE,OAAO,CAAC,sBAAsB,IAAI,IAAI;YAC9D,gBAAgB,EAAE,OAAO,CAAC,gBAAgB,IAAI,IAAI;SACrD,CAAC;IACN,CAAC;IAKD,GAAG,CAAC,OAAY;QACZ,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5B,IAAI,CAAC,IAAI,EAAE,CAAC;IAChB,CAAC;IAKD,OAAO,CAAC,QAAe;QACnB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;QAChC,IAAI,CAAC,IAAI,EAAE,CAAC;IAChB,CAAC;IAKD,oBAAoB,CAAC,OAAe,EAAE,SAAiB;QAEnD,IAAI,OAAO,IAAI,SAAS,EAAE,MAAM,EAAE,CAAC;YAC/B,IAAI,CAAC,GAAG,CAAC;gBACL,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,OAAO,IAAI,EAAE;gBACtB,MAAM,EAAE,WAAW;aACtB,CAAC,CAAC;QACP,CAAC;QAGD,IAAI,SAAS,EAAE,CAAC;YACZ,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;gBAC/B,IAAI,CAAC,GAAG,CAAC;oBACL,IAAI,EAAE,eAAe;oBACrB,EAAE,EAAE,QAAQ,CAAC,EAAE;oBACf,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE;oBACxC,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;oBAC5B,SAAS,EAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS;iBACzC,CAAC,CAAC;gBAEH,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;oBAClB,IAAI,CAAC,GAAG,CAAC;wBACL,IAAI,EAAE,sBAAsB;wBAC5B,EAAE,EAAE,QAAQ,CAAC,EAAE;wBACf,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE;wBACxC,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;wBAC5B,MAAM,EAAE,QAAQ,CAAC,MAAM;qBAC1B,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;QACL,CAAC;IACL,CAAC;IAKD,WAAW;QACP,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC;IAKD,KAAK;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;IAChC,CAAC;IAKD,KAAK;QACD,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,sBAAsB;YACtD,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;YACxE,CAAC,CAAC,EAAE,CAAC;QACT,IAAI,CAAC,QAAQ,GAAG,cAAc,CAAC;IACnC,CAAC;IAKO,IAAI;QACR,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YAC9E,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,sBAAsB;gBACtD,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;gBACxE,CAAC,CAAC,EAAE,CAAC;YAET,MAAM,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC;YACpG,MAAM,eAAe,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAE3E,IAAI,CAAC,QAAQ,GAAG,CAAC,GAAG,cAAc,EAAE,GAAG,eAAe,CAAC,CAAC;QAC5D,CAAC;QAGD,IAAI,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;YAChC,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC5B,CAAC;IACL,CAAC;IAKO,gBAAgB;QACpB,MAAM,SAAS,GAAkB,EAAE,CAAC;QACpC,IAAI,CAAC,GAAG,CAAC,CAAC;QAEV,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YAG7B,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACrD,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACpB,CAAC,EAAE,CAAC;gBAGJ,MAAM,SAAS,GAAG,EAAE,CAAC;gBACrB,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM;oBACxB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,eAAe;wBACzC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,sBAAsB,CAAC,EAAE,CAAC;oBACxD,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;oBACjC,CAAC,EAAE,CAAC;gBACR,CAAC;gBAGD,SAAS,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;YACjC,CAAC;iBAAM,CAAC;gBACJ,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACpB,CAAC,EAAE,CAAC;YACR,CAAC;QACL,CAAC;QAED,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;IAC9B,CAAC;IAKD,UAAU;QACN,MAAM,MAAM,GAAG;YACX,IAAI,EAAE,CAAC;YACP,SAAS,EAAE,CAAC;YACZ,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,WAAW,EAAE,CAAC;SACjB,CAAC;QAEF,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC9B,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC;gBACrC,MAAM,IAAI,GAAG,GAAG,CAAC,IAA2B,CAAC;gBAC7C,IAAI,IAAI,IAAI,MAAM,EAAE,CAAC;oBACjB,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnB,CAAC;YACL,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBACtC,MAAM,CAAC,SAAS,EAAE,CAAC;YACvB,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,sBAAsB,EAAE,CAAC;gBAC7C,MAAM,CAAC,WAAW,EAAE,CAAC;YACzB,CAAC;QACL,CAAC;QAED,OAAO,aAAa,IAAI,CAAC,QAAQ,CAAC,MAAM,WAAW,MAAM,CAAC,IAAI,gBAAgB,MAAM,CAAC,SAAS,aAAa,MAAM,CAAC,MAAM,YAAY,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,WAAW,GAAG,CAAC;IAClL,CAAC;IAKD,QAAQ,CAAC,SAAgC;QACrC,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACjD,IAAI,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC9B,OAAO,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC5B,CAAC;QACL,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;IAKD,yBAAyB;QACrB,IAAI,cAAc,GAAG,KAAK,CAAC;QAE3B,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACjD,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YAE7B,IAAI,GAAG,CAAC,IAAI,KAAK,eAAe,IAAI,CAAC,cAAc,EAAE,CAAC;gBAElD,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;oBAC7B,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBACrD,OAAO,IAAI,CAAC;oBAChB,CAAC;oBACD,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBAEhD,MAAM;oBACV,CAAC;gBACL,CAAC;YACL,CAAC;YAED,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACrD,cAAc,GAAG,IAAI,CAAC;YAC1B,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;CACJ"}

package/dist/utils/tool_builder.d.ts

@@ -0,0 +1,51 @@
+import { EnhancedToolFunction } from '../types/tool_types.js';
+export interface ToolBuilderOptions {
+    name: string;
+    description?: string;
+    category?: string;
+    priority?: number;
+    maxExecutions?: number;
+    cooldown?: number;
+    sideEffects?: boolean;
+}
+export declare class ToolBuilder {
+    private options;
+    private parameters;
+    private requiredParams;
+    private implementation?;
+    constructor(name: string);
+    description(desc: string): this;
+    category(cat: string): this;
+    param(name: string, schema: any, required?: boolean): this;
+    string(name: string, description?: string, required?: boolean): this;
+    number(name: string, description?: string, required?: boolean): this;
+    boolean(name: string, description?: string, required?: boolean): this;
+    array(name: string, itemType: string, description?: string, required?: boolean): this;
+    object(name: string, properties: Record<string, any>, description?: string, required?: boolean): this;
+    enum(name: string, values: string[], description?: string, required?: boolean): this;
+    constraints(opts: {
+        maxExecutions?: number;
+        cooldown?: number;
+        priority?: number;
+    }): this;
+    hasSideEffects(): this;
+    implement(fn: (...args: any[]) => any): this;
+    build(): EnhancedToolFunction;
+}
+export declare function tool(name: string): ToolBuilder;
+export declare function createControlTools(handlers: {
+    onComplete?: (result: any) => void;
+    onError?: (error: any) => void;
+    onClarification?: (question: string, options?: string[]) => void;
+}): EnhancedToolFunction[];
+export declare function createToolBatch<T>(baseConfig: {
+    category?: string;
+    priority?: number;
+    prefix?: string;
+}, tools: Array<{
+    name: string;
+    description: string;
+    params?: Record<string, any>;
+    implement: (args: T) => Promise<any>;
+}>): EnhancedToolFunction[];
+//# sourceMappingURL=tool_builder.d.ts.map

package/dist/utils/tool_builder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool_builder.d.ts","sourceRoot":"","sources":["../../utils/tool_builder.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAG9D,MAAM,WAAW,kBAAkB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,OAAO,CAAC;CACzB;AAKD,qBAAa,WAAW;IACpB,OAAO,CAAC,OAAO,CAAqB;IACpC,OAAO,CAAC,UAAU,CAA2B;IAC7C,OAAO,CAAC,cAAc,CAAgB;IACtC,OAAO,CAAC,cAAc,CAAC,CAA0B;gBAErC,IAAI,EAAE,MAAM;IAOxB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAQ/B,QAAQ,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAQ3B,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,QAAQ,UAAQ,GAAG,IAAI;IAWxD,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,UAAO,GAAG,IAAI;IAUjE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,UAAO,GAAG,IAAI;IAUjE,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,UAAO,GAAG,IAAI;IAUlE,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,UAAO,GAAG,IAAI;IAWlF,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,UAAO,GAAG,IAAI;IAWlG,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,UAAO,GAAG,IAAI;IAWjF,WAAW,CAAC,IAAI,EAAE;QACd,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACrB,GAAG,IAAI;IAQR,cAAc,IAAI,IAAI;IAQtB,SAAS,CAAC,EAAE,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,GAAG,IAAI;IAQ5C,KAAK,IAAI,oBAAoB;CAkChC;AAKD,wBAAgB,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,WAAW,CAE9C;AAKD,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE;IACzC,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,IAAI,CAAC;IACnC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IAC/B,eAAe,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;CACpE,GAAG,oBAAoB,EAAE,CAyDzB;AAKD,wBAAgB,eAAe,CAAC,CAAC,EAC7B,UAAU,EAAE;IACR,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;CACnB,EACD,KAAK,EAAE,KAAK,CAAC;IACT,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,SAAS,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;CACxC,CAAC,GACH,oBAAoB,EAAE,CAkBxB"}