@sema-agent/core 1.279.0

package/dist/engine/harness/agent-harness.js

@@ -0,0 +1,932 @@
+import { runAgentLoop } from "../loop/agent-loop.js";
+import { resolveAgentCoreStreamFn } from "../loop/runtime-deps.js";
+import { convertToLlm } from "./messages.js";
+import { AgentHarnessError, CompactionError, SessionError, toError, } from "./types.js";
+function createUserMessage(text, images) {
+    const content = [{ type: "text", text }];
+    if (images) {
+        content.push(...images);
+    }
+    return { role: "user", content, timestamp: Date.now() };
+}
+function createFailureMessage(model, error, aborted) {
+    return {
+        role: "assistant",
+        content: [{ type: "text", text: "" }],
+        api: model.api,
+        provider: model.provider,
+        model: model.id,
+        stopReason: aborted ? "aborted" : "error",
+        errorMessage: error instanceof Error ? error.message : String(error),
+        timestamp: Date.now(),
+        usage: {
+            input: 0,
+            output: 0,
+            cacheRead: 0,
+            cacheWrite: 0,
+            totalTokens: 0,
+            cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+        },
+    };
+}
+function cloneStreamOptions(streamOptions) {
+    return {
+        ...streamOptions,
+        headers: streamOptions?.headers ? { ...streamOptions.headers } : undefined,
+        metadata: streamOptions?.metadata ? { ...streamOptions.metadata } : undefined,
+    };
+}
+function mergeHeaders(...headers) {
+    const merged = {};
+    let hasHeaders = false;
+    for (const entry of headers) {
+        if (!entry) {
+            continue;
+        }
+        Object.assign(merged, entry);
+        hasHeaders = true;
+    }
+    return hasHeaders ? merged : undefined;
+}
+function applyStreamOptionsPatch(base, patch) {
+    const result = cloneStreamOptions(base);
+    if (!patch) {
+        return result;
+    }
+    if (Object.hasOwn(patch, "transport")) {
+        result.transport = patch.transport;
+    }
+    if (Object.hasOwn(patch, "timeoutMs")) {
+        result.timeoutMs = patch.timeoutMs;
+    }
+    if (Object.hasOwn(patch, "maxRetries")) {
+        result.maxRetries = patch.maxRetries;
+    }
+    if (Object.hasOwn(patch, "maxRetryDelayMs")) {
+        result.maxRetryDelayMs = patch.maxRetryDelayMs;
+    }
+    if (Object.hasOwn(patch, "cacheRetention")) {
+        result.cacheRetention = patch.cacheRetention;
+    }
+    if (Object.hasOwn(patch, "headers")) {
+        if (patch.headers === undefined) {
+            result.headers = undefined;
+        }
+        else {
+            const headers = { ...result.headers };
+            for (const [key, value] of Object.entries(patch.headers)) {
+                if (value === undefined) {
+                    delete headers[key];
+                }
+                else {
+                    headers[key] = value;
+                }
+            }
+            result.headers = Object.keys(headers).length > 0 ? headers : undefined;
+        }
+    }
+    if (Object.hasOwn(patch, "metadata")) {
+        if (patch.metadata === undefined) {
+            result.metadata = undefined;
+        }
+        else {
+            const metadata = { ...result.metadata };
+            for (const [key, value] of Object.entries(patch.metadata)) {
+                if (value === undefined) {
+                    delete metadata[key];
+                }
+                else {
+                    metadata[key] = value;
+                }
+            }
+            result.metadata = Object.keys(metadata).length > 0 ? metadata : undefined;
+        }
+    }
+    return result;
+}
+const SUBSCRIBER_EVENT_TYPE = "*";
+function normalizeHarnessError(error, fallbackCode) {
+    if (error instanceof AgentHarnessError) {
+        return error;
+    }
+    const cause = toError(error);
+    if (cause instanceof SessionError) {
+        return new AgentHarnessError("session", cause.message, cause);
+    }
+    if (cause instanceof CompactionError) {
+        return new AgentHarnessError("compaction", cause.message, cause);
+    }
+    return new AgentHarnessError(fallbackCode, cause.message, cause);
+}
+function normalizeHookError(error) {
+    return normalizeHarnessError(error, "hook");
+}
+export class AgentHarness {
+    env;
+    session;
+    phase = "idle";
+    _stopAfterTurn = false;
+    runAbortController;
+    runPromise;
+    pendingSessionWrites = [];
+    model;
+    thinkingLevel;
+    systemPrompt;
+    streamOptions;
+    getApiKeyAndHeaders;
+    runtime;
+    resources;
+    tools = new Map();
+    activeToolNames;
+    steerQueue = [];
+    steeringQueueMode;
+    maxOutputTokens;
+    maxOutputTokensPerCall;
+    callDeadlineMsPerCall;
+    loopTrace;
+    resilience;
+    maxToolConcurrency;
+    streamingToolExecution;
+    followUpQueue = [];
+    followUpQueueMode;
+    nextTurnQueue = [];
+    loopRecovery;
+    stopGate;
+    handlers = new Map();
+    constructor(options) {
+        this.env = options.env;
+        this.session = options.session;
+        this.resources = options.resources ?? {};
+        this.streamOptions = cloneStreamOptions(options.streamOptions);
+        this.systemPrompt = options.systemPrompt;
+        this.getApiKeyAndHeaders = options.getApiKeyAndHeaders;
+        this.runtime = options.runtime;
+        for (const tool of options.tools ?? []) {
+            this.tools.set(tool.name, tool);
+        }
+        this.model = options.model;
+        this.thinkingLevel = options.thinkingLevel ?? "off";
+        this.activeToolNames =
+            options.activeToolNames ?? (options.tools ?? []).map((tool) => tool.name);
+        this.steeringQueueMode = options.steeringMode ?? "one-at-a-time";
+        this.maxOutputTokens = options.maxOutputTokens;
+        this.maxOutputTokensPerCall = options.maxOutputTokensPerCall;
+        this.callDeadlineMsPerCall = options.callDeadlineMsPerCall;
+        this.loopTrace = options.loopTrace;
+        this.resilience = options.resilience;
+        this.maxToolConcurrency = options.maxToolConcurrency;
+        this.streamingToolExecution = options.streamingToolExecution;
+        this.followUpQueueMode = options.followUpMode ?? "one-at-a-time";
+    }
+    getHandlers(type) {
+        return this.handlers.get(type);
+    }
+    async emitOwn(event, signal) {
+        for (const listener of this.getHandlers(SUBSCRIBER_EVENT_TYPE) ?? []) {
+            try {
+                await listener(event, signal);
+            }
+            catch (error) {
+                throw normalizeHookError(error);
+            }
+        }
+    }
+    async emitAny(event, signal) {
+        for (const listener of this.getHandlers(SUBSCRIBER_EVENT_TYPE) ?? []) {
+            try {
+                await listener(event, signal);
+            }
+            catch (error) {
+                throw normalizeHookError(error);
+            }
+        }
+    }
+    async emitHook(event) {
+        const handlers = this.getHandlers(event.type);
+        if (!handlers || handlers.size === 0) {
+            return undefined;
+        }
+        let lastResult;
+        for (const handler of handlers) {
+            try {
+                const result = (await handler(event));
+                if (result !== undefined) {
+                    lastResult = result;
+                }
+            }
+            catch (error) {
+                throw normalizeHookError(error);
+            }
+        }
+        return lastResult;
+    }
+    async emitBeforeProviderRequest(model, sessionId, streamOptions) {
+        const handlers = this.getHandlers("before_provider_request");
+        let current = cloneStreamOptions(streamOptions);
+        if (!handlers || handlers.size === 0) {
+            return current;
+        }
+        for (const handler of handlers) {
+            try {
+                const result = (await handler({
+                    type: "before_provider_request",
+                    model,
+                    sessionId,
+                    streamOptions: cloneStreamOptions(current),
+                }));
+                if (result?.streamOptions) {
+                    current = applyStreamOptionsPatch(current, result.streamOptions);
+                }
+            }
+            catch (error) {
+                throw normalizeHookError(error);
+            }
+        }
+        return current;
+    }
+    async emitBeforeProviderPayload(model, payload) {
+        const handlers = this.getHandlers("before_provider_payload");
+        let current = payload;
+        if (!handlers || handlers.size === 0) {
+            return current;
+        }
+        for (const handler of handlers) {
+            try {
+                const result = (await handler({
+                    type: "before_provider_payload",
+                    model,
+                    payload: current,
+                }));
+                if (result !== undefined) {
+                    current = result.payload;
+                }
+            }
+            catch (error) {
+                throw normalizeHookError(error);
+            }
+        }
+        return current;
+    }
+    async emitQueueUpdate() {
+        await this.emitOwn({
+            type: "queue_update",
+            steer: [...this.steerQueue],
+            followUp: [...this.followUpQueue],
+            nextTurn: [...this.nextTurnQueue],
+        });
+    }
+    startRunPromise() {
+        let finish = () => { };
+        this.runPromise = new Promise((resolve) => {
+            finish = resolve;
+        });
+        return () => {
+            this.runPromise = undefined;
+            finish();
+        };
+    }
+    async createTurnState() {
+        const context = await this.session.buildContext();
+        const resources = this.getResources();
+        const sessionMetadata = await this.session.getMetadata();
+        const tools = [...this.tools.values()];
+        const activeTools = this.activeToolNames
+            .map((name) => this.tools.get(name))
+            .filter((tool) => tool !== undefined);
+        let systemPrompt = "You are a helpful assistant.";
+        if (typeof this.systemPrompt === "string") {
+            systemPrompt = this.systemPrompt;
+        }
+        else if (this.systemPrompt) {
+            systemPrompt = await this.systemPrompt({
+                env: this.env,
+                session: this.session,
+                model: this.model,
+                thinkingLevel: this.thinkingLevel,
+                activeTools,
+                resources,
+            });
+        }
+        return {
+            messages: context.messages,
+            resources,
+            streamOptions: cloneStreamOptions(this.streamOptions),
+            sessionId: sessionMetadata.id,
+            systemPrompt,
+            model: this.model,
+            thinkingLevel: this.thinkingLevel,
+            tools,
+            activeTools,
+        };
+    }
+    createContext(turnState, systemPrompt) {
+        return {
+            systemPrompt: systemPrompt ?? turnState.systemPrompt,
+            messages: turnState.messages.slice(),
+            tools: turnState.activeTools.slice(),
+        };
+    }
+    createStreamFn(getTurnState) {
+        return async (model, context, streamOptions) => {
+            const turnState = getTurnState();
+            const auth = await this.getApiKeyAndHeaders?.(model);
+            const snapshotOptions = {
+                ...turnState.streamOptions,
+                headers: mergeHeaders(turnState.streamOptions.headers, auth?.headers),
+            };
+            const requestOptions = await this.emitBeforeProviderRequest(model, turnState.sessionId, snapshotOptions);
+            return resolveAgentCoreStreamFn(this.runtime)(model, context, {
+                cacheRetention: requestOptions.cacheRetention,
+                headers: requestOptions.headers,
+                maxRetries: requestOptions.maxRetries,
+                maxRetryDelayMs: requestOptions.maxRetryDelayMs,
+                metadata: requestOptions.metadata,
+                onPayload: async (payload) => await this.emitBeforeProviderPayload(model, payload),
+                onResponse: async (response) => {
+                    const headers = { ...response.headers };
+                    await this.emitOwn({ type: "after_provider_response", status: response.status, headers }, streamOptions?.signal);
+                },
+                reasoning: streamOptions?.reasoning,
+                maxTokens: streamOptions?.maxTokens,
+                callDeadlineMs: streamOptions?.callDeadlineMs,
+                maxTokensDynamic: streamOptions?.maxTokensDynamic,
+                staticReasoningCutDowngrade: streamOptions?.staticReasoningCutDowngrade,
+                resilience: streamOptions?.resilience,
+                signal: streamOptions?.signal,
+                sessionId: turnState.sessionId,
+                timeoutMs: requestOptions.timeoutMs,
+                transport: requestOptions.transport,
+                apiKey: auth?.apiKey,
+            });
+        };
+    }
+    async drainQueuedMessages(queue, mode) {
+        const messages = mode === "all" ? queue.splice(0) : queue.splice(0, 1);
+        if (messages.length === 0) {
+            return messages;
+        }
+        try {
+            await this.emitQueueUpdate();
+            return messages;
+        }
+        catch (error) {
+            queue.unshift(...messages);
+            throw normalizeHookError(error);
+        }
+    }
+    createLoopConfig(getTurnState, setTurnState) {
+        const turnState = getTurnState();
+        const self = this;
+        return {
+            model: turnState.model,
+            reasoning: turnState.thinkingLevel === "off" ? undefined : turnState.thinkingLevel,
+            ...(this.maxOutputTokens !== undefined ? { maxTokens: this.maxOutputTokens } : {}),
+            ...(this.maxOutputTokensPerCall !== undefined ? { maxTokensPerCall: this.maxOutputTokensPerCall } : {}),
+            ...(this.callDeadlineMsPerCall !== undefined ? { callDeadlineMsPerCall: this.callDeadlineMsPerCall } : {}),
+            ...(this.resilience !== undefined ? { resilience: this.resilience } : {}),
+            ...(this.maxToolConcurrency !== undefined ? { maxToolConcurrency: this.maxToolConcurrency } : {}),
+            ...(this.streamingToolExecution === true && (this.getHandlers("tool_call")?.size ?? 0) === 0
+                ? { streamingToolExecution: true }
+                : {}),
+            convertToLlm,
+            shouldStopAfterTurn: () => this._stopAfterTurn,
+            transformContext: async (messages) => {
+                const result = await this.emitHook({ type: "context", messages: [...messages] });
+                return result?.messages ?? messages;
+            },
+            beforeToolCall: async ({ toolCall, args }) => {
+                const result = await this.emitHook({
+                    type: "tool_call",
+                    toolCallId: toolCall.id,
+                    toolName: toolCall.name,
+                    input: args,
+                });
+                return result;
+            },
+            afterToolCall: async ({ toolCall, args, result, isError }) => {
+                const patch = await this.emitHook({
+                    type: "tool_result",
+                    toolCallId: toolCall.id,
+                    toolName: toolCall.name,
+                    input: args,
+                    content: result.content,
+                    details: result.details,
+                    isError,
+                });
+                return patch
+                    ? {
+                        content: patch.content,
+                        details: patch.details,
+                        isError: patch.isError,
+                        terminate: patch.terminate,
+                    }
+                    : undefined;
+            },
+            prepareNextTurn: async () => {
+                await this.flushPendingSessionWrites();
+                await this.emitHook({ type: "turn_boundary", model: this.model });
+                const nextTurnState = await this.createTurnState();
+                setTurnState(nextTurnState);
+                return {
+                    context: this.createContext(nextTurnState),
+                    model: nextTurnState.model,
+                    thinkingLevel: nextTurnState.thinkingLevel,
+                };
+            },
+            getSteeringMessages: async () => this.drainQueuedMessages(this.steerQueue, this.steeringQueueMode),
+            getFollowUpMessages: async () => {
+                const queued = await this.drainQueuedMessages(this.followUpQueue, this.followUpQueueMode);
+                if (queued.length > 0)
+                    return queued;
+                return this.stopGate ? await this.stopGate() : queued;
+            },
+            recovery: {
+                get truncatedOutput() {
+                    return self.loopRecovery?.truncatedOutput;
+                },
+                get malformedToolUse() {
+                    return self.loopRecovery?.malformedToolUse;
+                },
+                get thinkingOnly() {
+                    return self.loopRecovery?.thinkingOnly;
+                },
+                get degenerateOutput() {
+                    return self.loopRecovery?.degenerateOutput;
+                },
+                get walltimeCutoff() {
+                    return self.loopRecovery?.walltimeCutoff;
+                },
+                get promptTooLong() {
+                    const ptl = self.loopRecovery?.promptTooLong;
+                    if (!ptl)
+                        return undefined;
+                    return {
+                        detect: ptl.detect,
+                        maxRetries: ptl.maxRetries,
+                        withholdErrorEvents: ptl.withholdErrorEvents,
+                        recover: async (_messages, attempt) => {
+                            let shrank = false;
+                            try {
+                                shrank = await ptl.recover(attempt);
+                            }
+                            catch {
+                                shrank = false;
+                            }
+                            if (!shrank)
+                                return undefined;
+                            const nextTurnState = await self.createTurnState();
+                            setTurnState(nextTurnState);
+                            return self.createContext(nextTurnState).messages;
+                        },
+                    };
+                },
+            },
+        };
+    }
+    setLoopRecovery(recovery) {
+        this.loopRecovery = recovery;
+    }
+    setStopGate(gate) {
+        this.stopGate = gate;
+    }
+    validateToolNames(toolNames, tools = this.tools) {
+        const missing = toolNames.filter((name) => !tools.has(name));
+        if (missing.length > 0) {
+            throw new AgentHarnessError("invalid_argument", `Unknown tool(s): ${missing.join(", ")}`);
+        }
+    }
+    async flushPendingSessionWrites() {
+        if (this.runAbortController?.signal.aborted) {
+            return;
+        }
+        while (this.pendingSessionWrites.length > 0) {
+            const write = this.pendingSessionWrites[0];
+            if (write.type === "message") {
+                await this.session.appendMessage(write.message);
+            }
+            else if (write.type === "model_change") {
+                await this.session.appendModelChange(write.provider, write.modelId);
+            }
+            else if (write.type === "thinking_level_change") {
+                await this.session.appendThinkingLevelChange(write.thinkingLevel);
+            }
+            else if (write.type === "custom") {
+                await this.session.appendCustomEntry(write.customType, write.data);
+            }
+            else if (write.type === "custom_message") {
+                await this.session.appendCustomMessageEntry(write.customType, write.content, write.display, write.details);
+            }
+            else if (write.type === "label") {
+                await this.session.appendLabel(write.targetId, write.label);
+            }
+            else if (write.type === "session_info") {
+                await this.session.appendSessionName(write.name ?? "");
+            }
+            else if (write.type === "leaf") {
+                await this.session.getStorage().setLeafId(write.targetId);
+            }
+            this.pendingSessionWrites.shift();
+        }
+    }
+    async handleAgentEvent(event, signal) {
+        if (event.type === "message_end") {
+            let entryId;
+            if (!isEmptyFailureAssistant(event.message)) {
+                entryId = await this.session.appendMessage(event.message);
+            }
+            await this.emitAny(entryId !== undefined ? { ...event, entryId } : event, signal);
+            return;
+        }
+        if (event.type === "turn_end") {
+            let eventError;
+            try {
+                await this.emitAny(event, signal);
+            }
+            catch (error) {
+                eventError = error;
+            }
+            const queued = this.pendingSessionWrites.length > 0;
+            await this.flushPendingSessionWrites();
+            const hadPendingMutations = queued && this.pendingSessionWrites.length === 0;
+            if (eventError) {
+                throw toLintErrorObject(eventError, "Non-Error thrown");
+            }
+            await this.emitOwn({ type: "save_point", hadPendingMutations });
+            return;
+        }
+        if (event.type === "agent_end") {
+            await this.flushPendingSessionWrites();
+            this.phase = "idle";
+            await this.emitAny(event, signal);
+            await this.emitOwn({ type: "settled", nextTurnCount: this.nextTurnQueue.length }, signal);
+            return;
+        }
+        await this.emitAny(event, signal);
+    }
+    async emitRunFailure(model, error, aborted, signal) {
+        const failureMessage = createFailureMessage(model, error, aborted);
+        await this.handleAgentEvent({ type: "message_start", message: failureMessage }, signal);
+        await this.handleAgentEvent({ type: "message_end", message: failureMessage }, signal);
+        await this.handleAgentEvent({ type: "turn_end", message: failureMessage, toolResults: [] }, signal);
+        await this.handleAgentEvent({ type: "agent_end", messages: [failureMessage] }, signal);
+        return [failureMessage];
+    }
+    async executeTurn(turnState, text, options) {
+        let activeTurnState = turnState;
+        let messages = [createUserMessage(text, options?.images)];
+        if (this.nextTurnQueue.length > 0) {
+            const queuedMessages = this.nextTurnQueue.splice(0);
+            try {
+                await this.emitQueueUpdate();
+            }
+            catch (error) {
+                this.nextTurnQueue.unshift(...queuedMessages);
+                throw normalizeHookError(error);
+            }
+            messages = [...queuedMessages, messages[0]];
+        }
+        const beforeResult = await this.emitHook({
+            type: "before_agent_start",
+            prompt: text,
+            images: options?.images,
+            systemPrompt: turnState.systemPrompt,
+            resources: turnState.resources,
+        });
+        if (beforeResult?.messages) {
+            messages = [...messages, ...beforeResult.messages];
+        }
+        const abortController = new AbortController();
+        const getTurnState = () => activeTurnState;
+        const setTurnState = (nextTurnState) => {
+            activeTurnState = nextTurnState;
+        };
+        this.runAbortController = abortController;
+        const runResultPromise = (async () => {
+            try {
+                return await runAgentLoop(messages, this.createContext(turnState, beforeResult?.systemPrompt), this.createLoopConfig(getTurnState, setTurnState), (event) => this.handleAgentEvent(event, abortController.signal), abortController.signal, this.createStreamFn(getTurnState), undefined, this.loopTrace);
+            }
+            catch (error) {
+                try {
+                    return await this.emitRunFailure(activeTurnState.model, error, abortController.signal.aborted, abortController.signal);
+                }
+                catch (failureError) {
+                    const cause = new AggregateError([toError(error), toError(failureError)], "Agent run failed and failure reporting failed");
+                    throw new AgentHarnessError("unknown", cause.message, cause);
+                }
+            }
+        })();
+        try {
+            const newMessages = await runResultPromise;
+            for (let i = newMessages.length - 1; i >= 0; i--) {
+                const message = newMessages[i];
+                if (message.role === "assistant") {
+                    return mergeTruncatedOutputChain(newMessages, i, this.loopRecovery?.walltimeCutoff?.detect);
+                }
+            }
+            throw new AgentHarnessError("invalid_state", "AgentHarness prompt completed without an assistant message");
+        }
+        finally {
+            try {
+                await this.flushPendingSessionWrites();
+            }
+            finally {
+                this.runAbortController = undefined;
+            }
+        }
+    }
+    async prompt(text, options) {
+        if (this.phase !== "idle") {
+            throw new AgentHarnessError("busy", "AgentHarness is busy");
+        }
+        this._stopAfterTurn = false;
+        this.phase = "turn";
+        const finishRunPromise = this.startRunPromise();
+        try {
+            await this.flushPendingSessionWrites();
+            const turnState = await this.createTurnState();
+            return await this.executeTurn(turnState, text, options);
+        }
+        catch (error) {
+            this.phase = "idle";
+            throw normalizeHarnessError(error, "unknown");
+        }
+        finally {
+            finishRunPromise();
+        }
+    }
+    async steer(text, options) {
+        if (this.phase === "idle") {
+            throw new AgentHarnessError("invalid_state", "Cannot steer while idle");
+        }
+        this.steerQueue.push(createUserMessage(text, options?.images));
+        await this.emitQueueUpdate();
+    }
+    async followUp(text, options) {
+        if (this.phase === "idle") {
+            throw new AgentHarnessError("invalid_state", "Cannot follow up while idle");
+        }
+        this.followUpQueue.push(createUserMessage(text, options?.images));
+        await this.emitQueueUpdate();
+    }
+    async nextTurn(text, options) {
+        this.nextTurnQueue.push(createUserMessage(text, options?.images));
+        await this.emitQueueUpdate();
+    }
+    async appendMessage(message) {
+        try {
+            if (this.phase === "idle") {
+                await this.session.appendMessage(message);
+            }
+            else {
+                this.pendingSessionWrites.push({ type: "message", message });
+            }
+        }
+        catch (error) {
+            throw normalizeHarnessError(error, "session");
+        }
+    }
+    getModel() {
+        return this.model;
+    }
+    getThinkingLevel() {
+        return this.thinkingLevel;
+    }
+    async setModel(model) {
+        try {
+            const previousModel = this.model;
+            if (this.phase === "idle") {
+                await this.session.appendModelChange(model.provider, model.id);
+            }
+            else {
+                this.pendingSessionWrites.push({
+                    type: "model_change",
+                    provider: model.provider,
+                    modelId: model.id,
+                });
+            }
+            this.model = model;
+            await this.emitOwn({ type: "model_select", model, previousModel, source: "set" });
+        }
+        catch (error) {
+            throw normalizeHarnessError(error, "session");
+        }
+    }
+    async setThinkingLevel(level) {
+        try {
+            const previousLevel = this.thinkingLevel;
+            if (this.phase === "idle") {
+                await this.session.appendThinkingLevelChange(level);
+            }
+            else {
+                this.pendingSessionWrites.push({ type: "thinking_level_change", thinkingLevel: level });
+            }
+            this.thinkingLevel = level;
+            await this.emitOwn({ type: "thinking_level_select", level, previousLevel });
+        }
+        catch (error) {
+            throw normalizeHarnessError(error, "session");
+        }
+    }
+    async setActiveTools(toolNames) {
+        try {
+            this.validateToolNames(toolNames);
+            this.activeToolNames = [...toolNames];
+        }
+        catch (error) {
+            throw normalizeHarnessError(error, "invalid_argument");
+        }
+    }
+    getSteeringMode() {
+        return this.steeringQueueMode;
+    }
+    async setSteeringMode(mode) {
+        this.steeringQueueMode = mode;
+    }
+    getFollowUpMode() {
+        return this.followUpQueueMode;
+    }
+    async setFollowUpMode(mode) {
+        this.followUpQueueMode = mode;
+    }
+    getResources() {
+        return {
+            skills: this.resources.skills?.slice(),
+            promptTemplates: this.resources.promptTemplates?.slice(),
+        };
+    }
+    async setResources(resources) {
+        const previousResources = this.getResources();
+        this.resources = {
+            skills: resources.skills?.slice(),
+            promptTemplates: resources.promptTemplates?.slice(),
+        };
+        await this.emitOwn({
+            type: "resources_update",
+            resources: this.getResources(),
+            previousResources,
+        });
+    }
+    getStreamOptions() {
+        return cloneStreamOptions(this.streamOptions);
+    }
+    async setStreamOptions(streamOptions) {
+        this.streamOptions = cloneStreamOptions(streamOptions);
+    }
+    async setTools(tools, activeToolNames) {
+        try {
+            const nextTools = new Map(tools.map((tool) => [tool.name, tool]));
+            const nextActiveToolNames = activeToolNames ? [...activeToolNames] : this.activeToolNames;
+            this.validateToolNames(nextActiveToolNames, nextTools);
+            this.tools = nextTools;
+            this.activeToolNames = [...nextActiveToolNames];
+        }
+        catch (error) {
+            throw normalizeHarnessError(error, "invalid_argument");
+        }
+    }
+    requestStopAfterTurn() {
+        this._stopAfterTurn = true;
+    }
+    async flushQueuedSessionWrites() {
+        try {
+            await this.flushPendingSessionWrites();
+        }
+        catch (error) {
+            throw normalizeHarnessError(error, "session");
+        }
+    }
+    async abort() {
+        const clearedSteer = [...this.steerQueue];
+        const clearedFollowUp = [...this.followUpQueue];
+        this.steerQueue = [];
+        this.followUpQueue = [];
+        this.runAbortController?.abort();
+        const errors = [];
+        try {
+            await this.emitQueueUpdate();
+        }
+        catch (error) {
+            errors.push(toError(error));
+        }
+        try {
+            await this.waitForIdle();
+        }
+        catch (error) {
+            errors.push(toError(error));
+        }
+        try {
+            await this.emitOwn({ type: "abort", clearedSteer, clearedFollowUp });
+        }
+        catch (error) {
+            errors.push(toError(error));
+        }
+        if (errors.length > 0) {
+            const cause = errors.length === 1 ? errors[0] : new AggregateError(errors, "Abort completed with errors");
+            throw normalizeHarnessError(cause, "hook");
+        }
+        return { clearedSteer, clearedFollowUp };
+    }
+    async waitForIdle() {
+        await this.runPromise;
+    }
+    subscribe(listener) {
+        let handlers = this.handlers.get(SUBSCRIBER_EVENT_TYPE);
+        if (!handlers) {
+            handlers = new Set();
+            this.handlers.set(SUBSCRIBER_EVENT_TYPE, handlers);
+        }
+        handlers.add(listener);
+        return () => handlers.delete(listener);
+    }
+    on(type, handler) {
+        let handlers = this.handlers.get(type);
+        if (!handlers) {
+            handlers = new Set();
+            this.handlers.set(type, handlers);
+        }
+        handlers.add(handler);
+        return () => handlers.delete(handler);
+    }
+}
+function isEmptyFailureAssistant(message) {
+    if (message.role !== "assistant")
+        return false;
+    const m = message;
+    if (m.stopReason !== "aborted" && m.stopReason !== "error")
+        return false;
+    return m.content.every((c) => c.type === "text" ? c.text.trim() === "" : c.type === "thinking" ? c.thinking.trim() === "" : false);
+}
+function mergeTruncatedOutputChain(messages, lastAssistantIndex, isWalltimeCutoff) {
+    const last = messages[lastAssistantIndex];
+    const lastText = last.content.map((c) => (c.type === "text" ? c.text : "")).join("");
+    const parts = [];
+    let i = lastAssistantIndex;
+    while (i - 2 >= 0) {
+        const mid = messages[i - 1];
+        const downstreamText = parts.join("") + lastText;
+        const walltimeGate = last.stopReason === "error" || downstreamText.trim() === "";
+        let head;
+        let walltimeStep = false;
+        let nextI = i - 2;
+        if (mid.role === "custom") {
+            const h = messages[i - 2];
+            if (h.role !== "assistant")
+                break;
+            const assistant = h;
+            const step = (mid.customType === "loop:truncated-output-continue" && assistant.stopReason === "length") ||
+                (mid.customType === "loop:midstream-partial-continue" && assistant.stopReason === "stop") ||
+                (mid.customType === "loop:walltime-cutoff-recovery" && assistant.stopReason === "error" && walltimeGate);
+            if (!step)
+                break;
+            head = assistant;
+            walltimeStep = mid.customType === "loop:walltime-cutoff-recovery";
+        }
+        else if (mid.role === "user" && isWalltimeCutoff) {
+            let j = i - 1;
+            while (j >= 0 && messages[j].role === "user")
+                j--;
+            if (j < 0 || j >= i - 1)
+                break;
+            const h = messages[j];
+            if (h.role !== "assistant")
+                break;
+            const assistant = h;
+            if (!(assistant.stopReason === "error" && isWalltimeCutoff(assistant) && walltimeGate))
+                break;
+            head = assistant;
+            walltimeStep = true;
+            nextI = j;
+        }
+        else {
+            break;
+        }
+        const headText = head.content.map((c) => (c.type === "text" ? c.text : "")).join("");
+        const needsBreak = walltimeStep && downstreamText !== "" && headText !== "";
+        parts.unshift(needsBreak ? `${headText}\n\n` : headText);
+        i = nextI;
+    }
+    if (parts.length === 0) {
+        return last;
+    }
+    return {
+        ...last,
+        content: [
+            { type: "text", text: parts.join("") + lastText },
+            ...last.content.filter((c) => c.type !== "text"),
+        ],
+    };
+}
+function toLintErrorObject(value, fallbackMessage) {
+    if (value instanceof Error) {
+        return value;
+    }
+    if (typeof value === "string") {
+        return new Error(value);
+    }
+    const error = new Error(fallbackMessage, { cause: value });
+    if ((typeof value === "object" && value !== null) || typeof value === "function") {
+        Object.assign(error, value);
+    }
+    return error;
+}
+//# sourceMappingURL=agent-harness.js.map