@agimon-ai/po-team 0.0.0-alpha.16 → 0.0.0-alpha.17

package/dist/extensions/server.cjs

@@ -19,6 +19,7 @@ let node_fs = require("node:fs");
 node_fs = require_atomicJson.__toESM(node_fs, 1);
 let node_path = require("node:path");
 node_path = require_atomicJson.__toESM(node_path, 1);
+let _agimon_ai_po_config = require("@agimon-ai/po-config");
 let node_os = require("node:os");
 node_os = require_atomicJson.__toESM(node_os, 1);
 let _agimon_ai_po_core_serverPlugin = require("@agimon-ai/po-core/serverPlugin");
@@ -1916,24 +1917,26 @@ function readWorkflowPreferences(payload) {
 	if (preferences.subagentThinking && !THINKING_LEVELS.some((level) => level === preferences.subagentThinking)) throw invalid(`subagentThinking must be one of: ${THINKING_LEVELS.join(", ")}.`);
 	return preferences;
 }
-function resolvedModelSelection(taskInput, agentConfig, runtime, parentModel, availableModels, teamPackageModels, preferences) {
+function resolvedModelSelection(taskInput, agentConfig, runtime, parentModel, availableModels, teamPackageModels, preferences, planning = {}) {
 	const parentModelId = isPiRuntime(runtime) && parentModel ? `${parentModel.provider}/${parentModel.id}` : void 0;
 	const agentModels = [...agentConfig.modelSource?.scope === "package" ? [] : [agentConfig.model], ...agentConfig.fallbackModels ?? []];
 	const candidates = (isPiRuntime(runtime) ? [
 		taskInput.model,
 		preferences.subagentModel,
+		planning.subagentModel,
 		...agentModels,
 		...teamPackageModels ?? [],
 		parentModelId
 	] : [
 		taskInput.model,
 		preferences.subagentModel,
+		planning.subagentModel,
 		...agentModels,
 		...teamPackageModels ?? []
 	]).filter((candidate) => Boolean(candidate?.trim()));
 	const [primaryModel, ...fallbackModels] = candidates;
 	const agentThinking = typeof agentConfig.thinking === "string" ? agentConfig.thinking : void 0;
-	if (!preferences.subagentModel && !preferences.subagentThinking) return {
+	if (!preferences.subagentModel && !preferences.subagentThinking && !planning.subagentModel && !planning.subagentThinking) return {
 		primaryModel,
 		fallbackModels,
 		model: selectAvailableModel(primaryModel, fallbackModels, availableModels),
@@ -1950,11 +1953,13 @@ function resolvedModelSelection(taskInput, agentConfig, runtime, parentModel, av
 	}
 	const selectedSuffix = model ? splitKnownThinkingSuffix(model).thinkingSuffix.slice(1) : void 0;
 	const explicitThinking = selectedCandidate === taskInput.model ? selectedSuffix : void 0;
-	const thinking = explicitThinking || preferences.subagentThinking || selectedSuffix || agentThinking;
-	const workflowThinking = !explicitThinking && (preferences.subagentThinking || selectedCandidate === preferences.subagentModel && selectedSuffix);
+	const planningThinking = !planning.subagentModel || selectedCandidate === planning.subagentModel ? planning.subagentThinking : void 0;
+	const preferredThinking = (!preferences.subagentModel || selectedCandidate === preferences.subagentModel || selectedCandidate === taskInput.model ? preferences.subagentThinking : void 0) || planningThinking;
+	const thinking = explicitThinking || preferredThinking || selectedSuffix || agentThinking;
+	const workflowThinking = !explicitThinking && (preferredThinking || selectedCandidate === preferences.subagentModel && selectedSuffix);
 	const modelInfo = findModelInfo(model, availableModels);
-	if (isPiRuntime(runtime) && workflowThinking && modelInfo && !getSupportedThinkingLevels(modelInfo).some((level) => level === thinking)) throw new PoTeamExpectedError(ERROR_CODE_INVALID_REQUEST, `WORKFLOW_RUN_CONFIG thinking '${thinking}' is unsupported by model '${modelInfo.fullId}'.`, false, `Choose a supported thinking level: ${getSupportedThinkingLevels(modelInfo).join(", ")}.`);
-	if (model && selectedSuffix && !explicitThinking && preferences.subagentThinking) model = splitKnownThinkingSuffix(model).baseModel;
+	if (isPiRuntime(runtime) && workflowThinking && modelInfo && !getSupportedThinkingLevels(modelInfo).some((level) => level === thinking)) throw new PoTeamExpectedError(ERROR_CODE_INVALID_REQUEST, `Subagent thinking '${thinking}' is unsupported by model '${modelInfo.fullId}'.`, false, `Choose a supported thinking level: ${getSupportedThinkingLevels(modelInfo).join(", ")}.`);
+	if (model && selectedSuffix && !explicitThinking && preferredThinking) model = splitKnownThinkingSuffix(model).baseModel;
 	return {
 		primaryModel,
 		fallbackModels,
@@ -2132,12 +2137,12 @@ var SpawnPlanner = class {
 	* the module header.
 	*/
 	async spawnOneChild(input) {
-		const { runId, operationId, agentConfig, identity, capabilityCeiling, excludeTools, teamPackageModels, workflowPreferences, taskInput, childIndex, fanout, fallbackCwd, parentSessionId, parentForkSource, sessionScope, environment, parentSessionFile, maxLiveRuns, admissionTimeoutMs, handshakeTimeoutMs, artifacts, artifactDir, availableModels, parentModel, runtime, runtimes, skillProjectionCache, sender } = input;
+		const { runId, operationId, agentConfig, identity, capabilityCeiling, excludeTools, teamPackageModels, workflowPreferences, planningPreferences, taskInput, childIndex, fanout, fallbackCwd, parentSessionId, parentForkSource, sessionScope, environment, parentSessionFile, maxLiveRuns, admissionTimeoutMs, handshakeTimeoutMs, artifacts, artifactDir, availableModels, parentModel, runtime, runtimes, skillProjectionCache, sender } = input;
 		const cwd = taskInput.cwd ?? fallbackCwd;
 		const task = taskInput.task ?? "";
 		const effectiveContext = resolveEffectiveContext(taskInput, agentConfig);
 		const effectiveRuntime = taskInput.runtime ?? runtime ?? agentConfig.runtime ?? "pi";
-		const modelSelection = resolvedModelSelection(taskInput, agentConfig, effectiveRuntime, parentModel, availableModels, teamPackageModels, workflowPreferences);
+		const modelSelection = resolvedModelSelection(taskInput, agentConfig, effectiveRuntime, parentModel, availableModels, teamPackageModels, workflowPreferences, planningPreferences);
 		if (modelSelection.primaryModel && availableModels !== void 0 && !modelSelection.model) return {
 			agent: agentConfig.name,
 			identity: identity.identity,
@@ -2159,7 +2164,11 @@ var SpawnPlanner = class {
 				error: error instanceof Error ? error.message : String(error)
 			};
 		}
-		const warnings = [...skillProjection.warnings, ...bestEffortRuntimeWarnings(effectiveRuntime, agentConfig, excludeTools)];
+		const warnings = [
+			...modelSelection.primaryModel && modelSelection.model && !selectAvailableModel(modelSelection.primaryModel, [], availableModels) ? [`[fallback] ${modelSelection.primaryModel} is unavailable. Using ${modelSelection.model}.`] : [],
+			...skillProjection.warnings,
+			...bestEffortRuntimeWarnings(effectiveRuntime, agentConfig, excludeTools)
+		];
 		const warning = warnings.length > 0 ? warnings.join(" ") : void 0;
 		const spawnInput = {
 			runId,
@@ -2231,6 +2240,7 @@ var SpawnPlanner = class {
 						source,
 						agent: agentConfig.name,
 						task,
+						...warning ? { meta: { warning } } : {},
 						sender: sender ?? MAIN_AGENT_SENDER,
 						cwd,
 						...modelSelection.model ? { model: modelSelection.model } : {},
@@ -2294,6 +2304,11 @@ var SpawnPlanner = class {
 	async spawn(request, config) {
 		const tasks = this.resolveTaskList(request);
 		const workflowPreferences = readWorkflowPreferences(request.environment?.WORKFLOW_RUN_CONFIG ?? this.environment.WORKFLOW_RUN_CONFIG);
+		const planningConfig = request.repoRoot && request.parentSelection?.state?.["minor-mode"]?.includes("plan") ? (0, _agimon_ai_po_config.loadPoConfig)(request.repoRoot, request.environment?.HOME ?? this.environment.HOME).modes?.planning?.subagents : void 0;
+		const planningPreferences = {
+			subagentModel: planningConfig?.model,
+			subagentThinking: planningConfig?.thinking
+		};
 		const fanout = tasks.length > 1;
 		const teamPackageExcludeTools = this.resolveTeamPackageExcludeTools();
 		const teamPackageModels = this.resolveTeamPackageModels();
@@ -2341,7 +2356,7 @@ var SpawnPlanner = class {
 			const cwd = taskInput.cwd ?? request.cwd;
 			this.validateCwd(cwd);
 			const runtime = taskInput.runtime ?? request.runtime ?? agent.runtime ?? "pi";
-			const modelSelection = resolvedModelSelection(taskInput, agent, runtime, request.parentModel, request.availableModels, teamPackageModels, workflowPreferences);
+			const modelSelection = resolvedModelSelection(taskInput, agent, runtime, request.parentModel, request.availableModels, teamPackageModels, workflowPreferences, planningPreferences);
 			if (modelSelection.primaryModel && request.availableModels !== void 0 && !modelSelection.model) throw new PoTeamExpectedError(ERROR_CODE_MODEL_UNAVAILABLE, `No authenticated model is available for '${agent.name}'.`, false, "Authenticate the requested model or choose an available model.");
 			if (taskInput.inlineAgent && !isPiRuntime(runtime)) throw new PoTeamExpectedError(ERROR_CODE_UNSUPPORTED_OPERATION, `Inline agent '${taskInput.agent}' requires ${PI_RUNTIME_REQUIREMENT} to enforce its read-only tools.`, false, "Remove the runtime override or use a discovered external-runtime agent without inlineAgent.");
 			if (capabilityCeiling && !isPiRuntime(runtime)) throw new PoTeamExpectedError(ERROR_CODE_UNSUPPORTED_OPERATION, `Runtime '${runtime}' cannot enforce the active capability ceiling.`, false, `Use ${PI_RUNTIME_REQUIREMENT} while plan mode or another capability ceiling is active.`);
@@ -2370,6 +2385,7 @@ var SpawnPlanner = class {
 			excludeTools: teamPackageExcludeTools,
 			teamPackageModels,
 			workflowPreferences,
+			planningPreferences,
 			capabilityCeiling,
 			taskInput,
 			childIndex,
@@ -2596,6 +2612,9 @@ function createDelegationBridge(deps) {
 				agentScope: "both",
 				sessionScope: session.sessionScope,
 				parentSessionId: session.sessionId,
+				parentSelection: session.parentSelection,
+				repoRoot: session.repoRoot,
+				environment: session.environment,
 				...forkRequestFields(forkSource),
 				availableModels: await session.availableModels(),
 				...session.parentModel ? { parentModel: session.parentModel } : {}
@@ -2624,6 +2643,11 @@ function createDelegationBridge(deps) {
 				requestId: request.requestId,
 				runId: outcome.runId
 			});
+			if (outcome.warning) try {
+				await session.reportWarning?.(outcome.warning);
+			} catch (error) {
+				process.emitWarning(`Delegation warning could not be delivered: ${error instanceof Error ? error.message : String(error)}`);
+			}
 			await monitor(entry, request, ctx, jobs, session.sessionId, session.sessionScope);
 		} catch (error) {
 			finish(entry, {
@@ -4892,6 +4916,8 @@ function createHeadlessSubagentTool(runtime, initialExecution) {
 						sessionScope: require_externalProcessIpc.createSessionScope(execution.sessionId),
 						environment: execution.environment,
 						parentSessionId: execution.sessionId,
+						parentSelection: execution.selection,
+						repoRoot: execution.repoRoot,
 						...params.concurrency === void 0 ? {} : { concurrency: params.concurrency },
 						...params.artifacts === void 0 ? {} : { artifacts: params.artifacts },
 						availableModels: await availableModelInfos(execution),
@@ -4992,7 +5018,7 @@ function subscribeNativeRunProjection(sessionId, listener) {
 		if (closed) return;
 		closed = true;
 		projection.listeners.delete(listener);
-		if (projection.listeners.size === 0 && projection.runs.size === 0) sessions.delete(sessionId);
+		if (projection.listeners.size === 0 && projection.runs.size === 0 && sessions.get(sessionId) === projection) sessions.delete(sessionId);
 	};
 }
 //#endregion
@@ -6386,9 +6412,17 @@ const root = (0, _agimon_ai_po_core_extensionFile.defineRoot)(({ context, host:
 					cwd: execution.cwd,
 					environment: serverHost.context.environment,
 					availableModels: () => availableModelInfos(execution),
+					repoRoot: execution.repoRoot,
+					get parentSelection() {
+						return execution.selection;
+					},
 					get parentModel() {
 						return execution.model;
 					},
+					reportWarning: (body) => execution.client.notify({
+						body,
+						level: "warning"
+					}),
 					captureForkSource: () => execution.session.forkSource?.()
 				},
 				directRunTracker: runtime.asyncJobTracker,
@@ -6521,39 +6555,43 @@ const root = (0, _agimon_ai_po_core_extensionFile.defineRoot)(({ context, host:
 		async onDispose() {
 			disposing = true;
 			activityAttached = false;
-			stopPeerDiscovery?.();
-			sessionPeers.clear();
-			execution.client.setStatus(TEAM_COST_STATUS, void 0);
-			channel.dispose();
-			runtime.completionNotifier.dispose();
-			const stopped = stopScopeRuns({
-				reason: "headless session ended",
-				jobs: jobs.list(),
-				stop: (runId, reason) => runtime.management.stop(runId, reason)
-			});
-			const nativeShutdown = runtime.nativeRuns.close().finally(() => nativeRunProjection.dispose(execution.sessionId));
-			const outcomes = await Promise.allSettled([stopped, nativeShutdown]);
-			for (const outcome of outcomes) if (outcome.status === "rejected") process.emitWarning(`Could not shut down headless Team runs: ${String(outcome.reason)}`);
-			unsubscribeCostTracker();
-			unsubscribeCostNative();
-			nativeRuns.clear();
-			childCosts.clear();
-			costText = void 0;
-			bridge.abandonAll();
-			fablePlan.abandonAll();
-			runtime.capabilityPolicies.clear();
-			runtime.management.releaseSessionScope(scope);
-			runtime.dispose();
-			runtime.asyncJobTracker.stop();
-			runtime.pollScheduler.stop();
-			const stoppedRuns = outcomes[0];
-			await telemetry.recordEvent("po_team.session_finished", {
-				reason: "headless session ended",
-				"team.stopped_count": stoppedRuns?.status === "fulfilled" ? stoppedRuns.value.stopped.length : 0,
-				"team.unstoppable_count": stoppedRuns?.status === "fulfilled" ? stoppedRuns.value.unstoppable.length : 0,
-				outcome: stoppedRuns?.status === "fulfilled" && !stoppedRuns.value.unstoppable.length ? "stopped" : "degraded"
-			});
-			await telemetry.shutdown();
+			let stoppedRuns;
+			await (0, _agimon_ai_po_core_extensionFile.runCleanups)([
+				() => stopPeerDiscovery?.(),
+				() => sessionPeers.clear(),
+				() => execution.client.setStatus(TEAM_COST_STATUS, void 0),
+				() => channel.dispose(),
+				() => runtime.completionNotifier.dispose(),
+				async () => {
+					stoppedRuns = await stopScopeRuns({
+						reason: "headless session ended",
+						jobs: jobs.list(),
+						stop: (runId, reason) => runtime.management.stop(runId, reason)
+					});
+				},
+				() => (0, _agimon_ai_po_core_extensionFile.runCleanups)([() => runtime.nativeRuns.close(), () => nativeRunProjection.dispose(execution.sessionId)], "Native Team runs cleanup failed."),
+				unsubscribeCostTracker,
+				unsubscribeCostNative,
+				() => nativeRuns.clear(),
+				() => childCosts.clear(),
+				() => {
+					costText = void 0;
+				},
+				() => bridge.abandonAll(),
+				() => fablePlan.abandonAll(),
+				() => runtime.capabilityPolicies.clear(),
+				() => runtime.management.releaseSessionScope(scope),
+				() => runtime.dispose(),
+				() => runtime.asyncJobTracker.stop(),
+				() => runtime.pollScheduler.stop(),
+				() => telemetry.recordEvent("po_team.session_finished", {
+					reason: "headless session ended",
+					"team.stopped_count": stoppedRuns?.stopped.length ?? 0,
+					"team.unstoppable_count": stoppedRuns?.unstoppable.length ?? 0,
+					outcome: stoppedRuns && !stoppedRuns.unstoppable.length ? "stopped" : "degraded"
+				}),
+				() => telemetry.shutdown()
+			], "Team cleanup failed.");
 		}
 	};
 });