pi-crew 0.1.24 → 0.1.26

package/src/extension/team-tool/context.ts

@@ -0,0 +1,48 @@
+import type { ExtensionContext } from "@mariozechner/pi-coding-agent";
+import type { TeamToolDetails } from "../team-tool-types.ts";
+import { toolResult, type PiTeamsToolResult } from "../tool-result.ts";
+
+export type TeamContext = Pick<ExtensionContext, "cwd"> & Partial<Pick<ExtensionContext, "model">> & {
+	modelRegistry?: unknown;
+	sessionManager?: { getBranch?: () => unknown[] };
+	events?: { emit?: (event: string, data: unknown) => void };
+	signal?: AbortSignal;
+	startForegroundRun?: (runner: (signal?: AbortSignal) => Promise<void>, runId?: string) => void;
+	onRunStarted?: (runId: string) => void;
+};
+
+export function result(text: string, details: TeamToolDetails, isError = false): PiTeamsToolResult {
+	return toolResult(text, details, isError);
+}
+
+export function formatScoped(name: string, source: string, description: string): string {
+	return `- ${name} (${source}): ${description}`;
+}
+
+function extractTextContent(content: unknown): string {
+	if (typeof content === "string") return content;
+	if (!Array.isArray(content)) return "";
+	return content.map((part) => part && typeof part === "object" && !Array.isArray(part) && typeof (part as { text?: unknown }).text === "string" ? (part as { text: string }).text : "").filter(Boolean).join("\n");
+}
+
+export function buildParentContext(ctx: TeamContext): string | undefined {
+	const branch = ctx.sessionManager?.getBranch?.();
+	if (!Array.isArray(branch) || branch.length === 0) return undefined;
+	const parts: string[] = [];
+	for (const entry of branch.slice(-20)) {
+		if (!entry || typeof entry !== "object" || Array.isArray(entry)) continue;
+		const record = entry as { type?: unknown; message?: unknown; summary?: unknown };
+		if (record.type === "compaction" && typeof record.summary === "string") parts.push(`[Summary]: ${record.summary}`);
+		const message = record.message && typeof record.message === "object" && !Array.isArray(record.message) ? record.message as { role?: unknown; content?: unknown } : undefined;
+		if (!message || (message.role !== "user" && message.role !== "assistant")) continue;
+		const text = extractTextContent(message.content).trim();
+		if (text) parts.push(`[${message.role === "user" ? "User" : "Assistant"}]: ${text}`);
+	}
+	if (!parts.length) return undefined;
+	return [`# Parent Conversation Context`, "The following context was inherited from the parent Pi session. Treat it as reference-only.", "", parts.join("\n\n")].join("\n");
+}
+
+export function configRecord(config: unknown): Record<string, unknown> {
+	if (!config || typeof config !== "object" || Array.isArray(config)) return {};
+	return config as Record<string, unknown>;
+}

package/src/extension/team-tool/doctor.ts

