@pasko70/pibo 1.12.0 → 1.12.1

package/dist/core/transcript-integrity.js

@@ -0,0 +1,431 @@
+import { sessionEntryToContextMessages, } from "@earendil-works/pi-coding-agent";
+export const PIBO_TRANSCRIPT_INTEGRITY_RESUME_MESSAGE_TYPE = "pibo-transcript-integrity-resume";
+export const PIBO_TRANSCRIPT_INTEGRITY_RESUME_PROMPT = "Continue the interrupted task autonomously from the repaired transcript. Do not rerun completed tools, wait for additional user input, or mention transcript repair unless it affects the result.";
+export const PIBO_TRANSCRIPT_INTEGRITY_ENTRY_TYPE = "pibo-transcript-integrity";
+const installedStates = new WeakMap();
+export class PiboTranscriptIntegrityError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = "PiboTranscriptIntegrityError";
+    }
+}
+export function validatePiboTranscriptIntegrityMessages(messages) {
+    return validateProjectedMessages(messages.map((message) => ({ message })));
+}
+export function installPiboTranscriptIntegrity(session) {
+    const existing = installedStates.get(session);
+    if (existing)
+        return [...existing.reports];
+    const manager = session.sessionManager;
+    const state = {
+        session,
+        originalAppendMessage: manager.appendMessage.bind(manager),
+        restoredAssistants: new Map(),
+        reports: [],
+        continuationPending: false,
+        continuationInProgress: false,
+    };
+    installedStates.set(session, state);
+    manager.appendMessage = ((message) => appendMessageWithIntegrity(state, message));
+    patchProviderBoundary(state);
+    patchCompactionBoundaries(state);
+    reconcilePiboTranscriptIntegrity(session, "load");
+    return [...state.reports];
+}
+export function reconcilePiboTranscriptIntegrity(session, boundary) {
+    const state = installedStates.get(session);
+    if (!state)
+        throw new PiboTranscriptIntegrityError("Transcript integrity is not installed for this session");
+    const manager = session.sessionManager;
+    const activeEntries = manager.buildContextEntries();
+    const projected = projectEntries(activeEntries);
+    const [issue] = validateProjectedMessages(projected, manager.getEntries());
+    if (!issue)
+        return undefined;
+    const invalidEntry = issue.entryId ? manager.getEntry(issue.entryId) : undefined;
+    if (!invalidEntry) {
+        return quarantineRuntimeTranscript(state, boundary, issue);
+    }
+    const invalidEntryIndex = activeEntries.findIndex((entry) => entry.id === invalidEntry.id);
+    const removedEntryIds = invalidEntryIndex >= 0
+        ? activeEntries.slice(invalidEntryIndex).map((entry) => entry.id)
+        : [invalidEntry.id];
+    const resultMessage = isToolResultMessageEntry(invalidEntry) ? invalidEntry.message : undefined;
+    const authoritativeAssistant = resultMessage
+        ? uniqueAuthoritativeAssistant(manager.getEntries(), issue.toolCallId)
+        : undefined;
+    let report;
+    branchTo(manager, invalidEntry.parentId);
+    if (authoritativeAssistant && resultMessage && (issue.relation === "orphan_result" || issue.relation === "wrong_branch_result")) {
+        const repairId = manager.appendCustomEntry(PIBO_TRANSCRIPT_INTEGRITY_ENTRY_TYPE, repairMetadata({
+            boundary,
+            relation: issue.relation,
+            action: "restored_pair",
+            toolCallIds: assistantToolCalls(authoritativeAssistant.message).map((call) => call.id),
+            removedEntryIds,
+            restoredEntryIds: [authoritativeAssistant.id, invalidEntry.id],
+            phase: "journal_started",
+        }));
+        const restoredAssistantEntryId = state.originalAppendMessage(authoritativeAssistant.message);
+        rememberRestoredAssistant(state, authoritativeAssistant.message, restoredAssistantEntryId);
+        const restoredResultEntryId = state.originalAppendMessage(resultMessage);
+        manager.appendCustomEntry(PIBO_TRANSCRIPT_INTEGRITY_ENTRY_TYPE, {
+            repairId,
+            phase: "journal_committed",
+            restoredAssistantEntryId,
+            restoredResultEntryId,
+        });
+        report = {
+            repairId,
+            boundary,
+            relation: issue.relation,
+            action: "restored_pair",
+            toolCallIds: assistantToolCalls(authoritativeAssistant.message).map((call) => call.id),
+            removedEntryIds,
+            restoredEntryIds: [authoritativeAssistant.id, invalidEntry.id],
+            continuationRequired: boundary === "load",
+        };
+    }
+    else {
+        const repairId = manager.appendCustomEntry(PIBO_TRANSCRIPT_INTEGRITY_ENTRY_TYPE, repairMetadata({
+            boundary,
+            relation: issue.relation,
+            action: "quarantined_tail",
+            toolCallIds: [issue.toolCallId],
+            removedEntryIds,
+            restoredEntryIds: [],
+            phase: "quarantined",
+        }));
+        report = {
+            repairId,
+            boundary,
+            relation: issue.relation,
+            action: "quarantined_tail",
+            toolCallIds: [issue.toolCallId],
+            removedEntryIds,
+            restoredEntryIds: [],
+            continuationRequired: boundary === "load",
+        };
+    }
+    syncAgentTranscript(session);
+    assertSessionTranscriptValid(session, boundary);
+    recordReport(state, report);
+    return report;
+}
+export function claimPiboTranscriptIntegrityContinuation(session) {
+    const state = installedStates.get(session);
+    if (!state || !state.continuationPending || state.continuationInProgress)
+        return [];
+    const reports = state.reports.filter((report) => report.continuationRequired);
+    if (reports.length === 0) {
+        state.continuationPending = false;
+        return [];
+    }
+    state.continuationPending = false;
+    state.continuationInProgress = true;
+    session.sessionManager.appendCustomEntry(PIBO_TRANSCRIPT_INTEGRITY_ENTRY_TYPE, {
+        phase: "continuation_started",
+        repairIds: reports.map((report) => report.repairId),
+    });
+    return reports;
+}
+export function settlePiboTranscriptIntegrityContinuation(session, outcome, error) {
+    const state = installedStates.get(session);
+    if (!state?.continuationInProgress)
+        return;
+    state.continuationInProgress = false;
+    session.sessionManager.appendCustomEntry(PIBO_TRANSCRIPT_INTEGRITY_ENTRY_TYPE, {
+        phase: `continuation_${outcome}`,
+        ...(error ? { errorClass: error instanceof Error ? error.name : "Error" } : {}),
+    });
+}
+function patchProviderBoundary(state) {
+    const agent = state.session.agent;
+    const previous = agent.transformContext?.bind(agent);
+    const repairedProviderMessages = async (signal) => {
+        const repaired = state.session.sessionManager.buildSessionContext().messages;
+        if (!previous)
+            return repaired;
+        const transformed = await previous(repaired, signal);
+        return validatePiboTranscriptIntegrityMessages(transformed).length === 0 ? transformed : repaired;
+    };
+    agent.transformContext = async (messages, signal) => {
+        const transformed = previous ? await previous(messages, signal) : messages;
+        const durableRepair = reconcilePiboTranscriptIntegrity(state.session, "before_provider");
+        if (durableRepair)
+            return repairedProviderMessages(signal);
+        const [runtimeIssue] = validatePiboTranscriptIntegrityMessages(transformed);
+        if (!runtimeIssue)
+            return transformed;
+        quarantineRuntimeTranscript(state, "before_provider", runtimeIssue);
+        return repairedProviderMessages(signal);
+    };
+}
+function patchCompactionBoundaries(state) {
+    const session = state.session;
+    const originalCompact = session.compact.bind(session);
+    session.compact = (async (...args) => {
+        reconcilePiboTranscriptIntegrity(session, "before_compaction");
+        const result = await originalCompact(...args);
+        reconcilePiboTranscriptIntegrity(session, "after_compaction");
+        return result;
+    });
+}
+function appendMessageWithIntegrity(state, message) {
+    if (isAssistantMessage(message)) {
+        const restoredEntryId = restoredAssistantEntryId(state, message);
+        return restoredEntryId ?? state.originalAppendMessage(message);
+    }
+    if (!isToolResultMessage(message))
+        return state.originalAppendMessage(message);
+    const activeMessages = state.session.sessionManager.buildSessionContext().messages;
+    if (activeToolCall(activeMessages, message.toolCallId))
+        return state.originalAppendMessage(message);
+    const authoritativeAssistant = uniqueAssistantMessage(state.session.agent.state.messages, message.toolCallId);
+    if (authoritativeAssistant) {
+        const toolCallIds = assistantToolCalls(authoritativeAssistant).map((call) => call.id);
+        const repairId = state.session.sessionManager.appendCustomEntry(PIBO_TRANSCRIPT_INTEGRITY_ENTRY_TYPE, repairMetadata({
+            boundary: "persistence",
+            relation: "orphan_result",
+            action: "restored_pair",
+            toolCallIds,
+            removedEntryIds: [],
+            restoredEntryIds: [],
+            phase: "journal_started",
+        }));
+        const assistantEntryId = state.originalAppendMessage(authoritativeAssistant);
+        rememberRestoredAssistant(state, authoritativeAssistant, assistantEntryId);
+        const resultEntryId = state.originalAppendMessage(message);
+        state.session.sessionManager.appendCustomEntry(PIBO_TRANSCRIPT_INTEGRITY_ENTRY_TYPE, {
+            repairId,
+            phase: "journal_committed",
+            assistantEntryId,
+            resultEntryId,
+        });
+        recordReport(state, {
+            repairId,
+            boundary: "persistence",
+            relation: "orphan_result",
+            action: "restored_pair",
+            toolCallIds,
+            removedEntryIds: [],
+            restoredEntryIds: [assistantEntryId, resultEntryId],
+            continuationRequired: false,
+        });
+        return resultEntryId;
+    }
+    const repairId = state.session.sessionManager.appendCustomEntry(PIBO_TRANSCRIPT_INTEGRITY_ENTRY_TYPE, repairMetadata({
+        boundary: "persistence",
+        relation: "orphan_result",
+        action: "quarantined_runtime_result",
+        toolCallIds: [message.toolCallId],
+        removedEntryIds: [],
+        restoredEntryIds: [],
+        phase: "quarantined",
+    }));
+    recordReport(state, {
+        repairId,
+        boundary: "persistence",
+        relation: "orphan_result",
+        action: "quarantined_runtime_result",
+        toolCallIds: [message.toolCallId],
+        removedEntryIds: [],
+        restoredEntryIds: [],
+        continuationRequired: false,
+    });
+    syncAgentTranscript(state.session);
+    return repairId;
+}
+function quarantineRuntimeTranscript(state, boundary, issue) {
+    const repairId = state.session.sessionManager.appendCustomEntry(PIBO_TRANSCRIPT_INTEGRITY_ENTRY_TYPE, repairMetadata({
+        boundary,
+        relation: issue.relation,
+        action: "quarantined_runtime_result",
+        toolCallIds: [issue.toolCallId],
+        removedEntryIds: [],
+        restoredEntryIds: [],
+        phase: "quarantined",
+    }));
+    const report = {
+        repairId,
+        boundary,
+        relation: issue.relation,
+        action: "quarantined_runtime_result",
+        toolCallIds: [issue.toolCallId],
+        removedEntryIds: [],
+        restoredEntryIds: [],
+        continuationRequired: boundary === "load",
+    };
+    syncAgentTranscript(state.session);
+    recordReport(state, report);
+    return report;
+}
+function validateProjectedMessages(projected, allEntries = []) {
+    const calls = new Map();
+    const results = new Map();
+    const issues = [];
+    const allCalls = allAssistantCallIds(allEntries);
+    for (let messageIndex = 0; messageIndex < projected.length; messageIndex += 1) {
+        const { message, entry } = projected[messageIndex];
+        if (isAssistantMessage(message)) {
+            for (const call of assistantToolCalls(message)) {
+                const existing = calls.get(call.id);
+                if (existing) {
+                    issues.push({ relation: "duplicate_call", toolCallId: call.id, toolName: call.name, messageIndex, entryId: entry?.id });
+                    continue;
+                }
+                calls.set(call.id, { name: call.name, messageIndex });
+            }
+            continue;
+        }
+        if (!isToolResultMessage(message))
+            continue;
+        if (results.has(message.toolCallId)) {
+            issues.push({ relation: "duplicate_result", toolCallId: message.toolCallId, toolName: message.toolName, messageIndex, entryId: entry?.id });
+            continue;
+        }
+        results.set(message.toolCallId, messageIndex);
+        const call = calls.get(message.toolCallId);
+        if (!call) {
+            issues.push({
+                relation: allCalls.has(message.toolCallId) ? "wrong_branch_result" : "orphan_result",
+                toolCallId: message.toolCallId,
+                toolName: message.toolName,
+                messageIndex,
+                entryId: entry?.id,
+            });
+            continue;
+        }
+        if (call.name !== message.toolName) {
+            issues.push({
+                relation: "tool_name_mismatch",
+                toolCallId: message.toolCallId,
+                toolName: message.toolName,
+                expectedToolName: call.name,
+                messageIndex,
+                entryId: entry?.id,
+            });
+        }
+    }
+    return issues;
+}
+function projectEntries(entries) {
+    return entries.flatMap((entry) => sessionEntryToContextMessages(entry).map((message) => ({ entry, message })));
+}
+function allAssistantCallIds(entries) {
+    const ids = new Set();
+    for (const entry of entries) {
+        if (!isAssistantMessageEntry(entry))
+            continue;
+        for (const call of assistantToolCalls(entry.message))
+            ids.add(call.id);
+    }
+    return ids;
+}
+function uniqueAuthoritativeAssistant(entries, toolCallId) {
+    const candidates = entries.filter((entry) => isAssistantMessageEntry(entry) && assistantToolCalls(entry.message).some((call) => call.id === toolCallId));
+    return candidates.length === 1 ? candidates[0] : undefined;
+}
+function uniqueAssistantMessage(messages, toolCallId) {
+    const candidates = messages.filter((message) => isAssistantMessage(message) && assistantToolCalls(message).some((call) => call.id === toolCallId));
+    return candidates.length === 1 ? candidates[0] : undefined;
+}
+function activeToolCall(messages, toolCallId) {
+    for (const message of messages) {
+        if (!isAssistantMessage(message))
+            continue;
+        const call = assistantToolCalls(message).find((candidate) => candidate.id === toolCallId);
+        if (call)
+            return call;
+    }
+    return undefined;
+}
+function assistantToolCalls(message) {
+    if (!isAssistantMessage(message))
+        return [];
+    return message.content.flatMap((part) => {
+        if (!part || typeof part !== "object")
+            return [];
+        const candidate = part;
+        return candidate.type === "toolCall" && typeof candidate.id === "string" && typeof candidate.name === "string"
+            ? [{ type: "toolCall", id: candidate.id, name: candidate.name, arguments: candidate.arguments }]
+            : [];
+    });
+}
+function isAssistantMessage(message) {
+    return Boolean(message && typeof message === "object" && message.role === "assistant" && Array.isArray(message.content));
+}
+function isToolResultMessage(message) {
+    return Boolean(message
+        && typeof message === "object"
+        && message.role === "toolResult"
+        && typeof message.toolCallId === "string"
+        && typeof message.toolName === "string");
+}
+function isAssistantMessageEntry(entry) {
+    return entry.type === "message" && isAssistantMessage(entry.message);
+}
+function isToolResultMessageEntry(entry) {
+    return entry.type === "message" && isToolResultMessage(entry.message);
+}
+function branchTo(manager, parentId) {
+    if (parentId)
+        manager.branch(parentId);
+    else
+        manager.resetLeaf();
+}
+function syncAgentTranscript(session) {
+    session.agent.state.messages = session.sessionManager.buildSessionContext().messages;
+}
+function assertSessionTranscriptValid(session, boundary) {
+    const issues = validateProjectedMessages(projectEntries(session.sessionManager.buildContextEntries()), session.sessionManager.getEntries());
+    if (issues.length === 0)
+        return;
+    const issue = issues[0];
+    throw new PiboTranscriptIntegrityError(`Transcript integrity repair failed at ${boundary}: ${issue.relation} for tool call ${issue.toolCallId}`);
+}
+function rememberRestoredAssistant(state, message, entryId) {
+    const fingerprint = assistantFingerprint(message);
+    for (const call of assistantToolCalls(message))
+        state.restoredAssistants.set(call.id, { entryId, fingerprint });
+}
+function restoredAssistantEntryId(state, message) {
+    const calls = assistantToolCalls(message);
+    if (calls.length === 0)
+        return undefined;
+    const fingerprint = assistantFingerprint(message);
+    const restored = calls.map((call) => state.restoredAssistants.get(call.id));
+    if (restored.some((item) => !item || item.fingerprint !== fingerprint))
+        return undefined;
+    const entryIds = new Set(restored.map((item) => item.entryId));
+    return entryIds.size === 1 ? restored[0].entryId : undefined;
+}
+function assistantFingerprint(message) {
+    return JSON.stringify({
+        content: message.content,
+        api: message.api,
+        provider: message.provider,
+        model: message.model,
+        responseId: message.responseId,
+        timestamp: message.timestamp,
+    });
+}
+function repairMetadata(input) {
+    return {
+        version: 1,
+        boundary: input.boundary,
+        relation: input.relation,
+        action: input.action,
+        phase: input.phase,
+        toolCallIds: input.toolCallIds,
+        removedEntryIds: input.removedEntryIds,
+        restoredEntryIds: input.restoredEntryIds,
+        removedCount: input.removedEntryIds.length,
+        restoredCount: input.restoredEntryIds.length,
+    };
+}
+function recordReport(state, report) {
+    state.reports.push(report);
+    if (report.continuationRequired)
+        state.continuationPending = true;
+}

