@selesai/code 0.5.28 → 0.5.29

package/dist/extensions/pi-subagents/src/runs/foreground/foreground-control.ts

@@ -4,7 +4,10 @@ interface BeginForegroundChildInput {
 	index: number;
 	agent: string;
 	description?: string;
+	model?: string;
+	thinking?: string;
 	interrupt: () => boolean;
+	detach?: () => boolean;
 }
 
 function copyProgress(target: ForegroundChildControl, progress: AgentProgress | undefined): void {
@@ -16,6 +19,10 @@ function copyProgress(target: ForegroundChildControl, progress: AgentProgress |
 	target.currentPath = progress.currentPath;
 	target.turnCount = progress.turnCount;
 	target.tokens = progress.tokens;
+	target.inputTokens = progress.inputTokens;
+	target.outputTokens = progress.outputTokens;
+	target.model = progress.model;
+	target.thinking = progress.thinking;
 	target.toolCount = progress.toolCount;
 }
 
@@ -30,8 +37,13 @@ function syncCurrentChild(control: ForegroundRunControl, child: ForegroundChildC
 	control.currentPath = child.currentPath;
 	control.turnCount = child.turnCount;
 	control.tokens = child.tokens;
+	control.inputTokens = child.inputTokens;
+	control.outputTokens = child.outputTokens;
+	control.model = child.model;
+	control.thinking = child.thinking;
 	control.toolCount = child.toolCount;
 	control.interrupt = child.interrupt;
+	control.detach = child.detach;
 	control.updatedAt = child.updatedAt;
 }
 
@@ -45,8 +57,27 @@ function clearCurrentChild(control: ForegroundRunControl): void {
 	control.currentPath = undefined;
 	control.turnCount = undefined;
 	control.tokens = undefined;
+	control.inputTokens = undefined;
+	control.outputTokens = undefined;
+	control.model = undefined;
+	control.thinking = undefined;
 	control.toolCount = undefined;
 	control.interrupt = undefined;
+	control.detach = undefined;
+}
+
+export function retainForegroundSchedulingOwner(control: ForegroundRunControl): void {
+	control.schedulingOwners = (control.schedulingOwners ?? 0) + 1;
+	control.updatedAt = Date.now();
+}
+
+export function settleForegroundSchedulingOwner(control: ForegroundRunControl): void {
+	control.schedulingOwners = Math.max(0, (control.schedulingOwners ?? 0) - 1);
+	control.updatedAt = Date.now();
+}
+
+export function foregroundSchedulingSettled(control: ForegroundRunControl): boolean {
+	return (control.schedulingOwners ?? 0) === 0;
 }
 
 export function beginForegroundChild(control: ForegroundRunControl, input: BeginForegroundChildInput): void {
@@ -57,6 +88,8 @@ export function beginForegroundChild(control: ForegroundRunControl, input: Begin
 		...(input.description ? { description: input.description } : {}),
 		startedAt: now,
 		updatedAt: now,
+		...(input.model ? { model: input.model } : {}),
+		...(input.thinking ? { thinking: input.thinking } : {}),
 	};
 	child.interrupt = () => {
 		if (!input.interrupt()) return false;
@@ -65,6 +98,15 @@ export function beginForegroundChild(control: ForegroundRunControl, input: Begin
 		syncCurrentChild(control, child);
 		return true;
 	};
+	if (input.detach) {
+		child.detach = () => {
+			if (!input.detach?.()) return false;
+			child.currentActivityState = undefined;
+			child.updatedAt = Date.now();
+			syncCurrentChild(control, child);
+			return true;
+		};
+	}
 	control.activeChildren ??= new Map();
 	control.activeChildren.set(input.index, child);
 	syncCurrentChild(control, child);