@tmustier/pi-agent-teams 0.5.1 → 0.5.3

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # Changelog
 
+## 0.5.3
+
+### Fixes
+
+- **Inherited parent-team attach warnings** — forked/branched child sessions now silently detach back to their own session team when they briefly inherit a parent team id without any attach claim. This removes the noisy warning seen in subagent-style flows while preserving normal `not_owner` warnings for real claim conflicts.
+
+## 0.5.2
+
+### Fixes
+
+- **Pi 0.62 metadata compatibility** — updated tool metadata wiring for recent Pi releases so teams tools continue to render the right prompt snippets/guidelines and stay compatible with current core APIs.
+- **Non-interactive exit hang** — leader polling timers now call `unref()` so print/json child sessions can exit cleanly instead of hanging after the agent finishes. This fixes subagent and other nested Pi flows that load the teams extension in the background.
+
 ## 0.5.1
 
 ### Features

package/extensions/teams/leader-teams-tool.ts

@@ -191,6 +191,11 @@ export function registerTeamsTool(opts: {
 			"Optional overrides: model='<provider>/<modelId>' and thinking (off|minimal|low|medium|high|xhigh).",
 			"For governance, the user can run /team delegate on (leader restricted to coordination) or /team spawn <name> plan (worker needs plan approval).",
 		].join(" "),
+		promptSnippet: "Delegate work across teammates, inspect member status, message workers, and manage team lifecycle/tasks.",
+		promptGuidelines: [
+			"Use this tool when the user wants parallel agent work, worker coordination, or team lifecycle/task management.",
+			"Prefer member_status before interrupting or reassigning active teammates when the current state is unclear.",
+		],
 		parameters: TeamsToolParamsSchema,
 
 		async execute(_toolCallId, params: TeamsToolParamsType, signal, _onUpdate, ctx): Promise<AgentToolResult<unknown>> {

package/extensions/teams/leader.ts

@@ -33,6 +33,7 @@ import {
 } from "./hooks.js";
 import { handleTeamCommand } from "./leader-team-command.js";
 import { registerTeamsTool } from "./leader-teams-tool.js";
+import { getParentSessionId, shouldSilenceInheritedParentAttachClaimWarning } from "./session-parent.js";
 import type { ContextMode, SpawnTeammateFn, SpawnTeammateResult, WorkspaceMode } from "./spawn-types.js";
 
 function getTeamsExtensionEntryPath(): string | null {
@@ -155,6 +156,7 @@ export function runLeader(pi: ExtensionAPI): void {
 	let delegateMode = process.env.PI_TEAMS_DELEGATE_MODE === "1";
 	let style: TeamsStyle = getTeamsStyleFromEnv();
 	let lastAttachClaimHeartbeatMs = 0;
+	let inheritedParentTeamId: string | null = null;
 
 	const stopLoops = () => {
 		if (refreshTimer) clearInterval(refreshTimer);
@@ -180,13 +182,22 @@ export function runLeader(pi: ExtensionAPI): void {
 		const result = await heartbeatTeamAttachClaim(getTeamDir(currentTeamId), sessionTeamId);
 		if (result === "updated") return;
 
-		ctx.ui.notify(
-			`Attach claim for team ${currentTeamId} is no longer owned by this session; detaching to session team.`,
-			"warning",
-		);
+		const lostTeamId = currentTeamId;
+		const shouldSilenceWarning = shouldSilenceInheritedParentAttachClaimWarning({
+			currentTeamId: lostTeamId,
+			parentSessionId: inheritedParentTeamId,
+			result,
+		});
+		inheritedParentTeamId = null;
 		currentTeamId = sessionTeamId;
 		taskListId = sessionTeamId;
 		delegationTracker.clear();
+		if (!shouldSilenceWarning) {
+			ctx.ui.notify(
+				`Attach claim for team ${lostTeamId} is no longer owned by this session; detaching to session team.`,
+				"warning",
+			);
+		}
 		await refreshTasks();
 		renderWidget();
 	};
@@ -709,6 +720,7 @@ export function runLeader(pi: ExtensionAPI): void {
 	pi.on("session_start", async (_event, ctx) => {
 		currentCtx = ctx;
 		currentTeamId = currentCtx.sessionManager.getSessionId();
+		inheritedParentTeamId = getParentSessionId(currentCtx.sessionManager);
 		// Keep the task list aligned with the active session. If you want a shared namespace,
 		// use `/team task use <taskListId>` after switching.
 		taskListId = currentTeamId;
@@ -750,6 +762,8 @@ export function runLeader(pi: ExtensionAPI): void {
 				refreshInFlight = false;
 			}
 		}, 1000);
+		// Don't keep non-interactive/child pi processes alive just because leader polling exists.
+		refreshTimer.unref?.();
 
 		inboxTimer = setInterval(async () => {
 			if (isStopping) return;
@@ -761,6 +775,7 @@ export function runLeader(pi: ExtensionAPI): void {
 				inboxInFlight = false;
 			}
 		}, 700);
+		inboxTimer.unref?.();
 	});
 
 	pi.on("session_switch", async (_event, ctx) => {
@@ -789,6 +804,7 @@ export function runLeader(pi: ExtensionAPI): void {
 
 		currentCtx = ctx;
 		currentTeamId = currentCtx.sessionManager.getSessionId();
+		inheritedParentTeamId = getParentSessionId(currentCtx.sessionManager);
 		// Keep the task list aligned with the active session. If you want a shared namespace,
 		// use `/team task use <taskListId>` after switching.
 		taskListId = currentTeamId;
@@ -820,6 +836,7 @@ export function runLeader(pi: ExtensionAPI): void {
 				refreshInFlight = false;
 			}
 		}, 1000);
+		refreshTimer.unref?.();
 
 		inboxTimer = setInterval(async () => {
 			if (isStopping) return;
@@ -831,6 +848,7 @@ export function runLeader(pi: ExtensionAPI): void {
 				inboxInFlight = false;
 			}
 		}, 700);
+		inboxTimer.unref?.();
 	});
 
 	pi.on("session_shutdown", async () => {
@@ -1067,6 +1085,7 @@ export function runLeader(pi: ExtensionAPI): void {
 				getActiveTeamId: () => currentTeamId ?? ctx.sessionManager.getSessionId(),
 				setActiveTeamId: (teamId) => {
 					currentTeamId = teamId;
+					inheritedParentTeamId = null;
 					delegationTracker.clear();
 				},
 				pendingPlanApprovals,

package/extensions/teams/session-parent.ts

@@ -0,0 +1,24 @@
+import { SessionManager } from "@mariozechner/pi-coding-agent";
+import type { TeamAttachClaimHeartbeatResult } from "./team-attach-claim.js";
+
+interface SessionManagerWithHeader {
+	getHeader(): { parentSession?: string } | null;
+}
+
+export function getParentSessionId(sessionManager: SessionManagerWithHeader): string | null {
+	const parentSessionPath = sessionManager.getHeader()?.parentSession;
+	if (!parentSessionPath) return null;
+	try {
+		return SessionManager.open(parentSessionPath).getSessionId();
+	} catch {
+		return null;
+	}
+}
+
+export function shouldSilenceInheritedParentAttachClaimWarning(opts: {
+	currentTeamId: string;
+	parentSessionId: string | null;
+	result: TeamAttachClaimHeartbeatResult;
+}): boolean {
+	return opts.result === "missing" && opts.parentSessionId !== null && opts.currentTeamId === opts.parentSessionId;
+}

package/extensions/teams/worker.ts

@@ -143,6 +143,11 @@ export function runWorker(pi: ExtensionAPI): void {
 		name: "team_message",
 		label: "Team Message",
 		description: "Send a message to a comrade. Use this to coordinate with peers on related tasks. Set urgent=true to interrupt their active turn (use sparingly — only for time-sensitive coordination).",
+		promptSnippet: "Send a coordination message to another teammate, optionally as an urgent interruption.",
+		promptGuidelines: [
+			"Use this tool for teammate-to-teammate coordination instead of overloading task status fields with freeform messages.",
+			"Set urgent=true only when the recipient must be interrupted before finishing their current turn.",
+		],
 		parameters: TeamMessageToolParamsSchema,
 		async execute(
 			_toolCallId,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tmustier/pi-agent-teams",
-  "version": "0.5.1",
+  "version": "0.5.3",
   "description": "Claude Code agent teams style workflow for Pi.",
   "license": "MIT",
   "author": "Thomas Mustier",
@@ -39,10 +39,10 @@
   },
   "devDependencies": {
     "@eslint/js": "^9.39.2",
-    "@mariozechner/pi-agent-core": "^0.52.8",
-    "@mariozechner/pi-ai": "^0.52.8",
-    "@mariozechner/pi-coding-agent": "^0.52.8",
-    "@mariozechner/pi-tui": "^0.52.8",
+    "@mariozechner/pi-agent-core": "^0.62.0",
+    "@mariozechner/pi-ai": "^0.62.0",
+    "@mariozechner/pi-coding-agent": "^0.62.0",
+    "@mariozechner/pi-tui": "^0.62.0",
     "@sinclair/typebox": "^0.34.48",
     "eslint": "^9.39.2",
     "tsx": "^4.20.5",

package/scripts/smoke-test.mts

@@ -70,7 +70,9 @@ import {
 	isPlanRejectedMessage,
 } from "../extensions/teams/protocol.js";
 import { pollLeaderInbox } from "../extensions/teams/leader-inbox.js";
-import type { ExtensionContext } from "@mariozechner/pi-coding-agent";
+import { getParentSessionId, shouldSilenceInheritedParentAttachClaimWarning } from "../extensions/teams/session-parent.js";
+import { SessionManager, type ExtensionContext } from "@mariozechner/pi-coding-agent";
+import type { AssistantMessage } from "@mariozechner/pi-ai";
 
 // ── helpers ──────────────────────────────────────────────────────────
 let passed = 0;
@@ -554,14 +556,18 @@ console.log("\n7. Pi extension loading");
 		return typeof c === "string" ? c : undefined;
 	})();
 
+	const versionOutput = `${res.stdout ?? ""}${res.stderr ?? ""}`.trim();
+
 	if (errCode === "ENOENT") {
 		console.log("  (skipped) pi CLI not found on PATH");
 	} else if (errCode === "ETIMEDOUT") {
 		console.log("  (skipped) pi --version timed out");
 	} else if (res.status !== 0) {
 		console.log("  (skipped) pi --version returned non-zero exit code");
+	} else if (versionOutput.length === 0) {
+		console.log("  (skipped) pi --version produced no output");
 	} else {
-		assert((res.stdout ?? "").trim().length > 0, "pi --version works");
+		assert(versionOutput.length > 0, "pi --version works");
 	}
 }
 
@@ -841,6 +847,67 @@ console.log("\n10. team discovery + attach claims");
 	}
 }
 
+// ── 10b. branched sessions + inherited attach claims ────────────────
+console.log("\n10b. branched sessions + inherited attach claims");
+{
+	const sessionsDir = path.join(tmpRoot, "branch-session-test");
+	const parent = SessionManager.create(tmpRoot, sessionsDir);
+	const assistantMessage: AssistantMessage = {
+		role: "assistant",
+		content: [{ type: "text", text: "ok" }],
+		api: "test",
+		provider: "test",
+		model: "test",
+		usage: {
+			input: 0,
+			output: 0,
+			cacheRead: 0,
+			cacheWrite: 0,
+			totalTokens: 0,
+			cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+		},
+		stopReason: "stop",
+		timestamp: Date.now(),
+	};
+	parent.appendMessage(assistantMessage);
+	const parentLeafId = parent.getLeafId();
+	const originalParentSessionId = parent.getSessionId();
+	assert(parentLeafId !== null, "branch test parent has a leaf entry");
+	if (parentLeafId) {
+		const branchedPath = parent.createBranchedSession(parentLeafId);
+		assert(branchedPath !== null, "createBranchedSession returns child session path");
+		if (branchedPath) {
+			const child = SessionManager.open(branchedPath, sessionsDir);
+			const parentSessionId = getParentSessionId(child);
+			assertEq(parentSessionId, originalParentSessionId, "getParentSessionId resolves branch parent session id");
+			assert(
+				shouldSilenceInheritedParentAttachClaimWarning({
+					currentTeamId: originalParentSessionId,
+					parentSessionId,
+					result: "missing",
+				}),
+				"silences missing inherited parent-team claim warnings",
+			);
+			assert(
+				!shouldSilenceInheritedParentAttachClaimWarning({
+					currentTeamId: originalParentSessionId,
+					parentSessionId,
+					result: "not_owner",
+				}),
+				"keeps non-owner inherited parent-team warnings visible",
+			);
+			assert(
+				!shouldSilenceInheritedParentAttachClaimWarning({
+					currentTeamId: child.getSessionId(),
+					parentSessionId,
+					result: "missing",
+				}),
+				"does not silence unrelated missing-claim warnings",
+			);
+		}
+	}
+}
+
 // ── 11. /team done (end-of-run cleanup) ──────────────────────────────
 console.log("\n11. /team done (end-of-run)");
 {