@pi-unipi/unipi 3.0.0-alpha.2 → 3.0.0-alpha.21
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/CHANGELOG.md +292 -0
- package/README.md +8 -11
- package/docs/prefix-cache-architecture.md +14 -0
- package/package.json +22 -24
- package/packages/ask-user/README.md +55 -107
- package/packages/ask-user/ask-ui.ts +306 -732
- package/packages/ask-user/clipboard.ts +50 -0
- package/packages/ask-user/config.ts +74 -78
- package/packages/ask-user/package.json +2 -2
- package/packages/ask-user/questions.ts +151 -0
- package/packages/ask-user/skills/ask-user/SKILL.md +59 -133
- package/packages/ask-user/tools.ts +198 -432
- package/packages/autocomplete/package.json +1 -1
- package/packages/autocomplete/src/constants.ts +38 -43
- package/packages/autocomplete/src/sorting.ts +8 -0
- package/packages/background-tasks/package.json +2 -2
- package/packages/background-tasks/src/config.ts +0 -73
- package/packages/background-tasks/src/index.ts +20 -16
- package/packages/background-tasks/src/registry.ts +1 -171
- package/packages/background-tasks/src/types.ts +0 -32
- package/packages/btw/README.md +13 -56
- package/packages/btw/extensions/btw.ts +403 -1796
- package/packages/btw/package.json +5 -2
- package/packages/btw/skills/btw/SKILL.md +7 -143
- package/packages/compactor/README.md +56 -287
- package/packages/compactor/package.json +5 -9
- package/packages/compactor/skills/compactor/SKILL.md +13 -20
- package/packages/compactor/src/card.ts +90 -0
- package/packages/compactor/src/commands/index.ts +73 -130
- package/packages/compactor/src/compaction/compact-args.ts +5 -3
- package/packages/compactor/src/compaction/filter-noise.ts +1 -1
- package/packages/compactor/src/compaction/hooks.ts +487 -311
- package/packages/compactor/src/compaction/source.ts +178 -0
- package/packages/compactor/src/compaction/summarize.ts +586 -52
- package/packages/compactor/src/config/manager.ts +130 -127
- package/packages/compactor/src/config/schema.ts +32 -48
- package/packages/compactor/src/index.ts +40 -431
- package/packages/compactor/src/info-screen.ts +44 -190
- package/packages/compactor/src/session/recall-blocks.ts +1 -1
- package/packages/compactor/src/stats.ts +76 -0
- package/packages/compactor/src/tools/context-budget.ts +17 -34
- package/packages/compactor/src/tools/ctx-doctor.ts +45 -36
- package/packages/compactor/src/tools/register.ts +29 -269
- package/packages/compactor/src/types.ts +47 -186
- package/packages/core/README.md +25 -0
- package/packages/core/child-agent.ts +1 -0
- package/packages/core/command-echo.ts +63 -0
- package/packages/core/compaction-context.ts +41 -0
- package/packages/core/constants.ts +8 -38
- package/packages/core/events.ts +0 -53
- package/packages/core/fusion-status.ts +4 -0
- package/packages/core/global.d.ts +10 -0
- package/packages/core/harness-messages.ts +392 -0
- package/packages/core/index.ts +14 -0
- package/packages/core/kanboard-status.ts +49 -0
- package/packages/core/long-horizon-owner-status.ts +46 -0
- package/packages/core/model-cache.ts +81 -7
- package/packages/core/package.json +6 -3
- package/packages/core/src/attach/detect.ts +163 -0
- package/packages/core/src/child-agent/child-args.ts +31 -0
- package/packages/core/src/child-agent/delivery.ts +41 -0
- package/packages/core/src/child-agent/index.ts +5 -0
- package/packages/core/src/child-agent/readers.ts +203 -0
- package/packages/core/src/child-agent/runtime.ts +584 -0
- package/packages/core/src/evidence.ts +78 -0
- package/packages/core/src/hints/browser.ts +278 -0
- package/packages/core/src/hints/crab-data.ts +19 -0
- package/packages/core/src/hints/index.ts +865 -0
- package/packages/core/src/hints/lines.ts +607 -0
- package/packages/core/src/hints/store.ts +120 -0
- package/packages/core/src/jev/chatter.ts +24 -0
- package/packages/core/src/jev/settings.ts +177 -14
- package/packages/core/src/package-colors.ts +2 -0
- package/packages/core/src/settings/catalog.ts +39 -15
- package/packages/core/src/settings/engine.ts +48 -2
- package/packages/core/src/settings/hub.ts +83 -20
- package/packages/core/src/settings/open.ts +49 -0
- package/packages/core/src/settings/schema.ts +27 -7
- package/packages/core/src/tui/hub-kit.ts +18 -5
- package/packages/core/src/tui/kit.ts +198 -0
- package/packages/core/src/tui/progress.ts +73 -0
- package/packages/core/src/turn/arbiter.ts +290 -0
- package/packages/footer/README.md +4 -6
- package/packages/footer/package.json +3 -3
- package/packages/footer/src/glance-editor.ts +19 -11
- package/packages/footer/src/help.ts +1 -1
- package/packages/footer/src/index.ts +36 -3
- package/packages/footer/src/presets.ts +3 -4
- package/packages/footer/src/rendering/icons.ts +0 -20
- package/packages/footer/src/segments/compactor.ts +52 -142
- package/packages/footer/src/segments/status-ext.ts +0 -1
- package/packages/fusion/README.md +14 -4
- package/packages/fusion/package.json +4 -3
- package/packages/fusion/src/child-args.ts +1 -0
- package/packages/fusion/src/index.ts +161 -8
- package/packages/fusion/src/pi-settings.ts +57 -0
- package/packages/fusion/src/picker.ts +186 -51
- package/packages/fusion/src/preset.ts +56 -2
- package/packages/fusion/src/prompts.ts +1 -1
- package/packages/fusion/src/sidekick-runtime.ts +1 -420
- package/packages/fusion/src/tools.ts +107 -128
- package/packages/fusion/src/transcript.ts +27 -85
- package/packages/info-screen/package.json +2 -2
- package/packages/input-shortcuts/README.md +18 -30
- package/packages/input-shortcuts/package.json +2 -2
- package/packages/input-shortcuts/src/burst.ts +88 -0
- package/packages/input-shortcuts/src/chord-overlay.ts +20 -70
- package/packages/input-shortcuts/src/cleared-input.ts +97 -0
- package/packages/input-shortcuts/src/index.ts +99 -193
- package/packages/input-shortcuts/src/last-response.ts +40 -0
- package/packages/input-shortcuts/src/registers.ts +8 -23
- package/packages/input-shortcuts/src/types.ts +4 -16
- package/packages/input-shortcuts/src/undo-redo.ts +59 -47
- package/packages/kanboard/README.md +51 -47
- package/packages/kanboard/index.ts +293 -55
- package/packages/kanboard/package.json +8 -8
- package/packages/kanboard/skills/kanboard/SKILL.md +61 -27
- package/packages/kanboard/src/badges.ts +93 -0
- package/packages/kanboard/src/bin.ts +4 -2
- package/packages/kanboard/src/commands.ts +218 -84
- package/packages/kanboard/src/debug.ts +20 -0
- package/packages/kanboard/src/guard.ts +267 -109
- package/packages/kanboard/src/monitor.ts +330 -0
- package/packages/kanboard/src/notice-buffer.ts +74 -0
- package/packages/kanboard/src/progress.ts +27 -0
- package/packages/kanboard/src/reminders.ts +234 -0
- package/packages/kanboard/src/settings.ts +69 -15
- package/packages/kanboard/src/shapes.ts +5 -0
- package/packages/long-horizon/index.ts +190 -13
- package/packages/long-horizon/package.json +3 -3
- package/packages/long-horizon/src/commands.ts +103 -7
- package/packages/long-horizon/src/compaction-brief.ts +35 -0
- package/packages/long-horizon/src/engine/continuation.ts +84 -15
- package/packages/long-horizon/src/engine/goal-state.ts +64 -13
- package/packages/long-horizon/src/engine/nudge-stash.ts +71 -0
- package/packages/long-horizon/src/engine/ralph.ts +3 -2
- package/packages/long-horizon/src/engine/runaway.ts +68 -11
- package/packages/long-horizon/src/engine/verifier.ts +19 -0
- package/packages/long-horizon/src/gate.ts +37 -12
- package/packages/long-horizon/src/judge/resolve.ts +1 -0
- package/packages/long-horizon/src/owner.ts +35 -2
- package/packages/long-horizon/src/progress.ts +87 -0
- package/packages/long-horizon/src/prompts/goal.ts +1 -1
- package/packages/long-horizon/src/runtime.ts +172 -18
- package/packages/long-horizon/src/settings.ts +54 -39
- package/packages/long-horizon/src/tools/goal.ts +6 -0
- package/packages/long-horizon/src/tools/graph.ts +1 -1
- package/packages/long-horizon/src/tools/swarm.ts +3 -3
- package/packages/mcp/package.json +2 -2
- package/packages/memory/commands.ts +7 -2
- package/packages/memory/index.ts +210 -32
- package/packages/memory/package.json +3 -3
- package/packages/memory/save-session.ts +264 -0
- package/packages/memory/settings.ts +17 -0
- package/packages/memory/tools.ts +253 -203
- package/packages/notify/package.json +2 -2
- package/packages/skill-registry/README.md +53 -16
- package/packages/skill-registry/index.ts +230 -0
- package/packages/skill-registry/package.json +19 -8
- package/packages/skill-registry/src/editor.ts +206 -0
- package/packages/skill-registry/src/judge.ts +289 -0
- package/packages/skill-registry/src/registry.ts +87 -0
- package/packages/skill-registry/src/settings.ts +259 -0
- package/packages/skill-registry/src/vault.ts +80 -0
- package/packages/subagents/README.md +39 -119
- package/packages/subagents/package.json +6 -13
- package/packages/subagents/skills/subagents/SKILL.md +18 -42
- package/packages/subagents/src/agents.ts +213 -0
- package/packages/subagents/src/cards.ts +182 -0
- package/packages/subagents/src/index.ts +609 -1084
- package/packages/subagents/src/manager.ts +429 -0
- package/packages/subagents/src/profiles.ts +130 -0
- package/packages/subagents/src/transcript.ts +154 -0
- package/packages/subagents/src/ui.ts +389 -0
- package/packages/unipi/bundled.js +39756 -45569
- package/packages/unipi/index.ts +55 -10
- package/packages/updater/package.json +2 -2
- package/packages/utility/README.md +76 -69
- package/packages/utility/package.json +2 -2
- package/packages/utility/skills/image/SKILL.md +37 -0
- package/packages/utility/skills/utility/SKILL.md +17 -60
- package/packages/utility/src/answer/extract.ts +128 -0
- package/packages/utility/src/answer/index.ts +213 -0
- package/packages/utility/src/answer/reply.ts +133 -0
- package/packages/utility/src/answer/web.ts +167 -0
- package/packages/utility/src/attach/index.ts +205 -0
- package/packages/utility/src/commands.ts +38 -142
- package/packages/utility/src/diagnostics/engine.ts +50 -5
- package/packages/utility/src/image/index.ts +33 -0
- package/packages/utility/src/image/models.ts +221 -0
- package/packages/{image/src → utility/src/image}/recognize.ts +2 -2
- package/packages/utility/src/image/settings.ts +178 -0
- package/packages/{image/src/image-source.ts → utility/src/image/source.ts} +1 -1
- package/packages/utility/src/image/tools.ts +168 -0
- package/packages/utility/src/image/transports.ts +186 -0
- package/packages/utility/src/index.ts +68 -389
- package/packages/utility/src/lifecycle/cleanup.ts +116 -311
- package/packages/utility/src/rename/gate.ts +102 -0
- package/packages/utility/src/rename/index.ts +191 -0
- package/packages/utility/src/rename/session.ts +170 -0
- package/packages/utility/src/render/delegated.ts +285 -0
- package/packages/utility/src/render/harness.ts +683 -0
- package/packages/utility/src/render/parse.ts +156 -0
- package/packages/utility/src/render/pi-settings.ts +38 -0
- package/packages/utility/src/render/reply-bg.ts +307 -0
- package/packages/utility/src/render/simple.ts +724 -0
- package/packages/utility/src/render/spacing.ts +235 -0
- package/packages/utility/src/render/styled.ts +251 -0
- package/packages/utility/src/render/tools.ts +353 -0
- package/packages/utility/src/settings.ts +166 -233
- package/packages/utility/src/types.ts +1 -222
- package/packages/watchdog/index.ts +17 -6
- package/packages/watchdog/package.json +3 -3
- package/packages/watchdog/src/config.ts +3 -2
- package/packages/web-api/package.json +2 -2
- package/packages/workflow/README.md +3 -0
- package/packages/workflow/index.ts +0 -2
- package/packages/workflow/package.json +2 -2
- package/packages/workflow/src/permission/decide.ts +6 -4
- package/packages/workflow/src/permission/index.ts +17 -7
- package/packages/workflow/src/permission/settings.ts +39 -4
- package/packages/workflow/src/plan/index.ts +19 -4
- package/packages/background-tasks/extensions/delegate-child.ts +0 -1
- package/packages/background-tasks/src/context-parent-snapshot.ts +0 -142
- package/packages/background-tasks/src/context-token-budget.ts +0 -903
- package/packages/background-tasks/src/context-visible-conversation-v2.ts +0 -551
- package/packages/background-tasks/src/delegate/artifacts.ts +0 -487
- package/packages/background-tasks/src/delegate/budget.ts +0 -415
- package/packages/background-tasks/src/delegate/hook-contract-evidence.json +0 -18
- package/packages/background-tasks/src/delegate/hook-contract.ts +0 -154
- package/packages/background-tasks/src/delegate/launch.ts +0 -484
- package/packages/background-tasks/src/delegate/result-package.ts +0 -459
- package/packages/background-tasks/src/delegate/runner.ts +0 -430
- package/packages/background-tasks/src/delegate/seed.ts +0 -423
- package/packages/background-tasks/src/delegate/types.ts +0 -323
- package/packages/background-tasks/src/delegate-child-extension.ts +0 -978
- package/packages/background-tasks/src/delegate-extension.ts +0 -654
- package/packages/background-tasks/src/fixtures/delegate-context-incident.json +0 -17
- package/packages/compactor/skills/compactor-detail/SKILL.md +0 -116
- package/packages/compactor/skills/compactor-doctor/SKILL.md +0 -71
- package/packages/compactor/skills/compactor-stats/SKILL.md +0 -51
- package/packages/compactor/src/compaction/build-sections.ts +0 -76
- package/packages/compactor/src/compaction/extract/goals.ts +0 -36
- package/packages/compactor/src/compaction/extract/preferences.ts +0 -40
- package/packages/compactor/src/compaction/format.ts +0 -53
- package/packages/compactor/src/compaction/merge.ts +0 -143
- package/packages/compactor/src/config/presets.ts +0 -160
- package/packages/compactor/src/executor/executor.ts +0 -233
- package/packages/compactor/src/executor/runtime.ts +0 -115
- package/packages/compactor/src/security/evaluator.ts +0 -89
- package/packages/compactor/src/security/policy.ts +0 -96
- package/packages/compactor/src/security/scanner.ts +0 -65
- package/packages/compactor/src/session/auto-inject.ts +0 -60
- package/packages/compactor/src/session/db.ts +0 -325
- package/packages/compactor/src/session/extract.ts +0 -144
- package/packages/compactor/src/session/resume-inject.ts +0 -64
- package/packages/compactor/src/session/snapshot.ts +0 -327
- package/packages/compactor/src/tools/ctx-batch-execute.ts +0 -41
- package/packages/compactor/src/tools/ctx-execute-file.ts +0 -23
- package/packages/compactor/src/tools/ctx-execute.ts +0 -23
- package/packages/compactor/src/tools/ctx-stats.ts +0 -96
- package/packages/image/README.md +0 -99
- package/packages/image/package.json +0 -61
- package/packages/image/skills/image/SKILL.md +0 -76
- package/packages/image/src/generate.ts +0 -256
- package/packages/image/src/index.ts +0 -135
- package/packages/image/src/models.ts +0 -519
- package/packages/image/src/openai-images-api.ts +0 -282
- package/packages/image/src/register-providers.ts +0 -209
- package/packages/image/src/settings.ts +0 -173
- package/packages/image/src/tools.ts +0 -326
- package/packages/input-shortcuts/src/clipboard.ts +0 -119
- package/packages/kanboard/src/runner.ts +0 -817
- package/packages/skill-registry/skills/auto/SKILL.md +0 -465
- package/packages/skill-registry/skills/brainstorm/SKILL.md +0 -219
- package/packages/skill-registry/skills/chore-create/SKILL.md +0 -315
- package/packages/skill-registry/skills/chore-execute/SKILL.md +0 -314
- package/packages/skill-registry/skills/consolidate/SKILL.md +0 -142
- package/packages/skill-registry/skills/consultant/SKILL.md +0 -97
- package/packages/skill-registry/skills/debug/SKILL.md +0 -228
- package/packages/skill-registry/skills/document/SKILL.md +0 -120
- package/packages/skill-registry/skills/fix/SKILL.md +0 -212
- package/packages/skill-registry/skills/gather-context/SKILL.md +0 -145
- package/packages/skill-registry/skills/plan/SKILL.md +0 -231
- package/packages/skill-registry/skills/quick-fix/SKILL.md +0 -135
- package/packages/skill-registry/skills/quick-work/SKILL.md +0 -110
- package/packages/skill-registry/skills/research/SKILL.md +0 -251
- package/packages/skill-registry/skills/review-work/SKILL.md +0 -187
- package/packages/skill-registry/skills/scan-issues/SKILL.md +0 -195
- package/packages/skill-registry/skills/work/SKILL.md +0 -218
- package/packages/skill-registry/skills/worktree-create/SKILL.md +0 -69
- package/packages/skill-registry/skills/worktree-list/SKILL.md +0 -78
- package/packages/skill-registry/skills/worktree-merge/SKILL.md +0 -89
- package/packages/subagents/agents/delegate.md +0 -14
- package/packages/subagents/agents/oracle.md +0 -78
- package/packages/subagents/agents/researcher.md +0 -52
- package/packages/subagents/agents/reviewer.md +0 -80
- package/packages/subagents/agents/scout.md +0 -50
- package/packages/subagents/agents/worker.md +0 -59
- package/packages/subagents/prompts/council.md +0 -48
- package/packages/subagents/prompts/gather-context-and-clarify.md +0 -13
- package/packages/subagents/prompts/parallel-cleanup.md +0 -59
- package/packages/subagents/prompts/parallel-research.md +0 -50
- package/packages/subagents/prompts/parallel-review.md +0 -54
- package/packages/subagents/prompts/review-loop.md +0 -43
- package/packages/subagents/skills/council-mode/SKILL.md +0 -230
- package/packages/subagents/skills/subagents/references/constraints-and-recipes.md +0 -259
- package/packages/subagents/skills/subagents/references/execution-controls.md +0 -454
- package/packages/subagents/skills/subagents/references/management-authoring-rpc.md +0 -161
- package/packages/subagents/skills/subagents/references/multi-lane-orchestration.md +0 -39
- package/packages/subagents/skills/subagents/references/prompting-and-roles.md +0 -267
- package/packages/subagents/src/acceptance.ts +0 -331
- package/packages/subagents/src/agent-manager.ts +0 -405
- package/packages/subagents/src/agent-memory.ts +0 -300
- package/packages/subagents/src/agent-overrides.ts +0 -183
- package/packages/subagents/src/agent-runner.ts +0 -271
- package/packages/subagents/src/async-runner.ts +0 -422
- package/packages/subagents/src/authority-policy.ts +0 -47
- package/packages/subagents/src/budgets.ts +0 -285
- package/packages/subagents/src/child-safety.ts +0 -140
- package/packages/subagents/src/config.ts +0 -249
- package/packages/subagents/src/conversation-viewer.ts +0 -281
- package/packages/subagents/src/core-compat.ts +0 -94
- package/packages/subagents/src/custom-agents.ts +0 -343
- package/packages/subagents/src/file-system-retry.ts +0 -50
- package/packages/subagents/src/fleet-data.ts +0 -105
- package/packages/subagents/src/fleet-view.ts +0 -322
- package/packages/subagents/src/foreground-detach.ts +0 -59
- package/packages/subagents/src/fork-context.ts +0 -285
- package/packages/subagents/src/global.d.ts +0 -7
- package/packages/subagents/src/guide.ts +0 -129
- package/packages/subagents/src/mission-state.ts +0 -133
- package/packages/subagents/src/mission-store.ts +0 -436
- package/packages/subagents/src/output-limits.ts +0 -142
- package/packages/subagents/src/parity-types.ts +0 -315
- package/packages/subagents/src/pi-args.ts +0 -167
- package/packages/subagents/src/result-files.ts +0 -232
- package/packages/subagents/src/result-watcher.ts +0 -221
- package/packages/subagents/src/retained-children.ts +0 -176
- package/packages/subagents/src/run-fanout-budget.ts +0 -356
- package/packages/subagents/src/scheduled-runs.ts +0 -349
- package/packages/subagents/src/schemas.ts +0 -263
- package/packages/subagents/src/slash-commands.ts +0 -80
- package/packages/subagents/src/supervisor-channel.ts +0 -249
- package/packages/subagents/src/tool-handler.ts +0 -1144
- package/packages/subagents/src/types.ts +0 -238
- package/packages/subagents/src/widget.ts +0 -457
- package/packages/subagents/src/workflow-script.ts +0 -737
- package/packages/subagents/src/workflow-worker.ts +0 -384
- package/packages/subagents/src/worktree.ts +0 -614
- package/packages/utility/src/analytics/collector.ts +0 -293
- package/packages/utility/src/info-screen.ts +0 -80
- package/packages/utility/src/lifecycle/process.ts +0 -162
- package/packages/utility/src/prefix-cache.ts +0 -263
- package/packages/utility/src/skill-discovery.ts +0 -542
- package/packages/utility/src/tools/env.ts +0 -113
- package/packages/utility/src/tui/name-badge-state.ts +0 -315
- package/packages/utility/src/tui/name-badge.ts +0 -109
- /package/packages/{subagents/src → core/src/child-agent}/pi-spawn.ts +0 -0
|
@@ -0,0 +1,584 @@
|
|
|
1
|
+
import { spawn as defaultSpawn, type ChildProcess } from "node:child_process";
|
|
2
|
+
import { mkdirSync, unlinkSync, writeFileSync } from "node:fs";
|
|
3
|
+
import { tmpdir } from "node:os";
|
|
4
|
+
import { dirname, join } from "node:path";
|
|
5
|
+
import { randomUUID } from "node:crypto";
|
|
6
|
+
import { getPiSpawnCommand } from "./pi-spawn.js";
|
|
7
|
+
export type { EffortLevel, ModelKey } from "@pi-unipi/fusion/src/preset.js";
|
|
8
|
+
import type { EffortLevel, ModelKey } from "@pi-unipi/fusion/src/preset.js";
|
|
9
|
+
function primaryArg(name: string, args: Record<string, unknown> | undefined): string {
|
|
10
|
+
if (!args) return "";
|
|
11
|
+
const truncate = (v: string, max: number) => (v.length > max ? `${v.slice(0, Math.max(0, max - 1))}…` : v);
|
|
12
|
+
const firstLine = (v: string) => v.split("\n", 1)[0] ?? "";
|
|
13
|
+
const value = name === "bash"
|
|
14
|
+
? args.command
|
|
15
|
+
: name === "read" || name === "edit" || name === "write"
|
|
16
|
+
? args.path ?? args.file_path ?? args.filePath
|
|
17
|
+
: Object.values(args).find((entry) => typeof entry === "string");
|
|
18
|
+
return typeof value === "string" ? truncate(firstLine(value), 100) : "";
|
|
19
|
+
}
|
|
20
|
+
import { leadExtensionArgs } from "./child-args.js";
|
|
21
|
+
|
|
22
|
+
/** A completed sidekick step, streamed to the lead chat as a UI-only
|
|
23
|
+
* `sidekick-step` custom entry (never model-facing). */
|
|
24
|
+
export type SidekickStep =
|
|
25
|
+
| { kind: "tool"; name: string; arg: string; output: string; isError: boolean; durationMs: number; args?: Record<string, unknown> }
|
|
26
|
+
| { kind: "text"; text: string; thinking?: string };
|
|
27
|
+
|
|
28
|
+
/** The subset of ExtensionUIContext the child's prompts get forwarded to. */
|
|
29
|
+
export interface SidekickUIForwarder {
|
|
30
|
+
select?: (title: string, options: unknown[], opts?: unknown) => Promise<unknown>;
|
|
31
|
+
confirm?: (title: string, message: string, opts?: unknown) => Promise<unknown>;
|
|
32
|
+
input?: (title: string, placeholder?: string, opts?: unknown) => Promise<unknown>;
|
|
33
|
+
}
|
|
34
|
+
|
|
35
|
+
export interface SidekickSpawnConfig {
|
|
36
|
+
cwd: string;
|
|
37
|
+
model: ModelKey;
|
|
38
|
+
thinking: EffortLevel;
|
|
39
|
+
sessionFile: string;
|
|
40
|
+
systemPrompt: string;
|
|
41
|
+
spawn?: typeof defaultSpawn;
|
|
42
|
+
command?: { command: string; args: string[] };
|
|
43
|
+
onProgress?: () => void;
|
|
44
|
+
/** Fired once per completed sidekick step (tool end, or a text segment that
|
|
45
|
+
* is not the final report). The final report text is never emitted — the
|
|
46
|
+
* lead reports it itself. */
|
|
47
|
+
onStep?: (step: SidekickStep) => void;
|
|
48
|
+
settleGraceMs?: number;
|
|
49
|
+
}
|
|
50
|
+
|
|
51
|
+
export interface SidekickUsage {
|
|
52
|
+
input: number;
|
|
53
|
+
output: number;
|
|
54
|
+
cacheRead: number;
|
|
55
|
+
cacheWrite: number;
|
|
56
|
+
cost: number;
|
|
57
|
+
}
|
|
58
|
+
|
|
59
|
+
export type SidekickEvent =
|
|
60
|
+
| { kind: "text"; text: string; open: boolean }
|
|
61
|
+
| { kind: "tool"; toolCallId: string; name: string; args: Record<string, unknown> | undefined; output: string; isError: boolean; done: boolean; startedAt: number; endedAt?: number };
|
|
62
|
+
|
|
63
|
+
/** The permission gate's "deny with a reason" option (workflow prompt.ts). */
|
|
64
|
+
export const DENY_WITH_NOTE = "Deny with note…";
|
|
65
|
+
export const BACKGROUND_DENY_NOTE =
|
|
66
|
+
"Permission denied for this tool: this agent is running in the background, where tools that would require approval are automatically denied. Do not retry variants of this call; finish what you can without it and say in your report what needs approval.";
|
|
67
|
+
|
|
68
|
+
export const MAX_EVENTS = 300;
|
|
69
|
+
export const MAX_TOOL_OUTPUT = 4000;
|
|
70
|
+
|
|
71
|
+
export interface HandoffProgress {
|
|
72
|
+
toolCalls: number;
|
|
73
|
+
recentTools: string[];
|
|
74
|
+
textTail: string;
|
|
75
|
+
startedAt: number;
|
|
76
|
+
events: SidekickEvent[];
|
|
77
|
+
droppedEvents: number;
|
|
78
|
+
}
|
|
79
|
+
|
|
80
|
+
export interface HandoffReport {
|
|
81
|
+
id: string;
|
|
82
|
+
status: "completed" | "aborted" | "error" | "interrupted";
|
|
83
|
+
text: string;
|
|
84
|
+
usage: SidekickUsage;
|
|
85
|
+
toolCalls: number;
|
|
86
|
+
durationMs: number;
|
|
87
|
+
events: SidekickEvent[];
|
|
88
|
+
error?: string;
|
|
89
|
+
}
|
|
90
|
+
|
|
91
|
+
interface PendingHandoff {
|
|
92
|
+
id: string;
|
|
93
|
+
message: string;
|
|
94
|
+
startedAt: number;
|
|
95
|
+
usage: SidekickUsage;
|
|
96
|
+
progress: HandoffProgress;
|
|
97
|
+
retriedPrompt: boolean;
|
|
98
|
+
openBgTasks: number;
|
|
99
|
+
settled: boolean;
|
|
100
|
+
settleTimer?: NodeJS.Timeout;
|
|
101
|
+
resolve: (report: HandoffReport) => void;
|
|
102
|
+
}
|
|
103
|
+
|
|
104
|
+
const emptyUsage = (): SidekickUsage => ({ input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0 });
|
|
105
|
+
|
|
106
|
+
export interface ChildAgentConfig extends SidekickSpawnConfig {
|
|
107
|
+
/** Appended to the spawn args (e.g. ["--tools","a,b"]). */
|
|
108
|
+
extraArgs?: string[];
|
|
109
|
+
/** Extra env vars for the child. */
|
|
110
|
+
extraEnv?: Record<string, string>;
|
|
111
|
+
/** Prefix for forwarded approval-prompt titles (default "Sidekick: "). */
|
|
112
|
+
promptPrefix?: string;
|
|
113
|
+
}
|
|
114
|
+
|
|
115
|
+
export class ChildAgentRuntime {
|
|
116
|
+
private readonly cfg: ChildAgentConfig;
|
|
117
|
+
private child: ChildProcess | undefined;
|
|
118
|
+
private promptPath: string | undefined;
|
|
119
|
+
private pending: PendingHandoff | undefined;
|
|
120
|
+
private latestHandoff: { id: string; done: Promise<HandoffReport>; report?: HandoffReport } | undefined;
|
|
121
|
+
private responseText: ((text: string) => void) | undefined;
|
|
122
|
+
private responseError: ((error: Error) => void) | undefined;
|
|
123
|
+
private stderrTail = "";
|
|
124
|
+
private inputBuffer = "";
|
|
125
|
+
/** Buffered latest text segment — emitted only if it is NOT the terminal
|
|
126
|
+
* report (the lead relays that itself). */
|
|
127
|
+
private bufferedText: { text: string; thinking: string } | undefined;
|
|
128
|
+
/** Attached lead UI for forwarding child approval prompts (blocking waits
|
|
129
|
+
* only); undefined means background → prompts are refused. */
|
|
130
|
+
private forwardedUi: SidekickUIForwarder | undefined;
|
|
131
|
+
|
|
132
|
+
/** Attach the lead's UI for child prompt forwarding (blocking waits). */
|
|
133
|
+
attachUi(ui: SidekickUIForwarder): void {
|
|
134
|
+
this.forwardedUi = ui;
|
|
135
|
+
}
|
|
136
|
+
|
|
137
|
+
detachUi(): void {
|
|
138
|
+
this.forwardedUi = undefined;
|
|
139
|
+
}
|
|
140
|
+
|
|
141
|
+
private emitStep(step: SidekickStep): void {
|
|
142
|
+
try {
|
|
143
|
+
this.cfg.onStep?.(step);
|
|
144
|
+
} catch {
|
|
145
|
+
// Rendering must not affect the handoff.
|
|
146
|
+
}
|
|
147
|
+
}
|
|
148
|
+
|
|
149
|
+
/** Flush the buffered text segment as a step (it proved not to be the final
|
|
150
|
+
* report — more work follows). */
|
|
151
|
+
private flushTextStep(): void {
|
|
152
|
+
const buffered = this.bufferedText;
|
|
153
|
+
if (buffered === undefined || buffered.text.trim().length === 0) {
|
|
154
|
+
this.bufferedText = undefined;
|
|
155
|
+
return;
|
|
156
|
+
}
|
|
157
|
+
this.emitStep({ kind: "text", text: buffered.text, thinking: buffered.thinking || undefined });
|
|
158
|
+
this.bufferedText = undefined;
|
|
159
|
+
}
|
|
160
|
+
/** Set after answering a background approval prompt with "Deny with note…":
|
|
161
|
+
* the follow-up note input gets BACKGROUND_DENY_NOTE. */
|
|
162
|
+
private denyNotePending = false;
|
|
163
|
+
|
|
164
|
+
/**
|
|
165
|
+
* No lead waiter → refuse. When the prompt is the permission gate's (it
|
|
166
|
+
* offers "Deny with note…"), deny WITH a reason so the child stops retrying
|
|
167
|
+
* variants; anything else is cancelled (which also denies).
|
|
168
|
+
*/
|
|
169
|
+
private refuseInBackground(method: string, id: unknown, message: Record<string, unknown>): void {
|
|
170
|
+
let reply: Record<string, unknown> = { type: "extension_ui_response", id, cancelled: true };
|
|
171
|
+
const options = Array.isArray(message.options) ? message.options : [];
|
|
172
|
+
if (method === "select" && options.includes(DENY_WITH_NOTE)) {
|
|
173
|
+
this.denyNotePending = true;
|
|
174
|
+
reply = { type: "extension_ui_response", id, value: DENY_WITH_NOTE };
|
|
175
|
+
} else if (method === "input" && this.denyNotePending) {
|
|
176
|
+
this.denyNotePending = false;
|
|
177
|
+
reply = { type: "extension_ui_response", id, value: BACKGROUND_DENY_NOTE };
|
|
178
|
+
}
|
|
179
|
+
try {
|
|
180
|
+
this.send(reply);
|
|
181
|
+
} catch { /* child gone */ }
|
|
182
|
+
}
|
|
183
|
+
private abortRequested = false;
|
|
184
|
+
private pendingError: string | undefined;
|
|
185
|
+
readonly reports = new Map<string, HandoffReport>();
|
|
186
|
+
readonly usage = emptyUsage();
|
|
187
|
+
|
|
188
|
+
constructor(cfg: ChildAgentConfig) {
|
|
189
|
+
this.cfg = cfg;
|
|
190
|
+
}
|
|
191
|
+
|
|
192
|
+
isAlive(): boolean {
|
|
193
|
+
return this.child !== undefined && (this.child.exitCode === null || this.child.exitCode === undefined) && !this.child.killed;
|
|
194
|
+
}
|
|
195
|
+
|
|
196
|
+
isBusy(): boolean {
|
|
197
|
+
return this.pending !== undefined;
|
|
198
|
+
}
|
|
199
|
+
|
|
200
|
+
totalToolCalls(): number {
|
|
201
|
+
let total = this.pending?.progress.toolCalls ?? 0;
|
|
202
|
+
for (const report of this.reports.values()) total += report.toolCalls;
|
|
203
|
+
return total;
|
|
204
|
+
}
|
|
205
|
+
|
|
206
|
+
private notifyProgress(): void {
|
|
207
|
+
try {
|
|
208
|
+
this.cfg.onProgress?.();
|
|
209
|
+
} catch {
|
|
210
|
+
// Progress updates must not affect the handoff.
|
|
211
|
+
}
|
|
212
|
+
}
|
|
213
|
+
|
|
214
|
+
private appendEvent(event: SidekickEvent): void {
|
|
215
|
+
const progress = this.pending?.progress;
|
|
216
|
+
if (!progress) return;
|
|
217
|
+
progress.events.push(event);
|
|
218
|
+
if (progress.events.length > MAX_EVENTS) {
|
|
219
|
+
progress.events.shift();
|
|
220
|
+
progress.droppedEvents += 1;
|
|
221
|
+
}
|
|
222
|
+
}
|
|
223
|
+
|
|
224
|
+
private closeOpenText(): void {
|
|
225
|
+
const events = this.pending?.progress.events;
|
|
226
|
+
const last = events?.at(-1);
|
|
227
|
+
if (last?.kind === "text" && last.open) last.open = false;
|
|
228
|
+
}
|
|
229
|
+
|
|
230
|
+
private toolOutput(result: unknown): string {
|
|
231
|
+
if (typeof result === "object" && result !== null && Array.isArray((result as { content?: unknown }).content)) {
|
|
232
|
+
return ((result as { content: unknown[] }).content)
|
|
233
|
+
.map((part) => typeof part === "object" && part !== null && typeof (part as { text?: unknown }).text === "string" ? (part as { text: string }).text : typeof part === "string" ? part : "")
|
|
234
|
+
.filter(Boolean)
|
|
235
|
+
.join("\n")
|
|
236
|
+
.slice(-MAX_TOOL_OUTPUT);
|
|
237
|
+
}
|
|
238
|
+
return String(result ?? "").slice(-MAX_TOOL_OUTPUT);
|
|
239
|
+
}
|
|
240
|
+
|
|
241
|
+
private send(value: Record<string, unknown>): void {
|
|
242
|
+
if (!this.child?.stdin?.writable) throw new Error("Sidekick process is not writable");
|
|
243
|
+
this.child.stdin.write(`${JSON.stringify(value)}\n`);
|
|
244
|
+
}
|
|
245
|
+
|
|
246
|
+
private cleanupPrompt(): void {
|
|
247
|
+
if (!this.promptPath) return;
|
|
248
|
+
try {
|
|
249
|
+
unlinkSync(this.promptPath);
|
|
250
|
+
} catch {
|
|
251
|
+
/* already removed */
|
|
252
|
+
}
|
|
253
|
+
this.promptPath = undefined;
|
|
254
|
+
}
|
|
255
|
+
|
|
256
|
+
private spawn(): void {
|
|
257
|
+
mkdirSync(dirname(this.cfg.sessionFile), { recursive: true });
|
|
258
|
+
this.promptPath = join(tmpdir(), `unipi-fusion-${randomUUID()}.txt`);
|
|
259
|
+
writeFileSync(this.promptPath, this.cfg.systemPrompt, "utf8");
|
|
260
|
+
const command = this.cfg.command ?? getPiSpawnCommand([
|
|
261
|
+
...leadExtensionArgs(process.argv),
|
|
262
|
+
"--mode", "rpc",
|
|
263
|
+
"--session", this.cfg.sessionFile,
|
|
264
|
+
"--model", this.cfg.model,
|
|
265
|
+
"--thinking", this.cfg.thinking,
|
|
266
|
+
"--append-system-prompt", this.promptPath,
|
|
267
|
+
"--no-skills",
|
|
268
|
+
...(this.cfg.extraArgs ?? []),
|
|
269
|
+
]);
|
|
270
|
+
const spawn = this.cfg.spawn ?? defaultSpawn;
|
|
271
|
+
const child = spawn(command.command, command.args, {
|
|
272
|
+
cwd: this.cfg.cwd,
|
|
273
|
+
env: { ...process.env, UNIPI_FUSION_CHILD: "1", UNIPI_SUBAGENT_CHILD: "1", ...(this.cfg.extraEnv ?? {}) },
|
|
274
|
+
stdio: ["pipe", "pipe", "pipe"],
|
|
275
|
+
});
|
|
276
|
+
this.child = child;
|
|
277
|
+
child.stdout?.on("data", (data: Buffer | string) => this.readStdout(String(data)));
|
|
278
|
+
child.stderr?.on("data", (data: Buffer | string) => {
|
|
279
|
+
this.stderrTail = `${this.stderrTail}${String(data)}`.slice(-2048);
|
|
280
|
+
});
|
|
281
|
+
child.on("close", () => {
|
|
282
|
+
this.cleanupPrompt();
|
|
283
|
+
if (this.pending !== undefined) this.finish("error", undefined, this.stderrTail || "Sidekick process exited");
|
|
284
|
+
});
|
|
285
|
+
child.on("error", (error) => {
|
|
286
|
+
if (this.pending !== undefined) this.finish("error", undefined, error.message);
|
|
287
|
+
});
|
|
288
|
+
}
|
|
289
|
+
|
|
290
|
+
private readStdout(data: string): void {
|
|
291
|
+
this.inputBuffer += data;
|
|
292
|
+
const lines = this.inputBuffer.split("\n");
|
|
293
|
+
this.inputBuffer = lines.pop() ?? "";
|
|
294
|
+
for (const raw of lines) {
|
|
295
|
+
const line = raw.endsWith("\r") ? raw.slice(0, -1) : raw;
|
|
296
|
+
if (line.length === 0) continue;
|
|
297
|
+
try {
|
|
298
|
+
this.handleMessage(JSON.parse(line) as Record<string, unknown>);
|
|
299
|
+
} catch {
|
|
300
|
+
continue;
|
|
301
|
+
}
|
|
302
|
+
}
|
|
303
|
+
}
|
|
304
|
+
|
|
305
|
+
private requestLastAssistantText(): void {
|
|
306
|
+
const current = this.pending;
|
|
307
|
+
if (current === undefined) return;
|
|
308
|
+
this.responseText = (text) => this.finish(this.abortRequested ? "aborted" : this.pendingError === undefined ? "completed" : "error", text, this.pendingError);
|
|
309
|
+
this.responseError = (error) => this.finish("error", undefined, error.message);
|
|
310
|
+
try {
|
|
311
|
+
this.send({ type: "get_last_assistant_text" });
|
|
312
|
+
} catch (error) {
|
|
313
|
+
this.finish("error", undefined, error instanceof Error ? error.message : String(error));
|
|
314
|
+
}
|
|
315
|
+
}
|
|
316
|
+
|
|
317
|
+
private handleMessage(message: Record<string, unknown>): void {
|
|
318
|
+
if (message.type === "response") {
|
|
319
|
+
const command = message.command;
|
|
320
|
+
if (command === "prompt" && message.success === false) {
|
|
321
|
+
const errorText = String(message.error ?? "Sidekick prompt rejected");
|
|
322
|
+
const current = this.pending;
|
|
323
|
+
if (current === undefined) return;
|
|
324
|
+
if (/already processing/i.test(errorText) && !current.retriedPrompt) {
|
|
325
|
+
current.retriedPrompt = true;
|
|
326
|
+
try {
|
|
327
|
+
this.send({ id: current.id, type: "prompt", message: current.message, streamingBehavior: "followUp" });
|
|
328
|
+
} catch (error) {
|
|
329
|
+
this.finish("error", undefined, error instanceof Error ? error.message : String(error));
|
|
330
|
+
}
|
|
331
|
+
} else {
|
|
332
|
+
this.finish("error", undefined, errorText);
|
|
333
|
+
}
|
|
334
|
+
} else if (command === "get_last_assistant_text") {
|
|
335
|
+
const data = message.data as Record<string, unknown> | undefined;
|
|
336
|
+
const text = typeof data?.text === "string" ? data.text : this.pending?.progress.textTail ?? "";
|
|
337
|
+
this.responseText?.(text);
|
|
338
|
+
this.responseText = undefined;
|
|
339
|
+
this.responseError = undefined;
|
|
340
|
+
}
|
|
341
|
+
return;
|
|
342
|
+
}
|
|
343
|
+
if (message.type === "extension_ui_request") {
|
|
344
|
+
const method = String(message.method ?? "");
|
|
345
|
+
const id = message.id;
|
|
346
|
+
const ui = this.forwardedUi;
|
|
347
|
+
// Devin rule: a sidekick approval prompt reaches the user only while a
|
|
348
|
+
// lead waiter is attached (blocking sidekick / read_subagent block:true).
|
|
349
|
+
// Background handoffs get an automatic refusal.
|
|
350
|
+
if (ui !== undefined && (method === "select" || method === "confirm" || method === "input")) {
|
|
351
|
+
const title = `${this.cfg.promptPrefix ?? "Sidekick: "}${String(message.title ?? "")}`;
|
|
352
|
+
void (async () => {
|
|
353
|
+
let value: unknown;
|
|
354
|
+
if (method === "select" && ui.select !== undefined) value = await ui.select(title, (message.options as unknown[]) ?? [], undefined);
|
|
355
|
+
else if (method === "confirm" && ui.confirm !== undefined) value = await ui.confirm(title, String(message.message ?? ""), undefined);
|
|
356
|
+
else if (method === "input" && ui.input !== undefined) value = await ui.input(title, typeof message.placeholder === "string" ? message.placeholder : undefined, undefined);
|
|
357
|
+
if (value === undefined || value === null) this.send({ type: "extension_ui_response", id, cancelled: true });
|
|
358
|
+
else this.send({ type: "extension_ui_response", id, value });
|
|
359
|
+
})().catch(() => {
|
|
360
|
+
try {
|
|
361
|
+
this.send({ type: "extension_ui_response", id, cancelled: true });
|
|
362
|
+
} catch { /* child gone */ }
|
|
363
|
+
});
|
|
364
|
+
} else {
|
|
365
|
+
this.refuseInBackground(method, id, message);
|
|
366
|
+
}
|
|
367
|
+
return;
|
|
368
|
+
}
|
|
369
|
+
if (this.pending === undefined) return;
|
|
370
|
+
if (message.type === "tool_execution_start") {
|
|
371
|
+
this.closeOpenText();
|
|
372
|
+
this.flushTextStep();
|
|
373
|
+
this.pending.progress.toolCalls += 1;
|
|
374
|
+
const args = message.args !== undefined && typeof message.args === "object" && message.args !== null ? message.args as Record<string, unknown> : undefined;
|
|
375
|
+
if (message.toolName === "bg_run" && args?.notifyOnCompletion !== false && args?.triggerOnCompletion !== false) this.pending.openBgTasks += 1;
|
|
376
|
+
const argsText = args === undefined ? "" : JSON.stringify(args).replace(/\s+/gu, " ");
|
|
377
|
+
const summary = `${String(message.toolName ?? "tool")}(${argsText})`.slice(0, 40);
|
|
378
|
+
this.pending.progress.recentTools = [...this.pending.progress.recentTools, summary].slice(-6);
|
|
379
|
+
this.appendEvent({ kind: "tool", toolCallId: String(message.toolCallId ?? ""), name: String(message.toolName ?? "tool"), args, output: "", isError: false, done: false, startedAt: Date.now() });
|
|
380
|
+
this.notifyProgress();
|
|
381
|
+
} else if (message.type === "tool_execution_update") {
|
|
382
|
+
// Partial output of a running tool (bash streams) — live views only.
|
|
383
|
+
const toolCallId = String(message.toolCallId ?? "");
|
|
384
|
+
const event = [...this.pending.progress.events].reverse().find((entry): entry is Extract<SidekickEvent, { kind: "tool" }> => entry.kind === "tool" && entry.toolCallId === toolCallId);
|
|
385
|
+
if (event && !event.done && message.partialResult !== undefined) {
|
|
386
|
+
event.output = this.toolOutput(message.partialResult);
|
|
387
|
+
this.notifyProgress();
|
|
388
|
+
}
|
|
389
|
+
} else if (message.type === "tool_execution_end") {
|
|
390
|
+
const toolCallId = String(message.toolCallId ?? "");
|
|
391
|
+
const event = [...this.pending.progress.events].reverse().find((entry): entry is Extract<SidekickEvent, { kind: "tool" }> => entry.kind === "tool" && entry.toolCallId === toolCallId);
|
|
392
|
+
if (event) {
|
|
393
|
+
event.done = true;
|
|
394
|
+
event.endedAt = Date.now();
|
|
395
|
+
event.isError = message.isError === true;
|
|
396
|
+
event.output = this.toolOutput(message.result);
|
|
397
|
+
if (event.name === "bg_run" && event.isError) this.pending.openBgTasks = Math.max(0, this.pending.openBgTasks - 1);
|
|
398
|
+
const capLine = (l: string) => (l.length > 300 ? `${l.slice(0, 299)}…` : l);
|
|
399
|
+
this.emitStep({
|
|
400
|
+
kind: "tool",
|
|
401
|
+
name: event.name,
|
|
402
|
+
args: event.args,
|
|
403
|
+
arg: event.args === undefined ? "" : primaryArg(event.name, event.args),
|
|
404
|
+
output: event.output.split("\n").slice(-40).map(capLine).join("\n"),
|
|
405
|
+
isError: event.isError,
|
|
406
|
+
durationMs: (event.endedAt ?? Date.now()) - event.startedAt,
|
|
407
|
+
});
|
|
408
|
+
}
|
|
409
|
+
this.notifyProgress();
|
|
410
|
+
} else if (message.type === "message_update") {
|
|
411
|
+
const streamEvent = (message.assistantMessageEvent ?? message) as Record<string, unknown>;
|
|
412
|
+
if (streamEvent.type === "text_delta") {
|
|
413
|
+
const delta = typeof streamEvent.delta === "string" ? streamEvent.delta : typeof streamEvent.text === "string" ? streamEvent.text : "";
|
|
414
|
+
this.pending.progress.textTail = `${this.pending.progress.textTail}${delta}`.slice(-400);
|
|
415
|
+
const last = this.pending.progress.events.at(-1);
|
|
416
|
+
if (last?.kind === "text" && last.open) last.text += delta;
|
|
417
|
+
else this.appendEvent({ kind: "text", text: delta, open: true });
|
|
418
|
+
(this.bufferedText ??= { text: "", thinking: "" }).text += delta;
|
|
419
|
+
this.notifyProgress();
|
|
420
|
+
} else if (streamEvent.type === "thinking_delta") {
|
|
421
|
+
const delta = typeof streamEvent.delta === "string" ? streamEvent.delta : "";
|
|
422
|
+
(this.bufferedText ??= { text: "", thinking: "" }).thinking += delta;
|
|
423
|
+
}
|
|
424
|
+
} else if (message.type === "message_end") {
|
|
425
|
+
const msg = message.message as Record<string, unknown> | undefined;
|
|
426
|
+
if (msg?.role === "custom" && msg.customType === "background-task-notification") {
|
|
427
|
+
this.pending.openBgTasks = Math.max(0, this.pending.openBgTasks - 1);
|
|
428
|
+
if (this.pending.openBgTasks === 0 && this.pending.settled) {
|
|
429
|
+
clearTimeout(this.pending.settleTimer);
|
|
430
|
+
this.pending.settleTimer = setTimeout(() => {
|
|
431
|
+
const current = this.pending;
|
|
432
|
+
if (current === undefined || !current.settled || current.openBgTasks > 0) return;
|
|
433
|
+
this.requestLastAssistantText();
|
|
434
|
+
}, this.cfg.settleGraceMs ?? 3000);
|
|
435
|
+
this.pending.settleTimer.unref();
|
|
436
|
+
}
|
|
437
|
+
this.notifyProgress();
|
|
438
|
+
} else if (msg?.role === "assistant") {
|
|
439
|
+
this.closeOpenText();
|
|
440
|
+
this.notifyProgress();
|
|
441
|
+
if (msg.stopReason === "error" && typeof msg.errorMessage === "string") this.pendingError = msg.errorMessage;
|
|
442
|
+
const usage = msg.usage as Record<string, unknown> | undefined;
|
|
443
|
+
if (usage) {
|
|
444
|
+
const cost = usage.cost as Record<string, unknown> | undefined;
|
|
445
|
+
this.addUsage(this.pending.usage, {
|
|
446
|
+
input: usage.input,
|
|
447
|
+
output: usage.output,
|
|
448
|
+
cacheRead: usage.cacheRead,
|
|
449
|
+
cacheWrite: usage.cacheWrite,
|
|
450
|
+
cost: cost?.total,
|
|
451
|
+
});
|
|
452
|
+
}
|
|
453
|
+
}
|
|
454
|
+
} else if (message.type === "message_start") {
|
|
455
|
+
this.flushTextStep();
|
|
456
|
+
} else if (message.type === "agent_start") {
|
|
457
|
+
clearTimeout(this.pending.settleTimer);
|
|
458
|
+
this.pending.settleTimer = undefined;
|
|
459
|
+
this.pending.settled = false;
|
|
460
|
+
this.notifyProgress();
|
|
461
|
+
} else if (message.type === "agent_settled") {
|
|
462
|
+
this.pending.settled = true;
|
|
463
|
+
if (this.pending.openBgTasks > 0) {
|
|
464
|
+
this.notifyProgress();
|
|
465
|
+
return;
|
|
466
|
+
}
|
|
467
|
+
this.requestLastAssistantText();
|
|
468
|
+
}
|
|
469
|
+
}
|
|
470
|
+
|
|
471
|
+
private addUsage(target: SidekickUsage, raw: Record<string, unknown>): void {
|
|
472
|
+
const number = (value: unknown) => (typeof value === "number" && Number.isFinite(value) ? value : 0);
|
|
473
|
+
target.input += number(raw.input);
|
|
474
|
+
target.output += number(raw.output);
|
|
475
|
+
target.cacheRead += number(raw.cacheRead);
|
|
476
|
+
target.cacheWrite += number(raw.cacheWrite);
|
|
477
|
+
target.cost += number(raw.cost);
|
|
478
|
+
if (target !== this.usage) this.addUsage(this.usage, raw);
|
|
479
|
+
}
|
|
480
|
+
|
|
481
|
+
private finish(status: HandoffReport["status"], text?: string, error?: string): void {
|
|
482
|
+
const current = this.pending;
|
|
483
|
+
if (current === undefined) return;
|
|
484
|
+
clearTimeout(current.settleTimer);
|
|
485
|
+
current.settleTimer = undefined;
|
|
486
|
+
this.pending = undefined;
|
|
487
|
+
this.responseText = undefined;
|
|
488
|
+
this.responseError = undefined;
|
|
489
|
+
this.bufferedText = undefined; // terminal report — never a step
|
|
490
|
+
const report: HandoffReport = {
|
|
491
|
+
id: current.id,
|
|
492
|
+
status,
|
|
493
|
+
text: text ?? current.progress.textTail,
|
|
494
|
+
usage: { ...current.usage },
|
|
495
|
+
toolCalls: current.progress.toolCalls,
|
|
496
|
+
durationMs: Date.now() - current.startedAt,
|
|
497
|
+
events: current.progress.events.map((event) => ({ ...event })),
|
|
498
|
+
...(error === undefined ? {} : { error }),
|
|
499
|
+
};
|
|
500
|
+
this.reports.set(report.id, report);
|
|
501
|
+
if (this.latestHandoff?.id === report.id) this.latestHandoff.report = report;
|
|
502
|
+
current.resolve(report);
|
|
503
|
+
this.notifyProgress();
|
|
504
|
+
this.abortRequested = false;
|
|
505
|
+
this.pendingError = undefined;
|
|
506
|
+
}
|
|
507
|
+
|
|
508
|
+
handoff(message: string): { id: string; done: Promise<HandoffReport> } {
|
|
509
|
+
if (this.pending !== undefined) {
|
|
510
|
+
this.send({ type: "steer", message });
|
|
511
|
+
return { id: this.pending.id, done: this.latestHandoff?.done ?? Promise.reject(new Error("Missing handoff")) };
|
|
512
|
+
}
|
|
513
|
+
if (!this.isAlive()) this.spawn();
|
|
514
|
+
const id = randomUUID();
|
|
515
|
+
const startedAt = Date.now();
|
|
516
|
+
let resolve!: (report: HandoffReport) => void;
|
|
517
|
+
const done = new Promise<HandoffReport>((res) => {
|
|
518
|
+
resolve = res;
|
|
519
|
+
});
|
|
520
|
+
this.pendingError = undefined;
|
|
521
|
+
this.pending = {
|
|
522
|
+
id,
|
|
523
|
+
message,
|
|
524
|
+
startedAt,
|
|
525
|
+
usage: emptyUsage(),
|
|
526
|
+
progress: { toolCalls: 0, recentTools: [], textTail: "", startedAt, events: [], droppedEvents: 0 },
|
|
527
|
+
retriedPrompt: false,
|
|
528
|
+
openBgTasks: 0,
|
|
529
|
+
settled: false,
|
|
530
|
+
resolve,
|
|
531
|
+
};
|
|
532
|
+
this.latestHandoff = { id, done };
|
|
533
|
+
try {
|
|
534
|
+
this.send({ id, type: "prompt", message });
|
|
535
|
+
} catch (error) {
|
|
536
|
+
this.finish("error", undefined, error instanceof Error ? error.message : String(error));
|
|
537
|
+
}
|
|
538
|
+
return { id, done };
|
|
539
|
+
}
|
|
540
|
+
|
|
541
|
+
progress(id?: string): HandoffProgress | undefined {
|
|
542
|
+
if (this.pending !== undefined && (id === undefined || id === this.pending.id)) {
|
|
543
|
+
return { ...this.pending.progress, recentTools: [...this.pending.progress.recentTools], events: this.pending.progress.events.map((event) => ({ ...event })) };
|
|
544
|
+
}
|
|
545
|
+
return undefined;
|
|
546
|
+
}
|
|
547
|
+
|
|
548
|
+
latest(): { id: string; done: Promise<HandoffReport>; report?: HandoffReport } | undefined {
|
|
549
|
+
return this.latestHandoff;
|
|
550
|
+
}
|
|
551
|
+
|
|
552
|
+
async abort(): Promise<void> {
|
|
553
|
+
if (!this.pending) return;
|
|
554
|
+
this.abortRequested = true;
|
|
555
|
+
this.send({ type: "abort" });
|
|
556
|
+
}
|
|
557
|
+
|
|
558
|
+
/** Update the thinking level: live over rpc when the child is running, and
|
|
559
|
+
* in the spawn args the next time a process is (re)started. */
|
|
560
|
+
setThinking(level: EffortLevel): void {
|
|
561
|
+
this.cfg.thinking = level;
|
|
562
|
+
if (!this.isAlive()) return;
|
|
563
|
+
try {
|
|
564
|
+
this.send({ type: "set_thinking_level", level });
|
|
565
|
+
} catch {
|
|
566
|
+
// A mid-exit write fails; the level still applies to the next spawn.
|
|
567
|
+
}
|
|
568
|
+
}
|
|
569
|
+
|
|
570
|
+
kill(): void {
|
|
571
|
+
const child = this.child;
|
|
572
|
+
if (!child) return;
|
|
573
|
+
this.cleanupPrompt();
|
|
574
|
+
child.kill("SIGTERM");
|
|
575
|
+
const timer = setTimeout(() => {
|
|
576
|
+
if (child.exitCode === null || child.exitCode === undefined) child.kill("SIGKILL");
|
|
577
|
+
}, 2000);
|
|
578
|
+
timer.unref();
|
|
579
|
+
this.child = undefined;
|
|
580
|
+
}
|
|
581
|
+
}
|
|
582
|
+
|
|
583
|
+
/** Back-compat alias for the fusion sidekick runtime. */
|
|
584
|
+
export { ChildAgentRuntime as SidekickRuntime };
|
|
@@ -0,0 +1,78 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Evidence contributors — how other modules hand the long-horizon verifier
|
|
3
|
+
* hard signals about work outside the transcript.
|
|
4
|
+
*
|
|
5
|
+
* Kanboard (step 4) registers one so a goal cannot complete while this
|
|
6
|
+
* session still holds an In Progress claim. Contributors are time-boxed and
|
|
7
|
+
* a throw counts as no contribution — evidence must never break settlement.
|
|
8
|
+
*/
|
|
9
|
+
|
|
10
|
+
export interface EvidenceContribution {
|
|
11
|
+
blocking: string[];
|
|
12
|
+
notes: string[];
|
|
13
|
+
}
|
|
14
|
+
|
|
15
|
+
export type EvidenceContributor = () => Promise<EvidenceContribution>;
|
|
16
|
+
|
|
17
|
+
const KEY = Symbol.for("unipi.evidence.contributors");
|
|
18
|
+
|
|
19
|
+
interface Holder {
|
|
20
|
+
contributors: Map<string, EvidenceContributor>;
|
|
21
|
+
}
|
|
22
|
+
|
|
23
|
+
function holder(): Holder {
|
|
24
|
+
const g = globalThis as { [KEY]?: Holder };
|
|
25
|
+
g[KEY] ??= { contributors: new Map() };
|
|
26
|
+
return g[KEY] as Holder;
|
|
27
|
+
}
|
|
28
|
+
|
|
29
|
+
/** Register a named contributor; re-registering a name replaces it. */
|
|
30
|
+
export function registerEvidenceContributor(
|
|
31
|
+
name: string,
|
|
32
|
+
contributor: () => Promise<EvidenceContribution>,
|
|
33
|
+
): () => void {
|
|
34
|
+
const h = holder();
|
|
35
|
+
h.contributors.set(name, contributor);
|
|
36
|
+
return () => {
|
|
37
|
+
const current = h.contributors.get(name);
|
|
38
|
+
if (current === contributor) h.contributors.delete(name);
|
|
39
|
+
};
|
|
40
|
+
}
|
|
41
|
+
|
|
42
|
+
/** Test/holder introspection hook. */
|
|
43
|
+
export function resetEvidenceForTests(): void {
|
|
44
|
+
holder().contributors.clear();
|
|
45
|
+
}
|
|
46
|
+
|
|
47
|
+
/**
|
|
48
|
+
* Gather every contribution, each time-boxed; a throw or timeout contributes
|
|
49
|
+
* nothing. `blocking` items force a `not_met` verdict downstream; `notes`
|
|
50
|
+
* ride the evidence brief as context.
|
|
51
|
+
*/
|
|
52
|
+
export async function gatherEvidence(timeoutMs = 2_000): Promise<EvidenceContribution> {
|
|
53
|
+
const contributors = [...holder().contributors.values()];
|
|
54
|
+
if (contributors.length === 0) return { blocking: [], notes: [] };
|
|
55
|
+
const blocking: string[] = [];
|
|
56
|
+
const notes: string[] = [];
|
|
57
|
+
await Promise.all(
|
|
58
|
+
contributors.map(async (contributor) => {
|
|
59
|
+
let timer: ReturnType<typeof setTimeout> | undefined;
|
|
60
|
+
try {
|
|
61
|
+
const contribution = await Promise.race([
|
|
62
|
+
Promise.resolve(contributor()),
|
|
63
|
+
new Promise<null>((resolve) => {
|
|
64
|
+
timer = setTimeout(() => resolve(null), timeoutMs);
|
|
65
|
+
}),
|
|
66
|
+
]);
|
|
67
|
+
if (contribution === null) return;
|
|
68
|
+
blocking.push(...contribution.blocking);
|
|
69
|
+
notes.push(...contribution.notes);
|
|
70
|
+
} catch {
|
|
71
|
+
// A failed contributor is no contribution.
|
|
72
|
+
} finally {
|
|
73
|
+
if (timer !== undefined) clearTimeout(timer);
|
|
74
|
+
}
|
|
75
|
+
}),
|
|
76
|
+
);
|
|
77
|
+
return { blocking, notes };
|
|
78
|
+
}
|