@yusukeshib/pi-babysit 0.3.13 → 0.3.14

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yusukeshib/pi-babysit",
-  "version": "0.3.13",
+  "version": "0.3.14",
   "description": "Run any shell command and pi subagents under babysit, with context-safe captured output.",
   "keywords": [
     "pi-package",

package/rpc-stream-proxy.mjs

@@ -7,15 +7,45 @@
  * are useful only while streaming; `message_end` is the authoritative final
  * message. Persisting every snapshot makes babysit logs grow quadratically.
  *
- * This proxy forwards stdin byte-for-byte and compacts only `message_update`
- * lines on stdout. All final, tool, lifecycle, response, and error events pass
- * through byte-for-byte.
+ * This proxy forwards stdin byte-for-byte. `message_update` snapshots are
+ * always compacted. In opt-in `compact` log mode, redundant payloads on
+ * lifecycle/tool events are also removed while authoritative `message_end`,
+ * response, and error events remain intact. Set PI_BABYSIT_RPC_LOG_MODE=standard
+ * to retain the legacy lifecycle payloads.
  */
 import { spawn } from "node:child_process";
 import { StringDecoder } from "node:string_decoder";
 import { pathToFileURL } from "node:url";
 
-export function compactRpcLine(line) {
+function isParkedMessages(messages) {
+	if (!Array.isArray(messages)) return false;
+	for (let index = messages.length - 1; index >= 0; index--) {
+		const message = messages[index];
+		if (message?.role !== "toolResult") continue;
+		if (message.toolName === "process") return true;
+		if (message.toolName !== "babysit_run") continue;
+		if (message.details?.kind === "process" && message.details.status === "started") {
+			return true;
+		}
+		const text = Array.isArray(message.content)
+			? message.content
+					.filter((part) => part?.type === "text" && typeof part.text === "string")
+					.map((part) => part.text)
+					.join("")
+			: typeof message.content === "string"
+				? message.content
+				: "";
+		if (/^Process started \(id: [^)]+\)\. \[notify-on-exit\]\nLog: /.test(text)) {
+			return true;
+		}
+	}
+	return false;
+}
+
+export function compactRpcLine(
+	line,
+	mode = process.env.PI_BABYSIT_RPC_LOG_MODE ?? "standard",
+) {
 	if (!line.trimStart().startsWith("{")) return line;
 
 	let event;
@@ -24,14 +54,44 @@ export function compactRpcLine(line) {
 	} catch {
 		return line;
 	}
-	if (event?.type !== "message_update") return line;
 
-	const { message: _cumulativeMessage, assistantMessageEvent, ...rest } = event;
-	if (!assistantMessageEvent || typeof assistantMessageEvent !== "object") {
-		return JSON.stringify(rest);
+	if (event?.type === "message_update") {
+		const { message: _cumulativeMessage, assistantMessageEvent, ...rest } = event;
+		if (!assistantMessageEvent || typeof assistantMessageEvent !== "object") {
+			return JSON.stringify(rest);
+		}
+		const { partial: _cumulativePartial, ...deltaEvent } = assistantMessageEvent;
+		return JSON.stringify({ ...rest, assistantMessageEvent: deltaEvent });
+	}
+	if (mode !== "compact") return line;
+
+	switch (event?.type) {
+		case "message_start": {
+			const { message, ...rest } = event;
+			return JSON.stringify({
+				...rest,
+				message:
+					message && typeof message === "object"
+						? { role: message.role, toolName: message.toolName, toolCallId: message.toolCallId }
+						: message,
+			});
+		}
+		case "turn_end": {
+			const { message: _message, toolResults: _toolResults, ...rest } = event;
+			return JSON.stringify(rest);
+		}
+		case "agent_end": {
+			const { messages, ...rest } = event;
+			return JSON.stringify({ ...rest, piBabysitParked: isParkedMessages(messages) });
+		}
+		case "tool_execution_end": {
+			if (event.isError) return line;
+			const { result: _result, ...rest } = event;
+			return JSON.stringify(rest);
+		}
+		default:
+			return line;
 	}
-	const { partial: _cumulativePartial, ...deltaEvent } = assistantMessageEvent;
-	return JSON.stringify({ ...rest, assistantMessageEvent: deltaEvent });
 }
 
 function transformRpcOutput(input, output) {

package/self-reap.ts

@@ -23,10 +23,8 @@
  */
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 
-// Keep in sync with NOTIFY_MARKER in index.ts (not imported: this file is
-// loaded standalone into the subagent process via --extension).
-const NOTIFY_MARKER = "[notify-on-exit]";
-
+// Loaded standalone into the subagent process via --extension, so the parked
+// result-shape check below intentionally mirrors index.ts without importing it.
 function parseDurMs(s: string | undefined): number | null {
 	if (!s || s === "none" || s === "off" || s === "0") return null;
 	const m = /^(\d+)(ms|s|m|h)?$/.exec(s.trim());
@@ -38,8 +36,15 @@ function parseDurMs(s: string | undefined): number | null {
 
 // Same parked-turn rule as the parent: scan the run because models sometimes
 // add an assistant note after the marker-bearing tool result.
-function isParked(
-	messages: { role?: string; toolName?: string; content?: unknown }[] | undefined,
+export function isParked(
+	messages:
+		| {
+				role?: string;
+				toolName?: string;
+				content?: unknown;
+				details?: { kind?: string; status?: string };
+			}[]
+		| undefined,
 ): boolean {
 	if (!messages) return false;
 	for (let index = messages.length - 1; index >= 0; index--) {
@@ -47,12 +52,20 @@ function isParked(
 		if (message?.role !== "toolResult") continue;
 		if (message.toolName === "process") return true; // legacy pi-processes
 		if (message.toolName !== "babysit_run") continue;
-		try {
-			const serialized = JSON.stringify(message.content ?? null);
-			if (serialized === "null" || serialized.includes(NOTIFY_MARKER)) return true;
-		} catch {
+		if (message.details?.kind === "process" && message.details.status === "started") {
 			return true;
 		}
+		const text = Array.isArray(message.content)
+			? message.content
+					.filter((part): part is { type: "text"; text: string } =>
+						Boolean(part && typeof part === "object" && part.type === "text" && typeof part.text === "string"),
+					)
+					.map((part) => part.text)
+					.join("")
+			: typeof message.content === "string"
+				? message.content
+				: "";
+		if (/^Process started \(id: [^)]+\)\. \[notify-on-exit\]\nLog: /.test(text)) return true;
 	}
 	return false;
 }