@just-every/ensemble 0.1.24 → 0.1.26

package/dist/types/tool_types.d.ts

@@ -0,0 +1,119 @@
+import { ToolFunction, ToolCall, StreamEvent, ResponseInput } from '../types.js';
+export declare enum ToolCallAction {
+    EXECUTE = "execute",
+    SKIP = "skip",
+    HALT = "halt",
+    DEFER = "defer",
+    RETRY = "retry",
+    REPLACE = "replace"
+}
+export interface ExecutionMetrics {
+    duration: number;
+    tokenCount: number;
+    toolCallCount: number;
+    modelCalls: number;
+    errors: number;
+    retries?: number;
+    timestamp: number;
+}
+export interface RequestContext {
+    shouldContinue: boolean;
+    metadata: Record<string, any>;
+    toolCallCount: number;
+    turnCount: number;
+    startTime: number;
+    messages: ResponseInput;
+    halt(): void;
+    pause(): void;
+    resume(): void;
+    setMetadata(key: string, value: any): void;
+    getMetadata<T = any>(key: string): T | undefined;
+    addMessage(message: any): void;
+    getHistory(): ResponseInput;
+    agent?: any;
+    isPaused: boolean;
+    isHalted: boolean;
+}
+export interface EnhancedToolFunction extends ToolFunction {
+    category?: 'control' | 'utility' | 'meta' | 'custom' | string;
+    priority?: number;
+    sideEffects?: boolean;
+    agentId?: string;
+    requiresContext?: string[];
+    maxExecutions?: number;
+    cooldown?: number;
+    timeout?: number;
+    dependsOn?: string[];
+    conflicts?: string[];
+}
+export interface ToolHandler {
+    context?: any;
+    onToolCall?: (toolCall: ToolCall, context: any) => Promise<ToolCallAction | {
+        action: ToolCallAction;
+        replacement?: any;
+    }>;
+    onToolComplete?: (toolCall: ToolCall, result: any, context: any) => Promise<void>;
+    onToolError?: (toolCall: ToolCall, error: Error, context: any) => Promise<any>;
+    executor?: (tool: ToolFunction, args: any, context: any) => Promise<any>;
+    executionMode?: 'sequential' | 'parallel' | 'batch';
+    errorStrategy?: 'throw' | 'return-error' | 'retry' | 'custom';
+    retryConfig?: {
+        maxAttempts?: number;
+        backoff?: 'linear' | 'exponential';
+        initialDelay?: number;
+    };
+}
+export interface LoopConfig {
+    maxIterations?: number;
+    maxDuration?: number;
+    continueCondition?: (context: RequestContext) => boolean | Promise<boolean>;
+    onIteration?: (iteration: number, context: RequestContext) => Promise<void>;
+    breakOnError?: boolean;
+    resetToolCount?: boolean;
+}
+export interface ToolResultTransformer {
+    transform?: (toolName: string, result: any, context: any) => any;
+    augment?: (toolName: string, result: any, metrics: ExecutionMetrics) => any;
+    format?: (toolName: string, result: any) => string;
+    validate?: (toolName: string, result: any) => boolean | {
+        valid: boolean;
+        error?: string;
+    };
+}
+export type ToolChoiceStrategy = (callCount: number, turnCount: number, context: any) => 'none' | 'auto' | 'required' | {
+    type: 'function';
+    function: {
+        name: string;
+    };
+};
+export interface EnhancedRequestOptions {
+    toolHandler?: ToolHandler;
+    toolCategories?: string[];
+    toolFilter?: (tool: ToolFunction) => boolean;
+    toolPriority?: (tools: ToolFunction[]) => ToolFunction[];
+    loop?: boolean | LoopConfig;
+    maxToolCalls?: number;
+    maxToolCallsPerTurn?: number;
+    toolChoiceStrategy?: ToolChoiceStrategy;
+    toolResultTransformer?: ToolResultTransformer;
+    allowedEvents?: string[];
+    eventEmitter?: (event: StreamEvent, context: any) => void | Promise<void>;
+    onStreamComplete?: (response: any, context: any) => Promise<boolean>;
+    cacheToolResults?: boolean;
+    parallelExecution?: number;
+    debug?: boolean | {
+        logToolCalls?: boolean;
+        logToolResults?: boolean;
+        logMessages?: boolean;
+        logMetrics?: boolean;
+    };
+}
+export declare function createRequestContext(initialData?: Partial<RequestContext>): RequestContext;
+export declare function isLoopConfig(value: boolean | LoopConfig): value is LoopConfig;
+export declare function normalizeToolChoice(choice: ReturnType<ToolChoiceStrategy>): 'none' | 'auto' | 'required' | {
+    type: 'function';
+    function: {
+        name: string;
+    };
+};
+//# sourceMappingURL=tool_types.d.ts.map

