merlion 0.1.0

package/dist/runtime/cost_gate.d.ts

@@ -0,0 +1,34 @@
+export type CostGateMode = 'off' | 'warn' | 'fail';
+export interface CostBaselineScenario {
+    total_tokens: number;
+    threshold_pct?: number;
+}
+export interface CostBaseline {
+    default_threshold_pct: number;
+    scenarios: Record<string, CostBaselineScenario>;
+}
+export type CostGateDecision = {
+    status: 'skip';
+    message: string;
+} | {
+    status: 'pass';
+    message: string;
+    thresholdTokens: number;
+} | {
+    status: 'warn';
+    message: string;
+    thresholdTokens: number;
+} | {
+    status: 'fail';
+    message: string;
+    thresholdTokens: number;
+};
+export declare function parseCostGateMode(raw: string | undefined): CostGateMode;
+export declare function evaluateCostGate(params: {
+    baseline: CostBaseline | null;
+    scenario: string;
+    totalTokens: number;
+    mode: CostGateMode;
+}): CostGateDecision;
+export declare function readCostBaseline(path: string): Promise<CostBaseline | null>;
+//# sourceMappingURL=cost_gate.d.ts.map

package/dist/runtime/cost_gate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost_gate.d.ts","sourceRoot":"","sources":["../../src/runtime/cost_gate.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,YAAY,GAAG,KAAK,GAAG,MAAM,GAAG,MAAM,CAAA;AAElD,MAAM,WAAW,oBAAoB;IACnC,YAAY,EAAE,MAAM,CAAA;IACpB,aAAa,CAAC,EAAE,MAAM,CAAA;CACvB;AAED,MAAM,WAAW,YAAY;IAC3B,qBAAqB,EAAE,MAAM,CAAA;IAC7B,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAA;CAChD;AAED,MAAM,MAAM,gBAAgB,GACxB;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GACnC;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,eAAe,EAAE,MAAM,CAAA;CAAE,GAC5D;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,eAAe,EAAE,MAAM,CAAA;CAAE,GAC5D;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,eAAe,EAAE,MAAM,CAAA;CAAE,CAAA;AAEhE,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAG,YAAY,CAKvE;AAED,wBAAgB,gBAAgB,CAAC,MAAM,EAAE;IACvC,QAAQ,EAAE,YAAY,GAAG,IAAI,CAAA;IAC7B,QAAQ,EAAE,MAAM,CAAA;IAChB,WAAW,EAAE,MAAM,CAAA;IACnB,IAAI,EAAE,YAAY,CAAA;CACnB,GAAG,gBAAgB,CAoCnB;AAMD,wBAAsB,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,CAqCjF"}

package/dist/runtime/cost_gate.js

@@ -0,0 +1,85 @@
+import { readFile } from 'node:fs/promises';
+export function parseCostGateMode(raw) {
+    const normalized = (raw ?? '').trim().toLowerCase();
+    if (normalized === 'off')
+        return 'off';
+    if (normalized === 'warn')
+        return 'warn';
+    return 'fail';
+}
+export function evaluateCostGate(params) {
+    if (params.mode === 'off') {
+        return { status: 'skip', message: '[cost-gate] skipped (mode=off)' };
+    }
+    if (!params.baseline) {
+        return { status: 'skip', message: '[cost-gate] skipped (no baseline loaded)' };
+    }
+    const scenarioBaseline = params.baseline.scenarios[params.scenario];
+    if (!scenarioBaseline) {
+        return { status: 'skip', message: `[cost-gate] skipped (no baseline for ${params.scenario})` };
+    }
+    const thresholdPct = Number.isFinite(scenarioBaseline.threshold_pct)
+        ? Math.max(0, scenarioBaseline.threshold_pct)
+        : Math.max(0, params.baseline.default_threshold_pct);
+    const baselineTokens = Math.max(0, Math.floor(scenarioBaseline.total_tokens));
+    const thresholdTokens = Math.ceil(baselineTokens * (1 + thresholdPct / 100));
+    const totalTokens = Math.max(0, Math.floor(params.totalTokens));
+    if (totalTokens <= thresholdTokens) {
+        return {
+            status: 'pass',
+            thresholdTokens,
+            message: `[cost-gate] pass ${params.scenario}: total=${totalTokens}, threshold=${thresholdTokens}`
+        };
+    }
+    const message = `[cost-gate] regression ${params.scenario}: total=${totalTokens} > threshold=${thresholdTokens} ` +
+        `(baseline=${baselineTokens}, threshold_pct=${thresholdPct})`;
+    if (params.mode === 'warn') {
+        return { status: 'warn', thresholdTokens, message };
+    }
+    return { status: 'fail', thresholdTokens, message };
+}
+function isObject(value) {
+    return typeof value === 'object' && value !== null;
+}
+export async function readCostBaseline(path) {
+    let text;
+    try {
+        text = await readFile(path, 'utf8');
+    }
+    catch {
+        return null;
+    }
+    let parsed;
+    try {
+        parsed = JSON.parse(text);
+    }
+    catch {
+        return null;
+    }
+    if (!isObject(parsed))
+        return null;
+    if (!isObject(parsed.scenarios))
+        return null;
+    const defaultThresholdRaw = parsed.default_threshold_pct;
+    const defaultThreshold = typeof defaultThresholdRaw === 'number' && Number.isFinite(defaultThresholdRaw)
+        ? defaultThresholdRaw
+        : 20;
+    const scenarios = {};
+    for (const [name, item] of Object.entries(parsed.scenarios)) {
+        if (!isObject(item))
+            continue;
+        const totalRaw = item.total_tokens;
+        if (typeof totalRaw !== 'number' || !Number.isFinite(totalRaw))
+            continue;
+        const thresholdRaw = item.threshold_pct;
+        scenarios[name] = {
+            total_tokens: totalRaw,
+            threshold_pct: typeof thresholdRaw === 'number' && Number.isFinite(thresholdRaw) ? thresholdRaw : undefined
+        };
+    }
+    return {
+        default_threshold_pct: defaultThreshold,
+        scenarios
+    };
+}
+//# sourceMappingURL=cost_gate.js.map

