@thurstonsand/pi-agent-switchboard 0.0.0 → 0.1.0

package/index.js

@@ -97,6 +97,7 @@ function branch(cwd) {
 }
 
 // packages/pi/src/index.ts
+import { Type } from "typebox";
 var KEY = Symbol.for("agent-switchboard.recorder");
 var store = globalThis;
 function claim() {
@@ -230,6 +231,16 @@ function src_default(pi) {
   pi.on("session_info_changed", (event, ctx) => {
     writeFor(ctx, (db, id) => db.run("UPDATE sessions SET title = ? WHERE session_id = ?", event.name ?? null, id));
   });
+  const markArchived = (ctx) => {
+    let archived = false;
+    writeFor(ctx, (db, id) => {
+      if (!db.get("SELECT 1 FROM sessions WHERE session_id = ?", id))
+        return;
+      db.run("INSERT INTO marks (session_id, archived_at) VALUES (?, ?) ON CONFLICT (session_id) DO UPDATE SET archived_at = excluded.archived_at", id, Date.now());
+      archived = true;
+    });
+    return archived;
+  };
   pi.registerCommand("archive", {
     description: "Archive this session in swb and quit",
     handler: async (_args, ctx) => {
@@ -237,19 +248,33 @@ function src_default(pi) {
         ctx.ui.notify("swb: turn running: wait for it to complete", "error");
         return;
       }
-      let archived = false;
-      writeFor(ctx, (db, id) => {
-        if (!db.get("SELECT 1 FROM sessions WHERE session_id = ?", id))
-          return;
-        db.run("INSERT INTO marks (session_id, archived_at) VALUES (?, ?) ON CONFLICT (session_id) DO UPDATE SET archived_at = excluded.archived_at", id, Date.now());
-        archived = true;
-      });
-      if (archived)
+      if (markArchived(ctx))
         ctx.shutdown();
       else if (!proc.failure)
         ctx.ui.notify("swb: nothing to archive before the first prompt", "error");
     }
   });
+  pi.registerTool({
+    name: "swb_archive",
+    label: "Archive session",
+    description: "Archive this session in Agent Switchboard. It leaves the user's roster of open sessions, and pi quits as soon as this turn ends. The user can restore it with `swb unarchive <id>`.",
+    promptSnippet: "Archive this session when the user asks",
+    promptGuidelines: [
+      "Use swb_archive only when the user asks to archive this session; never archive on your own initiative.",
+      "Call swb_archive as the last action of the turn, then end with a brief final message: pi quits once the turn ends."
+    ],
+    parameters: Type.Object({}),
+    async execute(_toolCallId, _params, _signal, _onUpdate, ctx) {
+      if (!markArchived(ctx)) {
+        throw new Error(proc.failure ? `swb: not recording: ${proc.failure}` : "swb: no session recorded yet");
+      }
+      ctx.shutdown();
+      return {
+        content: [{ type: "text", text: "Archived. pi quits when this turn ends." }],
+        details: undefined
+      };
+    }
+  });
   pi.on("session_shutdown", (event, ctx) => {
     if (event.reason !== "quit")
       return;

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@thurstonsand/pi-agent-switchboard",
-	"version": "0.0.0",
+	"version": "0.1.0",
 	"type": "module",
 	"description": "Records managed pi sessions for Agent Switchboard (swb)",
 	"license": "MIT",
@@ -17,6 +17,7 @@
 		]
 	},
 	"peerDependencies": {
-		"@earendil-works/pi-coding-agent": "*"
+		"@earendil-works/pi-coding-agent": "*",
+		"typebox": "*"
 	}
 }