@@ -0,0 +1,178 @@
+import { execFileSync, spawnSync } from "node:child_process";
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { allAgents, discoverAgents } from "../../agents/discover-agents.ts";
+import { allTeams, discoverTeams } from "../../teams/discover-teams.ts";
+import { allWorkflows, discoverWorkflows } from "../../workflows/discover-workflows.ts";
+import { loadConfig } from "../../config/config.ts";
+import { projectPiRoot, userPiRoot } from "../../utils/paths.ts";
+import type { TeamToolParamsValue } from "../../schema/team-tool-schema.ts";
+import { getPiSpawnCommand } from "../../runtime/pi-spawn.ts";
+import { validateResources } from "../validate-resources.ts";
+import type { PiTeamsToolResult } from "../tool-result.ts";
+import { configRecord, result, type TeamContext } from "./context.ts";
+
+interface DoctorCheck {
+	label: string;
+	ok: boolean;
+	detail: string;
+}
+
+function firstOutputLine(stdout: string | null | undefined, stderr: string | null | undefined): string {
+	const output = `${stdout ?? ""}\n${stderr ?? ""}`.trim();
+	return output.split(/\r?\n/).find((line) => line.trim().length > 0)?.trim() ?? "available";
+}
+
+function commandExists(command: string, args: string[]): { ok: boolean; detail: string } {
+	try {
+		const output = spawnSync(command, args, { encoding: "utf-8", stdio: ["ignore", "pipe", "pipe"] });
+		if (output.error) {
+			return { ok: false, detail: output.error.message };
+		}
+		if (output.status !== 0) {
+			return { ok: false, detail: firstOutputLine(output.stdout, output.stderr) || `status ${output.status}` };
+		}
+		return { ok: true, detail: firstOutputLine(output.stdout, output.stderr) };
+	} catch (error) {
+		return { ok: false, detail: error instanceof Error ? error.message : String(error) };
+	}
+}
+
+function piCommandExists(): { ok: boolean; detail: string } {
+	const spec = getPiSpawnCommand(["--version"]);
+	const output = commandExists(spec.command, spec.args);
+	if (!output.ok) return output;
+	const executable = spec.command === "pi" ? "pi" : `${spec.command} ${spec.args[0] ?? ""}`.trim();
+	return { ok: true, detail: `${output.detail} (${executable})` };
+}
+
+function checkWritableDir(dir: string): { ok: boolean; detail: string } {
+	try {
+		if (!fs.existsSync(dir)) return { ok: false, detail: `${dir}: missing` };
+		if (!fs.statSync(dir).isDirectory()) return { ok: false, detail: `${dir}: not a directory` };
+		fs.accessSync(dir, fs.constants.W_OK);
+		return { ok: true, detail: dir };
+	} catch (error) {
+		const message = error instanceof Error ? error.message : String(error);
+		return { ok: false, detail: `${dir}: ${message}` };
+	}
+}
+
+function makeLine(check: DoctorCheck): string {
+	return `- ${check.ok ? "OK" : "FAIL"} ${check.label}: ${check.detail}`;
+}
+
+function section(title: string, checks: () => DoctorCheck[]): string[] {
+	try {
+		return [title, ...checks().map(makeLine)];
+	} catch (error) {
+		const detail = error instanceof Error ? error.message : String(error);
+		return [title, `- FAIL ${title}: ${detail}`];
+	}
+}
+
+export interface TeamDoctorReportInput {
+	cwd: string;
+	configPath: string;
+	configErrors: string[];
+	configWarnings: string[];
+	model?: { provider: string; id: string };
+	validationErrors: number;
+	validationWarnings: number;
+	smokeChildPi?: { ok: boolean; detail: string };
+}
+
+export interface TeamDoctorReport {
+	text: string;
+	hasErrors: boolean;
+}
+
+export function buildTeamDoctorReport(input: TeamDoctorReportInput): TeamDoctorReport {
+	const sections = [
+		section("Runtime", () => {
+			const git = commandExists("git", ["--version"]);
+			const pi = piCommandExists();
+			return [
+				{ label: "cwd", ok: true, detail: input.cwd },
+				{ label: "platform", ok: true, detail: `${process.platform}/${process.arch} node=${process.version}` },
+				{ label: "pi command", ok: pi.ok, detail: pi.detail },
+				{ label: "git command", ok: git.ok, detail: git.detail },
+				{ label: "config", ok: input.configErrors.length === 0, detail: `${input.configPath} (${input.configErrors.length} errors)` },
+				{ label: "model", ok: true, detail: input.model ? `${input.model.provider}/${input.model.id}` : "not available in this context" },
+				{ label: "config warnings", ok: true, detail: `${input.configWarnings.length} warnings` },
+			];
+		}),
+		section("Filesystem", () => {
+			const userWritable = checkWritableDir(path.join(userPiRoot(), "extensions", "pi-crew"));
+			const projectWritable = checkWritableDir(path.join(projectPiRoot(input.cwd), "teams"));
+			return [
+				{ label: "user state", ok: userWritable.ok, detail: userWritable.detail },
+				{ label: "project state", ok: projectWritable.ok, detail: projectWritable.detail },
+				{ label: "project state root", ok: true, detail: path.join(projectPiRoot(input.cwd), "teams") },
+				{ label: "artifacts root", ok: true, detail: path.join(projectPiRoot(input.cwd), "artifacts") },
+			];
+		}),
+		section("Discovery", () => {
+			const discoveredAgents = allAgents(discoverAgents(input.cwd));
+			const discoveredTeams = allTeams(discoverTeams(input.cwd));
+			const discoveredWorkflows = allWorkflows(discoverWorkflows(input.cwd));
+			const agentModelHints = discoveredAgents.filter((agent) => agent.model || agent.fallbackModels?.length).length;
+			return [
+				{ label: "agents", ok: true, detail: `${discoveredAgents.length} discovered` },
+				{ label: "teams", ok: true, detail: `${discoveredTeams.length} discovered` },
+				{ label: "workflows", ok: true, detail: `${discoveredWorkflows.length} discovered` },
+				{ label: "resource model hints", ok: true, detail: `${agentModelHints} agents declare model/fallback preferences` },
+			];
+		}),
+		section("Resource validation", () => [{
+			label: "resource validation",
+			ok: input.validationErrors === 0,
+			detail: `${input.validationErrors} errors, ${input.validationWarnings} warnings`,
+		}]),
+	];
+	if (input.smokeChildPi) {
+		sections.push([`Child check`, `- ${input.smokeChildPi.ok ? "OK" : "FAIL"} child Pi smoke: ${input.smokeChildPi.detail}`]);
+	}
+	const lines = ["pi-crew doctor report"];
+	for (const block of sections) {
+		if (block.length > 0) {
+			lines.push(...block);
+			lines.push("");
+		}
+	}
+	if (lines.at(-1) === "") lines.pop();
+	const text = lines.join("\n");
+	return { text, hasErrors: sections.some((sectionLines) => sectionLines.some((line) => line.includes("FAIL"))) };
+}
+
+export function handleDoctor(ctx: TeamContext, params: TeamToolParamsValue = {}): PiTeamsToolResult {
+	const loadedConfig = loadConfig(ctx.cwd);
+	let smokeChildPi: { ok: boolean; detail: string } | undefined;
+	if (configRecord(params.config).smokeChildPi === true) {
+		try {
+			const spec = getPiSpawnCommand(["--mode", "json", "-p", "Reply with exactly PI-TEAMS-SMOKE-OK"]);
+			const output = execFileSync(spec.command, spec.args, {
+				cwd: ctx.cwd,
+				encoding: "utf-8",
+				stdio: ["ignore", "pipe", "pipe"],
+				timeout: 15_000,
+			}).trim();
+			smokeChildPi = { ok: output.includes("PI-TEAMS-SMOKE-OK"), detail: output.split("\n").slice(-1)[0] ?? "completed" };
+		} catch (error) {
+			const message = error instanceof Error ? error.message : String(error);
+			smokeChildPi = { ok: false, detail: message };
+		}
+	}
+	const validation = validateResources(ctx.cwd);
+	const { text, hasErrors } = buildTeamDoctorReport({
+		cwd: ctx.cwd,
+		configPath: loadedConfig.path,
+		configErrors: loadedConfig.error ? [loadedConfig.error] : [],
+		configWarnings: loadedConfig.warnings ?? [],
+		model: ctx.model,
+		validationErrors: validation.issues.filter((issue) => issue.level === "error").length,
+		validationWarnings: validation.issues.filter((issue) => issue.level === "warning").length,
+		smokeChildPi,
+	});
+	return result(text, { action: "doctor", status: hasErrors ? "error" : "ok" }, hasErrors);
+}