package/dist/runtime/cost_gate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost_gate.js","sourceRoot":"","sources":["../../src/runtime/cost_gate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAA;AAoB3C,MAAM,UAAU,iBAAiB,CAAC,GAAuB;IACvD,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAA;IACnD,IAAI,UAAU,KAAK,KAAK;QAAE,OAAO,KAAK,CAAA;IACtC,IAAI,UAAU,KAAK,MAAM;QAAE,OAAO,MAAM,CAAA;IACxC,OAAO,MAAM,CAAA;AACf,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAKhC;IACC,IAAI,MAAM,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;QAC1B,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,gCAAgC,EAAE,CAAA;IACtE,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACrB,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,0CAA0C,EAAE,CAAA;IAChF,CAAC;IAED,MAAM,gBAAgB,GAAG,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;IACnE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,wCAAwC,MAAM,CAAC,QAAQ,GAAG,EAAE,CAAA;IAChG,CAAC;IAED,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAC,aAAa,CAAC;QAClE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,gBAAgB,CAAC,aAAc,CAAC;QAC9C,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAA;IACtD,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC,CAAA;IAC7E,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC,GAAG,YAAY,GAAG,GAAG,CAAC,CAAC,CAAA;IAC5E,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAA;IAE/D,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;QACnC,OAAO;YACL,MAAM,EAAE,MAAM;YACd,eAAe;YACf,OAAO,EAAE,oBAAoB,MAAM,CAAC,QAAQ,WAAW,WAAW,eAAe,eAAe,EAAE;SACnG,CAAA;IACH,CAAC;IAED,MAAM,OAAO,GACX,0BAA0B,MAAM,CAAC,QAAQ,WAAW,WAAW,gBAAgB,eAAe,GAAG;QACjG,aAAa,cAAc,mBAAmB,YAAY,GAAG,CAAA;IAE/D,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC3B,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,OAAO,EAAE,CAAA;IACrD,CAAC;IACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,OAAO,EAAE,CAAA;AACrD,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAA;AACpD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,IAAY;IACjD,IAAI,IAAY,CAAA;IAChB,IAAI,CAAC;QACH,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;IACrC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAA;IACb,CAAC;IAED,IAAI,MAAe,CAAA;IACnB,IAAI,CAAC;QACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;IAC3B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAA;IACb,CAAC;IACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,IAAI,CAAA;IAClC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC;QAAE,OAAO,IAAI,CAAA;IAC5C,MAAM,mBAAmB,GAAG,MAAM,CAAC,qBAAqB,CAAA;IACxD,MAAM,gBAAgB,GAAG,OAAO,mBAAmB,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC;QACtG,CAAC,CAAC,mBAAmB;QACrB,CAAC,CAAC,EAAE,CAAA;IAEN,MAAM,SAAS,GAAyC,EAAE,CAAA;IAC1D,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;QAC5D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YAAE,SAAQ;QAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAA;QAClC,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,SAAQ;QACxE,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAA;QACvC,SAAS,CAAC,IAAI,CAAC,GAAG;YAChB,YAAY,EAAE,QAAQ;YACtB,aAAa,EAAE,OAAO,YAAY,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS;SAC5G,CAAA;IACH,CAAC;IAED,OAAO;QACL,qBAAqB,EAAE,gBAAgB;QACvC,SAAS;KACV,CAAA;AACH,CAAC"}

package/dist/runtime/executor.d.ts

@@ -0,0 +1,25 @@
+import type { ChatMessage, ToolCall } from '../types.js';
+import type { ToolContext, ToolUiPayload } from '../tools/types.js';
+import { ToolRegistry } from '../tools/registry.ts';
+export declare function partitionToolCalls(toolCalls: ToolCall[], registry: ToolRegistry): ToolCall[][];
+export interface ToolCallStartEvent {
+    call: ToolCall;
+    index: number;
+    total: number;
+}
+export interface ToolCallResultEvent extends ToolCallStartEvent {
+    message: ChatMessage;
+    isError: boolean;
+    durationMs: number;
+    uiPayload?: ToolUiPayload;
+}
+export interface ExecuteToolCallsOptions {
+    toolCalls: ToolCall[];
+    registry: ToolRegistry;
+    toolContext: ToolContext;
+    maxConcurrency: number;
+    onToolCallStart?: (event: ToolCallStartEvent) => Promise<void> | void;
+    onToolCallResult?: (event: ToolCallResultEvent) => Promise<void> | void;
+}
+export declare function executeToolCalls(options: ExecuteToolCallsOptions): Promise<ChatMessage[]>;
+//# sourceMappingURL=executor.d.ts.map

package/dist/runtime/executor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"executor.d.ts","sourceRoot":"","sources":["../../src/runtime/executor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAA;AACxD,OAAO,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAA;AACnE,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAA;AAsBnD,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,YAAY,GAAG,QAAQ,EAAE,EAAE,CAqB9F;AAqCD,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,QAAQ,CAAA;IACd,KAAK,EAAE,MAAM,CAAA;IACb,KAAK,EAAE,MAAM,CAAA;CACd;AAED,MAAM,WAAW,mBAAoB,SAAQ,kBAAkB;IAC7D,OAAO,EAAE,WAAW,CAAA;IACpB,OAAO,EAAE,OAAO,CAAA;IAChB,UAAU,EAAE,MAAM,CAAA;IAClB,SAAS,CAAC,EAAE,aAAa,CAAA;CAC1B;AA4CD,MAAM,WAAW,uBAAuB;IACtC,SAAS,EAAE,QAAQ,EAAE,CAAA;IACrB,QAAQ,EAAE,YAAY,CAAA;IACtB,WAAW,EAAE,WAAW,CAAA;IACxB,cAAc,EAAE,MAAM,CAAA;IACtB,eAAe,CAAC,EAAE,CAAC,KAAK,EAAE,kBAAkB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAA;IACrE,gBAAgB,CAAC,EAAE,CAAC,KAAK,EAAE,mBAAmB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAA;CACxE;AAED,wBAAsB,gBAAgB,CAAC,OAAO,EAAE,uBAAuB,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAuD/F"}

package/dist/runtime/executor.js

