@geminixiang/mikan 1.0.0-beta.9 → 1.0.0-beta.91

package/dist/harness/runner.js

@@ -1,440 +1,596 @@
-/**
- * MikanAgentSession — mikan's agent run loop.
- *
- * Owns a pi-agent-core `Agent` and layers on the behaviors mikan needs for
- * long-lived chat conversations:
- *
- * - message persistence into the conversation's {@link SessionStore}
- * - automatic context compaction (threshold and overflow recovery), using
- *   pi-agent-core's compaction pipeline over the session tree
- * - automatic retry with exponential backoff on transient provider errors
- * - extension hook dispatch ({@link ExtensionRegistry})
- *
- * Events mirror the agent's lifecycle events plus `compaction_start/_end`
- * and `auto_retry_start/_end`, preserving the event surface mikan's
- * platform adapters already render.
- */
-import { Agent, calculateContextTokens, compact, convertToLlm, estimateContextTokens, getOrThrow, prepareCompaction, shouldCompact, } from "@earendil-works/pi-agent-core";
-import { isContextOverflow, } from "@earendil-works/pi-ai";
+import { MikanModels } from "./models.js";
+import { SessionStore } from "../sessions/session-store.js";
+import { CONTROL_INPUT_CUSTOM_TYPE, RUN_CAUSE_CUSTOM_TYPE, } from "../sessions/types.js";
+import { MikanAgentSession, DEFAULT_EVENT_BUDGET } from "./session.js";
+import { runSubagent, DEFAULT_GLOBAL_SUBAGENT_SLOTS, SubagentSlotPool } from "./subagent.js";
+import { loadSubagentProfiles } from "./subagent-profiles.js";
+import { createMikanTools } from "./tools/index.js";
+import { createSubagentTool } from "./tools/subagent.js";
+import { createHistoryTool } from "./tools/history.js";
+import { adaptAgentTool } from "./tools/pi-tools.js";
+import { withSecretRedaction } from "./tools/secret-redaction.js";
+import { createSandboxExecutionEnv } from "./execution-env.js";
+import { mkdir } from "node:fs/promises";
+import { join } from "node:path";
+import { ActorExecutionResolver } from "./execution-resolver.js";
+import { assertSandboxSupportsWorkspacePolicy, createExecutor, getUnresolvedSandboxPathContext, } from "../sandbox/registry.js";
+import { resolveWorkspaceProjection } from "../office/projection.js";
+import { createHash } from "node:crypto";
+import { resolveConversationSettings } from "../settings/index.js";
+import { ensureDefaultOpenConnector } from "./open-connector.js";
+import { OfficeEventStore } from "../events/index.js";
+import { addLifecycleEvent, updateActiveSpanAttribution } from "../observability/index.js";
+import { conversationIdOf, isThreadSessionKey } from "../sessions/session-key.js";
+import { START_TASK_TOOL, TASK_STATUS_TOOL } from "./tools/task.js";
+import { loadMikanSkills } from "./skills.js";
+import { normalizeAttachRuntimePath, withStagedRuntimeFile, buildPromptPayload, buildSystemPrompt, buildTurnInstructions, getMemory, resolveTriggerAttribution, } from "./prompt.js";
+import { attachSessionEventHandlers, activateRunPresentation, createRunState, finalizeRunResponse, getFinalAssistantText, isEventTriggerAttribution, reportUsageSummary, } from "./presenter.js";
 import * as log from "../log.js";