package/dist/types/tool_types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool_types.d.ts","sourceRoot":"","sources":["../../types/tool_types.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAKjF,oBAAY,cAAc;IACtB,OAAO,YAAY;IACnB,IAAI,SAAS;IACb,IAAI,SAAS;IACb,KAAK,UAAU;IACf,KAAK,UAAU;IACf,OAAO,YAAY;CACtB;AAKD,MAAM,WAAW,gBAAgB;IAC7B,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;CACrB;AAKD,MAAM,WAAW,cAAc;IAE3B,cAAc,EAAE,OAAO,CAAC;IACxB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAG9B,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAGlB,QAAQ,EAAE,aAAa,CAAC;IAGxB,IAAI,IAAI,IAAI,CAAC;IACb,KAAK,IAAI,IAAI,CAAC;IACd,MAAM,IAAI,IAAI,CAAC;IACf,WAAW,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,IAAI,CAAC;IAC3C,WAAW,CAAC,CAAC,GAAG,GAAG,EAAE,GAAG,EAAE,MAAM,GAAG,CAAC,GAAG,SAAS,CAAC;IACjD,UAAU,CAAC,OAAO,EAAE,GAAG,GAAG,IAAI,CAAC;IAC/B,UAAU,IAAI,aAAa,CAAC;IAG5B,KAAK,CAAC,EAAE,GAAG,CAAC;IAGZ,QAAQ,EAAE,OAAO,CAAC;IAClB,QAAQ,EAAE,OAAO,CAAC;CACrB;AAKD,MAAM,WAAW,oBAAqB,SAAQ,YAAY;IAEtD,QAAQ,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,MAAM,GAAG,QAAQ,GAAG,MAAM,CAAC;IAC9D,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,OAAO,CAAC;IAGtB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAG3B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;IAGjB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;CACxB;AAKD,MAAM,WAAW,WAAW;IAExB,OAAO,CAAC,EAAE,GAAG,CAAC;IAGd,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,KAAK,OAAO,CAAC,cAAc,GAAG;QAAE,MAAM,EAAE,cAAc,CAAC;QAAC,WAAW,CAAC,EAAE,GAAG,CAAA;KAAE,CAAC,CAAC;IAC3H,cAAc,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAClF,WAAW,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAG/E,QAAQ,CAAC,EAAE,CAAC,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAGzE,aAAa,CAAC,EAAE,YAAY,GAAG,UAAU,GAAG,OAAO,CAAC;IACpD,aAAa,CAAC,EAAE,OAAO,GAAG,cAAc,GAAG,OAAO,GAAG,QAAQ,CAAC;IAC9D,WAAW,CAAC,EAAE;QACV,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,OAAO,CAAC,EAAE,QAAQ,GAAG,aAAa,CAAC;QACnC,YAAY,CAAC,EAAE,MAAM,CAAC;KACzB,CAAC;CACL;AAKD,MAAM,WAAW,UAAU;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,iBAAiB,CAAC,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC5E,WAAW,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5E,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC5B;AAKD,MAAM,WAAW,qBAAqB;IAElC,SAAS,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,KAAK,GAAG,CAAC;IAGjE,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,gBAAgB,KAAK,GAAG,CAAC;IAG5E,MAAM,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,KAAK,MAAM,CAAC;IAGnD,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,KAAK,OAAO,GAAG;QAAE,KAAK,EAAE,OAAO,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;CAC9F;AAKD,MAAM,MAAM,kBAAkB,GAAG,CAC7B,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,GAAG,KACX,MAAM,GAAG,MAAM,GAAG,UAAU,GAAG;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,QAAQ,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,CAAC;AAKrF,MAAM,WAAW,sBAAsB;IAEnC,WAAW,CAAC,EAAE,WAAW,CAAC;IAG1B,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,YAAY,KAAK,OAAO,CAAC;IAC7C,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK,YAAY,EAAE,CAAC;IAGzD,IAAI,CAAC,EAAE,OAAO,GAAG,UAAU,CAAC;IAG5B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAG7B,kBAAkB,CAAC,EAAE,kBAAkB,CAAC;IAGxC,qBAAqB,CAAC,EAAE,qBAAqB,CAAC;IAG9C,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,GAAG,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1E,gBAAgB,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAGrE,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAG3B,KAAK,CAAC,EAAE,OAAO,GAAG;QACd,YAAY,CAAC,EAAE,OAAO,CAAC;QACvB,cAAc,CAAC,EAAE,OAAO,CAAC;QACzB,WAAW,CAAC,EAAE,OAAO,CAAC;QACtB,UAAU,CAAC,EAAE,OAAO,CAAC;KACxB,CAAC;CACL;AAKD,wBAAgB,oBAAoB,CAAC,WAAW,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,cAAc,CA+C1F;AAKD,wBAAgB,YAAY,CAAC,KAAK,EAAE,OAAO,GAAG,UAAU,GAAG,KAAK,IAAI,UAAU,CAE7E;AAKD,wBAAgB,mBAAmB,CAC/B,MAAM,EAAE,UAAU,CAAC,kBAAkB,CAAC,GACvC,MAAM,GAAG,MAAM,GAAG,UAAU,GAAG;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,QAAQ,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,CAKjF"}