@@ -0,0 +1,136 @@
+import { applyToolResultBudget, resolveToolResultBudgetFromEnv } from "./budget.js";
+function parseToolArgs(raw) {
+    try {
+        return JSON.parse(raw);
+    }
+    catch {
+        return {};
+    }
+}
+function isConcurrencySafe(call, registry) {
+    const tool = registry.get(call.function.name);
+    return tool?.concurrencySafe === true;
+}
+export function partitionToolCalls(toolCalls, registry) {
+    const batches = [];
+    let currentSafeBatch = [];
+    for (const call of toolCalls) {
+        if (isConcurrencySafe(call, registry)) {
+            currentSafeBatch.push(call);
+            continue;
+        }
+        if (currentSafeBatch.length > 0) {
+            batches.push(currentSafeBatch);
+            currentSafeBatch = [];
+        }
+        batches.push([call]);
+    }
+    if (currentSafeBatch.length > 0) {
+        batches.push(currentSafeBatch);
+    }
+    return batches;
+}
+async function runToolCall(call, registry, toolContext) {
+    const tool = registry.get(call.function.name);
+    if (!tool) {
+        return {
+            message: {
+                role: 'tool',
+                tool_call_id: call.id,
+                content: `Unknown tool: ${call.function.name}`
+            },
+            isError: true
+        };
+    }
+    const args = parseToolArgs(call.function.arguments);
+    const result = await tool.execute(args, toolContext);
+    const budget = resolveToolResultBudgetFromEnv();
+    const budgeted = applyToolResultBudget(result.content, budget);
+    const content = budgeted.truncated
+        ? `${budgeted.content}\n[tool result truncated by budget]`
+        : budgeted.content;
+    return {
+        message: {
+            role: 'tool',
+            tool_call_id: call.id,
+            content: content && content.trim() !== '' ? content : '(no output)'
+        },
+        isError: result.isError,
+        uiPayload: result.uiPayload
+    };
+}
+async function executeSafeBatch(batch, registry, toolContext, maxConcurrency, indexById, total, onToolCallStart, onToolCallResult) {
+    const results = new Array(batch.length);
+    let cursor = 0;
+    async function worker() {
+        for (;;) {
+            const index = cursor;
+            cursor += 1;
+            if (index >= batch.length)
+                return;
+            const call = batch[index];
+            const displayIndex = indexById.get(call.id) ?? index + 1;
+            await onToolCallStart?.({ call, index: displayIndex, total });
+            const startedAt = Date.now();
+            const outcome = await runToolCall(call, registry, toolContext);
+            const durationMs = Date.now() - startedAt;
+            results[index] = outcome;
+            await onToolCallResult?.({
+                call,
+                index: displayIndex,
+                total,
+                message: outcome.message,
+                isError: outcome.isError,
+                durationMs,
+                uiPayload: outcome.uiPayload
+            });
+        }
+    }
+    const workerCount = Math.max(1, Math.min(maxConcurrency, batch.length));
+    await Promise.all(Array.from({ length: workerCount }, () => worker()));
+    return results;
+}
+export async function executeToolCalls(options) {
+    const { toolCalls, registry, toolContext } = options;
+    const batches = partitionToolCalls(toolCalls, registry);
+    const total = toolCalls.length;
+    const indexById = new Map();
+    toolCalls.forEach((call, i) => indexById.set(call.id, i + 1));
+    const resultsById = new Map();
+    for (const batch of batches) {
+        if (batch.length === 0)
+            continue;
+        const safe = batch.every((call) => isConcurrencySafe(call, registry));
+        const batchResults = safe
+            ? await executeSafeBatch(batch, registry, toolContext, options.maxConcurrency, indexById, total, options.onToolCallStart, options.onToolCallResult)
+            : await (async () => {
+                const call = batch[0];
+                const index = indexById.get(call.id) ?? 1;
+                await options.onToolCallStart?.({ call, index, total });
+                const startedAt = Date.now();
+                const outcome = await runToolCall(call, registry, toolContext);
+                const durationMs = Date.now() - startedAt;
+                await options.onToolCallResult?.({
+                    call,
+                    index,
+                    total,
+                    message: outcome.message,
+                    isError: outcome.isError,
+                    durationMs,
+                    uiPayload: outcome.uiPayload
+                });
+                return [outcome];
+            })();
+        for (const { message } of batchResults) {
+            if (message.tool_call_id) {
+                resultsById.set(message.tool_call_id, message);
+            }
+        }
+    }
+    return toolCalls.map((call) => (resultsById.get(call.id) ?? {
+        role: 'tool',
+        tool_call_id: call.id,
+        content: '[Tool execution missing result]'
+    }));
+}
+//# sourceMappingURL=executor.js.map