package/dist/data/pibo-store.js

@@ -39,6 +39,8 @@ export class PiboDataStore {
         this.telemetry = new TelemetryStore(this.db);
     }
     transaction(action) {
+        if (this.db.isTransaction)
+            return action();
         this.db.exec("BEGIN IMMEDIATE");
         try {
             const result = action();

package/dist/data/telemetry.js

@@ -86,6 +86,154 @@ export class TelemetryStore {
 		`).all(turnId);
         return rows.map(toolCallFromRow);
     }
+    listActiveTurns() {
+        const rows = this.db.prepare(`
+			SELECT * FROM telemetry_turns
+			WHERE status IN ('queued', 'running')
+			ORDER BY queued_at ASC, created_at ASC
+		`).all();
+        return rows.map(turnFromRow);
+    }
+    recoverInterruptedTurns(input = {}) {
+        const at = input.at ?? new Date().toISOString();
+        return this.transaction(() => this.listActiveTurns().map((turn) => {
+            const outcome = input.resolveOutcome?.(turn) ?? {
+                status: "aborted",
+                summary: "Turn was interrupted by gateway restart.",
+            };
+            const phaseStatus = outcome.status;
+            const providerStatus = outcome.status;
+            const toolStatus = outcome.status;
+            const phaseName = outcome.status === "timeout"
+                ? "timeout"
+                : outcome.status === "error"
+                    ? "error"
+                    : "abort";
+            for (const phase of this.listOpenPhasesForTurn(turn.turnId)) {
+                this.finishPhase(phase.phaseId, {
+                    status: phaseStatus,
+                    endedAt: at,
+                    lastProgressAt: at,
+                    summary: outcome.summary,
+                });
+            }
+            for (const request of this.listActiveProviderRequestsForTurn(turn.turnId)) {
+                this.upsertProviderRequest({
+                    providerRequestId: request.providerRequestId,
+                    piboSessionId: request.piboSessionId,
+                    rootSessionId: request.rootSessionId,
+                    roomId: request.roomId,
+                    turnId: request.turnId,
+                    phaseId: request.phaseId,
+                    provider: request.provider,
+                    api: request.api,
+                    model: request.model,
+                    transport: request.transport,
+                    serviceTier: request.serviceTier,
+                    status: providerStatus,
+                    responseHeadersAt: request.responseHeadersAt,
+                    firstByteAt: request.firstByteAt,
+                    lastRawEventAt: request.lastRawEventAt,
+                    lastNormalizedEventAt: request.lastNormalizedEventAt,
+                    completedAt: at,
+                    httpStatus: request.httpStatus,
+                    upstreamResponseId: request.upstreamResponseId,
+                    rawEventCount: request.rawEventCount,
+                    normalizedEventCount: request.normalizedEventCount,
+                    parseErrorCount: request.parseErrorCount,
+                    unknownEventCount: request.unknownEventCount,
+                    bytesReceived: request.bytesReceived,
+                    eventTypeCounts: request.eventTypeCounts,
+                    eventStreamId: request.eventStreamId,
+                    eventId: request.eventId,
+                    payloadRef: request.payloadRef,
+                    errorCategory: "runtime_restart",
+                    errorMessage: outcome.summary,
+                    captureMode: request.captureMode,
+                    retentionClass: request.retentionClass,
+                    updatedAt: at,
+                });
+            }
+            for (const toolCall of this.listActiveToolCallsForTurn(turn.turnId)) {
+                const executionEndedAt = toolCall.executionStartedAt ? at : toolCall.executionEndedAt;
+                this.upsertToolCall({
+                    toolCallId: toolCall.toolCallId,
+                    piboSessionId: toolCall.piboSessionId,
+                    rootSessionId: toolCall.rootSessionId,
+                    roomId: toolCall.roomId,
+                    turnId: toolCall.turnId,
+                    providerRequestId: toolCall.providerRequestId,
+                    providerItemId: toolCall.providerItemId,
+                    outputIndex: toolCall.outputIndex,
+                    toolName: toolCall.toolName,
+                    status: toolStatus,
+                    argsStartedAt: toolCall.argsStartedAt,
+                    firstDeltaAt: toolCall.firstDeltaAt,
+                    lastDeltaAt: toolCall.lastDeltaAt,
+                    argsCompletedAt: toolCall.argsCompletedAt,
+                    executionStartedAt: toolCall.executionStartedAt,
+                    executionEndedAt,
+                    durationMs: toolCall.executionStartedAt && executionEndedAt
+                        ? Math.max(0, Date.parse(executionEndedAt) - Date.parse(toolCall.executionStartedAt))
+                        : toolCall.durationMs,
+                    argsBytes: toolCall.argsBytes,
+                    parseStatus: toolCall.parseStatus,
+                    safeArgKeys: toolCall.safeArgKeys,
+                    eventStreamId: toolCall.eventStreamId,
+                    eventId: toolCall.eventId,
+                    payloadRef: toolCall.payloadRef,
+                    runId: toolCall.runId,
+                    errorCategory: "runtime_restart",
+                    errorMessage: outcome.summary,
+                    retentionClass: toolCall.retentionClass,
+                    updatedAt: at,
+                });
+            }
+            this.upsertPhase({
+                phaseId: `${turn.turnId}:runtime_restart`,
+                turnId: turn.turnId,
+                piboSessionId: turn.piboSessionId,
+                rootSessionId: turn.rootSessionId,
+                roomId: turn.roomId,
+                name: phaseName,
+                status: phaseStatus,
+                startedAt: at,
+                endedAt: at,
+                lastProgressAt: at,
+                eventId: turn.eventId,
+                runId: turn.runId,
+                summary: outcome.summary,
+                retentionClass: turn.retentionClass,
+                updatedAt: at,
+            });
+            const recovered = this.upsertTurn({
+                turnId: turn.turnId,
+                piboSessionId: turn.piboSessionId,
+                rootSessionId: turn.rootSessionId,
+                roomId: turn.roomId,
+                inputEventId: turn.inputEventId,
+                eventId: turn.eventId,
+                eventStreamId: turn.eventStreamId,
+                payloadRef: turn.payloadRef,
+                runId: turn.runId,
+                source: turn.source,
+                status: outcome.status,
+                currentPhase: phaseName,
+                queuedAt: turn.queuedAt,
+                startedAt: turn.startedAt,
+                completedAt: at,
+                lastProgressAt: at,
+                queuedBehind: turn.queuedBehind,
+                queueDepth: 0,
+                summary: outcome.summary,
+                retentionClass: turn.retentionClass,
+                createdAt: turn.createdAt,
+                updatedAt: at,
+                metadata: turn.metadata,
+            });
+            return { turn: recovered, outcome };
+        }));
+    }
     listProviderEventsPage(providerRequestId, input = {}) {
         return listTelemetryProviderEventsPage(this.db, providerRequestId, input);
     }

package/dist/debug/index.js

@@ -989,6 +989,17 @@ function compactRunRow(run) {
         timeoutAt: run.timeoutAt,
         timeoutPhase: run.timeoutPhase,
         serviceWarning: run.serviceWarning,
+        isolation: run.resources?.isolationMode,
+        resourceUnit: run.resources?.unitName,
+        resourceLimitReason: run.resources?.limitReason,
+        minimumHostAvailableBytes: run.resources?.minimumHostAvailableBytes,
+        peakMemoryFullPsiAvg10: run.resources?.peakMemoryFullPsiAvg10,
+        peakIoFullPsiAvg10: run.resources?.peakIoFullPsiAvg10,
+        memoryPeakBytes: run.resources?.cgroup?.memoryPeakBytes,
+        memoryMaxBytes: run.resources?.cgroup?.memoryMaxBytes,
+        tasksPeak: run.resources?.cgroup?.tasksPeak,
+        ioReadBytes: run.resources?.cgroup?.ioReadBytes,
+        ioWriteBytes: run.resources?.cgroup?.ioWriteBytes,
         summary: run.summary,
     };
 }

package/dist/gateway/server.js

@@ -149,6 +149,7 @@ export class PiboGatewayServer {
             pluginRegistry: this.pluginRegistry,
             sessionStore: this.sessionStore,
             messagePreflight: createLoopMessagePreflight({ path: this.options.loopStorePath }),
+            recoverInterruptedRuntimeState: true,
         });
         this.unsubscribe = this.router.subscribe((event) => this.broadcastRouterEvent(event));
         this.server = createServer((socket) => this.handleSocket(socket));

package/dist/reliability/store.js

@@ -144,7 +144,8 @@ export class PiboReliabilityStore {
 				timeout_ms INTEGER,
 				timeout_at TEXT,
 				timeout_phase TEXT,
-				service_warning TEXT
+				service_warning TEXT,
+				resource_json TEXT
 			);
 			CREATE INDEX IF NOT EXISTS idx_pibo_runs_controller_updated
 				ON pibo_runs(controller_pibo_session_id, updated_at);
@@ -155,6 +156,7 @@ export class PiboReliabilityStore {
         ensurePiboRunColumn(this.db, "timeout_at", "TEXT");
         ensurePiboRunColumn(this.db, "timeout_phase", "TEXT");
         ensurePiboRunColumn(this.db, "service_warning", "TEXT");
+        ensurePiboRunColumn(this.db, "resource_json", "TEXT");
         this.appendEventStatement = this.db.prepare(`
 			INSERT INTO pibo_event_stream (topic, key, event_id, idempotency_key, created_at, retention_class, payload_json)
 			VALUES (?, ?, ?, ?, ?, ?, ?)
@@ -496,10 +498,10 @@ export class PiboReliabilityStore {
 				INSERT INTO pibo_runs (
 					run_id, kind, controller_pibo_session_id, status, completion_policy, consumed, tool_name,
 					summary, result_json, error, notified_status, acknowledged_status, created_at, updated_at,
-					completed_at, job_id, retryable, max_attempts, timeout_ms, timeout_at, timeout_phase, service_warning
-				) VALUES (?, 'tool', ?, 'running', ?, 0, ?, ?, NULL, NULL, NULL, NULL, ?, ?, NULL, ?, ?, ?, ?, ?, NULL, ?)
+					completed_at, job_id, retryable, max_attempts, timeout_ms, timeout_at, timeout_phase, service_warning, resource_json
+				) VALUES (?, 'tool', ?, 'running', ?, 0, ?, ?, NULL, NULL, NULL, NULL, ?, ?, NULL, ?, ?, ?, ?, ?, NULL, ?, ?)
 			`)
-            .run(runId, input.controllerPiboSessionId, input.completionPolicy, input.toolName, `${input.toolName} run is running.`, timestamp, timestamp, job.jobId, input.retryable ? 1 : 0, maxAttempts, input.timeoutMs ?? null, timeoutAt ?? null, input.serviceWarning ?? null);
+            .run(runId, input.controllerPiboSessionId, input.completionPolicy, input.toolName, `${input.toolName} run is running.`, timestamp, timestamp, job.jobId, input.retryable ? 1 : 0, maxAttempts, input.timeoutMs ?? null, timeoutAt ?? null, input.serviceWarning ?? null, input.resources ? JSON.stringify(input.resources) : null);
         this.claimJob(job.jobId, input.workerId ?? `run-registry:${process.pid}`, 24 * 60 * 60 * 1000);
         return this.requireRun(runId);
     }
@@ -527,10 +529,11 @@ export class PiboReliabilityStore {
 					timeout_ms = ?,
 					timeout_at = ?,
 					timeout_phase = ?,
-					service_warning = ?
+					service_warning = ?,
+					resource_json = ?
 				WHERE run_id = ?
 			`)
-            .run(next.status, next.completionPolicy, next.consumed ? 1 : 0, next.summary ?? null, next.result ? JSON.stringify(next.result) : null, next.error ?? null, next.notifiedStatus ?? null, next.acknowledgedStatus ?? null, next.updatedAt, next.completedAt ?? null, next.jobId ?? null, next.retryable ? 1 : 0, next.maxAttempts, next.timeoutMs ?? null, next.timeoutAt ?? null, next.timeoutPhase ?? null, next.serviceWarning ?? null, runId);
+            .run(next.status, next.completionPolicy, next.consumed ? 1 : 0, next.summary ?? null, next.result ? JSON.stringify(next.result) : null, next.error ?? null, next.notifiedStatus ?? null, next.acknowledgedStatus ?? null, next.updatedAt, next.completedAt ?? null, next.jobId ?? null, next.retryable ? 1 : 0, next.maxAttempts, next.timeoutMs ?? null, next.timeoutAt ?? null, next.timeoutPhase ?? null, next.serviceWarning ?? null, next.resources ? JSON.stringify(next.resources) : null, runId);
         return this.requireRun(runId);
     }
     getRun(runId) {
@@ -801,6 +804,8 @@ function runFromRow(row) {
         output.timeoutPhase = row.timeout_phase;
     if (row.service_warning)
         output.serviceWarning = row.service_warning;
+    if (row.resource_json)
+        output.resources = JSON.parse(row.resource_json);
     return output;
 }
 function retryDelayMs(attempts, input) {