package/dist/types/tool_types.js

@@ -0,0 +1,55 @@
+export var ToolCallAction;
+(function (ToolCallAction) {
+    ToolCallAction["EXECUTE"] = "execute";
+    ToolCallAction["SKIP"] = "skip";
+    ToolCallAction["HALT"] = "halt";
+    ToolCallAction["DEFER"] = "defer";
+    ToolCallAction["RETRY"] = "retry";
+    ToolCallAction["REPLACE"] = "replace";
+})(ToolCallAction || (ToolCallAction = {}));
+export function createRequestContext(initialData) {
+    const context = {
+        shouldContinue: true,
+        metadata: {},
+        toolCallCount: 0,
+        turnCount: 0,
+        startTime: Date.now(),
+        messages: [],
+        isPaused: false,
+        isHalted: false,
+        ...initialData,
+        halt() {
+            this.shouldContinue = false;
+            this.isHalted = true;
+        },
+        pause() {
+            this.isPaused = true;
+        },
+        resume() {
+            this.isPaused = false;
+        },
+        setMetadata(key, value) {
+            this.metadata[key] = value;
+        },
+        getMetadata(key) {
+            return this.metadata[key];
+        },
+        addMessage(message) {
+            this.messages.push(message);
+        },
+        getHistory() {
+            return [...this.messages];
+        }
+    };
+    return context;
+}
+export function isLoopConfig(value) {
+    return typeof value === 'object' && value !== null;
+}
+export function normalizeToolChoice(choice) {
+    if (typeof choice === 'string') {
+        return choice;
+    }
+    return choice;
+}
+//# sourceMappingURL=tool_types.js.map

