@geminixiang/mikan 1.0.0-beta.2 → 1.0.0-beta.21

package/dist/harness/runner.d.ts

@@ -17,6 +17,7 @@
 import { Agent, type AgentEvent, type AgentMessage, type AgentTool, type ThinkingLevel } from "@earendil-works/pi-agent-core";
 import { type Api, type ImageContent, type Model } from "@earendil-works/pi-ai";
 import type { ExtensionRegistry } from "./extensions/registry.js";
+import type { RunOrigin } from "./extensions/types.js";
 import type { MikanModels } from "./models.js";
 import { resolveHarnessSettings, type BudgetSettings } from "./settings.js";
 import type { SessionStore } from "./session-store.js";
@@ -56,6 +57,11 @@ export type HarnessEvent = AgentEvent | {
     durationMs: number;
 };
 export type HarnessEventListener = (event: HarnessEvent) => void | Promise<void>;
+/** Outcome of a `prompt()` call that an extension blocked before the model ran. */
+export interface PromptBlockedOutcome {
+    blocked: true;
+    reason?: string;
+}
 export interface MikanAgentSessionOptions {
     systemPrompt: string;
     model: Model<Api>;
@@ -80,26 +86,51 @@ export declare class MikanAgentSession {
     private tally;
     private runBudget;
     private budgetExceededReason;
+    private runOrigin;
     constructor(options: MikanAgentSessionOptions);
     /** Conversation transcript currently held by the agent. */
     get messages(): AgentMessage[];
     get model(): Model<Api>;
+    /** Resource totals and terminal budget state from the most recent prompt. */
+    getLastRunStats(): Readonly<{
+        tokens: number;
+        costUsd: number;
+        llmCalls: number;
+        durationMs: number;
+        budgetExceededReason?: string;
+    }>;
+    /**
+     * Fold spend incurred outside this session's own LLM calls — e.g. subagent
+     * runs launched by a tool during the current prompt — into the run tally
+     * and enforce the run budget at the fold itself: a fold that lands over a
+     * resource ceiling aborts the run now instead of waiting for a next
+     * assistant message that may never come (and would be paid for). Counts
+     * tokens and cost only; external runs are not this session's turns.
+     */
+    foldExternalUsage(usage: {
+        tokens?: number;
+        costUsd?: number;
+    }): Promise<void>;
     subscribe(listener: HarnessEventListener): () => void;
     /** Replace the agent transcript from the persisted session tree. */
     reloadFromSession(): number;
     /**
      * Send a user prompt and run the agent loop to completion, including
      * automatic retries and compaction. Resolves when the turn (and any
-     * recovery continuations) settles.
+     * recovery continuations) settles. Returns a blocked outcome when a
+     * `before_agent_start` extension blocked the turn — the model was never
+     * called and nothing was persisted.
      *
      * @param options.budget Per-run resource ceilings that override the session
      *   defaults. Autonomous (event / trigger) runs should pass a budget so a
      *   runaway loop is stopped even with no human watching.
+     * @param options.origin Platform provenance forwarded to extension hooks.
      */
     prompt(text: string, options?: {
         images?: ImageContent[];
         budget?: BudgetSettings;
-    }): Promise<void>;
+        origin?: RunOrigin;
+    }): Promise<PromptBlockedOutcome | undefined>;
     /** Abort the active run, pending retry backoff, and in-flight compaction. */
     abort(): void;
     private ensureAuthConfigured;