package/dist/runtime/executor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../src/runtime/executor.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,qBAAqB,EAAE,8BAA8B,EAAE,MAAM,aAAa,CAAA;AAQnF,SAAS,aAAa,CAAC,GAAW;IAChC,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAA4B,CAAA;IACnD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAA;IACX,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAc,EAAE,QAAsB;IAC/D,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;IAC7C,OAAO,IAAI,EAAE,eAAe,KAAK,IAAI,CAAA;AACvC,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,SAAqB,EAAE,QAAsB;IAC9E,MAAM,OAAO,GAAiB,EAAE,CAAA;IAChC,IAAI,gBAAgB,GAAe,EAAE,CAAA;IAErC,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;QAC7B,IAAI,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,CAAC;YACtC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC3B,SAAQ;QACV,CAAC;QAED,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;YAC9B,gBAAgB,GAAG,EAAE,CAAA;QACvB,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;IACtB,CAAC;IAED,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChC,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;IAChC,CAAC;IACD,OAAO,OAAO,CAAA;AAChB,CAAC;AAED,KAAK,UAAU,WAAW,CACxB,IAAc,EACd,QAAsB,EACtB,WAAwB;IAExB,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;IAC7C,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;YACL,OAAO,EAAE;gBACP,IAAI,EAAE,MAAM;gBACZ,YAAY,EAAE,IAAI,CAAC,EAAE;gBACrB,OAAO,EAAE,iBAAiB,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE;aAC/C;YACD,OAAO,EAAE,IAAI;SACd,CAAA;IACH,CAAC;IACD,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAA;IACnD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,WAAW,CAAC,CAAA;IACpD,MAAM,MAAM,GAAG,8BAA8B,EAAE,CAAA;IAC/C,MAAM,QAAQ,GAAG,qBAAqB,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;IAC9D,MAAM,OAAO,GACX,QAAQ,CAAC,SAAS;QAChB,CAAC,CAAC,GAAG,QAAQ,CAAC,OAAO,qCAAqC;QAC1D,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAA;IACtB,OAAO;QACL,OAAO,EAAE;YACP,IAAI,EAAE,MAAM;YACZ,YAAY,EAAE,IAAI,CAAC,EAAE;YACrB,OAAO,EAAE,OAAO,IAAI,OAAO,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa;SACpE;QACD,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,SAAS,EAAE,MAAM,CAAC,SAAS;KAC5B,CAAA;AACH,CAAC;AAeD,KAAK,UAAU,gBAAgB,CAC7B,KAAiB,EACjB,QAAsB,EACtB,WAAwB,EACxB,cAAsB,EACtB,SAA8B,EAC9B,KAAa,EACb,eAAqE,EACrE,gBAAuE;IAEvE,MAAM,OAAO,GAA0B,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;IAC9D,IAAI,MAAM,GAAG,CAAC,CAAA;IAEd,KAAK,UAAU,MAAM;QACnB,SAAS,CAAC;YACR,MAAM,KAAK,GAAG,MAAM,CAAA;YACpB,MAAM,IAAI,CAAC,CAAA;YACX,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM;gBAAE,OAAM;YACjC,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAE,CAAA;YAC1B,MAAM,YAAY,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,KAAK,GAAG,CAAC,CAAA;YACxD,MAAM,eAAe,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAA;YAC7D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;YAC5B,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,IAAI,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAA;YAC9D,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAA;YACzC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAA;YACxB,MAAM,gBAAgB,EAAE,CAAC;gBACvB,IAAI;gBACJ,KAAK,EAAE,YAAY;gBACnB,KAAK;gBACL,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,UAAU;gBACV,SAAS,EAAE,OAAO,CAAC,SAAS;aAC7B,CAAC,CAAA;QACJ,CAAC;IACH,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAA;IACvE,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;IACtE,OAAO,OAAO,CAAA;AAChB,CAAC;AAWD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,OAAgC;IACrE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAA;IACpD,MAAM,OAAO,GAAG,kBAAkB,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAA;IACvD,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAA;IAC9B,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAA;IAC3C,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAA;IAC7D,MAAM,WAAW,GAAG,IAAI,GAAG,EAAuB,CAAA;IAElD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,SAAQ;QAChC,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAA;QACrE,MAAM,YAAY,GAA0B,IAAI;YAC9C,CAAC,CAAC,MAAM,gBAAgB,CACpB,KAAK,EACL,QAAQ,EACR,WAAW,EACX,OAAO,CAAC,cAAc,EACtB,SAAS,EACT,KAAK,EACL,OAAO,CAAC,eAAe,EACvB,OAAO,CAAC,gBAAgB,CACzB;YACH,CAAC,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE;gBAChB,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAE,CAAA;gBACtB,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,CAAA;gBACzC,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAA;gBACvD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;gBAC5B,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,IAAI,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAA;gBAC9D,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAA;gBACzC,MAAM,OAAO,CAAC,gBAAgB,EAAE,CAAC;oBAC/B,IAAI;oBACJ,KAAK;oBACL,KAAK;oBACL,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,UAAU;oBACV,SAAS,EAAE,OAAO,CAAC,SAAS;iBAC7B,CAAC,CAAA;gBACF,OAAO,CAAC,OAAO,CAAC,CAAA;YAClB,CAAC,CAAC,EAAE,CAAA;QAER,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,YAAY,EAAE,CAAC;YACvC,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;gBACzB,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,CAAA;YAChD,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAC7B,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI;QAC1B,IAAI,EAAE,MAAM;QACZ,YAAY,EAAE,IAAI,CAAC,EAAE;QACrB,OAAO,EAAE,iCAAiC;KAC3C,CACF,CAAC,CAAA;AACJ,CAAC"}

package/dist/runtime/loop.d.ts

@@ -0,0 +1,48 @@
+import type { ChatMessage, LoopState, LoopTerminal, ModelProvider } from '../types.js';
+import type { PermissionStore } from '../tools/types.js';
+import { type ToolCallResultEvent, type ToolCallStartEvent } from './executor.ts';
+import { ToolRegistry } from '../tools/registry.ts';
+export interface RunLoopOptions {
+    provider: ModelProvider;
+    registry: ToolRegistry;
+    systemPrompt: string;
+    userPrompt: string;
+    cwd: string;
+    permissions?: PermissionStore;
+    maxTurns?: number;
+    initialMessages?: ChatMessage[];
+    persistInitialMessages?: boolean;
+    onMessageAppended?: (message: ChatMessage) => Promise<void> | void;
+    onUsage?: (usage: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        cached_tokens?: number | null;
+        provider?: string;
+    }) => Promise<void> | void;
+    onTurnStart?: (event: {
+        turn: number;
+    }) => Promise<void> | void;
+    onAssistantResponse?: (event: {
+        turn: number;
+        finish_reason: string;
+        tool_calls_count: number;
+        content: string | null;
+    }) => Promise<void> | void;
+    onToolCallStart?: (event: ToolCallStartEvent) => Promise<void> | void;
+    onToolCallResult?: (event: ToolCallResultEvent) => Promise<void> | void;
+}
+export interface RunLoopResult {
+    terminal: LoopTerminal;
+    finalText: string;
+    state: LoopState;
+}
+/**
+ * Returns true when the model's response looks like a false start:
+ * it promised to take action but produced no tool calls.
+ *
+ * Deliberately conservative to avoid nudging genuine short completions
+ * ("done", "在", "yes") or complete summaries.
+ */
+export declare function shouldNudge(text: string, state: LoopState): boolean;
+export declare function runLoop(options: RunLoopOptions): Promise<RunLoopResult>;
+//# sourceMappingURL=loop.d.ts.map

