@narumitw/pi-subagents 0.53.0 → 0.54.0

package/src/auto-transport.ts

@@ -1,4 +1,5 @@
-import { discoverAgents, type SubagentSettings } from "./agents.js";
+import { discoverAgents } from "./agents/discovery.js";
+import type { SubagentSettings } from "./agents/types.js";
 import type { ManagedAgent, TurnOutcome } from "./registry.js";
 import { isWriteCapable } from "./stateful-safety.js";
 import type { SubagentTransport } from "./transport.js";

package/src/automation.ts

@@ -9,7 +9,9 @@ import {
 } from "@earendil-works/pi-coding-agent";
 import { Text } from "@earendil-works/pi-tui";
 import { type Static, Type } from "typebox";
-import { discoverAgents, getBuiltInAgent, type SubagentSettings } from "./agents.js";
+import { getBuiltInAgent } from "./agents/built-ins.js";
+import { discoverAgents } from "./agents/discovery.js";
+import type { SubagentSettings } from "./agents/types.js";
 import {
 	AutomationRequestSchema,
 	parseAutomationRequest,
@@ -39,7 +41,10 @@ import {
 	type SubagentDetails,
 } from "./runner.js";
 import { boundedPrivateText } from "./safe-text.js";
-import { DEFAULT_DELEGATION_CWD_POLICY, resolveBlockingMaxParallelTasks } from "./settings.js";
+import {
+	DEFAULT_DELEGATION_CWD_POLICY,
+	resolveBlockingMaxParallelTasks,
+} from "./settings/inspection.js";
 import {
 	type CompiledWorkflowPlan,
 	compileWorkflowPlan,

package/src/capability-router.ts

@@ -1,4 +1,4 @@
-import { type AgentConfig, resolveAgentToolNames } from "./agents.js";
+import { type AgentConfig, resolveAgentToolNames } from "./agents/types.js";
 import type { CapabilityHint } from "./capabilities.js";
 
 export interface CapabilityRouteRequest {

package/src/completion-delivery.ts

@@ -1,12 +1,11 @@
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
-import type { CompletionDelivery } from "./agents.js";
+import type { CompletionDelivery } from "./agents/types.js";
 import { redactPrivateText } from "./context.js";
-import { DEFAULT_MAX_CONTEXT_BYTES, truncateUtf8 } from "./limits.js";
+import { DEFAULT_MAX_CONTEXT_BYTES, MAX_TOOL_MESSAGE_BYTES, truncateUtf8 } from "./limits.js";
 import type { AgentTurnCompletion, ManagedAgent } from "./registry.js";
 import { safeTerminalText } from "./safe-text.js";
 import { PI_SUBAGENTS_RPC_PROTOCOL } from "./transport-types.js";
 
-const MAX_TOOL_MESSAGE_BYTES = 2 * 1024;
 const MAX_COMPLETION_ERROR_BYTES = 512;
 const MAX_COMPLETIONS_PER_MESSAGE = 16;
 const COMPLETION_BATCH_DELAY_MS = 10;

package/src/config-status.ts

@@ -4,11 +4,11 @@ import type {
 	ConsultationCwdPolicy,
 	ConsultResourcePolicy,
 	DelegationCwdPolicy,
-} from "./agents.js";
+} from "./agents/types.js";
 import type { SubagentSettingsRuntime } from "./config-ui.js";
 import { safeTerminalLine as safeTerminalText } from "./safe-text.js";
+import type { DelegationWorkflow } from "./settings/inspection.js";
 import {
-	type DelegationWorkflow,
 	inspectBlockingParallelLimitSettings,
 	inspectCompletionDeliverySettings,
 	inspectConsultResourceSettings,

package/src/config-ui.ts

@@ -1,11 +1,11 @@
 import type { ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
-import {
-	type CompletionDelivery,
-	type ConsultationCwdPolicy,
-	type ConsultResourcePolicy,
-	type DelegationCwdPolicy,
-	discoverAgents,
-} from "./agents.js";
+import { discoverAgents } from "./agents/discovery.js";
+import type {
+	CompletionDelivery,
+	ConsultationCwdPolicy,
+	ConsultResourcePolicy,
+	DelegationCwdPolicy,
+} from "./agents/types.js";
 import {
 	completionLabel,
 	consultationCwdLabel,
@@ -38,8 +38,8 @@ import {
 } from "./parallel-limit-ui.js";
 import type { ManagedAgent } from "./registry.js";
 import { safeTerminalLine as safeTerminalText } from "./safe-text.js";
+import type { DelegationWorkflow } from "./settings/inspection.js";
 import {
-	type DelegationWorkflow,
 	hasOwn,
 	inspectBlockingParallelLimitSettings,
 	inspectCompletionDeliverySettings,

package/src/consult-resources.ts

@@ -1,4 +1,4 @@
-import type { ConsultResourcePolicy } from "./agents.js";
+import type { ConsultResourcePolicy } from "./agents/types.js";
 import { DEFAULT_MAX_CONTEXT_BYTES, truncateUtf8 } from "./limits.js";
 import { resolvePiPromptResources } from "./prompt-resources.js";
 import type { ChildLaunchPolicy } from "./runner.js";

package/src/consult.ts

@@ -8,17 +8,16 @@ import {
 	type ToolDefinition,
 } from "@earendil-works/pi-coding-agent";
 import { type Static, Type } from "typebox";
+import { DEFAULT_AGENT_CATALOG_MAX_ITEMS } from "./agents/catalog.js";
+import { type AgentDiscoveryResult, discoverAgents } from "./agents/discovery.js";
 import {
 	type AgentConfig,
-	type AgentDiscoveryResult,
 	type AgentScope,
 	type ConsultResourcePolicy,
-	DEFAULT_AGENT_CATALOG_MAX_ITEMS,
-	discoverAgents,
 	isThinkingLevel,
 	type SubagentSettings,
 	THINKING_LEVELS,
-} from "./agents.js";
+} from "./agents/types.js";
 import { resolveConsultTools } from "./consult-policy.js";
 import { renderConsultCall, renderConsultResult } from "./consult-render.js";
 import { resolveConsultResourceLaunchPolicy } from "./consult-resources.js";
@@ -27,7 +26,10 @@ import {
 	type ResolvedSubagentTarget,
 	resolveSubagentTarget,
 } from "./cwd-policy.js";
-import { assertSubagentDepthAllowed, resolveDefaultSubagentTimeoutMs } from "./execution.js";
+import {
+	assertSubagentDepthAllowed,
+	resolveDefaultSubagentTimeoutMs,
+} from "./execution/runtime-policy.js";
 import {
 	DEFAULT_MAX_CONTEXT_BYTES,
 	DEFAULT_MAX_STDERR_BYTES,
@@ -45,8 +47,8 @@ import { boundedPrivateText, boundText, safeDisplayPath, safeTerminalLine } from
 import {
 	DEFAULT_CONSULT_RESOURCE_POLICY,
 	DEFAULT_CONSULTATION_CWD_POLICY,
-	resolveSubagentThinkingLevel,
-} from "./settings.js";
+} from "./settings/inspection.js";
+import { resolveSubagentThinkingLevel } from "./settings.js";
 
 const ConsultScopeSchema = StringEnum(["user", "project", "both"] as const, {
 	default: "user",

package/src/create-stateful-transport.ts

@@ -1,5 +1,6 @@
 import type { ModelRegistry } from "@earendil-works/pi-coding-agent";
-import { discoverAgents, type SubagentSettings, type SubagentTransportKind } from "./agents.js";
+import { discoverAgents } from "./agents/discovery.js";
+import type { SubagentSettings, SubagentTransportKind } from "./agents/types.js";
 import { AutoTransport } from "./auto-transport.js";
 import {
 	type ChildSessionFactory,

package/src/cwd-policy.ts

@@ -1,7 +1,7 @@
 import * as fs from "node:fs";
 import * as path from "node:path";
 import { getAgentDir, ProjectTrustStore } from "@earendil-works/pi-coding-agent";
-import type { ConsultationCwdPolicy, DelegationCwdPolicy } from "./agents.js";
+import type { ConsultationCwdPolicy, DelegationCwdPolicy } from "./agents/types.js";
 import { safeTerminalLine } from "./safe-text.js";
 
 export type TargetBoundary = "current-workspace" | "external";

package/src/execution/budget.ts

@@ -0,0 +1,56 @@
+import type { AgentConfig } from "../agents/types.js";
+import type { TurnLimits } from "../turn-budget.js";
+
+export interface ExecutionBudget {
+	timeoutMs: number;
+	workTimeoutReason: "work_timeout" | "orchestration_timeout";
+	workTimeoutReportLimit: number;
+}
+
+export function resolveConfiguredTimeout(
+	agents: readonly AgentConfig[],
+	agentName: string,
+	localTimeoutMs: number | undefined,
+	topLevelTimeoutMs: number | undefined,
+	defaultTimeoutMs: number,
+): number {
+	return (
+		localTimeoutMs ??
+		topLevelTimeoutMs ??
+		agents.find((agent) => agent.name === agentName)?.timeoutMs ??
+		defaultTimeoutMs
+	);
+}
+
+export function mergeTurnLimits(local: TurnLimits | undefined, fallback: TurnLimits): TurnLimits {
+	return {
+		idleTimeoutMs: local?.idleTimeoutMs ?? fallback.idleTimeoutMs,
+		maxTurns: local?.maxTurns ?? fallback.maxTurns,
+		maxToolCalls: local?.maxToolCalls ?? fallback.maxToolCalls,
+	};
+}
+
+export function calculateExecutionBudget(input: {
+	requestedTimeoutMs: number;
+	orchestrationDeadline?: number;
+	totalTimeoutMs?: number;
+	now: number;
+}): ExecutionBudget | undefined {
+	if (input.orchestrationDeadline === undefined) {
+		return {
+			timeoutMs: input.requestedTimeoutMs,
+			workTimeoutReason: "work_timeout",
+			workTimeoutReportLimit: input.requestedTimeoutMs,
+		};
+	}
+	const remaining = Math.floor(input.orchestrationDeadline - input.now);
+	if (remaining < 1) return undefined;
+	const orchestrationLimited = remaining < input.requestedTimeoutMs;
+	return {
+		timeoutMs: Math.min(input.requestedTimeoutMs, remaining),
+		workTimeoutReason: orchestrationLimited ? "orchestration_timeout" : "work_timeout",
+		workTimeoutReportLimit: orchestrationLimited
+			? Math.floor(input.totalTimeoutMs as number)
+			: input.requestedTimeoutMs,
+	};
+}

package/src/execution/runtime-policy.ts

@@ -0,0 +1,19 @@
+export const FALLBACK_TIMEOUT_MS = 10 * 60 * 1000;
+
+export function parsePositiveInteger(value: string | undefined): number | undefined {
+	if (!value) return undefined;
+	const parsed = Number.parseInt(value, 10);
+	return Number.isFinite(parsed) && parsed > 0 ? parsed : undefined;
+}
+
+export function resolveDefaultSubagentTimeoutMs(): number {
+	return parsePositiveInteger(process.env.PI_SUBAGENT_TIMEOUT_MS) ?? FALLBACK_TIMEOUT_MS;
+}
+
+export function assertSubagentDepthAllowed(): void {
+	const depth = Number.parseInt(process.env.PI_SUBAGENT_DEPTH ?? "0", 10) || 0;
+	const maxDepth = parsePositiveInteger(process.env.PI_SUBAGENT_MAX_DEPTH) ?? 1;
+	if (depth >= maxDepth) {
+		throw new Error(`Subagent recursion depth limit reached (${maxDepth})`);
+	}
+}

package/src/execution-plan.ts

@@ -8,7 +8,7 @@ import {
 	type AgentSource,
 	resolveAgentToolNames,
 	type SubagentThinkingLevel,
-} from "./agents.js";
+} from "./agents/types.js";
 import type { TargetPolicyAudit } from "./cwd-policy.js";
 import type { DelegationContract } from "./delegation-contract.js";
 import type { SubagentResultFormat } from "./result-contract.js";

package/src/execution-profiles.ts

@@ -1,4 +1,4 @@
-import type { SubagentThinkingLevel } from "./agents.js";
+import type { SubagentThinkingLevel } from "./agents/types.js";
 import { readSubagentSettings, updateAgentSettingsPatch } from "./settings.js";
 
 export const EXECUTION_PROFILES = ["fast", "balanced", "deep"] as const;

package/src/execution-ui.ts

@@ -1,5 +1,5 @@
 import type { ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
-import type { AgentConfig, SubagentThinkingLevel } from "./agents.js";
+import type { AgentConfig, SubagentThinkingLevel } from "./agents/types.js";
 import {
 	applyExecutionProfile,
 	EXECUTION_PROFILES,