@botbotgo/agent-harness 0.0.108 → 0.0.110

package/dist/runtime/harness/run/run-operations.d.ts

@@ -45,6 +45,53 @@ type CancelOperationRuntime = {
         error?: string;
     }) => Promise<unknown>;
 };
+type ExecuteQueuedRunOperationRuntime = {
+    persistence: {
+        getRun: (runId: string) => Promise<{
+            state: RunResult["state"];
+        } | null>;
+        clearRunRequest: (threadId: string, runId: string) => Promise<void>;
+    };
+    getRunCancellation: (runId: string) => Promise<{
+        requested: boolean;
+        reason?: string;
+    }>;
+    finalizeCancelledRun: (threadId: string, runId: string, previousState: RunResult["state"] | null, reason?: string) => Promise<RunResult>;
+    emit: (threadId: string, runId: string, sequence: number, eventType: string, payload: Record<string, unknown>) => Promise<unknown>;
+    setRunStateAndEmit: (threadId: string, runId: string, sequence: number, state: RunResult["state"], options: {
+        previousState: string | null;
+        checkpointRef?: string | null;
+        error?: string;
+    }) => Promise<unknown>;
+    invokeWithHistory: (binding: CompiledAgentBinding, input: MessageContent, threadId: string, runId: string, resumePayload?: unknown, priorHistory?: TranscriptMessage[], options?: {
+        context?: Record<string, unknown>;
+        state?: Record<string, unknown>;
+        files?: Record<string, unknown>;
+    }) => Promise<RunResult>;
+    finalizeContinuedRun: (binding: CompiledAgentBinding, threadId: string, runId: string, input: MessageContent, actual: RunResult, options: {
+        previousState: RunResult["state"] | null;
+        stateSequence: number;
+        approvalSequence?: number;
+    }) => Promise<RunResult>;
+    emitSyntheticFallback: (threadId: string, runId: string, agentId: string, error: unknown) => Promise<void>;
+    renderRuntimeFailure: (error: unknown) => string;
+};
 export declare function resumeRun(runtime: ResumeOperationRuntime, options: ResumeOptions): Promise<RunResult>;
 export declare function cancelRunOperation(runtime: CancelOperationRuntime, options: CancelOptions): Promise<RunResult>;
+export declare function executeQueuedRunOperation(runtime: ExecuteQueuedRunOperationRuntime, input: {
+    binding: CompiledAgentBinding;
+    message: MessageContent;
+    threadId: string;
+    runId: string;
+    agentId: string;
+    options?: {
+        context?: Record<string, unknown>;
+        state?: Record<string, unknown>;
+        files?: Record<string, unknown>;
+        priorHistory?: TranscriptMessage[];
+        previousState?: RunResult["state"];
+        stateSequence?: number;
+        approvalSequence?: number;
+    };
+}): Promise<RunResult>;
 export {};

package/dist/runtime/harness/run/run-operations.js

@@ -1,4 +1,14 @@
 import { isTerminalRunState } from "./helpers.js";