package/dist/runtime/loop.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"loop.d.ts","sourceRoot":"","sources":["../../src/runtime/loop.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,aAAa,CAAA;AACtF,OAAO,KAAK,EAAE,eAAe,EAAe,MAAM,mBAAmB,CAAA;AACrE,OAAO,EAAoB,KAAK,mBAAmB,EAAE,KAAK,kBAAkB,EAAE,MAAM,eAAe,CAAA;AAEnG,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAA;AAGnD,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,aAAa,CAAA;IACvB,QAAQ,EAAE,YAAY,CAAA;IACtB,YAAY,EAAE,MAAM,CAAA;IACpB,UAAU,EAAE,MAAM,CAAA;IAClB,GAAG,EAAE,MAAM,CAAA;IACX,WAAW,CAAC,EAAE,eAAe,CAAA;IAC7B,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,eAAe,CAAC,EAAE,WAAW,EAAE,CAAA;IAC/B,sBAAsB,CAAC,EAAE,OAAO,CAAA;IAChC,iBAAiB,CAAC,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAA;IAClE,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE;QAChB,aAAa,EAAE,MAAM,CAAA;QACrB,iBAAiB,EAAE,MAAM,CAAA;QACzB,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;QAC7B,QAAQ,CAAC,EAAE,MAAM,CAAA;KAClB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAA;IAC1B,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAA;IAC/D,mBAAmB,CAAC,EAAE,CAAC,KAAK,EAAE;QAC5B,IAAI,EAAE,MAAM,CAAA;QACZ,aAAa,EAAE,MAAM,CAAA;QACrB,gBAAgB,EAAE,MAAM,CAAA;QACxB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAA;KACvB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAA;IAC1B,eAAe,CAAC,EAAE,CAAC,KAAK,EAAE,kBAAkB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAA;IACrE,gBAAgB,CAAC,EAAE,CAAC,KAAK,EAAE,mBAAmB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAA;CACxE;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,YAAY,CAAA;IACtB,SAAS,EAAE,MAAM,CAAA;IACjB,KAAK,EAAE,SAAS,CAAA;CACjB;AAYD;;;;;;GAMG;AACH,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,GAAG,OAAO,CAUnE;AAmCD,wBAAsB,OAAO,CAAC,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,aAAa,CAAC,CAiJ7E"}

package/dist/runtime/loop.js

