pi-crew 0.2.10 → 0.2.12

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-crew",
-  "version": "0.2.10",
+  "version": "0.2.12",
   "description": "Pi extension for coordinated AI teams, workflows, worktrees, and async task orchestration",
   "author": "baphuongna",
   "license": "MIT",

package/src/extension/registration/commands.ts

@@ -387,12 +387,14 @@ export function registerTeamCommands(pi: ExtensionAPI, deps: RegisterTeamCommand
 
 	pi.registerCommand("team-dashboard", { description: "Open a pi-crew run dashboard overlay", handler: async (_args: string, ctx: ExtensionCommandContext) => {
 		for (;;) {
+			// Extract sessionId for workspace-scoped filtering
+			const sessionId = ctx.sessionManager?.getSessionId?.();
 			const runs = deps.getManifestCache(ctx.cwd).list(50);
 			const uiConfig = loadConfig(ctx.cwd).config.ui;
 			const rightPanel = (uiConfig?.dashboardPlacement ?? DEFAULT_UI.dashboardPlacement) === "right";
 			const width = rightPanel ? Math.min(90, Math.max(40, uiConfig?.dashboardWidth ?? DEFAULT_UI.dashboardWidth)) : "90%";
 			const { RunDashboard } = await ui();
-			const selection = await ctx.ui.custom<RunDashboardSelection | undefined>((tui, theme, _keybindings, done) => new RunDashboard(runs, done, theme, { placement: rightPanel ? "right" : "center", showModel: uiConfig?.showModel, showTokens: uiConfig?.showTokens, showTools: uiConfig?.showTools, snapshotCache: deps.getRunSnapshotCache?.(ctx.cwd), runProvider: () => deps.getManifestCache(ctx.cwd).list(50), registry: deps.getMetricRegistry?.(), requestRender: () => requestRenderTarget(tui) }), { overlay: true, overlayOptions: rightPanel ? { width, minWidth: 40, maxHeight: "100%", anchor: "top-right", offsetX: 0, offsetY: 0, margin: { top: 0, right: 0, bottom: 0, left: 0 } } : { width, maxHeight: "90%", anchor: "center", margin: 2 } });
+			const selection = await ctx.ui.custom<RunDashboardSelection | undefined>((tui, theme, _keybindings, done) => new RunDashboard(runs, done, theme, { placement: rightPanel ? "right" : "center", showModel: uiConfig?.showModel, showTokens: uiConfig?.showTokens, showTools: uiConfig?.showTools, snapshotCache: deps.getRunSnapshotCache?.(ctx.cwd), runProvider: () => deps.getManifestCache(ctx.cwd).list(50), registry: deps.getMetricRegistry?.(), workspaceId: sessionId, requestRender: () => requestRenderTarget(tui) }), { overlay: true, overlayOptions: rightPanel ? { width, minWidth: 40, maxHeight: "100%", anchor: "top-right", offsetX: 0, offsetY: 0, margin: { top: 0, right: 0, bottom: 0, left: 0 } } : { width, maxHeight: "90%", anchor: "center", margin: 2 } });
 			if (!selection) return;
 			if (selection.action === "reload") continue;
 			if (selection.action === "notifications-dismiss") {

package/src/extension/registration/subagent-tools.ts

@@ -3,6 +3,7 @@ import { Type } from "@sinclair/typebox";
 import type { TeamToolParamsValue } from "../../schema/team-tool-schema.ts";
 // Lazy-loaded: team-tool.ts pulls in entire runtime chain.
 import type { handleTeamTool as HandleTeamToolFn } from "../team-tool.ts";
+import { withSessionId } from "../team-tool/context.ts";
 let _cachedHandleTeamTool: typeof HandleTeamToolFn | undefined;
 async function handleTeamTool(params: Parameters<typeof HandleTeamToolFn>[0], ctx: Parameters<typeof HandleTeamToolFn>[1]): Promise<Awaited<ReturnType<typeof HandleTeamToolFn>>> {
 	if (!_cachedHandleTeamTool) {
@@ -63,7 +64,10 @@ export function registerSubagentTools(pi: ExtensionAPI, subagentManager: Subagen
 			const spawnOptions = __test__subagentSpawnParams(params as Record<string, unknown>, ctx);
 			spawnOptions.ownerSessionGeneration = options.ownerSessionGeneration?.();
 			if (!spawnOptions.prompt.trim()) return subagentToolResult(t("agent.requiresPrompt"), {}, true);
-			const runner = async (currentOptions: SubagentSpawnOptions, childSignal?: AbortSignal) => handleTeamTool({ action: "run", agent: currentOptions.type, goal: currentOptions.prompt, model: currentOptions.model, skill: currentOptions.skill, async: currentOptions.background, config: currentOptions.maxTurns ? { runtime: { maxTurns: currentOptions.maxTurns } } : undefined } as TeamToolParamsValue, { ...ctx, signal: childSignal, ...(options.startForegroundRun ? { startForegroundRun: (runRunner: (sig?: AbortSignal) => Promise<void>, runId?: string) => options.startForegroundRun!(ctx, runRunner, runId) } : {}) });
+			// Extract sessionId from sessionManager.getSessionId() so team runs created
+			// by the Agent tool have proper session ownership for isolation.
+			const ctxWithSession = withSessionId(ctx);
+			const runner = async (currentOptions: SubagentSpawnOptions, childSignal?: AbortSignal) => handleTeamTool({ action: "run", agent: currentOptions.type, goal: currentOptions.prompt, model: currentOptions.model, skill: currentOptions.skill, async: currentOptions.background, config: currentOptions.maxTurns ? { runtime: { maxTurns: currentOptions.maxTurns } } : undefined } as TeamToolParamsValue, { ...ctxWithSession, signal: childSignal, ...(options.startForegroundRun ? { startForegroundRun: (runRunner: (sig?: AbortSignal) => Promise<void>, runId?: string) => options.startForegroundRun!(ctxWithSession, runRunner, runId) } : {}) });
 			const record = subagentManager.spawn(spawnOptions, runner, spawnOptions.background ? undefined : signal);
 			if (spawnOptions.background || record.status === "queued") {
 				// Phase 1.1a: Terminate turn for background queued — no LLM follow-up needed.

package/src/ui/crew-widget.ts

@@ -175,7 +175,7 @@ function agentsFor(run: TeamRunManifest): CrewAgentRecord[] {
 let lastStaleReconcileAt = 0;
 const STALE_RECONCILE_INTERVAL_MS = 60_000;
 
-export function activeWidgetRuns(cwd: string, manifestCache?: ManifestCache, snapshotCache?: RunSnapshotCache, preloadedManifests?: TeamRunManifest[]): WidgetRun[] {
+export function activeWidgetRuns(cwd: string, manifestCache?: ManifestCache, snapshotCache?: RunSnapshotCache, preloadedManifests?: TeamRunManifest[], workspaceId?: string): WidgetRun[] {
 	// Evict stale live-agent handles (terminal status >10min, or running >30min with no update)
 	evictStaleLiveAgentHandles();
 	// Periodic stale reconciliation: detect ghost runs on disk with dead PIDs
@@ -185,7 +185,11 @@ export function activeWidgetRuns(cwd: string, manifestCache?: ManifestCache, sna
 		lastStaleReconcileAt = now;
 		try { reconcileAllStaleRuns(cwd, manifestCache); } catch { /* non-critical background maintenance */ }
 	}
-	const runs = preloadedManifests ?? (manifestCache ? manifestCache.list(20) : listRecentRuns(cwd, 20));
+	let runs = preloadedManifests ?? (manifestCache ? manifestCache.list(20) : listRecentRuns(cwd, 20));
+	// Filter by workspaceId for session isolation
+	if (workspaceId) {
+		runs = runs.filter((run) => !run.ownerSessionId || run.ownerSessionId === workspaceId);
+	}
 	return runs
 		.map((run) => {
 			try {
@@ -409,7 +413,7 @@ class CrewWidgetComponent implements WidgetComponent {
 }
 
 export function updateCrewWidget(
-	ctx: Pick<ExtensionContext, "cwd" | "hasUI" | "ui">,
+	ctx: Pick<ExtensionContext, "cwd" | "hasUI" | "ui" | "sessionManager">,
 	state: CrewWidgetState,
 	config?: CrewUiConfig,
 	manifestCache?: ManifestCache,
@@ -419,7 +423,8 @@ export function updateCrewWidget(
 	if (!ctx.hasUI) return;
 	state.frame += 1;
 	const maxLines = config?.widgetMaxLines ?? MAX_LINES_DEFAULT;
-	const runs = activeWidgetRuns(ctx.cwd, manifestCache, snapshotCache, preloadedManifests);
+	const workspaceId = ctx.sessionManager?.getSessionId?.();
+	const runs = activeWidgetRuns(ctx.cwd, manifestCache, snapshotCache, preloadedManifests, workspaceId);
 	const lines = buildCrewWidgetLines(ctx.cwd, state.frame, maxLines, runs, state.notificationCount ?? 0);
 	const placement = config?.widgetPlacement ?? DEFAULT_UI.widgetPlacement;
 	ctx.ui.setStatus(STATUS_KEY, lines.length ? statusSummary(runs) : undefined);

package/src/ui/dashboard-panes/agents-pane.ts

@@ -3,7 +3,7 @@ import { iconForStatus } from "../status-colors.ts";
 import type { RunUiSnapshot } from "../snapshot-types.ts";
 import { spinnerFrame } from "../spinner.ts";
 import type { CrewAgentRecord } from "../../runtime/crew-agent-runtime.ts";
-import { listLiveAgents, type LiveAgentHandle } from "../../runtime/live-agent-manager.ts";
+import { listLiveAgents, listLiveAgentsByWorkspace, type LiveAgentHandle } from "../../runtime/live-agent-manager.ts";
 
 /**
  * Returns true if this agent did real work (LLM call, tool use, or non-trivial duration).
@@ -52,7 +52,11 @@ function describeActivity(handle: LiveAgentHandle): string {
 export function renderAgentsPane(snapshot: RunUiSnapshot | undefined, options: RunDashboardOptions = {}): string[] {
 	if (!snapshot) return ["(snapshot unavailable)"];
 	if (!snapshot.agents.length) return ["(no agents)"];
-	const liveForRun = listLiveAgents().filter(h => h.runId === snapshot.runId);
+	// Filter live agents by workspaceId for session isolation
+	const allLive = options.workspaceId
+		? listLiveAgentsByWorkspace(options.workspaceId)
+		: listLiveAgents();
+	const liveForRun = allLive.filter(h => h.runId === snapshot.runId);
 	const { completed, total } = snapshot.progress;
 
 	const lines: string[] = [];

package/src/ui/run-dashboard.ts

@@ -41,6 +41,12 @@ export interface RunDashboardOptions {
 	snapshotCache?: RunSnapshotCache;
 	runProvider?: () => TeamRunManifest[];
 	registry?: MetricRegistry;
+	/**
+	 * Workspace/session ID for filtering runs and live agents. When provided,
+	 * only runs with matching ownerSessionId and live agents with matching
+	 * workspaceId are shown. This ensures session isolation in the UI.
+	 */
+	workspaceId?: string;
 	/**
 	 * Poke the host TUI to repaint after a state change. Must be wired from
 	 * `commands.ts` (`() => requestRenderTarget(tui)`) so keypresses and event-bus
@@ -279,7 +285,12 @@ export class RunDashboard implements DashboardComponent {
 		theme: unknown = {},
 		options: RunDashboardOptions = {},
 	) {
-		this.runs = runs;
+		// Filter runs by workspaceId for session isolation
+		// If workspaceId is provided, only show runs owned by that session or runs with no owner (legacy)
+		const filteredRuns = options.workspaceId
+			? runs.filter((run) => !run.ownerSessionId || run.ownerSessionId === options.workspaceId)
+			: runs;
+		this.runs = filteredRuns;
 		this.done = done;
 		this.theme = asCrewTheme(theme);
 		this.options = options;