@schilderlabs/pitown-package 0.2.6 → 0.2.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@schilderlabs/pitown-package",
-  "version": "0.2.6",
+  "version": "0.2.7",
   "type": "module",
   "types": "./dist/index.d.mts",
   "exports": {
@@ -17,7 +17,7 @@
   },
   "dependencies": {
     "@mariozechner/pi-ai": "0.58.3",
-    "@schilderlabs/pitown-core": "0.2.6"
+    "@schilderlabs/pitown-core": "0.2.7"
   },
   "devDependencies": {
     "@mariozechner/pi-coding-agent": "0.58.3",

package/pi/extensions/index.test.ts

@@ -175,7 +175,7 @@ describe("pi town extension", () => {
 				message: expect.objectContaining({
 					customType: "pitown-context",
 					display: false,
-					content: expect.stringContaining("Allowed Pi Town tools: pitown_board, pitown_delegate"),
+					content: expect.stringContaining("Recent inbox:\ninbox: empty"),
 				}),
 			}),
 		)
@@ -237,6 +237,46 @@ describe("pi town extension", () => {
 		)
 	})
 
+	it("shows queued mayor inbox updates as mayor UI notifications on the next turn", async () => {
+		const { artifactsDir, repoRoot, mayorSessionFile, workerSessionFile } = createRepoArtifacts()
+		const { handlers, tools } = setup()
+		const ctx = createToolContext(repoRoot, mayorSessionFile)
+
+		writeAgentState(
+			artifactsDir,
+			createAgentState({
+				agentId: "mayor",
+				role: "mayor",
+				status: "queued",
+				task: "coordinate town work",
+				lastMessage: "worker-001 completed task-123",
+				session: createAgentSessionRecord({
+					sessionDir: join(artifactsDir, "agents", "mayor", "sessions"),
+					sessionPath: mayorSessionFile,
+					sessionId: "mayor",
+				}),
+			}),
+		)
+
+		await tools.get("pitown_message_agent")?.execute(
+			"tool-call",
+			{ agentId: "mayor", body: "worker-001 completed task-123 (fix the failing auth flow): auth flow fixed" },
+			undefined,
+			() => {},
+			createToolContext(repoRoot, workerSessionFile),
+		)
+
+		await handlers["before_agent_start"]?.[1]?.(
+			{ systemPrompt: "base", prompt: "follow up", images: [] },
+			ctx,
+		)
+
+		expect(ctx.ui.notify).toHaveBeenCalledWith(
+			"worker-001 completed task-123 (fix the failing auth flow): auth flow fixed",
+			"info",
+		)
+	})
+
 	it("enables mayor plan mode with a read-only tool set", async () => {
 		const { repoRoot, mayorSessionFile } = createRepoArtifacts()
 		const { commands, activeTools, appendedEntries } = setup()

package/pi/extensions/town-tools.ts

@@ -121,6 +121,7 @@ function buildStartupContext(context: TownAgentContext): string {
 
 	const currentTask = state.task ?? "no active task"
 	const board = formatBoard(context.artifactsDir)
+	const inbox = formatMailbox(context.artifactsDir, context.agentId, "inbox")
 	const tools = getAllowedTools(context.role).join(", ")
 
 	return [
@@ -131,6 +132,9 @@ function buildStartupContext(context: TownAgentContext): string {
 		"",
 		"Current board:",
 		board,
+		"",
+		"Recent inbox:",
+		inbox,
 	].join("\n")
 }
 
@@ -142,7 +146,7 @@ function requireTownContext(ctx: ExtensionContext): TownAgentContext {
 	return context
 }
 
-const notifiedCompletions = new Set<string>()
+const notifiedMayorInboxMessages = new Set<string>()
 
 export function registerTownTools(pi: ExtensionAPI) {
 	pi.on("before_agent_start", async (_event, ctx) => {
@@ -165,23 +169,25 @@ export function registerTownTools(pi: ExtensionAPI) {
 
 		const agents = listAgentStates(context.artifactsDir)
 		const workers = agents.filter((a) => a.agentId !== "mayor")
+		const recentMayorMessages = readAgentMessages(context.artifactsDir, context.agentId, "inbox").slice(-5)
+
+		for (const record of recentMayorMessages) {
+			if (record.from === "human" || record.from === "system") continue
+
+			const key = `${record.createdAt}:${record.from}:${record.body}`
+			if (notifiedMayorInboxMessages.has(key)) continue
+
+			notifiedMayorInboxMessages.add(key)
+			const level = /\bblocked\b|\bfailed\b|\bstopped\b/i.test(record.body) ? "warning" : "info"
+			ctx.ui.notify(record.body, level)
+		}
+
 		if (workers.length === 0) {
 			ctx.ui.setStatus("pitown-workers", undefined)
 			ctx.ui.setWidget("pitown-workers", undefined)
 			return
 		}
 
-		for (const w of workers) {
-			if ((w.status === "idle" || w.status === "completed") && !notifiedCompletions.has(w.agentId)) {
-				notifiedCompletions.add(w.agentId)
-				ctx.ui.notify(`✓ ${w.agentId} finished: ${w.lastMessage ?? w.task ?? "done"}`, "info")
-			}
-			if ((w.status === "blocked" || w.status === "failed" || w.status === "stopped") && !notifiedCompletions.has(w.agentId)) {
-				notifiedCompletions.add(w.agentId)
-				ctx.ui.notify(`✗ ${w.agentId} blocked: ${w.lastMessage ?? "needs attention"}`, "warning")
-			}
-		}
-
 		const running = workers.filter((a) => a.status === "running" || a.status === "starting").length
 		const done = workers.filter((a) => a.status === "idle" || a.status === "completed").length
 		const blocked = workers.filter((a) => a.status === "blocked" || a.status === "failed" || a.status === "stopped").length
@@ -241,6 +247,15 @@ export function registerTownTools(pi: ExtensionAPI) {
 				agentId: input.agentId ?? null,
 				extensionPath: resolvePiTownExtensionPath(),
 				appendedSystemPrompt: input.role === "mayor" ? readPiTownMayorPrompt() : null,
+				completionAutoResumeTarget:
+					context.agentId === "mayor"
+						? {
+								agentId: "mayor",
+								message: "New agent check-ins arrived. Review the latest board and inbox updates, then decide the next bounded action.",
+								extensionPath: resolvePiTownExtensionPath(),
+								appendedSystemPrompt: readPiTownMayorPrompt(),
+							}
+						: null,
 			})
 
 			return {