package/src/extension/team-tool/run.ts

@@ -0,0 +1,133 @@
+import { allAgents, discoverAgents } from "../../agents/discover-agents.ts";
+import { allTeams, discoverTeams } from "../../teams/discover-teams.ts";
+import { allWorkflows, discoverWorkflows } from "../../workflows/discover-workflows.ts";
+import { loadConfig } from "../../config/config.ts";
+import type { TeamToolParamsValue } from "../../schema/team-tool-schema.ts";
+import { writeArtifact } from "../../state/artifact-store.ts";
+import { createRunManifest } from "../../state/state-store.ts";
+import { atomicWriteJson } from "../../state/atomic-write.ts";
+import { validateWorkflowForTeam } from "../../workflows/validate-workflow.ts";
+import { executeTeamRun } from "../../runtime/team-runner.ts";
+import { spawnBackgroundTeamRun } from "../../runtime/async-runner.ts";
+import { appendEvent } from "../../state/event-log.ts";
+import { resolveCrewRuntime } from "../../runtime/runtime-resolver.ts";
+import { expandParallelResearchWorkflow } from "../../runtime/parallel-research.ts";
+import type { PiTeamsToolResult } from "../tool-result.ts";
+import { buildParentContext, result, type TeamContext } from "./context.ts";
+import { effectiveRunConfig } from "./config-patch.ts";
+
+export async function handleRun(params: TeamToolParamsValue, ctx: TeamContext): Promise<PiTeamsToolResult> {
+	const goal = params.goal ?? params.task;
+	if (!goal) return result("Run requires goal or task.", { action: "run", status: "error" }, true);
+
+	const teams = allTeams(discoverTeams(ctx.cwd));
+	const workflows = allWorkflows(discoverWorkflows(ctx.cwd));
+	const agents = allAgents(discoverAgents(ctx.cwd));
+	const directAgent = params.agent ? agents.find((item) => item.name === params.agent) : undefined;
+	if (params.agent && !directAgent) return result(`Agent '${params.agent}' not found.`, { action: "run", status: "error" }, true);
+	const teamName = params.team ?? "default";
+	const team = directAgent ? {
+		name: `direct-${directAgent.name}`,
+		description: `Direct subagent run for ${directAgent.name}`,
+		source: "builtin" as const,
+		filePath: "<generated>",
+		roles: [{ name: params.role ?? "agent", agent: directAgent.name, description: directAgent.description }],
+		defaultWorkflow: "direct-agent",
+		workspaceMode: params.workspaceMode,
+	} : teams.find((item) => item.name === teamName);
+	if (!team) return result(`Team '${teamName}' not found.`, { action: "run", status: "error" }, true);
+	const workflowName = directAgent ? "direct-agent" : params.workflow ?? team.defaultWorkflow ?? "default";
+	const baseWorkflow = directAgent ? {
+		name: "direct-agent",
+		description: `Direct task for ${directAgent.name}`,
+		source: "builtin" as const,
+		filePath: "<generated>",
+		steps: [{ id: "01_agent", role: params.role ?? "agent", task: "{goal}", model: params.model }],
+	} : workflows.find((item) => item.name === workflowName);
+	if (!baseWorkflow) return result(`Workflow '${workflowName}' not found.`, { action: "run", status: "error" }, true);
+	const workflow = directAgent ? baseWorkflow : expandParallelResearchWorkflow(baseWorkflow, ctx.cwd);
+
+	const validationErrors = validateWorkflowForTeam(workflow, team);
+	if (validationErrors.length > 0) {
+		return result([`Workflow '${workflow.name}' is not valid for team '${team.name}':`, ...validationErrors.map((error) => `- ${error}`)].join("\n"), { action: "run", status: "error" }, true);
+	}
+
+	const { manifest, tasks, paths } = createRunManifest({
+		cwd: ctx.cwd,
+		team,
+		workflow,
+		goal,
+		workspaceMode: params.workspaceMode,
+	});
+	const goalArtifact = writeArtifact(paths.artifactsRoot, {
+		kind: "prompt",
+		relativePath: "goal.md",
+		content: `${goal}\n`,
+		producer: "team-tool",
+	});
+	const updatedManifest = { ...manifest, artifacts: [goalArtifact], summary: "Run manifest created; worker execution is not implemented yet." };
+	atomicWriteJson(paths.manifestPath, updatedManifest);
+
+	const loadedConfig = loadConfig(ctx.cwd);
+	const runAsync = params.async ?? loadedConfig.config.asyncByDefault ?? false;
+	if (runAsync) {
+		const spawned = spawnBackgroundTeamRun(updatedManifest);
+		const asyncManifest = { ...updatedManifest, async: { pid: spawned.pid, logPath: spawned.logPath, spawnedAt: new Date().toISOString() } };
+		atomicWriteJson(paths.manifestPath, asyncManifest);
+		appendEvent(updatedManifest.eventsPath, { type: "async.spawned", runId: updatedManifest.runId, data: { pid: spawned.pid, logPath: spawned.logPath } });
+		const text = [
+			`Started async pi-crew run ${updatedManifest.runId}.`,
+			`Team: ${team.name}`,
+			`Workflow: ${workflow.name}`,
+			`Status: ${updatedManifest.status}`,
+			`Tasks: ${tasks.length}`,
+			`State: ${updatedManifest.stateRoot}`,
+			`Artifacts: ${updatedManifest.artifactsRoot}`,
+			`Background log: ${spawned.logPath}`,
+			"",
+			`Check status with: team status runId=${updatedManifest.runId}`,
+		].join("\n");
+		return result(text, { action: "run", status: "ok", runId: updatedManifest.runId, artifactsRoot: updatedManifest.artifactsRoot });
+	}
+
+	const runtime = await resolveCrewRuntime(effectiveRunConfig(loadedConfig.config, params.config));
+	const executeWorkers = runtime.kind !== "scaffold";
+	const executedConfig = effectiveRunConfig(loadedConfig.config, params.config);
+	if (executeWorkers && ctx.startForegroundRun) {
+		ctx.onRunStarted?.(updatedManifest.runId);
+		ctx.startForegroundRun(async (signal) => {
+			await executeTeamRun({ manifest: updatedManifest, tasks, team, workflow, agents, executeWorkers, limits: executedConfig.limits, runtime, runtimeConfig: executedConfig.runtime, parentContext: buildParentContext(ctx), parentModel: ctx.model, modelRegistry: ctx.modelRegistry, modelOverride: params.model, signal });
+		}, updatedManifest.runId);
+		const text = [
+			`Started foreground pi-crew run ${updatedManifest.runId}.`,
+			`Team: ${team.name}`,
+			`Workflow: ${workflow.name}`,
+			"Status: running",
+			`Tasks: ${tasks.length}`,
+			`Runtime: ${runtime.kind}`,
+			`State: ${updatedManifest.stateRoot}`,
+			`Artifacts: ${updatedManifest.artifactsRoot}`,
+			"",
+			"The run continues in this Pi session without blocking the chat. It will be interrupted on session shutdown. Use /team-dashboard or /team-status to watch it.",
+		].join("\n");
+		return result(text, { action: "run", status: "ok", runId: updatedManifest.runId, artifactsRoot: updatedManifest.artifactsRoot });
+	}
+	const executed = await executeTeamRun({ manifest: updatedManifest, tasks, team, workflow, agents, executeWorkers, limits: executedConfig.limits, runtime, runtimeConfig: executedConfig.runtime, parentContext: buildParentContext(ctx), parentModel: ctx.model, modelRegistry: ctx.modelRegistry, modelOverride: params.model, signal: ctx.signal });
+	const text = [
+		`Created pi-crew run ${executed.manifest.runId}.`,
+		`Team: ${team.name}`,
+		`Workflow: ${workflow.name}`,
+		`Status: ${executed.manifest.status}`,
+		`Tasks: ${executed.tasks.length}`,
+		`State: ${executed.manifest.stateRoot}`,
+		`Artifacts: ${executed.manifest.artifactsRoot}`,
+		"",
+		`Runtime: ${runtime.kind}${runtime.fallback ? ` (fallback from ${runtime.requestedMode})` : ""}${runtime.reason ? ` - ${runtime.reason}` : ""}`,
+		runtime.kind === "child-process"
+			? "Child Pi worker execution is enabled by default; each task is launched as a separate Pi process. Set runtime.mode=scaffold or executeWorkers=false only for dry runs."
+			: runtime.kind === "live-session"
+				? "Experimental live-session worker execution was enabled."
+				: "Safe scaffold mode: child Pi workers were not launched because runtime.mode=scaffold or executeWorkers=false was configured.",
+	].join("\n");
+	return result(text, { action: "run", status: executed.manifest.status === "failed" ? "error" : "ok", runId: executed.manifest.runId, artifactsRoot: executed.manifest.artifactsRoot }, executed.manifest.status === "failed");
+}

package/src/extension/team-tool-types.ts

@@ -0,0 +1,6 @@
+export interface TeamToolDetails {
+	action: string;
+	status: "ok" | "error" | "planned";
+	runId?: string;
+	artifactsRoot?: string;
+}