package/dist/types/tool_types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool_types.js","sourceRoot":"","sources":["../../types/tool_types.ts"],"names":[],"mappings":"AAUA,MAAM,CAAN,IAAY,cAOX;AAPD,WAAY,cAAc;IACtB,qCAAmB,CAAA;IACnB,+BAAa,CAAA;IACb,+BAAa,CAAA;IACb,iCAAe,CAAA;IACf,iCAAe,CAAA;IACf,qCAAmB,CAAA;AACvB,CAAC,EAPW,cAAc,KAAd,cAAc,QAOzB;AAoLD,MAAM,UAAU,oBAAoB,CAAC,WAAqC;IACtE,MAAM,OAAO,GAAmB;QAE5B,cAAc,EAAE,IAAI;QACpB,QAAQ,EAAE,EAAE;QACZ,aAAa,EAAE,CAAC;QAChB,SAAS,EAAE,CAAC;QACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,QAAQ,EAAE,EAAE;QACZ,QAAQ,EAAE,KAAK;QACf,QAAQ,EAAE,KAAK;QAGf,GAAG,WAAW;QAGd,IAAI;YACA,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;YAC5B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACzB,CAAC;QAED,KAAK;YACD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACzB,CAAC;QAED,MAAM;YACF,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC;QAC1B,CAAC;QAED,WAAW,CAAC,GAAW,EAAE,KAAU;YAC/B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QAC/B,CAAC;QAED,WAAW,CAAU,GAAW;YAC5B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAM,CAAC;QACnC,CAAC;QAED,UAAU,CAAC,OAAY;YACnB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC;QAED,UAAU;YACN,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC;KACJ,CAAC;IAEF,OAAO,OAAO,CAAC;AACnB,CAAC;AAKD,MAAM,UAAU,YAAY,CAAC,KAA2B;IACpD,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACvD,CAAC;AAKD,MAAM,UAAU,mBAAmB,CAC/B,MAAsC;IAEtC,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC7B,OAAO,MAAM,CAAC;IAClB,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC"}

package/dist/unified_request.d.ts

@@ -0,0 +1,18 @@
+import { EnsembleStreamEvent, ResponseInput, RequestOptions as BaseRequestOptions, ModelSettings, ToolFunction, ModelClassID, EnsembleAgent } from './types.js';
+import { EnhancedRequestOptions } from './types/tool_types.js';
+import { MessageHistory } from './utils/message_history.js';
+declare class RequestAgent implements EnsembleAgent {
+    agent_id: string;
+    modelSettings?: ModelSettings;
+    modelClass?: ModelClassID;
+    private tools;
+    constructor(options: BaseRequestOptions);
+    getTools(): Promise<ToolFunction[]>;
+}
+interface UnifiedRequestOptions extends BaseRequestOptions, Partial<EnhancedRequestOptions> {
+    useEnhancedMode?: boolean;
+    messageHistory?: MessageHistory;
+}
+export declare function unifiedRequest(model: string, messages: ResponseInput, options?: UnifiedRequestOptions): AsyncGenerator<EnsembleStreamEvent>;
+export { RequestAgent, UnifiedRequestOptions };
+//# sourceMappingURL=unified_request.d.ts.map

package/dist/unified_request.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"unified_request.d.ts","sourceRoot":"","sources":["../unified_request.ts"],"names":[],"mappings":"AAIA,OAAO,EACH,mBAAmB,EACnB,aAAa,EAEb,cAAc,IAAI,kBAAkB,EACpC,aAAa,EACb,YAAY,EACZ,YAAY,EACZ,aAAa,EAChB,MAAM,YAAY,CAAC;AACpB,OAAO,EACH,sBAAsB,EAIzB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAI5D,cAAM,YAAa,YAAW,aAAa;IACvC,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,UAAU,CAAC,EAAE,YAAY,CAAC;IAC1B,OAAO,CAAC,KAAK,CAAiB;gBAElB,OAAO,EAAE,kBAAkB;IAOjC,QAAQ,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;CAG5C;AAGD,UAAU,qBAAsB,SAAQ,kBAAkB,EAAE,OAAO,CAAC,sBAAsB,CAAC;IAEvF,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,cAAc,CAAC,EAAE,cAAc,CAAC;CACnC;AAKD,wBAAuB,cAAc,CACjC,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,aAAa,EACvB,OAAO,GAAE,qBAA0B,GACpC,cAAc,CAAC,mBAAmB,CAAC,CAwErC;AAiRD,OAAO,EAAE,YAAY,EAAE,qBAAqB,EAAE,CAAC"}

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"}