+async function finalizeIfCancellationRequested(input) {
+    const cancellation = await input.getRunCancellation(input.runId);
+    if (!cancellation.requested) {
+        return null;
+    }
+    return input.finalizeCancelledRun(input.threadId, input.runId, input.previousState, cancellation.reason);
+}
+function normalizeContinuationState(previousState) {
+    return previousState === "queued" ? "running" : previousState;
+}
 export async function resumeRun(runtime, options) {
     const approvalById = options.approvalId ? await runtime.getApprovalById(options.approvalId) : null;
     const thread = options.threadId
@@ -17,9 +27,15 @@ export async function resumeRun(runtime, options) {
         throw new Error(`Unknown agent ${thread.agentId}`);
     }
     const resumePayload = runtime.buildResumePayload(binding, approval, options);
-    const cancellation = await runtime.getRunCancellation(runId);
-    if (cancellation.requested) {
-        return runtime.finalizeCancelledRun(threadId, runId, thread.status, cancellation.reason);
+    const cancelledBeforeResume = await finalizeIfCancellationRequested({
+        getRunCancellation: runtime.getRunCancellation,
+        finalizeCancelledRun: runtime.finalizeCancelledRun,
+        threadId,
+        runId,
+        previousState: thread.status,
+    });
+    if (cancelledBeforeResume) {
+        return cancelledBeforeResume;
     }
     const checkpointRef = `checkpoints/${threadId}/${runId}/cp-1`;
     await runtime.setRunState(threadId, runId, "resuming", checkpointRef);
@@ -53,9 +69,15 @@ export async function resumeRun(runtime, options) {
         try {
             const actual = await runtime.invoke(binding, "", threadId, runId, resumePayload, priorHistory);
             runtime.recordLlmSuccess(startedAt);
-            const cancelledAfterInvoke = await runtime.getRunCancellation(runId);
-            if (cancelledAfterInvoke.requested) {
-                return runtime.finalizeCancelledRun(threadId, runId, "resuming", cancelledAfterInvoke.reason);
+            const cancelledAfterInvoke = await finalizeIfCancellationRequested({
+                getRunCancellation: runtime.getRunCancellation,
+                finalizeCancelledRun: runtime.finalizeCancelledRun,
+                threadId,
+                runId,
+                previousState: "resuming",
+            });
+            if (cancelledAfterInvoke) {
+                return cancelledAfterInvoke;
             }
             await runtime.clearRecoveryIntent(threadId, runId);
             const finalized = await runtime.finalizeContinuedRun(binding, threadId, runId, runInput, actual, {
@@ -111,3 +133,83 @@ export async function cancelRunOperation(runtime, options) {
         output: options.reason ? `cancelling: ${options.reason}` : "cancelling",
     };
 }
+export async function executeQueuedRunOperation(runtime, input) {
+    const { binding, message, threadId, runId, agentId } = input;
+    const options = input.options ?? {};
+    const previousState = options.previousState ?? "running";
+    const currentRun = await runtime.persistence.getRun(runId);
+    if (currentRun?.state === "cancelled") {
+        return {
+            threadId,
+            runId,
+            agentId,
+            state: "cancelled",
+            output: "cancelled",
+        };
+    }
+    const cancelledBeforeInvoke = await finalizeIfCancellationRequested({
+        getRunCancellation: runtime.getRunCancellation,
+        finalizeCancelledRun: runtime.finalizeCancelledRun,
+        threadId,
+        runId,
+        previousState,
+    });
+    if (cancelledBeforeInvoke) {
+        return cancelledBeforeInvoke;
+    }
+    if (previousState === "queued") {
+        await runtime.emit(threadId, runId, 101, "run.dequeued", {
+            queuePosition: 0,
+            activeRunCount: 0,
+            maxConcurrentRuns: 0,
+            recoveredOnStartup: true,
+        });
+        await runtime.setRunStateAndEmit(threadId, runId, 102, "running", {
+            previousState: "queued",
+        });
+    }
+    const continuationState = normalizeContinuationState(previousState);
+    try {
+        const actual = await runtime.invokeWithHistory(binding, message, threadId, runId, undefined, options.priorHistory, {
+            context: options.context,
+            state: options.state,
+            files: options.files,
+        });
+        const cancelledAfterInvoke = await finalizeIfCancellationRequested({
+            getRunCancellation: runtime.getRunCancellation,
+            finalizeCancelledRun: runtime.finalizeCancelledRun,
+            threadId,
+            runId,
+            previousState: continuationState,
+        });
+        if (cancelledAfterInvoke) {
+            return cancelledAfterInvoke;
+        }
+        const finalized = await runtime.finalizeContinuedRun(binding, threadId, runId, message, actual, {
+            previousState: continuationState,
+            stateSequence: options.stateSequence ?? 103,
+            approvalSequence: options.approvalSequence ?? 104,
+        });
+        return {
+            ...finalized,
+            agentId,
+        };
+    }
+    catch (error) {
+        await runtime.emitSyntheticFallback(threadId, runId, agentId, error);
+        await runtime.setRunStateAndEmit(threadId, runId, 104, "failed", {
+            previousState: continuationState,
+            error: error instanceof Error ? error.message : String(error),
+        });
+        return {
+            threadId,
+            runId,
+            agentId,
+            state: "failed",
+            output: runtime.renderRuntimeFailure(error),
+        };
+    }
+    finally {
+        await runtime.persistence.clearRunRequest(threadId, runId);
+    }
+}

package/dist/runtime/harness/run/start-run.d.ts

@@ -0,0 +1,82 @@
+import type { HarnessEvent, InvocationEnvelope, MessageContent, RunResult, RunStartOptions, WorkspaceBundle } from "../../../contracts/types.js";
+import type { PersistedRunRequest } from "../../../persistence/types.js";
+import type { PolicyEngine } from "../system/policy-engine.js";
+type Binding = WorkspaceBundle["bindings"] extends Map<any, infer T> ? T : never;
+type EnsureThreadStartedRuntime = {
+    persistence: {
+        bootstrapRun?: (input: {
+            threadId: string;
+            agentId: string;
+            runId: string;
+            status: RunResult["state"];
+            createdAt: string;
+            executionMode: string;
+            adapterKind: string;
+            userMessage: {
+                role: "user";
+                content: MessageContent;
+                runId: string;
+                createdAt: string;
+            };
+            runRequest: PersistedRunRequest;
+            createThread: boolean;
+        }) => Promise<void>;
+        createThread: (input: {
+            threadId: string;
+            agentId: string;
+            runId: string;
+            status: RunResult["state"];
+            createdAt: string;
+        }) => Promise<void>;
+        appendThreadMessage: (threadId: string, message: {
+            role: "user";
+            content: MessageContent;
+            runId: string;
+            createdAt: string;
+        }) => Promise<void>;
+        createRun: (input: {
+            threadId: string;
+            runId: string;
+            agentId: string;
+            executionMode: string;
+            adapterKind: string;
+            createdAt: string;
+        }) => Promise<void>;
+        saveRunRequest: (threadId: string, runId: string, runRequest: PersistedRunRequest) => Promise<void>;
+    };
+};
+type PrepareRunStartRuntime = EnsureThreadStartedRuntime & {
+    workspace: WorkspaceBundle;
+    policyEngine: PolicyEngine;
+    resolveSelectedAgentId: (input: MessageContent, requestedAgentId?: string, threadId?: string) => Promise<string>;
+    emitRunCreated: (threadId: string, runId: string, payload: Record<string, unknown>) => Promise<HarnessEvent>;
+    acquireRunSlot: (threadId?: string, runId?: string, activeState?: RunResult["state"], priority?: number) => Promise<() => Promise<void>>;
+};
+export declare function ensureThreadStarted(runtime: EnsureThreadStartedRuntime, input: {
+    selectedAgentId: string;
+    binding: Binding;
+    message: MessageContent;
+    runRequest: PersistedRunRequest;
+    existingThreadId?: string;
+}): Promise<{
+    threadId: string;
+    runId: string;
+    createdAt: string;
+    isNewThread: boolean;
+}>;
+export declare function prepareRunStart(runtime: PrepareRunStartRuntime, input: {
+    options: Pick<RunStartOptions, "input" | "agentId" | "threadId" | "priority">;
+    invocation: InvocationEnvelope;
+    runCreatedPayload: (binding: Binding, selectedAgentId: string) => Record<string, unknown>;
+}): Promise<{
+    binding: Binding;
+    selectedAgentId: string;
+    priority: number;
+    threadId: string;
+    runId: string;
+    isNewThread: boolean;
+    runCreatedEventPromise: Promise<HarnessEvent>;
+    releaseRunSlotPromise: Promise<() => Promise<void>>;
+}>;
+export declare function defaultRequestedAgentId(agentId?: string): string;
+export {};

package/dist/runtime/harness/run/start-run.js

@@ -0,0 +1,88 @@
+import { AUTO_AGENT_ID } from "../../../contracts/types.js";
+import { createPersistentId } from "../../../utils/id.js";
+import { normalizeMessageContent } from "../../../utils/message-content.js";
+import { buildPersistedRunRequest, normalizeRunPriority } from "./helpers.js";
+import { getRequiredWorkspaceBinding } from "../bindings.js";
+import { getBindingAdapterKind } from "../../support/compiled-binding.js";
+export async function ensureThreadStarted(runtime, input) {
+    const { selectedAgentId, binding, message, runRequest, existingThreadId } = input;
+    const threadId = existingThreadId ?? createPersistentId();
+    const runId = createPersistentId();
+    const createdAt = new Date().toISOString();
+    const isNewThread = !existingThreadId;
+    const userMessage = {
+        role: "user",
+        content: normalizeMessageContent(message),
+        runId,
+        createdAt,
+    };
+    if (typeof runtime.persistence.bootstrapRun === "function") {
+        await runtime.persistence.bootstrapRun({
+            threadId,
+            agentId: binding.agent.id,
+            runId,
+            status: "running",
+            createdAt,
+            executionMode: getBindingAdapterKind(binding),
+            adapterKind: getBindingAdapterKind(binding),
+            userMessage,
+            runRequest,
+            createThread: isNewThread,
+        });
+    }
+    else {
+        if (isNewThread) {
+            await runtime.persistence.createThread({
+                threadId,
+                agentId: selectedAgentId,
+                runId,
+                status: "running",
+                createdAt,
+            });
+        }
+        await Promise.all([
+            runtime.persistence.appendThreadMessage(threadId, userMessage),
+            runtime.persistence.createRun({
+                threadId,
+                runId,
+                agentId: binding.agent.id,
+                executionMode: getBindingAdapterKind(binding),
+                adapterKind: getBindingAdapterKind(binding),
+                createdAt,
+            }),
+            runtime.persistence.saveRunRequest(threadId, runId, runRequest),
+        ]);
+    }
+    return { threadId, runId, createdAt, isNewThread };
+}
+export async function prepareRunStart(runtime, input) {
+    const { options, invocation, runCreatedPayload } = input;
+    const selectedAgentId = await runtime.resolveSelectedAgentId(options.input, options.agentId, options.threadId);
+    const binding = getRequiredWorkspaceBinding(runtime.workspace, selectedAgentId);
+    const policyDecision = runtime.policyEngine.evaluate(binding);
+    if (!policyDecision.allowed) {
+        throw new Error(`Policy evaluation blocked agent ${selectedAgentId}: ${policyDecision.reasons.join(", ")}`);
+    }
+    const priority = normalizeRunPriority(options.priority);
+    const runRequest = buildPersistedRunRequest(options.input, invocation, priority);
+    const { threadId, runId, isNewThread } = await ensureThreadStarted(runtime, {
+        selectedAgentId,
+        binding,
+        message: options.input,
+        runRequest,
+        existingThreadId: options.threadId,
+    });
+    return {
+        binding,
+        selectedAgentId,
+        priority,
+        threadId,
+        runId,
+        isNewThread,
+        runCreatedEventPromise: runtime.emitRunCreated(threadId, runId, runCreatedPayload(binding, selectedAgentId)),
+        releaseRunSlotPromise: runtime.acquireRunSlot(threadId, runId, "running", priority),
+    };
+}
+export function defaultRequestedAgentId(agentId) {
+    return agentId ?? AUTO_AGENT_ID;
+}

package/dist/runtime/harness/run/startup-runtime.d.ts

@@ -1,6 +1,7 @@
 import type { RunResult, ThreadSummary } from "../../../contracts/types.js";
 import type { RuntimePersistence } from "../../../persistence/types.js";
 import type { ConcurrencyConfig, RecoveryConfig } from "../../../workspace/support/workspace-ref-utils.js";
+import { type StartupRecoveryContext } from "./recovery.js";
 type Startable = {
     start(): Promise<void>;
 };
@@ -11,7 +12,7 @@ export declare function initializeHarnessRuntime(input: {
 export declare function recoverStartupRuns(input: {
     recoveryConfig: RecoveryConfig;
     persistence: RuntimePersistence;
-    createStartupRecoveryContext: () => unknown;
+    createStartupRecoveryContext: () => StartupRecoveryContext;
     reclaimExpiredClaimedRuns: (nowIso?: string) => Promise<void>;
 }): Promise<void>;
 export declare function reclaimExpiredClaimedRuns(input: {

package/dist/runtime/harness/run/startup-runtime.js

@@ -1,13 +1,48 @@
+import { recoverQueuedStartupRun, recoverResumingStartupRun, recoverRunningStartupRun, } from "./recovery.js";
 export async function initializeHarnessRuntime(input) {
     await input.persistence.initialize();
     await input.healthMonitor?.start();
 }
 export async function recoverStartupRuns(input) {
-    void input;
+    if (!input.recoveryConfig.enabled) {
+        return;
+    }
+    await input.reclaimExpiredClaimedRuns();
+    const threads = await input.persistence.listSessions();
+    if (threads.length === 0) {
+        return;
+    }
+    const context = input.createStartupRecoveryContext();
+    for (const thread of threads) {
+        if (await recoverQueuedStartupRun(context, thread)) {
+            continue;
+        }
+        if (await recoverRunningStartupRun(context, thread)) {
+            continue;
+        }
+        await recoverResumingStartupRun(context, thread);
+    }
 }
 export async function reclaimExpiredClaimedRuns(input, nowIso = new Date().toISOString()) {
-    void input;
-    void nowIso;
+    const expiredRuns = await input.persistence.listExpiredClaimedRuns(nowIso);
+    if (expiredRuns.length === 0) {
+        return;
+    }
+    const activeRunCount = input.getActiveRunSlots();
+    const maxConcurrentRuns = input.concurrencyConfig.maxConcurrentRuns;
+    for (const [index, claimedRun] of expiredRuns.entries()) {
+        await input.persistence.releaseRunClaim(claimedRun.runId);
+        await input.setRunStateAndEmit(claimedRun.threadId, claimedRun.runId, 100, "queued", {
+            previousState: "claimed",
+        });
+        await input.emit(claimedRun.threadId, claimedRun.runId, 101, "run.queued", {
+            queuePosition: activeRunCount + index + 1,
+            activeRunCount,
+            ...(maxConcurrentRuns !== undefined ? { maxConcurrentRuns } : {}),
+            priority: claimedRun.priority,
+            recoveryKind: "startup-claim-recovery",
+        });
+    }
 }
 export async function isStaleRunningRun(input, thread, nowMs = Date.now()) {
     const control = await input.persistence.getRunControl(thread.latestRunId);

package/dist/runtime/harness.d.ts

@@ -33,10 +33,14 @@ export declare class AgentHarnessRuntime {
     private pendingRunInsertionOrder;
     private readonly pendingRunSlots;
     private runtimeEventSequence;
+    private initialized;
+    private closed;
     private readonly backgroundEventRuntime;
+    private readonly runtimeEventOperations;
     private defaultRunRoot;
     private getDefaultHostAgentId;
     private resolveSelectedAgentId;
+    private createPrepareRunStartRuntime;
     constructor(workspace: WorkspaceBundle, runtimeAdapterOptions?: RuntimeAdapterOptions);
     private createHealthMonitor;
     private recordLlmSuccess;
@@ -75,7 +79,7 @@ export declare class AgentHarnessRuntime {
     }): Promise<string>;
     private emit;
     private trackBackgroundTask;
-    private ensureThreadStarted;
+    private resolveToolMcpServerTools;
     private loadPriorHistory;
     private loadRunInput;
     private getRunCancellation;
@@ -89,10 +93,8 @@ export declare class AgentHarnessRuntime {
     private requestApprovalAndEmit;
     private isDecisionRun;
     private notifyListener;
-    private prepareRunStart;
     private acquireRunSlot;
     private dropPendingRunSlot;
-    private dispatchRunListeners;
     run(options: RunOptions): Promise<RunResult>;
     streamEvents(options: RunStartOptions): AsyncGenerator<HarnessStreamItem>;
     resume(options: ResumeOptions): Promise<RunResult>;