pi-squad 0.6.1 → 0.6.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-squad",
-  "version": "0.6.1",
+  "version": "0.6.3",
   "description": "Multi-agent collaboration extension for pi — task decomposition, dependency management, parallel execution, TUI panel",
   "type": "module",
   "pi": {

package/src/index.ts

@@ -105,6 +105,8 @@ export default function (pi: ExtensionAPI) {
 		}
 
 		// When NO squad is active, nudge the agent to consider using squad for complex tasks
+		const allAgents = store.loadAllAgentDefs(ctx.cwd).filter((a) => a.name !== "planner" && !a.disabled);
+		const agentList = allAgents.map((a) => `${a.name} (${a.role})`).join(", ");
 		const squadNudge = [
 			`<squad_hint>`,
 			`You have the "squad" tool available for multi-agent collaboration.`,
@@ -112,8 +114,9 @@ export default function (pi: ExtensionAPI) {
 			`would benefit from parallel execution, or is too large for a single agent context.`,
 			`The squad tool decomposes work into tasks, assigns specialist agents, and runs them in parallel.`,
 			`When in doubt about whether a task is complex enough, prefer using squad — it handles the coordination for you.`,
+			allAgents.length > 0 ? `Available agents: ${agentList}. When providing tasks, the "agent" field must be one of these names.` : ``,
 			`</squad_hint>`,
-		].join("\n");
+		].filter(Boolean).join("\n");
 
 		return {
 			systemPrompt: event.systemPrompt + "\n\n" + squadNudge,
@@ -1176,6 +1179,16 @@ async function startSquad(
 				depends: t.depends || [],
 			})),
 		};
+
+		// Validate agent names — remap unknown agents to fullstack
+		for (const task of plan.tasks) {
+			const agentDef = store.loadAgentDef(task.agent, cwd);
+			if (!agentDef) {
+				const original = task.agent;
+				task.agent = "fullstack";
+				task.description = `[Note: agent "${original}" not found, remapped to fullstack]\n\n${task.description}`;
+			}
+		}
 	} else {
 		// Run planner to generate task breakdown
 		try {

package/src/planner.ts

@@ -65,7 +65,8 @@ export async function runPlanner(options: PlannerOptions): Promise<PlannerOutput
 			model: model || plannerDef?.model || undefined,
 		});
 
-		return parsePlannerOutput(output);
+		const agentNames = new Set(allAgents.map((a) => a.name));
+		return parsePlannerOutput(output, agentNames);
 	} finally {
 		try {
 			fs.unlinkSync(promptFile);
@@ -172,7 +173,7 @@ async function runPiJson(options: PiJsonOptions): Promise<string> {
 // Output Parsing
 // ============================================================================
 
-function parsePlannerOutput(text: string): PlannerOutput {
+function parsePlannerOutput(text: string, validAgents?: Set<string>): PlannerOutput {
 	// Try to extract JSON from the text (might be wrapped in markdown code blocks)
 	const jsonMatch = text.match(/```(?:json)?\s*\n?([\s\S]*?)\n?```/) || text.match(/(\{[\s\S]*\})/);
 
@@ -194,6 +195,13 @@ function parsePlannerOutput(text: string): PlannerOutput {
 			}
 			if (!task.depends) task.depends = [];
 			if (!task.description) task.description = "";
+
+			// Remap unknown agent names to fullstack (the generalist)
+			if (validAgents && !validAgents.has(task.agent)) {
+				const original = task.agent;
+				task.agent = "fullstack";
+				task.description = `[Note: planner assigned "${original}" which doesn't exist, remapped to fullstack]\n\n${task.description}`;
+			}
 		}
 
 		if (!parsed.agents) parsed.agents = {};