pi-crew 0.2.3 → 0.2.5
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/AGENTS.md +57 -32
- package/CHANGELOG.md +466 -448
- package/LICENSE +21 -21
- package/NOTICE.md +16 -16
- package/README.md +323 -323
- package/docs/FEATURE_INTAKE.md +126 -0
- package/docs/HARNESS.md +86 -0
- package/docs/HARNESS_BACKLOG.md +41 -0
- package/docs/TEST_MATRIX.md +49 -0
- package/docs/actions-reference.md +595 -595
- package/docs/architecture.md +180 -180
- package/docs/code-review-2026-05-11.md +592 -592
- package/docs/commands-reference.md +347 -347
- package/docs/comparison-pi-subagents-vs-pi-crew.md +303 -0
- package/docs/decisions/0001-durable-state.md +41 -0
- package/docs/decisions/0002-child-process-for-async.md +42 -0
- package/docs/decisions/0003-depth-guard.md +36 -0
- package/docs/decisions/0004-execfile-over-exec.md +34 -0
- package/docs/decisions/0005-no-parameter-properties.md +49 -0
- package/docs/decisions/0006-publish-bundled-esm.md +63 -0
- package/docs/decisions/0007-active-run-binary-index.md +54 -0
- package/docs/decisions/0008-child-pi-warm-pool.md +61 -0
- package/docs/decisions/README.md +23 -0
- package/docs/followup-review-round4-2026-05-13.md +107 -0
- package/docs/implementation-plan-top3.md +333 -0
- package/docs/live-mailbox-runtime.md +36 -36
- package/docs/next-upgrade-roadmap.md +808 -808
- package/docs/oh-my-pi-research.md +509 -0
- package/docs/perf/baseline-2026-05.md +113 -0
- package/docs/perf/final-report-2026-05.md +206 -0
- package/docs/perf/sprint-1-report.md +71 -0
- package/docs/perf/sprint-2-report.md +81 -0
- package/docs/perf/sprint-2.5-report.md +53 -0
- package/docs/perf/sprint-3-report.md +36 -0
- package/docs/perf/sprint-4-report.md +47 -0
- package/docs/perf/sprint-5-report.md +51 -0
- package/docs/perf/sprint-6-report.md +94 -0
- package/docs/perf/sprint-7-report.md +74 -0
- package/docs/perf/upgrade-plan-2026-05.md +147 -0
- package/docs/pi-subagents3-deep-analysis.md +508 -0
- package/docs/product/README.md +31 -0
- package/docs/product/platform.md +27 -0
- package/docs/product/runtime-safety.md +37 -0
- package/docs/product/team-run.md +39 -0
- package/docs/product/team-tool.md +37 -0
- package/docs/publishing.md +65 -65
- package/docs/resource-formats.md +134 -134
- package/docs/runtime-analysis-child-vs-live.md +171 -0
- package/docs/runtime-flow.md +148 -148
- package/docs/runtime-migration-in-process-analysis.md +250 -0
- package/docs/stories/README.md +30 -0
- package/docs/stories/backlog.md +36 -0
- package/docs/templates/decision.md +27 -0
- package/docs/templates/story.md +44 -0
- package/docs/templates/validation-report.md +32 -0
- package/docs/usage.md +238 -238
- package/index.ts +7 -6
- package/install.mjs +65 -65
- package/package.json +107 -100
- package/schema.json +222 -222
- package/skills/child-pi-spawning/SKILL.md +213 -0
- package/skills/context-artifact-hygiene/SKILL.md +32 -0
- package/skills/event-log-tracing/SKILL.md +299 -0
- package/skills/git-master/SKILL.md +225 -24
- package/skills/live-agent-lifecycle/SKILL.md +192 -0
- package/skills/mailbox-interactive/SKILL.md +300 -19
- package/skills/model-routing-context/SKILL.md +94 -0
- package/skills/multi-perspective-review/SKILL.md +88 -0
- package/skills/read-only-explorer/SKILL.md +250 -26
- package/skills/safe-bash/SKILL.md +307 -21
- package/skills/verification-before-done/SKILL.md +11 -2
- package/skills/widget-rendering/SKILL.md +258 -0
- package/skills/workspace-isolation/SKILL.md +202 -0
- package/skills/worktree-isolation/SKILL.md +202 -18
- package/src/adapters/claude-adapter.ts +25 -25
- package/src/adapters/codex-adapter.ts +21 -21
- package/src/adapters/cursor-adapter.ts +17 -17
- package/src/adapters/export-util.ts +137 -137
- package/src/adapters/index.ts +15 -15
- package/src/adapters/registry.ts +18 -18
- package/src/adapters/types.ts +23 -23
- package/src/agents/agent-config.ts +38 -38
- package/src/agents/agent-serializer.ts +38 -38
- package/src/agents/discover-agents.ts +121 -118
- package/src/config/config.ts +740 -858
- package/src/config/defaults.ts +96 -96
- package/src/config/drift-detector.ts +211 -211
- package/src/config/markers.ts +327 -327
- package/src/config/resilient-parser.ts +109 -108
- package/src/config/suggestions.ts +74 -74
- package/src/config/types.ts +199 -0
- package/src/extension/async-notifier.ts +123 -89
- package/src/extension/autonomous-policy.ts +169 -169
- package/src/extension/cross-extension-rpc.ts +104 -104
- package/src/extension/help.ts +47 -47
- package/src/extension/import-index.ts +69 -69
- package/src/extension/management.ts +395 -382
- package/src/extension/notification-router.ts +116 -116
- package/src/extension/notification-sink.ts +51 -51
- package/src/extension/project-init.ts +168 -168
- package/src/extension/register.ts +859 -668
- package/src/extension/registration/artifact-cleanup.ts +15 -15
- package/src/extension/registration/command-utils.ts +54 -54
- package/src/extension/registration/commands.ts +559 -452
- package/src/extension/registration/compaction-guard.ts +125 -125
- package/src/extension/registration/subagent-helpers.ts +102 -102
- package/src/extension/registration/subagent-tools.ts +220 -159
- package/src/extension/registration/team-tool.ts +159 -99
- package/src/extension/registration/viewers.ts +29 -0
- package/src/extension/result-watcher.ts +128 -128
- package/src/extension/run-bundle-schema.ts +89 -89
- package/src/extension/run-export.ts +73 -73
- package/src/extension/run-import.ts +84 -84
- package/src/extension/run-index.ts +94 -94
- package/src/extension/run-maintenance.ts +142 -142
- package/src/extension/session-summary.ts +8 -8
- package/src/extension/team-manager-command.ts +96 -96
- package/src/extension/team-recommendation.ts +188 -188
- package/src/extension/team-tool/api.ts +5 -2
- package/src/extension/team-tool/cancel.ts +224 -209
- package/src/extension/team-tool/config-patch.ts +36 -36
- package/src/extension/team-tool/context.ts +60 -60
- package/src/extension/team-tool/doctor.ts +242 -242
- package/src/extension/team-tool/handle-settings.ts +421 -195
- package/src/extension/team-tool/inspect.ts +41 -41
- package/src/extension/team-tool/lifecycle-actions.ts +139 -139
- package/src/extension/team-tool/parallel-dispatch.ts +156 -156
- package/src/extension/team-tool/plan.ts +19 -19
- package/src/extension/team-tool/respond.ts +112 -111
- package/src/extension/team-tool/run.ts +246 -229
- package/src/extension/team-tool/status.ts +110 -110
- package/src/extension/team-tool-types.ts +13 -13
- package/src/extension/team-tool.ts +344 -344
- package/src/extension/tool-result.ts +16 -16
- package/src/extension/validate-resources.ts +77 -77
- package/src/hooks/registry.ts +61 -61
- package/src/hooks/types.ts +40 -40
- package/src/i18n.ts +184 -184
- package/src/observability/correlation.ts +35 -35
- package/src/observability/event-to-metric.ts +68 -68
- package/src/observability/exporters/adapter.ts +30 -30
- package/src/observability/exporters/otlp-exporter.ts +106 -92
- package/src/observability/exporters/prometheus-exporter.ts +54 -54
- package/src/observability/metric-registry.ts +87 -87
- package/src/observability/metric-retention.ts +54 -54
- package/src/observability/metric-sink.ts +81 -56
- package/src/observability/metrics-primitives.ts +167 -167
- package/src/prompt/prompt-runtime.ts +72 -72
- package/src/runtime/adaptive-plan.ts +338 -0
- package/src/runtime/agent-control.ts +169 -169
- package/src/runtime/agent-memory.ts +72 -72
- package/src/runtime/agent-observability.ts +114 -114
- package/src/runtime/async-marker.ts +26 -26
- package/src/runtime/async-runner.ts +153 -153
- package/src/runtime/attention-events.ts +28 -28
- package/src/runtime/auto-resume.ts +100 -100
- package/src/runtime/background-runner.ts +122 -89
- package/src/runtime/cancellation.ts +61 -61
- package/src/runtime/capability-inventory.ts +116 -116
- package/src/runtime/child-pi-pool.ts +68 -0
- package/src/runtime/child-pi.ts +541 -461
- package/src/runtime/code-summary.ts +247 -247
- package/src/runtime/compaction-summary.ts +271 -271
- package/src/runtime/concurrency.ts +58 -58
- package/src/runtime/crash-recovery.ts +317 -301
- package/src/runtime/crew-agent-records.ts +379 -281
- package/src/runtime/crew-agent-runtime.ts +60 -60
- package/src/runtime/cross-extension-rpc.ts +72 -0
- package/src/runtime/custom-tools/irc-tool.ts +201 -201
- package/src/runtime/custom-tools/submit-result-tool.ts +90 -90
- package/src/runtime/deadletter.ts +47 -47
- package/src/runtime/delivery-coordinator.ts +176 -176
- package/src/runtime/delta-conflict.ts +360 -360
- package/src/runtime/diagnostic-export.ts +102 -102
- package/src/runtime/direct-run.ts +35 -35
- package/src/runtime/effectiveness.ts +82 -81
- package/src/runtime/errors/crew-errors.ts +166 -0
- package/src/runtime/event-stream-bridge.ts +92 -92
- package/src/runtime/foreground-control.ts +82 -82
- package/src/runtime/green-contract.ts +46 -46
- package/src/runtime/group-join.ts +234 -106
- package/src/runtime/heartbeat-watcher.ts +145 -124
- package/src/runtime/iteration-hooks.ts +267 -267
- package/src/runtime/live-agent-control.ts +88 -88
- package/src/runtime/live-agent-manager.ts +377 -179
- package/src/runtime/live-control-realtime.ts +36 -36
- package/src/runtime/live-session-runtime.ts +676 -600
- package/src/runtime/loop-gates.ts +129 -129
- package/src/runtime/manifest-cache.ts +263 -263
- package/src/runtime/mcp-proxy.ts +113 -113
- package/src/runtime/metric-parser.ts +40 -40
- package/src/runtime/model-fallback.ts +282 -274
- package/src/runtime/model-resolver.ts +118 -0
- package/src/runtime/output-validator.ts +187 -187
- package/src/runtime/overflow-recovery.ts +175 -175
- package/src/runtime/parallel-research.ts +44 -44
- package/src/runtime/parallel-utils.ts +156 -156
- package/src/runtime/parent-guard.ts +80 -80
- package/src/runtime/phase-progress.ts +217 -217
- package/src/runtime/pi-args.ts +165 -165
- package/src/runtime/pi-json-output.ts +111 -111
- package/src/runtime/pi-spawn.ts +167 -167
- package/src/runtime/policy-engine.ts +79 -79
- package/src/runtime/post-checks.ts +125 -125
- package/src/runtime/post-exit-stdio-guard.ts +86 -86
- package/src/runtime/process-status.ts +97 -73
- package/src/runtime/progress-event-coalescer.ts +43 -43
- package/src/runtime/recovery-recipes.ts +74 -74
- package/src/runtime/retry-executor.ts +81 -81
- package/src/runtime/role-permission.ts +39 -39
- package/src/runtime/run-tracker.ts +99 -0
- package/src/runtime/runtime-policy.ts +21 -0
- package/src/runtime/runtime-resolver.ts +94 -91
- package/src/runtime/scheduler.ts +294 -0
- package/src/runtime/semaphore.ts +131 -131
- package/src/runtime/sensitive-paths.ts +92 -92
- package/src/runtime/session-usage.ts +79 -79
- package/src/runtime/settings-store.ts +103 -0
- package/src/runtime/sidechain-output.ts +29 -29
- package/src/runtime/skill-instructions.ts +222 -222
- package/src/runtime/stale-reconciler.ts +198 -189
- package/src/runtime/streaming-output.ts +47 -0
- package/src/runtime/subagent-manager.ts +404 -400
- package/src/runtime/subprocess-tool-registry.ts +67 -67
- package/src/runtime/task-display.ts +38 -38
- package/src/runtime/task-graph-scheduler.ts +122 -122
- package/src/runtime/task-graph.ts +207 -207
- package/src/runtime/task-output-context.ts +177 -177
- package/src/runtime/task-packet.ts +93 -93
- package/src/runtime/task-quality.ts +207 -207
- package/src/runtime/task-runner/capabilities.ts +78 -78
- package/src/runtime/task-runner/live-executor.ts +131 -113
- package/src/runtime/task-runner/progress.ts +119 -119
- package/src/runtime/task-runner/prompt-builder.ts +139 -139
- package/src/runtime/task-runner/prompt-pipeline.ts +64 -64
- package/src/runtime/task-runner/result-utils.ts +14 -14
- package/src/runtime/task-runner/run-projection.ts +103 -103
- package/src/runtime/task-runner/state-helpers.ts +22 -22
- package/src/runtime/task-runner.ts +469 -459
- package/src/runtime/team-runner.ts +693 -945
- package/src/runtime/usage-tracker.ts +71 -0
- package/src/runtime/worker-heartbeat.ts +21 -21
- package/src/runtime/worker-startup.ts +57 -57
- package/src/runtime/workflow-state.ts +187 -187
- package/src/runtime/yield-handler.ts +190 -190
- package/src/schema/config-schema.ts +172 -168
- package/src/schema/team-tool-schema.ts +126 -126
- package/src/schema/validation-types.ts +151 -148
- package/src/skills/discover-skills.ts +67 -67
- package/src/skills/skill-templates.ts +374 -374
- package/src/state/active-run-registry.ts +227 -191
- package/src/state/artifact-store.ts +130 -129
- package/src/state/atomic-write.ts +262 -195
- package/src/state/blob-store.ts +116 -116
- package/src/state/contracts.ts +111 -111
- package/src/state/event-log-rotation.ts +161 -158
- package/src/state/event-log.ts +383 -303
- package/src/state/event-reconstructor.ts +217 -217
- package/src/state/jsonl-writer.ts +82 -82
- package/src/state/locks.ts +146 -146
- package/src/state/mailbox.ts +446 -405
- package/src/state/state-store.ts +364 -351
- package/src/state/task-claims.ts +44 -44
- package/src/state/types.ts +285 -285
- package/src/state/usage.ts +29 -29
- package/src/subagents/async-entry.ts +1 -1
- package/src/subagents/index.ts +3 -3
- package/src/subagents/live/control.ts +1 -1
- package/src/subagents/live/manager.ts +1 -1
- package/src/subagents/live/realtime.ts +1 -1
- package/src/subagents/live/session-runtime.ts +1 -1
- package/src/subagents/manager.ts +1 -1
- package/src/subagents/spawn.ts +1 -1
- package/src/teams/discover-teams.ts +116 -116
- package/src/teams/team-config.ts +27 -27
- package/src/teams/team-serializer.ts +38 -38
- package/src/types/diff.d.ts +18 -18
- package/src/ui/agent-management-overlay.ts +144 -144
- package/src/ui/crew-widget.ts +487 -370
- package/src/ui/dashboard-panes/agents-pane.ts +109 -28
- package/src/ui/dashboard-panes/cancellation-pane.ts +42 -42
- package/src/ui/dashboard-panes/capability-pane.ts +59 -59
- package/src/ui/dashboard-panes/health-pane.ts +30 -30
- package/src/ui/dashboard-panes/mailbox-pane.ts +35 -35
- package/src/ui/dashboard-panes/progress-pane.ts +30 -30
- package/src/ui/dashboard-panes/transcript-pane.ts +10 -10
- package/src/ui/heartbeat-aggregator.ts +63 -63
- package/src/ui/keybinding-map.ts +97 -94
- package/src/ui/live-conversation-overlay.ts +152 -0
- package/src/ui/live-run-sidebar.ts +180 -180
- package/src/ui/mascot.ts +442 -442
- package/src/ui/overlays/agent-picker-overlay.ts +57 -57
- package/src/ui/overlays/confirm-overlay.ts +58 -58
- package/src/ui/overlays/mailbox-compose-overlay.ts +144 -144
- package/src/ui/overlays/mailbox-compose-preview.ts +63 -63
- package/src/ui/overlays/mailbox-detail-overlay.ts +122 -122
- package/src/ui/pi-ui-compat.ts +57 -57
- package/src/ui/powerbar-publisher.ts +221 -197
- package/src/ui/render-scheduler.ts +216 -143
- package/src/ui/run-action-dispatcher.ts +118 -118
- package/src/ui/run-dashboard.ts +526 -464
- package/src/ui/run-event-bus.ts +208 -208
- package/src/ui/run-snapshot-cache.ts +826 -777
- package/src/ui/settings-overlay.ts +721 -0
- package/src/ui/snapshot-types.ts +86 -70
- package/src/ui/theme-adapter.ts +190 -190
- package/src/ui/tool-progress-formatter.ts +89 -0
- package/src/ui/transcript-cache.ts +94 -94
- package/src/ui/transcript-viewer.ts +335 -335
- package/src/utils/conflict-detect.ts +662 -0
- package/src/utils/file-coalescer.ts +86 -86
- package/src/utils/frontmatter.ts +68 -68
- package/src/utils/fs-watch.ts +88 -31
- package/src/utils/gh-protocol.ts +479 -0
- package/src/utils/ids.ts +17 -17
- package/src/utils/incremental-reader.ts +104 -104
- package/src/utils/internal-error.ts +6 -6
- package/src/utils/names.ts +27 -27
- package/src/utils/paths.ts +102 -63
- package/src/utils/redaction.ts +44 -44
- package/src/utils/safe-paths.ts +47 -47
- package/src/utils/scan-cache.ts +136 -136
- package/src/utils/sse-parser.ts +134 -134
- package/src/utils/task-name-generator.ts +337 -337
- package/src/utils/timings.ts +33 -33
- package/src/utils/visual.ts +243 -198
- package/src/workflows/discover-workflows.ts +139 -139
- package/src/workflows/validate-workflow.ts +40 -40
- package/src/workflows/workflow-config.ts +26 -26
- package/src/workflows/workflow-serializer.ts +32 -32
- package/src/worktree/branch-freshness.ts +45 -45
- package/src/worktree/cleanup.ts +75 -75
- package/src/worktree/worktree-manager.ts +188 -188
- package/teams/default.team.md +12 -12
- package/teams/fast-fix.team.md +11 -11
- package/teams/implementation.team.md +18 -18
- package/teams/parallel-research.team.md +14 -14
- package/teams/research.team.md +11 -11
- package/teams/review.team.md +12 -12
- package/tsconfig.json +19 -19
- package/workflows/default.workflow.md +30 -30
- package/workflows/fast-fix.workflow.md +23 -23
- package/workflows/implementation.workflow.md +43 -43
- package/workflows/parallel-research.workflow.md +46 -46
- package/workflows/research.workflow.md +22 -22
- package/workflows/review.workflow.md +30 -30
- package/skills/task-packet/SKILL.md +0 -28
- package/skills/verify-evidence/SKILL.md +0 -27
|
@@ -1,400 +1,404 @@
|
|
|
1
|
-
import * as fs from "node:fs";
|
|
2
|
-
import * as path from "node:path";
|
|
3
|
-
import { loadRunManifestById } from "../state/state-store.ts";
|
|
4
|
-
import type { PiTeamsToolResult } from "../extension/tool-result.ts";
|
|
5
|
-
import { DEFAULT_SUBAGENT } from "../config/defaults.ts";
|
|
6
|
-
import { projectCrewRoot } from "../utils/paths.ts";
|
|
7
|
-
import { DEFAULT_PATHS } from "../config/defaults.ts";
|
|
8
|
-
import { logInternalError } from "../utils/internal-error.ts";
|
|
9
|
-
import { redactSecrets } from "../utils/redaction.ts";
|
|
10
|
-
|
|
11
|
-
export type SubagentStatus = "queued" | "running" | "completed" | "failed" | "cancelled" | "error" | "blocked" | "stopped";
|
|
12
|
-
|
|
13
|
-
export interface SubagentSpawnOptions {
|
|
14
|
-
cwd: string;
|
|
15
|
-
type: string;
|
|
16
|
-
description: string;
|
|
17
|
-
prompt: string;
|
|
18
|
-
background: boolean;
|
|
19
|
-
model?: string;
|
|
20
|
-
skill?: string | string[] | false;
|
|
21
|
-
maxTurns?: number;
|
|
22
|
-
ownerSessionGeneration?: number;
|
|
23
|
-
}
|
|
24
|
-
|
|
25
|
-
export interface SubagentRecord {
|
|
26
|
-
id: string;
|
|
27
|
-
runId?: string;
|
|
28
|
-
type: string;
|
|
29
|
-
description: string;
|
|
30
|
-
prompt: string;
|
|
31
|
-
status: SubagentStatus;
|
|
32
|
-
startedAt: number;
|
|
33
|
-
completedAt?: number;
|
|
34
|
-
result?: string;
|
|
35
|
-
error?: string;
|
|
36
|
-
resultConsumed?: boolean;
|
|
37
|
-
model?: string;
|
|
38
|
-
skill?: string | string[] | false;
|
|
39
|
-
background: boolean;
|
|
40
|
-
ownerSessionGeneration?: number;
|
|
41
|
-
stuckNotified?: boolean;
|
|
42
|
-
blockedAt?: number;
|
|
43
|
-
promise?: Promise<void>;
|
|
44
|
-
// Phase 1.6: Telemetry baseline fields
|
|
45
|
-
turnCount?: number;
|
|
46
|
-
terminated?: boolean;
|
|
47
|
-
durationMs?: number;
|
|
48
|
-
}
|
|
49
|
-
|
|
50
|
-
type SpawnRunner = (options: SubagentSpawnOptions, signal?: AbortSignal) => Promise<PiTeamsToolResult>;
|
|
51
|
-
type Notify = (record: SubagentRecord) => void;
|
|
52
|
-
type NotifyEvent = (type: string, data: Record<string, unknown>) => void;
|
|
53
|
-
|
|
54
|
-
interface QueuedSpawn {
|
|
55
|
-
record: SubagentRecord;
|
|
56
|
-
options: SubagentSpawnOptions;
|
|
57
|
-
runner: SpawnRunner;
|
|
58
|
-
signal?: AbortSignal;
|
|
59
|
-
}
|
|
60
|
-
|
|
61
|
-
function isValidSubagentId(id: string): boolean {
|
|
62
|
-
return /^[a-z0-9_]+$/i.test(id) && id.length <= 128;
|
|
63
|
-
}
|
|
64
|
-
|
|
65
|
-
function persistedSubagentPath(cwd: string, id: string): string {
|
|
66
|
-
if (!isValidSubagentId(id)) throw new Error(`Invalid subagent id: ${id}`);
|
|
67
|
-
return path.join(projectCrewRoot(cwd), DEFAULT_PATHS.state.subagentsSubdir, `${id}.json`);
|
|
68
|
-
}
|
|
69
|
-
|
|
70
|
-
function serializableRecord(record: SubagentRecord): SubagentRecord {
|
|
71
|
-
const { promise: _promise, ...rest } = record;
|
|
72
|
-
return rest;
|
|
73
|
-
}
|
|
74
|
-
|
|
75
|
-
export function savePersistedSubagentRecord(cwd: string, record: SubagentRecord): void {
|
|
76
|
-
try {
|
|
77
|
-
const filePath = persistedSubagentPath(cwd, record.id);
|
|
78
|
-
fs.mkdirSync(path.dirname(filePath), { recursive: true });
|
|
79
|
-
fs.writeFileSync(filePath, `${JSON.stringify(redactSecrets(serializableRecord(record)), null, 2)}\n`, "utf-8");
|
|
80
|
-
} catch (error) {
|
|
81
|
-
logInternalError("subagent-manager.save", error, `id=${record.id}`);
|
|
82
|
-
}
|
|
83
|
-
}
|
|
84
|
-
|
|
85
|
-
export function readPersistedSubagentRecord(cwd: string, id: string): SubagentRecord | undefined {
|
|
86
|
-
try {
|
|
87
|
-
const parsed = JSON.parse(fs.readFileSync(persistedSubagentPath(cwd, id), "utf-8"));
|
|
88
|
-
return parsed && typeof parsed === "object" && !Array.isArray(parsed) ? parsed as SubagentRecord : undefined;
|
|
89
|
-
} catch {
|
|
90
|
-
return undefined;
|
|
91
|
-
}
|
|
92
|
-
}
|
|
93
|
-
|
|
94
|
-
function resultText(result: PiTeamsToolResult): string {
|
|
95
|
-
return result.content?.map((item) => item.type === "text" ? item.text : "").filter(Boolean).join("\n") ?? "";
|
|
96
|
-
}
|
|
97
|
-
|
|
98
|
-
function detailsRunId(result: PiTeamsToolResult): string | undefined {
|
|
99
|
-
const details = result.details as { runId?: unknown } | undefined;
|
|
100
|
-
return typeof details?.runId === "string" ? details.runId : undefined;
|
|
101
|
-
}
|
|
102
|
-
|
|
103
|
-
function totalRunTurns(cwd: string, runId: string | undefined): number | undefined {
|
|
104
|
-
if (!runId) return undefined;
|
|
105
|
-
const loaded = loadRunManifestById(cwd, runId);
|
|
106
|
-
if (!loaded) return undefined;
|
|
107
|
-
let total = 0;
|
|
108
|
-
let hasTurns = false;
|
|
109
|
-
for (const task of loaded.tasks) {
|
|
110
|
-
const turns = task.usage?.turns ?? task.agentProgress?.turns;
|
|
111
|
-
if (typeof turns === "number" && Number.isFinite(turns)) {
|
|
112
|
-
total += turns;
|
|
113
|
-
hasTurns = true;
|
|
114
|
-
}
|
|
115
|
-
}
|
|
116
|
-
return hasTurns ? total : undefined;
|
|
117
|
-
}
|
|
118
|
-
|
|
119
|
-
export class SubagentManager {
|
|
120
|
-
private readonly records = new Map<string, SubagentRecord>();
|
|
121
|
-
private readonly cwdByRecord = new Map<string, string>();
|
|
122
|
-
private readonly controllers = new Map<string, AbortController>();
|
|
123
|
-
private readonly controllerCleanup = new Map<string, () => void>();
|
|
124
|
-
private queue: QueuedSpawn[] = [];
|
|
125
|
-
private runningBackground = 0;
|
|
126
|
-
private counter = 0;
|
|
127
|
-
private maxConcurrent: number;
|
|
128
|
-
private readonly onComplete?: Notify;
|
|
129
|
-
private readonly onEvent?: NotifyEvent;
|
|
130
|
-
private readonly pollIntervalMs: number;
|
|
131
|
-
|
|
132
|
-
constructor(maxConcurrent = 4, onComplete?: Notify, pollIntervalMs = 1000, onEvent?: NotifyEvent) {
|
|
133
|
-
this.maxConcurrent = maxConcurrent;
|
|
134
|
-
this.onComplete = onComplete;
|
|
135
|
-
this.onEvent = onEvent;
|
|
136
|
-
this.pollIntervalMs = pollIntervalMs;
|
|
137
|
-
}
|
|
138
|
-
|
|
139
|
-
spawn(options: SubagentSpawnOptions, runner: SpawnRunner, signal?: AbortSignal): SubagentRecord {
|
|
140
|
-
const record: SubagentRecord = {
|
|
141
|
-
id: `agent_${Date.now().toString(36)}_${(++this.counter).toString(36)}`,
|
|
142
|
-
type: options.type,
|
|
143
|
-
description: options.description,
|
|
144
|
-
prompt: options.prompt,
|
|
145
|
-
status: options.background && this.runningBackground >= this.maxConcurrent ? "queued" : "running",
|
|
146
|
-
startedAt: Date.now(),
|
|
147
|
-
model: options.model,
|
|
148
|
-
skill: options.skill,
|
|
149
|
-
background: options.background,
|
|
150
|
-
ownerSessionGeneration: options.ownerSessionGeneration,
|
|
151
|
-
};
|
|
152
|
-
this.records.set(record.id, record);
|
|
153
|
-
this.cwdByRecord.set(record.id, options.cwd);
|
|
154
|
-
savePersistedSubagentRecord(options.cwd, record);
|
|
155
|
-
if (record.status === "queued") {
|
|
156
|
-
this.queue.push({ record, options, runner, signal });
|
|
157
|
-
return record;
|
|
158
|
-
}
|
|
159
|
-
this.start(record, options, runner, signal);
|
|
160
|
-
return record;
|
|
161
|
-
}
|
|
162
|
-
|
|
163
|
-
getRecord(id: string): SubagentRecord | undefined {
|
|
164
|
-
return this.records.get(id);
|
|
165
|
-
}
|
|
166
|
-
|
|
167
|
-
listAgents(): SubagentRecord[] {
|
|
168
|
-
return [...this.records.values()].sort((a, b) => b.startedAt - a.startedAt);
|
|
169
|
-
}
|
|
170
|
-
|
|
171
|
-
abort(id: string): boolean {
|
|
172
|
-
const record = this.records.get(id);
|
|
173
|
-
if (!record) return false;
|
|
174
|
-
if (record.status === "queued") {
|
|
175
|
-
this.queue = this.queue.filter((entry) => entry.record.id !== id);
|
|
176
|
-
this.markStopped(record);
|
|
177
|
-
return true;
|
|
178
|
-
}
|
|
179
|
-
if (record.status !== "running" && record.status !== "blocked") return false;
|
|
180
|
-
this.controllers.get(id)?.abort();
|
|
181
|
-
this.markStopped(record);
|
|
182
|
-
return true;
|
|
183
|
-
}
|
|
184
|
-
|
|
185
|
-
abortAll(): number {
|
|
186
|
-
let count = 0;
|
|
187
|
-
|
|
188
|
-
|
|
189
|
-
|
|
190
|
-
|
|
191
|
-
|
|
192
|
-
|
|
193
|
-
|
|
194
|
-
|
|
195
|
-
this.
|
|
196
|
-
|
|
197
|
-
|
|
198
|
-
|
|
199
|
-
|
|
200
|
-
|
|
201
|
-
|
|
202
|
-
|
|
203
|
-
|
|
204
|
-
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
|
|
214
|
-
|
|
215
|
-
if (record
|
|
216
|
-
if (record.
|
|
217
|
-
|
|
218
|
-
|
|
219
|
-
|
|
220
|
-
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
this.
|
|
224
|
-
|
|
225
|
-
|
|
226
|
-
|
|
227
|
-
|
|
228
|
-
|
|
229
|
-
record.
|
|
230
|
-
record.
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
|
|
236
|
-
|
|
237
|
-
record.
|
|
238
|
-
record.
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
|
|
242
|
-
record.
|
|
243
|
-
|
|
244
|
-
|
|
245
|
-
|
|
246
|
-
|
|
247
|
-
|
|
248
|
-
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
|
|
252
|
-
|
|
253
|
-
|
|
254
|
-
|
|
255
|
-
|
|
256
|
-
|
|
257
|
-
|
|
258
|
-
|
|
259
|
-
|
|
260
|
-
if (
|
|
261
|
-
|
|
262
|
-
|
|
263
|
-
|
|
264
|
-
|
|
265
|
-
|
|
266
|
-
|
|
267
|
-
|
|
268
|
-
|
|
269
|
-
|
|
270
|
-
|
|
271
|
-
|
|
272
|
-
|
|
273
|
-
|
|
274
|
-
|
|
275
|
-
|
|
276
|
-
|
|
277
|
-
|
|
278
|
-
|
|
279
|
-
|
|
280
|
-
|
|
281
|
-
|
|
282
|
-
|
|
283
|
-
|
|
284
|
-
|
|
285
|
-
|
|
286
|
-
if (signal) {
|
|
287
|
-
|
|
288
|
-
signal
|
|
289
|
-
|
|
290
|
-
|
|
291
|
-
|
|
292
|
-
|
|
293
|
-
|
|
294
|
-
|
|
295
|
-
|
|
296
|
-
|
|
297
|
-
|
|
298
|
-
|
|
299
|
-
|
|
300
|
-
|
|
301
|
-
|
|
302
|
-
|
|
303
|
-
|
|
304
|
-
|
|
305
|
-
|
|
306
|
-
|
|
307
|
-
|
|
308
|
-
|
|
309
|
-
|
|
310
|
-
|
|
311
|
-
|
|
312
|
-
|
|
313
|
-
|
|
314
|
-
|
|
315
|
-
if (loaded
|
|
316
|
-
|
|
317
|
-
|
|
318
|
-
|
|
319
|
-
|
|
320
|
-
|
|
321
|
-
|
|
322
|
-
|
|
323
|
-
|
|
324
|
-
record
|
|
325
|
-
|
|
326
|
-
|
|
327
|
-
|
|
328
|
-
|
|
329
|
-
|
|
330
|
-
|
|
331
|
-
|
|
332
|
-
record
|
|
333
|
-
|
|
334
|
-
|
|
335
|
-
|
|
336
|
-
|
|
337
|
-
|
|
338
|
-
|
|
339
|
-
|
|
340
|
-
|
|
341
|
-
|
|
342
|
-
|
|
343
|
-
|
|
344
|
-
|
|
345
|
-
|
|
346
|
-
|
|
347
|
-
|
|
348
|
-
|
|
349
|
-
|
|
350
|
-
|
|
351
|
-
|
|
352
|
-
|
|
353
|
-
|
|
354
|
-
|
|
355
|
-
|
|
356
|
-
|
|
357
|
-
|
|
358
|
-
|
|
359
|
-
|
|
360
|
-
|
|
361
|
-
|
|
362
|
-
|
|
363
|
-
|
|
364
|
-
|
|
365
|
-
|
|
366
|
-
current.
|
|
367
|
-
|
|
368
|
-
|
|
369
|
-
|
|
370
|
-
|
|
371
|
-
|
|
372
|
-
|
|
373
|
-
|
|
374
|
-
|
|
375
|
-
|
|
376
|
-
|
|
377
|
-
|
|
378
|
-
|
|
379
|
-
|
|
380
|
-
|
|
381
|
-
|
|
382
|
-
|
|
383
|
-
|
|
384
|
-
|
|
385
|
-
|
|
386
|
-
|
|
387
|
-
|
|
388
|
-
|
|
389
|
-
|
|
390
|
-
|
|
391
|
-
|
|
392
|
-
|
|
393
|
-
|
|
394
|
-
|
|
395
|
-
|
|
396
|
-
}
|
|
397
|
-
|
|
398
|
-
|
|
399
|
-
|
|
400
|
-
}
|
|
1
|
+
import * as fs from "node:fs";
|
|
2
|
+
import * as path from "node:path";
|
|
3
|
+
import { loadRunManifestById } from "../state/state-store.ts";
|
|
4
|
+
import type { PiTeamsToolResult } from "../extension/tool-result.ts";
|
|
5
|
+
import { DEFAULT_SUBAGENT } from "../config/defaults.ts";
|
|
6
|
+
import { projectCrewRoot } from "../utils/paths.ts";
|
|
7
|
+
import { DEFAULT_PATHS } from "../config/defaults.ts";
|
|
8
|
+
import { logInternalError } from "../utils/internal-error.ts";
|
|
9
|
+
import { redactSecrets } from "../utils/redaction.ts";
|
|
10
|
+
|
|
11
|
+
export type SubagentStatus = "queued" | "running" | "completed" | "failed" | "cancelled" | "error" | "blocked" | "stopped";
|
|
12
|
+
|
|
13
|
+
export interface SubagentSpawnOptions {
|
|
14
|
+
cwd: string;
|
|
15
|
+
type: string;
|
|
16
|
+
description: string;
|
|
17
|
+
prompt: string;
|
|
18
|
+
background: boolean;
|
|
19
|
+
model?: string;
|
|
20
|
+
skill?: string | string[] | false;
|
|
21
|
+
maxTurns?: number;
|
|
22
|
+
ownerSessionGeneration?: number;
|
|
23
|
+
}
|
|
24
|
+
|
|
25
|
+
export interface SubagentRecord {
|
|
26
|
+
id: string;
|
|
27
|
+
runId?: string;
|
|
28
|
+
type: string;
|
|
29
|
+
description: string;
|
|
30
|
+
prompt: string;
|
|
31
|
+
status: SubagentStatus;
|
|
32
|
+
startedAt: number;
|
|
33
|
+
completedAt?: number;
|
|
34
|
+
result?: string;
|
|
35
|
+
error?: string;
|
|
36
|
+
resultConsumed?: boolean;
|
|
37
|
+
model?: string;
|
|
38
|
+
skill?: string | string[] | false;
|
|
39
|
+
background: boolean;
|
|
40
|
+
ownerSessionGeneration?: number;
|
|
41
|
+
stuckNotified?: boolean;
|
|
42
|
+
blockedAt?: number;
|
|
43
|
+
promise?: Promise<void>;
|
|
44
|
+
// Phase 1.6: Telemetry baseline fields
|
|
45
|
+
turnCount?: number;
|
|
46
|
+
terminated?: boolean;
|
|
47
|
+
durationMs?: number;
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
type SpawnRunner = (options: SubagentSpawnOptions, signal?: AbortSignal) => Promise<PiTeamsToolResult>;
|
|
51
|
+
type Notify = (record: SubagentRecord) => void;
|
|
52
|
+
type NotifyEvent = (type: string, data: Record<string, unknown>) => void;
|
|
53
|
+
|
|
54
|
+
interface QueuedSpawn {
|
|
55
|
+
record: SubagentRecord;
|
|
56
|
+
options: SubagentSpawnOptions;
|
|
57
|
+
runner: SpawnRunner;
|
|
58
|
+
signal?: AbortSignal;
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
function isValidSubagentId(id: string): boolean {
|
|
62
|
+
return /^[a-z0-9_]+$/i.test(id) && id.length <= 128;
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
function persistedSubagentPath(cwd: string, id: string): string {
|
|
66
|
+
if (!isValidSubagentId(id)) throw new Error(`Invalid subagent id: ${id}`);
|
|
67
|
+
return path.join(projectCrewRoot(cwd), DEFAULT_PATHS.state.subagentsSubdir, `${id}.json`);
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
function serializableRecord(record: SubagentRecord): SubagentRecord {
|
|
71
|
+
const { promise: _promise, ...rest } = record;
|
|
72
|
+
return rest;
|
|
73
|
+
}
|
|
74
|
+
|
|
75
|
+
export function savePersistedSubagentRecord(cwd: string, record: SubagentRecord): void {
|
|
76
|
+
try {
|
|
77
|
+
const filePath = persistedSubagentPath(cwd, record.id);
|
|
78
|
+
fs.mkdirSync(path.dirname(filePath), { recursive: true });
|
|
79
|
+
fs.writeFileSync(filePath, `${JSON.stringify(redactSecrets(serializableRecord(record)), null, 2)}\n`, "utf-8");
|
|
80
|
+
} catch (error) {
|
|
81
|
+
logInternalError("subagent-manager.save", error, `id=${record.id}`);
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
|
|
85
|
+
export function readPersistedSubagentRecord(cwd: string, id: string): SubagentRecord | undefined {
|
|
86
|
+
try {
|
|
87
|
+
const parsed = JSON.parse(fs.readFileSync(persistedSubagentPath(cwd, id), "utf-8"));
|
|
88
|
+
return parsed && typeof parsed === "object" && !Array.isArray(parsed) ? parsed as SubagentRecord : undefined;
|
|
89
|
+
} catch {
|
|
90
|
+
return undefined;
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
|
|
94
|
+
function resultText(result: PiTeamsToolResult): string {
|
|
95
|
+
return result.content?.map((item) => item.type === "text" ? item.text : "").filter(Boolean).join("\n") ?? "";
|
|
96
|
+
}
|
|
97
|
+
|
|
98
|
+
function detailsRunId(result: PiTeamsToolResult): string | undefined {
|
|
99
|
+
const details = result.details as { runId?: unknown } | undefined;
|
|
100
|
+
return typeof details?.runId === "string" ? details.runId : undefined;
|
|
101
|
+
}
|
|
102
|
+
|
|
103
|
+
function totalRunTurns(cwd: string, runId: string | undefined): number | undefined {
|
|
104
|
+
if (!runId) return undefined;
|
|
105
|
+
const loaded = loadRunManifestById(cwd, runId);
|
|
106
|
+
if (!loaded) return undefined;
|
|
107
|
+
let total = 0;
|
|
108
|
+
let hasTurns = false;
|
|
109
|
+
for (const task of loaded.tasks) {
|
|
110
|
+
const turns = task.usage?.turns ?? task.agentProgress?.turns;
|
|
111
|
+
if (typeof turns === "number" && Number.isFinite(turns)) {
|
|
112
|
+
total += turns;
|
|
113
|
+
hasTurns = true;
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
return hasTurns ? total : undefined;
|
|
117
|
+
}
|
|
118
|
+
|
|
119
|
+
export class SubagentManager {
|
|
120
|
+
private readonly records = new Map<string, SubagentRecord>();
|
|
121
|
+
private readonly cwdByRecord = new Map<string, string>();
|
|
122
|
+
private readonly controllers = new Map<string, AbortController>();
|
|
123
|
+
private readonly controllerCleanup = new Map<string, () => void>();
|
|
124
|
+
private queue: QueuedSpawn[] = [];
|
|
125
|
+
private runningBackground = 0;
|
|
126
|
+
private counter = 0;
|
|
127
|
+
private maxConcurrent: number;
|
|
128
|
+
private readonly onComplete?: Notify;
|
|
129
|
+
private readonly onEvent?: NotifyEvent;
|
|
130
|
+
private readonly pollIntervalMs: number;
|
|
131
|
+
|
|
132
|
+
constructor(maxConcurrent = 4, onComplete?: Notify, pollIntervalMs = 1000, onEvent?: NotifyEvent) {
|
|
133
|
+
this.maxConcurrent = maxConcurrent;
|
|
134
|
+
this.onComplete = onComplete;
|
|
135
|
+
this.onEvent = onEvent;
|
|
136
|
+
this.pollIntervalMs = pollIntervalMs;
|
|
137
|
+
}
|
|
138
|
+
|
|
139
|
+
spawn(options: SubagentSpawnOptions, runner: SpawnRunner, signal?: AbortSignal): SubagentRecord {
|
|
140
|
+
const record: SubagentRecord = {
|
|
141
|
+
id: `agent_${Date.now().toString(36)}_${(++this.counter).toString(36)}`,
|
|
142
|
+
type: options.type,
|
|
143
|
+
description: options.description,
|
|
144
|
+
prompt: options.prompt,
|
|
145
|
+
status: options.background && this.runningBackground >= this.maxConcurrent ? "queued" : "running",
|
|
146
|
+
startedAt: Date.now(),
|
|
147
|
+
model: options.model,
|
|
148
|
+
skill: options.skill,
|
|
149
|
+
background: options.background,
|
|
150
|
+
ownerSessionGeneration: options.ownerSessionGeneration,
|
|
151
|
+
};
|
|
152
|
+
this.records.set(record.id, record);
|
|
153
|
+
this.cwdByRecord.set(record.id, options.cwd);
|
|
154
|
+
savePersistedSubagentRecord(options.cwd, record);
|
|
155
|
+
if (record.status === "queued") {
|
|
156
|
+
this.queue.push({ record, options, runner, signal });
|
|
157
|
+
return record;
|
|
158
|
+
}
|
|
159
|
+
this.start(record, options, runner, signal);
|
|
160
|
+
return record;
|
|
161
|
+
}
|
|
162
|
+
|
|
163
|
+
getRecord(id: string): SubagentRecord | undefined {
|
|
164
|
+
return this.records.get(id);
|
|
165
|
+
}
|
|
166
|
+
|
|
167
|
+
listAgents(): SubagentRecord[] {
|
|
168
|
+
return [...this.records.values()].sort((a, b) => b.startedAt - a.startedAt);
|
|
169
|
+
}
|
|
170
|
+
|
|
171
|
+
abort(id: string, reason?: string): boolean {
|
|
172
|
+
const record = this.records.get(id);
|
|
173
|
+
if (!record) return false;
|
|
174
|
+
if (record.status === "queued") {
|
|
175
|
+
this.queue = this.queue.filter((entry) => entry.record.id !== id);
|
|
176
|
+
this.markStopped(record, reason ?? "Aborted by caller.");
|
|
177
|
+
return true;
|
|
178
|
+
}
|
|
179
|
+
if (record.status !== "running" && record.status !== "blocked") return false;
|
|
180
|
+
this.controllers.get(id)?.abort();
|
|
181
|
+
this.markStopped(record, reason ?? "Aborted by caller.");
|
|
182
|
+
return true;
|
|
183
|
+
}
|
|
184
|
+
|
|
185
|
+
abortAll(reason?: string): number {
|
|
186
|
+
let count = 0;
|
|
187
|
+
const stopReason = reason ?? "Aborted (session switch or shutdown).";
|
|
188
|
+
for (const entry of this.queue) {
|
|
189
|
+
this.markStopped(entry.record, stopReason);
|
|
190
|
+
count++;
|
|
191
|
+
}
|
|
192
|
+
this.queue = [];
|
|
193
|
+
for (const record of this.records.values()) {
|
|
194
|
+
if (record.status === "running" || record.status === "blocked") {
|
|
195
|
+
this.controllers.get(record.id)?.abort();
|
|
196
|
+
this.markStopped(record, stopReason);
|
|
197
|
+
count++;
|
|
198
|
+
}
|
|
199
|
+
}
|
|
200
|
+
return count;
|
|
201
|
+
}
|
|
202
|
+
|
|
203
|
+
async waitForAll(): Promise<void> {
|
|
204
|
+
while (true) {
|
|
205
|
+
this.drainQueue();
|
|
206
|
+
const pending = this.listAgents().filter((record) => record.status === "running" || record.status === "queued").map((record) => record.promise).filter((promise): promise is Promise<void> => Boolean(promise));
|
|
207
|
+
if (!pending.length) break;
|
|
208
|
+
await Promise.allSettled(pending);
|
|
209
|
+
}
|
|
210
|
+
}
|
|
211
|
+
|
|
212
|
+
async waitForRecord(id: string): Promise<SubagentRecord | undefined> {
|
|
213
|
+
while (true) {
|
|
214
|
+
const record = this.records.get(id);
|
|
215
|
+
if (!record) return undefined;
|
|
216
|
+
if (record.status !== "running" && record.status !== "queued") return record;
|
|
217
|
+
if (record.promise) await record.promise.catch(() => { /* status already set to error */ });
|
|
218
|
+
else await new Promise((resolve) => setTimeout(resolve, 100));
|
|
219
|
+
}
|
|
220
|
+
}
|
|
221
|
+
|
|
222
|
+
setMaxConcurrent(value: number): void {
|
|
223
|
+
this.maxConcurrent = Math.max(1, Math.floor(value));
|
|
224
|
+
this.drainQueue();
|
|
225
|
+
}
|
|
226
|
+
|
|
227
|
+
private start(record: SubagentRecord, options: SubagentSpawnOptions, runner: SpawnRunner, signal?: AbortSignal): void {
|
|
228
|
+
if (options.background) this.runningBackground++;
|
|
229
|
+
record.status = "running";
|
|
230
|
+
record.startedAt = Date.now();
|
|
231
|
+
record.completedAt = undefined;
|
|
232
|
+
const runSignal = this.createRunSignal(record.id, signal);
|
|
233
|
+
savePersistedSubagentRecord(options.cwd, record);
|
|
234
|
+
record.promise = (async () => {
|
|
235
|
+
try {
|
|
236
|
+
const result = await runner(options, runSignal);
|
|
237
|
+
if (record.status === "stopped") return;
|
|
238
|
+
record.runId = detailsRunId(result);
|
|
239
|
+
record.result = resultText(result);
|
|
240
|
+
savePersistedSubagentRecord(options.cwd, record);
|
|
241
|
+
if (result.isError) {
|
|
242
|
+
record.status = "error";
|
|
243
|
+
record.error = record.result;
|
|
244
|
+
throw new Error(record.error);
|
|
245
|
+
}
|
|
246
|
+
if (record.runId) await this.pollRunToTerminal(options.cwd, record);
|
|
247
|
+
else record.status = "completed";
|
|
248
|
+
} catch (error) {
|
|
249
|
+
if (record.status === "stopped" || runSignal.aborted) {
|
|
250
|
+
const abortReason = runSignal.aborted ? "Signal aborted — agent cancelled by parent (session switch, user cancel, or tool timeout)." : undefined;
|
|
251
|
+
record.status = "stopped";
|
|
252
|
+
if (!record.error) record.error = abortReason ?? (error instanceof Error ? error.message : String(error));
|
|
253
|
+
return;
|
|
254
|
+
}
|
|
255
|
+
record.status = "error";
|
|
256
|
+
record.error = error instanceof Error ? error.message : String(error);
|
|
257
|
+
throw error; // H4: Propagate rejection so callers awaiting record.promise see the error
|
|
258
|
+
} finally {
|
|
259
|
+
this.cleanupRunSignal(record.id);
|
|
260
|
+
if (options.background) this.runningBackground = Math.max(0, this.runningBackground - 1);
|
|
261
|
+
if (record.status !== "blocked") record.completedAt = record.completedAt ?? Date.now();
|
|
262
|
+
savePersistedSubagentRecord(options.cwd, record);
|
|
263
|
+
if (record.status === "completed" || record.status === "failed" || record.status === "cancelled" || record.status === "error" || record.status === "stopped") {
|
|
264
|
+
// Phase 1.6: Populate telemetry fields
|
|
265
|
+
record.turnCount = record.turnCount ?? totalRunTurns(options.cwd, record.runId);
|
|
266
|
+
record.durationMs = record.completedAt ? Math.max(0, record.completedAt - record.startedAt) : undefined;
|
|
267
|
+
savePersistedSubagentRecord(options.cwd, record);
|
|
268
|
+
this.onComplete?.(record);
|
|
269
|
+
}
|
|
270
|
+
this.drainQueue();
|
|
271
|
+
}
|
|
272
|
+
})();
|
|
273
|
+
}
|
|
274
|
+
|
|
275
|
+
private markStopped(record: SubagentRecord, reason?: string): void {
|
|
276
|
+
record.status = "stopped";
|
|
277
|
+
record.completedAt = Date.now();
|
|
278
|
+
if (reason && !record.error) record.error = reason;
|
|
279
|
+
const cwd = this.cwdByRecord.get(record.id);
|
|
280
|
+
if (cwd) savePersistedSubagentRecord(cwd, record);
|
|
281
|
+
}
|
|
282
|
+
|
|
283
|
+
private createRunSignal(id: string, signal?: AbortSignal): AbortSignal {
|
|
284
|
+
const controller = new AbortController();
|
|
285
|
+
this.controllers.set(id, controller);
|
|
286
|
+
if (signal?.aborted) {
|
|
287
|
+
controller.abort();
|
|
288
|
+
return controller.signal;
|
|
289
|
+
}
|
|
290
|
+
if (signal) {
|
|
291
|
+
const abort = (): void => controller.abort();
|
|
292
|
+
signal.addEventListener("abort", abort, { once: true });
|
|
293
|
+
this.controllerCleanup.set(id, () => signal.removeEventListener("abort", abort));
|
|
294
|
+
}
|
|
295
|
+
return controller.signal;
|
|
296
|
+
}
|
|
297
|
+
|
|
298
|
+
private cleanupRunSignal(id: string): void {
|
|
299
|
+
this.controllerCleanup.get(id)?.();
|
|
300
|
+
this.controllerCleanup.delete(id);
|
|
301
|
+
this.controllers.delete(id);
|
|
302
|
+
}
|
|
303
|
+
|
|
304
|
+
private drainQueue(): void {
|
|
305
|
+
while (this.queue.length > 0 && this.runningBackground < this.maxConcurrent) {
|
|
306
|
+
const next = this.queue.shift();
|
|
307
|
+
if (!next || next.record.status !== "queued") continue;
|
|
308
|
+
this.start(next.record, next.options, next.runner, next.signal);
|
|
309
|
+
}
|
|
310
|
+
}
|
|
311
|
+
|
|
312
|
+
private async pollRunToTerminal(cwd: string, record: SubagentRecord): Promise<void> {
|
|
313
|
+
while (record.runId && (record.status === "running" || record.status === "blocked")) {
|
|
314
|
+
const loaded = loadRunManifestById(cwd, record.runId);
|
|
315
|
+
if (!loaded) {
|
|
316
|
+
await new Promise((resolve) => setTimeout(resolve, this.pollIntervalMs));
|
|
317
|
+
continue;
|
|
318
|
+
}
|
|
319
|
+
if (loaded.manifest.status === "completed") {
|
|
320
|
+
record.status = "completed";
|
|
321
|
+
record.error = undefined;
|
|
322
|
+
record.turnCount = record.turnCount ?? totalRunTurns(cwd, record.runId);
|
|
323
|
+
record.completedAt = Date.now();
|
|
324
|
+
savePersistedSubagentRecord(cwd, record);
|
|
325
|
+
return;
|
|
326
|
+
}
|
|
327
|
+
if (loaded.manifest.status === "failed" || loaded.manifest.status === "cancelled") {
|
|
328
|
+
record.status = loaded.manifest.status;
|
|
329
|
+
record.error = loaded.manifest.summary;
|
|
330
|
+
record.turnCount = record.turnCount ?? totalRunTurns(cwd, record.runId);
|
|
331
|
+
record.completedAt = Date.now();
|
|
332
|
+
savePersistedSubagentRecord(cwd, record);
|
|
333
|
+
return;
|
|
334
|
+
}
|
|
335
|
+
if (loaded.manifest.status === "blocked") {
|
|
336
|
+
record.status = "blocked";
|
|
337
|
+
record.error = undefined;
|
|
338
|
+
if (!record.blockedAt) {
|
|
339
|
+
record.blockedAt = Date.now();
|
|
340
|
+
record.stuckNotified = false;
|
|
341
|
+
record.completedAt = undefined;
|
|
342
|
+
this.onComplete?.(record);
|
|
343
|
+
this.scheduleStuckBlockedNotify(cwd, record);
|
|
344
|
+
this.scheduleBlockedTerminalPoll(cwd, record);
|
|
345
|
+
}
|
|
346
|
+
savePersistedSubagentRecord(cwd, record);
|
|
347
|
+
return;
|
|
348
|
+
}
|
|
349
|
+
await new Promise((resolve) => setTimeout(resolve, this.pollIntervalMs));
|
|
350
|
+
}
|
|
351
|
+
}
|
|
352
|
+
|
|
353
|
+
private scheduleBlockedTerminalPoll(cwd: string, record: SubagentRecord): void {
|
|
354
|
+
const poll = (): void => {
|
|
355
|
+
const current = this.records.get(record.id);
|
|
356
|
+
if (!current || current.status !== "blocked" || !current.runId) return;
|
|
357
|
+
const loaded = loadRunManifestById(cwd, current.runId);
|
|
358
|
+
if (!loaded || loaded.manifest.status === "blocked" || loaded.manifest.status === "running" || loaded.manifest.status === "planning" || loaded.manifest.status === "queued") {
|
|
359
|
+
const timer = setTimeout(poll, this.pollIntervalMs);
|
|
360
|
+
timer.unref();
|
|
361
|
+
return;
|
|
362
|
+
}
|
|
363
|
+
const persisted = readPersistedSubagentRecord(cwd, current.id);
|
|
364
|
+
current.resultConsumed = current.resultConsumed || persisted?.resultConsumed;
|
|
365
|
+
if (loaded.manifest.status === "completed") {
|
|
366
|
+
current.status = "completed";
|
|
367
|
+
current.error = undefined;
|
|
368
|
+
} else if (loaded.manifest.status === "failed" || loaded.manifest.status === "cancelled") {
|
|
369
|
+
current.status = loaded.manifest.status;
|
|
370
|
+
current.error = loaded.manifest.summary;
|
|
371
|
+
} else return;
|
|
372
|
+
current.completedAt = Date.now();
|
|
373
|
+
current.turnCount = current.turnCount ?? totalRunTurns(cwd, current.runId);
|
|
374
|
+
current.durationMs = Math.max(0, current.completedAt - current.startedAt);
|
|
375
|
+
savePersistedSubagentRecord(cwd, current);
|
|
376
|
+
this.onComplete?.(current);
|
|
377
|
+
};
|
|
378
|
+
const timer = setTimeout(poll, this.pollIntervalMs);
|
|
379
|
+
timer.unref();
|
|
380
|
+
}
|
|
381
|
+
|
|
382
|
+
private scheduleStuckBlockedNotify(cwd: string, record: SubagentRecord): void {
|
|
383
|
+
const threshold = DEFAULT_SUBAGENT.stuckBlockedNotifyMs;
|
|
384
|
+
const fire = (): void => {
|
|
385
|
+
const current = this.records.get(record.id);
|
|
386
|
+
if (!current || current.status !== "blocked" || !current.blockedAt || current.stuckNotified) return;
|
|
387
|
+
current.stuckNotified = true;
|
|
388
|
+
this.onEvent?.("subagent.stuck-blocked", {
|
|
389
|
+
event: "subagent.stuck-blocked",
|
|
390
|
+
id: current.id,
|
|
391
|
+
runId: current.runId,
|
|
392
|
+
durationMs: Math.max(0, Date.now() - current.blockedAt),
|
|
393
|
+
ownerSessionGeneration: current.ownerSessionGeneration,
|
|
394
|
+
});
|
|
395
|
+
savePersistedSubagentRecord(cwd, current);
|
|
396
|
+
};
|
|
397
|
+
if (threshold <= 0) {
|
|
398
|
+
fire();
|
|
399
|
+
return;
|
|
400
|
+
}
|
|
401
|
+
const timer = setTimeout(fire, threshold);
|
|
402
|
+
timer.unref();
|
|
403
|
+
}
|
|
404
|
+
}
|