@tshu-w/pi-agents 0.1.0 → 0.1.1

package/bin/daemon.mjs

@@ -152,7 +152,7 @@ function openWorkbench(channel) {
   channel.onMessage(message => {
     if (typeof message?.id !== 'string') return;
     if (message.type === 'interrupt') channels.get(message.id)?.send({ type: 'interrupt', id: message.id });
-    else if (message.type === 'show' || message.type === 'hide') {
+    else if (message.type === 'show' || message.type === 'hide' || message.type === 'see') {
       sessions[message.type](message.id);
       changed();
     }

package/docs/WORKBENCH.md

@@ -106,7 +106,8 @@ still running (`●`) or has exited (`○`), and the color marks a finished (gre
 or failed (red) turn, or a dialog waiting for the user (yellow). The header
 shows Pi's logo, the size of each group, and key hints.
 
-A turn counts as seen once it has been shown in an attached terminal. Sessions
+A turn counts as seen once it has been shown in an attached terminal or a
+preview. Sessions
 from before the workbench start out seen. A Session that holds its lock but is
 not reporting to the daemon, as while its Pi exits, shows as unknown (`?`); the
 daemon checks it again every second until that changes.
@@ -127,7 +128,8 @@ Keys:
   - `Esc` clears a typed message; with none typed, it closes the preview.
 - `Ctrl+X` stops the current turn, as `app.interrupt` does.
 - `Ctrl+D` removes the selected Session from the list. Its file stays, and
-  `/resume` brings it back. A running Session must quit first.
+  `/resume` brings it back. An idle Session in a host that no terminal is
+  attached to quits first; any other running Session must quit itself.
 - `Ctrl+Z` (`pi-agents.detach`) exits the workbench, as detaching leaves a
   Session; every Session keeps running.
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@tshu-w/pi-agents",
-  "version": "0.1.0",
-  "description": "Multi-agent runtime for Pi: async agents, programmatic delegation, and inter-agent messaging",
+  "version": "0.1.1",
+  "description": "Multi-agent runtime for Pi: async agents, programmatic delegation, inter-agent messaging, and a workbench for your Sessions",
   "license": "AGPL-3.0-only",
   "repository": {
     "type": "git",
@@ -21,8 +21,10 @@
   "keywords": [
     "pi-package",
     "pi-extension",
-    "agents",
-    "multi-agent"
+    "pi-coding-agent",
+    "pi",
+    "multi-agent",
+    "rlm"
   ],
   "pi": {
     "extensions": [
@@ -30,7 +32,7 @@
     ]
   },
   "scripts": {
-    "test": "node --test --test-timeout=30000 tests/",
+    "test": "node --test --test-timeout=30000 \"tests/*.test.mjs\"",
     "typecheck": "tsc -p tsconfig.json"
   },
   "dependencies": {

package/src/daemon/sessions.mjs

@@ -87,6 +87,13 @@ export async function createSessions(paths, sessionDir) {
       (marks[id] ??= {}).listed = true;
       save();
     },
+    /** Marks a Session's last turn seen, as when the user previews it. */
+    see(id) {
+      const mark = marks[id];
+      if (!mark?.last || (mark.seen ?? 0) >= mark.last.at) return;
+      mark.seen = mark.last.at;
+      save();
+    },
     /** Removes a Session from the workbench list and clears its unseen turn; its file stays. */
     hide(id) {
       const mark = marks[id];

package/src/host/client.mjs

@@ -27,7 +27,7 @@ function hostStatus(path) {
   });
 }
 
-function alive(pid) {
+export function alive(pid) {
   try { process.kill(pid, 0); return true; }
   catch (error) { return error.code !== 'ESRCH'; }
 }

package/src/workbench.mjs

@@ -7,7 +7,7 @@ import { createRequire } from 'node:module';
 import { join } from 'node:path';
 import { pathToFileURL } from 'node:url';
 import { setTimeout as delay } from 'node:timers/promises';
-import { attach, listHosts, startHost } from './host/client.mjs';
+import { alive, attach, listHosts, startHost } from './host/client.mjs';
 import { detachKey, detachMatcher } from './host/keys.mjs';
 import { piPackage } from './pi.mjs';
 import { ensureDaemon, sendViaDaemon } from './daemon/client.mjs';
@@ -66,6 +66,8 @@ function age(ms) {
 
 const time = session => Math.max(session.finishedAt ?? 0, Date.parse(session.updatedAt ?? '') || 0);
 const label = session => session.name || session.title || session.id.slice(0, 8);
+// Notices quote the label, since a title can read as part of the notice.
+const named = session => `"${label(session)}"`;
 const COMMANDS = [
   { value: '/resume', label: 'resume', description: 'Open a past Session' },
   { value: '/quit', label: 'quit', description: 'Exit the workbench' },
@@ -283,7 +285,7 @@ export async function workbench({ pi: executable, paths, sessionDir }) {
         return say('Press Ctrl+C again to quit');
       }
       if (T.matchesKey(data, 'ctrl+x')) return interrupt();
-      if (empty && keybindings.matches(data, 'app.session.delete')) return remove(selected()?.session);
+      if (empty && keybindings.matches(data, 'app.session.delete')) return void remove(selected()?.session);
       // The preview's mention stays: an edit that would remove or change it is undone.
       if (pane) {
         const before = editor.getText();
@@ -366,9 +368,9 @@ export async function workbench({ pi: executable, paths, sessionDir }) {
       const hosted = (await listHosts(paths)).find(host => host.session?.id === session.id);
       if (hosted) return await attachTo(hosted.socket);
       if (session.running || session.status === 'unknown') {
-        return say(`${label(session)} runs outside a host; attach it after it exits.`, 'warning');
+        return say(`${named(session)} runs outside a host; attach it after it exits.`, 'warning');
       }
-      say(`Starting ${label(session)}…`);
+      say(`Starting ${named(session)}…`);
       const socket = await startHost(paths, {
         pi: executable, args: ['--session', session.sessionFile], cwd: session.cwd,
         cols: process.stdout.columns, rows: process.stdout.rows, waitForTerminal: true,
@@ -385,20 +387,32 @@ export async function workbench({ pi: executable, paths, sessionDir }) {
     const session = selected()?.session;
     if (!['working', 'blocked'].includes(session?.status) || !stream) return say('The selected Session is not working.', 'warning');
     stream.send({ type: 'interrupt', id: session.id });
-    say(`Stopping ${label(session)}…`);
+    say(`Stopping ${named(session)}…`);
   }
 
-  /** Removes a Session from the list; `/resume` brings it back. A running Session always stays listed. */
-  function remove(session) {
+  /**
+   * Removes a Session from the list; `/resume` brings it back. An idle Session in a host no terminal
+   * is attached to quits first, as Pi does on SIGTERM; any other running Session always stays listed.
+   */
+  async function remove(session) {
     if (!session || !stream) return;
-    if (session.running || session.status !== 'idle') return say(`Quit ${label(session)} before removing it.`, 'warning');
+    if (session.running) {
+      const host = session.status === 'idle' && (await listHosts(paths)).find(host => host.session?.id === session.id);
+      if (!host || host.attached) return say(`Quit ${named(session)} before removing it.`, 'warning');
+      say(`Quitting ${named(session)}…`);
+      process.kill(host.pid, 'SIGTERM');
+      for (let wait = 0; alive(host.pid); wait += 100) {
+        if (wait >= 5000) return say(`${named(session)} did not quit; attach it to see why.`, 'warning');
+        await delay(100);
+      }
+    } else if (session.status !== 'idle') return say(`Quit ${named(session)} before removing it.`, 'warning');
     stream.send({ type: 'hide', id: session.id });
     if (pane?.session.id === session.id) pane.close();
     const choices = selectable();
     const index = choices.findIndex(row => row.key === session.id);
     selectedKey = (choices[index + 1] ?? choices[index - 1])?.key;
     sessions = sessions.filter(other => other.id !== session.id);
-    say(`Removed ${label(session)}; /resume brings it back.`);
+    say(`Removed ${named(session)}; /resume brings it back.`);
   }
 
   async function preview(session) {
@@ -409,17 +423,22 @@ export async function workbench({ pi: executable, paths, sessionDir }) {
     let body = conversation(messages);
     const border = themes.getEditorTheme().borderColor;
     let scroll = 0, page = 1, timer;
+    // A turn the preview shows counts as seen.
+    const see = () => stream?.send({ type: 'see', id: session.id });
+    see();
     // Pi appends each finished message to the file, so the preview follows the Session as it works.
     const watcher = watch(session.sessionFile, () => {
       clearTimeout(timer);
       timer = setTimeout(async () => {
         try { body = conversation(await branchMessages(session.sessionFile)); }
         catch (error) { return say(error.message, 'error'); }
+        see();
         ui.requestRender();
       }, 100);
     });
     const close = () => {
       watcher.close();
+      see();
       clearTimeout(timer);
       // A draft stays, still addressed to the Session.
       if (editor.getText() === pane?.mention) editor.setText('');
@@ -507,12 +526,12 @@ export async function workbench({ pi: executable, paths, sessionDir }) {
   }
 
   async function send(session, body) {
-    say(`Sending to ${label(session)}…`);
+    say(`Sending to ${named(session)}…`);
     const message = { id: randomUUID(), user: true, recipient: session.id, body };
     try {
       const receipt = await sendViaDaemon(paths, message, { sessionDir, launch: { command: executable, args: [], env } });
-      if (receipt?.accepted !== true || receipt.messageId !== message.id) throw new Error(`Invalid acknowledgement from ${label(session)}; it may have received the message.`);
-      say(`Sent to ${label(session)}.`);
+      if (receipt?.accepted !== true || receipt.messageId !== message.id) throw new Error(`Invalid acknowledgement from ${named(session)}; it may have received the message.`);
+      say(`Sent to ${named(session)}.`);
     } catch (error) { say(error.message, 'error'); }
   }