-import { resolveHarnessSettings } from "./settings.js";
-// Transient provider failures worth retrying: overload/rate-limit responses,
-// 5xx statuses, and network/stream interruptions.
-const RETRYABLE_ERROR_PATTERN = /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;
-function sleep(ms, signal) {
-    return new Promise((resolve, reject) => {
-        if (signal.aborted) {
-            reject(new Error("aborted"));
-            return;
-        }
-        const timer = setTimeout(() => {
-            signal.removeEventListener("abort", onAbort);
-            resolve();
-        }, ms);
-        const onAbort = () => {
-            clearTimeout(timer);
-            reject(new Error("aborted"));
-        };
-        signal.addEventListener("abort", onAbort, { once: true });
+import { errorMessage } from "../unknown-values.js";
+const globalSubagentSlots = new SubagentSlotPool(DEFAULT_GLOBAL_SUBAGENT_SLOTS);
+async function createConfiguredAgentSession(params) {
+    const { workspaceDir, systemPrompt, model, thinkingLevel, tools, toolContext, sessionStore, models, } = params;
+    const loadedProfiles = loadSubagentProfiles(workspaceDir);
+    for (const diagnostic of loadedProfiles.diagnostics) {
+        log.logWarning(`Subagent profile ignored: ${diagnostic.path}`, diagnostic.message);
+    }
+    const availableToolNames = new Set(tools.map((tool) => tool.name));
+    const runnableProfiles = new Map([...loadedProfiles.profiles].filter(([, profile]) => profile.tools.every((tool) => availableToolNames.has(tool))));
+    let session;
+    const subagentTool = createSubagentTool(async (request, hooks) => runSubagent({
+        request,
+        onActivity: hooks?.onActivity,
+        defaultModel: model,
+        thinkingLevel,
+        models,
+        availableTools: tools.filter((tool) => ![START_TASK_TOOL, TASK_STATUS_TOOL].includes(tool.name)),
+        profiles: runnableProfiles,
+        slots: globalSubagentSlots,
+        toolContext,
+        parentEntries: await session.sessionStore.getContextEntries(),
+        onUsage: session.captureExternalUsageSink(),
+    }), runnableProfiles);
+    session = new MikanAgentSession({
+        systemPrompt,
+        model,
+        thinkingLevel,
+        tools: [...tools, adaptAgentTool(subagentTool)],
+        toolContext,
+        models,
+        sessionStore,
     });
+    return session;
 }
-export class MikanAgentSession {
-    constructor(options) {
-        this.listeners = new Set();
-        this.retryAttempt = 0;
-        this.overflowRecoveryAttempted = false;
-        this.tally = { tokens: 0, costUsd: 0, llmCalls: 0, startedAt: 0 };
-        this.runBudget = {};
-        this.models = options.models;
-        this.sessionStore = options.sessionStore;
-        this.settings = resolveHarnessSettings(options.settings);
-        this.extensions = options.extensions;
-        const tools = [...options.tools, ...(options.extensions?.getContributedTools() ?? [])];
-        this.agent = new Agent({
-            initialState: {
-                systemPrompt: options.systemPrompt,
-                model: options.model,
-                thinkingLevel: options.thinkingLevel,
-                tools,
-            },
-            convertToLlm,
-            streamFn: (model, context, streamOptions) => this.models.models.streamSimple(model, context, streamOptions),
-            beforeToolCall: async ({ toolCall, args }) => {
-                if (!this.extensions?.hasHandlers("tool_call"))
-                    return undefined;
-                const result = await this.extensions.emit("tool_call", {
-                    toolCallId: toolCall.id,
-                    toolName: toolCall.name,
-                    args,
-                });
-                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;
-            },
-        });
-        this.agent.subscribe(async (event) => {
-            await this.handleAgentEvent(event);
+function createRunnerExecutionContext(sandboxConfig, vaultManager, provisioner, workspace) {
+    const executionResolver = vaultManager && sandboxConfig.type !== "host"
+        ? new ActorExecutionResolver(sandboxConfig, vaultManager, provisioner, workspace)
+        : undefined;
+    let activeExecutor = executionResolver !== undefined
+        ? createExecutor({ type: "host" })
+        : createExecutor(sandboxConfig);
+    const executor = {
+        exec(command, options) {
+            return activeExecutor.exec(command, options);
+        },
+        readFile(path, options) {
+            return activeExecutor.readFile(path, options);
+        },
+        readFileBase64(path, options) {
+            return activeExecutor.readFileBase64(path, options);
+        },
+        writeFile(path, content, options) {
+            return activeExecutor.writeFile(path, content, options);
+        },
+        getWorkspacePath(hostPath) {
+            return activeExecutor.getWorkspacePath(hostPath);
+        },
+        getSandboxConfig() {
+            return activeExecutor.getSandboxConfig();
+        },
+        getPathContext(hostWorkspaceRoot) {
+            return activeExecutor.getPathContext(hostWorkspaceRoot);
+        },
+    };
+    return {
+        executor,
+        async resolveForRun(context) {
+            if (executionResolver) {
+                const decision = await executionResolver.resolve(context);
+                activeExecutor = decision.executor;
+                return {
+                    pathContext: decision.pathContext,
+                    projection: decision.projection,
+                };
+            }
+            const office = workspace.office(context.address);
+            const projection = resolveWorkspaceProjection(office);
+            assertSandboxSupportsWorkspacePolicy(sandboxConfig, projection.visibility, office.key);
+            return {
+                pathContext: executor.getPathContext(workspace.root),
+                projection,
+            };
+        },
+    };
+}
+const THREAD_SESSION_NAME_MAX_CHARS = 80;
+function buildThreadSessionName(message) {
+    const firstLine = message?.text
+        ?.split("\n")
+        .map((line) => line.trim())
+        .find(Boolean);
+    if (!firstLine)
+        return undefined;
+    const chars = Array.from(firstLine);
+    const title = chars.length > THREAD_SESSION_NAME_MAX_CHARS
+        ? `${chars.slice(0, THREAD_SESSION_NAME_MAX_CHARS).join("")}…`
+        : firstLine;
+    const userLabel = message?.userName || message?.user || "unknown";
+    return `[${userLabel}]: ${title}`;
+}
+async function ensureDefaultMcpServers(options) {
+    if (options.trustModel === "open-trigger")
+        return;
+    try {
+        await ensureDefaultOpenConnector({
+            office: options.office,
+            platformWorkspaceId: options.platformWorkspaceId,
+            defaultServer: options.openConnector,
+            signal: options.signal,
         });
     }
-    /** Conversation transcript currently held by the agent. */
-    get messages() {
-        return this.agent.state.messages;
+    catch (error) {
+        options.signal?.throwIfAborted();
+        log.logWarning(`[${options.office.address.conversationId}] OpenConnector default provisioning failed`, errorMessage(error));
     }
-    get model() {
-        return this.agent.state.model;
-    }
-    subscribe(listener) {
-        this.listeners.add(listener);
-        return () => this.listeners.delete(listener);
+}
+async function preparePromptContext(params) {
+    const { message, platform, office, executor, resolveForRun, session } = params;
+    const conversationId = office.address.conversationId;
+    const decision = await resolveForRun({
+        address: message.address,
+        userId: message.userId,
+        trustModel: platform.trustModel,
+    });
+    const { pathContext, projection } = decision;
+    const memory = getMemory(projection);
+    const conversationSkillLoad = loadMikanSkills(office, pathContext.runtimeWorkspaceRoot, projection);
+    const triggerAttribution = resolveTriggerAttribution(message);
+    const systemPrompt = buildSystemPrompt({
+        workspacePath: pathContext.runtimeWorkspaceRoot,
+        office,
+        memory,
+        sandboxConfig: executor.getSandboxConfig(),
+        platform,
+        skills: conversationSkillLoad.skills,
+        projection,
+        skippedSkillLinks: conversationSkillLoad.skippedSkillLinks,
+    });
+    session.setSystemPrompt(systemPrompt);
+    const promptHash = createHash("sha256").update(systemPrompt).digest("hex").slice(0, 8);
+    log.logInfo(`[${conversationId}] System prompt (base): ${systemPrompt.length} chars, sha ${promptHash}`);
+    return { pathContext, memory, systemPrompt, triggerAttribution };
+}
+function bindRunCapabilities(params, pathContext) {
+    const { message, responder, platform, office, executor, setEventContext, setSandboxContext, setUploadFunction, setImageUploadFunction, setReactFunction, bindTasks, bindPlatformToolPacks, } = params;
+    setEventContext({
+        platform: platform.name,
+        conversationId: office.address.conversationId,
+        conversationKind: message.conversationKind,
+        userId: message.userId,
+    });
+    setSandboxContext({ address: message.address, userId: message.userId });
+    setUploadFunction(async (filePath, title) => {
+        const runtimePath = normalizeAttachRuntimePath(filePath, pathContext.runtimeWorkspaceRoot);
+        await withStagedRuntimeFile(executor, runtimePath, (stagedPath) => responder.uploadFile(stagedPath, title));
+    });
+    setImageUploadFunction(async (hostPath, title) => {
+        await responder.uploadFile(hostPath, title);
+    });
+    bindTasks(responder);
+    setReactFunction(responder.react ? async (emoji) => responder.react(emoji) : null);
+    bindPlatformToolPacks({
+        conversationId: office.address.conversationId,
+        platformName: platform.name,
+        threadTs: message.threadTs,
+    });
+}
+async function prepareRunContext(params) {
+    const { message, platform, office, executor } = params;
+    const sessionConversation = conversationIdOf(message.sessionKey);
+    await mkdir(join(office.dir, "scratch"), { recursive: true });
+    const { pathContext, memory, systemPrompt, triggerAttribution } = await preparePromptContext(params);
+    bindRunCapabilities(params, pathContext);
+    log.logInfo(`Context sizes - system: ${systemPrompt.length} chars, memory: ${memory.length} chars`);
+    log.logInfo(`Channels: ${platform.channels.length}, Users: ${platform.users.length}`);
+    const { userMessage, imageAttachments } = await buildPromptPayload(message, pathContext.runtimeWorkspaceRoot, pathContext, (runtimePath) => executor.readFileBase64(runtimePath));
+    const turnInstructions = buildTurnInstructions(message.id.startsWith("event:"));
+    const finalUserMessage = turnInstructions ? `${turnInstructions}\n\n${userMessage}` : userMessage;
+    return {
+        sessionConversation,
+        userMessage: finalUserMessage,
+        imageAttachments,
+        triggerAttribution,
+    };
+}
+async function buildInitialSystemPrompt(params) {
+    const { office, pathContext, projection, sandboxConfig } = params;
+    const memory = getMemory(projection);
+    const { skills, skippedSkillLinks } = loadMikanSkills(office, pathContext.runtimeWorkspaceRoot, projection);
+    return buildSystemPrompt({
+        workspacePath: pathContext.runtimeWorkspaceRoot,
+        office,
+        memory,
+        sandboxConfig,
+        platform: {
+            name: "chat",
+            formattingGuide: "",
+            channels: [],
+            users: [],
+            trustModel: "membership",
+        },
+        skills,
+        projection,
+        skippedSkillLinks,
+    });
+}
+async function rollbackRunnerResource(label, cleanup) {
+    try {
+        await cleanup();
     }
-    /** Replace the agent transcript from the persisted session tree. */
-    reloadFromSession() {
-        const context = this.sessionStore.buildSessionContext();
-        if (context.messages.length > 0) {
-            this.agent.state.messages = context.messages;
-        }
-        return context.messages.length;
+    catch (error) {
+        log.logWarning(`Runner rollback failed to ${label}`, String(error));
     }
-    /**
-     * 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.
-     *
-     * @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.
-     */
-    async prompt(text, options) {
-        if (this.agent.state.isStreaming) {
-            throw new Error("Agent is already processing a prompt");
-        }
-        const model = this.agent.state.model;
-        await this.ensureAuthConfigured(model);
-        this.retryAttempt = 0;
-        this.overflowRecoveryAttempted = false;
-        this.runBudget = { ...this.settings.budget, ...options?.budget };
-        this.budgetExceededReason = undefined;
-        this.tally = { tokens: 0, costUsd: 0, llmCalls: 0, startedAt: Date.now() };
-        // 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 systemPrompt = this.agent.state.systemPrompt;
-        if (this.extensions?.hasHandlers("before_agent_start")) {
-            const result = await this.extensions.emit("before_agent_start", {
-                prompt: text,
-                images: options?.images,
-                systemPrompt,
-            });
-            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 ?? [])],
-            timestamp: Date.now(),
-        };
-        await this.agent.prompt([userMessage]);
-        while (await this.handlePostRun()) {
-            await this.agent.continue();
-        }
-        if (this.extensions?.hasHandlers("turn_end")) {
-            await this.extensions.emit("turn_end", { messages: this.agent.state.messages });
+}
+async function openRunnerSessionManager(params) {
+    const { office, sessionKey, threadRootMessage } = params;
+    const sessionManager = await SessionStore.open(office, sessionKey);
+    try {
+        const threadSessionName = buildThreadSessionName(threadRootMessage);
+        if (isThreadSessionKey(sessionKey) &&
+            threadSessionName &&
+            (await sessionManager.getSessionName()) !== threadSessionName) {
+            await sessionManager.setSessionName(threadSessionName);
         }
+        return sessionManager;
     }
-    /** Abort the active run, pending retry backoff, and in-flight compaction. */
-    abort() {
-        this.retryAbortController?.abort();
-        this.compactionAbortController?.abort();
-        this.agent.abort();
-    }
-    async ensureAuthConfigured(model) {
-        const auth = await this.models.getAuth(model);
-        if (!auth) {
-            throw new Error(`No credentials for provider "${model.provider}". Set the provider API key environment variable or add it to auth.json.`);
-        }
+    catch (error) {
+        await rollbackRunnerResource("close the session writer", () => sessionManager.close());
+        throw error;
     }
-    async emit(event) {
-        for (const listener of this.listeners) {
-            try {
-                await listener(event);
-            }
-            catch (err) {
-                log.logWarning("Harness event listener failed", err instanceof Error ? err.message : String(err));
-            }
-        }
+}
+async function createRunnerAgentSession(params) {
+    const { workspaceDir, systemPrompt, model, agentConfig, tools, toolContext, sessionManager, modelRegistry, signal, } = params;
+    const mcpTools = await sessionManager.connectMcp(agentConfig.mcpServers ?? {}, signal);
+    return createConfiguredAgentSession({
+        workspaceDir,
+        systemPrompt,
+        model,
+        thinkingLevel: agentConfig.thinkingLevel,
+        tools: [...tools, ...mcpTools.map(withSecretRedaction)],
+        toolContext,
+        sessionStore: sessionManager,
+        models: modelRegistry,
+    });
+}
+async function runPreparedTurn(params) {
+    const { prepared, presentation, message, responder, platform, runState, session, model, agentConfig, sessionUuid, conversationId, office, sessionView, } = params;
+    if (runState.logCtx) {
+        log.logAgentRunStart(runState.logCtx, model.provider, model.id, model.name);
     }
-    async handleAgentEvent(event) {
-        await this.emit(event);
-        if (event.type !== "message_end")
-            return;
-        const message = event.message;
-        if (message.role === "user" || message.role === "assistant" || message.role === "toolResult") {
-            this.sessionStore.appendMessage(message);
-        }
-        else if (message.role === "custom") {
-            const custom = message;
-            this.sessionStore.appendCustomMessageEntry(custom.customType, custom.content, custom.display, custom.details);
-        }
-        if (this.extensions?.hasHandlers("message_end")) {
-            await this.extensions.emit("message_end", { message });
-        }
-        if (message.role === "assistant") {
-            this.recordUsage(message);
-            await this.enforceBudget();
-            if (message.stopReason !== "error") {
-                this.overflowRecoveryAttempted = false;
-                if (this.retryAttempt > 0) {
-                    await this.emit({ type: "auto_retry_end", success: true, attempt: this.retryAttempt });
-                    this.retryAttempt = 0;
-                }
+    updateActiveSpanAttribution({
+        provider: model.provider,
+        model: model.id,
+        channel_id: prepared.sessionConversation,
+        session_id: sessionUuid,
+        "mikan.input.message_count": 1,
+        "mikan.input.characters": prepared.userMessage.length,
+        "mikan.input.attachment_count": message.attachments?.length ?? 0,
+        "mikan.input.image_count": prepared.imageAttachments.length,
+        "mikan.input.has_text": prepared.userMessage.length > 0,
+    });
+    addLifecycleEvent("agent.prompt.sent", {
+        provider: model.provider,
+        model: agentConfig.model,
+        channel_id: prepared.sessionConversation,
+        session_id: sessionUuid,
+        attachment_count: message.attachments?.length ?? 0,
+        image_attachment_count: prepared.imageAttachments.length,
+    });
+    const isEventRun = message.id.startsWith("event:");
+    await session.sessionStore.appendCustomEntry(RUN_CAUSE_CUSTOM_TYPE, { messageId: message.id });
+    await session.prompt(prepared.userMessage, {
+        allowTaskHandoff: responder.startTask !== undefined,
+        allowTaskStatus: responder.getTaskStatus !== undefined,
+        images: prepared.imageAttachments.length > 0 ? prepared.imageAttachments : undefined,
+        budget: isEventRun ? DEFAULT_EVENT_BUDGET : undefined,
+    });
+    await presentation.wait();
+    if (session.getLastRunStats().status === "aborted")
+        runState.stopReason = "aborted";
+    const sessionViewTokenStore = sessionView?.tokenStore;
+    const sessionViewPortalBaseUrl = sessionView?.portalBaseUrl;
+    let sessionViewLink;
+    const createSessionViewLink = sessionViewTokenStore && sessionViewPortalBaseUrl
+        ? () => {
+            if (!sessionViewLink) {
+                const token = sessionViewTokenStore.create({
+                    platform: platform.name,
+                    platformUserId: message.userId,
+                    conversationId,
+                    sessionKey: message.sessionKey,
+                    office,
+                    platformUserName: message.userName,
+                });
+                sessionViewLink = `${sessionViewPortalBaseUrl}/session?token=${token.token}`;
             }
+            return sessionViewLink;
         }
+        : undefined;
+    await finalizeRunResponse(responder, session, runState, {
+        initialTask: message.id.startsWith("task:"),
+        triggerSessionLink: isEventTriggerAttribution(prepared.triggerAttribution)
+            ? createSessionViewLink?.()
+            : undefined,
+        createOverflowLink: createSessionViewLink,
+        platform: platform.name,
+        model,
+        sessionConversation: prepared.sessionConversation,
+        sessionUuid,
+    });
+    await reportUsageSummary({
+        session,
+        runState,
+        responder,
+        platform,
+        model,
+        sessionConversation: prepared.sessionConversation,
+        sessionUuid,
+        waitForQueue: presentation.wait,
+    });
+    return {
+        stopReason: runState.stopReason,
+        errorMessage: runState.errorMessage,
+        finalText: getFinalAssistantText(session),
+    };
+}
+async function steerRun(session, activeMessage, message) {
+    if (!activeMessage)
+        return false;
+    await session.sessionStore.appendCustomEntry(CONTROL_INPUT_CUSTOM_TYPE, {
+        messageId: message.id,
+    });
+    if (!session.isActiveRun)
+        throw new Error("Task is preparing or settling. Please retry this update shortly.");
+    if (message.userId !== activeMessage.userId)
+        throw new Error("Only the task's current actor can guide this run.");
+    if (message.attachments?.length)
+        throw new Error("Stop the task before continuing with new attachments.");
+    if (!(await session.steer(`[${message.userName ?? message.userId}]: ${message.text}`))) {
+        throw new Error("Task settled while receiving this update. Please send it again to continue.");
     }
-    /** Fold one assistant turn's usage into the running per-run tally. */
-    recordUsage(message) {
-        this.tally.llmCalls += 1;
-        const usage = message.usage;
-        if (!usage)
-            return;
-        this.tally.tokens += usage.input + usage.output + usage.cacheRead + usage.cacheWrite;
-        this.tally.costUsd += usage.cost?.total ?? 0;
-    }
-    /** Compare the tally against the run budget; abort the run when a cap is exceeded. */
-    async enforceBudget() {
-        if (this.budgetExceededReason)
-            return;
-        const reason = this.overBudgetReason();
-        if (!reason)
-            return;
-        this.budgetExceededReason = reason;
-        await this.emit({
-            type: "budget_exceeded",
-            reason,
-            tokens: this.tally.tokens,
-            costUsd: this.tally.costUsd,
-            llmCalls: this.tally.llmCalls,
-            durationMs: Date.now() - this.tally.startedAt,
-        });
-        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) {
-            return `${this.tally.llmCalls} LLM calls >= ${maxLlmCalls} limit`;
-        }
-        if (maxTokens !== undefined && this.tally.tokens >= maxTokens) {
-            return `${this.tally.tokens} tokens >= ${maxTokens} limit`;
-        }
-        if (maxCostUsd !== undefined && this.tally.costUsd >= maxCostUsd) {
-            return `cost ${this.tally.costUsd.toFixed(2)} USD >= ${maxCostUsd} USD limit`;
-        }
-        if (maxDurationMs !== undefined && Date.now() - this.tally.startedAt >= maxDurationMs) {
-            return `${Date.now() - this.tally.startedAt}ms >= ${maxDurationMs}ms limit`;
-        }
-        return undefined;
-    }
-    findLastAssistantMessage() {
-        const messages = this.agent.state.messages;
-        for (let i = messages.length - 1; i >= 0; i--) {
-            const message = messages[i];
-            if (message.role === "assistant")
-                return message;
-        }
-        return undefined;
+    return true;
+}
+async function publishRunLifecycle(publish, message, run) {
+    publish?.({
+        type: "run_started",
+        userName: message.userName ?? message.userId,
+        text: message.text,
+    });
+    try {
+        return await run();
     }
-    /** Decide what to do after a run settles. Returns true when the agent should continue. */
-    async handlePostRun() {
-        // A run stopped for going over budget must not be retried or compacted back
-        // into another continuation — that would defeat the circuit breaker.
-        if (this.budgetExceededReason)
-            return false;
-        const lastAssistant = this.findLastAssistantMessage();
-        if (!lastAssistant)
-            return false;
-        if (lastAssistant.stopReason === "error") {
-            const contextWindow = this.agent.state.model.contextWindow || 0;
-            if (isContextOverflow(lastAssistant, contextWindow)) {
-                return this.handleOverflow();
-            }
-            if (this.isRetryableError(lastAssistant)) {
-                return this.prepareRetry(lastAssistant);
-            }
-            return false;
-        }
-        return this.checkThresholdCompaction(lastAssistant);
+    finally {
+        publish?.({ type: "run_ended" });
     }
-    async handleOverflow() {
-        if (this.overflowRecoveryAttempted) {
-            await this.emit({
-                type: "compaction_end",
-                reason: "overflow",
-                aborted: false,
-                errorMessage: "Context overflow recovery failed after one compact-and-retry attempt. Try /new or a larger-context model.",
+}
+function prepareRunnerTurn(params, message, responder, platform) {
+    const { office, executor, resolveForRun, session, toolBindings } = params;
+    return prepareRunContext({
+        message,
+        responder,
+        platform,
+        office,
+        executor,
+        resolveForRun,
+        session,
+        setEventContext: toolBindings.setEventContext,
+        setSandboxContext: toolBindings.setSandboxContext,
+        setUploadFunction: toolBindings.setUploadFunction,
+        setImageUploadFunction: toolBindings.setImageUploadFunction,
+        setReactFunction: toolBindings.setReactFunction,
+        bindTasks: toolBindings.bindTasks,
+        bindPlatformToolPacks: toolBindings.bindPlatformToolPacks,
+    });
+}
+function createRunnerInterface(params) {
+    const { conversationId, sessionKey, office, sessionUuid, sessionView, runEvents, runState, session, model, agentConfig, sessionManager, chatSessionManager, } = params;
+    const publishRunEvent = runEvents
+        ? (event) => runEvents.publish(office.address, sessionKey, event)
+        : undefined;
+    let activeMessage;
+    let stopped = false;
+    return {
+        steer: (message) => steerRun(session, activeMessage, message),
+        async syncChatHistory(currentMessageId) {
+            await chatSessionManager.syncSessionManager({
+                office,
+                sessionKey,
+                sessionManager,
+                currentMessageId,
             });
-            return false;
-        }
-        this.overflowRecoveryAttempted = true;
-        this.dropTrailingErrorMessage();
-        return this.runCompaction("overflow", true);
-    }
-    async checkThresholdCompaction(lastAssistant) {
-        const settings = this.settings.compaction;
-        if (!settings.enabled)
-            return false;
-        const contextWindow = this.agent.state.model.contextWindow || 0;
-        if (contextWindow <= 0)
-            return false;
-        let contextTokens = lastAssistant.usage ? calculateContextTokens(lastAssistant.usage) : 0;
-        if (contextTokens === 0) {
-            const estimate = estimateContextTokens(this.agent.state.messages);
-            if (estimate.lastUsageIndex === null)
-                return false;
-            contextTokens = estimate.tokens;
-        }
-        if (!shouldCompact(contextTokens, contextWindow, settings))
-            return false;
-        return this.runCompaction("threshold", false);
-    }
-    async runCompaction(reason, willRetry) {
-        let started = false;
-        try {
-            const pathEntries = this.sessionStore.getBranch();
-            const preparation = getOrThrow(prepareCompaction(pathEntries, this.settings.compaction));
-            if (!preparation)
-                return false;
-            await this.emit({ type: "compaction_start", reason });
-            started = true;
-            this.compactionAbortController = new AbortController();
-            const signal = this.compactionAbortController.signal;
-            const result = getOrThrow(await compact(preparation, this.models.models, this.agent.state.model, undefined, signal, this.agent.state.thinkingLevel));
-            if (signal.aborted) {
-                await this.emit({ type: "compaction_end", reason, aborted: true });
-                return false;
-            }
-            const entryId = this.sessionStore.appendCompaction(result.summary, result.firstKeptEntryId, result.tokensBefore, result.details);
-            const context = this.sessionStore.buildSessionContext();
-            this.agent.state.messages = context.messages;
-            if (this.extensions?.hasHandlers("session_compact")) {
-                const entry = this.sessionStore.getEntry(entryId);
-                if (entry?.type === "compaction") {
-                    await this.extensions.emit("session_compact", {
-                        entry: entry,
-                        reason,
+        },
+        async run(message, responder, platform) {
+            activeMessage = message;
+            stopped = false;
+            let presentation;
+            return publishRunLifecycle(publishRunEvent, message, async () => {
+                try {
+                    const prepared = await prepareRunnerTurn(params, message, responder, platform);
+                    if (stopped)
+                        return { stopReason: "aborted" };
+                    presentation = activateRunPresentation(runState, {
+                        responder,
+                        sessionConversation: prepared.sessionConversation,
+                        userName: message.userName,
+                        sessionUuid,
+                        triggerAttribution: prepared.triggerAttribution,
+                        publishRunEvent,
+                    });
+                    return await runPreparedTurn({
+                        prepared,
+                        presentation,
+                        message,
+                        responder,
+                        platform,
+                        runState,
+                        session,
+                        model,
+                        agentConfig,
+                        sessionUuid,
+                        conversationId,
+                        office,
+                        sessionView,
                     });
                 }
-            }
-            await this.emit({
-                type: "compaction_end",
-                reason,
-                result: {
-                    summary: result.summary,
-                    firstKeptEntryId: result.firstKeptEntryId,
-                    tokensBefore: result.tokensBefore,
-                },
-                aborted: false,
-            });
-            if (willRetry) {
-                this.dropTrailingErrorMessage();
-                return true;
-            }
-            return this.agent.hasQueuedMessages();
-        }
-        catch (error) {
-            const errorMessage = error instanceof Error ? error.message : "compaction failed";
-            if (started) {
-                await this.emit({
-                    type: "compaction_end",
-                    reason,
-                    aborted: false,
-                    errorMessage: reason === "overflow"
-                        ? `Context overflow recovery failed: ${errorMessage}`
-                        : `Auto-compaction failed: ${errorMessage}`,
-                });
-            }
-            else {
-                log.logWarning("Compaction preparation failed", errorMessage);
-            }
-            return false;
-        }
-        finally {
-            this.compactionAbortController = undefined;
-        }
-    }
-    isRetryableError(message) {
-        if (message.stopReason !== "error" || !message.errorMessage)
-            return false;
-        return RETRYABLE_ERROR_PATTERN.test(message.errorMessage);
-    }
-    async prepareRetry(message) {
-        const settings = this.settings.retry;
-        if (!settings.enabled)
-            return false;
-        this.retryAttempt++;
-        if (this.retryAttempt > settings.maxRetries) {
-            this.retryAttempt--;
-            await this.emit({
-                type: "auto_retry_end",
-                success: false,
-                attempt: this.retryAttempt,
-                finalError: message.errorMessage,
+                finally {
+                    activeMessage = undefined;
+                    presentation?.dispose();
+                }
             });
-            this.retryAttempt = 0;
-            return false;
-        }
-        const delayMs = settings.baseDelayMs * 2 ** (this.retryAttempt - 1);
-        await this.emit({
-            type: "auto_retry_start",
-            attempt: this.retryAttempt,
-            maxAttempts: settings.maxRetries,
-            delayMs,
-            errorMessage: message.errorMessage || "Unknown error",
+        },
+        abort() {
+            stopped = true;
+            session.abort();
+        },
+        async dispose() {
+            await sessionManager.close();
+        },
+        getCurrentStep() {
+            const first = runState.pendingTools.values().next().value;
+            if (!first)
+                return undefined;
+            return {
+                toolName: first.toolName,
+                label: first.args?.label,
+            };
+        },
+    };
+}
+async function finishRunnerCreation(params) {
+    const { options, conversationId, workspaceDir, executor, resolveForRun, model, modelRegistry, agentConfig, systemPrompt, sessionManager, toolBindings, platformToolRoles, toolContext, } = params;
+    const { sessionKey, office, sessionView, chatHistory, runEvents } = options;
+    try {
+        const sessionUuid = sessionManager.getSessionId();
+        const session = await createRunnerAgentSession({
+            workspaceDir,
+            systemPrompt,
+            model,
+            agentConfig,
+            tools: [
+                ...toolBindings.tools,
+                withSecretRedaction(adaptAgentTool(createHistoryTool({ office, sessionKey }))),
+            ],
+            toolContext,
+            sessionManager,
+            modelRegistry,
+            conversationId,
+            signal: options.signal,
+        });
+        options.signal?.throwIfAborted();
+        const runState = createRunState();
+        attachSessionEventHandlers({ session, runState, model, agentConfig, platformToolRoles });
+        return createRunnerInterface({
+            conversationId,
+            sessionKey,
+            office,
+            sessionUuid,
+            sessionView,
+            runEvents,
+            runState,
+            executor,
+            resolveForRun,
+            session,
+            model,
+            agentConfig,
+            sessionManager,
+            chatSessionManager: chatHistory,
+            toolBindings,
         });
-        this.dropTrailingErrorMessage();
-        this.retryAbortController = new AbortController();
-        try {
-            await sleep(delayMs, this.retryAbortController.signal);
-        }
-        catch {
-            const attempt = this.retryAttempt;
-            this.retryAttempt = 0;
-            await this.emit({
-                type: "auto_retry_end",
-                success: false,
-                attempt,
-                finalError: "Retry cancelled",
-            });
-            return false;
-        }
-        finally {
-            this.retryAbortController = undefined;
-        }
-        return true;
     }
-    dropTrailingErrorMessage() {
-        const messages = this.agent.state.messages;
-        if (messages.length > 0 && messages[messages.length - 1].role === "assistant") {
-            this.agent.state.messages = messages.slice(0, -1);
-        }
+    catch (error) {
+        await rollbackRunnerResource("close the session writer", () => sessionManager.close());
+        throw error;
     }
 }
+export async function createRunner(options) {
+    options.signal?.throwIfAborted();
+    const { sandboxConfig, sessionKey, office, trustModel, sessionScope, vaultManager, provisioner, resourceController, platformToolPackFactories, } = options;
+    const conversationId = office.address.conversationId;
+    const workspaceDir = office.workspace.root;
+    await ensureDefaultMcpServers({
+        office,
+        trustModel,
+        platformWorkspaceId: options.platformWorkspaceId,
+        openConnector: options.openConnector,
+        signal: options.signal,
+    });
+    options.signal?.throwIfAborted();
+    const resolvedAgentConfig = resolveConversationSettings(office);
+    const agentConfig = {
+        ...resolvedAgentConfig,
+        mcpServers: trustModel === "open-trigger" ? {} : (resolvedAgentConfig.mcpServers ?? {}),
+    };
+    const projection = resolveWorkspaceProjection(office);
+    assertSandboxSupportsWorkspacePolicy(sandboxConfig, projection.visibility, office.key);
+    const { executor, resolveForRun } = createRunnerExecutionContext(sandboxConfig, vaultManager, provisioner, office.workspace);
+    const pathContext = getUnresolvedSandboxPathContext(sandboxConfig, workspaceDir);
+    const toolContext = {
+        env: createSandboxExecutionEnv(executor, sandboxConfig.type, pathContext.runtimeWorkspaceRoot),
+    };
+    const modelRegistry = options.models ?? MikanModels.create();
+    if (modelRegistry.getError()) {
+        log.logWarning("models.json load error", modelRegistry.getError());
+    }
+    const model = modelRegistry.resolve(agentConfig.provider, agentConfig.model);
+    const platformToolPacks = (platformToolPackFactories ?? []).map((createPack) => createPack());
+    const toolBindings = createMikanTools(executor, new OfficeEventStore(office, options.eventScheduler), { sandbox: sandboxConfig, resourceController: resourceController ?? provisioner }, platformToolPacks, {
+        model,
+        getApiKey: () => modelRegistry.getApiKeyForProvider(model.provider),
+        outputDir: office.dir,
+    });
+    const systemPrompt = await buildInitialSystemPrompt({
+        office,
+        pathContext,
+        projection,
+        sandboxConfig,
+    });
+    options.signal?.throwIfAborted();
+    const { threadRootMessage } = sessionScope;
+    const sessionManager = await openRunnerSessionManager({
+        office,
+        sessionKey,
+        threadRootMessage,
+    });
+    return finishRunnerCreation({
+        options,
+        conversationId,
+        workspaceDir,
+        executor,
+        resolveForRun,
+        model,
+        modelRegistry,
+        agentConfig,
+        systemPrompt,
+        sessionManager,
+        toolBindings,
+        platformToolRoles: collectPlatformToolRoles(platformToolPacks),
+        toolContext,
+    });
+}
+function collectPlatformToolRoles(packs) {
+    return {
+        platformTools: new Set(packs.flatMap((pack) => pack.tools.map((tool) => tool.name))),
+        finalResponseTools: new Set(packs.flatMap((pack) => pack.finalResponseTools ?? [])),
+    };
+}
 //# sourceMappingURL=runner.js.map