@botbotgo/agent-harness 0.0.108 → 0.0.109

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/run/stream-runtime.d.ts

@@ -0,0 +1,48 @@
+import type { CompiledAgentBinding, HarnessEvent, MessageContent, RunResult, TranscriptMessage } from "../../../contracts/types.js";
+import type { RuntimePersistence } from "../../../persistence/types.js";
+import type { AgentRuntimeAdapter } from "../../agent-runtime-adapter.js";
+export declare function createStreamRunRuntime(input: {
+    persistence: RuntimePersistence;
+    runtimeAdapter: AgentRuntimeAdapter;
+    emit: (threadId: string, runId: string, sequence: number, eventType: string, payload: Record<string, unknown>) => Promise<HarnessEvent>;
+    setRunStateAndEmit: (threadId: string, runId: string, sequence: number, state: RunResult["state"], options: {
+        previousState: string | null;
+        checkpointRef?: string | null;
+        error?: string;
+    }) => Promise<HarnessEvent>;
+    requestApprovalAndEmit: (threadId: string, runId: string, input: MessageContent, interruptContent: string | undefined, checkpointRef: string, sequence: number) => Promise<{
+        approval: {
+            approvalId: string;
+            pendingActionId: string;
+        };
+        event: HarnessEvent;
+    }>;
+    loadPriorHistory: (threadId: string, runId: string) => Promise<TranscriptMessage[]>;
+    invokeWithHistory: (binding: CompiledAgentBinding, input: MessageContent, threadId: string, runId: string) => Promise<RunResult>;
+    emitSyntheticFallback: (threadId: string, runId: string, selectedAgentId: string, error: unknown) => Promise<void>;
+}): {
+    loadPriorHistory: (threadId: string, runId: string) => Promise<TranscriptMessage[]>;
+    stream: (binding: CompiledAgentBinding, message: MessageContent, threadId: string, priorHistory: TranscriptMessage[], options: {
+        context?: Record<string, unknown>;
+        state?: Record<string, unknown>;
+        files?: Record<string, unknown>;
+        runId: string;
+    }) => AsyncGenerator<string | import("../../parsing/stream-event-parsing.js").RuntimeStreamChunk, any, any>;
+    invokeWithHistory: (binding: CompiledAgentBinding, message: MessageContent, threadId: string, runId: string) => Promise<RunResult>;
+    emit: (threadId: string, runId: string, sequence: number, eventType: string, payload: Record<string, unknown>) => Promise<HarnessEvent>;
+    setRunStateAndEmit: (threadId: string, runId: string, sequence: number, state: RunResult["state"], options: {
+        previousState: string | null;
+        checkpointRef?: string | null;
+        error?: string;
+    }) => Promise<HarnessEvent>;
+    requestApprovalAndEmit: (threadId: string, runId: string, message: MessageContent, interruptContent: string | undefined, checkpointRef: string, sequence: number) => Promise<{
+        approval: {
+            approvalId: string;
+            pendingActionId: string;
+        };
+        event: HarnessEvent;
+    }>;
+    appendAssistantMessage: (threadId: string, runId: string, content?: string) => Promise<void>;
+    clearRunRequest: (threadId: string, runId: string) => Promise<void>;
+    emitSyntheticFallback: (threadId: string, runId: string, selectedAgentId: string, error: unknown) => Promise<void>;
+};

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

@@ -0,0 +1,14 @@
+import { appendAssistantMessage as appendLifecycleAssistantMessage } from "./run-lifecycle.js";
+export function createStreamRunRuntime(input) {
+    return {
+        loadPriorHistory: (threadId, runId) => input.loadPriorHistory(threadId, runId),
+        stream: (binding, message, threadId, priorHistory, options) => input.runtimeAdapter.stream(binding, message, threadId, priorHistory, options),
+        invokeWithHistory: (binding, message, threadId, runId) => input.invokeWithHistory(binding, message, threadId, runId),
+        emit: (threadId, runId, sequence, eventType, payload) => input.emit(threadId, runId, sequence, eventType, payload),
+        setRunStateAndEmit: (threadId, runId, sequence, state, options) => input.setRunStateAndEmit(threadId, runId, sequence, state, options),
+        requestApprovalAndEmit: (threadId, runId, message, interruptContent, checkpointRef, sequence) => input.requestApprovalAndEmit(threadId, runId, message, interruptContent, checkpointRef, sequence),
+        appendAssistantMessage: (threadId, runId, content) => appendLifecycleAssistantMessage(input.persistence, threadId, runId, content),
+        clearRunRequest: (threadId, runId) => input.persistence.clearRunRequest(threadId, runId),
+        emitSyntheticFallback: (threadId, runId, selectedAgentId, error) => input.emitSyntheticFallback(threadId, runId, selectedAgentId, error),
+    };
+}

package/dist/runtime/harness.d.ts

@@ -33,7 +33,10 @@ 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;
@@ -75,7 +78,7 @@ export declare class AgentHarnessRuntime {
     }): Promise<string>;
     private emit;
     private trackBackgroundTask;
-    private ensureThreadStarted;
+    private resolveToolMcpServerTools;
     private loadPriorHistory;
     private loadRunInput;
     private getRunCancellation;
@@ -89,10 +92,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>;