@henryqw/pi-subagent 2.0.2 → 2.1.0

package/README.md

@@ -1,6 +1,6 @@
 # `@henryqw/pi-subagent`
 
-Delegate one bounded task to one isolated Pi process. Main chooses the role and model class per call.
+Delegate one bounded task to one isolated Pi process. Main chooses the role and may override shared task-model effort per call.
 
 ## Install
 
@@ -23,9 +23,9 @@ pi install npm:@henryqw/pi-subagent
 | --- | --- | --- |
 | `delegate_task` | tool | Start one isolated child for `role`, `task`, and optional `modelClass`. |
 
-`modelClass` is `fast`, `balanced`, or `frontier`. Omitted class uses `balanced`. Primary route is resolved against current scoped text models; fallback is tried only before launch. If no route is usable, delegation rejects with `Run /task-models`. A started child is never retried.
+`modelClass` is `fast`, `balanced`, or `frontier`. Omitted class uses the shared `pi-subagent/delegateTask` assignment, which defaults to `balanced`. Primary route is resolved against current scoped text models; fallback is tried only before launch. If no route is usable, delegation rejects with `Run /task-models`. A started child is never retried.
 
-Each call starts one isolated child (`pi --mode json -p --no-session`). Ambient extensions and skills are off; only role resources load. Child uses the delegated working directory and Main's project approval. Abort kills the child process group. Streaming output is capped at 50 KiB.
+Each call starts one isolated child (`pi --mode json -p --no-session`). Ambient extensions and skills are off; only role resources load. Child uses the delegated working directory and Main's project approval. Abort kills the child process group. Streaming output is capped at 50 KiB. Unused JSON event types are discarded before payload buffering; consumed or unclassifiable events above 1 MiB fail delegation.
 
 TUI shows one row per Subagent with role, route, task, tokens, and elapsed time. Terminal rows drop after one second.
 

package/extensions/subagent.ts

