@skydiveai/pi-extensions 0.1.0-beta.1544 → 0.1.0-beta.1545

package/dist/index.mjs

@@ -2264,6 +2264,19 @@ async function postBackgroundTaskDone({ messageId, content }) {
 	} });
 	if (!res.ok) throw new Error(`bg-task-done POST failed: ${res.status}`);
 }
+function postBackgroundTasksSnapshot({ messageId, tasks }) {
+	const client = sandboxClient();
+	if (!client || !messageId) return;
+	client["bg-tasks"].$post({ json: {
+		messageId,
+		tasks
+	} }).catch((err) => {
+		log$10.debug({
+			err,
+			event: "bg_tasks_snapshot_failed"
+		}, "bg-tasks snapshot publish failed");
+	});
+}
 async function postSubagentSpawn({ messageId, tasks }) {
 	const client = sandboxClient();
 	if (!client) throw new Error("no api url for subagent-spawn");
@@ -3695,6 +3708,7 @@ const WATCHDOG_INTERVAL_MS = 3e4;
 const KEEPALIVE_EVERY_MS = 6e4;
 const KEEPALIVE_MAX_MS = 3600 * 1e3;
 const STALL_HINT_AFTER_MS = 120 * 1e3;
+const PUBLISH_DEBOUNCE_MS = 300;
 const MAX_LOG_BYTES = 100 * 1024 * 1024;
 const DEFAULT_TAIL_LINES = 30;
 const TAIL_READ_BYTES = 64 * 1024;
@@ -3888,6 +3902,80 @@ This is a background-task completion, not a message from the user. If it needs n
 				});
 			} else log.info({ taskId: meta.id }, "bg task completed idle with no origin message; deferring to next session_start");
 		}
+		const SNAPSHOT_TAIL_LINES = 40;
+		let lastPublishedSignature = null;
+		let publishInFlight = null;
+		let publishQueued = false;
+		function snapshotSignature(snapshot) {
+			return JSON.stringify(snapshot.map((t) => ({
+				id: t.id,
+				state: t.state,
+				exitCode: t.exitCode,
+				killedReason: t.killedReason,
+				outputTail: t.outputTail
+			})));
+		}
+		async function doPublishSnapshot() {
+			if (!messageId) return;
+			const mine = [...tasks.values()].filter((t) => sameScope(t, scopeKey()));
+			try {
+				const snapshot = await Promise.all(mine.map(async (t) => ({
+					id: t.id,
+					command: t.command,
+					description: t.description,
+					startedAt: t.startedAt,
+					state: t.running ? "running" : "finished",
+					exitCode: t.exitCode,
+					killedReason: t.killedReason,
+					outputTail: await tailLog(t.id, SNAPSHOT_TAIL_LINES)
+				})));
+				const signature = snapshotSignature(snapshot);
+				if (signature === lastPublishedSignature) return;
+				lastPublishedSignature = signature;
+				postBackgroundTasksSnapshot({
+					messageId,
+					tasks: snapshot
+				});
+			} catch (err) {
+				log.debug({
+					err,
+					event: "bg_tasks_snapshot_build_failed"
+				}, "building bg-tasks snapshot failed");
+			}
+		}
+		async function publishSnapshotNow() {
+			if (publishInFlight) {
+				publishQueued = true;
+				return;
+			}
+			publishInFlight = (async () => {
+				try {
+					do {
+						publishQueued = false;
+						await doPublishSnapshot();
+					} while (publishQueued);
+				} finally {
+					publishInFlight = null;
+				}
+			})();
+			await publishInFlight;
+		}
+		let publishTimer = null;
+		function schedulePublishSnapshot() {
+			if (publishTimer) return;
+			publishTimer = setTimeout(() => {
+				publishTimer = null;
+				publishSnapshotNow();
+			}, PUBLISH_DEBOUNCE_MS);
+			publishTimer.unref?.();
+		}
+		async function flushPublishSnapshot() {
+			if (publishTimer) {
+				clearTimeout(publishTimer);
+				publishTimer = null;
+			}
+			await publishSnapshotNow();
+		}
 		async function launchTask({ command, description, cwd }) {
 			taskCounter += 1;
 			const id = `bg-${process.pid.toString(36)}-${taskCounter}`;
@@ -3950,6 +4038,7 @@ This is a background-task completion, not a message from the user. If it needs n
 					taskId: id,
 					exitCode: meta.exitCode
 				}, "bg task finished");
+				schedulePublishSnapshot();
 				await notifyCompletion(meta);
 			});
 			ensureWatchdog();
@@ -3957,6 +4046,7 @@ This is a background-task completion, not a message from the user. If it needs n
 				taskId: id,
 				conversationId
 			}, "bg task started");
+			schedulePublishSnapshot();
 			return meta;
 		}
 		function knownTaskIds() {
@@ -3979,6 +4069,8 @@ This is a background-task completion, not a message from the user. If it needs n
 				count: unnotified.length
 			}, "injected completed bg tasks at session_start");
 			if ([...tasks.values()].some((t) => sameScope(t, scopeKey()) && t.running)) ensureWatchdog();
+			lastPublishedSignature = null;
+			await flushPublishSnapshot();
 		});
 		function err(text) {
 			return {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skydiveai/pi-extensions",
-  "version": "0.1.0-beta.1544",
+  "version": "0.1.0-beta.1545",
   "homepage": "https://skydive.com",
   "license": "MIT",
   "author": "Create, Inc.",