@@ -0,0 +1,184 @@
+import { executeToolCalls } from "./executor.js";
+import { withRetry } from "./retry.js";
+import { compactMessages, estimateMessagesChars } from "../context/compact.js";
+// Patterns that signal the model promised action but made no tool calls.
+// Only tested on responses ≥50 chars to avoid flagging conversational short replies.
+const WILL_DO_PATTERNS = [
+    /\bi('ll| will)\s+(start|begin|look|check|read|analyze|examine|fix|help|try|write|create|update|run|search|find)/i,
+    /\blet me\s+(start|begin|look|check|read|analyze|examine|fix|help|try|write|create|update|run|search|find)/i,
+    /\bi('m| am) going to\s+\w/i,
+    /\bfirst,?\s+i('ll| will)\s+\w/i,
+    /\bto (start|begin|proceed),?\s+i('ll| will)\s+\w/i,
+];
+/**
+ * Returns true when the model's response looks like a false start:
+ * it promised to take action but produced no tool calls.
+ *
+ * Deliberately conservative to avoid nudging genuine short completions
+ * ("done", "在", "yes") or complete summaries.
+ */
+export function shouldNudge(text, state) {
+    // Hard cap: never nudge more than twice per session
+    if (state.nudgeCount >= 2)
+        return false;
+    // Very short text is always conversational — never nudge
+    // "在", "done", "ok", "yes", "finished", etc.
+    if (text.trim().length < 50)
+        return false;
+    // Core signal: contains a "will do X" promise without having done X
+    return WILL_DO_PATTERNS.some((p) => p.test(text));
+}
+function createState(systemPrompt, userPrompt, initialMessages) {
+    const messages = initialMessages
+        ? [...initialMessages]
+        : [
+            { role: 'system', content: systemPrompt },
+            { role: 'user', content: userPrompt },
+        ];
+    if (initialMessages && userPrompt.trim() !== '') {
+        messages.push({ role: 'user', content: userPrompt });
+    }
+    return {
+        messages,
+        turnCount: 0,
+        maxOutputTokensRecoveryCount: 0,
+        hasAttemptedReactiveCompact: false,
+        nudgeCount: 0,
+    };
+}
+function parsePositiveInt(raw, fallback) {
+    if (raw === undefined || raw.trim() === '')
+        return fallback;
+    const parsed = Number(raw);
+    if (!Number.isFinite(parsed))
+        return fallback;
+    const value = Math.floor(parsed);
+    return value > 0 ? value : fallback;
+}
+export async function runLoop(options) {
+    const state = createState(options.systemPrompt, options.userPrompt, options.initialMessages);
+    const maxTurns = options.maxTurns ?? 100;
+    let finalText = '';
+    let emptyStopRecoveryCount = 0;
+    let awaitingPostToolSummary = false;
+    const defaultPermissions = { ask: async () => 'allow' };
+    const toolContext = {
+        cwd: options.cwd,
+        permissions: options.permissions ?? defaultPermissions,
+    };
+    if (options.persistInitialMessages !== false) {
+        for (const msg of state.messages) {
+            await options.onMessageAppended?.(msg);
+        }
+    }
+    for (;;) {
+        if (state.turnCount >= maxTurns) {
+            return { terminal: 'max_turns_exceeded', finalText, state };
+        }
+        let assistant;
+        try {
+            const compactTriggerChars = parsePositiveInt(process.env.MERLION_COMPACT_TRIGGER_CHARS, 60_000);
+            const keepRecent = parsePositiveInt(process.env.MERLION_COMPACT_KEEP_RECENT, 10);
+            const chars = estimateMessagesChars(state.messages);
+            if (chars > compactTriggerChars && !state.hasAttemptedReactiveCompact) {
+                const compacted = compactMessages(state.messages, { keepRecent });
+                if (compacted.compacted) {
+                    state.messages = compacted.messages;
+                    state.hasAttemptedReactiveCompact = true;
+                }
+            }
+            await options.onTurnStart?.({ turn: state.turnCount + 1 });
+            assistant = await withRetry(() => options.provider.complete(state.messages, options.registry.getAll()), { maxAttempts: 5, baseDelayMs: 200, maxDelayMs: 32_000 });
+        }
+        catch {
+            return { terminal: 'model_error', finalText, state };
+        }
+        state.turnCount += 1;
+        const assistantMessage = {
+            role: 'assistant',
+            content: assistant.content,
+            tool_calls: assistant.tool_calls,
+        };
+        state.messages.push(assistantMessage);
+        await options.onMessageAppended?.(assistantMessage);
+        await options.onUsage?.(assistant.usage);
+        await options.onAssistantResponse?.({
+            turn: state.turnCount,
+            finish_reason: assistant.finish_reason,
+            tool_calls_count: assistant.tool_calls?.length ?? 0,
+            content: assistant.content
+        });
+        // ── tool_calls ──────────────────────────────────────────────────────────
+        if (assistant.finish_reason === 'tool_calls' &&
+            assistant.tool_calls &&
+            assistant.tool_calls.length > 0) {
+            const maxConcurrency = Number(process.env.MERLION_MAX_TOOL_CONCURRENCY ?? '10');
+            const toolMessages = await executeToolCalls({
+                toolCalls: assistant.tool_calls,
+                registry: options.registry,
+                toolContext,
+                maxConcurrency: Number.isFinite(maxConcurrency) ? Math.max(1, Math.floor(maxConcurrency)) : 10,
+                onToolCallStart: options.onToolCallStart,
+                onToolCallResult: options.onToolCallResult,
+            });
+            for (const toolMsg of toolMessages) {
+                state.messages.push(toolMsg);
+                await options.onMessageAppended?.(toolMsg);
+            }
+            continue;
+        }
+        // ── length (output truncated) ────────────────────────────────────────────
+        if (assistant.finish_reason === 'length' && state.maxOutputTokensRecoveryCount < 3) {
+            state.maxOutputTokensRecoveryCount += 1;
+            const continueMessage = {
+                role: 'user',
+                content: 'Output was cut off. Continue directly from where you stopped. No recap, no apology.',
+            };
+            state.messages.push(continueMessage);
+            await options.onMessageAppended?.(continueMessage);
+            continue;
+        }
+        // ── content_filter ───────────────────────────────────────────────────────
+        if (assistant.finish_reason === 'content_filter') {
+            return { terminal: 'model_error', finalText, state };
+        }
+        // ── stop (and length-recovery exhausted) ────────────────────────────────
+        const text = assistant.content ?? '';
+        const previousMessage = state.messages[state.messages.length - 2];
+        const previousWasTool = previousMessage?.role === 'tool';
+        const shouldRecoverEmptyStop = previousWasTool || awaitingPostToolSummary;
+        if (assistant.finish_reason === 'stop' && text.trim() === '' && shouldRecoverEmptyStop) {
+            if (emptyStopRecoveryCount < 1) {
+                emptyStopRecoveryCount += 1;
+                awaitingPostToolSummary = true;
+                const summaryRequest = {
+                    role: 'user',
+                    content: 'You just finished tool execution. Provide a concise final summary of what changed and any next steps.',
+                };
+                state.messages.push(summaryRequest);
+                await options.onMessageAppended?.(summaryRequest);
+                continue;
+            }
+            finalText = 'Task completed via tool execution, but the model returned no final summary.';
+            return { terminal: 'completed', finalText, state };
+        }
+        if (text.trim() !== '') {
+            awaitingPostToolSummary = false;
+        }
+        // Nudge: model promised action but made no tool call
+        if (shouldNudge(text, state)) {
+            const nudgeMessage = {
+                role: 'user',
+                content: 'Continue with the task. Use your tools to make progress. ' +
+                    'If you have completed everything, describe what was done.',
+            };
+            state.messages.push(nudgeMessage);
+            await options.onMessageAppended?.(nudgeMessage);
+            state.nudgeCount += 1;
+            continue;
+        }
+        finalText = text;
+        return { terminal: 'completed', finalText, state };
+    }
+}
+//# sourceMappingURL=loop.js.map

package/dist/runtime/loop.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loop.js","sourceRoot":"","sources":["../../src/runtime/loop.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAqD,MAAM,eAAe,CAAA;AACnG,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAA;AAEtC,OAAO,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAA;AAoC9E,yEAAyE;AACzE,qFAAqF;AACrF,MAAM,gBAAgB,GAAa;IACjC,kHAAkH;IAClH,4GAA4G;IAC5G,4BAA4B;IAC5B,gCAAgC;IAChC,mDAAmD;CACpD,CAAA;AAED;;;;;;GAMG;AACH,MAAM,UAAU,WAAW,CAAC,IAAY,EAAE,KAAgB;IACxD,oDAAoD;IACpD,IAAI,KAAK,CAAC,UAAU,IAAI,CAAC;QAAE,OAAO,KAAK,CAAA;IAEvC,yDAAyD;IACzD,6CAA6C;IAC7C,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,EAAE;QAAE,OAAO,KAAK,CAAA;IAEzC,oEAAoE;IACpE,OAAO,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;AACnD,CAAC;AAED,SAAS,WAAW,CAClB,YAAoB,EACpB,UAAkB,EAClB,eAA+B;IAE/B,MAAM,QAAQ,GAAkB,eAAe;QAC7C,CAAC,CAAC,CAAC,GAAG,eAAe,CAAC;QACtB,CAAC,CAAC;YACE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE;YACzC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE;SACtC,CAAA;IAEL,IAAI,eAAe,IAAI,UAAU,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QAChD,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAA;IACtD,CAAC;IAED,OAAO;QACL,QAAQ;QACR,SAAS,EAAE,CAAC;QACZ,4BAA4B,EAAE,CAAC;QAC/B,2BAA2B,EAAE,KAAK;QAClC,UAAU,EAAE,CAAC;KACd,CAAA;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,GAAuB,EAAE,QAAgB;IACjE,IAAI,GAAG,KAAK,SAAS,IAAI,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE;QAAE,OAAO,QAAQ,CAAA;IAC3D,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAA;IAC1B,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,QAAQ,CAAA;IAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;IAChC,OAAO,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAA;AACrC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,OAAuB;IACnD,MAAM,KAAK,GAAG,WAAW,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,eAAe,CAAC,CAAA;IAC5F,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,GAAG,CAAA;IACxC,IAAI,SAAS,GAAG,EAAE,CAAA;IAClB,IAAI,sBAAsB,GAAG,CAAC,CAAA;IAC9B,IAAI,uBAAuB,GAAG,KAAK,CAAA;IAEnC,MAAM,kBAAkB,GAAoB,EAAE,GAAG,EAAE,KAAK,IAAI,EAAE,CAAC,OAAO,EAAE,CAAA;IACxE,MAAM,WAAW,GAAgB;QAC/B,GAAG,EAAE,OAAO,CAAC,GAAG;QAChB,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,kBAAkB;KACvD,CAAA;IAED,IAAI,OAAO,CAAC,sBAAsB,KAAK,KAAK,EAAE,CAAC;QAC7C,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC,GAAG,CAAC,CAAA;QACxC,CAAC;IACH,CAAC;IAED,SAAS,CAAC;QACR,IAAI,KAAK,CAAC,SAAS,IAAI,QAAQ,EAAE,CAAC;YAChC,OAAO,EAAE,QAAQ,EAAE,oBAAoB,EAAE,SAAS,EAAE,KAAK,EAAE,CAAA;QAC7D,CAAC;QAED,IAAI,SAAS,CAAA;QACb,IAAI,CAAC;YACH,MAAM,mBAAmB,GAAG,gBAAgB,CAAC,OAAO,CAAC,GAAG,CAAC,6BAA6B,EAAE,MAAM,CAAC,CAAA;YAC/F,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,EAAE,CAAC,CAAA;YAChF,MAAM,KAAK,GAAG,qBAAqB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;YACnD,IAAI,KAAK,GAAG,mBAAmB,IAAI,CAAC,KAAK,CAAC,2BAA2B,EAAE,CAAC;gBACtE,MAAM,SAAS,GAAG,eAAe,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,UAAU,EAAE,CAAC,CAAA;gBACjE,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;oBACxB,KAAK,CAAC,QAAQ,GAAG,SAAS,CAAC,QAAQ,CAAA;oBACnC,KAAK,CAAC,2BAA2B,GAAG,IAAI,CAAA;gBAC1C,CAAC;YACH,CAAC;YACD,MAAM,OAAO,CAAC,WAAW,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,SAAS,GAAG,CAAC,EAAE,CAAC,CAAA;YAC1D,SAAS,GAAG,MAAM,SAAS,CACzB,GAAG,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,EAC1E,EAAE,WAAW,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,EAAE,CACzD,CAAA;QACH,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,SAAS,EAAE,KAAK,EAAE,CAAA;QACtD,CAAC;QAED,KAAK,CAAC,SAAS,IAAI,CAAC,CAAA;QAEpB,MAAM,gBAAgB,GAAgB;YACpC,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,SAAS,CAAC,OAAO;YAC1B,UAAU,EAAE,SAAS,CAAC,UAAU;SACjC,CAAA;QACD,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QACrC,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC,gBAAgB,CAAC,CAAA;QACnD,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,CAAA;QACxC,MAAM,OAAO,CAAC,mBAAmB,EAAE,CAAC;YAClC,IAAI,EAAE,KAAK,CAAC,SAAS;YACrB,aAAa,EAAE,SAAS,CAAC,aAAa;YACtC,gBAAgB,EAAE,SAAS,CAAC,UAAU,EAAE,MAAM,IAAI,CAAC;YACnD,OAAO,EAAE,SAAS,CAAC,OAAO;SAC3B,CAAC,CAAA;QAEF,2EAA2E;QAC3E,IACE,SAAS,CAAC,aAAa,KAAK,YAAY;YACxC,SAAS,CAAC,UAAU;YACpB,SAAS,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAC/B,CAAC;YACD,MAAM,cAAc,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,4BAA4B,IAAI,IAAI,CAAC,CAAA;YAC/E,MAAM,YAAY,GAAG,MAAM,gBAAgB,CAAC;gBAC1C,SAAS,EAAE,SAAS,CAAC,UAAU;gBAC/B,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,WAAW;gBACX,cAAc,EAAE,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;gBAC9F,eAAe,EAAE,OAAO,CAAC,eAAe;gBACxC,gBAAgB,EAAE,OAAO,CAAC,gBAAgB;aAC3C,CAAC,CAAA;YAEF,KAAK,MAAM,OAAO,IAAI,YAAY,EAAE,CAAC;gBACnC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;gBAC5B,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC,OAAO,CAAC,CAAA;YAC5C,CAAC;YACD,SAAQ;QACV,CAAC;QAED,4EAA4E;QAC5E,IAAI,SAAS,CAAC,aAAa,KAAK,QAAQ,IAAI,KAAK,CAAC,4BAA4B,GAAG,CAAC,EAAE,CAAC;YACnF,KAAK,CAAC,4BAA4B,IAAI,CAAC,CAAA;YACvC,MAAM,eAAe,GAAgB;gBACnC,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,qFAAqF;aAC/F,CAAA;YACD,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;YACpC,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC,eAAe,CAAC,CAAA;YAClD,SAAQ;QACV,CAAC;QAED,4EAA4E;QAC5E,IAAI,SAAS,CAAC,aAAa,KAAK,gBAAgB,EAAE,CAAC;YACjD,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,SAAS,EAAE,KAAK,EAAE,CAAA;QACtD,CAAC;QAED,2EAA2E;QAC3E,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,IAAI,EAAE,CAAA;QACpC,MAAM,eAAe,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;QACjE,MAAM,eAAe,GAAG,eAAe,EAAE,IAAI,KAAK,MAAM,CAAA;QACxD,MAAM,sBAAsB,GAAG,eAAe,IAAI,uBAAuB,CAAA;QAEzE,IAAI,SAAS,CAAC,aAAa,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,sBAAsB,EAAE,CAAC;YACvF,IAAI,sBAAsB,GAAG,CAAC,EAAE,CAAC;gBAC/B,sBAAsB,IAAI,CAAC,CAAA;gBAC3B,uBAAuB,GAAG,IAAI,CAAA;gBAC9B,MAAM,cAAc,GAAgB;oBAClC,IAAI,EAAE,MAAM;oBACZ,OAAO,EACL,uGAAuG;iBAC1G,CAAA;gBACD,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;gBACnC,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC,cAAc,CAAC,CAAA;gBACjD,SAAQ;YACV,CAAC;YACD,SAAS,GAAG,6EAA6E,CAAA;YACzF,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,CAAA;QACpD,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YACvB,uBAAuB,GAAG,KAAK,CAAA;QACjC,CAAC;QAED,qDAAqD;QACrD,IAAI,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM,YAAY,GAAgB;gBAChC,IAAI,EAAE,MAAM;gBACZ,OAAO,EACL,2DAA2D;oBAC3D,2DAA2D;aAC9D,CAAA;YACD,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;YACjC,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC,YAAY,CAAC,CAAA;YAC/C,KAAK,CAAC,UAAU,IAAI,CAAC,CAAA;YACrB,SAAQ;QACV,CAAC;QAED,SAAS,GAAG,IAAI,CAAA;QAChB,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,CAAA;IACpD,CAAC;AACH,CAAC"}

package/dist/runtime/retry.d.ts

@@ -0,0 +1,7 @@
+export interface RetryOptions {
+    maxAttempts: number;
+    baseDelayMs: number;
+    maxDelayMs: number;
+}
+export declare function withRetry<T>(fn: () => Promise<T>, options: RetryOptions): Promise<T>;
+//# sourceMappingURL=retry.d.ts.map

package/dist/runtime/retry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"retry.d.ts","sourceRoot":"","sources":["../../src/runtime/retry.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,YAAY;IAC3B,WAAW,EAAE,MAAM,CAAA;IACnB,WAAW,EAAE,MAAM,CAAA;IACnB,UAAU,EAAE,MAAM,CAAA;CACnB;AA4BD,wBAAsB,SAAS,CAAC,CAAC,EAC/B,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EACpB,OAAO,EAAE,YAAY,GACpB,OAAO,CAAC,CAAC,CAAC,CAiBZ"}

package/dist/runtime/retry.js

@@ -0,0 +1,42 @@
+function extractStatusCode(message) {
+    const match = message.match(/\b(400|401|403|429|500|502|503|529)\b/);
+    return match ? Number(match[1]) : null;
+}
+function isRetryableError(error) {
+    const message = String(error);
+    const code = extractStatusCode(message);
+    if (code === null) {
+        return /ECONNRESET|EPIPE|ENOTFOUND|ETIMEDOUT/i.test(message);
+    }
+    return code === 429 || code === 500 || code === 502 || code === 503 || code === 529;
+}
+function isPermanentError(error) {
+    const message = String(error);
+    const code = extractStatusCode(message);
+    return code === 400 || code === 401 || code === 403;
+}
+function delay(attempt, baseDelayMs, maxDelayMs) {
+    const exp = Math.min(maxDelayMs, baseDelayMs * Math.pow(2, Math.max(0, attempt - 1)));
+    const jitter = Math.floor(exp * 0.25 * Math.random());
+    return new Promise((resolve) => setTimeout(resolve, exp + jitter));
+}
+export async function withRetry(fn, options) {
+    let attempt = 0;
+    let lastError = undefined;
+    while (attempt < options.maxAttempts) {
+        attempt += 1;
+        try {
+            return await fn();
+        }
+        catch (error) {
+            lastError = error;
+            if (isPermanentError(error))
+                throw error;
+            if (!isRetryableError(error) || attempt >= options.maxAttempts)
+                throw error;
+            await delay(attempt, options.baseDelayMs, options.maxDelayMs);
+        }
+    }
+    throw lastError ?? new Error('Retry exhausted');
+}
+//# sourceMappingURL=retry.js.map

package/dist/runtime/retry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retry.js","sourceRoot":"","sources":["../../src/runtime/retry.ts"],"names":[],"mappings":"AAMA,SAAS,iBAAiB,CAAC,OAAe;IACxC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAA;IACpE,OAAO,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;AACxC,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAA;IAC7B,MAAM,IAAI,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAA;IACvC,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QAClB,OAAO,uCAAuC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;IAC9D,CAAC;IACD,OAAO,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,CAAA;AACrF,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAA;IAC7B,MAAM,IAAI,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAA;IACvC,OAAO,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,CAAA;AACrD,CAAC;AAED,SAAS,KAAK,CAAC,OAAe,EAAE,WAAmB,EAAE,UAAkB;IACrE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;IACrF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAA;IACrD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,GAAG,MAAM,CAAC,CAAC,CAAA;AACpE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,EAAoB,EACpB,OAAqB;IAErB,IAAI,OAAO,GAAG,CAAC,CAAA;IACf,IAAI,SAAS,GAAY,SAAS,CAAA;IAElC,OAAO,OAAO,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACrC,OAAO,IAAI,CAAC,CAAA;QACZ,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAA;QACnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,SAAS,GAAG,KAAK,CAAA;YACjB,IAAI,gBAAgB,CAAC,KAAK,CAAC;gBAAE,MAAM,KAAK,CAAA;YACxC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,OAAO,IAAI,OAAO,CAAC,WAAW;gBAAE,MAAM,KAAK,CAAA;YAC3E,MAAM,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE,OAAO,CAAC,UAAU,CAAC,CAAA;QAC/D,CAAC;IACH,CAAC;IAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAA;AACjD,CAAC"}

package/dist/runtime/session.d.ts

@@ -0,0 +1,26 @@
+import type { ChatMessage } from '../types.js';
+export interface SessionFiles {
+    sessionId: string;
+    projectHash: string;
+    projectDir: string;
+    transcriptPath: string;
+    usagePath: string;
+}
+export interface UsageEntry {
+    timestamp: string;
+    session_id: string;
+    model: string;
+    provider?: string;
+    prompt_tokens: number;
+    completion_tokens: number;
+    cached_tokens: number | null;
+    tool_schema_tokens_estimate: number;
+}
+export declare function redactSecrets(input: string): string;
+export declare function createSessionFiles(cwd: string): Promise<SessionFiles>;
+export declare function getSessionFilesForResume(cwd: string, sessionId: string): Promise<SessionFiles>;
+export declare function appendSessionMeta(transcriptPath: string, sessionId: string, model: string, projectPath: string): Promise<void>;
+export declare function appendTranscriptMessage(transcriptPath: string, message: ChatMessage): Promise<void>;
+export declare function appendUsage(usagePath: string, entry: UsageEntry): Promise<void>;
+export declare function loadSessionMessages(transcriptPath: string): Promise<ChatMessage[]>;
+//# sourceMappingURL=session.d.ts.map