@@ -109,7 +140,11 @@ export declare class MikanAgentSession {
     private recordUsage;
     /** Compare the tally against the run budget; abort the run when a cap is exceeded. */
     private enforceBudget;
+    /** Mark the run over budget, notify listeners and extensions, abort the agent. */
+    private exceedBudget;
     private overBudgetReason;
+    /** The message-independent budget checks, shared with external-spend folds. */
+    private resourceOverBudgetReason;
     private findLastAssistantMessage;
     /** Decide what to do after a run settles. Returns true when the agent should continue. */
     private handlePostRun;

package/dist/harness/runner.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runner.d.ts","sourceRoot":"","sources":["../../src/harness/runner.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AACH,OAAO,EACL,KAAK,EAQL,KAAK,UAAU,EACf,KAAK,YAAY,EACjB,KAAK,SAAS,EAEd,KAAK,aAAa,EACnB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAEL,KAAK,GAAG,EAER,KAAK,YAAY,EACjB,KAAK,KAAK,EACX,MAAM,uBAAuB,CAAC;AAE/B,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,sBAAsB,EAAE,KAAK,cAAc,EAAwB,MAAM,eAAe,CAAC;AAClG,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGvD,MAAM,MAAM,gBAAgB,GAAG,WAAW,GAAG,UAAU,GAAG,QAAQ,CAAC;AAUnE,UAAU,uBAAuB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,MAAM,YAAY,GACpB,UAAU,GACV;IAAE,IAAI,EAAE,kBAAkB,CAAC;IAAC,MAAM,EAAE,gBAAgB,CAAA;CAAE,GACtD;IACE,IAAI,EAAE,gBAAgB,CAAC;IACvB,MAAM,EAAE,gBAAgB,CAAC;IACzB,MAAM,CAAC,EAAE,uBAAuB,CAAC;IACjC,OAAO,EAAE,OAAO,CAAC;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,GACD;IACE,IAAI,EAAE,kBAAkB,CAAC;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;CACtB,GACD;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,OAAO,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,GAClF;IACE,IAAI,EAAE,iBAAiB,CAAC;IACxB,6DAA6D;IAC7D,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAEN,MAAM,MAAM,oBAAoB,GAAG,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAyBjF,MAAM,WAAW,wBAAwB;IACvC,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IAClB,aAAa,EAAE,aAAa,CAAC;IAC7B,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,MAAM,EAAE,WAAW,CAAC;IACpB,YAAY,EAAE,YAAY,CAAC;IAC3B,QAAQ,CAAC,EAAE,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,UAAU,CAAC,EAAE,iBAAiB,CAAC;CAChC;AAED,qBAAa,iBAAiB;IAC5B,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC;IACtB,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;IACpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAc;IACrC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAkB;IAC3C,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAgC;IAC3D,OAAO,CAAC,SAAS,CAAmC;IACpD,OAAO,CAAC,YAAY,CAAK;IACzB,OAAO,CAAC,yBAAyB,CAAS;IAC1C,OAAO,CAAC,oBAAoB,CAA8B;IAC1D,OAAO,CAAC,yBAAyB,CAA8B;IAC/D,OAAO,CAAC,KAAK,CAAkE;IAC/E,OAAO,CAAC,SAAS,CAAsB;IACvC,OAAO,CAAC,oBAAoB,CAAqB;IAEjD,YAAY,OAAO,EAAE,wBAAwB,EA2C5C;IAED,2DAA2D;IAC3D,IAAI,QAAQ,IAAI,YAAY,EAAE,CAE7B;IAED,IAAI,KAAK,IAAI,KAAK,CAAC,GAAG,CAAC,CAEtB;IAED,SAAS,CAAC,QAAQ,EAAE,oBAAoB,GAAG,MAAM,IAAI,CAGpD;IAED,oEAAoE;IACpE,iBAAiB,IAAI,MAAM,CAM1B;IAED;;;;;;;;OAQG;IACG,MAAM,CACV,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,YAAY,EAAE,CAAC;QAAC,MAAM,CAAC,EAAE,cAAc,CAAA;KAAE,GAC7D,OAAO,CAAC,IAAI,CAAC,CAgDf;IAED,6EAA6E;IAC7E,KAAK,IAAI,IAAI,CAIZ;YAEa,oBAAoB;YASpB,IAAI;YAaJ,gBAAgB;IAmC9B,sEAAsE;IACtE,OAAO,CAAC,WAAW;IAQnB,sFAAsF;YACxE,aAAa;IAkB3B,OAAO,CAAC,gBAAgB;IAiBxB,OAAO,CAAC,wBAAwB;IAShC,0FAA0F;YAC5E,aAAa;YAsBb,cAAc;YAgBd,wBAAwB;YAiBxB,aAAa;IAmF3B,OAAO,CAAC,gBAAgB;YAKV,YAAY;IA+C1B,OAAO,CAAC,wBAAwB;CAMjC"}
+{"version":3,"file":"runner.d.ts","sourceRoot":"","sources":["../../src/harness/runner.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AACH,OAAO,EACL,KAAK,EAQL,KAAK,UAAU,EACf,KAAK,YAAY,EACjB,KAAK,SAAS,EAEd,KAAK,aAAa,EACnB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAEL,KAAK,GAAG,EAER,KAAK,YAAY,EACjB,KAAK,KAAK,EACX,MAAM,uBAAuB,CAAC;AAE/B,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,sBAAsB,EAAE,KAAK,cAAc,EAAwB,MAAM,eAAe,CAAC;AAClG,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGvD,MAAM,MAAM,gBAAgB,GAAG,WAAW,GAAG,UAAU,GAAG,QAAQ,CAAC;AAUnE,UAAU,uBAAuB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,MAAM,YAAY,GACpB,UAAU,GACV;IAAE,IAAI,EAAE,kBAAkB,CAAC;IAAC,MAAM,EAAE,gBAAgB,CAAA;CAAE,GACtD;IACE,IAAI,EAAE,gBAAgB,CAAC;IACvB,MAAM,EAAE,gBAAgB,CAAC;IACzB,MAAM,CAAC,EAAE,uBAAuB,CAAC;IACjC,OAAO,EAAE,OAAO,CAAC;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,GACD;IACE,IAAI,EAAE,kBAAkB,CAAC;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;CACtB,GACD;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,OAAO,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,GAClF;IACE,IAAI,EAAE,iBAAiB,CAAC;IACxB,6DAA6D;IAC7D,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAEN,MAAM,MAAM,oBAAoB,GAAG,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAEjF,mFAAmF;AACnF,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,IAAI,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAyBD,MAAM,WAAW,wBAAwB;IACvC,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IAClB,aAAa,EAAE,aAAa,CAAC;IAC7B,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,MAAM,EAAE,WAAW,CAAC;IACpB,YAAY,EAAE,YAAY,CAAC;IAC3B,QAAQ,CAAC,EAAE,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,UAAU,CAAC,EAAE,iBAAiB,CAAC;CAChC;AAED,qBAAa,iBAAiB;IAC5B,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC;IACtB,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;IACpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAc;IACrC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAkB;IAC3C,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAgC;IAC3D,OAAO,CAAC,SAAS,CAAmC;IACpD,OAAO,CAAC,YAAY,CAAK;IACzB,OAAO,CAAC,yBAAyB,CAAS;IAC1C,OAAO,CAAC,oBAAoB,CAA8B;IAC1D,OAAO,CAAC,yBAAyB,CAA8B;IAC/D,OAAO,CAAC,KAAK,CAAkE;IAC/E,OAAO,CAAC,SAAS,CAAsB;IACvC,OAAO,CAAC,oBAAoB,CAAqB;IACjD,OAAO,CAAC,SAAS,CAAwB;IAEzC,YAAY,OAAO,EAAE,wBAAwB,EA8C5C;IAED,2DAA2D;IAC3D,IAAI,QAAQ,IAAI,YAAY,EAAE,CAE7B;IAED,IAAI,KAAK,IAAI,KAAK,CAAC,GAAG,CAAC,CAEtB;IAED,6EAA6E;IAC7E,eAAe,IAAI,QAAQ,CAAC;QAC1B,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,oBAAoB,CAAC,EAAE,MAAM,CAAC;KAC/B,CAAC,CAQD;IAED;;;;;;;OAOG;IACG,iBAAiB,CAAC,KAAK,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAMnF;IAED,SAAS,CAAC,QAAQ,EAAE,oBAAoB,GAAG,MAAM,IAAI,CAGpD;IAED,oEAAoE;IACpE,iBAAiB,IAAI,MAAM,CAM1B;IAED;;;;;;;;;;;OAWG;IACG,MAAM,CACV,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,YAAY,EAAE,CAAC;QAAC,MAAM,CAAC,EAAE,cAAc,CAAC;QAAC,MAAM,CAAC,EAAE,SAAS,CAAA;KAAE,GACjF,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC,CAmF3C;IAED,6EAA6E;IAC7E,KAAK,IAAI,IAAI,CAIZ;YAEa,oBAAoB;YASpB,IAAI;YAaJ,gBAAgB;IAmC9B,sEAAsE;IACtE,OAAO,CAAC,WAAW;IAQnB,sFAAsF;YACxE,aAAa;IAO3B,kFAAkF;YACpE,YAAY;IAwB1B,OAAO,CAAC,gBAAgB;IAUxB,+EAA+E;IAC/E,OAAO,CAAC,wBAAwB;IAchC,OAAO,CAAC,wBAAwB;IAShC,0FAA0F;YAC5E,aAAa;YAsBb,cAAc;YAgBd,wBAAwB;YAiBxB,aAAa;IAmF3B,OAAO,CAAC,gBAAgB;YAKV,YAAY;IA+C1B,OAAO,CAAC,wBAAwB;CAMjC"}

package/dist/harness/runner.js

@@ -66,20 +66,24 @@ export class MikanAgentSession {
                     toolCallId: toolCall.id,
                     toolName: toolCall.name,
                     args,
+                    origin: this.runOrigin,
                 });
                 return result ?? undefined;
             },
             afterToolCall: async ({ toolCall, args, result, isError }) => {
-                if (this.extensions?.hasHandlers("tool_result")) {
-                    await this.extensions.emit("tool_result", {
-                        toolCallId: toolCall.id,
-                        toolName: toolCall.name,
-                        args,
-                        content: result.content,
-                        isError,
-                    });
-                }
-                return undefined;
+                if (!this.extensions?.hasHandlers("tool_result"))
+                    return undefined;
+                // Chained extension rewrites (redaction, truncation) become a
+                // pi-agent-core AfterToolCallResult override; undefined keeps the
+                // executed result untouched.
+                return this.extensions.emitToolResult({
+                    toolCallId: toolCall.id,
+                    toolName: toolCall.name,
+                    args,
+                    content: result.content,
+                    isError,
+                    origin: this.runOrigin,
+                });
             },
         });
         this.agent.subscribe(async (event) => {
@@ -93,6 +97,33 @@ export class MikanAgentSession {
     get model() {
         return this.agent.state.model;
     }
+    /** Resource totals and terminal budget state from the most recent prompt. */
+    getLastRunStats() {
+        return {
+            tokens: this.tally.tokens,
+            costUsd: this.tally.costUsd,
+            llmCalls: this.tally.llmCalls,
+            durationMs: this.tally.startedAt > 0 ? Date.now() - this.tally.startedAt : 0,
+            ...(this.budgetExceededReason ? { budgetExceededReason: this.budgetExceededReason } : {}),
+        };
+    }
+    /**
+     * Fold spend incurred outside this session's own LLM calls — e.g. subagent
+     * runs launched by a tool during the current prompt — into the run tally
+     * and enforce the run budget at the fold itself: a fold that lands over a
+     * resource ceiling aborts the run now instead of waiting for a next
+     * assistant message that may never come (and would be paid for). Counts
+     * tokens and cost only; external runs are not this session's turns.
+     */
+    async foldExternalUsage(usage) {
+        this.tally.tokens += usage.tokens ?? 0;
+        this.tally.costUsd += usage.costUsd ?? 0;
+        if (this.budgetExceededReason || !this.agent.state.isStreaming)
+            return;
+        const reason = this.resourceOverBudgetReason();
+        if (reason)
+            await this.exceedBudget(reason);
+    }
     subscribe(listener) {
         this.listeners.add(listener);
         return () => this.listeners.delete(listener);
@@ -108,11 +139,14 @@ export class MikanAgentSession {
     /**
      * Send a user prompt and run the agent loop to completion, including
      * automatic retries and compaction. Resolves when the turn (and any
-     * recovery continuations) settles.
+     * recovery continuations) settles. Returns a blocked outcome when a
+     * `before_agent_start` extension blocked the turn — the model was never
+     * called and nothing was persisted.
      *
      * @param options.budget Per-run resource ceilings that override the session
      *   defaults. Autonomous (event / trigger) runs should pass a budget so a
      *   runaway loop is stopped even with no human watching.
+     * @param options.origin Platform provenance forwarded to extension hooks.
      */
     async prompt(text, options) {
         if (this.agent.state.isStreaming) {
@@ -125,36 +159,66 @@ export class MikanAgentSession {
         this.runBudget = { ...this.settings.budget, ...options?.budget };
         this.budgetExceededReason = undefined;
         this.tally = { tokens: 0, costUsd: 0, llmCalls: 0, startedAt: Date.now() };
+        this.runOrigin = options?.origin;
         // A previous turn may have ended over the threshold (for example after an
         // abort); compact before adding new context on top.
         const lastAssistant = this.findLastAssistantMessage();
         if (lastAssistant && lastAssistant.stopReason !== "error") {
             await this.checkThresholdCompaction(lastAssistant);
         }
+        let promptText = text;
         let systemPrompt = this.agent.state.systemPrompt;
         if (this.extensions?.hasHandlers("before_agent_start")) {
-            const result = await this.extensions.emit("before_agent_start", {
-                prompt: text,
+            const result = await this.extensions.emitBeforeAgentStart({
+                prompt: promptText,
                 images: options?.images,
                 systemPrompt,
+                origin: this.runOrigin,
             });
-            if (result?.systemPrompt) {
+            if (result?.block) {
+                // Blocked before agent.prompt(): the user message never enters the
+                // transcript or session store, so the blocked turn leaves no trace.
+                log.logInfo(`Extension blocked turn${result.reason ? `: ${result.reason}` : ""}`);
+                return { blocked: true, reason: result.reason };
+            }
+            if (result?.prompt !== undefined && result.prompt !== promptText) {
+                log.logInfo(`Extension rewrote user prompt: ${promptText.length} → ${result.prompt.length} chars`);
+                promptText = result.prompt;
+            }
+            if (result?.systemPrompt && result.systemPrompt !== systemPrompt) {
+                // An extension rewrote the system prompt for this turn. Log the new
+                // size so the diagnostic in agent.ts (which logs the pre-hook prompt)
+                // isn't mistaken for the final prompt sent to the model.
+                log.logInfo(`Extension rewrote system prompt: ${systemPrompt.length} → ${result.systemPrompt.length} chars`);
                 systemPrompt = result.systemPrompt;
                 this.agent.state.systemPrompt = systemPrompt;
             }
         }
         const userMessage = {
             role: "user",
-            content: [{ type: "text", text }, ...(options?.images ?? [])],
+            content: [{ type: "text", text: promptText }, ...(options?.images ?? [])],
             timestamp: Date.now(),
         };
         await this.agent.prompt([userMessage]);
         while (await this.handlePostRun()) {
             await this.agent.continue();
         }
+        // The turn settled on an error (retries exhausted or not retryable):
+        // give monitoring extensions the final failure, once per turn.
+        const settled = this.findLastAssistantMessage();
+        if (settled?.stopReason === "error" && this.extensions?.hasHandlers("agent_error")) {
+            await this.extensions.emit("agent_error", {
+                errorMessage: settled.errorMessage || "Unknown error",
+                origin: this.runOrigin,
+            });
+        }
         if (this.extensions?.hasHandlers("turn_end")) {
-            await this.extensions.emit("turn_end", { messages: this.agent.state.messages });
+            await this.extensions.emit("turn_end", {
+                messages: this.agent.state.messages,
+                origin: this.runOrigin,
+            });
         }
+        return undefined;
     }
     /** Abort the active run, pending retry backoff, and in-flight compaction. */
     abort() {
@@ -191,11 +255,11 @@ export class MikanAgentSession {
             this.sessionStore.appendCustomMessageEntry(custom.customType, custom.content, custom.display, custom.details);
         }
         if (this.extensions?.hasHandlers("message_end")) {
-            await this.extensions.emit("message_end", { message });
+            await this.extensions.emit("message_end", { message, origin: this.runOrigin });
         }
         if (message.role === "assistant") {
             this.recordUsage(message);
-            await this.enforceBudget();
+            await this.enforceBudget(message);
             if (message.stopReason !== "error") {
                 this.overflowRecoveryAttempted = false;
                 if (this.retryAttempt > 0) {
@@ -215,12 +279,16 @@ export class MikanAgentSession {
         this.tally.costUsd += usage.cost?.total ?? 0;
     }
     /** Compare the tally against the run budget; abort the run when a cap is exceeded. */
-    async enforceBudget() {
+    async enforceBudget(message) {
         if (this.budgetExceededReason)
             return;
-        const reason = this.overBudgetReason();
+        const reason = this.overBudgetReason(message);
         if (!reason)
             return;
+        await this.exceedBudget(reason);
+    }
+    /** Mark the run over budget, notify listeners and extensions, abort the agent. */
+    async exceedBudget(reason) {
         this.budgetExceededReason = reason;
         await this.emit({
             type: "budget_exceeded",
@@ -230,14 +298,30 @@ export class MikanAgentSession {
             llmCalls: this.tally.llmCalls,
             durationMs: Date.now() - this.tally.startedAt,
         });
+        if (this.extensions?.hasHandlers("budget_exceeded")) {
+            await this.extensions.emit("budget_exceeded", {
+                reason,
+                tokens: this.tally.tokens,
+                costUsd: this.tally.costUsd,
+                llmCalls: this.tally.llmCalls,
+                durationMs: Date.now() - this.tally.startedAt,
+                origin: this.runOrigin,
+            });
+        }
         log.logWarning("Run budget exceeded — aborting", reason);
         this.agent.abort();
     }
-    overBudgetReason() {
-        const { maxTokens, maxCostUsd, maxDurationMs, maxLlmCalls } = this.runBudget;
-        if (maxLlmCalls !== undefined && this.tally.llmCalls >= maxLlmCalls) {
+    overBudgetReason(message) {
+        const { maxLlmCalls } = this.runBudget;
+        const needsAnotherCall = message.stopReason === "error" || message.content.some((part) => part.type === "toolCall");
+        if (needsAnotherCall && maxLlmCalls !== undefined && this.tally.llmCalls >= maxLlmCalls) {
             return `${this.tally.llmCalls} LLM calls >= ${maxLlmCalls} limit`;
         }
+        return this.resourceOverBudgetReason();
+    }
+    /** The message-independent budget checks, shared with external-spend folds. */
+    resourceOverBudgetReason() {
+        const { maxTokens, maxCostUsd, maxDurationMs } = this.runBudget;
         if (maxTokens !== undefined && this.tally.tokens >= maxTokens) {
             return `${this.tally.tokens} tokens >= ${maxTokens} limit`;
         }

package/dist/harness/runner.js.map

@@ -1 +1 @@
-{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/harness/runner.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AACH,OAAO,EACL,KAAK,EACL,sBAAsB,EACtB,OAAO,EACP,YAAY,EACZ,qBAAqB,EACrB,UAAU,EACV,iBAAiB,EACjB,aAAa,GAMd,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,iBAAiB,GAKlB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,KAAK,GAAG,MAAM,WAAW,CAAC;AAGjC,OAAO,EAAE,sBAAsB,EAA6C,MAAM,eAAe,CAAC;AAkDlG,6EAA6E;AAC7E,kDAAkD;AAClD,MAAM,uBAAuB,GAC3B,wcAAwc,CAAC;AAE3c,SAAS,KAAK,CAAC,EAAU,EAAE,MAAmB;IAC5C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC;YAC7B,OAAO;QACT,CAAC;QACD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC7C,OAAO,EAAE,CAAC;QACZ,CAAC,EAAE,EAAE,CAAC,CAAC;QACP,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC;QAC/B,CAAC,CAAC;QACF,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC;AAaD,MAAM,OAAO,iBAAiB;IAe5B,YAAY,OAAiC;QATrC,cAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;QAC5C,iBAAY,GAAG,CAAC,CAAC;QACjB,8BAAyB,GAAG,KAAK,CAAC;QAGlC,UAAK,GAAa,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;QACvE,cAAS,GAAmB,EAAE,CAAC;QAIrC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,QAAQ,GAAG,sBAAsB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QAErC,MAAM,KAAK,GAAG,CAAC,GAAG,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,OAAO,CAAC,UAAU,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC;YACrB,YAAY,EAAE;gBACZ,YAAY,EAAE,OAAO,CAAC,YAAY;gBAClC,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,aAAa,EAAE,OAAO,CAAC,aAAa;gBACpC,KAAK;aACN;YACD,YAAY;YACZ,QAAQ,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE,CAC1C,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,aAAa,CAAC;YAChE,cAAc,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE;gBAC3C,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,WAAW,CAAC;oBAAE,OAAO,SAAS,CAAC;gBACjE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,EAAE;oBACrD,UAAU,EAAE,QAAQ,CAAC,EAAE;oBACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,IAAI;iBACL,CAAC,CAAC;gBACH,OAAO,MAAM,IAAI,SAAS,CAAC;YAC7B,CAAC;YACD,aAAa,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE;gBAC3D,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,aAAa,CAAC,EAAE,CAAC;oBAChD,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,EAAE;wBACxC,UAAU,EAAE,QAAQ,CAAC,EAAE;wBACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;wBACvB,IAAI;wBACJ,OAAO,EAAE,MAAM,CAAC,OAAO;wBACvB,OAAO;qBACR,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,SAAS,CAAC;YACnB,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACnC,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,2DAA2D;IAC3D,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC;IACnC,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC;IAChC,CAAC;IAED,SAAS,CAAC,QAA8B;QACtC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7B,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED,oEAAoE;IACpE,iBAAiB;QACf,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,CAAC;QACxD,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAC/C,CAAC;QACD,OAAO,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;IACjC,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,MAAM,CACV,IAAY,EACZ,OAA8D;QAE9D,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC;QACrC,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAEvC,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;QACtB,IAAI,CAAC,yBAAyB,GAAG,KAAK,CAAC;QACvC,IAAI,CAAC,SAAS,GAAG,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,CAAC;QACjE,IAAI,CAAC,oBAAoB,GAAG,SAAS,CAAC;QACtC,IAAI,CAAC,KAAK,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QAE3E,0EAA0E;QAC1E,oDAAoD;QACpD,MAAM,aAAa,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QACtD,IAAI,aAAa,IAAI,aAAa,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;YAC1D,MAAM,IAAI,CAAC,wBAAwB,CAAC,aAAa,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;QACjD,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,oBAAoB,CAAC,EAAE,CAAC;YACvD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,oBAAoB,EAAE;gBAC9D,MAAM,EAAE,IAAI;gBACZ,MAAM,EAAE,OAAO,EAAE,MAAM;gBACvB,YAAY;aACb,CAAC,CAAC;YACH,IAAI,MAAM,EAAE,YAAY,EAAE,CAAC;gBACzB,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;gBACnC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,GAAG,YAAY,CAAC;YAC/C,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAAiB;YAChC,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,GAAG,CAAC,OAAO,EAAE,MAAM,IAAI,EAAE,CAAC,CAAC;YAC7D,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;QACvC,OAAO,MAAM,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;YAClC,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC9B,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC;YAC7C,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;QAClF,CAAC;IACH,CAAC;IAED,6EAA6E;IAC7E,KAAK;QACH,IAAI,CAAC,oBAAoB,EAAE,KAAK,EAAE,CAAC;QACnC,IAAI,CAAC,yBAAyB,EAAE,KAAK,EAAE,CAAC;QACxC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAAC,KAAiB;QAClD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC9C,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CACb,gCAAgC,KAAK,CAAC,QAAQ,0EAA0E,CACzH,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,IAAI,CAAC,KAAmB;QACpC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,IAAI,CAAC;gBACH,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,GAAG,CAAC,UAAU,CACZ,+BAA+B,EAC/B,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CACjD,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,KAAiB;QAC9C,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEvB,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa;YAAE,OAAO;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;QAE9B,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC7F,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAC3C,CAAC;aAAM,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACrC,MAAM,MAAM,GAAG,OAAwB,CAAC;YACxC,IAAI,CAAC,YAAY,CAAC,wBAAwB,CACxC,MAAM,CAAC,UAAU,EACjB,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,OAAO,CACf,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,aAAa,CAAC,EAAE,CAAC;YAChD,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACjC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAC1B,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;YAC3B,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;gBACnC,IAAI,CAAC,yBAAyB,GAAG,KAAK,CAAC;gBACvC,IAAI,IAAI,CAAC,YAAY,GAAG,CAAC,EAAE,CAAC;oBAC1B,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;oBACvF,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,sEAAsE;IAC9D,WAAW,CAAC,OAAyB;QAC3C,IAAI,CAAC,KAAK,CAAC,QAAQ,IAAI,CAAC,CAAC;QACzB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC5B,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC;QACrF,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,IAAI,CAAC,CAAC;IAC/C,CAAC;IAED,sFAAsF;IAC9E,KAAK,CAAC,aAAa;QACzB,IAAI,IAAI,CAAC,oBAAoB;YAAE,OAAO;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,IAAI,CAAC,MAAM;YAAE,OAAO;QAEpB,IAAI,CAAC,oBAAoB,GAAG,MAAM,CAAC;QACnC,MAAM,IAAI,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,iBAAiB;YACvB,MAAM;YACN,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM;YACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO;YAC3B,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ;YAC7B,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS;SAC9C,CAAC,CAAC;QACH,GAAG,CAAC,UAAU,CAAC,gCAAgC,EAAE,MAAM,CAAC,CAAC;QACzD,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAEO,gBAAgB;QACtB,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,aAAa,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;QAC7E,IAAI,WAAW,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,IAAI,WAAW,EAAE,CAAC;YACpE,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,iBAAiB,WAAW,QAAQ,CAAC;QACpE,CAAC;QACD,IAAI,SAAS,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;YAC9D,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,cAAc,SAAS,QAAQ,CAAC;QAC7D,CAAC;QACD,IAAI,UAAU,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,UAAU,EAAE,CAAC;YACjE,OAAO,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,UAAU,YAAY,CAAC;QAChF,CAAC;QACD,IAAI,aAAa,KAAK,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,IAAI,aAAa,EAAE,CAAC;YACtF,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,SAAS,aAAa,UAAU,CAAC;QAC9E,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,wBAAwB;QAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC3C,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9C,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC5B,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;gBAAE,OAAO,OAAO,CAAC;QACnD,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,0FAA0F;IAClF,KAAK,CAAC,aAAa;QACzB,4EAA4E;QAC5E,qEAAqE;QACrE,IAAI,IAAI,CAAC,oBAAoB;YAAE,OAAO,KAAK,CAAC;QAE5C,MAAM,aAAa,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QACtD,IAAI,CAAC,aAAa;YAAE,OAAO,KAAK,CAAC;QAEjC,IAAI,aAAa,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;YACzC,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;YAChE,IAAI,iBAAiB,CAAC,aAAa,EAAE,aAAa,CAAC,EAAE,CAAC;gBACpD,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC;YAC/B,CAAC;YACD,IAAI,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,EAAE,CAAC;gBACzC,OAAO,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YAC1C,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,IAAI,CAAC,wBAAwB,CAAC,aAAa,CAAC,CAAC;IACtD,CAAC;IAEO,KAAK,CAAC,cAAc;QAC1B,IAAI,IAAI,CAAC,yBAAyB,EAAE,CAAC;YACnC,MAAM,IAAI,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,gBAAgB;gBACtB,MAAM,EAAE,UAAU;gBAClB,OAAO,EAAE,KAAK;gBACd,YAAY,EACV,2GAA2G;aAC9G,CAAC,CAAC;YACH,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,CAAC,yBAAyB,GAAG,IAAI,CAAC;QACtC,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAChC,OAAO,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;IAC9C,CAAC;IAEO,KAAK,CAAC,wBAAwB,CAAC,aAA+B;QACpE,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;QAC1C,IAAI,CAAC,QAAQ,CAAC,OAAO;YAAE,OAAO,KAAK,CAAC;QACpC,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;QAChE,IAAI,aAAa,IAAI,CAAC;YAAE,OAAO,KAAK,CAAC;QAErC,IAAI,aAAa,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,sBAAsB,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1F,IAAI,aAAa,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,QAAQ,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAClE,IAAI,QAAQ,CAAC,cAAc,KAAK,IAAI;gBAAE,OAAO,KAAK,CAAC;YACnD,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC;QAClC,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,aAAa,EAAE,aAAa,EAAE,QAAQ,CAAC;YAAE,OAAO,KAAK,CAAC;QACzE,OAAO,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,MAAwB,EAAE,SAAkB;QACtE,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;YAClD,MAAM,WAAW,GAAG,UAAU,CAAC,iBAAiB,CAAC,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC;YACzF,IAAI,CAAC,WAAW;gBAAE,OAAO,KAAK,CAAC;YAE/B,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,MAAM,EAAE,CAAC,CAAC;YACtD,OAAO,GAAG,IAAI,CAAC;YACf,IAAI,CAAC,yBAAyB,GAAG,IAAI,eAAe,EAAE,CAAC;YACvD,MAAM,MAAM,GAAG,IAAI,CAAC,yBAAyB,CAAC,MAAM,CAAC;YAErD,MAAM,MAAM,GAAG,UAAU,CACvB,MAAM,OAAO,CACX,WAAW,EACX,IAAI,CAAC,MAAM,CAAC,MAAM,EAClB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EACtB,SAAS,EACT,MAAM,EACN,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,CAC/B,CACF,CAAC;YACF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnE,OAAO,KAAK,CAAC;YACf,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAChD,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,gBAAgB,EACvB,MAAM,CAAC,YAAY,EACnB,MAAM,CAAC,OAAO,CACf,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,CAAC;YACxD,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;YAE7C,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACpD,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAClD,IAAI,KAAK,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;oBACjC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,iBAAiB,EAAE;wBAC5C,KAAK,EAAE,KAAwB;wBAC/B,MAAM;qBACP,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,MAAM,IAAI,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,gBAAgB;gBACtB,MAAM;gBACN,MAAM,EAAE;oBACN,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;oBACzC,YAAY,EAAE,MAAM,CAAC,YAAY;iBAClC;gBACD,OAAO,EAAE,KAAK;aACf,CAAC,CAAC;YAEH,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,CAAC,wBAAwB,EAAE,CAAC;gBAChC,OAAO,IAAI,CAAC;YACd,CAAC;YACD,OAAO,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC;QACxC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,mBAAmB,CAAC;YAClF,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,IAAI,CAAC,IAAI,CAAC;oBACd,IAAI,EAAE,gBAAgB;oBACtB,MAAM;oBACN,OAAO,EAAE,KAAK;oBACd,YAAY,EACV,MAAM,KAAK,UAAU;wBACnB,CAAC,CAAC,qCAAqC,YAAY,EAAE;wBACrD,CAAC,CAAC,2BAA2B,YAAY,EAAE;iBAChD,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,UAAU,CAAC,+BAA+B,EAAE,YAAY,CAAC,CAAC;YAChE,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,yBAAyB,GAAG,SAAS,CAAC;QAC7C,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,OAAyB;QAChD,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY;YAAE,OAAO,KAAK,CAAC;QAC1E,OAAO,uBAAuB,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAC5D,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,OAAyB;QAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QACrC,IAAI,CAAC,QAAQ,CAAC,OAAO;YAAE,OAAO,KAAK,CAAC;QAEpC,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,IAAI,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC5C,IAAI,CAAC,YAAY,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,IAAI,CAAC,YAAY;gBAC1B,UAAU,EAAE,OAAO,CAAC,YAAY;aACjC,CAAC,CAAC;YACH,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;YACtB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,WAAW,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC;QACpE,MAAM,IAAI,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,kBAAkB;YACxB,OAAO,EAAE,IAAI,CAAC,YAAY;YAC1B,WAAW,EAAE,QAAQ,CAAC,UAAU;YAChC,OAAO;YACP,YAAY,EAAE,OAAO,CAAC,YAAY,IAAI,eAAe;SACtD,CAAC,CAAC;QAEH,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAEhC,IAAI,CAAC,oBAAoB,GAAG,IAAI,eAAe,EAAE,CAAC;QAClD,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QACzD,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC;YAClC,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;YACtB,MAAM,IAAI,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,KAAK;gBACd,OAAO;gBACP,UAAU,EAAE,iBAAiB;aAC9B,CAAC,CAAC;YACH,OAAO,KAAK,CAAC;QACf,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,oBAAoB,GAAG,SAAS,CAAC;QACxC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,wBAAwB;QAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC3C,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9E,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;CACF","sourcesContent":["/**\n * MikanAgentSession — mikan's agent run loop.\n *\n * Owns a pi-agent-core `Agent` and layers on the behaviors mikan needs for\n * long-lived chat conversations:\n *\n * - message persistence into the conversation's {@link SessionStore}\n * - automatic context compaction (threshold and overflow recovery), using\n *   pi-agent-core's compaction pipeline over the session tree\n * - automatic retry with exponential backoff on transient provider errors\n * - extension hook dispatch ({@link ExtensionRegistry})\n *\n * Events mirror the agent's lifecycle events plus `compaction_start/_end`\n * and `auto_retry_start/_end`, preserving the event surface mikan's\n * platform adapters already render.\n */\nimport {\n  Agent,\n  calculateContextTokens,\n  compact,\n  convertToLlm,\n  estimateContextTokens,\n  getOrThrow,\n  prepareCompaction,\n  shouldCompact,\n  type AgentEvent,\n  type AgentMessage,\n  type AgentTool,\n  type CustomMessage,\n  type ThinkingLevel,\n} from \"@earendil-works/pi-agent-core\";\nimport {\n  isContextOverflow,\n  type Api,\n  type AssistantMessage,\n  type ImageContent,\n  type Model,\n} from \"@earendil-works/pi-ai\";\nimport * as log from \"../log.js\";\nimport type { ExtensionRegistry } from \"./extensions/registry.js\";\nimport type { MikanModels } from \"./models.js\";\nimport { resolveHarnessSettings, type BudgetSettings, type HarnessSettings } from \"./settings.js\";\nimport type { SessionStore } from \"./session-store.js\";\nimport type { CompactionEntry } from \"./types.js\";\n\nexport type CompactionReason = \"threshold\" | \"overflow\" | \"manual\";\n\n/** Running resource tally for the current `prompt()` call, matched against the budget. */\ninterface RunTally {\n  tokens: number;\n  costUsd: number;\n  llmCalls: number;\n  startedAt: number;\n}\n\ninterface CompactionResultSummary {\n  summary: string;\n  firstKeptEntryId: string;\n  tokensBefore: number;\n}\n\nexport type HarnessEvent =\n  | AgentEvent\n  | { type: \"compaction_start\"; reason: CompactionReason }\n  | {\n      type: \"compaction_end\";\n      reason: CompactionReason;\n      result?: CompactionResultSummary;\n      aborted: boolean;\n      errorMessage?: string;\n    }\n  | {\n      type: \"auto_retry_start\";\n      attempt: number;\n      maxAttempts: number;\n      delayMs: number;\n      errorMessage: string;\n    }\n  | { type: \"auto_retry_end\"; success: boolean; attempt: number; finalError?: string }\n  | {\n      type: \"budget_exceeded\";\n      /** Which cap was hit, e.g. \"cost 2.01 USD > 2 USD limit\". */\n      reason: string;\n      tokens: number;\n      costUsd: number;\n      llmCalls: number;\n      durationMs: number;\n    };\n\nexport type HarnessEventListener = (event: HarnessEvent) => void | Promise<void>;\n\n// Transient provider failures worth retrying: overload/rate-limit responses,\n// 5xx statuses, and network/stream interruptions.\nconst RETRYABLE_ERROR_PATTERN =\n  /overloaded|provider.?returned.?error|rate.?limit|too many requests|429|500|502|503|504|service.?unavailable|server.?error|internal.?error|network.?error|connection.?error|connection.?refused|connection.?lost|websocket.?closed|websocket.?error|other side closed|fetch failed|upstream.?connect|reset before headers|socket hang up|ended without|stream ended before message_stop|http2 request did not get a response|timed? out|timeout|terminated|retry delay/i;\n\nfunction sleep(ms: number, signal: AbortSignal): Promise<void> {\n  return new Promise((resolve, reject) => {\n    if (signal.aborted) {\n      reject(new Error(\"aborted\"));\n      return;\n    }\n    const timer = setTimeout(() => {\n      signal.removeEventListener(\"abort\", onAbort);\n      resolve();\n    }, ms);\n    const onAbort = () => {\n      clearTimeout(timer);\n      reject(new Error(\"aborted\"));\n    };\n    signal.addEventListener(\"abort\", onAbort, { once: true });\n  });\n}\n\nexport interface MikanAgentSessionOptions {\n  systemPrompt: string;\n  model: Model<Api>;\n  thinkingLevel: ThinkingLevel;\n  tools: AgentTool[];\n  models: MikanModels;\n  sessionStore: SessionStore;\n  settings?: Parameters<typeof resolveHarnessSettings>[0];\n  extensions?: ExtensionRegistry;\n}\n\nexport class MikanAgentSession {\n  readonly agent: Agent;\n  readonly sessionStore: SessionStore;\n  private readonly models: MikanModels;\n  private readonly settings: HarnessSettings;\n  private readonly extensions: ExtensionRegistry | undefined;\n  private listeners = new Set<HarnessEventListener>();\n  private retryAttempt = 0;\n  private overflowRecoveryAttempted = false;\n  private retryAbortController: AbortController | undefined;\n  private compactionAbortController: AbortController | undefined;\n  private tally: RunTally = { tokens: 0, costUsd: 0, llmCalls: 0, startedAt: 0 };\n  private runBudget: BudgetSettings = {};\n  private budgetExceededReason: string | undefined;\n\n  constructor(options: MikanAgentSessionOptions) {\n    this.models = options.models;\n    this.sessionStore = options.sessionStore;\n    this.settings = resolveHarnessSettings(options.settings);\n    this.extensions = options.extensions;\n\n    const tools = [...options.tools, ...(options.extensions?.getContributedTools() ?? [])];\n    this.agent = new Agent({\n      initialState: {\n        systemPrompt: options.systemPrompt,\n        model: options.model,\n        thinkingLevel: options.thinkingLevel,\n        tools,\n      },\n      convertToLlm,\n      streamFn: (model, context, streamOptions) =>\n        this.models.models.streamSimple(model, context, streamOptions),\n      beforeToolCall: async ({ toolCall, args }) => {\n        if (!this.extensions?.hasHandlers(\"tool_call\")) return undefined;\n        const result = await this.extensions.emit(\"tool_call\", {\n          toolCallId: toolCall.id,\n          toolName: toolCall.name,\n          args,\n        });\n        return result ?? undefined;\n      },\n      afterToolCall: async ({ toolCall, args, result, isError }) => {\n        if (this.extensions?.hasHandlers(\"tool_result\")) {\n          await this.extensions.emit(\"tool_result\", {\n            toolCallId: toolCall.id,\n            toolName: toolCall.name,\n            args,\n            content: result.content,\n            isError,\n          });\n        }\n        return undefined;\n      },\n    });\n\n    this.agent.subscribe(async (event) => {\n      await this.handleAgentEvent(event);\n    });\n  }\n\n  /** Conversation transcript currently held by the agent. */\n  get messages(): AgentMessage[] {\n    return this.agent.state.messages;\n  }\n\n  get model(): Model<Api> {\n    return this.agent.state.model;\n  }\n\n  subscribe(listener: HarnessEventListener): () => void {\n    this.listeners.add(listener);\n    return () => this.listeners.delete(listener);\n  }\n\n  /** Replace the agent transcript from the persisted session tree. */\n  reloadFromSession(): number {\n    const context = this.sessionStore.buildSessionContext();\n    if (context.messages.length > 0) {\n      this.agent.state.messages = context.messages;\n    }\n    return context.messages.length;\n  }\n\n  /**\n   * Send a user prompt and run the agent loop to completion, including\n   * automatic retries and compaction. Resolves when the turn (and any\n   * recovery continuations) settles.\n   *\n   * @param options.budget Per-run resource ceilings that override the session\n   *   defaults. Autonomous (event / trigger) runs should pass a budget so a\n   *   runaway loop is stopped even with no human watching.\n   */\n  async prompt(\n    text: string,\n    options?: { images?: ImageContent[]; budget?: BudgetSettings },\n  ): Promise<void> {\n    if (this.agent.state.isStreaming) {\n      throw new Error(\"Agent is already processing a prompt\");\n    }\n\n    const model = this.agent.state.model;\n    await this.ensureAuthConfigured(model);\n\n    this.retryAttempt = 0;\n    this.overflowRecoveryAttempted = false;\n    this.runBudget = { ...this.settings.budget, ...options?.budget };\n    this.budgetExceededReason = undefined;\n    this.tally = { tokens: 0, costUsd: 0, llmCalls: 0, startedAt: Date.now() };\n\n    // A previous turn may have ended over the threshold (for example after an\n    // abort); compact before adding new context on top.\n    const lastAssistant = this.findLastAssistantMessage();\n    if (lastAssistant && lastAssistant.stopReason !== \"error\") {\n      await this.checkThresholdCompaction(lastAssistant);\n    }\n\n    let systemPrompt = this.agent.state.systemPrompt;\n    if (this.extensions?.hasHandlers(\"before_agent_start\")) {\n      const result = await this.extensions.emit(\"before_agent_start\", {\n        prompt: text,\n        images: options?.images,\n        systemPrompt,\n      });\n      if (result?.systemPrompt) {\n        systemPrompt = result.systemPrompt;\n        this.agent.state.systemPrompt = systemPrompt;\n      }\n    }\n\n    const userMessage: AgentMessage = {\n      role: \"user\",\n      content: [{ type: \"text\", text }, ...(options?.images ?? [])],\n      timestamp: Date.now(),\n    };\n\n    await this.agent.prompt([userMessage]);\n    while (await this.handlePostRun()) {\n      await this.agent.continue();\n    }\n\n    if (this.extensions?.hasHandlers(\"turn_end\")) {\n      await this.extensions.emit(\"turn_end\", { messages: this.agent.state.messages });\n    }\n  }\n\n  /** Abort the active run, pending retry backoff, and in-flight compaction. */\n  abort(): void {\n    this.retryAbortController?.abort();\n    this.compactionAbortController?.abort();\n    this.agent.abort();\n  }\n\n  private async ensureAuthConfigured(model: Model<Api>): Promise<void> {\n    const auth = await this.models.getAuth(model);\n    if (!auth) {\n      throw new Error(\n        `No credentials for provider \"${model.provider}\". Set the provider API key environment variable or add it to auth.json.`,\n      );\n    }\n  }\n\n  private async emit(event: HarnessEvent): Promise<void> {\n    for (const listener of this.listeners) {\n      try {\n        await listener(event);\n      } catch (err) {\n        log.logWarning(\n          \"Harness event listener failed\",\n          err instanceof Error ? err.message : String(err),\n        );\n      }\n    }\n  }\n\n  private async handleAgentEvent(event: AgentEvent): Promise<void> {\n    await this.emit(event);\n\n    if (event.type !== \"message_end\") return;\n    const message = event.message;\n\n    if (message.role === \"user\" || message.role === \"assistant\" || message.role === \"toolResult\") {\n      this.sessionStore.appendMessage(message);\n    } else if (message.role === \"custom\") {\n      const custom = message as CustomMessage;\n      this.sessionStore.appendCustomMessageEntry(\n        custom.customType,\n        custom.content,\n        custom.display,\n        custom.details,\n      );\n    }\n\n    if (this.extensions?.hasHandlers(\"message_end\")) {\n      await this.extensions.emit(\"message_end\", { message });\n    }\n\n    if (message.role === \"assistant\") {\n      this.recordUsage(message);\n      await this.enforceBudget();\n      if (message.stopReason !== \"error\") {\n        this.overflowRecoveryAttempted = false;\n        if (this.retryAttempt > 0) {\n          await this.emit({ type: \"auto_retry_end\", success: true, attempt: this.retryAttempt });\n          this.retryAttempt = 0;\n        }\n      }\n    }\n  }\n\n  /** Fold one assistant turn's usage into the running per-run tally. */\n  private recordUsage(message: AssistantMessage): void {\n    this.tally.llmCalls += 1;\n    const usage = message.usage;\n    if (!usage) return;\n    this.tally.tokens += usage.input + usage.output + usage.cacheRead + usage.cacheWrite;\n    this.tally.costUsd += usage.cost?.total ?? 0;\n  }\n\n  /** Compare the tally against the run budget; abort the run when a cap is exceeded. */\n  private async enforceBudget(): Promise<void> {\n    if (this.budgetExceededReason) return;\n    const reason = this.overBudgetReason();\n    if (!reason) return;\n\n    this.budgetExceededReason = reason;\n    await this.emit({\n      type: \"budget_exceeded\",\n      reason,\n      tokens: this.tally.tokens,\n      costUsd: this.tally.costUsd,\n      llmCalls: this.tally.llmCalls,\n      durationMs: Date.now() - this.tally.startedAt,\n    });\n    log.logWarning(\"Run budget exceeded — aborting\", reason);\n    this.agent.abort();\n  }\n\n  private overBudgetReason(): string | undefined {\n    const { maxTokens, maxCostUsd, maxDurationMs, maxLlmCalls } = this.runBudget;\n    if (maxLlmCalls !== undefined && this.tally.llmCalls >= maxLlmCalls) {\n      return `${this.tally.llmCalls} LLM calls >= ${maxLlmCalls} limit`;\n    }\n    if (maxTokens !== undefined && this.tally.tokens >= maxTokens) {\n      return `${this.tally.tokens} tokens >= ${maxTokens} limit`;\n    }\n    if (maxCostUsd !== undefined && this.tally.costUsd >= maxCostUsd) {\n      return `cost ${this.tally.costUsd.toFixed(2)} USD >= ${maxCostUsd} USD limit`;\n    }\n    if (maxDurationMs !== undefined && Date.now() - this.tally.startedAt >= maxDurationMs) {\n      return `${Date.now() - this.tally.startedAt}ms >= ${maxDurationMs}ms limit`;\n    }\n    return undefined;\n  }\n\n  private findLastAssistantMessage(): AssistantMessage | undefined {\n    const messages = this.agent.state.messages;\n    for (let i = messages.length - 1; i >= 0; i--) {\n      const message = messages[i];\n      if (message.role === \"assistant\") return message;\n    }\n    return undefined;\n  }\n\n  /** Decide what to do after a run settles. Returns true when the agent should continue. */\n  private async handlePostRun(): Promise<boolean> {\n    // A run stopped for going over budget must not be retried or compacted back\n    // into another continuation — that would defeat the circuit breaker.\n    if (this.budgetExceededReason) return false;\n\n    const lastAssistant = this.findLastAssistantMessage();\n    if (!lastAssistant) return false;\n\n    if (lastAssistant.stopReason === \"error\") {\n      const contextWindow = this.agent.state.model.contextWindow || 0;\n      if (isContextOverflow(lastAssistant, contextWindow)) {\n        return this.handleOverflow();\n      }\n      if (this.isRetryableError(lastAssistant)) {\n        return this.prepareRetry(lastAssistant);\n      }\n      return false;\n    }\n\n    return this.checkThresholdCompaction(lastAssistant);\n  }\n\n  private async handleOverflow(): Promise<boolean> {\n    if (this.overflowRecoveryAttempted) {\n      await this.emit({\n        type: \"compaction_end\",\n        reason: \"overflow\",\n        aborted: false,\n        errorMessage:\n          \"Context overflow recovery failed after one compact-and-retry attempt. Try /new or a larger-context model.\",\n      });\n      return false;\n    }\n    this.overflowRecoveryAttempted = true;\n    this.dropTrailingErrorMessage();\n    return this.runCompaction(\"overflow\", true);\n  }\n\n  private async checkThresholdCompaction(lastAssistant: AssistantMessage): Promise<boolean> {\n    const settings = this.settings.compaction;\n    if (!settings.enabled) return false;\n    const contextWindow = this.agent.state.model.contextWindow || 0;\n    if (contextWindow <= 0) return false;\n\n    let contextTokens = lastAssistant.usage ? calculateContextTokens(lastAssistant.usage) : 0;\n    if (contextTokens === 0) {\n      const estimate = estimateContextTokens(this.agent.state.messages);\n      if (estimate.lastUsageIndex === null) return false;\n      contextTokens = estimate.tokens;\n    }\n\n    if (!shouldCompact(contextTokens, contextWindow, settings)) return false;\n    return this.runCompaction(\"threshold\", false);\n  }\n\n  private async runCompaction(reason: CompactionReason, willRetry: boolean): Promise<boolean> {\n    let started = false;\n    try {\n      const pathEntries = this.sessionStore.getBranch();\n      const preparation = getOrThrow(prepareCompaction(pathEntries, this.settings.compaction));\n      if (!preparation) return false;\n\n      await this.emit({ type: \"compaction_start\", reason });\n      started = true;\n      this.compactionAbortController = new AbortController();\n      const signal = this.compactionAbortController.signal;\n\n      const result = getOrThrow(\n        await compact(\n          preparation,\n          this.models.models,\n          this.agent.state.model,\n          undefined,\n          signal,\n          this.agent.state.thinkingLevel,\n        ),\n      );\n      if (signal.aborted) {\n        await this.emit({ type: \"compaction_end\", reason, aborted: true });\n        return false;\n      }\n\n      const entryId = this.sessionStore.appendCompaction(\n        result.summary,\n        result.firstKeptEntryId,\n        result.tokensBefore,\n        result.details,\n      );\n      const context = this.sessionStore.buildSessionContext();\n      this.agent.state.messages = context.messages;\n\n      if (this.extensions?.hasHandlers(\"session_compact\")) {\n        const entry = this.sessionStore.getEntry(entryId);\n        if (entry?.type === \"compaction\") {\n          await this.extensions.emit(\"session_compact\", {\n            entry: entry as CompactionEntry,\n            reason,\n          });\n        }\n      }\n\n      await this.emit({\n        type: \"compaction_end\",\n        reason,\n        result: {\n          summary: result.summary,\n          firstKeptEntryId: result.firstKeptEntryId,\n          tokensBefore: result.tokensBefore,\n        },\n        aborted: false,\n      });\n\n      if (willRetry) {\n        this.dropTrailingErrorMessage();\n        return true;\n      }\n      return this.agent.hasQueuedMessages();\n    } catch (error) {\n      const errorMessage = error instanceof Error ? error.message : \"compaction failed\";\n      if (started) {\n        await this.emit({\n          type: \"compaction_end\",\n          reason,\n          aborted: false,\n          errorMessage:\n            reason === \"overflow\"\n              ? `Context overflow recovery failed: ${errorMessage}`\n              : `Auto-compaction failed: ${errorMessage}`,\n        });\n      } else {\n        log.logWarning(\"Compaction preparation failed\", errorMessage);\n      }\n      return false;\n    } finally {\n      this.compactionAbortController = undefined;\n    }\n  }\n\n  private isRetryableError(message: AssistantMessage): boolean {\n    if (message.stopReason !== \"error\" || !message.errorMessage) return false;\n    return RETRYABLE_ERROR_PATTERN.test(message.errorMessage);\n  }\n\n  private async prepareRetry(message: AssistantMessage): Promise<boolean> {\n    const settings = this.settings.retry;\n    if (!settings.enabled) return false;\n\n    this.retryAttempt++;\n    if (this.retryAttempt > settings.maxRetries) {\n      this.retryAttempt--;\n      await this.emit({\n        type: \"auto_retry_end\",\n        success: false,\n        attempt: this.retryAttempt,\n        finalError: message.errorMessage,\n      });\n      this.retryAttempt = 0;\n      return false;\n    }\n\n    const delayMs = settings.baseDelayMs * 2 ** (this.retryAttempt - 1);\n    await this.emit({\n      type: \"auto_retry_start\",\n      attempt: this.retryAttempt,\n      maxAttempts: settings.maxRetries,\n      delayMs,\n      errorMessage: message.errorMessage || \"Unknown error\",\n    });\n\n    this.dropTrailingErrorMessage();\n\n    this.retryAbortController = new AbortController();\n    try {\n      await sleep(delayMs, this.retryAbortController.signal);\n    } catch {\n      const attempt = this.retryAttempt;\n      this.retryAttempt = 0;\n      await this.emit({\n        type: \"auto_retry_end\",\n        success: false,\n        attempt,\n        finalError: \"Retry cancelled\",\n      });\n      return false;\n    } finally {\n      this.retryAbortController = undefined;\n    }\n    return true;\n  }\n\n  private dropTrailingErrorMessage(): void {\n    const messages = this.agent.state.messages;\n    if (messages.length > 0 && messages[messages.length - 1].role === \"assistant\") {\n      this.agent.state.messages = messages.slice(0, -1);\n    }\n  }\n}\n"]}
+{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/harness/runner.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AACH,OAAO,EACL,KAAK,EACL,sBAAsB,EACtB,OAAO,EACP,YAAY,EACZ,qBAAqB,EACrB,UAAU,EACV,iBAAiB,EACjB,aAAa,GAMd,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,iBAAiB,GAKlB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,KAAK,GAAG,MAAM,WAAW,CAAC;AAIjC,OAAO,EAAE,sBAAsB,EAA6C,MAAM,eAAe,CAAC;AAwDlG,6EAA6E;AAC7E,kDAAkD;AAClD,MAAM,uBAAuB,GAC3B,wcAAwc,CAAC;AAE3c,SAAS,KAAK,CAAC,EAAU,EAAE,MAAmB;IAC5C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC;YAC7B,OAAO;QACT,CAAC;QACD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC7C,OAAO,EAAE,CAAC;QACZ,CAAC,EAAE,EAAE,CAAC,CAAC;QACP,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC;QAC/B,CAAC,CAAC;QACF,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC;AAaD,MAAM,OAAO,iBAAiB;IAgB5B,YAAY,OAAiC;QAVrC,cAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;QAC5C,iBAAY,GAAG,CAAC,CAAC;QACjB,8BAAyB,GAAG,KAAK,CAAC;QAGlC,UAAK,GAAa,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;QACvE,cAAS,GAAmB,EAAE,CAAC;QAKrC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,QAAQ,GAAG,sBAAsB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QAErC,MAAM,KAAK,GAAG,CAAC,GAAG,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,OAAO,CAAC,UAAU,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC;YACrB,YAAY,EAAE;gBACZ,YAAY,EAAE,OAAO,CAAC,YAAY;gBAClC,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,aAAa,EAAE,OAAO,CAAC,aAAa;gBACpC,KAAK;aACN;YACD,YAAY;YACZ,QAAQ,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE,CAC1C,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,aAAa,CAAC;YAChE,cAAc,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE;gBAC3C,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,WAAW,CAAC;oBAAE,OAAO,SAAS,CAAC;gBACjE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,EAAE;oBACrD,UAAU,EAAE,QAAQ,CAAC,EAAE;oBACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,IAAI;oBACJ,MAAM,EAAE,IAAI,CAAC,SAAS;iBACvB,CAAC,CAAC;gBACH,OAAO,MAAM,IAAI,SAAS,CAAC;YAC7B,CAAC;YACD,aAAa,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE;gBAC3D,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,aAAa,CAAC;oBAAE,OAAO,SAAS,CAAC;gBACnE,8DAA8D;gBAC9D,kEAAkE;gBAClE,6BAA6B;gBAC7B,OAAO,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC;oBACpC,UAAU,EAAE,QAAQ,CAAC,EAAE;oBACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,IAAI;oBACJ,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,OAAO;oBACP,MAAM,EAAE,IAAI,CAAC,SAAS;iBACvB,CAAC,CAAC;YACL,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACnC,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,2DAA2D;IAC3D,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC;IACnC,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC;IAChC,CAAC;IAED,6EAA6E;IAC7E,eAAe;QAOb,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM;YACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO;YAC3B,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ;YAC7B,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,IAAI,CAAC,oBAAoB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC1F,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,iBAAiB,CAAC,KAA4C;QAClE,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,IAAI,CAAC,CAAC;QACzC,IAAI,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW;YAAE,OAAO;QACvE,MAAM,MAAM,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAC/C,IAAI,MAAM;YAAE,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;IAED,SAAS,CAAC,QAA8B;QACtC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7B,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED,oEAAoE;IACpE,iBAAiB;QACf,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,CAAC;QACxD,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAC/C,CAAC;QACD,OAAO,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;IACjC,CAAC;IAED;;;;;;;;;;;OAWG;IACH,KAAK,CAAC,MAAM,CACV,IAAY,EACZ,OAAkF;QAElF,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC;QACrC,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAEvC,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;QACtB,IAAI,CAAC,yBAAyB,GAAG,KAAK,CAAC;QACvC,IAAI,CAAC,SAAS,GAAG,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,CAAC;QACjE,IAAI,CAAC,oBAAoB,GAAG,SAAS,CAAC;QACtC,IAAI,CAAC,KAAK,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QAC3E,IAAI,CAAC,SAAS,GAAG,OAAO,EAAE,MAAM,CAAC;QAEjC,0EAA0E;QAC1E,oDAAoD;QACpD,MAAM,aAAa,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QACtD,IAAI,aAAa,IAAI,aAAa,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;YAC1D,MAAM,IAAI,CAAC,wBAAwB,CAAC,aAAa,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,UAAU,GAAG,IAAI,CAAC;QACtB,IAAI,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;QACjD,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,oBAAoB,CAAC,EAAE,CAAC;YACvD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAAC;gBACxD,MAAM,EAAE,UAAU;gBAClB,MAAM,EAAE,OAAO,EAAE,MAAM;gBACvB,YAAY;gBACZ,MAAM,EAAE,IAAI,CAAC,SAAS;aACvB,CAAC,CAAC;YACH,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;gBAClB,mEAAmE;gBACnE,oEAAoE;gBACpE,GAAG,CAAC,OAAO,CAAC,yBAAyB,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAClF,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;YAClD,CAAC;YACD,IAAI,MAAM,EAAE,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;gBACjE,GAAG,CAAC,OAAO,CACT,kCAAkC,UAAU,CAAC,MAAM,MAAM,MAAM,CAAC,MAAM,CAAC,MAAM,QAAQ,CACtF,CAAC;gBACF,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC;YAC7B,CAAC;YACD,IAAI,MAAM,EAAE,YAAY,IAAI,MAAM,CAAC,YAAY,KAAK,YAAY,EAAE,CAAC;gBACjE,oEAAoE;gBACpE,sEAAsE;gBACtE,yDAAyD;gBACzD,GAAG,CAAC,OAAO,CACT,oCAAoC,YAAY,CAAC,MAAM,MAAM,MAAM,CAAC,YAAY,CAAC,MAAM,QAAQ,CAChG,CAAC;gBACF,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;gBACnC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,GAAG,YAAY,CAAC;YAC/C,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAAiB;YAChC,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,GAAG,CAAC,OAAO,EAAE,MAAM,IAAI,EAAE,CAAC,CAAC;YACzE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;QACvC,OAAO,MAAM,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;YAClC,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC9B,CAAC;QAED,qEAAqE;QACrE,+DAA+D;QAC/D,MAAM,OAAO,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAChD,IAAI,OAAO,EAAE,UAAU,KAAK,OAAO,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,aAAa,CAAC,EAAE,CAAC;YACnF,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,EAAE;gBACxC,YAAY,EAAE,OAAO,CAAC,YAAY,IAAI,eAAe;gBACrD,MAAM,EAAE,IAAI,CAAC,SAAS;aACvB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC;YAC7C,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,EAAE;gBACrC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ;gBACnC,MAAM,EAAE,IAAI,CAAC,SAAS;aACvB,CAAC,CAAC;QACL,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,6EAA6E;IAC7E,KAAK;QACH,IAAI,CAAC,oBAAoB,EAAE,KAAK,EAAE,CAAC;QACnC,IAAI,CAAC,yBAAyB,EAAE,KAAK,EAAE,CAAC;QACxC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAAC,KAAiB;QAClD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC9C,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CACb,gCAAgC,KAAK,CAAC,QAAQ,0EAA0E,CACzH,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,IAAI,CAAC,KAAmB;QACpC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,IAAI,CAAC;gBACH,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,GAAG,CAAC,UAAU,CACZ,+BAA+B,EAC/B,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CACjD,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,KAAiB;QAC9C,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEvB,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa;YAAE,OAAO;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;QAE9B,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC7F,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAC3C,CAAC;aAAM,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACrC,MAAM,MAAM,GAAG,OAAwB,CAAC;YACxC,IAAI,CAAC,YAAY,CAAC,wBAAwB,CACxC,MAAM,CAAC,UAAU,EACjB,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,OAAO,CACf,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,aAAa,CAAC,EAAE,CAAC;YAChD,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QACjF,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACjC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAC1B,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YAClC,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;gBACnC,IAAI,CAAC,yBAAyB,GAAG,KAAK,CAAC;gBACvC,IAAI,IAAI,CAAC,YAAY,GAAG,CAAC,EAAE,CAAC;oBAC1B,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;oBACvF,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,sEAAsE;IAC9D,WAAW,CAAC,OAAyB;QAC3C,IAAI,CAAC,KAAK,CAAC,QAAQ,IAAI,CAAC,CAAC;QACzB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC5B,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC;QACrF,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,IAAI,CAAC,CAAC;IAC/C,CAAC;IAED,sFAAsF;IAC9E,KAAK,CAAC,aAAa,CAAC,OAAyB;QACnD,IAAI,IAAI,CAAC,oBAAoB;YAAE,OAAO;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM;YAAE,OAAO;QACpB,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IAED,kFAAkF;IAC1E,KAAK,CAAC,YAAY,CAAC,MAAc;QACvC,IAAI,CAAC,oBAAoB,GAAG,MAAM,CAAC;QACnC,MAAM,IAAI,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,iBAAiB;YACvB,MAAM;YACN,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM;YACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO;YAC3B,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ;YAC7B,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS;SAC9C,CAAC,CAAC;QACH,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,iBAAiB,EAAE;gBAC5C,MAAM;gBACN,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM;gBACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO;gBAC3B,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ;gBAC7B,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS;gBAC7C,MAAM,EAAE,IAAI,CAAC,SAAS;aACvB,CAAC,CAAC;QACL,CAAC;QACD,GAAG,CAAC,UAAU,CAAC,gCAAgC,EAAE,MAAM,CAAC,CAAC;QACzD,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAEO,gBAAgB,CAAC,OAAyB;QAChD,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;QACvC,MAAM,gBAAgB,GACpB,OAAO,CAAC,UAAU,KAAK,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QAC7F,IAAI,gBAAgB,IAAI,WAAW,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,IAAI,WAAW,EAAE,CAAC;YACxF,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,iBAAiB,WAAW,QAAQ,CAAC;QACpE,CAAC;QACD,OAAO,IAAI,CAAC,wBAAwB,EAAE,CAAC;IACzC,CAAC;IAED,+EAA+E;IACvE,wBAAwB;QAC9B,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;QAChE,IAAI,SAAS,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;YAC9D,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,cAAc,SAAS,QAAQ,CAAC;QAC7D,CAAC;QACD,IAAI,UAAU,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,UAAU,EAAE,CAAC;YACjE,OAAO,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,UAAU,YAAY,CAAC;QAChF,CAAC;QACD,IAAI,aAAa,KAAK,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,IAAI,aAAa,EAAE,CAAC;YACtF,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,SAAS,aAAa,UAAU,CAAC;QAC9E,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,wBAAwB;QAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC3C,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9C,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC5B,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;gBAAE,OAAO,OAAO,CAAC;QACnD,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,0FAA0F;IAClF,KAAK,CAAC,aAAa;QACzB,4EAA4E;QAC5E,qEAAqE;QACrE,IAAI,IAAI,CAAC,oBAAoB;YAAE,OAAO,KAAK,CAAC;QAE5C,MAAM,aAAa,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QACtD,IAAI,CAAC,aAAa;YAAE,OAAO,KAAK,CAAC;QAEjC,IAAI,aAAa,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;YACzC,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;YAChE,IAAI,iBAAiB,CAAC,aAAa,EAAE,aAAa,CAAC,EAAE,CAAC;gBACpD,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC;YAC/B,CAAC;YACD,IAAI,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,EAAE,CAAC;gBACzC,OAAO,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YAC1C,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,IAAI,CAAC,wBAAwB,CAAC,aAAa,CAAC,CAAC;IACtD,CAAC;IAEO,KAAK,CAAC,cAAc;QAC1B,IAAI,IAAI,CAAC,yBAAyB,EAAE,CAAC;YACnC,MAAM,IAAI,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,gBAAgB;gBACtB,MAAM,EAAE,UAAU;gBAClB,OAAO,EAAE,KAAK;gBACd,YAAY,EACV,2GAA2G;aAC9G,CAAC,CAAC;YACH,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,CAAC,yBAAyB,GAAG,IAAI,CAAC;QACtC,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAChC,OAAO,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;IAC9C,CAAC;IAEO,KAAK,CAAC,wBAAwB,CAAC,aAA+B;QACpE,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;QAC1C,IAAI,CAAC,QAAQ,CAAC,OAAO;YAAE,OAAO,KAAK,CAAC;QACpC,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;QAChE,IAAI,aAAa,IAAI,CAAC;YAAE,OAAO,KAAK,CAAC;QAErC,IAAI,aAAa,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,sBAAsB,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1F,IAAI,aAAa,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,QAAQ,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAClE,IAAI,QAAQ,CAAC,cAAc,KAAK,IAAI;gBAAE,OAAO,KAAK,CAAC;YACnD,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC;QAClC,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,aAAa,EAAE,aAAa,EAAE,QAAQ,CAAC;YAAE,OAAO,KAAK,CAAC;QACzE,OAAO,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,MAAwB,EAAE,SAAkB;QACtE,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;YAClD,MAAM,WAAW,GAAG,UAAU,CAAC,iBAAiB,CAAC,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC;YACzF,IAAI,CAAC,WAAW;gBAAE,OAAO,KAAK,CAAC;YAE/B,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,MAAM,EAAE,CAAC,CAAC;YACtD,OAAO,GAAG,IAAI,CAAC;YACf,IAAI,CAAC,yBAAyB,GAAG,IAAI,eAAe,EAAE,CAAC;YACvD,MAAM,MAAM,GAAG,IAAI,CAAC,yBAAyB,CAAC,MAAM,CAAC;YAErD,MAAM,MAAM,GAAG,UAAU,CACvB,MAAM,OAAO,CACX,WAAW,EACX,IAAI,CAAC,MAAM,CAAC,MAAM,EAClB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EACtB,SAAS,EACT,MAAM,EACN,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,CAC/B,CACF,CAAC;YACF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnE,OAAO,KAAK,CAAC;YACf,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAChD,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,gBAAgB,EACvB,MAAM,CAAC,YAAY,EACnB,MAAM,CAAC,OAAO,CACf,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,CAAC;YACxD,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;YAE7C,IAAI,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACpD,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAClD,IAAI,KAAK,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;oBACjC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,iBAAiB,EAAE;wBAC5C,KAAK,EAAE,KAAwB;wBAC/B,MAAM;qBACP,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,MAAM,IAAI,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,gBAAgB;gBACtB,MAAM;gBACN,MAAM,EAAE;oBACN,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;oBACzC,YAAY,EAAE,MAAM,CAAC,YAAY;iBAClC;gBACD,OAAO,EAAE,KAAK;aACf,CAAC,CAAC;YAEH,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,CAAC,wBAAwB,EAAE,CAAC;gBAChC,OAAO,IAAI,CAAC;YACd,CAAC;YACD,OAAO,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC;QACxC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,mBAAmB,CAAC;YAClF,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,IAAI,CAAC,IAAI,CAAC;oBACd,IAAI,EAAE,gBAAgB;oBACtB,MAAM;oBACN,OAAO,EAAE,KAAK;oBACd,YAAY,EACV,MAAM,KAAK,UAAU;wBACnB,CAAC,CAAC,qCAAqC,YAAY,EAAE;wBACrD,CAAC,CAAC,2BAA2B,YAAY,EAAE;iBAChD,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,UAAU,CAAC,+BAA+B,EAAE,YAAY,CAAC,CAAC;YAChE,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,yBAAyB,GAAG,SAAS,CAAC;QAC7C,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,OAAyB;QAChD,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY;YAAE,OAAO,KAAK,CAAC;QAC1E,OAAO,uBAAuB,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAC5D,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,OAAyB;QAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QACrC,IAAI,CAAC,QAAQ,CAAC,OAAO;YAAE,OAAO,KAAK,CAAC;QAEpC,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,IAAI,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC5C,IAAI,CAAC,YAAY,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,IAAI,CAAC,YAAY;gBAC1B,UAAU,EAAE,OAAO,CAAC,YAAY;aACjC,CAAC,CAAC;YACH,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;YACtB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,WAAW,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC;QACpE,MAAM,IAAI,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,kBAAkB;YACxB,OAAO,EAAE,IAAI,CAAC,YAAY;YAC1B,WAAW,EAAE,QAAQ,CAAC,UAAU;YAChC,OAAO;YACP,YAAY,EAAE,OAAO,CAAC,YAAY,IAAI,eAAe;SACtD,CAAC,CAAC;QAEH,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAEhC,IAAI,CAAC,oBAAoB,GAAG,IAAI,eAAe,EAAE,CAAC;QAClD,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QACzD,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC;YAClC,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;YACtB,MAAM,IAAI,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,KAAK;gBACd,OAAO;gBACP,UAAU,EAAE,iBAAiB;aAC9B,CAAC,CAAC;YACH,OAAO,KAAK,CAAC;QACf,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,oBAAoB,GAAG,SAAS,CAAC;QACxC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,wBAAwB;QAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC3C,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9E,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;CACF","sourcesContent":["/**\n * MikanAgentSession — mikan's agent run loop.\n *\n * Owns a pi-agent-core `Agent` and layers on the behaviors mikan needs for\n * long-lived chat conversations:\n *\n * - message persistence into the conversation's {@link SessionStore}\n * - automatic context compaction (threshold and overflow recovery), using\n *   pi-agent-core's compaction pipeline over the session tree\n * - automatic retry with exponential backoff on transient provider errors\n * - extension hook dispatch ({@link ExtensionRegistry})\n *\n * Events mirror the agent's lifecycle events plus `compaction_start/_end`\n * and `auto_retry_start/_end`, preserving the event surface mikan's\n * platform adapters already render.\n */\nimport {\n  Agent,\n  calculateContextTokens,\n  compact,\n  convertToLlm,\n  estimateContextTokens,\n  getOrThrow,\n  prepareCompaction,\n  shouldCompact,\n  type AgentEvent,\n  type AgentMessage,\n  type AgentTool,\n  type CustomMessage,\n  type ThinkingLevel,\n} from \"@earendil-works/pi-agent-core\";\nimport {\n  isContextOverflow,\n  type Api,\n  type AssistantMessage,\n  type ImageContent,\n  type Model,\n} from \"@earendil-works/pi-ai\";\nimport * as log from \"../log.js\";\nimport type { ExtensionRegistry } from \"./extensions/registry.js\";\nimport type { RunOrigin } from \"./extensions/types.js\";\nimport type { MikanModels } from \"./models.js\";\nimport { resolveHarnessSettings, type BudgetSettings, type HarnessSettings } from \"./settings.js\";\nimport type { SessionStore } from \"./session-store.js\";\nimport type { CompactionEntry } from \"./types.js\";\n\nexport type CompactionReason = \"threshold\" | \"overflow\" | \"manual\";\n\n/** Running resource tally for the current `prompt()` call, matched against the budget. */\ninterface RunTally {\n  tokens: number;\n  costUsd: number;\n  llmCalls: number;\n  startedAt: number;\n}\n\ninterface CompactionResultSummary {\n  summary: string;\n  firstKeptEntryId: string;\n  tokensBefore: number;\n}\n\nexport type HarnessEvent =\n  | AgentEvent\n  | { type: \"compaction_start\"; reason: CompactionReason }\n  | {\n      type: \"compaction_end\";\n      reason: CompactionReason;\n      result?: CompactionResultSummary;\n      aborted: boolean;\n      errorMessage?: string;\n    }\n  | {\n      type: \"auto_retry_start\";\n      attempt: number;\n      maxAttempts: number;\n      delayMs: number;\n      errorMessage: string;\n    }\n  | { type: \"auto_retry_end\"; success: boolean; attempt: number; finalError?: string }\n  | {\n      type: \"budget_exceeded\";\n      /** Which cap was hit, e.g. \"cost 2.01 USD > 2 USD limit\". */\n      reason: string;\n      tokens: number;\n      costUsd: number;\n      llmCalls: number;\n      durationMs: number;\n    };\n\nexport type HarnessEventListener = (event: HarnessEvent) => void | Promise<void>;\n\n/** Outcome of a `prompt()` call that an extension blocked before the model ran. */\nexport interface PromptBlockedOutcome {\n  blocked: true;\n  reason?: string;\n}\n\n// Transient provider failures worth retrying: overload/rate-limit responses,\n// 5xx statuses, and network/stream interruptions.\nconst RETRYABLE_ERROR_PATTERN =\n  /overloaded|provider.?returned.?error|rate.?limit|too many requests|429|500|502|503|504|service.?unavailable|server.?error|internal.?error|network.?error|connection.?error|connection.?refused|connection.?lost|websocket.?closed|websocket.?error|other side closed|fetch failed|upstream.?connect|reset before headers|socket hang up|ended without|stream ended before message_stop|http2 request did not get a response|timed? out|timeout|terminated|retry delay/i;\n\nfunction sleep(ms: number, signal: AbortSignal): Promise<void> {\n  return new Promise((resolve, reject) => {\n    if (signal.aborted) {\n      reject(new Error(\"aborted\"));\n      return;\n    }\n    const timer = setTimeout(() => {\n      signal.removeEventListener(\"abort\", onAbort);\n      resolve();\n    }, ms);\n    const onAbort = () => {\n      clearTimeout(timer);\n      reject(new Error(\"aborted\"));\n    };\n    signal.addEventListener(\"abort\", onAbort, { once: true });\n  });\n}\n\nexport interface MikanAgentSessionOptions {\n  systemPrompt: string;\n  model: Model<Api>;\n  thinkingLevel: ThinkingLevel;\n  tools: AgentTool[];\n  models: MikanModels;\n  sessionStore: SessionStore;\n  settings?: Parameters<typeof resolveHarnessSettings>[0];\n  extensions?: ExtensionRegistry;\n}\n\nexport class MikanAgentSession {\n  readonly agent: Agent;\n  readonly sessionStore: SessionStore;\n  private readonly models: MikanModels;\n  private readonly settings: HarnessSettings;\n  private readonly extensions: ExtensionRegistry | undefined;\n  private listeners = new Set<HarnessEventListener>();\n  private retryAttempt = 0;\n  private overflowRecoveryAttempted = false;\n  private retryAbortController: AbortController | undefined;\n  private compactionAbortController: AbortController | undefined;\n  private tally: RunTally = { tokens: 0, costUsd: 0, llmCalls: 0, startedAt: 0 };\n  private runBudget: BudgetSettings = {};\n  private budgetExceededReason: string | undefined;\n  private runOrigin: RunOrigin | undefined;\n\n  constructor(options: MikanAgentSessionOptions) {\n    this.models = options.models;\n    this.sessionStore = options.sessionStore;\n    this.settings = resolveHarnessSettings(options.settings);\n    this.extensions = options.extensions;\n\n    const tools = [...options.tools, ...(options.extensions?.getContributedTools() ?? [])];\n    this.agent = new Agent({\n      initialState: {\n        systemPrompt: options.systemPrompt,\n        model: options.model,\n        thinkingLevel: options.thinkingLevel,\n        tools,\n      },\n      convertToLlm,\n      streamFn: (model, context, streamOptions) =>\n        this.models.models.streamSimple(model, context, streamOptions),\n      beforeToolCall: async ({ toolCall, args }) => {\n        if (!this.extensions?.hasHandlers(\"tool_call\")) return undefined;\n        const result = await this.extensions.emit(\"tool_call\", {\n          toolCallId: toolCall.id,\n          toolName: toolCall.name,\n          args,\n          origin: this.runOrigin,\n        });\n        return result ?? undefined;\n      },\n      afterToolCall: async ({ toolCall, args, result, isError }) => {\n        if (!this.extensions?.hasHandlers(\"tool_result\")) return undefined;\n        // Chained extension rewrites (redaction, truncation) become a\n        // pi-agent-core AfterToolCallResult override; undefined keeps the\n        // executed result untouched.\n        return this.extensions.emitToolResult({\n          toolCallId: toolCall.id,\n          toolName: toolCall.name,\n          args,\n          content: result.content,\n          isError,\n          origin: this.runOrigin,\n        });\n      },\n    });\n\n    this.agent.subscribe(async (event) => {\n      await this.handleAgentEvent(event);\n    });\n  }\n\n  /** Conversation transcript currently held by the agent. */\n  get messages(): AgentMessage[] {\n    return this.agent.state.messages;\n  }\n\n  get model(): Model<Api> {\n    return this.agent.state.model;\n  }\n\n  /** Resource totals and terminal budget state from the most recent prompt. */\n  getLastRunStats(): Readonly<{\n    tokens: number;\n    costUsd: number;\n    llmCalls: number;\n    durationMs: number;\n    budgetExceededReason?: string;\n  }> {\n    return {\n      tokens: this.tally.tokens,\n      costUsd: this.tally.costUsd,\n      llmCalls: this.tally.llmCalls,\n      durationMs: this.tally.startedAt > 0 ? Date.now() - this.tally.startedAt : 0,\n      ...(this.budgetExceededReason ? { budgetExceededReason: this.budgetExceededReason } : {}),\n    };\n  }\n\n  /**\n   * Fold spend incurred outside this session's own LLM calls — e.g. subagent\n   * runs launched by a tool during the current prompt — into the run tally\n   * and enforce the run budget at the fold itself: a fold that lands over a\n   * resource ceiling aborts the run now instead of waiting for a next\n   * assistant message that may never come (and would be paid for). Counts\n   * tokens and cost only; external runs are not this session's turns.\n   */\n  async foldExternalUsage(usage: { tokens?: number; costUsd?: number }): Promise<void> {\n    this.tally.tokens += usage.tokens ?? 0;\n    this.tally.costUsd += usage.costUsd ?? 0;\n    if (this.budgetExceededReason || !this.agent.state.isStreaming) return;\n    const reason = this.resourceOverBudgetReason();\n    if (reason) await this.exceedBudget(reason);\n  }\n\n  subscribe(listener: HarnessEventListener): () => void {\n    this.listeners.add(listener);\n    return () => this.listeners.delete(listener);\n  }\n\n  /** Replace the agent transcript from the persisted session tree. */\n  reloadFromSession(): number {\n    const context = this.sessionStore.buildSessionContext();\n    if (context.messages.length > 0) {\n      this.agent.state.messages = context.messages;\n    }\n    return context.messages.length;\n  }\n\n  /**\n   * Send a user prompt and run the agent loop to completion, including\n   * automatic retries and compaction. Resolves when the turn (and any\n   * recovery continuations) settles. Returns a blocked outcome when a\n   * `before_agent_start` extension blocked the turn — the model was never\n   * called and nothing was persisted.\n   *\n   * @param options.budget Per-run resource ceilings that override the session\n   *   defaults. Autonomous (event / trigger) runs should pass a budget so a\n   *   runaway loop is stopped even with no human watching.\n   * @param options.origin Platform provenance forwarded to extension hooks.\n   */\n  async prompt(\n    text: string,\n    options?: { images?: ImageContent[]; budget?: BudgetSettings; origin?: RunOrigin },\n  ): Promise<PromptBlockedOutcome | undefined> {\n    if (this.agent.state.isStreaming) {\n      throw new Error(\"Agent is already processing a prompt\");\n    }\n\n    const model = this.agent.state.model;\n    await this.ensureAuthConfigured(model);\n\n    this.retryAttempt = 0;\n    this.overflowRecoveryAttempted = false;\n    this.runBudget = { ...this.settings.budget, ...options?.budget };\n    this.budgetExceededReason = undefined;\n    this.tally = { tokens: 0, costUsd: 0, llmCalls: 0, startedAt: Date.now() };\n    this.runOrigin = options?.origin;\n\n    // A previous turn may have ended over the threshold (for example after an\n    // abort); compact before adding new context on top.\n    const lastAssistant = this.findLastAssistantMessage();\n    if (lastAssistant && lastAssistant.stopReason !== \"error\") {\n      await this.checkThresholdCompaction(lastAssistant);\n    }\n\n    let promptText = text;\n    let systemPrompt = this.agent.state.systemPrompt;\n    if (this.extensions?.hasHandlers(\"before_agent_start\")) {\n      const result = await this.extensions.emitBeforeAgentStart({\n        prompt: promptText,\n        images: options?.images,\n        systemPrompt,\n        origin: this.runOrigin,\n      });\n      if (result?.block) {\n        // Blocked before agent.prompt(): the user message never enters the\n        // transcript or session store, so the blocked turn leaves no trace.\n        log.logInfo(`Extension blocked turn${result.reason ? `: ${result.reason}` : \"\"}`);\n        return { blocked: true, reason: result.reason };\n      }\n      if (result?.prompt !== undefined && result.prompt !== promptText) {\n        log.logInfo(\n          `Extension rewrote user prompt: ${promptText.length} → ${result.prompt.length} chars`,\n        );\n        promptText = result.prompt;\n      }\n      if (result?.systemPrompt && result.systemPrompt !== systemPrompt) {\n        // An extension rewrote the system prompt for this turn. Log the new\n        // size so the diagnostic in agent.ts (which logs the pre-hook prompt)\n        // isn't mistaken for the final prompt sent to the model.\n        log.logInfo(\n          `Extension rewrote system prompt: ${systemPrompt.length} → ${result.systemPrompt.length} chars`,\n        );\n        systemPrompt = result.systemPrompt;\n        this.agent.state.systemPrompt = systemPrompt;\n      }\n    }\n\n    const userMessage: AgentMessage = {\n      role: \"user\",\n      content: [{ type: \"text\", text: promptText }, ...(options?.images ?? [])],\n      timestamp: Date.now(),\n    };\n\n    await this.agent.prompt([userMessage]);\n    while (await this.handlePostRun()) {\n      await this.agent.continue();\n    }\n\n    // The turn settled on an error (retries exhausted or not retryable):\n    // give monitoring extensions the final failure, once per turn.\n    const settled = this.findLastAssistantMessage();\n    if (settled?.stopReason === \"error\" && this.extensions?.hasHandlers(\"agent_error\")) {\n      await this.extensions.emit(\"agent_error\", {\n        errorMessage: settled.errorMessage || \"Unknown error\",\n        origin: this.runOrigin,\n      });\n    }\n\n    if (this.extensions?.hasHandlers(\"turn_end\")) {\n      await this.extensions.emit(\"turn_end\", {\n        messages: this.agent.state.messages,\n        origin: this.runOrigin,\n      });\n    }\n    return undefined;\n  }\n\n  /** Abort the active run, pending retry backoff, and in-flight compaction. */\n  abort(): void {\n    this.retryAbortController?.abort();\n    this.compactionAbortController?.abort();\n    this.agent.abort();\n  }\n\n  private async ensureAuthConfigured(model: Model<Api>): Promise<void> {\n    const auth = await this.models.getAuth(model);\n    if (!auth) {\n      throw new Error(\n        `No credentials for provider \"${model.provider}\". Set the provider API key environment variable or add it to auth.json.`,\n      );\n    }\n  }\n\n  private async emit(event: HarnessEvent): Promise<void> {\n    for (const listener of this.listeners) {\n      try {\n        await listener(event);\n      } catch (err) {\n        log.logWarning(\n          \"Harness event listener failed\",\n          err instanceof Error ? err.message : String(err),\n        );\n      }\n    }\n  }\n\n  private async handleAgentEvent(event: AgentEvent): Promise<void> {\n    await this.emit(event);\n\n    if (event.type !== \"message_end\") return;\n    const message = event.message;\n\n    if (message.role === \"user\" || message.role === \"assistant\" || message.role === \"toolResult\") {\n      this.sessionStore.appendMessage(message);\n    } else if (message.role === \"custom\") {\n      const custom = message as CustomMessage;\n      this.sessionStore.appendCustomMessageEntry(\n        custom.customType,\n        custom.content,\n        custom.display,\n        custom.details,\n      );\n    }\n\n    if (this.extensions?.hasHandlers(\"message_end\")) {\n      await this.extensions.emit(\"message_end\", { message, origin: this.runOrigin });\n    }\n\n    if (message.role === \"assistant\") {\n      this.recordUsage(message);\n      await this.enforceBudget(message);\n      if (message.stopReason !== \"error\") {\n        this.overflowRecoveryAttempted = false;\n        if (this.retryAttempt > 0) {\n          await this.emit({ type: \"auto_retry_end\", success: true, attempt: this.retryAttempt });\n          this.retryAttempt = 0;\n        }\n      }\n    }\n  }\n\n  /** Fold one assistant turn's usage into the running per-run tally. */\n  private recordUsage(message: AssistantMessage): void {\n    this.tally.llmCalls += 1;\n    const usage = message.usage;\n    if (!usage) return;\n    this.tally.tokens += usage.input + usage.output + usage.cacheRead + usage.cacheWrite;\n    this.tally.costUsd += usage.cost?.total ?? 0;\n  }\n\n  /** Compare the tally against the run budget; abort the run when a cap is exceeded. */\n  private async enforceBudget(message: AssistantMessage): Promise<void> {\n    if (this.budgetExceededReason) return;\n    const reason = this.overBudgetReason(message);\n    if (!reason) return;\n    await this.exceedBudget(reason);\n  }\n\n  /** Mark the run over budget, notify listeners and extensions, abort the agent. */\n  private async exceedBudget(reason: string): Promise<void> {\n    this.budgetExceededReason = reason;\n    await this.emit({\n      type: \"budget_exceeded\",\n      reason,\n      tokens: this.tally.tokens,\n      costUsd: this.tally.costUsd,\n      llmCalls: this.tally.llmCalls,\n      durationMs: Date.now() - this.tally.startedAt,\n    });\n    if (this.extensions?.hasHandlers(\"budget_exceeded\")) {\n      await this.extensions.emit(\"budget_exceeded\", {\n        reason,\n        tokens: this.tally.tokens,\n        costUsd: this.tally.costUsd,\n        llmCalls: this.tally.llmCalls,\n        durationMs: Date.now() - this.tally.startedAt,\n        origin: this.runOrigin,\n      });\n    }\n    log.logWarning(\"Run budget exceeded — aborting\", reason);\n    this.agent.abort();\n  }\n\n  private overBudgetReason(message: AssistantMessage): string | undefined {\n    const { maxLlmCalls } = this.runBudget;\n    const needsAnotherCall =\n      message.stopReason === \"error\" || message.content.some((part) => part.type === \"toolCall\");\n    if (needsAnotherCall && maxLlmCalls !== undefined && this.tally.llmCalls >= maxLlmCalls) {\n      return `${this.tally.llmCalls} LLM calls >= ${maxLlmCalls} limit`;\n    }\n    return this.resourceOverBudgetReason();\n  }\n\n  /** The message-independent budget checks, shared with external-spend folds. */\n  private resourceOverBudgetReason(): string | undefined {\n    const { maxTokens, maxCostUsd, maxDurationMs } = this.runBudget;\n    if (maxTokens !== undefined && this.tally.tokens >= maxTokens) {\n      return `${this.tally.tokens} tokens >= ${maxTokens} limit`;\n    }\n    if (maxCostUsd !== undefined && this.tally.costUsd >= maxCostUsd) {\n      return `cost ${this.tally.costUsd.toFixed(2)} USD >= ${maxCostUsd} USD limit`;\n    }\n    if (maxDurationMs !== undefined && Date.now() - this.tally.startedAt >= maxDurationMs) {\n      return `${Date.now() - this.tally.startedAt}ms >= ${maxDurationMs}ms limit`;\n    }\n    return undefined;\n  }\n\n  private findLastAssistantMessage(): AssistantMessage | undefined {\n    const messages = this.agent.state.messages;\n    for (let i = messages.length - 1; i >= 0; i--) {\n      const message = messages[i];\n      if (message.role === \"assistant\") return message;\n    }\n    return undefined;\n  }\n\n  /** Decide what to do after a run settles. Returns true when the agent should continue. */\n  private async handlePostRun(): Promise<boolean> {\n    // A run stopped for going over budget must not be retried or compacted back\n    // into another continuation — that would defeat the circuit breaker.\n    if (this.budgetExceededReason) return false;\n\n    const lastAssistant = this.findLastAssistantMessage();\n    if (!lastAssistant) return false;\n\n    if (lastAssistant.stopReason === \"error\") {\n      const contextWindow = this.agent.state.model.contextWindow || 0;\n      if (isContextOverflow(lastAssistant, contextWindow)) {\n        return this.handleOverflow();\n      }\n      if (this.isRetryableError(lastAssistant)) {\n        return this.prepareRetry(lastAssistant);\n      }\n      return false;\n    }\n\n    return this.checkThresholdCompaction(lastAssistant);\n  }\n\n  private async handleOverflow(): Promise<boolean> {\n    if (this.overflowRecoveryAttempted) {\n      await this.emit({\n        type: \"compaction_end\",\n        reason: \"overflow\",\n        aborted: false,\n        errorMessage:\n          \"Context overflow recovery failed after one compact-and-retry attempt. Try /new or a larger-context model.\",\n      });\n      return false;\n    }\n    this.overflowRecoveryAttempted = true;\n    this.dropTrailingErrorMessage();\n    return this.runCompaction(\"overflow\", true);\n  }\n\n  private async checkThresholdCompaction(lastAssistant: AssistantMessage): Promise<boolean> {\n    const settings = this.settings.compaction;\n    if (!settings.enabled) return false;\n    const contextWindow = this.agent.state.model.contextWindow || 0;\n    if (contextWindow <= 0) return false;\n\n    let contextTokens = lastAssistant.usage ? calculateContextTokens(lastAssistant.usage) : 0;\n    if (contextTokens === 0) {\n      const estimate = estimateContextTokens(this.agent.state.messages);\n      if (estimate.lastUsageIndex === null) return false;\n      contextTokens = estimate.tokens;\n    }\n\n    if (!shouldCompact(contextTokens, contextWindow, settings)) return false;\n    return this.runCompaction(\"threshold\", false);\n  }\n\n  private async runCompaction(reason: CompactionReason, willRetry: boolean): Promise<boolean> {\n    let started = false;\n    try {\n      const pathEntries = this.sessionStore.getBranch();\n      const preparation = getOrThrow(prepareCompaction(pathEntries, this.settings.compaction));\n      if (!preparation) return false;\n\n      await this.emit({ type: \"compaction_start\", reason });\n      started = true;\n      this.compactionAbortController = new AbortController();\n      const signal = this.compactionAbortController.signal;\n\n      const result = getOrThrow(\n        await compact(\n          preparation,\n          this.models.models,\n          this.agent.state.model,\n          undefined,\n          signal,\n          this.agent.state.thinkingLevel,\n        ),\n      );\n      if (signal.aborted) {\n        await this.emit({ type: \"compaction_end\", reason, aborted: true });\n        return false;\n      }\n\n      const entryId = this.sessionStore.appendCompaction(\n        result.summary,\n        result.firstKeptEntryId,\n        result.tokensBefore,\n        result.details,\n      );\n      const context = this.sessionStore.buildSessionContext();\n      this.agent.state.messages = context.messages;\n\n      if (this.extensions?.hasHandlers(\"session_compact\")) {\n        const entry = this.sessionStore.getEntry(entryId);\n        if (entry?.type === \"compaction\") {\n          await this.extensions.emit(\"session_compact\", {\n            entry: entry as CompactionEntry,\n            reason,\n          });\n        }\n      }\n\n      await this.emit({\n        type: \"compaction_end\",\n        reason,\n        result: {\n          summary: result.summary,\n          firstKeptEntryId: result.firstKeptEntryId,\n          tokensBefore: result.tokensBefore,\n        },\n        aborted: false,\n      });\n\n      if (willRetry) {\n        this.dropTrailingErrorMessage();\n        return true;\n      }\n      return this.agent.hasQueuedMessages();\n    } catch (error) {\n      const errorMessage = error instanceof Error ? error.message : \"compaction failed\";\n      if (started) {\n        await this.emit({\n          type: \"compaction_end\",\n          reason,\n          aborted: false,\n          errorMessage:\n            reason === \"overflow\"\n              ? `Context overflow recovery failed: ${errorMessage}`\n              : `Auto-compaction failed: ${errorMessage}`,\n        });\n      } else {\n        log.logWarning(\"Compaction preparation failed\", errorMessage);\n      }\n      return false;\n    } finally {\n      this.compactionAbortController = undefined;\n    }\n  }\n\n  private isRetryableError(message: AssistantMessage): boolean {\n    if (message.stopReason !== \"error\" || !message.errorMessage) return false;\n    return RETRYABLE_ERROR_PATTERN.test(message.errorMessage);\n  }\n\n  private async prepareRetry(message: AssistantMessage): Promise<boolean> {\n    const settings = this.settings.retry;\n    if (!settings.enabled) return false;\n\n    this.retryAttempt++;\n    if (this.retryAttempt > settings.maxRetries) {\n      this.retryAttempt--;\n      await this.emit({\n        type: \"auto_retry_end\",\n        success: false,\n        attempt: this.retryAttempt,\n        finalError: message.errorMessage,\n      });\n      this.retryAttempt = 0;\n      return false;\n    }\n\n    const delayMs = settings.baseDelayMs * 2 ** (this.retryAttempt - 1);\n    await this.emit({\n      type: \"auto_retry_start\",\n      attempt: this.retryAttempt,\n      maxAttempts: settings.maxRetries,\n      delayMs,\n      errorMessage: message.errorMessage || \"Unknown error\",\n    });\n\n    this.dropTrailingErrorMessage();\n\n    this.retryAbortController = new AbortController();\n    try {\n      await sleep(delayMs, this.retryAbortController.signal);\n    } catch {\n      const attempt = this.retryAttempt;\n      this.retryAttempt = 0;\n      await this.emit({\n        type: \"auto_retry_end\",\n        success: false,\n        attempt,\n        finalError: \"Retry cancelled\",\n      });\n      return false;\n    } finally {\n      this.retryAbortController = undefined;\n    }\n    return true;\n  }\n\n  private dropTrailingErrorMessage(): void {\n    const messages = this.agent.state.messages;\n    if (messages.length > 0 && messages[messages.length - 1].role === \"assistant\") {\n      this.agent.state.messages = messages.slice(0, -1);\n    }\n  }\n}\n"]}

package/dist/harness/session-store.d.ts

@@ -39,7 +39,10 @@ export declare class SessionStore {
     static create(path: string, cwd: string, options?: {
         id?: string;
     }): SessionStore;
-    getSessionFile(): string;
+    /** Create an ephemeral session that never writes a session file. */
+    static inMemory(cwd?: string): SessionStore;
+    getSessionFile(): string | undefined;
+    isPersisted(): boolean;
     getCwd(): string;
     /** Session header as read from disk, or null when the file had none. */
     getHeader(): SessionHeader | null;

package/dist/harness/session-store.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"session-store.d.ts","sourceRoot":"","sources":["../../src/harness/session-store.ts"],"names":[],"mappings":"AAiBA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEvE,OAAO,EAEL,KAAK,cAAc,EACnB,KAAK,YAAY,EACjB,KAAK,gBAAgB,EACrB,KAAK,aAAa,EACnB,MAAM,YAAY,CAAC;AAkBpB,mEAAmE;AACnE,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,gBAAgB,EAAE,CAmB3E;AAMD;;;;GAIG;AACH,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,gBAAgB,EAAE,CAM3E;AAQD;;;;;;GAMG;AACH,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,YAAY,CAAU;IAC9B,OAAO,CAAC,OAAO,CAAsB;IACrC,OAAO,CAAC,IAAI,CAAmC;IAC/C,OAAO,CAAC,MAAM,CAAuB;IAErC,OAAO,eAiBN;IAED;;;;;;;OAOG;IACH,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,YAAY,CAW5D;IAED,kFAAkF;IAClF,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,YAAY,CAUhF;IAED,cAAc,IAAI,MAAM,CAEvB;IAED,MAAM,IAAI,MAAM,CAEf;IAED,wEAAwE;IACxE,SAAS,IAAI,aAAa,GAAG,IAAI,CAEhC;IAED,8EAA8E;IAC9E,YAAY,IAAI,MAAM,CAErB;IAED,SAAS,IAAI,MAAM,GAAG,IAAI,CAEzB;IAED,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS,CAE7C;IAED,qEAAqE;IACrE,UAAU,IAAI,YAAY,EAAE,CAE3B;IAED,iDAAiD;IACjD,cAAc,IAAI,MAAM,GAAG,SAAS,CAMnC;IAED;;;OAGG;IACH,SAAS,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,YAAY,EAAE,CAUzC;IAED;;;OAGG;IACH,mBAAmB,IAAI,cAAc,CAEpC;IAED,kFAAkF;IAClF,aAAa,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM,CAE3C;IAED,0EAA0E;IAC1E,iBAAiB,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,MAAM,CAE5D;IAED,sEAAsE;IACtE,wBAAwB,CACtB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,GAAG,CAAC,WAAW,GAAG,YAAY,CAAC,EAAE,EAChD,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,OAAO,GAChB,MAAM,CAER;IAED,0CAA0C;IAC1C,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAEtC;IAED,8DAA8D;IAC9D,gBAAgB,CACd,OAAO,EAAE,MAAM,EACf,gBAAgB,EAAE,MAAM,EACxB,YAAY,EAAE,MAAM,EACpB,OAAO,CAAC,EAAE,OAAO,EACjB,QAAQ,CAAC,EAAE,OAAO,GACjB,MAAM,CASR;IAED,OAAO,CAAC,WAAW;IAcnB,OAAO,CAAC,OAAO;CAiBhB"}
+{"version":3,"file":"session-store.d.ts","sourceRoot":"","sources":["../../src/harness/session-store.ts"],"names":[],"mappings":"AAiBA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEvE,OAAO,EAEL,KAAK,cAAc,EACnB,KAAK,YAAY,EACjB,KAAK,gBAAgB,EACrB,KAAK,aAAa,EACnB,MAAM,YAAY,CAAC;AAkBpB,mEAAmE;AACnE,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,gBAAgB,EAAE,CAmB3E;AAMD;;;;GAIG;AACH,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,gBAAgB,EAAE,CAM3E;AAQD;;;;;;GAMG;AACH,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAgB;IAC5C,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,YAAY,CAAU;IAC9B,OAAO,CAAC,OAAO,CAAsB;IACrC,OAAO,CAAC,IAAI,CAAmC;IAC/C,OAAO,CAAC,MAAM,CAAuB;IAErC,OAAO,eAiBN;IAED;;;;;;;OAOG;IACH,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,YAAY,CAW5D;IAED,kFAAkF;IAClF,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,YAAY,CAUhF;IAED,oEAAoE;IACpE,MAAM,CAAC,QAAQ,CAAC,GAAG,SAAgB,GAAG,YAAY,CAEjD;IAED,cAAc,IAAI,MAAM,GAAG,SAAS,CAEnC;IAED,WAAW,IAAI,OAAO,CAErB;IAED,MAAM,IAAI,MAAM,CAEf;IAED,wEAAwE;IACxE,SAAS,IAAI,aAAa,GAAG,IAAI,CAEhC;IAED,8EAA8E;IAC9E,YAAY,IAAI,MAAM,CAErB;IAED,SAAS,IAAI,MAAM,GAAG,IAAI,CAEzB;IAED,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS,CAE7C;IAED,qEAAqE;IACrE,UAAU,IAAI,YAAY,EAAE,CAE3B;IAED,iDAAiD;IACjD,cAAc,IAAI,MAAM,GAAG,SAAS,CAMnC;IAED;;;OAGG;IACH,SAAS,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,YAAY,EAAE,CAUzC;IAED;;;OAGG;IACH,mBAAmB,IAAI,cAAc,CAEpC;IAED,kFAAkF;IAClF,aAAa,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM,CAE3C;IAED,0EAA0E;IAC1E,iBAAiB,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,MAAM,CAE5D;IAED,sEAAsE;IACtE,wBAAwB,CACtB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,GAAG,CAAC,WAAW,GAAG,YAAY,CAAC,EAAE,EAChD,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,OAAO,GAChB,MAAM,CAER;IAED,0CAA0C;IAC1C,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAEtC;IAED,8DAA8D;IAC9D,gBAAgB,CACd,OAAO,EAAE,MAAM,EACf,gBAAgB,EAAE,MAAM,EACxB,YAAY,EAAE,MAAM,EACpB,OAAO,CAAC,EAAE,OAAO,EACjB,QAAQ,CAAC,EAAE,OAAO,GACjB,MAAM,CASR;IAED,OAAO,CAAC,WAAW;IAcnB,OAAO,CAAC,OAAO;CAkBhB"}

package/dist/harness/session-store.js

@@ -83,7 +83,7 @@ export class SessionStore {
         this.entries = [];
         this.byId = new Map();
         this.leafId = null;
-        this.sessionFile = resolve(sessionFile);
+        this.sessionFile = sessionFile === null ? null : resolve(sessionFile);
         this.header = fileEntries.length > 0 && isSessionHeader(fileEntries[0]) ? fileEntries[0] : null;
         this.headerOnDisk = this.header !== null;
         this.sessionId = this.header?.id ?? randomUUID();
@@ -131,8 +131,15 @@ export class SessionStore {
         atomicWritePrivateFile(path, `${JSON.stringify(header)}\n`);
         return new SessionStore(path, cwd, [header]);
     }
+    /** Create an ephemeral session that never writes a session file. */
+    static inMemory(cwd = process.cwd()) {
+        return new SessionStore(null, cwd, []);
+    }
     getSessionFile() {
-        return this.sessionFile;
+        return this.sessionFile ?? undefined;
+    }
+    isPersisted() {
+        return this.sessionFile !== null;
     }
     getCwd() {
         return this.cwd;
@@ -228,6 +235,8 @@ export class SessionStore {
         return entry.id;
     }
     persist(entry) {
+        if (this.sessionFile === null)
+            return;
         if (!this.headerOnDisk) {
             this.header ??= {
                 type: "session",