@@ -8,6 +8,7 @@ import { StringEnum } from "@earendil-works/pi-ai";
 import { type ExtensionAPI, type ExtensionContext, getAgentDir, parseFrontmatter, type Theme } from "@earendil-works/pi-coding-agent";
 import { type Component, truncateToWidth, type TUI, visibleWidth } from "@earendil-works/pi-tui";
 import {
+	DEFAULT_TASK_ASSIGNMENTS,
 	modelReference,
 	orderedProfileRoutes,
 	PROFILE_NAMES,
@@ -19,10 +20,14 @@ import {
 import { Type } from "typebox";
 
 const MODEL_CLASSES = PROFILE_NAMES;
+const SUBAGENT_TASK = "pi-subagent/delegateTask";
+const DEFAULT_MODEL_CLASS = DEFAULT_TASK_ASSIGNMENTS[SUBAGENT_TASK];
 const CODEX_ALIAS = /^openai-codex-(?:[2-9]|[1-9]\d+)$/;
 const MULTI_CODEX_EXTENSION = fileURLToPath(import.meta.resolve("@henryqw/pi-multi-codex/extensions/multi-codex.ts"));
 const MAX_OUTPUT_BYTES = 50 * 1024;
 const MAX_JSON_EVENT_BYTES = 1024 * 1024;
+const CONSUMED_JSON_EVENTS = new Set(["message_start", "message_update", "message_end"]);
+const JSON_EVENT_TYPE = /^\s*\{\s*"type"\s*:\s*"([^"\\]+)"/;
 const WIDGET_KEY = "subagent-status";
 const WIDGET_INTERVAL_MS = 80;
 const TERMINAL_DISPLAY_MS = 1_000;
@@ -60,7 +65,7 @@ const isModelClass = (value: unknown): value is ModelClass =>
 	typeof value === "string" && MODEL_CLASSES.includes(value as ModelClass);
 const isNumberedCodexProvider = (provider: string): boolean => CODEX_ALIAS.test(provider);
 
-function resolveTaskRoute(ctx: ExtensionContext, modelClass: ModelClass): ResolvedTaskRoute {
+function resolveTaskRoute(ctx: ExtensionContext, modelClass?: ModelClass): ResolvedTaskRoute {
 	let config;
 	try {
 		config = readTaskModelsConfig();
@@ -68,13 +73,14 @@ function resolveTaskRoute(ctx: ExtensionContext, modelClass: ModelClass): Resolv
 		throw new Error("Couldn't read task model config. Run /task-models.");
 	}
 
-	const profile = config.profiles[modelClass];
-	if (!profile) throw new Error(`No ${modelClass} task model profile is configured. Run /task-models.`);
+	const profileName = modelClass ?? config.tasks[SUBAGENT_TASK] ?? DEFAULT_MODEL_CLASS;
+	const profile = config.profiles[profileName];
+	if (!profile) throw new Error(`No ${profileName} task model profile is configured. Run /task-models.`);
 	for (const route of orderedProfileRoutes(profile)) {
 		const resolved = resolveTaskModelRoute(ctx, route);
 		if (resolved) return resolved;
 	}
-	throw new Error(`No usable ${modelClass} task model route. Run /task-models.`);
+	throw new Error(`No usable ${profileName} task model route. Run /task-models.`);
 }
 
 const cleanText = (value: unknown, field: string, file: string): string => {
@@ -306,6 +312,8 @@ async function runPi(
 		child.stderr.setEncoding("utf8");
 		let lineParts: string[] = [];
 		let lineBytes = 0;
+		let linePrefix = "";
+		let ignoreLine = false;
 		let output = "";
 		const stderr = { prefix: "", totalBytes: 0 };
 		const partial = { prefix: "", totalBytes: 0 };
@@ -396,17 +404,29 @@ async function runPi(
 				const newline = data.indexOf("\n", offset);
 				const end = newline === -1 ? data.length : newline;
 				const part = data.slice(offset, end);
-				lineBytes += Buffer.byteLength(part, "utf8");
-				if (lineBytes > MAX_JSON_EVENT_BYTES) {
-					protocolError = new Error(`Subagent JSON event exceeds ${MAX_JSON_EVENT_BYTES} bytes.`);
-					killTree(true);
-					return;
+				if (!ignoreLine) {
+					linePrefix += part.slice(0, Math.max(0, 256 - linePrefix.length));
+					const eventType = JSON_EVENT_TYPE.exec(linePrefix)?.[1];
+					if (eventType && !CONSUMED_JSON_EVENTS.has(eventType)) {
+						ignoreLine = true;
+						lineParts = [];
+						lineBytes = 0;
+					} else {
+						lineBytes += Buffer.byteLength(part, "utf8");
+						if (lineBytes > MAX_JSON_EVENT_BYTES) {
+							protocolError = new Error(`Subagent JSON event exceeds ${MAX_JSON_EVENT_BYTES} bytes.`);
+							killTree(true);
+							return;
+						}
+						if (part) lineParts.push(part);
+					}
 				}
-				if (part) lineParts.push(part);
 				if (newline === -1) return;
-				processLine(lineParts.join(""));
+				if (!ignoreLine) processLine(lineParts.join(""));
 				lineParts = [];
 				lineBytes = 0;
+				linePrefix = "";
+				ignoreLine = false;
 				offset = newline + 1;
 			}
 		});
@@ -438,7 +458,7 @@ const Parameters = Type.Object({
 	role: Type.String({ description: "Configured Subagent role name" }),
 	task: Type.String({ description: "One bounded task with needed context and expected result" }),
 	modelClass: Type.Optional(StringEnum(MODEL_CLASSES, {
-		description: "Classify task complexity: fast for narrow lookups or mechanical edits; balanced for normal bounded work; frontier for ambiguous, cross-cutting, or high-risk reasoning. Defaults to the shared balanced profile.",
+		description: "Classify task complexity: fast for narrow lookups or mechanical edits; balanced for normal bounded work; frontier for ambiguous, cross-cutting, or high-risk reasoning. Defaults to the shared pi-subagent/delegateTask assignment.",
 	})),
 });
 
@@ -557,7 +577,7 @@ export default function subagentExtension(pi: ExtensionAPI): void {
 	pi.registerTool({
 		name: "delegate_task",
 		label: "Subagent",
-		description: `Delegate one bounded task to one isolated Pi Subagent. Roles: ${roleSummary()}. Choose fast for narrow work, balanced for normal work, or frontier for ambiguous and high-risk work. Request concise conclusions and file/line references; split broad scouting work.`,
+		description: `Delegate one bounded task to one isolated Pi Subagent. Roles: ${roleSummary()}. Choose fast for narrow work, balanced for normal work, or frontier for ambiguous and high-risk work; omit modelClass to use shared task-model settings. Request concise conclusions and file/line references; split broad scouting work.`,
 		parameters: Parameters,
 		async execute(toolCallId, params, signal, onUpdate, ctx) {
 			const task = cleanText(params.task, "task", "delegate_task");
@@ -570,8 +590,7 @@ export default function subagentExtension(pi: ExtensionAPI): void {
 			if (params.modelClass !== undefined && !isModelClass(params.modelClass)) {
 				throw new Error("delegate_task modelClass must be fast, balanced, or frontier.");
 			}
-			const modelClass = params.modelClass ?? "balanced";
-			const resolvedRoute = resolveTaskRoute(ctx, modelClass);
+			const resolvedRoute = resolveTaskRoute(ctx, params.modelClass);
 			const model = resolvedRoute.model;
 			const modelReferenceValue = modelReference(model);
 			const thinkingLevel = resolvedRoute.thinkingLevel;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@henryqw/pi-subagent",
-  "version": "2.0.2",
+  "version": "2.1.0",
   "description": "Delegate one task to an isolated Pi role with explicit extensions and skills.",
   "keywords": [
     "pi-package",
@@ -47,6 +47,6 @@
   },
   "dependencies": {
     "@henryqw/pi-multi-codex": "^0.3.8",
-    "@henryqw/pi-task-models": "^0.1.0"
+    "@henryqw/pi-task-models": "^0.2.0"
   }
 }