antechamber 0.1.1 → 0.2.0

package/dist/pi-event-translation.js

@@ -2,7 +2,7 @@
 import { randomUUID } from "node:crypto";
 import { applyProviderToolResults, finishProviderToolParts, mapTool, providerToolParts, resultMetadata, resultText, } from "./tool-map.js";
 import { TodoBridge } from "./todo-bridge.js";
-import { recordTurn } from "./session-tree.js";
+import { readSessionTree, recordTurn } from "./session-tree.js";
 import { touchedFiles } from "./session-diff.js";
 import { getAgentName } from "./agent-name.js";
 import { debugLog } from "./misc-routes.js";
@@ -42,7 +42,7 @@ export function textOf(content) {
         return "";
     return content.map((b) => (b?.type === "text" && typeof b.text === "string" ? b.text : "")).join("");
 }
-export function newPart(kind, sessionID, messageID) {
+function newPart(kind, sessionID, messageID) {
     const base = {
         id: id("prt"),
         sessionID,
@@ -54,7 +54,7 @@ export function newPart(kind, sessionID, messageID) {
         ? { ...base, type: "text" }
         : { ...base, type: "reasoning" };
 }
-export function applyUsage(msg, usage) {
+function applyUsage(msg, usage) {
     if (!usage)
         return;
     msg.tokens = {
@@ -75,9 +75,17 @@ export function syncSessionCost(s) {
     if (s.messages === undefined)
         return false;
     let cost = 0;
+    const tokens = { input: 0, output: 0, reasoning: 0, cache: { read: 0, write: 0 } };
     for (const message of s.messages)
-        if (message.info.role === "assistant")
+        if (message.info.role === "assistant") {
             cost += message.info.cost;
+            tokens.input += message.info.tokens.input;
+            tokens.output += message.info.tokens.output;
+            tokens.reasoning += message.info.tokens.reasoning;
+            tokens.cache.read += message.info.tokens.cache.read;
+            tokens.cache.write += message.info.tokens.cache.write;
+        }
+    s.info.tokens = tokens;
     if ((s.info.cost ?? 0) === cost)
         return false; // absent means zero; avoid no-op updates
     s.info.cost = cost;
@@ -86,9 +94,8 @@ export function syncSessionCost(s) {
 /**
  * An OmO compaction as OpenCode shows one: a user turn holding a `compaction`
  * part (OpenChamber renders it as `/compact`) answered by a `summary` message.
- * IDs derive from the compaction's first kept entry, which the live
- * `compaction_end` result and the persisted entry share, so a history reload
- * reconciles with the live messages instead of duplicating them.
+ * IDs derive from the native compaction entry, not its retained boundary:
+ * multiple compactions can retain the same entry.
  */
 export function compactionMessages(s, key, summary, created, model) {
     const sessionID = s.info.id;
@@ -135,6 +142,33 @@ export class PiEventTranslator {
     constructor(services) {
         this.services = services;
     }
+    /** Close dead-runtime UI state before an error or idle event can reach the client. */
+    finishInterruptedRun(s, error) {
+        const now = Date.now();
+        s.approvals?.finishTurn();
+        for (const part of s.streaming?.parts.values() ?? []) {
+            if (part.type === "tool")
+                s.toolByCallId.set(part.callID, part);
+            else if ((part.type === "text" || part.type === "reasoning") &&
+                part.time !== undefined &&
+                part.time.end === undefined) {
+                part.time.end = now;
+                this.emitPart(s, part);
+            }
+        }
+        this.finalizeDanglingTools(s);
+        this.services.subagents.finishCalls(s.info.id);
+        if (s.streaming !== undefined) {
+            const message = s.streaming.message;
+            message.finish = "error";
+            message.error = error;
+            message.time.completed = now;
+            this.services.emit(s.info.directory, { type: "message.updated", properties: { info: message } });
+            this.finishStreaming(s);
+        }
+        delete s.compactingFromIdle;
+        delete s.info.time.compacting;
+    }
     setStatus(s, status) {
         s.status = status;
         this.services.emit(s.info.directory, {
@@ -142,6 +176,14 @@ export class PiEventTranslator {
             properties: { sessionID: s.info.id, status },
         });
         if (status.type === "idle") {
+            s.info.time.idle = Date.now();
+            const last = [...(s.messages ?? [])].reverse().find((message) => message.info.role === "assistant");
+            s.info.outcome =
+                s.interruptRequested || (last?.info.role === "assistant" && last.info.error?.name === "MessageAbortedError")
+                    ? "interrupted"
+                    : last?.info.role === "assistant" && last.info.finish === "error"
+                        ? "failed"
+                        : "succeeded";
             this.services.emit(s.info.directory, { type: "session.idle", properties: { sessionID: s.info.id } });
         }
     }
@@ -206,28 +248,59 @@ export class PiEventTranslator {
     }
     onPiEvent(s, ev) {
         this.services.subagents.wake(s, ev);
-        const dir = s.info.directory;
         if (envSetting("DEBUG")) {
             const ame = ev.assistantMessageEvent?.type;
             debugLog(`[pi-event] ${String(ev.type)}${ame ? `/${ame}` : ""} ${JSON.stringify(ev).slice(0, 300)}`);
         }
         switch (ev.type) {
-            case "entry_appended": {
-                s.todoBridge ??= new TodoBridge(s.info.id);
-                const event = s.todoBridge.entry(ev.entry, (part) => this.emitPart(s, part));
-                if (event !== undefined)
-                    this.services.emit(dir, event);
+            case "entry_appended":
+                this.onTodoEntry(s, ev);
                 break;
-            }
             case "session_info_changed":
                 this.services.applyNativeTitle(s, ev.name);
                 break;
             case "agent_start":
-                this.setStatus(s, { type: "busy" });
+            case "agent_idle":
+            case "agent_end":
+            case "auto_retry_start":
+            case "turn_end":
+            case "extension_error":
+                this.onLifecycleEvent(s, ev);
+                break;
+            case "tool_execution_start":
+            case "tool_execution_update":
+            case "tool_execution_end":
+                this.onToolEvent(s, ev);
                 break;
-            case "agent_end": {
+            case "message_start":
+            case "message_update":
+            case "message_end":
+                this.onMessageEvent(s, ev);
+                break;
+            case "compaction_start":
+            case "compaction_end":
+                this.onCompactionEvent(s, ev);
+                break;
+            default:
+                break; // turn_start, queue_update — future milestones
+        }
+    }
+    onTodoEntry(s, ev) {
+        s.todoBridge ??= new TodoBridge(s.info.id);
+        const event = s.todoBridge.entry(ev.entry, (part) => this.emitPart(s, part));
+        if (event !== undefined)
+            this.services.emit(s.info.directory, event);
+    }
+    onLifecycleEvent(s, ev) {
+        switch (ev.type) {
+            case "agent_start":
+                this.setStatus(s, { type: "busy" });
+                return;
+            case "agent_end":
                 if (ev.willRetry === true)
-                    break; // auto-retry keeps the session busy
+                    return; // auto-retry keeps the session busy
+                if (s.interruptRequested)
+                    this.finishInterruptedRun(s, { name: "MessageAbortedError", data: { message: "Cancelled" } });
                 s.approvals?.finishTurn();
                 this.finalizeDanglingTools(s);
                 this.services.subagents.finishCalls(s.info.id);
@@ -235,18 +308,28 @@ export class PiEventTranslator {
                 this.services.emitDiff(s);
                 s.info.time.updated = Date.now();
                 this.services.persistMeta(s);
-                this.services.emit(dir, { type: "session.updated", properties: { info: s.info } });
+                this.services.emit(s.info.directory, { type: "session.updated", properties: { info: s.info } });
                 if (s.piPath !== undefined)
                     this.services.store.update(s.info.id, { updated: Date.now() });
                 recordTurn(s);
                 this.services.persistMeta(s);
-                break;
-            }
+                return;
             case "agent_idle":
                 // agent_end closes one run, not the session: OmO may still retry or
                 // start a deferred continuation. Only agent_idle settles that work.
+                if (s.interruptRequested) {
+                    this.finishInterruptedRun(s, { name: "MessageAbortedError", data: { message: "Cancelled" } });
+                    this.services.emit(s.info.directory, {
+                        type: "session.error",
+                        properties: {
+                            sessionID: s.info.id,
+                            error: { name: "MessageAbortedError", data: { message: "Cancelled" } },
+                        },
+                    });
+                    delete s.interruptRequested;
+                }
                 this.setStatus(s, { type: "idle" });
-                break;
+                return;
             case "turn_end": {
                 s.approvals?.finishTurn();
                 // pi delivers tool results here, matched back to parts by toolCallId.
@@ -254,18 +337,41 @@ export class PiEventTranslator {
                 if (Array.isArray(results))
                     this.applyToolResults(s, results);
                 this.services.emitDiff(s);
-                break;
+                return;
             }
+            case "auto_retry_start":
+                this.setStatus(s, {
+                    type: "retry",
+                    attempt: Number(ev.attempt ?? 1),
+                    message: String(ev.errorMessage ?? "retrying"),
+                    next: Date.now() + Number(ev.delayMs ?? 0),
+                });
+                return;
+            case "extension_error":
+                this.services.emit(s.info.directory, {
+                    type: "session.error",
+                    properties: {
+                        sessionID: s.info.id,
+                        error: { name: "UnknownError", data: { message: String(ev.error ?? "pi extension error") } },
+                    },
+                });
+                return;
+            default:
+                return;
+        }
+    }
+    onToolEvent(s, ev) {
+        switch (ev.type) {
             case "tool_execution_start": {
                 if (typeof ev.toolCallId === "string" && typeof ev.toolName === "string") {
                     s.approvals?.noteToolStart(ev.toolCallId, ev.toolName, ev.args);
                 }
                 if (ev.toolName === "task")
-                    break; // Expanded when the argument stream closes.
+                    return; // Expanded when the argument stream closes.
                 const callID = typeof ev.toolCallId === "string" ? ev.toolCallId : undefined;
                 const part = callID !== undefined ? s.toolByCallId.get(callID) : undefined;
                 if (part === undefined)
-                    break;
+                    return;
                 const name = typeof ev.toolName === "string" ? ev.toolName : part.tool;
                 const mapped = mapTool(name, name === "todo" ? part.state.input : ev.args);
                 part.tool = mapped.tool;
@@ -277,31 +383,37 @@ export class PiEventTranslator {
                     time: { start: part.state.status !== "pending" ? part.state.time.start : Date.now() },
                 };
                 this.emitPart(s, part);
-                break;
+                return;
             }
             case "tool_execution_update": {
                 if (typeof ev.toolCallId === "string" &&
                     this.services.subagents.result(s, ev.toolCallId, ev.partialResult !== null && typeof ev.partialResult === "object" ? ev.partialResult : {}, false))
-                    break;
+                    return;
                 const part = typeof ev.toolCallId === "string" ? s.toolByCallId.get(ev.toolCallId) : undefined;
                 if (part === undefined || part.state.status !== "running")
-                    break;
+                    return;
                 this.emitPart(s, part);
-                break;
+                return;
             }
             case "tool_execution_end": {
                 if (typeof ev.toolCallId === "string")
                     s.approvals?.noteToolEnd(ev.toolCallId);
                 const result = ev.result !== null && typeof ev.result === "object" ? ev.result : {};
                 this.applyToolResults(s, [{ ...result, toolCallId: ev.toolCallId, isError: ev.isError }]);
-                break;
+                return;
             }
+            default:
+                return;
+        }
+    }
+    onMessageEvent(s, ev) {
+        switch (ev.type) {
             case "message_start": {
                 if (this.reconcileUser(s, ev))
-                    break;
+                    return;
                 const msg = ev.message;
                 if (msg?.role !== "assistant")
-                    break; // user message already recorded at prompt()
+                    return; // user message already recorded at prompt()
                 // pi stamps the assistant message with the model really serving it.
                 // Trust that over our snapshot, which goes stale whenever pi switches on
                 // its own side (subagent routing, an in-session /model, a resumed file).
@@ -324,91 +436,121 @@ export class PiEventTranslator {
                 };
                 s.streaming = { message: assistant, parts: new Map() };
                 s.messages?.push({ info: assistant, parts: [] });
-                this.services.emit(dir, { type: "message.updated", properties: { info: assistant } });
-                break;
+                this.services.emit(s.info.directory, { type: "message.updated", properties: { info: assistant } });
+                return;
             }
             case "message_update":
                 this.onMessageUpdate(s, ev);
-                break;
+                return;
             case "message_end": {
                 if (this.reconcileUser(s, ev))
-                    break;
+                    return;
                 const ended = ev.message;
-                if (ended?.role === "assistant" && s.streaming) {
-                    this.onProviderTools(s, ev.message);
-                    if (ended.stopReason === "aborted" || ended.stopReason === "error") {
-                        for (const part of finishProviderToolParts(s.toolByCallId.values(), Date.now(), ended.stopReason === "aborted" ? "Cancelled" : "No result"))
-                            this.emitPart(s, part);
-                    }
-                    // Last chance to correct attribution: some providers only reveal the
-                    // concrete model once the stream has closed.
-                    if (this.services.applyPiModel(s, ended.provider, piModelId(ended.responseModel, ended.model)) &&
-                        s.model !== undefined) {
-                        s.streaming.message.providerID = s.model.providerID;
-                        s.streaming.message.modelID = s.model.modelID;
-                    }
-                    applyUsage(s.streaming.message, ended.usage);
-                    Object.assign(s.streaming.message, piStopReason(ended.stopReason, ended.errorMessage));
-                    s.streaming.message.time.completed = Date.now();
-                    this.services.emit(dir, { type: "message.updated", properties: { info: s.streaming.message } });
-                    this.finishStreaming(s);
-                    if (syncSessionCost(s))
-                        this.services.emit(dir, { type: "session.updated", properties: { info: s.info } });
+                if (ended?.role !== "assistant" || !s.streaming)
+                    return;
+                this.onProviderTools(s, ev.message);
+                if (ended.stopReason === "aborted" || ended.stopReason === "error") {
+                    for (const part of finishProviderToolParts(s.toolByCallId.values(), Date.now(), ended.stopReason === "aborted" ? "Cancelled" : "No result"))
+                        this.emitPart(s, part);
                 }
-                break;
+                // Last chance to correct attribution: some providers only reveal the
+                // concrete model once the stream has closed.
+                if (this.services.applyPiModel(s, ended.provider, piModelId(ended.responseModel, ended.model)) &&
+                    s.model !== undefined) {
+                    s.streaming.message.providerID = s.model.providerID;
+                    s.streaming.message.modelID = s.model.modelID;
+                }
+                applyUsage(s.streaming.message, ended.usage);
+                Object.assign(s.streaming.message, piStopReason(ended.stopReason, ended.errorMessage));
+                s.streaming.message.time.completed = Date.now();
+                this.services.emit(s.info.directory, { type: "message.updated", properties: { info: s.streaming.message } });
+                this.finishStreaming(s);
+                if (syncSessionCost(s))
+                    this.services.emit(s.info.directory, { type: "session.updated", properties: { info: s.info } });
+                return;
             }
-            case "auto_retry_start":
-                this.setStatus(s, {
-                    type: "retry",
-                    attempt: Number(ev.attempt ?? 1),
-                    message: String(ev.errorMessage ?? "retrying"),
-                    next: Date.now() + Number(ev.delayMs ?? 0),
-                });
-                break;
-            case "extension_error":
-                this.services.emit(dir, {
-                    type: "session.error",
+            default:
+                return;
+        }
+    }
+    onCompactionEvent(s, ev) {
+        switch (ev.type) {
+            case "compaction_start":
+                s.compactionID = id("compaction");
+                this.services.emit(s.info.directory, {
+                    type: "session.compaction.started",
                     properties: {
                         sessionID: s.info.id,
-                        error: { name: "UnknownError", data: { message: String(ev.error ?? "pi extension error") } },
+                        inputID: s.compactionID,
+                        reason: typeof ev.reason === "string" ? ev.reason : "manual",
                     },
                 });
-                break;
-            case "compaction_start": {
                 s.info.time.compacting = Date.now();
-                this.services.emit(dir, { type: "session.updated", properties: { info: s.info } });
+                this.services.emit(s.info.directory, { type: "session.updated", properties: { info: s.info } });
                 // A manual /compact runs no agent turn, so nothing else marks the session busy.
                 if (s.status.type === "idle") {
                     s.compactingFromIdle = true;
                     this.setStatus(s, { type: "busy" });
                 }
-                break;
-            }
+                return;
             case "compaction_end": {
                 delete s.info.time.compacting;
                 const result = ev.result;
+                const inputID = s.compactionID ?? id("compaction");
+                delete s.compactionID;
+                this.services.emit(s.info.directory, ev.aborted === true || typeof result?.summary !== "string"
+                    ? {
+                        type: "session.compaction.failed",
+                        properties: {
+                            sessionID: s.info.id,
+                            inputID,
+                            reason: typeof ev.reason === "string" ? ev.reason : "manual",
+                            error: {
+                                name: "CompactionError",
+                                data: { message: ev.aborted === true ? "Compaction cancelled" : "Compaction failed" },
+                            },
+                        },
+                    }
+                    : {
+                        type: "session.compaction.ended",
+                        properties: {
+                            sessionID: s.info.id,
+                            inputID,
+                            reason: typeof ev.reason === "string" ? ev.reason : "manual",
+                            text: result.summary,
+                            recent: "",
+                        },
+                    });
                 if (ev.aborted !== true && typeof result?.summary === "string" && typeof result.firstKeptEntryId === "string") {
                     const messages = this.services.ensureMessages(s);
-                    for (const message of compactionMessages(s, result.firstKeptEntryId, result.summary, Date.now(), s.model)) {
+                    let key;
+                    // compaction_end follows persistence, but its result omits the entry ID.
+                    if (s.piPath !== undefined) {
+                        for (const entry of readSessionTree(s.piPath).entries) {
+                            if (entry.type === "compaction")
+                                key = entry.id;
+                        }
+                    }
+                    for (const message of compactionMessages(s, key ?? id("compaction"), result.summary, Date.now(), s.model)) {
                         if (messages.some((item) => item.info.id === message.info.id))
                             continue;
                         // Keep a hidden redo branch after the new turn, as appendMessage does.
                         const boundary = messages.findIndex((item) => item.info.id === s.tree.revert?.messageID);
                         messages.splice(boundary < 0 ? messages.length : boundary, 0, message);
-                        this.services.emit(dir, { type: "message.updated", properties: { info: message.info } });
+                        this.services.emit(s.info.directory, { type: "message.updated", properties: { info: message.info } });
                         for (const part of message.parts)
                             this.emitPart(s, part);
                     }
                 }
-                this.services.emit(dir, { type: "session.updated", properties: { info: s.info } });
+                this.services.emit(s.info.directory, { type: "session.updated", properties: { info: s.info } });
                 if (s.compactingFromIdle) {
                     delete s.compactingFromIdle;
                     this.setStatus(s, { type: "idle" });
                 }
-                break;
+                return;
             }
             default:
-                break; // turn_start, queue_update — future milestones
+                return;
         }
     }
     finishStreaming(s) {
@@ -449,7 +591,19 @@ export class PiEventTranslator {
         const textPart = part;
         if (ame.type === `${wire}_delta` && typeof ame.delta === "string") {
             textPart.text += ame.delta;
-            this.emitPart(s, textPart, ame.delta);
+            this.services.emit(dir, {
+                type: "message.part.delta",
+                properties: {
+                    sessionID: s.info.id,
+                    messageID: textPart.messageID,
+                    partID: textPart.id,
+                    field: "text",
+                    delta: ame.delta,
+                },
+            });
+            // Append before publishing the authoritative snapshot. Sending another
+            // append delta on the snapshot would duplicate text in streaming clients.
+            this.emitPart(s, textPart);
         }
         else if (ame.type === `${wire}_end`) {
             if (typeof ame.content === "string")

package/dist/pi-rpc.d.ts

@@ -21,7 +21,20 @@ export declare const RPC_COMMAND_TIMEOUT_MS = 60000;
  * long before a large session finished, leaving the run orphaned.
  */
 export declare const COMPACT_RPC_TIMEOUT_MS: number;
-export declare const MAX_JSONL_FRAME_BYTES: number;
+export declare class UpstreamProtocolError extends Error {
+    constructor(message: string);
+}
+/** A process's group and kernel start time: with the PID, the identity that survives PID reuse checks. */
+interface ProcessIdentity {
+    readonly pid: number;
+    readonly group: number;
+    readonly start: string;
+}
+/** Reads process identities so disposal can tell owned survivors from a reused PID. */
+export interface ProcessTable {
+    identity(pid: number): ProcessIdentity | undefined;
+    groupMembers(group: number): ProcessIdentity[];
+}
 export type PiRpcResponse<T = unknown> = {
     readonly id: string;
     readonly type: "response";
@@ -35,13 +48,14 @@ export type PiRpcResponse<T = unknown> = {
     readonly error?: string;
     readonly data?: never;
 });
-export interface PiRpcClientOptions {
+interface PiRpcClientOptions {
     cwd: string;
     /** Path/name of the pi binary. Defaults to ANTECHAMBER_PI_BINARY env or "omo" from PATH. */
     piBinary?: string;
     /** Extra CLI flags, e.g. ["--no-session"] or ["--provider", "x"]. */
     args?: string[];
     clock?: LifecycleClock;
+    processTable?: ProcessTable;
 }
 export declare class PiRpcClient extends EventEmitter {
     private options;
@@ -61,6 +75,9 @@ export declare class PiRpcClient extends EventEmitter {
     constructor(options: PiRpcClientOptions);
     private drain;
     private frameOverflow;
+    /** Node has reaped the leader, so its PID (and group ID) may already belong to another process. */
+    private reaped;
+    private closeForTransportFailure;
     private handleLine;
     private handleExtensionUI;
     respondExtensionUI(response: Record<string, unknown>): void;
@@ -72,3 +89,4 @@ export declare class PiRpcClient extends EventEmitter {
     notify(cmd: Omit<PiRpcCommand, "id">): void;
     dispose(): Promise<void>;
 }
+export {};