diff --git a/.agents/skills/dsh-find-simplifications/SKILL.md b/.agents/skills/dsh-find-simplifications/SKILL.md index 8fad167a60..59dde3df0d 100644 --- a/.agents/skills/dsh-find-simplifications/SKILL.md +++ b/.agents/skills/dsh-find-simplifications/SKILL.md @@ -41,6 +41,14 @@ Use parallel subagents when the user asks for breadth or many candidates. Give e If subagents are unavailable, simulate the same breadth yourself. Do not let the first good candidate stop the survey. +Start with the largest production-code deltas. A broad simplification audit that stops after obvious unused symbols can miss the files where duplicated lifecycle or defensive machinery carries most of the cost. + +## Audit Trust And Lifecycle Boundaries + +Classify every defensive copy, freeze, validator, and callback capture by the boundary it crosses. Same-process typed service/plugin calls ordinarily borrow readonly values; parser/config, queue, model/tool JSON, durable/file, worker, process, and wire boundaries own or validate data. Tests built around hostile getters, fake typed objects, callback replacement, or mutation after a same-process handoff are evidence of a potentially speculative contract, not automatic justification for keeping it. + +For complex asynchronous code, draw the ownership graph and map each sentinel, readiness promise, cancellation path, disposer, and state flag to a distinct owner or transition. When several mechanisms mirror the same liveness or settlement fact, propose one transaction or lifecycle controller instead. Preserve separate machinery where it protects a real boundary: synchronous publication and rollback, callback containment, first-terminal-outcome arbitration, worker/process ownership, or dispose-to-quiescence. + ## Prove Or Reject Each Candidate For every symbol or behavior, classify consumers before writing: @@ -60,7 +68,7 @@ Reject or downgrade a candidate when: ## Write The RFC -Create one file per durable proposal under `docs/rfc/proposed/yyyy-mm-dd-topic.md` and add it to the Proposed table in `docs/rfc/README.md`. Keep prose paragraphs on one physical line and use relative Markdown links. +Create one file per durable proposal under `docs/rfc///yyyy-mm-dd-topic.md`, following the lifecycle/classification contract in `docs/rfc/README.md`. Regenerate `docs/rfc/INDEX.md`; never add a manual RFC table to the README. Keep prose paragraphs on one physical line and use relative Markdown links. Prefer this shape, adjusting when the idea needs it: diff --git a/docs/architecture.md b/docs/architecture.md index 15a1142ac3..2012be0946 100644 --- a/docs/architecture.md +++ b/docs/architecture.md @@ -58,7 +58,7 @@ A **session** is one agent's append-only event log. A **turn** drains one queued ### Turn Flow ```text -reserve ids -> mint agent.ctx -> await unpublished setup +prepare private session + agent.ctx -> await unpublished setup -> enter session + agent -> session/created -> agent/created -> enable driving -> agent/session-start(source) -> start driver forever: @@ -93,7 +93,7 @@ forever: checkpoint persistence and notify idle/running status ``` -Prompt assembly is single-path: `renderPrompt(assemble({ agent }))` IS the system prompt sent to the model. Plugins contribute ordered sections (static or computed from the per-call `AssembleContext`), tool schemas, and named variables interpolated as `{{name}}` at render — strictly, so an unknown or valueless reference fails the turn instead of shipping a hole. `dsh-system-prompt` owns the openers — the static `harness:identity` section (order −100) and the deployment's persona (order 0, its `persona` config, shared context-wide) — while the shipped loop registers the `model`/`cwd` variables; prompt-fact ownership is pinned by the [prompt-variables RFC](rfc/implemented/architecture/2026-07-05-prompt-variables-and-tool-guidance-ownership.md). +Prompt assembly is single-path: the loop sends `renderPrompt(await assemble(assembleContextFor(agent)))`; the helper couples the explicit agent and scope. Plugins contribute ordered sections, tool schemas, and named variables interpolated as `{{name}}` at render — strictly, so an unknown or valueless reference fails the turn instead of shipping a hole. `dsh-system-prompt` owns the openers — the static `harness:identity` section (order −100) and the deployment's persona (order 0, its `persona` config, shared context-wide) — while the loop registers the `model`/`cwd` variables; prompt-fact ownership is pinned by the [prompt-variables RFC](rfc/implemented/architecture/2026-07-05-prompt-variables-and-tool-guidance-ownership.md). Post-tool context lands after all tool results so tool-call/result adjacency stays stable. Steering drains between steps; ordinary leftover steering after a turn is re-queued as input. A terminal `agent/turn-stop` is the explicit exception: it runs after ordinary continuation and steering folding, then remains authoritative through turn close and flush so steering from those later listeners is discarded rather than becoming another step or turn; ordinary queued prompts are preserved. @@ -109,7 +109,7 @@ Every session event is turn-enclosed. Reloading a crashed session preserves the ### Agent Scope -Every live agent owns a scope context, `agent.ctx` ([`dsh-scope`](../packages/core/scope/README.md), key = the agent). Registrations through it — tools, prompt sections/variables, listeners, `tools.restrict()` masks — are visible to that agent alone, SHADOW same-named global contributions for it (per-agent personas and tool variants), and unwind with the agent; an `agent.ctx` listener hears only that agent's dispatches, while events about one agent dispatch with its scope carrier. `CreateAgentOptions.setup(agentCtx)` composes a child's scoped world at creation (the subagent seam's `persona`/`toolFilter`) — setup registers, never drives. Dev invariants enforce carrier/subject identity; `verify-scoped-dispatch` pins enforced ⇔ documented. Rationale: [agent-scope RFC](rfc/implemented/architecture/2026-07-08-agent-scope-contexts.md). +Every live agent owns `agent.ctx` ([`dsh-scope`](../packages/core/scope/README.md), keyed by the agent). Its registrations are visible only to that agent, shadow same-named globals, and unwind with it. Its listeners hear only that agent's dispatches; an opaque carrier routes while the real subject stays explicit. `CreateAgentOptions.setup(agentCtx)` composes this world before publication and does not drive. Dev invariants and `verify-scoped-dispatch` keep carrier/subject identity aligned with event declarations. Rationale: [agent-scope RFC](rfc/implemented/architecture/2026-07-08-agent-scope-contexts.md); subagent `persona`, `toolFilter`, and `maxDepth` are the separate [composition-controls feature](rfc/implemented/feature/2026-07-12-subagent-persona-tool-filter-and-depth.md). ## State @@ -160,6 +160,7 @@ New behavior should attach to a documented extension point; changing the shipped The [extension cookbook](cookbook/extension-cookbook.md) carries plugin skeletons and the feature-to-seam map; step-by-step guides cover [packages](cookbook/adding-a-package.md), [tools](cookbook/adding-a-tool.md), [LLM adapters](cookbook/adding-an-llm-adapter.md), and [vendored packages](cookbook/adding-a-vendored-package.md). ## Quick Reference +- Domain terms in the [glossary](glossary.md) - Type definitions in [core-data-structures/](core-data-structures/core.md) - Exact event and service signatures in [events](cordis-catalog/events.md) - [services](cordis-catalog/services.md) catalogs diff --git a/docs/config-catalog.md b/docs/config-catalog.md index 57f8bc961a..63d1303fc6 100644 --- a/docs/config-catalog.md +++ b/docs/config-catalog.md @@ -113,30 +113,15 @@ Source: [`packages/core/agent-core/src/index.ts:87`](../packages/core/agent-core Requires: `agents` · `sessions` · `llm` · `tools` · `systemPrompt` ```ts config-catalog -/** - * Plugin config: the agents to create — or resume, via `resumeSessionId` — - * declaratively at startup, so a cordis.yml deployment needs no code. - */ +/** Plugin configuration for declarative startup agents. */ export interface Config { - /** Agents created from configuration at startup. */ + /** Agents created or resumed at plugin startup. */ agents: (AgentOptions & { - /** Agent id to register under; also seeds the fresh per-run session id (`${id}-session-`). */ + /** Registry identity for the live agent. */ id: AgentId - /** Optional workspace cwd for the config-created fresh session. */ + /** Optional workspace for a fresh session. */ cwd?: string - /** - * If set, the config agent RESUMES this persisted session id instead of - * starting a fresh `${id}-session-`. Sourced from an env var in - * cordis.yml (`resumeSessionId: !!js process.env.RESUME_SESSION_ID`), so a - * demo can continue a prior conversation without code changes. Requires a - * `dsh-session-persistence` backend; the resume is deferred until that - * service is available (via `ctx.inject`) and the loaded session's events - * seed the live session so history continues. - * - * The schema accepts a plain string at runtime (cordis.yml values are - * untyped); the brand is compile-time only — the config format is the - * boundary where an id enters, so the TYPE declares the brand here. - */ + /** Persisted session to resume instead of creating a fresh session. */ resumeSessionId?: SessionId })[] } @@ -144,7 +129,7 @@ export interface Config { Depends on: [`AgentId`](../packages/core/agent/src/index.ts) · [`AgentOptions`](../packages/core/agent/src/index.ts) · [`SessionId`](../packages/core/session/src/index.ts) -Source: [`packages/core/agent-loop/src/index.ts:119`](../packages/core/agent-loop/src/index.ts) +Source: [`packages/core/agent-loop/src/index.ts:325`](../packages/core/agent-loop/src/index.ts) ## `@deepseek-ai/dsh-bash-local` @@ -560,12 +545,12 @@ Source: [`packages/session-persistence/session-persistence-sqlite/src/index.ts:5 ```ts config-catalog /** Skill registry configuration. */ export interface Config { - /** Maximum number of completed cwd/provider catalog snapshots kept in memory. */ - collectCacheMaxEntries?: number + /** Maximum number of completed cwd/provider catalogs kept in memory. */ + readonly collectCacheMaxEntries?: number } ``` -Source: [`packages/skill/skill/src/index.ts:114`](../packages/skill/skill/src/index.ts) +Source: [`packages/skill/skill/src/index.ts:113`](../packages/skill/skill/src/index.ts) ## `@deepseek-ai/dsh-skill-local` @@ -729,7 +714,7 @@ export interface Config { Depends on: [`SubagentCapabilities`](../packages/subagent/subagent/src/index.ts) · [`SubagentStopReason`](../packages/subagent/subagent/src/index.ts) -Source: [`packages/support/subagent-mock/src/index.ts:87`](../packages/support/subagent-mock/src/index.ts) +Source: [`packages/support/subagent-mock/src/index.ts:97`](../packages/support/subagent-mock/src/index.ts) ## `@deepseek-ai/dsh-subagent-spawn` @@ -792,7 +777,7 @@ export interface Config { } ``` -Source: [`packages/core/system-prompt/src/index.ts:315`](../packages/core/system-prompt/src/index.ts) +Source: [`packages/core/system-prompt/src/index.ts:257`](../packages/core/system-prompt/src/index.ts) ## `@deepseek-ai/dsh-tool-cordis` @@ -973,7 +958,7 @@ export interface Config { export type ToolPresentationMode = 'native' | 'code' | 'both' ``` -Source: [`packages/core/tools/src/index.ts:409`](../packages/core/tools/src/index.ts) +Source: [`packages/core/tools/src/index.ts:407`](../packages/core/tools/src/index.ts) ## `@deepseek-ai/dsh-user-approval` @@ -986,7 +971,7 @@ export interface Config { * (fail-closed with none); `'never'` auto-rejects every ask without * prompting (the deterministic CI/unattended stance). */ - policy?: ApprovalPolicy + readonly policy?: ApprovalPolicy } /** @@ -1004,7 +989,7 @@ export interface Config { export type ApprovalPolicy = 'ask' | 'never' ``` -Source: [`packages/ui/user-approval/src/index.ts:281`](../packages/ui/user-approval/src/index.ts) +Source: [`packages/ui/user-approval/src/index.ts:270`](../packages/ui/user-approval/src/index.ts) ## `@deepseek-ai/dsh-web` diff --git a/docs/cookbook/adding-a-tool.md b/docs/cookbook/adding-a-tool.md index 3832f44dd0..2912fde6e4 100644 --- a/docs/cookbook/adding-a-tool.md +++ b/docs/cookbook/adding-a-tool.md @@ -34,7 +34,7 @@ Registration is effect-based: disposing the plugin fiber unregisters the tool (w ## Rules of the execute() contract - **Args are validated for you.** `defineTool` validates the model-generated `arguments` against the `SchemaSpec` before `execute` runs (type, required keys, enum membership, nested objects/arrays — [runtime arg validation](../rfc/implemented/architecture/2026-06-11-runtime-arg-validation.md)), so inside `execute` the args already match `InferArgs`. You still hand-check value constraints the DSL can't express (non-empty strings, positive numbers, cross-field rules); throw a descriptive Error for those. Raw JSON-Schema tools registered directly (MCP) are NOT validated by the harness — they validate their own input. -- **Registration snapshots your definition.** Parameters must be losslessly JSON-serializable; the registry reads them once and materializes the detached stored value in one recursive pass, copies the scalar fields, binds each callback once to your definition as its method receiver, and freezes the stored record. Reassigning `definition.execute` after registration does not hot-swap the tool—dispose and register a new definition through the owning effect instead. Deliberate mutable state inside the callback's closure or receiver remains ordinary plugin state. +- **Registration borrows your readonly definition.** A typed same-process contribution is not a serialization boundary; do not mutate its schema or replace callbacks after registration. `schemas()` materializes only the explicit model-facing projection. To hot-swap a tool, dispose its owning effect and register the replacement; mutable state inside the callback's closure remains ordinary plugin state. - **Execution identity is protected.** The registry materializes `arguments` as detached lossless JSON in one recursive pass, freezes that value before policy starts, and assigns an opaque `exec.token`; `callId`, `name`, `arguments`, `agent`, `token`, and an optional enclosing-transport `parent` token stay immutable through dispatch. `parent` is identity-only and exposes no live outer execution. Treat `args` as readonly input. An around-dispatch wrapper may add, replace, or remove only `exec.signal` to impose cancellation or a deadline. - **Throwing or returning non-JSON data means isError.** The registry catches anything `execute()` throws and materializes the complete post-policy result as lossless JSON before final observers run. A throw, malformed result, or non-JSON content/context/meta becomes `{isError: true}` so the live outcome cannot succeed and then fail at the durable log. Use errors for infrastructure failures (bad input, spawn errors, aborts), but report domain failures in the result text instead (for example, tool-bash returns `[exit code: 9]` with `isError: false` because the model decides what a failing command means). - **Honor `exec.signal`.** Cancel in-flight work when it fires. diff --git a/docs/cookbook/extension-cookbook.md b/docs/cookbook/extension-cookbook.md index a16cda8799..ac62cdd75b 100644 --- a/docs/cookbook/extension-cookbook.md +++ b/docs/cookbook/extension-cookbook.md @@ -56,7 +56,7 @@ export function apply(ctx: Context) { ## A client-driver plugin (external protocol bridge) -A *client driver* is a UI plugin whose "user" is another program speaking a wire protocol rather than a human at a terminal. It owns the process's stdio (so it must run with **no stdout logger** — every non-protocol byte corrupts the stream), creates/resumes agents on demand through the `dsh-agent` factory seam, translates harness events (`session/event`, `agent/*`) into outbound protocol messages, and translates inbound requests back into `agent.send()` / `agent.cancel()`. Two harness-specific contracts make it correct: resolve each request exactly once from the durable `turn/end` session event, using `agent/status` only as defensive reconciliation against the canonical log, and tear each agent down through its `AgentHandle.dispose()` (which stops the loop, `await`s its exit, and unregisters), not just `cancel()` — disposal must *reach* quiescence, not merely request it. +A *client driver* is a UI plugin whose "user" is another program speaking a wire protocol rather than a human at a terminal. It owns the process's stdio (so it must run with **no stdout logger** — every non-protocol byte corrupts the stream), creates/resumes agents on demand through the `dsh-agent` factory seam, translates harness events (`session/event`, `agent/*`) into outbound protocol messages, and translates inbound requests back into `agent.send()` / `agent.cancel()`. Two harness-specific contracts make it correct: correlate and settle each request exactly once from the durable `turn/end` session event even if rendering fails, and tear each agent down through its `AgentHandle.dispose()` (which stops the loop, `await`s its exit, and unregisters), not just `cancel()` — disposal must *reach* quiescence, not merely request it. `packages/ui/acp` is the worked example: it bridges the agent to the Agent Client Protocol (JSON-RPC over stdio) so Zed and other ACP editors can drive it. See its README for the full method surface and the permission-prompt answerer it registers on the approval seam. @@ -94,14 +94,14 @@ Every product feature maps to a listener on a documented extension seam — the | Hook system (user + project level) | listeners on `agent/session-start`, `agent/prompt-submit`, `agent/request`, `agent/step-result`, `tools/pre-execute`, `tools/post-execute`, `agent/turn-continuation` — each interception waterfall returns a typed Decision; the `dsh-hooks-claude` / `dsh-hooks-codex` bridges map hook config files onto these seams | | `/goal` | force-continue via `agent/turn-continuation` + `steer()` reminders | | `/loop` | on the `turn/end` session event, `send()` the next iteration; or force-continue | -| Dynamic workflow | `ctx.workflows` + the worker-thread engine + the `workflow` tool; structured in-process children enforce output with scoped prompt protection, a monotonic tool guard, final `tools/result` commit (including enclosing `run_code`), and terminal `agent/turn-stop` | +| Dynamic workflow | `ctx.workflows` + the worker-thread engine + the `workflow` tool; structured in-process children enforce output with scoped owner-final prompt/tool contributions, a monotonic tool guard, final `tools/result` commit (including enclosing `run_code`), and terminal `agent/turn-stop` | | Queued + steering messages | core `Agent.send()` / `Agent.steer()` | | Context compaction (auto + manual) | the `ctx.compact` seam + a backend (`dsh-compact-basic`) on the serial `agent/pre-step` seam; auto = token-pressure check before each step; a manual trigger invokes the same `ctx.compact` routine ([compaction RFC](../rfc/implemented/feature/2026-06-18-compaction-capability-seam.md) — the model-facing `/compact` consumer tool is deferred) | -| System prompt configurability | `ctx.systemPrompt.section()` with ordering; an owner uses `systemPrompt.protect()` only when its canonical section/tool presence is a correctness invariant | +| System prompt configurability | `ctx.systemPrompt.section()` with ordering; a protocol owner sets `ownerFinal: true` on the section or tool only when canonical presence is a correctness invariant | | AGENTS.md (root) | a section provider reading the file | | AGENTS.md (subdir, on-touch) + file-change notices | `agent.inject()` from a watcher / tool-result listener | | Built-in tools | `ctx.tools.register()`; schemas flow into the assembly automatically — the `dsh-tool-*` families (bash, fs, web, subagent, todo) are the shipped examples | -| ToolSearch / progressive disclosure | filter ordinary capabilities at `system-prompt/assemble` (the loop logs the result as the request header); owner-protected transport and correctness entries retain their canonical presence or absence | +| ToolSearch / progressive disclosure | filter ordinary capabilities at `system-prompt/assemble` (the loop logs the result as the request header); owner-final transport and correctness contributions retain their canonical presence or absence | | Tool deadline / retry / metrics | wrap core dispatch with `tools/execute`; a wrapper may replace `exec.signal`, delegate, and inspect the normalized result in one lexical lifetime | | Final tool-result metrics / audit / capture | observe immutable authoritative outcomes with `tools/result`; use `tools/post-execute` instead only when the plugin must transform the result or attach context | | Monotonic terminal turn policy | return `{ action: 'stop' }` from serial `agent/turn-stop`, after continuation and steering have already been folded | diff --git a/docs/cordis-catalog/events.md b/docs/cordis-catalog/events.md index 21e69f7a25..e0149c7c67 100644 --- a/docs/cordis-catalog/events.md +++ b/docs/cordis-catalog/events.md @@ -15,7 +15,7 @@ Dispatch modes: **emit** (fire-and-forget), **waterfall** (each listener gets `n ### `agent/created` — emit -An agent's fully composed scoped world was published in the AgentRegistry. Its session is already live in the session store, but concrete factories may keep driving verbs locked until the subsequent `agent/session-start` boundary; that event is the first supported place to inject or queue work during startup. A synchronous listener throw vetoes publication and rollback emits the matching disposal edges; returned-promise rejection is observed and logged but cannot retroactively veto this synchronous boundary. A synchronous listener that requests the advanced registry detach does not remove the entry immediately: removal and the paired `agent/disposed` edge wait until the creation dispatch unwinds, so no later creation listener observes a disposal that preceded its own creation callback. +An agent's fully composed scoped world was published in the AgentRegistry. Its session is already live in the session store. Setup is composition-only by contract; the subsequent `agent/session-start` boundary is the first supported place to inject or queue startup work. A synchronous listener throw vetoes publication and rollback emits the matching disposal edges; returned-promise rejection is observed and logged but cannot retroactively veto this synchronous boundary. A synchronous listener that requests the advanced registry detach does not remove the entry immediately: removal and the paired `agent/disposed` edge wait until the creation dispatch unwinds, so no later creation listener observes a disposal that preceded its own creation callback. ```ts cordis-catalog 'agent/created'(this: Scoped, agent: Agent): void @@ -47,7 +47,7 @@ A step or turn errored. The loop reports a failure here (plus the logger) even w Types: [Agent](../core-data-structures/core.md) -Source: [`packages/core/agent/src/types.ts:608`](../../packages/core/agent/src/types.ts) +Source: [`packages/core/agent/src/types.ts:606`](../../packages/core/agent/src/types.ts) ### `agent/pre-step` — serial @@ -61,7 +61,7 @@ Serial (awaited in registration order), not a waterfall: a listener mutates the Types: [Agent](../core-data-structures/core.md) · [Message](../core-data-structures/core.md) -Source: [`packages/core/agent/src/types.ts:440`](../../packages/core/agent/src/types.ts) +Source: [`packages/core/agent/src/types.ts:438`](../../packages/core/agent/src/types.ts) ### `agent/prompt-submit` — waterfall @@ -73,11 +73,11 @@ Waterfall: decide what happens to ONE drained queued message before it becomes a Types: [Agent](../core-data-structures/core.md) · [ContentBlock](../core-data-structures/core.md) · [MessageSource](../core-data-structures/core.md) -Source: [`packages/core/agent/src/types.ts:458`](../../packages/core/agent/src/types.ts) +Source: [`packages/core/agent/src/types.ts:456`](../../packages/core/agent/src/types.ts) ### `agent/queued` — emit -A message entered the agent's inbox (queued or steering). Content and the resolved source are the detached, deeply-frozen values retained by the inbox; the `info` wrapper is frozen too, so one listener cannot rewrite what another listener observes. `source` has defaults applied and is not the caller's raw options. +A message entered the agent's inbox (queued or steering). Content and the resolved source are the detached, deeply-frozen values retained by the inbox. `source` has defaults applied and is not the caller's raw options. ```ts cordis-catalog 'agent/queued'(this: Scoped, agent: Agent, content: ContentBlock[], info: { source: MessageSource; steering: boolean }): void @@ -85,7 +85,7 @@ A message entered the agent's inbox (queued or steering). Content and the resolv Types: [Agent](../core-data-structures/core.md) · [ContentBlock](../core-data-structures/core.md) · [MessageSource](../core-data-structures/core.md) -Source: [`packages/core/agent/src/types.ts:362`](../../packages/core/agent/src/types.ts) +Source: [`packages/core/agent/src/types.ts:360`](../../packages/core/agent/src/types.ts) ### `agent/request` — waterfall @@ -97,7 +97,7 @@ Waterfall: shape the step's call configuration — model switching, sampling ove Types: [Agent](../core-data-structures/core.md) · [LlmCallConfig](../core-data-structures/core.md) -Source: [`packages/core/agent/src/types.ts:487`](../../packages/core/agent/src/types.ts) +Source: [`packages/core/agent/src/types.ts:485`](../../packages/core/agent/src/types.ts) ### `agent/session-prefix` — waterfall @@ -113,7 +113,7 @@ The seed is a frozen empty list; a contributing listener returns a NEW array — Types: [Agent](../core-data-structures/core.md) · [Message](../core-data-structures/core.md) -Source: [`packages/core/agent/src/types.ts:539`](../../packages/core/agent/src/types.ts) +Source: [`packages/core/agent/src/types.ts:537`](../../packages/core/agent/src/types.ts) ### `agent/session-start` — emit @@ -125,7 +125,7 @@ The agent's session lifecycle began, fired once before its first turn. `source` Types: [Agent](../core-data-structures/core.md) · [SessionStartSource](../core-data-structures/core.md) -Source: [`packages/core/agent/src/types.ts:383`](../../packages/core/agent/src/types.ts) +Source: [`packages/core/agent/src/types.ts:381`](../../packages/core/agent/src/types.ts) ### `agent/status` — emit @@ -149,7 +149,7 @@ Waterfall: post-process the assembled assistant Message before tool dispatch (va Types: [Agent](../core-data-structures/core.md) · [Message](../core-data-structures/core.md) -Source: [`packages/core/agent/src/types.ts:554`](../../packages/core/agent/src/types.ts) +Source: [`packages/core/agent/src/types.ts:552`](../../packages/core/agent/src/types.ts) ### `agent/turn-continuation` — waterfall @@ -161,7 +161,7 @@ Waterfall: override the turn-continuation decision via a typed ContinuationDecis Types: [Agent](../core-data-structures/core.md) -Source: [`packages/core/agent/src/types.ts:572`](../../packages/core/agent/src/types.ts) +Source: [`packages/core/agent/src/types.ts:570`](../../packages/core/agent/src/types.ts) ### `agent/turn-stop` — serial @@ -173,13 +173,13 @@ Serial terminal-stop checkpoint after the ordinary `agent/turn-continuation` wat Types: [Agent](../core-data-structures/core.md) -Source: [`packages/core/agent/src/types.ts:591`](../../packages/core/agent/src/types.ts) +Source: [`packages/core/agent/src/types.ts:589`](../../packages/core/agent/src/types.ts) ## `approval/*` ### `approval/request` — waterfall -Waterfall asking the composed answerers to decide one approval request. Dispatched only from ApprovalService.request — callers go through the service (which owns cancellation and the audit events), never through `ctx.waterfall` directly. A listener that can answer for this request's agent returns an outcome WITHOUT calling `next()` (the decision slot is single-occupancy, first listener to answer wins); a listener that does not recognize the agent MUST call `next()` so another answerer — or the fail-closed default `'unavailable'` — gets the question. Throwing is contained by the service and yields `'unavailable'`. Scope-filtered dispatch (`@deepseek-ai/dsh-scope`) keys the carrier by `req.agent`: a listener registered through `agent.ctx` receives only that agent's questions, while a plain-context listener receives every agent's. `req` is the service's shallow-frozen acceptance snapshot: later caller mutation cannot redirect the question, while the `agent` and `signal` identity capabilities remain exact. +Waterfall asking the composed answerers to decide one approval request. Dispatched only from ApprovalService.request — callers go through the service (which owns cancellation and the audit events), never through `ctx.waterfall` directly. A listener that can answer for this request's agent returns an outcome WITHOUT calling `next()` (the decision slot is single-occupancy, first listener to answer wins); a listener that does not recognize the agent MUST call `next()` so another answerer — or the fail-closed default `'unavailable'` — gets the question. Throwing is contained by the service and yields `'unavailable'`. Scope-filtered dispatch (`@deepseek-ai/dsh-scope`) keys the carrier by `req.agent`: a listener registered through `agent.ctx` receives only that agent's questions, while a plain-context listener receives every agent's. `req` is a readonly same-process value borrowed from the caller. ```ts cordis-catalog 'approval/request'(this: Scoped, req: ApprovalRequest, next: () => Promise): Promise @@ -187,7 +187,7 @@ Waterfall asking the composed answerers to decide one approval request. Dispatch Types: [ApprovalOutcome](../core-data-structures/approval.md) · [ApprovalRequest](../core-data-structures/approval.md) -Source: [`packages/ui/user-approval/src/index.ts:72`](../../packages/ui/user-approval/src/index.ts) +Source: [`packages/ui/user-approval/src/index.ts:70`](../../packages/ui/user-approval/src/index.ts) ## `fs/*` @@ -295,7 +295,7 @@ A skill provider became resolvable in the `ctx.skills` registry. Consumers can o 'skill/provider-added'(provider: SkillProvider): void ``` -Source: [`packages/skill/skill/src/index.ts:132`](../../packages/skill/skill/src/index.ts) +Source: [`packages/skill/skill/src/index.ts:131`](../../packages/skill/skill/src/index.ts) ### `skill/provider-removed` — emit @@ -305,49 +305,49 @@ A skill provider left the registry because its plugin fiber was disposed. 'skill/provider-removed'(name: string): void ``` -Source: [`packages/skill/skill/src/index.ts:138`](../../packages/skill/skill/src/index.ts) +Source: [`packages/skill/skill/src/index.ts:137`](../../packages/skill/skill/src/index.ts) ## `subagent/*` ### `subagent/end` — emit -A started subagent run settled — emitted when SubagentRun.result resolves (any stop reason) or rejects (reported as `error`). Paired with Events['subagent/start']; a run whose readiness rejected emits neither event. Scope-filtered dispatch (`@deepseek-ai/dsh-scope`): the carrier is keyed by the DELEGATING PARENT — a listener registered through the parent's `agent.ctx` observes only its own delegations; a plain plugin listener observes every run. +A ready child settled. Scope-filtered dispatch uses the same delegating parent carrier as `subagent/start`, so the lifecycle pair reaches the same scoped audience. ```ts cordis-catalog 'subagent/end'(this: Scoped, info: SubagentRunEndInfo): void ``` -Source: [`packages/subagent/subagent/src/index.ts:134`](../../packages/subagent/subagent/src/index.ts) +Source: [`packages/subagent/subagent/src/index.ts:90`](../../packages/subagent/subagent/src/index.ts) ### `subagent/provider-added` — emit -A provider became resolvable in the SubagentService registry. Consumers that derive state from a named provider (e.g. the model-facing tool wording in `dsh-tool-subagent`) react HERE instead of assuming load order — the cordis Loader starts sibling plugins concurrently, so "listed earlier in cordis.yml" does not mean "registered earlier". +A provider became resolvable in the registry. ```ts cordis-catalog 'subagent/provider-added'(provider: SubagentProvider): void ``` -Source: [`packages/subagent/subagent/src/index.ts:95`](../../packages/subagent/subagent/src/index.ts) +Source: [`packages/subagent/subagent/src/index.ts:66`](../../packages/subagent/subagent/src/index.ts) ### `subagent/provider-removed` — emit -A provider left the registry (its plugin's fiber was disposed — an unload or an HMR reload). Consumers holding provider-derived state drop it here; a reload re-fires `subagent/provider-added` with the fresh provider. Delivered with per-listener containment: a throwing subscriber is logged, never starves later subscribers, and never disrupts the provider's teardown. +A provider left the registry. Accepted runs remain holder-owned. ```ts cordis-catalog 'subagent/provider-removed'(name: string): void ``` -Source: [`packages/subagent/subagent/src/index.ts:106`](../../packages/subagent/subagent/src/index.ts) +Source: [`packages/subagent/subagent/src/index.ts:72`](../../packages/subagent/subagent/src/index.ts) ### `subagent/start` — emit -A subagent run started — emitted only after SubagentRun.started fulfills, when the provider has established a live child. For an in-process provider, `ctx.agents.get(info.id)` is therefore guaranteed to resolve during this notification. A readiness rejection emits neither lifecycle event; every emitted start is paired with Events['subagent/end']. Scope-filtered dispatch (`@deepseek-ai/dsh-scope`): the carrier is keyed by the DELEGATING PARENT — a listener registered through the parent's `agent.ctx` observes only its own delegations; a plain plugin listener observes every run. +A provider established a ready child. For in-process providers, `ctx.agents.get(info.id)` resolves during this notification. Scope-filtered dispatch keys the carrier by the delegating parent, so a parent-scoped listener observes only its own delegations. Paired with `subagent/end`. ```ts cordis-catalog 'subagent/start'(this: Scoped, info: SubagentRunInfo): void ``` -Source: [`packages/subagent/subagent/src/index.ts:121`](../../packages/subagent/subagent/src/index.ts) +Source: [`packages/subagent/subagent/src/index.ts:82`](../../packages/subagent/subagent/src/index.ts) ## `system-prompt/*` @@ -359,17 +359,17 @@ Waterfall around prompt assembly — mutate or extend the PromptAssembly (sectio 'system-prompt/assemble'(this: Scoped, assembly: PromptAssembly, context: AssembleContext, next: () => Promise): Promise ``` -Source: [`packages/core/system-prompt/src/index.ts:46`](../../packages/core/system-prompt/src/index.ts) +Source: [`packages/core/system-prompt/src/index.ts:45`](../../packages/core/system-prompt/src/index.ts) ### `system-prompt/change` — emit -A section, tool provider, variable provider, or protection was registered or unregistered (the assembly inputs changed — possibly for one scope only). An UNFILTERED registry-subject notification, deliberately not scope-filtered dispatch: a global change concerns every agent's next assembly, so a scoped listener subscribing here sees every change, not just its own scope's. +A section, tool provider, or variable provider was registered or unregistered (the assembly inputs changed — possibly for one scope only). An UNFILTERED registry-subject notification, deliberately not scope-filtered dispatch: a global change concerns every agent's next assembly, so a scoped listener subscribing here sees every change, not just its own scope's. ```ts cordis-catalog 'system-prompt/change'(): void ``` -Source: [`packages/core/system-prompt/src/index.ts:56`](../../packages/core/system-prompt/src/index.ts) +Source: [`packages/core/system-prompt/src/index.ts:55`](../../packages/core/system-prompt/src/index.ts) ## `tools/*` @@ -381,7 +381,7 @@ A tool was registered or unregistered, or a scoped restriction changed (the avai 'tools/change'(): void ``` -Source: [`packages/core/tools/src/index.ts:176`](../../packages/core/tools/src/index.ts) +Source: [`packages/core/tools/src/index.ts:173`](../../packages/core/tools/src/index.ts) ### `tools/execute` — waterfall @@ -393,23 +393,23 @@ Around-dispatch waterfall wrapping the registry's core tool dispatch, between th Types: [ToolExecution](../core-data-structures/tools.md) · [ToolExecutionResult](../core-data-structures/tools.md) -Source: [`packages/core/tools/src/index.ts:131`](../../packages/core/tools/src/index.ts) +Source: [`packages/core/tools/src/index.ts:128`](../../packages/core/tools/src/index.ts) ### `tools/post-execute` — waterfall Waterfall AFTER a tool runs — where hook plugins inspect the result and accept it (optionally REPLACING the model-facing content, and/or attaching `additionalContext` for the next request) or block it with corrective `feedback` (Claude Code's `PostToolUse`). Listeners receive `(exec, result, next)`: call `next()` to delegate to the default (accept unchanged), or return a PostToolDecision to override. Core tool dispatch runs earlier as the base `next()` of the `tools/execute` waterfall, all inside `execute`'s outer try/catch (and the tool body keeps its own inner try/catch, so a thrown tool still reaches `post-execute` as an `isError` result). Scope-filtered dispatch (`@deepseek-ai/dsh-scope`): the carrier is keyed by `exec.agent` — a listener registered through `agent.ctx` fires only for that agent's calls; a plain plugin listener fires for every call (including agent-less ones, which dispatch subject-less). ```ts cordis-catalog -'tools/post-execute'(this: Scoped, exec: ToolExecution, result: ToolExecutionResult, next: () => Promise): Promise +'tools/post-execute'(this: Scoped, exec: ToolExecution, result: Readonly, next: () => Promise): Promise ``` Types: [ToolExecution](../core-data-structures/tools.md) · [ToolExecutionResult](../core-data-structures/tools.md) -Source: [`packages/core/tools/src/index.ts:151`](../../packages/core/tools/src/index.ts) +Source: [`packages/core/tools/src/index.ts:148`](../../packages/core/tools/src/index.ts) ### `tools/pre-execute` — waterfall -Waterfall BEFORE a tool runs — the gate where sandbox, permission, and hook plugins allow or deny a call (Claude Code's `PreToolUse`). Listeners receive `(exec, next)`: call `next()` to delegate to the default (allow), or return a PreToolDecision without calling `next()` to short-circuit. A `deny` skips dispatch and yields an `isError` result; the tool body never runs. Input rewrite is deliberately NOT offered here (see PreToolDecision); `ask` is serviced by the `ctx.approval` seam when one is mounted, and degrades to deny otherwise. The returned union is validated as an exact runtime shape before approval or guards run; a malformed JavaScript/casted decision fails closed as an `isError` result and the tool body never runs. Scope-filtered dispatch (`@deepseek-ai/dsh-scope`) keys the carrier by `exec.agent`: a listener registered through `agent.ctx` fires only for that agent's calls, while a plain plugin listener fires for every call (including agent-less ones, which dispatch subject-less). +Waterfall BEFORE a tool runs — the gate where sandbox, permission, and hook plugins allow or deny a call (Claude Code's `PreToolUse`). Listeners receive `(exec, next)`: call `next()` to delegate to the default (allow), or return a PreToolDecision without calling `next()` to short-circuit. A `deny` skips dispatch and yields an `isError` result; the tool body never runs. Input rewrite is deliberately NOT offered here (see PreToolDecision); `ask` is serviced by the `ctx.approval` seam when one is mounted, and degrades to deny otherwise. Scope-filtered dispatch (`@deepseek-ai/dsh-scope`) keys the carrier by `exec.agent`: a listener registered through `agent.ctx` fires only for that agent's calls, while a plain plugin listener fires for every call (including agent-less ones, which dispatch subject-less). ```ts cordis-catalog 'tools/pre-execute'(this: Scoped, exec: ToolExecution, next: () => Promise): Promise @@ -417,7 +417,7 @@ Waterfall BEFORE a tool runs — the gate where sandbox, permission, and hook pl Types: [ToolExecution](../core-data-structures/tools.md) -Source: [`packages/core/tools/src/index.ts:104`](../../packages/core/tools/src/index.ts) +Source: [`packages/core/tools/src/index.ts:101`](../../packages/core/tools/src/index.ts) ### `tools/result` — parallel @@ -429,7 +429,7 @@ Awaited notification of the authoritative FINAL tool outcome, after the complete Types: [ToolExecution](../core-data-structures/tools.md) · [ToolExecutionResult](../core-data-structures/tools.md) -Source: [`packages/core/tools/src/index.ts:166`](../../packages/core/tools/src/index.ts) +Source: [`packages/core/tools/src/index.ts:163`](../../packages/core/tools/src/index.ts) ## `workflow/*` @@ -441,17 +441,17 @@ One `agent()` call settled (clean result, child failure, or run cancellation). P 'workflow/agent-end'(info: WorkflowRunInfo, agent: WorkflowAgentEndInfo): void ``` -Source: [`packages/workflow/workflow/src/index.ts:98`](../../packages/workflow/workflow/src/index.ts) +Source: [`packages/workflow/workflow/src/index.ts:96`](../../packages/workflow/workflow/src/index.ts) ### `workflow/agent-start` — emit -One `agent()` call established a ready child run. Paired with Events['workflow/agent-end'] by `agent.seq`. A call that never crosses the provider's publication/readiness boundary emits neither event in this pair. +One `agent()` call established a ready child run. Paired with Events['workflow/agent-end'] by `agent.seq`. A call that never receives a ready run from the provider emits neither event in this pair. ```ts cordis-catalog 'workflow/agent-start'(info: WorkflowRunInfo, agent: WorkflowAgentInfo): void ``` -Source: [`packages/workflow/workflow/src/index.ts:87`](../../packages/workflow/workflow/src/index.ts) +Source: [`packages/workflow/workflow/src/index.ts:85`](../../packages/workflow/workflow/src/index.ts) ### `workflow/end` — emit @@ -461,7 +461,7 @@ A workflow run settled (any stop reason). Fired when WorkflowRun.result resolves 'workflow/end'(info: WorkflowRunInfo, result: WorkflowResultInfo): void ``` -Source: [`packages/workflow/workflow/src/index.ts:108`](../../packages/workflow/workflow/src/index.ts) +Source: [`packages/workflow/workflow/src/index.ts:106`](../../packages/workflow/workflow/src/index.ts) ### `workflow/log` — emit @@ -471,7 +471,7 @@ The script emitted a narration line (a `log(message)` call). 'workflow/log'(info: WorkflowRunInfo, message: string): void ``` -Source: [`packages/workflow/workflow/src/index.ts:77`](../../packages/workflow/workflow/src/index.ts) +Source: [`packages/workflow/workflow/src/index.ts:75`](../../packages/workflow/workflow/src/index.ts) ### `workflow/phase` — emit @@ -481,7 +481,7 @@ The script entered a phase (a `phase(title)` call) — progress grouping for obs 'workflow/phase'(info: WorkflowRunInfo, title: string): void ``` -Source: [`packages/workflow/workflow/src/index.ts:70`](../../packages/workflow/workflow/src/index.ts) +Source: [`packages/workflow/workflow/src/index.ts:68`](../../packages/workflow/workflow/src/index.ts) ### `workflow/start` — emit @@ -491,7 +491,7 @@ A workflow run started — the script's meta block validated, the body about to 'workflow/start'(info: WorkflowRunInfo): void ``` -Source: [`packages/workflow/workflow/src/index.ts:62`](../../packages/workflow/workflow/src/index.ts) +Source: [`packages/workflow/workflow/src/index.ts:60`](../../packages/workflow/workflow/src/index.ts) ## Inherited events (cordis core + loader/hmr/timer) diff --git a/docs/cordis-catalog/services.md b/docs/cordis-catalog/services.md index 67a6b7a725..2e508e0a35 100644 --- a/docs/cordis-catalog/services.md +++ b/docs/cordis-catalog/services.md @@ -11,9 +11,7 @@ The **harness tier** below (the `@deepseek-ai/dsh-*` packages) is the vocabulary ## `ctx.agentLoop` — `AgentLoop` -The agent-loop plugin (`ctx.agentLoop`): creates ReactLoopAgents, runs their loops, and registers them in `ctx.agents`. Also implements the AgentFactory seam, so plugins create/resume agents through `ctx.agents` (the interface) without depending on this concrete package. - -The loop itself is deliberately thin — every behavior beyond "call the model, run the tools, repeat" belongs to plugins listening on the event taxonomy declared in @deepseek-ai/dsh-agent. +Concrete ReactLoopAgent factory and driver service. ```ts cordis-catalog create(id: AgentId, options: AgentOptions = {}, meta: Pick = {}): ReactLoopAgent @@ -21,27 +19,26 @@ async createAgent(ownerCtx: Context, options: CreateAgentOptions): Promise ``` -Source: [`packages/core/agent-loop/src/index.ts:153`](../../packages/core/agent-loop/src/index.ts) +Source: [`packages/core/agent-loop/src/index.ts:338`](../../packages/core/agent-loop/src/index.ts) ## `ctx.agents` — `AgentRegistry` Agent registry (`ctx.agents`): tracks live agents so UI, hook, and orchestrator plugins can find them without depending on the concrete loop package. Agent *creation* is provided by whichever plugin implements the AgentFactory (`@deepseek-ai/dsh-agent-loop`), registered via setFactory. ```ts cordis-catalog -reserve(id: AgentId): AgentRegistrationReservation setFactory(factory: AgentFactory): () => Promise | void async create(options: CreateAgentOptions): Promise async resume(options: ResumeAgentOptions): Promise register(agent: Agent): () => Promise | void -enter(agent: Agent, reservation?: AgentRegistrationReservation): () => void +enter(agent: Agent): () => void announce(agent: Agent): void get(id: AgentId): Agent | undefined list(): Agent[] ``` -Types: [Agent](../core-data-structures/core.md) · [AgentRegistrationReservation](../core-data-structures/core.md) +Types: [Agent](../core-data-structures/core.md) -Source: [`packages/core/agent/src/index.ts:250`](../../packages/core/agent/src/index.ts) +Source: [`packages/core/agent/src/index.ts:203`](../../packages/core/agent/src/index.ts) ## `ctx.approval` — `ApprovalService` @@ -55,7 +52,7 @@ async request(req: ApprovalRequest): Promise Types: [ApprovalOutcome](../core-data-structures/approval.md) · [ApprovalRequest](../core-data-structures/approval.md) -Source: [`packages/ui/user-approval/src/index.ts:305`](../../packages/ui/user-approval/src/index.ts) +Source: [`packages/ui/user-approval/src/index.ts:294`](../../packages/ui/user-approval/src/index.ts) ## `ctx.bash` — `BashExecutor` (abstract seam) @@ -211,10 +208,9 @@ In-memory session store (`ctx.sessions`). Persistence is intentionally not implemented here — persistence plugins subscribe to `session/event` and flush on `session/flush` / dispose. ```ts cordis-catalog -reserve(id: SessionId): SessionRegistrationReservation create(id?: SessionId, options?: CreateSessionOptions): Session prepare(id?: SessionId, options?: CreateSessionOptions): Session -enter(session: Session, reservation?: SessionRegistrationReservation): () => void +enter(session: Session): () => void announce(session: Session): void async flush(session: Session): Promise get(id: SessionId): Session | undefined @@ -222,9 +218,7 @@ list(): Session[] fork(source: SessionForkSource, boundary?: number, childSessionId?: SessionId): Session ``` -Types: [SessionRegistrationReservation](../core-data-structures/session.md) - -Source: [`packages/core/session/src/index.ts:761`](../../packages/core/session/src/index.ts) +Source: [`packages/core/session/src/index.ts:591`](../../packages/core/session/src/index.ts) ## `ctx.skills` — `SkillService` @@ -237,55 +231,52 @@ async list(options: SkillLookupOptions = {}): Promise async get(name: string, options: SkillLookupOptions = {}): Promise ``` -Source: [`packages/skill/skill/src/index.ts:159`](../../packages/skill/skill/src/index.ts) +Source: [`packages/skill/skill/src/index.ts:158`](../../packages/skill/skill/src/index.ts) ## `ctx.subagents` — `SubagentService` -The `subagents` service: a registry of named SubagentProviders and a capability-checked start surface. +Named provider registry and capability-checked start surface. ```ts cordis-catalog registerProvider(provider: SubagentProvider): () => Promise | void getProvider(name: string): SubagentProvider | undefined list(): string[] -start(name: string, request: SubagentStartRequest): SubagentRun +async start(name: string, request: SubagentStartRequest): Promise ``` -Source: [`packages/subagent/subagent/src/index.ts:180`](../../packages/subagent/subagent/src/index.ts) +Source: [`packages/subagent/subagent/src/index.ts:123`](../../packages/subagent/subagent/src/index.ts) ## `ctx.systemPrompt` — `SystemPrompt` -Registry service (`ctx.systemPrompt`): plugins contribute ordered text sections, tool-schema providers, named prompt variables, and owner-final contribution protections; the agent loop calls `assemble(context)` once per step. Registers the harness-owned `harness:identity` and `deployment:persona` sections itself (see Config.persona). +Registry service (`ctx.systemPrompt`): plugins contribute ordered text sections, tool-schema providers, named prompt variables, and owner-final contributions; the agent loop calls `assemble(context)` once per step. Registers the harness-owned `harness:identity` and `deployment:persona` sections itself (see Config.persona). ```ts cordis-catalog section(section: PromptSection): () => Promise | void tools(provider: (context: AssembleContext) => ToolProviderResult): () => Promise | void variable(name: string, provider: (context: AssembleContext) => string | undefined): () => Promise | void -protect(protection: PromptProtection): () => Promise | void async assemble(context: AssembleContext = {}): Promise ``` -Source: [`packages/core/system-prompt/src/index.ts:430`](../../packages/core/system-prompt/src/index.ts) +Source: [`packages/core/system-prompt/src/index.ts:372`](../../packages/core/system-prompt/src/index.ts) ## `ctx.tools` — `ToolRegistry` Tool registry (`ctx.tools`): tool plugins register definitions; the agent loop executes calls through the `tools/pre-execute` → guards → `tools/execute` → `tools/post-execute` → `tools/result` pipeline. The registry contributes its schemas into the system-prompt assembly — WHICH schemas is governed by its `mode` config (see Config.mode); under a non-native mode it also owns the reserved `run_code` presentation transport and the `tools:sdk` prompt section. -Two registration layers (`@deepseek-ai/dsh-scope`): a registration through a plain plugin context is GLOBAL (visible to every agent); one through a scoped context (`agent.ctx`) is filed in that scope's layer — visible to that agent alone, disposed with the scope, and SHADOWING a global tool of the same name for that agent (most-specific-wins; within one layer a duplicate name still throws). restrict masks the global layer per scope. One visibility function (visible) feeds prompt assembly, get, and execute — and, under a non-native mode, the SDK section and `run_code`'s bindings — so what the model is shown, what a presenter renders, what a program can call, and what dispatches can never disagree. +Two registration layers (`@deepseek-ai/dsh-scope`): a registration through a plain plugin context is GLOBAL (visible to every agent); one through a scoped context (`agent.ctx`) is filed in that scope's layer — visible to that agent alone, disposed with the scope, and SHADOWING a global tool of the same name for that agent (most-specific-wins; within one layer a duplicate name still throws). restrict masks the global layer per scope. One private visibility resolver feeds prompt assembly, get, and execute — and, under a non-native mode, the SDK section and `run_code`'s bindings — so what the model is shown, what a presenter renders, what a program can call, and what dispatches can never disagree. ```ts cordis-catalog register(definition: ToolDefinition): () => Promise | void restrict(filter: ToolRestriction): () => Promise | void guard(guard: ToolGuard): () => Promise | void -visible(scope?: ScopeKey): ToolDefinition[] get(name: string, scope?: ScopeKey): ToolDefinition | undefined schemas(scope?: ScopeKey): ToolSchema[] -knownNames(scope?: ScopeKey): string[] async execute(exec: ToolExecutionInput): Promise ``` Types: [ToolDefinition](../core-data-structures/tools.md) · [ToolExecutionInput](../core-data-structures/tools.md) · [ToolExecutionResult](../core-data-structures/tools.md) -Source: [`packages/core/tools/src/index.ts:484`](../../packages/core/tools/src/index.ts) +Source: [`packages/core/tools/src/index.ts:500`](../../packages/core/tools/src/index.ts) ## `ctx.userInteraction` — `UserInteractionService` @@ -327,7 +318,7 @@ Abstract workflow execution service. Subclass, implement start, and load the sub Semantics every implementation must honor: - start throws synchronously for a request that cannot begin (an unparseable script, an invalid meta block). Once it returns a WorkflowRun, `result` NEVER rejects — every failure resolves with `stopReason: 'error'` (or `'cancelled'`) — and once the run is cancelled, `result` SETTLES within the implementation's bounded grace even if the script itself never settles (a consumer awaiting `result` must never be wedged past a cancellation). -- The `workflow/*` events fire through emitWorkflowEvent (data snapshots, per-listener containment); `workflow/end` fires exactly once per started run, after `result` is settled or as it settles. +- The `workflow/*` events fire through emitWorkflowEvent (borrowed immutable data, per-listener containment); `workflow/end` fires exactly once per started run, after `result` is settled or as it settles. - `dispose()` reaches quiescence within a bounded grace: it cancels, waits for the script to settle AND its started children to finish disposing, and abandons whatever is left rather than hanging its caller (the engine documents what abandonment leaves behind). - Runs are HOLDER-OWNED: the engine hands control (`cancel`/`dispose`) to the `start()` caller and does not track its live runs — disposing the engine's own fiber mid-run deliberately leaves those runs to their holders' teardown, so an engine reload cannot yank a run out from under the consumer awaiting it. @@ -335,7 +326,7 @@ Semantics every implementation must honor: abstract start(request: WorkflowStartRequest): WorkflowRun ``` -Source: [`packages/workflow/workflow/src/index.ts:214`](../../packages/workflow/workflow/src/index.ts) +Source: [`packages/workflow/workflow/src/index.ts:211`](../../packages/workflow/workflow/src/index.ts) ## Inherited `ctx` members (cordis core + loader/hmr/timer) diff --git a/docs/core-data-structures/approval.md b/docs/core-data-structures/approval.md index 997f4e7ab2..c5fa1fe13b 100644 --- a/docs/core-data-structures/approval.md +++ b/docs/core-data-structures/approval.md @@ -39,21 +39,21 @@ interface ApprovalRequest { * UI answerer only answers for agents it owns) and receives the audit * events on its session log. */ - agent: Agent + readonly agent: Agent /** The tool the question is about (presentation and audit). */ - toolName: string + readonly toolName: string /** * The exact tool call being decided, when the asker has one — lets a UI * attach the prompt to the tool call it already streamed. */ - callId?: CallId + readonly callId?: CallId /** The asker's human-readable explanation of WHY it is asking. */ - reason?: string + readonly reason?: string /** * Aborting withdraws the question: the request settles `'cancelled'` * immediately and a late answer from a still-pending answerer is discarded. */ - signal?: AbortSignal + readonly signal?: AbortSignal } ``` diff --git a/docs/core-data-structures/core.md b/docs/core-data-structures/core.md index bd533fe5c4..ecf075b20f 100644 --- a/docs/core-data-structures/core.md +++ b/docs/core-data-structures/core.md @@ -355,19 +355,6 @@ interface Agent { `AgentStatus` is `'idle' | 'running' | 'disposed'`. `AgentId` is a branded string. `AgentOptions` (`model?`) is merge-extensible — plugins add creation options by declaration merging; the persona is NOT an agent option but the `dsh-system-prompt` plugin's `persona` config, shared context-wide. The `agent/*` event taxonomy (lifecycle emits incl. `agent/session-start`, serial `agent/pre-step`/`agent/turn-stop` checkpoints, and the `agent/prompt-submit`/`agent/request`/`agent/session-prefix`/`agent/step-result`/`agent/turn-continuation` waterfalls) is in [architecture.md § Event taxonomy](../architecture.md#event-taxonomy); turn/step boundaries are durable `session/event` records, not `agent/*` emits. -### `AgentRegistrationReservation` — unpublished identity ownership - -An agent factory reserves its public `AgentId` before awaiting setup, so setup code cannot publish either the intended agent or a replacement under that id ahead of the transaction. The opaque capability authorizes exactly the later `enter()` call. Its `release` function is the exact Cordis owner effect disposer, letting the lifecycle adopt it by identity and place release after scope quiescence while owner disposal remains the abandoned-transaction backstop. Ordinary plugins use `register()` and never hold this type. - -Source: [`packages/core/agent/src/index.ts`](../../packages/core/agent/src/index.ts) - -```ts type-equiv -interface AgentRegistrationReservation { - readonly id: AgentId - release(): void -} -``` - ## Interception decisions Each `agent/*` interception waterfall returns a small, seam-specific typed union — the unified Decision idiom (the tool seams' `PreToolDecision`/`PostToolDecision` in [tools.md](tools.md) follow the same shape). A CC/Codex hook bridge maps its `permissionDecision`/`decision`/`continue`/`additionalContext` fields onto these; a native plugin returns them directly. They share one envelope for model-facing context, `HookContext`, which is `inject()`ed as a `context/message` and so carries a REQUIRED `source` (a missing source would default to `{kind:'user'}` and mislabel plugin context as a user prompt). diff --git a/docs/core-data-structures/persistence.md b/docs/core-data-structures/persistence.md index 4dbb8fb2af..7bf102924b 100644 --- a/docs/core-data-structures/persistence.md +++ b/docs/core-data-structures/persistence.md @@ -25,15 +25,15 @@ interface SessionHeader { * session is created. A persistence backend rejects any other version on load * (no migration — see the constant). */ - version: number + readonly version: number /** The session's id (mirrors the {@link Session}'s id). */ - id: SessionId + readonly id: SessionId /** Unix epoch milliseconds when the session was created. */ - createdAt: number + readonly createdAt: number /** Absolute working directory the session was created in (if any). */ - cwd?: string + readonly cwd?: string /** The session this one was forked from (seed lineage), if any. */ - parentSession?: SessionId + readonly parentSession?: SessionId /** * How many leading events were INHERITED via a seed rather than produced by * this session — the seed boundary. Set when a fork seeds a child with a @@ -43,7 +43,7 @@ interface SessionHeader { * harness can skip the inherited prefix when deriving the child's OWN script * (the seeded events are the parent's, not this child's model calls). */ - seedLength?: number + readonly seedLength?: number } ``` @@ -54,7 +54,7 @@ Creating a `Session` through the store takes a `seed` (replay/fork an existing e ```ts type-equiv interface CreateSessionOptions { /** Events to seed the new session with (replay/fork). */ - seed?: SessionEvent[] + readonly seed?: readonly SessionEvent[] /** * Creation metadata. The store fills in `version`/`id` and defaults * `createdAt` to now; the caller supplies the storage-level fields (validated @@ -67,7 +67,12 @@ interface CreateSessionOptions { * length, not the original boundary — the caller must pass the persisted * boundary back. A fresh fork passes its actual seeded-prefix length. */ - meta?: { cwd?: string; parentSession?: SessionId; createdAt?: number; seedLength?: number } + readonly meta?: { + readonly cwd?: string + readonly parentSession?: SessionId + readonly createdAt?: number + readonly seedLength?: number + } } ``` diff --git a/docs/core-data-structures/scope.md b/docs/core-data-structures/scope.md index 9ccbfcc1fe..66d5f40de9 100644 --- a/docs/core-data-structures/scope.md +++ b/docs/core-data-structures/scope.md @@ -1,6 +1,6 @@ # Scoped Registration -The [scope package](../../packages/core/scope) supplies the identity and carrier vocabulary that makes one registration context mean both per-agent visibility and shared lifetime ownership. It is a library primitive rather than a Cordis service; the [agent-scope runtime-design RFC](../rfc/implemented/architecture/2026-07-12-agent-scope-runtime-design.md#scope-mechanism-context-key-and-lifetime) owns the implementation rationale, while the package [README](../../packages/core/scope/README.md) owns the callable API and filtering semantics. +The [scope package](../../packages/core/scope) supplies the identity and carrier vocabulary that makes one registration context mean both per-agent visibility and shared lifetime ownership. It is a library primitive rather than a Cordis service; the [agent-scope runtime-design RFC](../rfc/implemented/architecture/2026-07-12-agent-scope-runtime-design.md#scope-routing-one-opaque-key-selects-one-layer) owns the implementation rationale, while the package [README](../../packages/core/scope/README.md) owns the callable API and filtering semantics. Source: [`packages/core/scope/src/index.ts`](../../packages/core/scope/src/index.ts). @@ -12,10 +12,10 @@ Source: [`packages/core/scope/src/index.ts`](../../packages/core/scope/src/index type ScopeKey = object ``` -`Scoped` is the compile-time brand on the proxy returned by `scopeTarget(base, key)`. Scope-filtered event declarations require this carrier as their `this` type, preventing an ordinary subject object from type-checking as the dispatch carrier. +`Scoped` is the compile-time brand on the opaque routing receiver returned by `scopeTarget(base, key)`. Scope-filtered event declarations require this carrier as their `this` type, while the real event subject remains an explicit argument. ```ts type-equiv -type Scoped = T & { readonly [ScopedBrand]: 'dsh.scope.carrier' } +type Scoped = object & { readonly [ScopedBrand]: T } ``` ## Owned registration context diff --git a/docs/core-data-structures/session.md b/docs/core-data-structures/session.md index 3ed02b41a9..1b12c3fc48 100644 --- a/docs/core-data-structures/session.md +++ b/docs/core-data-structures/session.md @@ -196,20 +196,6 @@ export interface SurfaceNode { } ``` -## `SessionRegistrationReservation` — unpublished identity and construction ownership - -A session factory reserves its public `SessionId` before awaiting persistence load or scoped setup, so concurrent code cannot create, prepare, or enter a session under that id ahead of the transaction. The opaque capability may construct exactly one unpublished `Session` and authorizes exactly that object at `enter()`. Its `release` function is the exact Cordis owner effect disposer, letting the lifecycle adopt it by identity and place release after scope quiescence while owner disposal remains the abandoned-transaction backstop. Ordinary session consumers use `create()` and never hold this type. - -Source: [`packages/core/session/src/index.ts`](../../packages/core/session/src/index.ts) - -```ts type-equiv -interface SessionRegistrationReservation { - readonly id: SessionId - prepare(options?: CreateSessionOptions): Session - release(): void -} -``` - ## Derived history: `deriveMessages()` and `deriveEventMessage()` `Session.deriveMessages()` projects the event log into the `Message[]` the model sees — cached (each surface node projected once, when first seen; a surface rewrite rebuilds) and frozen (a fresh array per call over shared, deep-frozen messages, so mutating logged history through a projection is unrepresentable). `deriveEventMessage(event)` is the per-node pure function the fold applies — public so external reconstructors and the dev invariant project a log prefix with exactly the same rules and cannot disagree with the cache. The projection rules: diff --git a/docs/core-data-structures/skills.md b/docs/core-data-structures/skills.md index 0c5bfcae13..56ac91a6da 100644 --- a/docs/core-data-structures/skills.md +++ b/docs/core-data-structures/skills.md @@ -6,13 +6,13 @@ Source: [`packages/skill/skill/src/index.ts`](../../packages/skill/skill/src/ind ## Provider registry -`ctx.skills` is a multi-provider registry. Providers can represent local directories, embedded plugin data, HTTP catalogs, or another source. Provider plugins register synchronously during `apply()`; remote initialization, authentication, and discovery are awaited by `list()`. Each lookup snapshots its read-only options before provider work, and each provider candidate becomes registry-owned data while its opaque locator retains provider-owned identity. The registry validates candidates, resolves duplicate skill names first-wins by rank/provider order/local order, and sorts the final summaries by `name` for deterministic consumers. A provider `list()` rejection is logged and skipped without caching the degraded catalog; malformed candidates still fail fast because they violate the provider contract. +`ctx.skills` is a multi-provider registry. Providers can represent local directories, embedded plugin data, HTTP catalogs, or another source. Provider plugins register synchronously during `apply()`; remote initialization, authentication, and discovery are awaited by `list()`. Provider objects, lookup options, and candidates are readonly same-process contracts, so the registry borrows them instead of manufacturing defensive snapshots. The registry still validates semantic fields, resolves duplicate skill names first-wins by rank/provider order/local order, and sorts the final summaries by `name` for deterministic consumers. A provider `list()` rejection is logged and skipped without caching the degraded catalog; malformed candidates still fail fast because they violate the provider contract. ```ts type-equiv interface SkillProvider { - name: string - list(options: SkillLookupOptions): Promise - get(candidate: SkillCandidate, options: SkillLookupOptions): Promise + readonly name: string + readonly list: (options: SkillLookupOptions) => Promise + readonly get: (candidate: SkillCandidate, options: SkillLookupOptions) => Promise } ``` @@ -44,13 +44,13 @@ type SkillSource = 'project-dsh' | 'project-agents' | 'runtime' | 'user-dsh' | ' ```ts type-equiv interface SkillSummary { - name: string - description: string - whenToUse?: string - disableModelInvocation?: boolean - source: SkillSource - provider: string - resourceBase?: SkillResourceBase + readonly name: string + readonly description: string + readonly whenToUse?: string + readonly disableModelInvocation?: boolean + readonly source: SkillSource + readonly provider: string + readonly resourceBase?: SkillResourceBase } ``` @@ -58,10 +58,10 @@ interface SkillSummary { ```ts type-equiv interface SkillCandidate extends SkillSummary { - rank: number - locator: unknown - path?: string - metadata?: Record + readonly rank: number + readonly locator: unknown + readonly path?: string + readonly metadata?: Readonly> } ``` @@ -69,16 +69,16 @@ interface SkillCandidate extends SkillSummary { ```ts type-equiv type SkillResourceBase = - | { kind: 'directory'; path: string } - | { kind: 'url'; url: string } - | { kind: 'opaque'; description: string } + | { readonly kind: 'directory'; readonly path: string } + | { readonly kind: 'url'; readonly url: string } + | { readonly kind: 'opaque'; readonly description: string } ``` ```ts type-equiv interface SkillDefinition extends SkillSummary { - content: string - path?: string - metadata?: Record + readonly content: string + readonly path?: string + readonly metadata?: Readonly> } ``` @@ -86,13 +86,13 @@ Runtime skills use the same complete shape and participate in the same first-win ```ts type-equiv type SkillRegistration = Omit & { - provider?: string + readonly provider?: string } ``` ## Lookup and configuration -Skill lookup is cwd-sensitive because providers may expose workspace-local skills, and its optional signal cancels provider work for the caller. The registry captures both fields once and providers receive the same read-only snapshot used for cache identity and loading. Cancellation is checked before and after catalog selection, including cache hits, and races both discovery and full-definition loading. If no git root is found, the local provider treats the supplied cwd itself as the project root. +Skill lookup is cwd-sensitive because providers may expose workspace-local skills, and its optional signal cancels provider work for the caller. Providers receive the same readonly options object used for cache identity and loading. Cancellation is checked before and after catalog selection, including cache hits, and races both discovery and full-definition loading. If no git root is found, the local provider treats the supplied cwd itself as the project root. ```ts type-equiv interface SkillLookupOptions { @@ -105,7 +105,7 @@ The registry owns only its discovery-cache bound. The local provider owns filesy ```ts type-equiv interface Config { - collectCacheMaxEntries?: number + readonly collectCacheMaxEntries?: number } ``` diff --git a/docs/core-data-structures/subagent.md b/docs/core-data-structures/subagent.md index 2f3e089a22..5454f326b3 100644 --- a/docs/core-data-structures/subagent.md +++ b/docs/core-data-structures/subagent.md @@ -12,10 +12,10 @@ A provider advertises its **start-time** features on a static descriptor the ser ```ts type-equiv interface SubagentCapabilities { - outputSchema: boolean - depthLimit: boolean - toolFilter: boolean - persona: boolean + readonly outputSchema: boolean + readonly depthLimit: boolean + readonly toolFilter: boolean + readonly persona: boolean } ``` @@ -25,26 +25,28 @@ What a caller asks for when starting a subagent. The tool layer builds this from ```ts type-equiv interface SubagentStartRequest { - prompt: ContentBlock[] - parent: Agent - signal?: AbortSignal - agentOptions?: AgentOptions - outputSchema?: StructuredOutputSchema - maxDepth?: number - toolFilter?: { allow?: string[]; deny?: string[] } - persona?: string + readonly prompt: ContentBlock[] + readonly parent: Agent + readonly signal: AbortSignal + readonly agentOptions?: AgentOptions + readonly outputSchema?: StructuredOutputSchema + readonly maxDepth?: number + readonly toolFilter?: ToolRestriction + readonly persona?: string } ``` +`signal` is the single cancellation channel before and after readiness. The [subagent composition-controls RFC](../rfc/implemented/feature/2026-07-12-subagent-persona-tool-filter-and-depth.md) owns the persona, live global-tool filter, absolute-depth, and visibility-not-authority rationale. + ## The terminal result: `SubagentResult` The outcome of a run, resolved by `SubagentRun.result`. `structured` is present only after a requested `outputSchema` was successfully satisfied; requesting a schema does not guarantee it, and a provider may return `stopReason: 'error'` when the child fails or finishes without a valid capture. A non-`completed` `stopReason` means `output` may be partial — the consumer maps it to an `isError` tool result rather than reporting partial output as success. ```ts type-equiv interface SubagentResult { - output: ContentBlock[] - structured?: unknown - stopReason: SubagentStopReason + readonly output: ContentBlock[] + readonly structured?: unknown + readonly stopReason: SubagentStopReason } ``` @@ -62,17 +64,15 @@ interface SubagentStopReasonMap { ## A live run: `SubagentRun` -The handle the consumer holds while a child executes. `started` is the provider's publication boundary: it resolves only after an in-process agent is live in `ctx.agents` or a remote transport has created its child session, and rejects when the attempt fails or is cancelled before that point. The consumer normally awaits `result`, may `cancel` mid-flight, and MUST `dispose` on every path to reach child quiescence (no leaked idle child / session). `result` does NOT reject on a child-level failure — a model/transport failure resolves with `stopReason: 'error'` — so the consumer maps a non-`completed` reason to an `isError` result; it rejects only on an infrastructure fault the seam cannot represent. `sendMessage` and `resume` are OPTIONAL: a provider that supports the runtime capability defines the method; one that doesn't omits it. +The handle the consumer holds after a provider has established a ready child. The consumer awaits `result` and MUST `dispose` on every path to cancel remaining work and reach child quiescence. `result` does NOT reject on a child-level failure — a model/transport failure resolves with `stopReason: 'error'` — so the consumer maps a non-`completed` reason to an `isError` result; it rejects only on an infrastructure fault the seam cannot represent. `sendMessage` and `resume` are OPTIONAL: a provider that supports the runtime capability defines the method; one that doesn't omits it. ```ts type-equiv interface SubagentRun { readonly id: AgentId - readonly started: Promise readonly result: Promise - cancel(reason?: string): void dispose(): Promise sendMessage?(content: ContentBlock[]): void - resume?(content: ContentBlock[]): SubagentRun + resume?(content: ContentBlock[]): Promise } ``` @@ -85,15 +85,15 @@ interface SubagentProvider { readonly name: string readonly capabilities: SubagentCapabilities readonly inheritsParentContext: boolean - start(request: SubagentStartRequest): SubagentRun + start(request: SubagentStartRequest): Promise } ``` -The service (`ctx.subagents`) emits `subagent/start` only after `run.started` fulfills and emits the paired `subagent/end` when that started run settles (see the [events catalog](../cordis-catalog/events.md)); a pre-publication readiness rejection emits neither event. For an in-process provider, a start listener can therefore resolve the live child with `ctx.agents.get(info.id)`; a remote provider need not publish into the local registry. `subagent/end` carries `lastAssistantMessage` (the child's final `output`) on the settle path, so an observer sees WHAT the subagent produced without holding the run (absent when the run rejected at the infrastructure level — no result was produced). These are **observe-only** events: both are plain `emit`s, so a subscriber observes but cannot change the run. Result settlement is observed immediately even while readiness is pending, then its cloned end payload is buffered until start has been announced; this prevents an early rejection from becoming unhandled while preserving start-before-end order and protecting the caller's result from listener mutation. Both emits contain a thrown listener **per listener** (logged, never propagated): one bad subscriber can neither strand a live run, surface as an unhandled rejection on the detached settle hook, nor starve the listeners registered after it. +`SubagentProvider.start()` and `ctx.subagents.start()` are the publication boundary: their promises fulfill only with a ready run. The service attaches result observation, emits `subagent/start`, and returns the same holder-owned run; a rejected start has already cleaned provider-owned partial resources and emits neither lifecycle event. For an in-process provider, a start listener can resolve the live child with `ctx.agents.get(info.id)`; a remote provider need not publish into the local registry. `subagent/end` carries `lastAssistantMessage` (the child's final `output`) on the settle path and reports `error` on infrastructure rejection. Both lifecycle events are observe-only emits with per-listener exception containment. ## In-process backends: depth and seed -The two in-process backends ([dsh-subagent-spawn](../../packages/subagent/subagent-spawn) fresh, [dsh-subagent-fork](../../packages/subagent/subagent-fork) seeded) run the child as a child `Agent` on the same application. They synchronously snapshot caller-owned data, install provider ownership before attaching the abort listener, create one run-owner fiber under `parent.ctx`, and invoke the factory through that fiber: parent teardown, provider teardown, and manual run disposal share the same pre-publication ownership and quiescence boundary, while the child still receives a flat new scope rather than inheriting the parent's registrations. Their `started` promise projects the factory's successful publication and the result driver awaits that same promise before sending the prompt. Two pieces of vocabulary ride on the existing agent/session types rather than new core types: +The two in-process backends ([dsh-subagent-spawn](../../packages/subagent/subagent-spawn) fresh, [dsh-subagent-fork](../../packages/subagent/subagent-fork) seeded) run the child as an ordinary `Agent` in the same application. The provider creates it directly through `parent.ctx`, passes the required signal into the core creation transaction, and delegates quiescent disposal to the returned `AgentHandle`. Provider removal prevents new starts but does not revoke an accepted run. The child receives a flat new scope rather than inheriting the parent's registrations. Two pieces of vocabulary ride on the existing agent/session types rather than new core types: - **Delegation depth** is a merge-extensible `AgentOptions.subagentDepth` field (`0` for a top-level agent, parent + 1 for a child). Only `undefined` means top level; every stored present value must be a non-negative safe integer. The seam owns it — the loop neither sets nor reads it — so a nested spawn validates its parent's stored depth, rejects a derived child depth outside the safe-integer domain, and applies a defined absolute `request.maxDepth` cap to that child. - **Fork seeding** uses `CreateAgentOptions.seed` (a `SessionEvent[]` prefix threaded through `AgentLoop.createAgent` → `ctx.sessions.prepare({ seed })`, the same primitive `resume` uses). The fork backend passes a *balanced completed-turn prefix* of the parent's log — the parent's events up to and including its last `turn/end` — so the seed is contiguous-from-0 and the [invariants](../../packages/support/invariants) replay accepts it (the in-flight, unbalanced turn is excluded). diff --git a/docs/core-data-structures/system-prompt.md b/docs/core-data-structures/system-prompt.md index 0406ddb1a3..bfd34b3626 100644 --- a/docs/core-data-structures/system-prompt.md +++ b/docs/core-data-structures/system-prompt.md @@ -16,22 +16,25 @@ interface AssembleContext { ## Tool-provider result -`ToolProviderResult.schemas` is the model-visible set for the current assembly. `knownNames` is the provider's pre-restriction name universe used to distinguish a configured-name typo from a known tool that is deliberately hidden in this scope. +`ToolProviderResult.schemas` is the model-visible set for the current assembly. `knownNames` is the provider's pre-restriction name universe used to distinguish a configured-name typo from a known tool that is deliberately hidden in this scope. `ownerFinalNames` identifies tool contributions whose canonical presence or absence survives the assembly waterfall. ```ts type-equiv interface ToolProviderResult { - schemas: ToolSchema[] - knownNames?: readonly string[] + readonly schemas: readonly ToolSchema[] + readonly knownNames?: readonly string[] + readonly ownerFinalNames?: readonly string[] } ``` -## Canonical contribution protection +## Prompt sections and owner finality -`PromptProtection` names section and tool contributions whose canonical registry output remains authoritative after the assembly waterfall. Either field may be omitted, but a registration with no names is rejected. +`PromptSection` is a readonly same-process registration contract. `ownerFinal` is reserved for protocol-owned instructions whose canonical presence and definition must survive the complete assembly waterfall; ordinary sections remain transformable. Tool definitions declare the equivalent fact on their own contribution, and the tool provider reports the resolved names through `ownerFinalNames` above. ```ts type-equiv -interface PromptProtection { - sections?: readonly string[] - tools?: readonly string[] +interface PromptSection { + readonly name: string + readonly order: number + readonly text: string | ((context: AssembleContext) => string) + readonly ownerFinal?: boolean } ``` diff --git a/docs/core-data-structures/tools.md b/docs/core-data-structures/tools.md index 1766b4c9ce..4785de84a3 100644 --- a/docs/core-data-structures/tools.md +++ b/docs/core-data-structures/tools.md @@ -19,6 +19,12 @@ interface ToolDefinition extends ToolSchema { * cooperative implementation that can reach quiescence when the signal aborts. */ timeoutMs?: number + /** + * Whether this tool name's canonical wire presence or absence survives the + * complete system-prompt assembly waterfall. Reserved for protocol tools + * whose owner must retain the final definition. + */ + readonly ownerFinal?: boolean /** * Optional: how to present the PENDING state of one call in a UI, derived from * the call's `args` (parsed arguments, `unknown` — the tool validates/narrows @@ -81,16 +87,25 @@ type InferArgs = Simplify< `defineTool({ name, description, parameters, execute, … })` ties it together: `parameters` is a `SchemaSpec`, `execute(args, exec)` gets `args: InferArgs`, and the helper converts the spec to JSON Schema (`schemaSpecToJsonSchema`) for the wire and validates model-generated args (`validateArgs`) before the typed body runs. A mismatch throws `ToolArgsError` (`code: 'INVALID_ARGS'`), which the registry turns into an `isError` result so the model can self-correct. Why a custom DSL and not schemastery: tool parameters need JSON Schema (the LLM wire format), not validation/transformation — the lightweight DSL gives the best authoring DX with the smallest surface. -Registration is a value boundary. `ToolRegistry.register()` reads every top-level field once, validates fixed scalar and callback types, materializes `ToolDefinition.parameters` as detached lossless JSON in one recursive pass, binds the accepted execute/presentation callbacks once to the original definition as their method receiver, and deep-freezes the stored record. Replacing a callback property on the caller-owned definition later does not change dispatch. `get()`/`visible()` expose only that frozen snapshot, while `schemas()` produces detached projections, so the model-visible and executable views cannot drift through a leaked mutable registry object. +Registration is a trusted same-process contract. The registry borrows the typed definition as readonly input and validates only semantic requirements such as a positive finite `timeoutMs`; `schemas()` materializes the explicit model-facing projection at the model boundary so execution and presentation share one resolved definition without leaking callbacks onto the wire. + +## `ToolRestriction` — one scope's live global filter + +`ToolRestriction` applies only to the live deployment-global tool layer. The registry compiles readonly names into private sets, intersects multiple restrictions, then overlays scope-local tools. A deny-only filter admits later unlisted globals, while an allow-list excludes them. + +```ts type-equiv +interface ToolRestriction { + readonly allow?: readonly string[] + readonly deny?: readonly string[] +} +``` ## Execution: extensible waterfalls plus monotonic policy -`ctx.tools.execute()` accepts a caller-owned `ToolExecutionInput`, snapshots it into a pipeline-owned `ToolExecution`, and runs that call through `tools/pre-execute` (the reorderable allow/deny/ask waterfall) → registered monotonic guards → `tools/execute` (around-dispatch wrappers) → `tools/post-execute` (inspect/replace the result) → `tools/result` (the immutable authoritative outcome). The outcome is a `ToolExecutionResult`. +`ctx.tools.execute()` accepts a caller-owned `ToolExecutionInput`, materializes its parsed JSON arguments once into a pipeline-owned `ToolExecution`, and runs that call through `tools/pre-execute` (the reorderable allow/deny/ask waterfall) → registered monotonic guards → `tools/execute` (around-dispatch wrappers) → `tools/post-execute` (inspect/replace the result) → `tools/result` (the immutable authoritative outcome). The outcome is a `ToolExecutionResult`. ```ts type-equiv -interface ToolExecutionToken { - readonly [toolExecutionTokenBrand]: true -} +type ToolExecutionToken = symbol & { readonly [toolExecutionTokenBrand]: true } ``` ```ts type-equiv @@ -118,7 +133,7 @@ interface ToolExecution extends ToolExecutionInput { } ``` -`ToolExecutionToken` is a compile-time opaque type and a frozen, property-free object at runtime; identity comparison is its only operation. Before policy runs, `ctx.tools.execute()` reads each caller-owned field once, materializes `arguments` as detached lossless JSON in one recursive pass, assigns a fresh token, and deep-freezes the accepted arguments. A mutable exotic such as `Map` is rejected and normalized to an error before policy; one-pass materialization prevents a stateful getter from supplying different values to validation and storage. `token`, `callId`, `name`, `arguments`, `agent`, and the optional `parent` token are non-writable throughout all waterfalls, so a listener cannot change which tool or scope the pipeline accepted or reach a live enclosing execution; an around-dispatch wrapper may add, replace, or remove only optional `signal`. After the complete pipeline the registry freezes the execution and exposes its stable identity to `tools/result` observers, where the execution remains usable as a `WeakMap` key without mutation races. +`ToolExecutionToken` is a compile-time opaque fresh `Symbol` at runtime; identity comparison is its only operation. Before policy runs, `ctx.tools.execute()` materializes `arguments` as detached lossless JSON, assigns the token, and deep-freezes the accepted arguments. A non-JSON value is normalized to an error before policy. `token`, `callId`, `name`, `arguments`, `agent`, and the optional `parent` token are readonly throughout the waterfalls, while an around-dispatch wrapper may add, replace, or remove only optional `signal`. After the complete pipeline the registry freezes the execution and exposes its stable identity to `tools/result` observers. A `ToolGuard` is scope-aware final pre-dispatch policy. Its shape deliberately has no allow result: `undefined` preserves the waterfall decision, while a returned reason can only reduce permission, so a later listener cannot undo it. @@ -158,7 +173,7 @@ interface ToolExecutionResult { } ``` -The registry rebuilds and validates the complete authoritative result after post-policy. Its content, structured error, additional context, and presentation metadata must round-trip losslessly through JSON; a malformed or non-JSON value becomes a JSON-safe `isError` result before `tools/result` observers run, so the live outcome is always safe for the later durable `tool/result` append. +The registry materializes and freezes the final accepted result immediately before `tools/result`. Its content, structured error, additional context, and presentation metadata must round-trip losslessly through JSON; an invalid outcome becomes a JSON-safe `isError` result, so the observed live outcome is safe for the later durable `tool/result` append. Each interception waterfall returns a typed **Decision** (the idiom shared with the `agent/*` seams). `tools/pre-execute` listeners receive `(exec, next)` and return a `PreToolDecision`; `tools/execute` wrappers return a `ToolExecutionResult`; `tools/post-execute` listeners receive `(exec, result, next)` and return a `PostToolDecision`: diff --git a/docs/event-producer-consumer.md b/docs/event-producer-consumer.md index 4f2aec7b44..aac25e71b9 100644 --- a/docs/event-producer-consumer.md +++ b/docs/event-producer-consumer.md @@ -9,18 +9,18 @@ This matrix shows which packages dispatch each harness-owned event and which pac | --- | --- | --- | --- | --- | | `agent/created` | `emit` | [`packages/core/agent/src/types.ts:316`](../packages/core/agent/src/types.ts) | [`agent`](../packages/core/agent) (`events.dispatch`) | [`stdio-agent`](../packages/ui/stdio-agent) | | `agent/disposed` | `emit` | [`packages/core/agent/src/types.ts:331`](../packages/core/agent/src/types.ts) | [`agent`](../packages/core/agent) (`events.dispatch`) | [`stdio-agent`](../packages/ui/stdio-agent) | -| `agent/error` | `emit` | [`packages/core/agent/src/types.ts:608`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`emit`) | - | -| `agent/pre-step` | `serial` | [`packages/core/agent/src/types.ts:440`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`serial`) | [`compact-basic`](../packages/compact/compact-basic), [`user-approval`](../packages/ui/user-approval) | -| `agent/prompt-submit` | `waterfall` | [`packages/core/agent/src/types.ts:458`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`waterfall`) | [`acp`](../packages/ui/acp), [`hooks-claude`](../packages/hooks/hooks-claude), [`hooks-codex`](../packages/hooks/hooks-codex), [`repeat-tool-guard`](../packages/guard/repeat-tool-guard) | -| `agent/queued` | `emit` | [`packages/core/agent/src/types.ts:362`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`emit`) | - | -| `agent/request` | `waterfall` | [`packages/core/agent/src/types.ts:487`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`waterfall`) | - | -| `agent/session-prefix` | `waterfall` | [`packages/core/agent/src/types.ts:539`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`waterfall`) | [`tool-skill`](../packages/skill/tool-skill) | -| `agent/session-start` | `emit` | [`packages/core/agent/src/types.ts:383`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`emit`) | [`hooks-claude`](../packages/hooks/hooks-claude), [`hooks-codex`](../packages/hooks/hooks-codex), [`invariants`](../packages/support/invariants) | -| `agent/status` | `emit` | [`packages/core/agent/src/types.ts:345`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`emit`) | [`acp`](../packages/ui/acp), [`invariants`](../packages/support/invariants), [`repeat-tool-guard`](../packages/guard/repeat-tool-guard), [`stdio-agent`](../packages/ui/stdio-agent) | -| `agent/step-result` | `waterfall` | [`packages/core/agent/src/types.ts:554`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`waterfall`) | - | -| `agent/turn-continuation` | `waterfall` | [`packages/core/agent/src/types.ts:572`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`waterfall`) | [`hooks-claude`](../packages/hooks/hooks-claude), [`hooks-codex`](../packages/hooks/hooks-codex) | -| `agent/turn-stop` | `serial` | [`packages/core/agent/src/types.ts:591`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`strictSerial (serial)`) | [`subagent-inprocess`](../packages/subagent/subagent-inprocess) | -| `approval/request` | `waterfall` | [`packages/ui/user-approval/src/index.ts:72`](../packages/ui/user-approval/src/index.ts) | [`user-approval`](../packages/ui/user-approval) (`waterfall`) | [`acp`](../packages/ui/acp) | +| `agent/error` | `emit` | [`packages/core/agent/src/types.ts:606`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`emit`) | - | +| `agent/pre-step` | `serial` | [`packages/core/agent/src/types.ts:438`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`serial`) | [`compact-basic`](../packages/compact/compact-basic), [`user-approval`](../packages/ui/user-approval) | +| `agent/prompt-submit` | `waterfall` | [`packages/core/agent/src/types.ts:456`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`waterfall`) | [`acp`](../packages/ui/acp), [`hooks-claude`](../packages/hooks/hooks-claude), [`hooks-codex`](../packages/hooks/hooks-codex), [`repeat-tool-guard`](../packages/guard/repeat-tool-guard) | +| `agent/queued` | `emit` | [`packages/core/agent/src/types.ts:360`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`emit`) | - | +| `agent/request` | `waterfall` | [`packages/core/agent/src/types.ts:485`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`waterfall`) | - | +| `agent/session-prefix` | `waterfall` | [`packages/core/agent/src/types.ts:537`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`waterfall`) | [`tool-skill`](../packages/skill/tool-skill) | +| `agent/session-start` | `emit` | [`packages/core/agent/src/types.ts:381`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`emit`) | [`hooks-claude`](../packages/hooks/hooks-claude), [`hooks-codex`](../packages/hooks/hooks-codex), [`invariants`](../packages/support/invariants) | +| `agent/status` | `emit` | [`packages/core/agent/src/types.ts:345`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`emit`) | [`invariants`](../packages/support/invariants), [`repeat-tool-guard`](../packages/guard/repeat-tool-guard), [`stdio-agent`](../packages/ui/stdio-agent) | +| `agent/step-result` | `waterfall` | [`packages/core/agent/src/types.ts:552`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`waterfall`) | - | +| `agent/turn-continuation` | `waterfall` | [`packages/core/agent/src/types.ts:570`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`waterfall`) | [`hooks-claude`](../packages/hooks/hooks-claude), [`hooks-codex`](../packages/hooks/hooks-codex) | +| `agent/turn-stop` | `serial` | [`packages/core/agent/src/types.ts:589`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`serial`) | [`subagent-inprocess`](../packages/subagent/subagent-inprocess) | +| `approval/request` | `waterfall` | [`packages/ui/user-approval/src/index.ts:70`](../packages/ui/user-approval/src/index.ts) | [`user-approval`](../packages/ui/user-approval) (`waterfall`) | [`acp`](../packages/ui/acp) | | `fs/edit-intent` | `waterfall` | [`packages/fs/fs/src/index.ts:123`](../packages/fs/fs/src/index.ts) | [`tool-fs`](../packages/fs/tool-fs) (`waterfall`) | [`fs-policy`](../packages/fs/fs-policy) | | `fs/observed` | `emit` | [`packages/fs/fs/src/index.ts:138`](../packages/fs/fs/src/index.ts) | [`tool-fs`](../packages/fs/tool-fs) (`emit`) | [`fs-policy`](../packages/fs/fs-policy) | | `fs/write-intent` | `waterfall` | [`packages/fs/fs/src/index.ts:109`](../packages/fs/fs/src/index.ts) | [`tool-fs`](../packages/fs/tool-fs) (`waterfall`) | [`fs-policy`](../packages/fs/fs-policy) | @@ -29,25 +29,25 @@ This matrix shows which packages dispatch each harness-owned event and which pac | `session/disposed` | `emit` | [`packages/core/session/src/index.ts:64`](../packages/core/session/src/index.ts) | [`session`](../packages/core/session) (`events.dispatch`) | - | | `session/event` | `emit` | [`packages/core/session/src/index.ts:83`](../packages/core/session/src/index.ts) | [`session`](../packages/core/session) (`events.dispatch`) | [`acp`](../packages/ui/acp), [`invariants`](../packages/support/invariants), [`session-persistence`](../packages/session-persistence/session-persistence), [`stdio-agent`](../packages/ui/stdio-agent) | | `session/flush` | `parallel` | [`packages/core/session/src/index.ts:101`](../packages/core/session/src/index.ts) | [`session`](../packages/core/session) (`events.dispatch`) | [`session-persistence`](../packages/session-persistence/session-persistence) | -| `skill/provider-added` | `emit` | [`packages/skill/skill/src/index.ts:132`](../packages/skill/skill/src/index.ts) | [`skill`](../packages/skill/skill) (`emit`) | - | -| `skill/provider-removed` | `emit` | [`packages/skill/skill/src/index.ts:138`](../packages/skill/skill/src/index.ts) | [`skill`](../packages/skill/skill) (`emit`) | - | -| `subagent/end` | `emit` | [`packages/subagent/subagent/src/index.ts:134`](../packages/subagent/subagent/src/index.ts) | [`subagent`](../packages/subagent/subagent) (`events.dispatch`) | [`hooks-claude`](../packages/hooks/hooks-claude) | -| `subagent/provider-added` | `emit` | [`packages/subagent/subagent/src/index.ts:95`](../packages/subagent/subagent/src/index.ts) | [`subagent`](../packages/subagent/subagent) (`emit`) | [`tool-subagent`](../packages/subagent/tool-subagent) | -| `subagent/provider-removed` | `emit` | [`packages/subagent/subagent/src/index.ts:106`](../packages/subagent/subagent/src/index.ts) | [`subagent`](../packages/subagent/subagent) (`events.dispatch`) | [`tool-subagent`](../packages/subagent/tool-subagent) | -| `subagent/start` | `emit` | [`packages/subagent/subagent/src/index.ts:121`](../packages/subagent/subagent/src/index.ts) | [`subagent`](../packages/subagent/subagent) (`events.dispatch`) | [`hooks-claude`](../packages/hooks/hooks-claude) | -| `system-prompt/assemble` | `waterfall` | [`packages/core/system-prompt/src/index.ts:46`](../packages/core/system-prompt/src/index.ts) | [`system-prompt`](../packages/core/system-prompt) (`waterfall`) | - | -| `system-prompt/change` | `emit` | [`packages/core/system-prompt/src/index.ts:56`](../packages/core/system-prompt/src/index.ts) | [`system-prompt`](../packages/core/system-prompt) (`emit`) | - | -| `tools/change` | `emit` | [`packages/core/tools/src/index.ts:176`](../packages/core/tools/src/index.ts) | [`tools`](../packages/core/tools) (`emit`) | - | -| `tools/execute` | `waterfall` | [`packages/core/tools/src/index.ts:131`](../packages/core/tools/src/index.ts) | [`tools`](../packages/core/tools) (`waterfall`) | [`timeout-policy`](../packages/timeout/timeout-policy) | -| `tools/post-execute` | `waterfall` | [`packages/core/tools/src/index.ts:151`](../packages/core/tools/src/index.ts) | [`tools`](../packages/core/tools) (`waterfall`) | [`hooks-claude`](../packages/hooks/hooks-claude), [`hooks-codex`](../packages/hooks/hooks-codex), [`repeat-tool-guard`](../packages/guard/repeat-tool-guard) | -| `tools/pre-execute` | `waterfall` | [`packages/core/tools/src/index.ts:104`](../packages/core/tools/src/index.ts) | [`tools`](../packages/core/tools) (`waterfall`) | [`hooks-claude`](../packages/hooks/hooks-claude), [`hooks-codex`](../packages/hooks/hooks-codex) | -| `tools/result` | `parallel` | [`packages/core/tools/src/index.ts:166`](../packages/core/tools/src/index.ts) | [`tools`](../packages/core/tools) (`events.dispatch`) | [`subagent-inprocess`](../packages/subagent/subagent-inprocess) | -| `workflow/agent-end` | `emit` | [`packages/workflow/workflow/src/index.ts:98`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | -| `workflow/agent-start` | `emit` | [`packages/workflow/workflow/src/index.ts:87`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | -| `workflow/end` | `emit` | [`packages/workflow/workflow/src/index.ts:108`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | -| `workflow/log` | `emit` | [`packages/workflow/workflow/src/index.ts:77`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | -| `workflow/phase` | `emit` | [`packages/workflow/workflow/src/index.ts:70`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | -| `workflow/start` | `emit` | [`packages/workflow/workflow/src/index.ts:62`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | +| `skill/provider-added` | `emit` | [`packages/skill/skill/src/index.ts:131`](../packages/skill/skill/src/index.ts) | [`skill`](../packages/skill/skill) (`emit`) | - | +| `skill/provider-removed` | `emit` | [`packages/skill/skill/src/index.ts:137`](../packages/skill/skill/src/index.ts) | [`skill`](../packages/skill/skill) (`emit`) | - | +| `subagent/end` | `emit` | [`packages/subagent/subagent/src/index.ts:90`](../packages/subagent/subagent/src/index.ts) | [`subagent`](../packages/subagent/subagent) (`events.dispatch`) | [`hooks-claude`](../packages/hooks/hooks-claude) | +| `subagent/provider-added` | `emit` | [`packages/subagent/subagent/src/index.ts:66`](../packages/subagent/subagent/src/index.ts) | [`subagent`](../packages/subagent/subagent) (`emit`) | [`tool-subagent`](../packages/subagent/tool-subagent) | +| `subagent/provider-removed` | `emit` | [`packages/subagent/subagent/src/index.ts:72`](../packages/subagent/subagent/src/index.ts) | [`subagent`](../packages/subagent/subagent) (`events.dispatch`) | [`tool-subagent`](../packages/subagent/tool-subagent) | +| `subagent/start` | `emit` | [`packages/subagent/subagent/src/index.ts:82`](../packages/subagent/subagent/src/index.ts) | [`subagent`](../packages/subagent/subagent) (`events.dispatch`) | [`hooks-claude`](../packages/hooks/hooks-claude) | +| `system-prompt/assemble` | `waterfall` | [`packages/core/system-prompt/src/index.ts:45`](../packages/core/system-prompt/src/index.ts) | [`system-prompt`](../packages/core/system-prompt) (`waterfall`) | - | +| `system-prompt/change` | `emit` | [`packages/core/system-prompt/src/index.ts:55`](../packages/core/system-prompt/src/index.ts) | [`system-prompt`](../packages/core/system-prompt) (`emit`) | - | +| `tools/change` | `emit` | [`packages/core/tools/src/index.ts:173`](../packages/core/tools/src/index.ts) | [`tools`](../packages/core/tools) (`emit`) | - | +| `tools/execute` | `waterfall` | [`packages/core/tools/src/index.ts:128`](../packages/core/tools/src/index.ts) | [`tools`](../packages/core/tools) (`waterfall`) | [`timeout-policy`](../packages/timeout/timeout-policy) | +| `tools/post-execute` | `waterfall` | [`packages/core/tools/src/index.ts:148`](../packages/core/tools/src/index.ts) | [`tools`](../packages/core/tools) (`waterfall`) | [`hooks-claude`](../packages/hooks/hooks-claude), [`hooks-codex`](../packages/hooks/hooks-codex), [`repeat-tool-guard`](../packages/guard/repeat-tool-guard) | +| `tools/pre-execute` | `waterfall` | [`packages/core/tools/src/index.ts:101`](../packages/core/tools/src/index.ts) | [`tools`](../packages/core/tools) (`waterfall`) | [`hooks-claude`](../packages/hooks/hooks-claude), [`hooks-codex`](../packages/hooks/hooks-codex) | +| `tools/result` | `parallel` | [`packages/core/tools/src/index.ts:163`](../packages/core/tools/src/index.ts) | [`tools`](../packages/core/tools) (`events.dispatch`) | [`subagent-inprocess`](../packages/subagent/subagent-inprocess) | +| `workflow/agent-end` | `emit` | [`packages/workflow/workflow/src/index.ts:96`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | +| `workflow/agent-start` | `emit` | [`packages/workflow/workflow/src/index.ts:85`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | +| `workflow/end` | `emit` | [`packages/workflow/workflow/src/index.ts:106`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | +| `workflow/log` | `emit` | [`packages/workflow/workflow/src/index.ts:75`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | +| `workflow/phase` | `emit` | [`packages/workflow/workflow/src/index.ts:68`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | +| `workflow/start` | `emit` | [`packages/workflow/workflow/src/index.ts:60`](../packages/workflow/workflow/src/index.ts) | [`workflow`](../packages/workflow/workflow) (`events.dispatch`) | - | ## Non-harness or undeclared event strings seen in package source diff --git a/CONTEXT.md b/docs/glossary.md similarity index 78% rename from CONTEXT.md rename to docs/glossary.md index 2b3584aff4..81b9b4f84a 100644 --- a/CONTEXT.md +++ b/docs/glossary.md @@ -1,10 +1,12 @@ -# Context glossary +# Glossary -Domain vocabulary for the DeepSeek Harness SDK — one canonical term per concept. Terms link with `[[name]]`; implementation detail stays in the package READMEs and RFCs. +Domain vocabulary for the DeepSeek Harness SDK uses one canonical term per concept. Terms link to their entries with standard Markdown anchors; implementation detail stays in package READMEs and RFCs. + +FIXME(glossary-completeness): Expand this glossary before the first release so it covers the SDK's other core and capability subsystems, not only agent scope. ## agent-scope -- **scope** — the unit of per-agent registration: a contribution (tool, prompt section, variable, restriction, listener) is either *global* (visible to every agent) or *scoped* (owned by exactly one [[scope-key]]). Two levels, flat: scoped registrations do not inherit down to subagents; subtree behavior is expressed with [[lineage]] data, never scope structure. +- **scope** — the unit of per-agent registration: a contribution (tool, prompt section, variable, restriction, listener) is either *global* (visible to every agent) or *scoped* (owned by exactly one [scope key](#scope-key)). Two levels, flat: scoped registrations do not inherit down to subagents; subtree behavior is expressed with [lineage](#lineage) data, never scope structure. - **scope key** — the opaque identity a scope is keyed by, compared by object identity. The harness convention: a live agent is the key of its own scope. - **agent context (`agent.ctx`)** — the agent's scoped context; registrations through it are scope-visible AND scope-lifetime (one fact drives both), and listeners on it participate in that agent's scope-filtered dispatches. Registry-subject events may remain deliberately unfiltered under their own event contracts. - **scope carrier** — the `thisArg` a scope-filtered dispatch carries (built by `scopeTarget`); its filter admits untagged listeners plus the subject's own. A *subject-less* carrier (no key) admits untagged listeners only. diff --git a/docs/module-graph.md b/docs/module-graph.md index 8d0993b59a..2acee6ffc5 100644 --- a/docs/module-graph.md +++ b/docs/module-graph.md @@ -120,13 +120,17 @@ flowchart TD pkg_session --> pkg_brand pkg_session --> pkg_llm pkg_session --> pkg_scope + pkg_system_prompt --> pkg_llm + pkg_system_prompt --> pkg_scope pkg_fs --> pkg_brand pkg_fs --> pkg_llm pkg_web --> pkg_llm pkg_sandbox --> pkg_llm - pkg_system_prompt --> pkg_llm - pkg_system_prompt --> pkg_scope - pkg_system_prompt --> pkg_session + pkg_agent --> pkg_brand + pkg_agent --> pkg_llm + pkg_agent --> pkg_scope + pkg_agent --> pkg_session + pkg_agent --> pkg_system_prompt pkg_bash --> pkg_brand pkg_bash --> pkg_sandbox pkg_bash --> pkg_session @@ -146,26 +150,22 @@ flowchart TD pkg_llm_replay --> pkg_session pkg_sandbox_local --> pkg_llm pkg_sandbox_local --> pkg_sandbox - pkg_agent --> pkg_brand - pkg_agent --> pkg_llm - pkg_agent --> pkg_scope - pkg_agent --> pkg_session - pkg_agent --> pkg_system_prompt pkg_bash_local --> pkg_bash pkg_bash_local --> pkg_timeout + pkg_compact_basic --> pkg_agent + pkg_compact_basic --> pkg_compact + pkg_compact_basic --> pkg_llm + pkg_compact_basic --> pkg_session pkg_hook_protocol --> pkg_bash pkg_hook_protocol --> pkg_session pkg_session_persistence_jsonl --> pkg_session pkg_session_persistence_jsonl --> pkg_session_persistence pkg_session_persistence_sqlite --> pkg_session pkg_session_persistence_sqlite --> pkg_session_persistence - pkg_bash_sandbox --> pkg_bash - pkg_bash_sandbox --> pkg_bash_local - pkg_bash_sandbox --> pkg_sandbox - pkg_compact_basic --> pkg_agent - pkg_compact_basic --> pkg_compact - pkg_compact_basic --> pkg_llm - pkg_compact_basic --> pkg_session + pkg_invariants --> pkg_agent + pkg_invariants --> pkg_llm + pkg_invariants --> pkg_scope + pkg_invariants --> pkg_session pkg_user_approval --> pkg_agent pkg_user_approval --> pkg_brand pkg_user_approval --> pkg_llm @@ -184,6 +184,9 @@ flowchart TD pkg_tools --> pkg_session pkg_tools --> pkg_system_prompt pkg_tools --> pkg_user_approval + pkg_bash_sandbox --> pkg_bash + pkg_bash_sandbox --> pkg_bash_local + pkg_bash_sandbox --> pkg_sandbox pkg_agent_loop --> pkg_agent pkg_agent_loop --> pkg_llm pkg_agent_loop --> pkg_scope @@ -210,7 +213,6 @@ flowchart TD pkg_subagent --> pkg_agent pkg_subagent --> pkg_llm pkg_subagent --> pkg_scope - pkg_subagent --> pkg_session pkg_subagent --> pkg_tools pkg_tool_web --> pkg_llm pkg_tool_web --> pkg_system_prompt @@ -229,12 +231,6 @@ flowchart TD pkg_hooks_codex --> pkg_llm pkg_hooks_codex --> pkg_session pkg_hooks_codex --> pkg_tools - pkg_invariants --> pkg_agent - pkg_invariants --> pkg_llm - pkg_invariants --> pkg_scope - pkg_invariants --> pkg_session - pkg_invariants --> pkg_system_prompt - pkg_invariants --> pkg_tools pkg_acp --> pkg_agent pkg_acp --> pkg_bash pkg_acp --> pkg_llm @@ -333,10 +329,11 @@ flowchart TD | [`llm-deepseek`](../packages/llm/llm-deepseek) | `llm` | [`llm`](../packages/llm/llm) | | [`llm-pi-ai`](../packages/llm/llm-pi-ai) | `llm` | [`llm`](../packages/llm/llm) | | [`session`](../packages/core/session) | `core` | [`brand`](../packages/util/brand), [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope) | +| [`system-prompt`](../packages/core/system-prompt) | `core` | [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope) | | [`fs`](../packages/fs/fs) | `fs` | [`brand`](../packages/util/brand), [`llm`](../packages/llm/llm) | | [`web`](../packages/web/web) | `web` | [`llm`](../packages/llm/llm) | | [`sandbox`](../packages/sandbox/sandbox) | `sandbox` | [`llm`](../packages/llm/llm) | -| [`system-prompt`](../packages/core/system-prompt) | `core` | [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope), [`session`](../packages/core/session) | +| [`agent`](../packages/core/agent) | `core` | [`brand`](../packages/util/brand), [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope), [`session`](../packages/core/session), [`system-prompt`](../packages/core/system-prompt) | | [`bash`](../packages/bash/bash) | `bash` | [`brand`](../packages/util/brand), [`sandbox`](../packages/sandbox/sandbox), [`session`](../packages/core/session) | | [`fs-local`](../packages/fs/fs-local) | `fs` | [`fs`](../packages/fs/fs) | | [`fs-policy`](../packages/fs/fs-policy) | `fs` | [`fs`](../packages/fs/fs) | @@ -349,28 +346,27 @@ flowchart TD | [`session-persistence`](../packages/session-persistence/session-persistence) | `session-persistence` | [`session`](../packages/core/session) | | [`llm-replay`](../packages/support/llm-replay) | `support` | [`llm`](../packages/llm/llm), [`session`](../packages/core/session) | | [`sandbox-local`](../packages/sandbox/sandbox-local) | `sandbox` | [`llm`](../packages/llm/llm), [`sandbox`](../packages/sandbox/sandbox) | -| [`agent`](../packages/core/agent) | `core` | [`brand`](../packages/util/brand), [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope), [`session`](../packages/core/session), [`system-prompt`](../packages/core/system-prompt) | | [`bash-local`](../packages/bash/bash-local) | `bash` | [`bash`](../packages/bash/bash), [`timeout`](../packages/util/timeout) | +| [`compact-basic`](../packages/compact/compact-basic) | `compact` | [`agent`](../packages/core/agent), [`compact`](../packages/compact/compact), [`llm`](../packages/llm/llm), [`session`](../packages/core/session) | | [`hook-protocol`](../packages/hooks/hook-protocol) | `hooks` | [`bash`](../packages/bash/bash), [`session`](../packages/core/session) | | [`session-persistence-jsonl`](../packages/session-persistence/session-persistence-jsonl) | `session-persistence` | [`session`](../packages/core/session), [`session-persistence`](../packages/session-persistence/session-persistence) | | [`session-persistence-sqlite`](../packages/session-persistence/session-persistence-sqlite) | `session-persistence` | [`session`](../packages/core/session), [`session-persistence`](../packages/session-persistence/session-persistence) | -| [`bash-sandbox`](../packages/bash/bash-sandbox) | `bash` | [`bash`](../packages/bash/bash), [`bash-local`](../packages/bash/bash-local), [`sandbox`](../packages/sandbox/sandbox) | -| [`compact-basic`](../packages/compact/compact-basic) | `compact` | [`agent`](../packages/core/agent), [`compact`](../packages/compact/compact), [`llm`](../packages/llm/llm), [`session`](../packages/core/session) | +| [`invariants`](../packages/support/invariants) | `support` | [`agent`](../packages/core/agent), [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope), [`session`](../packages/core/session) | | [`user-approval`](../packages/ui/user-approval) | `ui` | [`agent`](../packages/core/agent), [`brand`](../packages/util/brand), [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope), [`session`](../packages/core/session), [`system-prompt`](../packages/core/system-prompt) | | [`user-interaction`](../packages/ui/user-interaction) | `ui` | [`agent`](../packages/core/agent), [`llm`](../packages/llm/llm) | | [`workflow`](../packages/workflow/workflow) | `workflow` | [`agent`](../packages/core/agent), [`brand`](../packages/util/brand), [`llm`](../packages/llm/llm) | | [`tools`](../packages/core/tools) | `core` | [`agent`](../packages/core/agent), [`code-runtime`](../packages/code-runtime/code-runtime), [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope), [`session`](../packages/core/session), [`system-prompt`](../packages/core/system-prompt), [`user-approval`](../packages/ui/user-approval) | +| [`bash-sandbox`](../packages/bash/bash-sandbox) | `bash` | [`bash`](../packages/bash/bash), [`bash-local`](../packages/bash/bash-local), [`sandbox`](../packages/sandbox/sandbox) | | [`agent-loop`](../packages/core/agent-loop) | `core` | [`agent`](../packages/core/agent), [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope), [`session`](../packages/core/session), [`session-persistence`](../packages/session-persistence/session-persistence), [`system-prompt`](../packages/core/system-prompt), [`tools`](../packages/core/tools) | | [`tool-bash`](../packages/bash/tool-bash) | `bash` | [`agent`](../packages/core/agent), [`bash`](../packages/bash/bash), [`llm`](../packages/llm/llm), [`sandbox`](../packages/sandbox/sandbox), [`system-prompt`](../packages/core/system-prompt), [`tools`](../packages/core/tools), [`user-approval`](../packages/ui/user-approval) | | [`tool-fs`](../packages/fs/tool-fs) | `fs` | [`fs`](../packages/fs/fs), [`llm`](../packages/llm/llm), [`session`](../packages/core/session), [`system-prompt`](../packages/core/system-prompt), [`tools`](../packages/core/tools) | | [`tool-skill`](../packages/skill/tool-skill) | `skill` | [`agent`](../packages/core/agent), [`llm`](../packages/llm/llm), [`skill`](../packages/skill/skill), [`tools`](../packages/core/tools) | -| [`subagent`](../packages/subagent/subagent) | `subagent` | [`agent`](../packages/core/agent), [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope), [`session`](../packages/core/session), [`tools`](../packages/core/tools) | +| [`subagent`](../packages/subagent/subagent) | `subagent` | [`agent`](../packages/core/agent), [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope), [`tools`](../packages/core/tools) | | [`tool-web`](../packages/web/tool-web) | `web` | [`llm`](../packages/llm/llm), [`system-prompt`](../packages/core/system-prompt), [`tools`](../packages/core/tools), [`web`](../packages/web/web) | | [`timeout-policy`](../packages/timeout/timeout-policy) | `timeout` | [`llm`](../packages/llm/llm), [`timeout`](../packages/util/timeout), [`tools`](../packages/core/tools) | | [`tool-todo`](../packages/todo/tool-todo) | `todo` | [`agent`](../packages/core/agent), [`session`](../packages/core/session), [`tools`](../packages/core/tools) | | [`tool-cordis`](../packages/cordis/tool-cordis) | `cordis` | [`scope`](../packages/core/scope), [`tools`](../packages/core/tools) | | [`hooks-codex`](../packages/hooks/hooks-codex) | `hooks` | [`agent`](../packages/core/agent), [`hook-protocol`](../packages/hooks/hook-protocol), [`llm`](../packages/llm/llm), [`session`](../packages/core/session), [`tools`](../packages/core/tools) | -| [`invariants`](../packages/support/invariants) | `support` | [`agent`](../packages/core/agent), [`llm`](../packages/llm/llm), [`scope`](../packages/core/scope), [`session`](../packages/core/session), [`system-prompt`](../packages/core/system-prompt), [`tools`](../packages/core/tools) | | [`acp`](../packages/ui/acp) | `ui` | [`agent`](../packages/core/agent), [`bash`](../packages/bash/bash), [`llm`](../packages/llm/llm), [`sandbox`](../packages/sandbox/sandbox), [`session`](../packages/core/session), [`session-persistence`](../packages/session-persistence/session-persistence), [`tools`](../packages/core/tools), [`user-approval`](../packages/ui/user-approval), [`user-interaction`](../packages/ui/user-interaction) | | [`tool-ask-user`](../packages/ui/tool-ask-user) | `ui` | [`agent`](../packages/core/agent), [`tools`](../packages/core/tools), [`user-interaction`](../packages/ui/user-interaction) | | [`repeat-tool-guard`](../packages/guard/repeat-tool-guard) | `guard` | [`agent`](../packages/core/agent), [`tools`](../packages/core/tools) | diff --git a/docs/persistence-catalog.md b/docs/persistence-catalog.md index cb0e8ce0dc..245d25fa5f 100644 --- a/docs/persistence-catalog.md +++ b/docs/persistence-catalog.md @@ -23,7 +23,7 @@ An approval question was put to the answerer chain — log-only audit (like `hoo Types: [CallId](core-data-structures/core.md) -Source: [`packages/ui/user-approval/src/index.ts:86`](../packages/ui/user-approval/src/index.ts) +Source: [`packages/ui/user-approval/src/index.ts:84`](../packages/ui/user-approval/src/index.ts) #### `approval/decided` — log-only @@ -33,7 +33,7 @@ The outcome of a prior `approval/asked` (same `id`) — log-only audit. Exactly 'approval/decided': { id: ApprovalRequestId; outcome: ApprovalOutcome } ``` -Source: [`packages/ui/user-approval/src/index.ts:97`](../packages/ui/user-approval/src/index.ts) +Source: [`packages/ui/user-approval/src/index.ts:95`](../packages/ui/user-approval/src/index.ts) #### `approval/policy` — log-only @@ -43,7 +43,7 @@ The session's approval policy was switched — log-only, durable, replayable, ne 'approval/policy': { policy: ApprovalPolicy } ``` -Source: [`packages/ui/user-approval/src/index.ts:109`](../packages/ui/user-approval/src/index.ts) +Source: [`packages/ui/user-approval/src/index.ts:107`](../packages/ui/user-approval/src/index.ts) ### `assistant/*` @@ -57,7 +57,7 @@ Raw stream chunk — token-level replay fidelity. Types: [StreamChunk](core-data-structures/llm-streaming.md) -Source: [`packages/core/session/src/types.ts:317`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:322`](../packages/core/session/src/types.ts) #### `assistant/message` — surface @@ -69,7 +69,7 @@ Assembled assistant message for one step (derived history uses this). Carries th Types: [ContentBlock](core-data-structures/core.md) · [TokenUsage](core-data-structures/llm-streaming.md) -Source: [`packages/core/session/src/types.ts:324`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:329`](../packages/core/session/src/types.ts) ### `bash/*` @@ -129,7 +129,7 @@ In-session context injection (file-change notices, subdir AGENTS.md, skill conte Types: [ContentBlock](core-data-structures/core.md) · [MessageSource](core-data-structures/core.md) -Source: [`packages/core/session/src/types.ts:315`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:320`](../packages/core/session/src/types.ts) ### `hook/*` @@ -165,7 +165,7 @@ A queued prompt an `agent/prompt-submit` listener VETOED — the durable record Types: [ContentBlock](core-data-structures/core.md) · [MessageSource](core-data-structures/core.md) -Source: [`packages/core/session/src/types.ts:309`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:314`](../packages/core/session/src/types.ts) ### `request/*` @@ -177,7 +177,7 @@ Full snapshot of the EpochHeader the NEXT request is built under, with the Reque 'request/header': { header: EpochHeader; reason: RequestHeaderReason } ``` -Source: [`packages/core/session/src/types.ts:369`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:374`](../packages/core/session/src/types.ts) #### `request/header-delta` — log-only @@ -187,7 +187,7 @@ Amendment to the folded EpochHeader: at least one of a SystemDelta, a ToolsDelta 'request/header-delta': { system?: SystemDelta; tools?: ToolsDelta; config?: LlmCallConfig; messagePrefix?: Message[] } ``` -Source: [`packages/core/session/src/types.ts:386`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:391`](../packages/core/session/src/types.ts) ### `steering/*` @@ -201,7 +201,7 @@ Steering content injected between steps of a running turn. Types: [ContentBlock](core-data-structures/core.md) · [MessageSource](core-data-structures/core.md) -Source: [`packages/core/session/src/types.ts:342`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:347`](../packages/core/session/src/types.ts) ### `step/*` @@ -213,7 +213,7 @@ Closes step `step` of turn `turn`. 'step/end': { turn: number; step: number } ``` -Source: [`packages/core/session/src/types.ts:296`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:301`](../packages/core/session/src/types.ts) #### `step/start` — log-only @@ -223,7 +223,7 @@ Opens step `step` of turn `turn` — one model call plus the tool executions it 'step/start': { turn: number; step: number } ``` -Source: [`packages/core/session/src/types.ts:294`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:299`](../packages/core/session/src/types.ts) ### `todo/*` @@ -239,7 +239,7 @@ NOT a SurfaceEventType: it produces no LLM message and never reaches `deriveMess Types: [TodoItem](core-data-structures/session.md) -Source: [`packages/core/session/src/types.ts:356`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:361`](../packages/core/session/src/types.ts) ### `tool/*` @@ -253,7 +253,7 @@ The model requested one tool invocation: `name` with the raw `arguments` JSON st Types: [CallId](core-data-structures/core.md) -Source: [`packages/core/session/src/types.ts:330`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:335`](../packages/core/session/src/types.ts) #### `tool/code-dispatch` — log-only @@ -277,7 +277,7 @@ A completed tool call's model-facing result, plus an optional tool-private `meta Types: [CallId](core-data-structures/core.md) · [ContentBlock](core-data-structures/core.md) -Source: [`packages/core/session/src/types.ts:340`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:345`](../packages/core/session/src/types.ts) ### `turn/*` @@ -291,7 +291,7 @@ Closes turn `turn` with the TurnEndReason that ended it. The loop fires the awai Types: [TurnEndReason](core-data-structures/session.md) -Source: [`packages/core/session/src/types.ts:292`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:297`](../packages/core/session/src/types.ts) #### `turn/start` — log-only @@ -303,7 +303,7 @@ Opens turn `turn`. `trigger` records what started it — a drained message batch Types: [TurnTrigger](core-data-structures/session.md) -Source: [`packages/core/session/src/types.ts:286`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:291`](../packages/core/session/src/types.ts) ### `user/*` @@ -317,4 +317,4 @@ A user-visible prompt (queued message drained at turn start). Types: [ContentBlock](core-data-structures/core.md) · [MessageSource](core-data-structures/core.md) -Source: [`packages/core/session/src/types.ts:298`](../packages/core/session/src/types.ts) +Source: [`packages/core/session/src/types.ts:303`](../packages/core/session/src/types.ts) diff --git a/docs/rfc/INDEX.md b/docs/rfc/INDEX.md index 9596aecd78..ccb8911ab0 100644 --- a/docs/rfc/INDEX.md +++ b/docs/rfc/INDEX.md @@ -71,6 +71,7 @@ Generated by `pnpm run gen-rfc-index` from the RFC tree — never edit by hand; | [The session prefix — request-only messages in front of the derived history](implemented/feature/2026-07-07-session-prefix.md) | 2026-07-07 | | [Repeat-tool-call guard plugin](implemented/feature/2026-07-08-repeat-tool-guard.md) | 2026-07-08 | | [The self-referential cordis toolset](implemented/feature/2026-07-08-self-referential-cordis-toolset.md) | 2026-07-08 | +| [Configure subagent persona, tool visibility, and depth](implemented/feature/2026-07-12-subagent-persona-tool-filter-and-depth.md) | 2026-07-12 | ### Simplification diff --git a/docs/rfc/implemented/architecture/2026-06-18-agent-lifecycle-and-ownership-seams.md b/docs/rfc/implemented/architecture/2026-06-18-agent-lifecycle-and-ownership-seams.md index 80ca18b56f..d64329262e 100644 --- a/docs/rfc/implemented/architecture/2026-06-18-agent-lifecycle-and-ownership-seams.md +++ b/docs/rfc/implemented/architecture/2026-06-18-agent-lifecycle-and-ownership-seams.md @@ -16,7 +16,7 @@ A new `cancel()` verb on the `Agent` interface — the single public stop primit ### 2. `AgentHandle` async disposer -`ctx.agents.create`/`resume` (and the `AgentFactory` interface) return `AgentHandle = { agent: Agent; dispose(): Promise }`. The disposer is a **consumer capability** — a registry observer holding only the bare `Agent` cannot tear it down. The caller fiber and registered factory provider are structural co-owners: caller unload enforces structured ownership, while provider unload must stop old instances whose scoped dependency surface resolves through that provider. All three paths reach the same memoized teardown: stop the loop, `await` its exit (true quiescence, not just the `disposed` status flip), unregister it, remove its session from the store, unwind its scope, and only then release both public IDs. Config-created agents are already owned by the `AgentLoop` fiber (the handle is discarded). ACP holds each session's disposer in its `SessionRecord` and runs it on disconnect/teardown, so a bare client disconnect leaves no registered agent and no session-store entry — even when `session/load` races teardown (the just-resumed handle is disposed before the closed-guard throw). +`ctx.agents.create`/`resume` (and the `AgentFactory` interface) return `AgentHandle = { agent: Agent; dispose(): Promise }`. The disposer is a **consumer capability** — a registry observer holding only the bare `Agent` cannot tear it down. The caller fiber and registered factory provider are structural co-owners: caller unload enforces structured ownership, while provider unload must stop old instances whose scoped dependency surface resolves through that provider. All three paths reach the same memoized teardown: stop the loop, await its exit and idle flushes (true quiescence, not just the `disposed` status flip), detach the agent, detach its session, and unwind its scope. Each public ID becomes reusable when its exact registry entry detaches; there is no separate reservation-release phase. Config-created agents are already owned by the `AgentLoop` fiber (the handle is discarded). ACP holds each session's disposer in its `SessionRecord` and runs it on disconnect/teardown, so a bare client disconnect leaves no registered agent and no session-store entry — even when `session/load` races teardown (the just-resumed handle is disposed before the closed-guard throw). **Teardown ORDER is load-bearing for durability**, and the implementation folds the session lifecycle into the agent's SINGLE composite cordis effect (`SessionStore.prepare`/`enter`/`announce`, replacing a sibling-effect split). A fiber unload disposes sibling effects concurrently (`Promise.all`), which would race removing the session store's append publication hooks against the loop's closing `session/flush` and drop the closing `turn/end`; inside one effect the disposers run as an ordered LIFO chain (loop stopped + `await agent.done` BEFORE the session detaches), so the loop's final flush is captured on BOTH the handle's `dispose()` and a fiber unload. The contained `agent/disposed` and `session/disposed` notifications cannot reject the chain or skip later teardown. @@ -35,7 +35,7 @@ These invariants hold and are pinned by tests: ## Session owner tokens are unique among live agents -The bash owner-token comparison relies on `session.header.id` being unique among live agents. `SessionStore.enter()` rejects a duplicate live session id, and the async agent factory reserves both agent and session ids across persistence loading and unpublished setup before rechecking the store at publication. A programmatic caller therefore cannot publish two live agents with one session token. The access *policy* (token comparison) stays in `tool-bash` (the consumer); the bash seam stores only an opaque `owner` string and never interprets it — the correct interface/implementation/consumer split. +The bash owner-token comparison relies on `session.header.id` being unique among live agents. Concurrent same-ID operations may both prepare privately, but publication enters the session and agent in order; `SessionStore.enter()` rejects a duplicate live session id, and every losing transaction rolls its private state back. A programmatic caller therefore cannot publish two live agents with one session token. The access *policy* (token comparison) stays in `tool-bash` (the consumer); the bash seam stores only an opaque `owner` string and never interprets it — the correct interface/implementation/consumer split. ## Alternatives considered diff --git a/docs/rfc/implemented/architecture/2026-07-08-agent-scope-contexts.md b/docs/rfc/implemented/architecture/2026-07-08-agent-scope-contexts.md index 024eb05f82..76c40805f8 100644 --- a/docs/rfc/implemented/architecture/2026-07-08-agent-scope-contexts.md +++ b/docs/rfc/implemented/architecture/2026-07-08-agent-scope-contexts.md @@ -4,30 +4,28 @@ Status: implemented ## Problem -One application needs to share infrastructure across many agents while giving each agent a coherent local world. Model adapters, persistence, user interfaces, and most tool implementations belong to the deployment; personas, visible tools, live policy, event listeners, and cleanup often belong to one agent. +One application needs to share infrastructure across many agents while letting each agent have its own tools, prompt contributions, policies, and listeners. Shared adapters, persistence, and user interfaces belong to the deployment; a persona, tool variant, or listener often belongs to one agent. -A separate service graph per agent duplicates shared infrastructure. One global registration graph has the opposite failure: an agent-specific tool, prompt section, restriction, or listener can leak into unrelated agents. Contributors need one way to compose local behavior without learning a different registration API for every service. +A separate service graph per agent duplicates shared infrastructure. One global registration graph has the opposite failure: an agent-specific contribution can leak into unrelated agents. Contributors need one ordinary registration mechanism that determines both who can see a contribution and when it is cleaned up. -The mechanism also needs a clear lifetime. An agent must not become visible before its local registrations exist, and final loop work must not lose those registrations before it settles. Parent-owned subagents make both failures easy to trigger because several differently configured agents can exist concurrently inside one application. +The mechanism also needs a publication boundary. An agent must not become visible before its local world is complete, and teardown must retain that world until final work has stopped. ## Decision -Every live agent owns one flat registration layer through `agent.ctx`. Code registers through the context that owns the contribution; scope-aware services resolve deployment-global registrations plus exactly one matching agent layer; scoped events route by the operation's real agent; and the layer is published and revoked with the agent lifecycle. +Every live agent owns one flat registration layer exposed as `agent.ctx`. Code registers through the context that owns a contribution; scope-aware services combine deployment-global registrations with exactly one matching agent layer; operations choose that layer from their real agent; and the layer exists for the agent's complete published lifetime. -A Cordis **context** is the object through which code accesses services and registers owned effects. The [Cordis primer](../../../cordis-primer.md) explains the framework beyond that concept. +Cordis is the plugin framework underneath the SDK. A Cordis **context** is the object plugins use to access services and register effects whose cleanup follows that context. The [Cordis primer](../../../cordis-primer.md) explains the framework in more detail. -The contract has four parts: +For most contributors, the complete contract is four rules: -| Contributor question | Contract | +| Question | Rule | |---|---| -| Where do I register agent-local behavior? | Use the ordinary service API through `agent.ctx` | -| What does an agent see? | Deployment globals plus its own layer, with service-specific merge rules | -| Which scoped listeners run? | By default, unscoped listeners plus listeners for the operation's agent; an explicit global-listener exception is described below | -| How long does local behavior exist? | Assembled during unpublished setup, observable only after creation succeeds, and retained through quiescent teardown | +| Where do I register behavior for one agent? | Call the ordinary registration API through `agent.ctx` | +| What does an operation for an agent see? | Deployment globals plus that agent's layer, using the owning service's merge rules | +| Which scoped listeners run? | Unscoped listeners plus listeners registered for the operation's agent | +| How long does the layer exist? | Setup completes before publication; disposal keeps it until work reaches quiescence | -The scope is deliberately flat. Resolution never walks parent or sibling scopes. Parent ownership links lifetimes without importing registrations. - -For scope-aware registries and default listener routing, the whole mechanism can be read from left to right: the registering context chooses a layer, while the agent named by an operation chooses which one local layer joins the deployment-global layer. +The scope is flat. Resolution never walks parent or sibling scopes, and lifetime ownership does not imply registration inheritance. ```mermaid flowchart LR @@ -35,41 +33,32 @@ flowchart LR agentAContext["agentA.ctx
cleanup follows Agent A"] -->|"registers into"| agentALayer["Agent A layer"] agentBContext["agentB.ctx
cleanup follows Agent B"] -->|"registers into"| agentBLayer["Agent B layer"] - operationA["Operation for Agent A"] -->|"selects"| agentAView["Agent A view
eligible globals plus A local only"] + operationA["Operation for Agent A"] -->|"selects"| agentAView["Agent A view
globals plus A local"] globalLayer --> agentAView agentALayer --> agentAView - operationB["Operation for Agent B"] -->|"selects"| agentBView["Agent B view
eligible globals plus B local only"] + operationB["Operation for Agent B"] -->|"selects"| agentBView["Agent B view
globals plus B local"] globalLayer --> agentBView agentBLayer --> agentBView ``` -The missing cross-edges describe registry resolution and default listener routing: Agent A's registered values and ordinary scoped listeners do not enter Agent B's view, and a parent's layer does not enter a child's view merely because the parent owns the child's lifetime. For scope-filtered events, `{ global: true }` is the explicit opt-in exception; it can observe across scopes while cleanup still follows the registering agent. Registry-membership notifications are a separate unfiltered event class described below. +The missing cross-edges are the isolation rule: Agent A's local registrations do not enter Agent B's view, and a parent's registrations do not enter a child merely because the parent owns the child's lifetime. -The companion [runtime-design RFC](2026-07-12-agent-scope-runtime-design.md) explains how the implementation preserves this contract under Cordis dispatch, JavaScript mutation and reentrancy, asynchronous setup, rollback, and racing disposal. +The companion [runtime-design RFC](2026-07-12-agent-scope-runtime-design.md) explains the implementation and correctness reasoning. The [subagent composition-controls RFC](../feature/2026-07-12-subagent-persona-tool-filter-and-depth.md) owns the separate `persona`, `toolFilter`, and `maxDepth` feature. -### Registration origin selects visibility and cleanup +### Registration origin chooses visibility and cleanup -A contribution made through a plain plugin context is deployment-global and is disposed with that plugin. The same method called through `agent.ctx` contributes only to that agent and is disposed with the agent scope. +A registration made through a plain plugin context is deployment-global and is disposed with that plugin. The same method called through `agent.ctx` contributes to one agent and is disposed with that agent's scope. -| Registration origin | Registration layer and default audience | Disposed with | +| Registration origin | Default visibility | Disposed with | |---|---|---| -| Plain plugin context | Deployment-global; eligible for every agent view, subject to service merge and restriction rules | Registering plugin | -| `agent.ctx` | Agent-local; visible to that agent by default | Agent scope | +| Plain plugin context | Every eligible agent view | Registering plugin | +| `agent.ctx` | Exactly that agent's view | Agent scope | -This applies to tools, prompt sections and variables, restrictions, protections, guards, and scoped event listeners. Named scoped values ordinarily shadow same-named global values; an owning service may reserve a protected name and reject the shadow instead. Duplicate names within one layer fail. Event listeners have the explicit `{ global: true }` audience exception described below. +Tools, prompt sections and variables, tool restrictions, guards, and scoped event listeners adopt this contract. Named local values ordinarily shadow a same-named global value for that agent; each owning service documents exceptions and merge behavior. -The public pattern is ordinary registration inside `setup`: +The ordinary contributor pattern is to register the complete local world during agent setup: ```js -const reviewSummaryTool = { - name: 'review_summary', - description: 'Return the review summary.', - parameters: { type: 'object', properties: {} }, - async execute() { - return [{ type: 'text', text: 'review complete' }] - }, -} - const handle = await ctx.agents.create({ agentId: AgentId('reviewer'), sessionId: SessionId('reviewer-session'), @@ -80,130 +69,108 @@ const handle = await ctx.agents.create({ order: 0, text: 'Review code, but do not modify files.', }) - agentCtx.tools.restrict({ allow: ['read'] }) - agentCtx.tools.register(reviewSummaryTool) + agentCtx.tools.register({ + name: 'review_summary', + description: 'Return the review summary.', + parameters: { type: 'object', properties: {} }, + async execute() { + return [{ type: 'text', text: 'review complete' }] + }, + }) }, }) -const reviewer = handle.agent -ctx.tools.get('read', reviewer) // global and allowed -ctx.tools.get('bash', reviewer) // undefined: filtered global -ctx.tools.get('review_summary') // undefined: not global -ctx.tools.get('review_summary', reviewer) // reviewer-local +ctx.tools.get('review_summary') // undefined: not global +ctx.tools.get('review_summary', handle.agent) // the reviewer-local tool await handle.dispose() -ctx.tools.get('review_summary', reviewer) // undefined: scope is gone +ctx.tools.get('review_summary', handle.agent) // undefined: scope is gone ``` -### The operation selects the view +Setup receives a full trusted Cordis context so it can compose ordinary plugins and services. Its contract is composition-only: driving or publishing the in-flight agent through casts or internal registry calls is unsupported. -Registration origin and operation subject are separate facts. Calling a read method through `agent.ctx` does not implicitly select that agent; lookup, execution, prompt assembly, and event dispatch still receive the agent or scope they act for. +### The operation chooses the view -For example, `agent.ctx.systemPrompt.assemble()` without an assembly scope requests the global view. `ctx.tools.get(name, agent)` and `ctx.tools.execute({ ..., agent })` select that agent's tool view explicitly. This lets one shared service act for any agent without binding the service instance itself to one scope. +Registration origin and operation subject are separate facts. Calling a service through `agent.ctx` selects where a new registration belongs; it does not bind later reads to that agent. -`agent.ctx.agent` is the associated agent for setup code, but it is not a general scope-selection shortcut. Contributors creating nested generic scopes use the nearest scope tag as the registration key; inheriting an `agent` property does not import the outer registration layer. +Tool lookup and execution receive the agent they act for. Prompt assembly receives an assembly context for the agent whose request is being built. Event dispatch receives its domain subject. This keeps shared service instances reusable across agents while making each operation's view explicit. -### Scoped events follow the operation's real subject +Only services that adopt the scope contract resolve an agent layer. `agent.ctx` does not automatically change arbitrary Cordis service calls. -By default, an event about agent A reaches unscoped listeners and A-scoped listeners, not B-scoped listeners; agent-less dispatch reaches only unscoped listeners. Product helpers and service-owned paths couple the routing key to the value the operation already owns—such as `ToolExecution.agent`, `ApprovalRequest.agent`, the prompt assembly scope, or the session's captured owner. Advanced code that constructs a low-level carrier or assembly context directly must keep its subject and scope fields aligned; development invariants detect mismatches, but the low-level types do not make every mismatch unrepresentable. +### Scoped events keep routing separate from event data -Cordis listeners have one explicit exception. `{ global: true }` bypasses contextual filtering, so a listener registered through `agent.ctx` can observe other agents and subjectless dispatches while its cleanup still belongs to that agent scope. Use it only for deliberate cross-scope observation. +An event about Agent A normally reaches unscoped listeners and A-scoped listeners, not B-scoped listeners. An event without an agent subject reaches only unscoped listeners. -Registry-membership notifications remain unfiltered because they describe shared registry state rather than an operation for one agent. The generated [event catalog](../../../cordis-catalog/events.md) is the exhaustive reference for event signatures and modes. +At the Cordis level, `Scoped` is an opaque routing receiver. It carries the filter used to choose listeners but is not the domain object. Event signatures therefore keep the real `Agent`, tool execution, approval request, or other subject as an explicit argument that listeners can inspect. -### Creation publishes after setup; disposal revokes after work stops +A listener registered with `{ global: true }` deliberately bypasses contextual audience filtering while its cleanup still follows the registering context. Registry-membership notifications remain unfiltered because they describe shared registry state rather than one agent's operation. The generated [event catalog](../../../cordis-catalog/events.md) is the exhaustive event reference. -`ctx.agents.create()` and `resume()` construct an unpublished agent. Their optional `setup(agentCtx)` callback may await child-plugin activation and register the complete local world. During setup, neither the agent nor its session is visible in the public registries, and driving methods reject. +### Creation publishes last and disposal revokes last -The returned promise resolves only after setup, ordered lifecycle notification, and loop start succeed. Setup failure or owner loss rolls the unpublished world back and releases its IDs. A caller therefore never receives a handle to a partially configured agent. +`ctx.agents.create()` and `resume()` build an unpublished session, scope, agent, and driver. They await `setup`, admit the final session and agent entries, announce them in order, start the loop, and only then return a handle. -`AgentHandle.dispose()` performs the reverse boundary. It stops and drains the loop, preserves the session and scoped listeners through final events and flushes, detaches the agent and session, unwinds the scope, and releases IDs. Repeated or racing calls join the same completion promise. +An optional creation signal cancels work only while create or resume is pending. After the promise resolves, the returned `AgentHandle` owns explicit disposal. -The calling Cordis context and AgentLoop are structural co-owners. Unloading either disposes the agent, so creation through a short-lived plugin context intentionally gives the agent that shorter lifetime. +If loading, setup, admission, or publication fails, the private transaction rolls back everything it prepared. Concurrent operations using the same caller-supplied live ID may both reach setup, but final registry entry admits only one; every loser rejects and cleans its private resources. Sequential reuse after awaited disposal remains valid. -The lifecycle keeps the local layer private until setup succeeds and keeps it alive until final work has drained: +`AgentHandle.dispose()` reverses the boundary. It deactivates creation or driving, waits for synchronous publication to unwind, stops and drains the driver and final session flushes, detaches the agent and session, and finally disposes the scope. Repeated or racing disposal requests join one completion promise. + +The calling Cordis context and the concrete AgentLoop factory are structural co-owners. Unloading either disposes the transaction or live agent. ```mermaid flowchart TB - request["Create or resume"] --> reserve["Reserve agent and session IDs"] - reserve --> privateWorld["Load or build private session, scope, and driver"] - privateWorld --> setup["Await setup through agent.ctx"] - setup --> publish["Publish session and agent, then start the loop"] - publish --> live["Return the live handle"] + request["Create or resume"] --> privateWorld["Build private session, scope, agent, and driver"] + privateWorld --> setup["Await composition through agent.ctx"] + setup --> admission["Admit final session and agent entries"] + admission --> publish["Announce lifecycle and start the driver"] + publish --> live["Return AgentHandle"] - privateWorld -->|"load or preparation failure, or owner loss"| rollback["Rollback startup
no handle escapes"] - setup -->|"setup failure or owner loss"| rollback - publish -->|"publication failure or owner loss"| rollback - live -->|"handle disposal, owner unload, or AgentLoop unload"| settle["Quiesce prepared or running work"] - rollback --> settle - settle --> detach["Detach any published agent, then session"] - detach --> revoke["Dispose any created agent scope"] - revoke --> release["Release acquired IDs"] + privateWorld -->|"failure, cancellation, or owner loss"| rollback["Rollback private work"] + setup -->|"failure, cancellation, or owner loss"| rollback + admission -->|"duplicate or owner loss"| rollback + publish -->|"listener failure or owner loss"| rollback + live -->|"handle or owner disposal"| quiesce["Stop and drain work"] + rollback --> quiesce + quiesce --> detach["Detach agent, then session"] + detach --> revoke["Dispose the agent scope"] ``` -Contributors should put agent-local activation inside `setup` and always dispose the returned handle. Code that needs to observe a live agent waits for `create()`/`resume()` to resolve rather than polling the registries during setup. +### Subagent controls are an independent feature -## Tool restrictions resolve against a live flat view +In-process subagents consume agent scope by installing their local composition during unpublished setup. Their optional persona, live global-tool filter, and absolute depth cap are not intrinsic scope semantics; the [subagent composition-controls RFC](../feature/2026-07-12-subagent-persona-tool-filter-and-depth.md) defines those controls, provider capability checks, and dynamic tool behavior. -A tool restriction filters the live deployment-global end-capability layer, after which scope-local tools are added. `allow` retains named globals, `deny` removes named globals, multiple restrictions intersect, and a hidden global tool is absent from both registry presentation and executable lookup. +`inheritsParentContext` describes conversation-history seeding only. It says nothing about Cordis scope, injected services, tools, or authority. -Filter presence is explicit: omitting a filter installs no restriction, `restrict({})` rejects, and `allow: []` deliberately hides every global end capability. +## Security and authority are non-goals -Because globals are live, allow- and deny-lists intentionally differ when a new global tool appears: +Agent scopes compose trusted same-process registrations. They do not sandbox plugins, define a parent-to-child authority lattice, freeze grants at creation, or guarantee that a child can do no more than its parent. -```text -at time 0: - global tools = { read, bash } - deny { bash } view = { read } - allow { read } view = { read } +A parent may own a child whose visible tools are wider than its own because lifetime ownership does not donate or cap registrations. A plugin holding a Cordis context also runs in the same process and can call available services directly. -after registering global tool web: - deny { bash } view = { read, web } - allow { read } view = { read } -``` - -Scope-local tools are merged after the filter. A local tool can therefore exist even when it is absent from an allow-list over globals. This is composition behavior, not an authorization promise. - -Reserved Code Mode presentation is not part of the filterable end-capability layer. The [Code Mode RFC](../feature/2026-06-15-code-mode.md) owns the `run_code`, SDK, `toolOrder`, and presentation-versus-execution contracts; contributors changing Code Mode behavior follow that decision rather than inferring new authority semantics from agent scope. - -## Security and authority are explicit non-goals - -Agent scopes compose trusted in-process registrations. They do not sandbox plugins, define a parent-to-child authority lattice, freeze a creation-time grant set, or guarantee that a child can do no more than its parent. A plugin holding a Cordis context runs in the same process and can call the services injected into that context. - -A parent can own a child whose visible tool set is wider than its own. For example, a parent restricted to global `read` can spawn a child with no restriction; the child then sees later global tools plus its own local registrations. The parent owns the child's lifetime but does not donate or cap the child's registration layer. - -Deployments that need non-escalation require a separate authority representation, propagation rule, and execution check. Authority-versus-visibility ledgers, parent-subset grants, explicit future-grant APIs, and generic capability/output/termination tags are outside this decision. - -## Subagents use the same composition rule - -In-process subagents are a consumer of agent scope, not a second scoping model. A child gets a fresh flat layer during unpublished setup; its persona, tool filter, structured-output protocol, and listeners are ordinary registrations through the child's context. Parent teardown, backend teardown, and manual run disposal own the child lifetime without importing the parent's registrations. - -`inheritsParentContext` describes conversation-history seeding only, not Cordis scope, service injection, tools, or authority. The [subagent capability RFC](../feature/2026-06-21-subagent-capability-seam.md) owns run usage and the provider contract, while the [runtime-design RFC](2026-07-12-agent-scope-runtime-design.md) explains in-process structured output and workflow race handling. +Deployments that need non-escalation require a separate authority representation, propagation rule, and execution check. Parent-subset grants, creation-time authorization snapshots, explicit future-grant APIs, and generic capability/output/termination tags are outside this decision. ## Alternatives considered -The rejected architectures either separate visibility from cleanup, scope only behavior but not registered data, duplicate shared infrastructure, or conflate parent ownership with registration inheritance. +The rejected designs either separate visibility from cleanup, cover only one registration family, duplicate shared infrastructure, or conflate lifetime ownership with inheritance. ### Pass an agent option to every registration -An API such as `tools.register(definition, { agent })` leaves global registration as the leak-by-omission default and repeats scope plumbing in every registry. It can also express “visible to A, disposed with unrelated plugin B,” which registration through `agent.ctx` prevents. +An API such as `tools.register(definition, { agent })` repeats scope plumbing in every registry and permits visibility ownership to drift from cleanup ownership. Registering through `agent.ctx` makes both facts follow one Cordis effect owner. ### Filter events while keeping registries global -Listener filtering prevents a hook from intercepting the wrong agent but does not scope tool schemas, executable lookup, prompt sections, variables, or Code Mode bindings. Agent-local composition would still require temporary global mutation. +Listener filtering prevents the wrong hook from running but does not scope tool schemas, executable lookup, prompt sections, variables, or other registered data. Agent-local composition would still require temporary global mutation. -### Create one isolated service graph per agent +### Create one service graph per agent -Service isolation chooses one registry instance, while the desired view is deployment globals plus one agent layer. Per-agent graphs duplicate adapters and force shared persistence and UI infrastructure to discover every instance. Independent applications still deserve separate graphs; collaborating agents inside one deployment do not. +The required view is shared deployment services plus one local registration layer. Per-agent graphs duplicate adapters and complicate shared persistence, provider registries, and application boot. -### Inherit the parent's registrations into a child +### Inherit parent registration scopes -Hierarchical inheritance silently imports every parent-scoped tool and policy. Flat layers plus parent-owned disposal separate lifetime from composition: the parent owns the child without deciding the child's local world. This choice deliberately makes authorization a separate design. +Parentage describes lifetime and conversation lineage, not a universal merge policy. Hierarchical lookup makes unrelated services inherit accidentally and cannot define security without a separate authority model. ## Consequences -Contributors use the same registration methods at both deployment and agent scope; changing the calling context changes visibility and cleanup together. Model-visible tool lookup, execution, prompt assembly, policy, observation, and teardown agree on one agent key instead of maintaining parallel per-feature scope options. +Contributors use one familiar pattern: register shared behavior through a plugin context, register local behavior through `agent.ctx`, select the real agent on operations, and dispose the returned handle. Setup is atomic from an observer's perspective, and teardown preserves local behavior until work stops. -The cost is explicit subject selection on reads and dispatch, asynchronous programmatic creation, disciplined handle disposal, and awareness that flat registration scope is not authority. Registries retain service-specific merge behavior, and only services that adopt the scope contract become agent-scoped automatically. - -The decision applies to tools, prompt state, scoped events, session lifecycle and scoped session events, approvals, and in-process subagent composition. Filesystem policy, LLM interception, background backend state, and other registries retain their own subject or policy mechanisms until their designs explicitly adopt agent scope. +The cost is explicit subject selection, asynchronous programmatic creation, and service-specific scope adoption. Flat registration scope is intentionally not authority, and subagent composition controls remain a separate feature rather than hidden scope semantics. diff --git a/docs/rfc/implemented/architecture/2026-07-12-agent-scope-runtime-design.md b/docs/rfc/implemented/architecture/2026-07-12-agent-scope-runtime-design.md index 35d1b50c16..786dce250e 100644 --- a/docs/rfc/implemented/architecture/2026-07-12-agent-scope-runtime-design.md +++ b/docs/rfc/implemented/architecture/2026-07-12-agent-scope-runtime-design.md @@ -4,999 +4,390 @@ Status: implemented ## Problem -The [agent-scope contract](2026-07-08-agent-scope-contexts.md) defines the contributor-visible result: registrations made through `agent.ctx` form one flat local layer, operations resolve that layer by their real agent, setup remains unpublished, and teardown preserves the layer until work stops. The implementation must make those claims true inside a cooperative plugin framework and mutable JavaScript runtime. +The [agent-scope contract](2026-07-08-agent-scope-contexts.md) is simple for contributors: register through `agent.ctx`, resolve one global-plus-agent view, publish only after setup, and retain the scope until work stops. The runtime must preserve that contract across a cooperative plugin framework, asynchronous creation, reentrant listeners, durable session commits, and worker or process failure. -Four failure classes interact in the paths this change hardens: +The main design risk is adding a second mechanism for every race. Separate reservations, readiness sentinels, cancellation relays, snapshot layers, and protection registries can mirror the same fact until no reader can tell which one is authoritative. That machinery also encourages the runtime to treat trusted typed calls as hostile serialization boundaries. -| Proof obligation | Failure if implemented locally or incompletely | -|---|---| -| Registration and dispatch select the same key | Prompt data resolves for one agent while behavior listeners run for another | -| Construction and teardown have continuous ownership | Reentrant unload publishes a partial world, leaks IDs, or revokes policy before final work | -| Covered validation and later use observe one accepted value | Stateful accessors or caller mutation make checked, executed, logged, and observed data disagree | -| Protocol invariants survive extensible middleware | Listener ordering removes required prompt state, re-allows denied work, commits a failed result, or forces another model step | - -Cordis already supplies derived contexts, effect ownership, receiver-based filtering, and waterfalls, but none alone establishes all four obligations. Context association is not a scope key, raw disposers are not await-idempotent, waterfall listeners can short-circuit or wrap each other, and JavaScript `readonly` types do not constrain runtime accessors. Async persistence, setup, publication callbacks, subagent providers, and worker messages add reentrancy and race boundaries around those primitives. +The implementation needs enough state to preserve real ownership and settlement boundaries, but no more. A correctness reviewer must be able to follow one fact from acceptance through publication and teardown without reconciling parallel representations. ## Decision -The runtime implements agent scope as four coupled mechanisms rather than one generic framework feature: +The runtime uses one mechanism per independent fact. Scope routing has an opaque carrier; each live registry object has one entry record; each create or resume operation has one transaction; typed same-process calls borrow readonly values; real data boundaries materialize once; and worker/process code retains separate terminal and quiescence state only where different owners can genuinely race. -| Mechanism | Implementation decision | +The design can be skimmed as seven choices: + +| Problem | Authoritative mechanism | |---|---| -| Layer and routing | A scope key tags registration effects; scope-aware contribution registries merge globals plus one layer; dispatch carriers filter listeners by the operation subject | -| Transactional lifetime | Scope, session, registry entry, driver, reservations, caller ownership, and AgentLoop ownership publish and unwind as one ordered transaction | -| Boundary ownership | The hardened acceptance-sensitive paths listed below capture caller fields once and retain stable identities or owner-controlled representations | -| Owner-final policy | Four narrow service-owned boundaries restore named prompt state, deny monotonically, observe final results, and stop terminal turns | +| Select global plus one agent's registrations | Opaque scope key and routing carrier | +| Own one live agent or session | One registry entry captured by its disposer | +| Coordinate create/resume | One `AgentCreationTransaction` | +| Protect durable, queued, model, or wire data | Materialize once at that boundary | +| Pass typed values inside one process | Readonly borrowed contract | +| Preserve an owner's final prompt/tool policy | Contribution-owned finality and one final observer point | +| Coordinate subagent, worker, and process shutdown | One cancellation signal plus the independent terminal/quiescence facts of that boundary | -The same mechanisms carry into in-process subagents and the workflow bridge. Subagents are the composition proof because child setup, structured output, readiness, cancellation, result settlement, and disposal exercise all four obligations concurrently. +The rest of this RFC expands those choices in dependency order. It first explains the Cordis mechanics, then scope routing, creation and session commit, tools and prompts, subagents and workflows, and finally the checks that make the reasoning executable. -This RFC owns the implementation rationale, algorithms, race handling, and correctness enforcement. The [agent-scope contract](2026-07-08-agent-scope-contexts.md) owns contributor-facing behavior, tool-filter semantics, usage examples, and the security non-goal; this document links to that contract rather than redefining authority or public scope inheritance. +The [July 8 RFC](2026-07-08-agent-scope-contexts.md) remains the contributor contract. The separate [subagent composition-controls RFC](../feature/2026-07-12-subagent-persona-tool-filter-and-depth.md) owns `persona`, `toolFilter`, and `maxDepth`; this document discusses only how their setup fits the lifecycle. -## Implementation model: domain terms and Cordis mechanics +## Cordis model: context, fiber, effect, receiver, and waterfall -Readers need four domain terms and four Cordis mechanics to follow the implementation. Readers already familiar with this codebase and Cordis can skim this section. +Five Cordis ideas are required to understand the implementation. A context selects services and registration ownership; a fiber is one live plugin or child lifecycle; an effect attaches cleanup to a fiber; an event receiver selects listeners; and a waterfall lets listeners transform or veto an operation in sequence. -### Recurring domain terms +### A context is an ownership path through one service graph -Four domain terms keep the rest of the RFC compact. A **Session** is an agent run's append-only event log, from which model history and durable replay are derived. **Lossless JSON** is the JSON subset that can be copied without changing meaning: primitives, dense arrays, and plain objects; cycles, sparse arrays, exotic prototypes, non-finite numbers, negative zero, `undefined`, `bigint`, functions, and symbols are rejected. An **end capability** is an actual callable tool implementation, whether the model sees it as a native schema or a Code Mode binding. **Code Mode** gives the model a generated SDK and a reserved `run_code` transport. Pure `code` presentation replaces native advertisement with that interface; `both` presentation retains native schemas alongside it. +All agents share one Cordis service graph. A derived context does not clone `ToolRegistry`, `SystemPrompt`, persistence, or model adapters; it changes how registrations made through that context are tagged and which effects own their cleanup. -### Four Cordis mechanics +`agent.ctx` is such a derived context. Service calls still reach the shared instances, while a registration can inspect its calling context and store a contribution under the nearest scope key. Ordinary plugin contexts carry no scope key and therefore register globally. -Contexts select service access and registration origin, fibers own effects, waterfalls provide cooperative transformation, and dispatch receivers select listeners. +### Fibers and effects make cleanup structural -| Cordis concept | Meaning in this RFC | -|---|---| -| Context | The object through which a plugin reaches services and registers contributions; a derived context can carry a different registration scope | -| Fiber and effect | The runtime owner and one owned piece of setup/cleanup; disposing the fiber unwinds its effects | -| Waterfall | Ordered around-middleware whose listener calls `next()` to include downstream work and may transform or short-circuit the result | -| Dispatch receiver | The `this` object used by Cordis listener filtering; a scope carrier encodes the operation's agent key | +A Cordis fiber is the live instance created when a plugin or child context is activated. Its state records whether that lifecycle is active, unloading, failed, or disposed. `ctx.effect()` and `ctx.on()` return disposers and also attach those disposers to the registering fiber, so unloading a plugin or agent scope removes everything registered through that context without a separate inventory. -#### Context selects both service access and registration origin +The vendored Cordis fiber implementation establishes ownership before arbitrary setup or `internal/plugin` observers run. A reentrant unload can see the child fiber or effect that has started, reject effects added after unload begins, and join cleanup already started through a public single-shot disposer. Teardown observers are contained individually so one callback cannot prevent structural cleanup. -A Cordis `Context` is the object through which code calls services such as `ctx.tools`, `ctx.systemPrompt`, and `ctx.sessions`. A service can recover the context through which it was accessed, so the same method can register globally from a plain plugin context or locally from `agent.ctx` without adding an `agent` option to every registration API. Cordis implements contextual service access with a **traced receiver**: a proxy that carries the accessing context while forwarding calls to the concrete service object. +These are framework lifecycle guarantees rather than agent-specific policy. Agent creation depends on them because setup can activate arbitrary plugins and synchronously reenter owner disposal. -```js -ctx.tools.register(globalTool) -agent.ctx.tools.register(agentOnlyTool) +### Receivers route listeners; waterfalls compose decisions -ctx.on('tools/result', globalObserver) -agent.ctx.on('tools/result', agentObserver) -``` +Cordis filters listeners using the dispatch receiver (`this`), while harness listeners need an explicit agent, execution, request, or other subject. `Scoped` marks the receiver expected by a scoped event declaration, but the runtime carrier deliberately exposes no subject API. -A context also exposes the dependency view injected into the plugin that minted it. `agent.ctx` therefore carries the agent loop's deliberate service surface; it is not an ambient root context or a security boundary. +Product helpers therefore construct the carrier and pass the domain subject separately. This prevents listener routing from becoming an alternate object model and keeps event signatures understandable without knowledge of carrier internals. -#### Effects make cleanup follow ownership +A Cordis waterfall is middleware-style dispatch. Each listener receives `next()`: calling it delegates to the remaining listeners and base operation, while returning without it vetoes or replaces the downstream result. Waterfalls power prompt assembly and tool policy; ordinary emit events notify synchronously, and parallel events await all listeners without a veto result. -An effect is setup whose cleanup belongs to a fiber. Tool registration, prompt contribution, event subscription, and an agent scope are effects, so normal disposal, failure, and hot module reload all follow the same ownership graph. +## Scope routing: one opaque key selects one layer -```js -ctx.effect(() => { - const resource = openResource() - return async () => { - await resource.close() - } -}) -``` +The scope package implements the smallest object needed for Cordis routing. Its carrier holds only a composed service filter and scope predicate, while the package records the opaque key privately and exposes the scope fiber's quiescent disposer separately. -Cordis also supports generator effects that nest child effects in a chosen teardown order. The lifecycle section explains why construction must become owner-visible before arbitrary callbacks run. +### Scope identity uses object identity -#### Waterfalls remain cooperative extension points +A `ScopeKey` is an opaque object compared by identity. The harness uses the live `Agent` as its own key, but the primitive is domain-neutral and supports other scoped owners. -A waterfall listener wraps downstream work. Calling `next()` includes the remaining listeners and base implementation; returning directly skips that downstream portion. +`createScope(parent, key)` returns a scope whose `ctx` shares the parent's services and whose effects are tagged with that key. `scopeOf(ctx)` reads the nearest registration key. `scopeTarget(base, key)` creates the event receiver whose filter preserves the base receiver's Cordis service filter, then admits unscoped listeners and listeners with that exact key. -```js -ctx.on('system-prompt/assemble', async (_assembly, _context, next) => { - const downstream = await next() - return { - ...downstream, - sections: [...downstream.sections, extraSection], - } -}) +The receiver is a small carrier rather than a transparent proxy for the domain object. Code that needs the agent receives the explicit event argument; code that needs registration ownership receives `agent.ctx`. -ctx.on('system-prompt/assemble', async () => replacementAssembly) -// The direct return skips this listener's downstream/base. An outer listener -// that already awaited next() still resumes around replacementAssembly. -``` +### Registry reads overlay one exact map -This flexibility is intentional for ordinary policy, but it cannot express a fact that must remain true after every wrapper and short-circuit. [Owner-final policy](#owner-final-policy-four-narrow-boundaries) adds only the four final checkpoints that need stronger semantics. +Scope-aware registries store global contributions separately from identity-keyed local contributions. A read resolves the global layer and at most one local layer; it never traverses parentage. -#### Dispatch receivers select scoped listeners +Each service retains its domain rule. Named prompt values and tools use local shadowing, tool restrictions filter globals before local tools are added, and events select listener audiences rather than registered data. Scope supplies identity and ownership, not a universal merge algorithm. -Cordis filters listeners using the dispatch receiver, the object visible as `this` inside a function-style listener. `dsh-scope` builds a receiver carrying the operation's scope key, allowing global listeners plus listeners registered for that exact key while rejecting other agents' listeners. +### Fused dispatch helpers prevent subject drift -The receiver is live coordination state, not durable session data. For example, `tools/result` is a live final-outcome notification, while `tool/result` is an append-only session event used for replay and model history. +`agentEvents(context, agent)` constructs the agent's carrier and injects the same agent as the event subject. Session, tool, approval, prompt, and subagent services likewise derive routing from the object they already own instead of accepting an unrelated key. -## Registration and delivery: global plus exactly one agent layer +The type marker rejects ordinary bare-receiver mistakes, and development invariants cover direct JavaScript or casted dispatch. The subject remains explicit because routing correctness and useful event data are different concerns. -Within services that adopt the agent-scope contract, one scope key controls both registered data and registered behavior. Contribution reads combine the deployment-global layer with exactly one agent layer, while scoped event dispatch admits global listeners plus the listeners for that same agent. +## Agent creation: one transaction owns the complete operation -Scope keys are opaque objects compared by identity; a live `Agent` is its own registration key. There is no name-based equality or parent traversal. +Create and resume are one asynchronous lifecycle with several phases, not several lifecycles. `AgentCreationTransaction` owns caller and factory liveness, optional cancellation, private resources, publication, rollback, and the memoized teardown observed by every owner. -### Scope mechanism: context, key, and lifetime +### Registry entries are the only live identity records -The registration context selects the layer, the scope primitive binds that layer to cleanup, and the nearest scope tag—not an inherited convenience property—selects the key. +AgentRegistry and SessionStore each keep one entry per live object. The entry holds the stable ID, object, scoped carrier, and the small amount of publication or append state that belongs to that object. -#### The calling context selects visibility and cleanup +A detach closure captures its exact entry. It deletes only when the map still points to that entry, so an old disposer cannot delete a later object that reuses the same ID. No registry rereads a mutable caller object to decide identity. -A scope-aware registry method recovers the Cordis context through which its service receiver was accessed and calls `scopeOf(context)` once while installing the registration effect. An absent key selects the global store; a key selects the per-scope store. Cleanup closes over that accepted store and key, so later context mutation or a same-named replacement cannot redirect disposal. Event listeners follow a different Cordis path: `ctx.on()` retains the registering context, and targeted dispatch reads its scope while filtering listeners; `{ global: true }` deliberately bypasses that audience filter without changing cleanup ownership. +There is no reservation API. Caller-supplied IDs are admitted at final entry. Concurrent same-ID operations may both complete private setup; exactly one final `enter()` succeeds, and every loser rolls its private resources back. Sequential reuse is valid after the earlier disposer reaches quiescence. -The [public contract](2026-07-08-agent-scope-contexts.md#registration-origin-selects-visibility-and-cleanup) owns the visibility table, shadowing rule, and `{ global: true }` listener exception. Internally, registry resolution overlays one exact identity-keyed map on the global map and never traverses an ancestry relation: +### The transaction owns preparation before awaiting it -```text -resolveLayer(agentA): - visible = copy(global registrations) - visible.overlay(registrations from agentA.ctx) - return visible -``` +The transaction is installed under both the calling Cordis context and the concrete AgentLoop factory before persistence load or setup can suspend. It also observes an optional create/resume signal until the public operation settles. -The one-overlay algorithm is why the generic primitive needs only opaque identity and effect ownership; parent/child meaning stays outside `dsh-scope`. +Create prepares a new Session. Resume loads and validates the persisted Session before preparing the same live session identity. Both paths then build the scope, agent, and driver and invoke the same setup/publication algorithm. -#### The scope primitive keeps layer and owner together +The factory stores concrete trace targets but invokes them through a caller-bound Cordis trace. This preserves dependency origin and caller ownership without stacking trace proxies. -`dsh-scope` exposes only the operations needed to mint a tagged ownership layer, read its key, target dispatch, and reach quiescent cleanup. For ordinary scope-aware registries, using a separate `{ scope }` option could express “stored in A's layer, disposed with B”; registration through the scoped context makes that mismatch unrepresentable. The `{ global: true }` listener option is an intentional audience exception, and the low-level `scopeTarget(base, key)` primitive still relies on its service-owned caller to supply matching facts. +### Setup is trusted composition inside a private world -| Operation | Responsibility | -|---|---| -| `createScope(context, key)` | Mount an ownership fiber and return its tagged context | -| `scopeOf(context)` | Read the nearest inherited scope key | -| `scopeTarget(base, key)` | Build a scope-filtered dispatch receiver around the existing base receiver | -| `Scope.dispose()` | Return one shared idempotent promise that reaches cleanup quiescence | -| `Scope.rawDispose` | Expose the exact Cordis disposer for ordered generator composition | +Setup receives the full child context and may await plugin activation. It can register tools, prompt sections, restrictions, listeners, and other effects, but the public contract does not support driving or publishing the in-flight agent through casts or internal registry calls. -`Scope.dispose()` and `rawDispose` serve different callers. Cordis raw disposers are single-shot, so a repeated raw call need not wait for an earlier asynchronous teardown; the public method follows the backing fiber's in-flight cleanup and gives racing callers the same completion promise. Generator lifecycles use `rawDispose` because Cordis recognizes nested ownership by exact disposer identity. +The transaction races asynchronous load and setup against deactivation rather than waiting forever for a promise owned by external code. If cancellation or owner unload wins, public creation rejects after transaction-owned cleanup even when the external promise never settles. -The primitive has one essential shape: +### Publication has one ordered commit path -```text -createScope(parentContext, key): - fiber = mount no-op plugin under parentContext - scopedContext = derive fiber.context with nearest-scope-tag = key +Publication admits and announces resources in the order required by observers: - rawDispose = fiber's exact disposer - dispose = memoized operation that: - invoke rawDispose if teardown has not started - follow fiber's in-flight cleanup until quiescent +1. Enter the session. +2. Enter the agent. +3. Announce `session/created`. +4. Announce `agent/created`. +5. Enable public driving. +6. Emit `agent/session-start`. +7. Start the driver. - return { ctx: scopedContext, rawDispose, dispose } -``` +The agent never drives before both registries and creation notifications agree. A synchronous listener may veto or dispose an owner; the transaction records publication in progress and waits for that callback stack to unwind before teardown continues. Every creation announcement that begins has a matching disposal announcement during rollback. -Derived contexts inherit the nearest tag. Mounting a plugin under `agent.ctx` preserves the agent scope; deliberately creating another scope replaces the tag below it. - -#### `ctx.agent` is an association; `scopeOf()` selects the layer - -`agent.ctx.agent` gives setup code convenient access to the associated agent, but the nearest scope tag remains authoritative for resolution. A nested scope can inherit the ergonomic `agent` property while replacing the registration key. - -```js -const auditKey = {} -const auditScope = createScope(agent.ctx, auditKey) - -auditScope.ctx.agent === agent // true: inherited association -scopeOf(auditScope.ctx) === auditKey // true: nearest registration key - -await auditScope.dispose() -``` - -This separation keeps the generic scope package independent of the agent package. - -### Resolution contracts preserve domain semantics - -The shared scope selects two layers, but each registry retains its own merge rules and must keep presentation, lookup, and execution coherent within the view it owns. - -#### Registries retain domain-specific merge rules - -The shared primitive answers “which layer?” and “who owns cleanup?”; each service still defines how its values combine. Prompt sections, variables, and tools use scoped-over-global shadowing by name. Tool-schema providers are additive. Tool lookup and execution receive an agent or scope explicitly, while prompt assembly receives an `AssembleContext` whose `scope` selects the layer. - -Calling a read method through `agent.ctx` does not silently choose an agent subject. For example, `agent.ctx.systemPrompt.assemble()` without an assembly scope still requests the global view. Registration origin and operation subject remain explicit, allowing one shared service to act for any agent. - -#### The tool view is live and executable - -`ToolRegistry` owns one resolver rather than separate presentation and execution stores. It snapshots restriction definitions at registration, applies them to the current global map, overlays the matching scoped map, and derives lookup, dispatch, schemas, SDK bindings, timeouts, inspection, and UI presentation from that resolved map. A filtered global implementation therefore cannot remain executable through a second path. - -The [public RFC](2026-07-08-agent-scope-contexts.md#tool-restrictions-resolve-against-a-live-flat-view) owns the exact allow/deny/future-global/local-overlay behavior. The internal distinction needed here is that `ToolRegistry.knownNames()` validates restrictions against the pre-restriction end-capability universe, while the system-prompt provider validates `toolOrder` against the presentation mode's wire universe. - -Final prompt assembly can include schemas from other `systemPrompt.tools()` providers or assembly listeners. The coherent-view proof therefore covers `ToolRegistry`'s own schemas, bindings, lookup, execution, and presentation; another provider owns coherence for the unrelated schemas it contributes. - -Reserved `run_code` presentation sits outside both registration maps. The [Code Mode RFC](../feature/2026-06-15-code-mode.md) owns its mode, SDK, and `toolOrder` semantics; this design relies only on the fact that the transport is resolved separately from filterable end capabilities. - -### Dispatch contract follows the operation subject - -Service-owned dispatch paths derive or couple the scope key with the operation subject, and a carrier composes that key with the chosen base receiver's existing dispatch behavior. For agent events the agent is both base and operation subject; tool, approval, and prompt dispatch instead wrap their owning service while selecting listeners with the operation's agent key. The low-level primitives can still represent mismatched facts, so helper use and development invariants—rather than the type system alone—protect direct internal callers. - -#### The operation subject selects the listener set - -The [public dispatch rule](2026-07-08-agent-scope-contexts.md#scoped-events-follow-the-operations-real-subject) and generated [event catalog](../../../cordis-catalog/events.md) own listener visibility and the exhaustive event-family mapping. The implementation problem is to prevent each service from choosing its carrier, subject argument, and scope key independently. - -Fused helpers keep values that must agree together. `agentEvents(context, agent)` uses one agent as the subject, scope key, and first event argument. `assembleContextFor(agent)` sets both prompt facts and the scope selector. The session store captures its carrier when a session enters because later appends and flushes may occur without the original agent context. - -#### The carrier preserves base-receiver behavior - -Function-style listeners receive the carrier as `this`. Agent listeners may call methods on the agent base; service listeners rely on the owning service's contextual receiver behavior. The carrier is therefore a proxy that selects listeners while reading, writing, and invoking through the real base receiver. - -The implementation uses a dedicated surrogate proxy target with an immutable composed-filter slot. It combines the base receiver's existing `Context.filter` with the scope predicate instead of replacing it. Methods bind to the real base; callable carriers preserve call and construct shape; descriptor queries normalize configurable flags as required by Proxy invariants; and definitions through the carrier require an explicitly configurable descriptor. Stable built-in references protect the composed filter from accidental `.call` replacement. - -Those mechanics preserve observable JavaScript behavior, including private-field method identity: - -```js -class Base { - #count = 0 - increment() { this.#count += 1 } -} - -const base = new Base() -const key = {} -new Proxy(base, {}).increment() // TypeError: proxy lacks Base's private identity - -const carrier = scopeTarget(base, key) -carrier.increment() // works: method is bound to base -carrier === base // false: carrier has distinct identity -``` - -Together these constraints keep listener selection correct while preserving the base-receiver behavior listeners expect. - -The TypeScript-only `Scoped` marker requires a carrier at typed dispatch sites. Runtime marks and development invariants cover JavaScript, casts, and direct Cordis dispatch; they detect routing mistakes but do not confine hostile same-process code. - -## Lifecycle: compose privately, publish once, tear down in reverse - -Scope, session, registry entry, and driver form one transaction with two owners. Request fields are captured first; AgentLoop tracking and both identity reservations precede asynchronous work; the caller owns the prepared lifecycle before setup; publication proceeds in synchronous observable phases; and every teardown path reaches one reverse-order quiescence boundary. - -Two services split the public API from the implementation. `AgentRegistry`, reached as `ctx.agents`, stores live agents and is the front door for `create()` and `resume()`. Its registered `AgentFactory` is concretely implemented by `AgentLoop`, which constructs and drives agents using its own injected dependencies. The rest of this section calls that concrete co-owner the **AgentLoop factory**. - -| Phase | Public state | Ownership fact | -|---|---|---| -| Reserve | IDs unavailable to competitors | AgentLoop tracking and exact reservations cover the next await | -| Prepare or load | Persistence data is loading, or session, scope, and driver exist privately | Resume's load sentinel covers persistence; the complete caller lifecycle covers setup | -| Setup | `setup(agent.ctx)` may await and register | Neither ID is published | -| Publish and start | Session, agent, and lifecycle notifications appear in order | Liveness is checked between observable phases | -| Dispose | Driver drains, registries detach, scope unwinds, IDs release | All owner paths join one completion promise | - -The [public lifecycle contract](2026-07-08-agent-scope-contexts.md#creation-publishes-after-setup-disposal-revokes-after-work-stops) defines what callers observe. The following sections justify each ownership and ordering fact behind that contract. - -### Reservations precede awaiting; lifecycle ownership precedes setup - -AgentLoop tracking and exact identity reservations precede the first await. Resume adds a caller sentinel across persistence loading; create and resume both establish the complete caller-owned lifecycle before invoking setup. - -#### The prepared lifecycle is owned before setup callbacks - -The caller context owns the work it requested and receives the consumer-facing `AgentHandle`. The AgentLoop factory is a structural co-owner because a live agent continues to depend on its injected services. Either owner can deactivate the transaction; both converge on the same lifecycle disposer. - -| Owner mechanism | Covers | Retires when | -|---|---|---| -| Caller lifecycle sentinel | Caller-fiber loss from lifecycle preparation through live lifecycle | Shared lifecycle reaches quiescence | -| Resume load sentinel | Caller-fiber loss across persistence load and lifecycle handoff | Load rollback or the adopted lifecycle reaches quiescence | -| AgentLoop tracker | AgentLoop unload and structural dependency loss | Transaction and lifecycle settle | -| ID reservations | Competing agent/session insertion | Ordered teardown releases both IDs | - -A **sentinel** is an owner-visible effect that follows work whose final disposer is not yet available. It adopts the exact reservation disposers immediately, then follows the complete lifecycle disposer once preparation establishes it. - -Cordis must make construction owner-visible before setup can reenter teardown. An effect's cleanup wrapper enters its owner list before its setup body runs, a child fiber receives its parent-owned disposer before Cordis's child-plugin notification (`internal/plugin`) announces it, and a fiber already unloading rejects new effects after taking its cleanup snapshot. Teardown observers are contained independently so one callback cannot starve peers or interrupt cleanup. These are domain-neutral lifecycle rules; `dsh-scope` uses them by mounting a no-op plugin fiber as the ownership bucket for one scope. - -#### Caller ownership and factory dependency lookup stay separate - -Factory delegation carries two contexts because ownership and dependency origin are different facts. `ownerCtx` is the caller-bound context whose fiber and optional scope own the requested lifecycle. The factory method receiver is the accepted factory traced through that access so the concrete service retains its own injected dependency view. - -```text -callerCtx.agents.create(options) - ownerCtx = context carrying callerCtx's fiber and scope - factoryThis = concrete accepted factory traced through ownerCtx - Reflect.apply(capturedCreateAgent, factoryThis, [ownerCtx, options]) -``` - -`setFactory()` captures the concrete target and its `createAgent` and `resume` callbacks once. It canonicalizes an already traced service before retracing, avoiding a second proxy layer that would break raw-identity state. Plain factory objects receive the explicit `ownerCtx` without depending on Cordis tracing. - -#### Create and resume reserve identities before awaiting - -Programmatic create and resume reserve both agent and session IDs before any operation can await. Create prepares a new or seeded session; resume loads persisted data while a caller sentinel and AgentLoop load tracker already own the interval in which no `Agent` object exists. - -Reservations are capabilities, not advisory sets. Setup code cannot reserve, prepare, create, register, or enter a substitute under the same IDs. A session reservation prepares at most one exact object, and publication requires the matching factory-held capabilities. A failed or abandoned transaction therefore cannot publish a substitute or wedge an ID indefinitely. - -Resume transfers ownership rather than opening a gap: - -```text -resume(ownerCtx, request): - snapshot request identity, options, and setup callback - reserve agentId and sessionId - install caller sentinel adopting both reservation disposers - track load under AgentLoop - - persisted = await firstOf(persistence.load(sessionId), deactivated) - session = sessionReservation.prepare(reconstruct persisted data) - starting = startOwned(ownerCtx, session, reservations, setup) - caller sentinel follows starting.dispose - return await starting.result -``` - -If deactivation wins, a backend load may still settle internally but has no path back to publication. Preparation failure still returns a rollback-backed lifecycle result, so both owners can wait for actual cleanup instead of mistaking a rejected async result for successful installation. - -### Setup composes an unpublished world - -`setup(agentCtx)` may register tools, prompt state, restrictions, listeners, protections, or child plugins and may await their activation. The new agent is available as `agentCtx.agent`, but neither agent nor session is visible in its global registry. - -The complete rollback skeleton exists before setup runs. If setup throws, rejects, or loses either owner, the scope and prepared resources unwind and the IDs become reusable. After setup settles, a microtask checkpoint and liveness checks let a same-turn owner unload win before publication. - -Setup composes but cannot drive. `send`, `steer`, `inject`, and `cancel` reject until publication reaches the session-start boundary. The driver lock and inbox use runtime-private state, and only factory-held controls enable and start the loop; JavaScript casts cannot call a public start method or write directly into the queue. - -```text -startOwned(ownerCtx, snapshot, preparedSession): - world = prepareLifecycleWithCompleteRollback(ownerCtx, snapshot, preparedSession) - - result = async: - require world active - await firstOf(runOptionalSetup(snapshot.setup, world.agent.ctx), world.deactivated) - await oneMicrotask() - require caller, factory, owner fiber, and owner agent still active - world.publish(snapshot.source) - return handle(world.agent, world.dispose) - - on any error: - await world.dispose() - rethrow -``` - -### Publication is ordered, observable, and rollback-covered - -Publication is one synchronous sequence with liveness checks between three observable notification phases. Both registry entries exist before the first listener runs, but driving stays locked until immediately before `agent/session-start`. - -1. Enter the session store and capture its scope carrier. -2. Enter the agent registry without announcing it. -3. Recheck caller and factory liveness. -4. Emit `session/created`. -5. Recheck liveness. -6. Emit `agent/created`. -7. Recheck liveness. -8. Enable driving. -9. Emit `agent/session-start`. -10. Recheck liveness. -11. Start the driver. - -```text -publish(world): - world.beginSynchronousPublication() - try: - world.detachSession = sessions.enter(world.session, sessionReservation) - world.detachAgent = agents.enter(world.agent, agentReservation) - require callerAndFactoryActive - sessions.announce(world.session) - require callerAndFactoryActive - agents.announce(world.agent) - require callerAndFactoryActive - world.driver.enableDrivingVerbs() - emitNonVetoing(agent/session-start) - require callerAndFactoryActive - world.driver.start() - finally: - world.endSynchronousPublication() -``` - -#### Creation is paired, not atomic - -Observers run between publication steps, so the sequence is not described as atomic. Effects already performed by an earlier listener cannot be retracted if a later listener throws. Instead, each registry marks a creation announcement as begun before dispatch and emits exactly one matching disposal edge during rollback. An entered object that was never announced has no disposal notification because no observer was told it existed. - -A detach requested during `session/created` or `agent/created` is deferred until that dispatch unwinds. Stable captured carriers and exact-object guards prevent a later listener from observing `disposed` before `created` or a stale detach from deleting a replacement with the same ID. The outer publication barrier likewise prevents caller or AgentLoop teardown from removing the other registry entry or unwinding `agent.ctx` while an announcement remains on the stack. - -Creation listener synchronous throws remain vetoes. Returned promise rejections are observed and logged but not awaited: publication has no asynchronous gap in which such a result could roll back safely. Disposal notifications and `agent/session-start` are non-vetoing and independently contain both synchronous throws and returned-promise rejections so one listener cannot block cleanup or later observers. - -### Teardown stops work before revoking registrations - -Every owner path reaches one memoized reverse-order transaction. It marks the lifecycle inactive, waits for an in-progress synchronous publication phase, stops the driver through actual exit and final durability work, detaches the agent and session, unwinds the scope, and releases IDs last. - -Final turn events, the turn-ending flush, and any outstanding session flush started while the agent was idle therefore run while the session and scoped listeners still exist. `agent/disposed` observes an already quiescent and unregistered concrete agent while its session remains live; `session/disposed` follows after event feed detachment and store removal. Both use the stable carrier captured for their matching creation edge. - -```text -disposeOwnedAgent(world): - mark world inactive - await world.synchronousPublicationIfRunning() - await world.stopDriver() # loop exit plus agent-started flushes - world.detachAgent() - world.detachSession() - await world.scope.dispose() - world.releaseSessionReservation() - world.releaseAgentReservation() -``` - -`AgentHandle.dispose()` gives repeated and racing consumers the same completion promise. The lifecycle-long caller sentinel follows that promise even when handle disposal wins first, while the AgentLoop ledger independently stops new transactions and waits for every structurally dependent agent before the service disappears. - -AgentLoop co-ownership follows dependency shape, not a blanket “creator owns every returned value” rule. An AgentLoop-created agent continues to depend on the loop's services, so AgentLoop unload stops it. - -## Boundary ownership: hardened paths accept once and own the accepted value - -The acceptance-sensitive paths enumerated below read caller-owned fields once and retain only owner-controlled identities or snapshots before crossing asynchronous, reentrant, model-visible, or durable-log code. This is a boundary-by-boundary implementation property, not a blanket claim about every public API. The protection is independent of TypeScript: `readonly` annotations vanish at runtime, and JavaScript accessors can return a different value on every read. - -The shared shape distinguishes identity-bearing references from data. Agent objects and abort signals are retained by identity after one read. Boundaries whose contract requires lossless JSON—such as session events and subagent payloads—validate and materialize it in one traversal; other boundaries use their own owned representation, such as `structuredClone` for agent options. Scalars and callbacks are captured once, then each boundary applies the validation promised by its API before downstream use. - -```text -accept(input): - read every relevant top-level field exactly once - retain identity-bearing references without rereading them - validate acceptance-time fields from those captures - copy or pin data in the representation owned by this boundary - bind accepted callbacks once when method receiver state is intentional - expose only owner-controlled identities, frozen records, or detached results -``` - -Capture does not imply uniform eager callback type-checking. Agent `setup` is captured once and any invocation failure enters rollback; a tool guard is likewise captured, and an invalid cast becomes a normalized execution error. The invariant is that later work never rereads caller fields to choose a different value. - -| Boundary | Identity retained | Data detached or pinned | -|---|---|---| -| Tool and `SubagentProvider` registration | Original callback receiver | Name, flags, schemas, scalar config | -| Agent create/resume | Caller context, setup callback | IDs, options, session metadata and seed | -| Agent send/steer | None | Content blocks and resolved message source | -| Approval request | Agent and abort signal | Tool name, call ID, and reason | -| Tool execution | Agent, signal, registry-minted parent token | Call identity and arguments | -| Session append/load | Session identity | Header and event envelopes | -| Subagent start/result | Parent and signal | Prompt, filters, schema, options, result | - -Before agent setup can run, the concrete agent pins its accepted ID, options, and session and binds `ctx` once. Registry detach closures likewise close over their accepted keys instead of rereading mutable public fields. - -`send()` and running `steer()` resolve the message source once and materialize `{ content, source }` as one detached, deeply frozen lossless-JSON record before `agent/queued` or inbox insertion. The notification and FIFO share that accepted content and source; its metadata wrapper is frozen separately, so neither retained caller references nor an earlier notification listener can rewrite what a later listener, the session log, or the model sees. Invalid content or source throws synchronously without notification, enqueue, or loop wakeup; idle `steer()` delegates to the same `send()` boundary. The later `agent/prompt-submit` waterfall can still replace a queued prompt by returning new content; ownership forbids in-place mutation, not the explicit rewrite protocol. - -The inbox path makes that accepted-value boundary concrete. Getter evaluation happens during materialization, so liveness is rechecked before the accepted record crosses into an inbox FIFO: +The sequence diagram isolates the non-obvious race: a synchronous creation listener can request disposal while the publication call stack still owns both registry entries. Teardown must deactivate immediately but wait for that stack to unwind before stopping and detaching anything. ```mermaid -flowchart TB - callerInput["Caller-owned content and source"] --> initialCheck["Require a live, drive-enabled agent"] - initialCheck --> accept["Resolve source once; materialize and deep-freeze one record"] - accept -->|"invalid lossless JSON"| invalidReject["Throw synchronously; no inbox insertion, agent/queued, or loop wakeup"] - accept -->|"accepted"| liveness["Recheck disposal after caller getters"] - liveness -->|"disposed reentrantly"| disposedReject["Throw disposed; do not insert or announce the message"] - liveness -->|"still live"| inbox["Insert the record into the queued or steering FIFO"] - inbox -->|"same frozen content and source"| queued["Emit agent/queued with a frozen metadata wrapper"] - inbox -->|"if later drained, read the same owned record"| drain["Loop-owned delivery"] - inbox -->|"cancel before drain"| cancelled["Clear the pending record without delivery"] - inbox -->|"disposal wins before drain"| disposed["Stop delivery; the disposed agent may retain the pending record"] - drain -->|"queued prompt"| prompt["agent/prompt-submit may block or explicitly replace"] - drain -->|"steering consumed by an active turn"| steering["Append steering/message"] +sequenceDiagram + participant Tx as AgentCreationTransaction + participant Registries + participant Listener as Synchronous listener + participant Driver + + Tx->>Tx: mark publication in progress + Tx->>Registries: announce agent/created + Registries->>Listener: invoke inside the same call stack + Listener->>Tx: dispose reentrantly + Tx->>Tx: deactivate, teardown waits for publication + Tx-->>Listener: disposal request accepted + Listener-->>Registries: return + Registries-->>Tx: announcement unwound + Tx->>Tx: resolve publication settlement + Tx->>Driver: stop and drain + Tx->>Registries: detach agent, then session + Tx->>Tx: dispose scope and resolve teardown ``` -A stateful getter shows why validation and ownership must use the same capture: +### Teardown preserves work before revoking registrations -```js -let reads = 0 -const input = { - get name() { - reads += 1 - return reads === 1 ? 'safe_tool' : 'different_tool' - }, -} +Every teardown request joins one memoized path. The order is: -// Wrong: validation and storage observe different values. -validateName(input.name) -storeName(input.name) +1. Deactivate creation or driving and let synchronous publication finish. +2. Stop and drain the driver, including idle injection flushes. +3. Detach the agent. +4. Detach the session. +5. Dispose the agent scope. +6. Retire transaction ownership tracking. -// Right: one accepted value drives both. -reads = 0 -const acceptedName = input.name -validateName(acceptedName) -storeName(acceptedName) -``` +This order lets final agent and session events use the matching scoped listeners and keeps persistence observers attached through the final flush. Scope disposal comes last because registration revocation is the externally visible lifetime boundary. -### Registered definitions are frozen snapshots +## Session append: materialize, validate, commit, notify -Tool registration creates the stored definition identity once; changes occur through explicit unregister/register effects rather than mutation of a caller-retained object. Parameters are materialized in one traversal, callbacks bind once to the accepted definition receiver, and the stored record is deep-frozen. +Session events cross a durable boundary, so append owns their data. The rest of the algorithm uses one attached entry and one commit point. -The first-party `defineTool()` helper applies the same boundary before registration. It captures each option once, materializes the authoring `SchemaSpec`, and derives both the wire schema and later execution/presentation validation from that owned spec. +### Durable data is materialized once -```text -defineTool(options): - accepted = read each option exactly once - parameterSpec = snapshotLosslessJson(accepted.parameters) - wireSchema = snapshotLosslessJson(convertToJsonSchema(parameterSpec)) - build execute and presentation validation over parameterSpec +Session headers, seeds, and appended events are lossless JSON data. The Session constructor or append path materializes and validates them before storage and exposes frozen snapshots, so later caller mutation cannot change persistence, replay, or model reconstruction. -registerTool(context, definition): - accepted = read each definition field exactly once - stored = deepFreeze({ - accepted name, description, timeout, - parameters: snapshotLosslessJson(accepted.parameters), - execute: bind accepted.execute to definition, - presentation callbacks: bind accepted callbacks when present - }) - layerFor(scopeOf(context)).add(stored.name, stored) -``` +This is a real ownership boundary: the values leave the caller, may be persisted, and must reconstruct the same request later. It is intentionally stricter than a typed same-process callback or registry definition. -`get()` and `visible()` return the frozen stored definitions; `schemas()` returns detached projections. Replacing `definition.execute` after registration has no effect, while a callback can deliberately read live state from its closure or original receiver. +### Pre-commit listeners can veto; post-commit observers cannot -Factory and backend registration use different reentrancy orderings around the same ownership rule. `AgentFactory` registration claims its single slot before reading callback accessors. `SubagentProvider` registration first snapshots the provider fields, then its effect checks and enters the accepted name. Both capture callback identity and intentional receiver state once, and hot-reload cleanup closes over the accepted slot or key instead of rereading a mutable public property. +Append follows one sequence: -### Durable session ownership carries the scope key +1. Materialize the durable event and surface intent. +2. Claim the SessionEntry and reject reentrant append on that entry. +3. Resolve scoped callbacks and run internal invariant validation. +4. Push exactly once; this is the commit point. +5. Notify each observer independently, containing synchronous and asynchronous failures. +6. Release append state and honor a detach requested during publication. -The [session-immutability RFC](2026-06-11-dev-invariants-over-deep-readonly.md#session-owns-immutable-history) owns header, event, and snapshot semantics. Agent-scope correctness adds one requirement: the store keeps append publication, accepted registry IDs, and captured scope carriers in private owner state rather than caller-writable fields. Outside JavaScript therefore cannot rename a stored session or redirect later `session/event` delivery by mutating visible state. +No observer error makes a committed event look uncommitted, and one bad listener cannot starve later listeners. Session invariants stage their transition before commit and apply it only when the same event reaches the contained post-commit observer. -An entered session treats append as one synchronous acceptance-and-publication boundary: +`flush()` starts every persistence listener and awaits every result before reporting failure. This deliberate all-settled behavior prevents a synchronous failure from starving another backend or final flush. -1. Capture the current store attachment and its private attachment epoch, keep the attachment live, then materialize and deep-freeze the caller's event data. -2. Reject if caller getters changed either value; the epoch catches even a transient attach-then-detach that restores the original hook lookup. The event must not become live without the store hooks that accepted it. -3. Resolve the exact scoped `session/event` callback list before commit. Cordis runs `internal/dispatch` during this step, so development invariants can still reject a bad candidate while the log is unchanged. Resolution uses a throwaway mutable argument array; replacing its accepted session or event rejects before commit, and product callbacks later receive a fresh fixed tuple. -4. Push the event into the log. This is the commit point. -5. Invoke the captured callbacks with per-listener containment and best-effort non-throwing failure reporting, then release the attachment barrier and honor any detach requested during acceptance or publication. +## Trust boundaries: copy only when ownership actually changes -The boundary rejects a reentrant `append()` until the outer callback list drains. Without that guard, an early observer could append event N+1 before a later persistence observer had received event N, reversing delivery relative to the log. Detach is deferred for the same interval, so no event can commit after `session/disposed` or lose its publication hooks. Once the push occurs, synchronous observer throws and returned-promise rejections are logged and contained rather than escaping as a false append failure or starving later observers. +The runtime distinguishes typed in-process contracts from serialization and durability boundaries. This is the main simplification rule for values and callbacks. -`SessionStore.flush()` uses the same pre-dispatch fixed-tuple check but remains an awaited durability barrier rather than an observe-only publication. It starts every captured listener synchronously, converts a synchronous throw into that listener's rejected result so later listeners still start, waits for every result to settle, and only then rejects with the first failed listener in registration order. One broken backend therefore cannot make the caller return while another backend is still flushing. - -Approval requests follow the same async boundary at smaller scale: one capture preserves exact agent/signal identities, copies scalar fields, captures the session once, and drives `approval/asked`, scoped policy, cancellation, and `approval/decided` from that record. - -### Tool execution has pipeline-owned identity - -The [interception-seams RFC](../feature/2026-06-30-interception-seams.md) owns the public tool-pipeline contract. For agent-scope correctness, `ctx.tools.execute(input)` must turn caller-owned input into one pipeline-owned `ToolExecution` before any scoped policy or dispatch runs. It first reads `callId` and `name` once and requires strings; a failure there rejects because even an error result would lack trustworthy correlation identity. Once those strings are accepted, later input failures can become normal final error outcomes. - -Arguments are materialized once and deep-frozen. The registry assigns a frozen property-free `ToolExecutionToken`; callers cannot choose it. `token`, `callId`, `name`, `arguments`, `agent`, and optional opaque `parent` token become non-writable and non-configurable before policy. `signal` is the only operational field an around-dispatch wrapper may replace or remove. - -```text -prepareExecution(input): - callId = read input.callId exactly once - name = read input.name exactly once - require both are strings - - accepted = read arguments, agent, parent, and signal exactly once - require parent is absent or a registry-minted token - arguments = deepFreeze(snapshotLosslessJson(accepted.arguments)) - - execution = { - token: new frozen property-free object, - callId, name, arguments, - agent: accepted.agent, - parent: accepted.parent, - signal: accepted.signal - } - protect every field except signal - return execution -``` - -Stable execution identity prevents middleware from changing which tool or scope policy accepted. It also gives structured-output commit a safe `WeakMap` key when an adapter reuses a string call ID. Code Mode correlates an SDK sub-call with its enclosing `run_code` using only the outer execution's opaque token, never a mutable reference to the live outer object. - -Result boundaries apply the same ownership rule. Each transform returns data that is captured field-by-field, validated, materialized, and ultimately deep-frozen for final observers; malformed outcomes normalize to JSON-safe error results rather than reaching the session log as apparent success. - -## Owner-final policy: four narrow boundaries - -Waterfalls remain the ordinary extension mechanism; each of four protocol invariants runs after the last extension point capable of violating that specific invariant. Each owner-final API has the weakest one-way power that can preserve its guarantee. - -Here **canonical** means the named registry or tool-schema-provider output assembled before the waterfall—not “all output the service approves.” Protection restores only the names its owner declares. - -| Invariant | Cooperative extension point | Owner-final boundary | Guarantee | -|---|---|---|---| -| Named prompt/tool contribution | `system-prompt/assemble` waterfall | `systemPrompt.protect()` finalization | Canonical presence, absence, definition, and local anchor survive | -| Non-overridable tool denial | `tools/pre-execute` allow/deny/ask waterfall | Synchronous `tools.guard()` | A denial cannot become allow | -| Authoritative live outcome | Execute and post-execute waterfalls | Awaited `tools/result` notification | Observers receive one immutable final result | -| Terminal protocol completion | Continuation waterfall and pending steering | Serial `agent/turn-stop` | No middleware or late steering creates another step | - -### Prompt protection restores named canonical contributions - -`systemPrompt.protect({ sections, tools })` snapshots the requested names and restores their canonical registry or tool-schema-provider output after the complete assembly waterfall. Global and matching scoped protections compose by set union; a waterfall failure still fails assembly rather than triggering recovery. - -Protection covers both presence and absence. If the canonical assembly omits a protected name, finalization removes a listener-fabricated entry; this is how Code Mode keeps a native schema absent while preserving the SDK/transport form. Tool providers likewise expose one captured coherent record for schemas and optional known names, so a stateful getter cannot validate one name and display another. - -#### Global section protection reserves its name - -A globally protected section name cannot be shadowed by a scoped section. Scoped registration under an already protected name fails, and adding protection fails if a scoped shadow already exists. This check occurs before assembly because scoped-over-global merge would otherwise make the shadow itself appear canonical. - -Tool-schema protection does not create a blanket reservation for unrelated schema names. Providers are additive and may deliberately contribute other executable schemas; the owner-final guarantee covers only the named canonical contribution. - -#### Restoration preserves a useful local anchor - -Protection does not reset the whole assembly. It removes protected names from the waterfall result and reinserts each canonical entry before the first surviving later unprotected canonical neighbor, or at the end if none survives. Unprotected entries retain the order and definitions chosen by middleware. - -```text -assemble(context): - assembly = assemble registries for context.scope - canonical = snapshot protected section/tool inputs - transformed = await systemPromptAssembleWaterfall(assembly) - - for each protected canonical name: - remove every transformed entry with that name - if canonical includes the name: - insert before first surviving later canonical neighbor, else append - - return transformed -``` - -Code Mode globally protects `tools:sdk` and reserved `run_code`; structured output adds scoped protection for its instruction and capture schema. - -### Tool guards deny monotonically - -`ctx.tools.guard()` installs a global or scoped synchronous check after the complete `tools/pre-execute` waterfall and before dispatch. A guard returns a denial reason or `undefined`; it has no allow result. - -Pre-execute hooks still compose ordinary allow, deny, and ask decisions. An ask resolves through the optional approval service, where only `allowed-once` becomes allow and absence or any non-grant becomes deny. Guards run afterward, so listener order cannot convert their denial into dispatched work. - -```js -agent.ctx.on( - 'tools/pre-execute', - async () => ({ kind: 'allow' }), - { prepend: true }, -) - -agent.ctx.tools.guard(execution => - execution.name === 'bash' - ? 'reviewer agents are read-only' - : undefined, -) -``` - -Even a later prepended allow listener cannot bypass the guard. A denied call still becomes an error outcome that flows through result transformation and final observation. - -### `tools/result` observes the final live outcome - -For a successfully prepared execution, the live pipeline is `tools/pre-execute` → guards → `tools/execute` → `tools/post-execute` → `tools/result`. Malformed non-identity input instead takes the error-shell path directly to final observation, as the algorithm below shows. The first, execute, and post stages are transformable waterfalls; `tools/result` is an awaited observe-only notification after every transform and outer error normalization. - -Every observer receives the same frozen execution and a separate deep-frozen snapshot of the owned result returned to the caller. Listener failures are contained independently, so they cannot change that returned result or starve peers. Routing uses `execution.agent`. - -`tools/result` is not the durable `tool/result` session event. The live notification also fires for direct programmatic executions and is the source of truth for in-process commit logic. The agent loop later appends the durable event for replay, UI reconstruction, and model history. - -```text -execute(input): - accept trustworthy callId and name - try to prepare pipeline-owned execution - on preparation failure: - create an identity-bearing error shell - ownedResult = owned error result - freeze execution - observerResult = deepFreeze(snapshotLosslessJson(ownedResult)) - await every tools/result observer independently with observerResult - return ownedResult - - gate = await tools/pre-execute(execution) - resolve ask through approval when needed - denial = policy denial or first guard denial - - if denied: - result = errorResult(denial) - else: - result = await tools/execute(execution, dispatchRegisteredTool) - - result = await tools/post-execute(execution, result) - ownedResult = normalize into owned lossless JSON - freeze execution - observerResult = deepFreeze(snapshotLosslessJson(ownedResult)) - await every tools/result observer independently with observerResult - return ownedResult -``` - -Waterfalls transform only at their named stages; guards only deny; final observers only observe. - -### `agent/turn-stop` makes continuation terminal - -Steering is input for another model step inside the current turn; queued prompts wait for a future turn. Ordinary continuation remains extensible: the loop computes a default, runs `agent/turn-continuation`, records any force-continue reason as steering, and treats pending steering as a reason to continue. - -The scoped serial `agent/turn-stop` checkpoint runs after that folding. A listener returns `{ action: 'stop' }` or abstains with `undefined`; malformed values and throws close the current turn with an error. A stop is terminal, so later listeners and steering cannot restore continuation. - -The loop uses `strictSerial` because ordinary Cordis serial dispatch treats `null` and `false` as abstentions. This terminal protocol permits only `undefined` to abstain, making accidental return values fail closed. - -Terminal state remains active through `turn/end` and the durability flush. Steering added by continuation, turn-close, or flush listeners is discarded after a terminal stop, while the ordinary queued-prompt FIFO remains untouched. - -```text -afterSuccessfulStep(turn): - decision = await agent/turn-continuation(defaultDecision) - record decision.reason as steering when present - if steering is pending: decision = continue - - terminal = await strictSerial(agent/turn-stop) - if terminal == stop: - discard steering - terminalStopped = true - decision = stop - - append turn/end - await session/flush - - if terminalStopped: - discard steering added by turn/end or flush listeners - else: - move leftover steering to the next-turn queue -``` - -This stronger control is reserved for terminal protocols such as a completed structured child; ordinary continuation policy remains cooperative. - -## Subagents: the composition proof - -In-process subagents add no second scoping model. They create a fresh flat child scope during unpublished setup, install ordinary scoped persona/filter/protocol registrations, own the child through a run handle, and use the same owner-final checkpoints for structured output. - -The roles and phases are explicit: - -| Role | Responsibility | +| Boundary | Ownership rule | |---|---| -| Caller | Supplies parent, prompt, optional child configuration, and eventual disposal | -| `SubagentService` | Validates capabilities, owns the public wrapper, normalizes result and lifecycle telemetry | -| `SubagentProvider` backend | Chooses transport and creates one run | -| In-process driver | Owns child creation, setup, prompt drive, result read, cancellation, and teardown | -| Child `Agent` | Uses the ordinary agent lifecycle and its fresh `agent.ctx` | +| Typed service/plugin call in the same process | Borrow readonly values and callbacks | +| Parsed plugin configuration or external file | Validate semantic and structural input | +| Queued inbox message | Materialize before asynchronous consumption | +| Model/tool JSON input or output | Materialize at the model/tool boundary | +| Durable session or persistence data | Materialize and validate before commit | +| Worker, process, or wire message | Serialize, validate, and own the decoded value | -```text -recommended caller order: start -> await run.started -> await run.result -> await run.dispose() -internal observation: started and result may settle in either order; lifecycle publication waits for started -ownership: dispose may race any phase and joins one cleanup promise -``` +Tests that fabricate hostile getters, replace typed callbacks after handoff, or cast fake service objects do not define a production contract by themselves. The runtime keeps checks where data crosses a parser, queue, model, durable, file, worker, process, or wire boundary and relies on readonly types plus plugin discipline inside the trusted process. -The [agent-scope contract](2026-07-08-agent-scope-contexts.md#subagents-use-the-same-composition-rule) gives the contributor-facing example, and the [subagent capability RFC](../feature/2026-06-21-subagent-capability-seam.md) owns the public `SubagentRun` contract. This section follows only the in-process ownership and terminal-protocol implementation. +Callback containment is separate from data ownership. Listeners are arbitrary extension code and can throw even when their arguments are trusted; publication and post-commit paths still contain failures according to their event contract. -### The child world uses ordinary registrations +## Tools and prompts: one view, one execution identity, explicit finality -A child persona is a scoped `deployment:persona` section. Its tool filter is a scoped restriction over the live global tool layer. Structured output is a bundle of scoped tool, prompt, protection, guard, and listener registrations. +Tool presentation and execution share one private resolver, while prompt/tool owners declare the few contributions that cooperative middleware may not alter finally. No second registry mirrors ownership. -```js -let structured -const setup = childCtx => { - if (persona !== undefined) { - childCtx.systemPrompt.section({ - name: 'deployment:persona', - order: 0, - text: persona, - }) - } - if (toolFilter !== undefined) childCtx.tools.restrict(toolFilter) - if (schema !== undefined) { - structured = attachStructuredRuntime(childCtx, schema) - } -} -``` +### One resolver defines the tool view -The driver creates one run-owner fiber under `parent.ctx` and calls the child factory through it. Parent teardown, `spawn` backend teardown, and manual run disposal reach the same node, but the child still receives a new registration key. Lifetime inheritance therefore does not imply registration inheritance. +The private resolver applies the current presentation mode, live global restrictions, exact local overlay, and local shadowing. Schemas, lookup, execution, Code Mode SDK generation, restriction validation, and owner-final name derivation all use that resolver or its pre-restriction global-name view. -### Structured output is a child-owned terminal protocol +The [subagent composition-controls RFC](../feature/2026-07-12-subagent-persona-tool-filter-and-depth.md#tool-filtering-is-one-live-global-view-rule) owns the user-visible allow/deny semantics. The implementation requirement is agreement: a filtered-away global cannot remain executable through a different lookup path, and a locally shadowed definition is the same definition presented and executed. -A structured child registers a real-schema `structured_output` tool and instruction in its own scope. Concurrent children can use different schemas without a global placeholder, reference count, or remove-for-everyone pass. +`ToolRestriction` accepts readonly allow/deny names and compiles them into internal sets. Multiple restrictions intersect. Public `visible()` and `knownNames()` methods are unnecessary because only the registry needs the intermediate views. -The [Code Mode RFC](../feature/2026-06-15-code-mode.md) owns advertised wire routes and SDK behavior. The correctness distinction here is execution nesting: a native capture has one tool execution, while an SDK capture is an inner execution whose parent token identifies the enclosing `run_code`. Tool mode is presentation rather than an execution allowlist, so a direct unadvertised capture still follows the native commit path; a deployment that forbids that route uses an execution guard. +### Tool execution owns identity and boundary materialization -Named protection restores this child's canonical capture contribution and instruction without erasing unrelated schemas deliberately added by another assembly provider. +The registry assigns every execution a fresh branded `Symbol` token. Nested Code Mode calls carry the outer token as `parent`, so structured output can correlate an inner capture with its enclosing `run_code` result by identity. -#### Native calls commit once; Code Mode SDK calls commit twice +A fresh registry-assigned Symbol provides collision-free execution identity without a WeakSet membership registry. Callers cannot supply the execution's own token through `ToolExecutionInput`; they only receive the pipeline-owned `ToolExecution` after the registry creates it. This is a trusted typed contract, not a runtime defense against arbitrary casts or JavaScript callers. -The capture body validates and stages a cloned value by stable `ToolExecution` identity. The scoped final-result observer commits a native capture only if that exact execution's final result succeeds. +Arguments are materialized once where model/tool JSON enters the pipeline. Pre-, around-, and post-execute listeners operate on the typed execution and decisions. Call ID correlation, approval, monotonic guards, and Code Mode nesting remain explicit relational checks. -A schema-validation failure becomes the ordinary `INVALID_ARGS` tool result, so the model can correct the value and call the capture tool again within the same turn. +After the last post-execute listener, the registry materializes and freezes the accepted final result once. Every `tools/result` observer receives that exact committed object, and observer failures are awaited and contained individually. An outer pipeline failure is normalized into a committed error result, so observers can discard staged work against the same authoritative boundary. -```text -structured_output.body(value, execution): - validate value against this child's schema - staged[execution] = clone(value) - return ordinary success +### Contribution-owned finality protects only named invariants -on tools/result(execution, finalResult): - if execution is staged: - value = staged.remove(execution) - if finalResult succeeded: - captured = value -``` +Most prompt assembly remains a cooperative waterfall: listeners may reorder, replace, or remove ordinary sections and schemas. A contribution sets `ownerFinal: true` only when its owner must retain final control over that named entry. -For a Code Mode SDK call, successful inner observation records a pending value against the opaque outer `run_code` token. Commit waits for the outer transport's own successful final result because an inner side effect can succeed while the program or its post-policy still fails. +Prompt sections carry owner-finality directly. Tool definitions carry it through the tool provider's `ownerFinalNames`, including canonical absence when a presentation mode intentionally omits a tool. `tools:sdk`, `run_code`, and structured-output instruction/schema contributions use this flag. -```text -on tools/result(innerStructuredCall, innerResult): - if innerStructuredCall is staged: - value = staged.remove(innerStructuredCall) - if innerResult succeeded: - pending = { outerToken: innerStructuredCall.parent, value } +An owner-final name is reserved across the global and scoped layers: a scoped shadow cannot be added beneath a global owner-final contribution, and a global contribution cannot become owner-final while any scoped shadow already exists. This makes the registered owner definition unambiguous before assembly begins. -on tools/result(outerRunCodeCall, outerResult): - if pending.outerToken == outerRunCodeCall.token: - value = pending.value - pending = none - if outerResult succeeded: - captured = value -``` +Assembly takes one private canonical snapshot before the waterfall. After listeners finish, it restores only owner-final names to their canonical presence, absence, definition, and relative anchor among surviving entries. Unrelated listener additions and reordering remain untouched. -Once capture is staged against an outer transport or committed, the scoped guard denies later calls in that response. After commit, `agent/turn-stop` ends the turn after ordinary continuation and steering fold. A child that otherwise completes cleanly without a committed capture returns an error rather than being re-prompted; requesting a schema makes output mandatory, not guaranteed. +Attaching finality to the owning contribution has two benefits. Registration and cleanup cannot drift from a separate protection registry, and the reader can see why a particular prompt/tool entry is special at its definition. -### The run protocol separates acceptance, readiness, result, and disposal +### Structured output commits only authoritative outcomes -`SubagentService.start()` returns synchronously, but `run.started` is the publication boundary. Callers treat the child as live only after readiness, consume `result`, and always dispose the run. +Structured output uses the final prompt/tool boundaries as a two-phase commit. The child-scoped `structured_output` tool and its instruction are owner-final; the tool body validates a candidate and stages it by the current `ToolExecution`, but successful capture is decided only by immutable `tools/result` observations. -Pre-readiness cancellation of an in-process run deactivates the run-owner fiber, prevents publication, rejects `started`, resolves `result` as `aborted`, and emits neither subagent lifecycle edge. +For a native call, the observer deletes the stage and commits its value only when that exact execution's final result succeeds. A post-execute block or outer pipeline failure therefore cannot leave a captured value behind. -`SubagentProvider` registration captures name, capability flags, the `inheritsParentContext` conversation-history descriptor, and the bound start callback once. The descriptor says whether completed parent turns seed the child's conversation; it says nothing about scope, services, tools, or authority. +For a Code Mode SDK call, the inner successful result records `{ parentToken, value }` rather than committing. The observer waits for the `run_code` execution whose token matches `parentToken` and commits only if that outer final result also succeeds. Program failure, runtime abort, or outer post-policy denial discards the pending value. -Starting a run captures every request field once. Parent and abort signal remain identity references; prompt, filter, schema, and options are detached lossless JSON; fixed `persona` and absolute `maxDepth` values validate before backend ownership. The in-process backend separately snapshots its optional session seed, and the service snapshots the terminal result when it settles. +Once a value is pending or committed, a scoped monotonic guard denies later tool calls. After commit, the ordinary serial `agent/turn-stop` listener returns a stop decision after continuation and steering have already folded. A schema-validation failure remains an ordinary `INVALID_ARGS` tool error and leaves the child able to retry within the same turn. -Depth validation repeats at each public entry while one helper owns the accepted domain: +Pure Code Mode omits `structured_output` from native wire schemas and exposes it through the generated SDK. Contribution-owned finality preserves that canonical absence, preventing an assembly listener from fabricating a second native route while keeping the instruction and SDK declaration intact. -```text -tool-subagent plugin load: - assertSubagentMaxDepth(config.maxDepth) +### Four final boundaries have four narrow powers -SubagentService.start(request): - capture and validate request.maxDepth +Owner-final behavior is not a general priority system. Four domain owners need four different one-way powers after cooperative extension points: -startInProcessRun(request): - capture and validate request.maxDepth - parentDepth = validated depthOf(parent) - childDepth = parentDepth + 1 - reject if childDepth is not a safe integer - reject if maxDepth exists and childDepth > maxDepth -``` - -Only `undefined` means parent depth zero. Present depth and cap values must be non-negative safe integers and must not be negative zero; derived overflow rejects even when no request cap exists. - -The service does not expose the backend-owned run handle directly. It captures `id`, `started`, `result`, and methods once; binds methods to that handle; wraps result in one detached frozen record; and installs a shared disposal promise before calling untrusted backend cleanup. Once a callable backend disposer has been captured, a malformed later field triggers rollback; if no callable disposer can be captured, rollback is impossible and acceptance fails immediately. A backend disposer that directly returns the wrapper's reentrant promise is rejected as a cycle instead of hanging. - -```text -startInProcessRun(backendContext, acceptedRequest): - install backend ownership - attach accepted abort signal - create run-owner fiber under accepted parent.ctx - create child through runOwner.ctx.agents with unpublished setup - - started = child creation publication - result = after started: - send accepted prompt - await child idle - derive owned terminal result - dispose = dispose run owner and await quiescence - -SubagentService.start(...): - backendRun = backend.start(detached request) - serviceRun = freeze accepted id, readiness, bound methods, normalized result - observe result immediately - after readiness: - emit subagent/start, then buffered/eventual subagent/end - on readiness failure: - emit neither lifecycle edge -``` - -The service observes result settlement immediately even while readiness is pending, preventing an early rejection from becoming temporarily unhandled. Lifecycle listeners receive one frozen payload; their throws and returned-promise rejections are contained independently and cannot veto the run. - -## Workflow integration preserves the subagent contract - -The [dynamic-workflows RFC](../feature/2026-07-05-dynamic-workflows.md) owns workflow behavior. The agent-scope concern is whether the worker bridge preserves the same readiness, terminal-claim, and bounded-cleanup boundaries across a message port. It never announces an unready child, never lets cleanup rewrite an already chosen result, and never suppresses disposal merely because another terminal fact already won. - -The worker executes the workflow script and exchanges protocol messages; the host owns `SubagentService`, which invokes `SubagentProvider` backends and returns normalized run wrappers that the host retains. Their lifetimes follow dependency shape: an AgentLoop-created agent stops when its loop unloads, while a workflow run captures its holder-bound `SubagentService` at start, so unloading the workflow engine prevents new runs without revoking an already returned run. - -Three state dimensions remain separate: - -| Dimension | Question | Winning rule | +| Boundary | Final power | Why ordinary listener order is insufficient | |---|---|---| -| Admission | May a worker message still start or announce a child? | Closed admission refuses the exact run and cleans it up | -| Terminal claim | Which external result does the workflow expose? | Earlier accepted external cancellation wins; otherwise first result/death claim wins | -| Physical cleanup | Which registered children and worker resources remain? | Every path may still dispose survivors through per-call gates | +| Prompt assembly | Restore named canonical contributions | A later listener can remove or replace an invariant schema or instruction | +| Tool pre-policy | Deny monotonically | A later listener must not re-allow an already denied call | +| Tool result | Observe the immutable committed outcome | Structured output must commit only the result that actually escaped the pipeline | +| Turn continuation | Stop after ordinary continuation folding | A committed terminal output must end the turn | -### Child admission waits for readiness +`ToolGuard` remains the monotonic policy registry. Final tool observation is the contained `tools/result` point described above. Terminal structured output listens on the ordinary serial `agent/turn-stop` fold after normal continuation and steering decisions; no public `strictSerial()` dispatcher is needed for the typed listener contract. -After `SubagentService.start()` returns its normalized wrapper, the host registers that exact wrapper before awaiting, attaches result observers immediately, and rechecks admission both then and when `started` settles. A closed boundary claims cancellation and disposal for that exact entry, removes it only when disposal settles, and reports `ChildStartError` only while the worker reply channel remains open. +### Skill and approval services trust typed callers -The backend's nested `start()` may synchronously reenter workflow cancellation before the service wrapper reaches the host registry. The immediate post-start check and exact-wrapper identity guard close that interval; a backend that later fulfills its own readiness cannot resurrect workflow admission. +Skill registry definitions and approval policies are readonly same-process contracts. Their services do not clone callback objects or defend against post-handoff callback replacement. -```text -after subagents.start returns its run wrapper: - register exact wrapper for cancellation - observe and snapshot result immediately - if admission closed: refuse and clean exact wrapper - else await run.started +Skill still validates external skill files and parsed provider output, routes catalogs through the calling agent's tool view, and disposes registrations exactly. Approval still resolves policy, observes cancellation, routes `approval/request` by `request.agent`, records the durable audit pair, and contains answerer and post-commit observer failures. - on ready: - if admission closed: refuse and clean exact run - else send ChildStarted, then buffered/eventual outcome +## Subagents: readiness is the start promise - on readiness failure: - send ChildStartError only if the worker reply channel remains open - dispose exact wrapper if still registered -``` +Subagent startup has one ownership transfer. The provider owns partial resources until its start promise fulfills with a ready published run; the caller owns the returned run and must dispose it. -### Each terminal contender claims before its own callbacks +### The service contract has one cancellation channel -Each terminal path records the state it owns before invoking its own callback fanout. External `cancel()` records the accepted cancellation reason before invoking child cancellation. On the Result path, the worker queues its `Result` message before settlement cleanup messages on the same port, and the host records the winning result before any Result-triggered abort or cancellation. Reentry therefore observes the fact that already won instead of rewriting it. +`SubagentProvider.start()` and `SubagentService.start()` return `Promise`. The promise fulfills only after the backend has established the child it promises, so callers and `subagent/start` observers never need a second `run.started` readiness promise. -```text -on workflow Result: - cancellationWasAlreadyAccepted = external cancellation is in flight - claim chosen result: - if earlier external cancellation and result is not cancelled: - cancelled result - else: - worker result +`SubagentStartRequest.signal` is required. Aborting it requests cancellation during startup and after readiness. `SubagentRun.dispose()` also requests cancellation and awaits quiescence. There is no separate public `run.cancel()` channel. - if not cancellationWasAlreadyAccepted: - abort shared child-request signal - cancel every registered child through its at-most-once gate - settle chosen result -``` +Optional `sendMessage()` supports a live backend that can accept steering. Optional `resume()` returns `Promise` because the resumed child has the same asynchronous readiness boundary. -The worker may also send a later `ChildCancel`; host fanout and the worker message share one per-call cancellation gate, so an arbitrary backend's `cancel()` need not be idempotent. Each callback is contained independently. +The service validates provider capabilities and request semantics before calling the provider. A provider rejection cleans any partial resources before the rejection escapes and emits no `subagent/start`/`subagent/end` pair. After fulfillment, the service attaches result observation, emits scoped start, and returns the run. Provider removal prevents later starts but does not revoke a run already accepted by the provider. -### Worker death, exit, and disposal remain separate +### In-process providers reuse the core transaction -The first worker death signal closes message admission, claims a death result unless an earlier terminal fact won, cancels and disposes registered children, and synthesizes missing lifecycle ends. A queued message can arrive between Node's `error` and `exit`, so the logical admission barrier—not physical exit—prevents late child creation or narration. +Spawn and fork share one in-process driver. It creates the child through `parent.ctx`, passes the required signal into the core creation transaction, and installs persona, tool restriction, and structured-output contributions during unpublished setup. -Physical exit performs a final disposal-only sweep without repeating explicit cancellation. A cancellation grace period bounds how long the host waits for cooperative settlement before terminating the worker; a grace result can already be chosen while exit cleanup still needs to dispose surviving child handles. The bound is real: after grace expires, public disposal may return after invoking child disposal and reaping host resources even if a slow backend disposer has not reached quiescence. +The provider awaits creation and returns only the published run. At the handoff, core creation detaches its creation-only abort listener; the provider immediately rechecks the signal before installing the live-run listener, so an abort in that narrow interval disposes the new handle instead of escaping cancellation. Parent teardown follows the child because the operation belongs to `parent.ctx`; provider unload blocks new starts but does not become a second revocation owner for accepted runs. The run disposer cancels the child and awaits the AgentHandle's ordered teardown. -Public `handle.dispose()` claims its shared promise before invoking cancellation or child callbacks. Each `disposeChild` likewise claims its call-ID promise before invoking the backend disposer. Public-first reentry joins the public promise; worker-first reentry lets the holder traversal join the already claimed child promise. Settled `dispose()` still drives a host-side reap before awaiting quiescence, so a fire-and-forget child cannot remain alive merely because workflow result settlement already occurred. +Spawn uses an empty session seed. Fork uses a validated completed-turn prefix. Conversation seeding changes history only and does not import scope, tools, services, or authority. -Together these rules ensure `workflow/agent-start` names only ready children, external result precedence is stable, and every surviving child reaches disposal. +### ACP providers own the process until readiness or cleanup + +An ACP provider crosses a real process and wire boundary, so it retains validation, environment scrubbing, message serialization, abort/process races, and kill-to-exit quiescence. + +Start resolves only after `initialize` and `newSession` succeed. Abort, spawn failure, RPC failure, or invalid startup response reaps the process before rejection. After readiness, result maps the ACP prompt outcome and streamed output; dispose requests cancellation, closes the connection, and awaits process exit through one memoized path. + +## Workflows and ACP UI: retain only independent async facts + +Worker and editor bridges need more state than same-process registries because messages, process death, and rendering can settle independently. Their state is organized around those real facts rather than duplicate cancellation protocols. + +### Workflow children are pending starts or published records + +The workflow host keeps pending provider-start promises and published child records. A child moves from pending to published only when async `SubagentService.start()` fulfills; rejected starts clean their partial provider work and produce no child lifecycle pair. + +One host-owned AbortController supplies the required signal to pending and live children. Closing workflow admission aborts that signal, so there is no duplicate `ChildCancel` worker RPC or explicit host-side `run.cancel()` fanout. Quiescence waits for both pending starts and published child disposal. + +The worker boundary still serializes requests and outcomes. The host retains first-terminal-outcome arbitration, exact child accounting, worker-death handling, grace termination, late/duplicate message rejection, and bounded cleanup because result receipt, worker exit, and child quiescence are genuinely independent facts. + +### Terminal result and physical cleanup remain separate + +The workflow result records the first accepted terminal outcome according to the public precedence rules. Cleanup can continue after that result is chosen: live children still need disposal, a worker still needs termination, and a slow external backend may outlive the configured grace bound. + +Public disposal claims its memoized promise before invoking callbacks. Worker death closes admission before processing any queued late child request, synthesizes missing lifecycle ends, and starts child/process cleanup without rewriting an outcome already claimed. + +### ACP prompt settlement does not depend on rendering success + +The ACP UI correlates a prompt with its observed turn directly. It does not scan from a `logWatermark` or use session status as a second reconciliation oracle. + +Prompt handling settles correlation in a `finally` around transcript rendering. A rendering failure can fail presentation, but it cannot skip prompt settlement or leave the session permanently in flight. Concurrent loads of the same persisted caller-supplied session ID remain excluded because that is a real persistence identity race, not a UUID collision concern. ## Correctness enforcement -The runtime rule is checked at four escape boundaries: API shape couples related subjects, TypeScript marks typed dispatch, development invariants inspect actual dispatch, and repository gates keep declarations aligned with enforcement. +The design is enforced at types, runtime escape points, generated contracts, and behavioral tests. No one layer is asked to prove what it cannot observe. -### API shape couples values that must agree +### Types make the ordinary path hard to misuse -`agentEvents(context, agent)` couples carrier, subject, and first event argument. `assembleContextFor(agent)` couples prompt facts with scope selection. `SessionStore.flush(session)` owns lookup of the carrier captured when the session entered. +Readonly contracts describe borrowed same-process values. `Scoped` marks event receivers, `agentEvents()` fuses carrier and subject, tool inputs omit registry-owned tokens, and subagent async return types expose readiness directly. -```text -assembleContextFor(agent): - return { agent, scope: agent } +TypeScript cannot govern JavaScript casts, direct Cordis dispatch, process messages, or durable files, so runtime enforcement remains at those escape points. -agentEvents(context, agent): - carrier = scopeTarget(agent, agent) - return dispatcher that always injects agent as the event subject -``` +### Runtime invariants cover cross-service facts -These helpers make a mismatch harder to express than the correct spelling. +The invariants plugin verifies that every declared scoped event uses a marked carrier and that event families exposing a subject use the matching key. Session trace validation stages before append commit and advances after the same event commits. -### Type markers cover every scoped event declaration +The plugin does not police trusted setup by scanning registries or reject prompt assembly objects fabricated through casts. Those checks would turn composition contracts into speculative runtime machinery without protecting a real external boundary. -Scoped agent, approval, tool, prompt, session, and subagent lifecycle events declare a `Scoped` receiver. TypeScript rejects a bare subject at typed dispatch sites, including subagent lifecycle events scoped to the delegating parent. +### Generated artifacts keep public contracts aligned -The marker is compile-time only; JavaScript, casts, and direct Cordis dispatch can bypass it. +The event catalog, service catalog, producer/consumer matrix, configuration catalog, module graph, tool catalog, and type-equivalence blocks are generated or freshness-gated from source. `verify-scoped-dispatch` keeps the declared scoped-event set aligned with runtime invariant coverage. -### Development invariants inspect actual dispatch - -The invariants plugin uses Cordis's internal dispatch as the pre-delivery enforcement point. Every scoped event requires a marked carrier, and events whose arguments expose the subject require the carrier key to be the same object. For `session/event`, callback resolution also precedes the log push: the plugin validates and stages the exact candidate there, then advances its live trace only when the same committed event reaches its contained post-commit listener. A later internal check can therefore veto without advancing either log or trace. Both halves of this oracle are explicitly global, so mounting the plugin under a scoped context cannot stage a foreign event without also applying its committed transition. - -Session and subagent payloads do not expose their owner key directly, so their service centralizes key selection and the invariant proves carrier presence. Additional invariants reject an assembly whose `agent` and `scope` disagree and a turn opened before `agent/session-start`. - -Dedicated `dsh-scope` unit tests cover the carrier's advanced Proxy behavior: private-field method binding, call/construct shape, primordial filter invocation, own-key/descriptor consistency, and explicit configurable definitions. These are implementation tests, not checks performed by the invariants plugin. - -### Repository gates keep declarations and dispatchers aligned - -`verify-scoped-dispatch` compares declared scoped events with the runtime invariant table, and the generated event matrix requires every declaration to name a recognized dispatcher. Source JSDoc generates the [event catalog](../../../cordis-catalog/events.md), which remains the exhaustive signature and mode reference. +Behavioral tests pin scoped routing and disposal, final-entry collision cleanup, publication rollback, ordered quiescence, durable pre/post-commit behavior, live tool filtering across presentation and execution, owner-final Code Mode and structured output, async subagent startup and signal cancellation, worker terminal arbitration, ACP settlement, and process teardown. ## Alternatives considered -The [agent-scope contract](2026-07-08-agent-scope-contexts.md#alternatives-considered) owns the rejected public architectures: explicit agent parameters, event-only filtering, per-agent service graphs, and hierarchical registration inheritance. This RFC records the implementation alternatives rejected after choosing the public contract. +The [July 8 RFC](2026-07-08-agent-scope-contexts.md#alternatives-considered) owns alternatives to the public flat-scope contract. The alternatives here concern implementation shape. -### Publish the agent before running setup +### Use a transparent proxy as the scope carrier -Early publication lets setup find the agent in global registries but lets observers act on a partially configured world. Rollback can remove entries but cannot retract external effects from listeners that already ran. +A proxy that impersonates the subject must preserve property, callable, constructable, private-field, descriptor, and proxy-invariant behavior that listener routing never needs. A small opaque carrier keeps the filter and key while the explicit event argument carries the subject. -The unpublished setup callback already receives `agent.ctx` and `ctx.agent`, so early global lookup is unnecessary. +### Reserve agent and session IDs before setup -### Allow only synchronous setup +Reservations prevent duplicate private setup work but require cross-service capabilities, release ordering, abandoned-reservation cleanup, and prepared-object binding. IDs are caller-supplied and concurrent reuse is caller error; final entry can choose the winner while the losing transaction rolls back cleanly. -Synchronous setup cannot honestly compose child plugins whose activation is asynchronous. TypeScript also permits a promise-returning callback where a void return is expected, so a synchronous-looking type would not reliably contain accidental async work. +### Snapshot every typed same-process argument -Awaited setup makes the transaction explicit and keeps first publication and prompt assembly behind it. +Universal copying defends against stateful getters and callers that violate readonly contracts, but it adds allocation, duplicated validators, and paths that can forget to copy. Materialization belongs at parser, queue, model, durable, worker, process, and wire boundaries where ownership actually changes. -### Validate caller data, then clone it +### Give readiness, cancellation, and disposal separate controllers -Validation followed by a separate clone rereads accessors, so it can approve one value and retain another. A generic JSON clone can also erase or coerce exotic prototypes and unsupported values. The lossless-JSON traversal validates and materializes one captured value in the same operation. +Parallel sentinels can all mirror whether one operation is live. One transaction or start promise owns the operation; separate promises remain only where publication unwind, external work, terminal result, and physical quiescence can settle independently. -### Enforce invariants with prepended waterfall listeners +### Keep synchronous subagent start plus `run.started` -A prepended listener is not permanently outermost: another plugin can prepend later, a short-circuit can skip inner work, and an outer wrapper can replace a downstream result. The same defect appears in prompt assembly, tool decisions, result commit, and turn continuation. +This splits provider acceptance from readiness and forces every consumer to register a partial run, attach result observation, await readiness, and clean up readiness failure. An async start promise makes provider-to-caller ownership transfer the readiness boundary itself. -The four owner-final APIs express the exact one-way power required: restore named canonical data, deny monotonically, observe immutable final outcome, or stop after ordinary continuation folding. +### Keep a separate prompt-protection registry -### Put agent-scope policy inside vendored Cordis +A protection registration mirrors the names and lifetime already owned by prompt sections and tool definitions. `ownerFinal` keeps the exceptional policy on the contribution and lets assembly derive the canonical set directly. -Cordis already supplies derived contexts, effect ownership, and receiver-based filtering. The harness-level primitive composes those domain-neutral mechanisms rather than teaching Cordis about agents, tools, prompts, or global-plus-agent merge rules. +### Remove worker/process lifecycle guards with same-process hardening -The lifecycle hardening remains correctly inside Cordis because effect pre-registration, parent ownership before child publication, and rejection of late effects protect every plugin under reentrant hot reload, not only agent scopes. +Worker messages, process death, and durable input do cross ownership and serialization boundaries. First-outcome arbitration, validation, environment scrubbing, and quiescent process cleanup remain necessary even though hostile same-process callback machinery does not. ## Consequences -The implementation makes the contributor contract locally checkable at each escape boundary. Its cost is explicit runtime machinery for key coherence, continuous ownership, accepted-value stability, and post-middleware finality. +The implementation is smaller and its proof follows the same shape as its ownership graph. One key selects a layer, one entry owns a live registry object, one transaction owns creation, one resolver owns a tool view, and one async promise transfers subagent ownership. -### Correctness properties +### What the design guarantees -The mechanisms compose into five properties: +- A scoped contribution is visible only in its exact agent view and is disposed with that scope. +- Create and resume expose no partially configured handle; final-entry losers and publication failures clean every prepared resource. +- Disposal retains scoped listeners and persistence through driver drain and final session work, then revokes the scope. +- Durable, queued, model, worker, process, and wire values are owned at their real boundary; typed same-process values follow readonly contracts. +- Tool presentation and execution resolve the same live view, and committed results have one immutable observation point. +- Owner-final prompt/tool contributions survive cooperative assembly without freezing unrelated middleware behavior. +- Subagent start returns only a ready run, required signals cancel pending or live work, and disposal reaches the backend's quiescence contract. +- Worker/process result precedence and cleanup remain correct under death, late messages, and bounded teardown. -- Registry layers and event carriers derive from one opaque key, while fused helpers couple subjects that must agree. -- Reservations, sentinels, provider tracking, publication barriers, and reverse teardown cover every asynchronous or reentrant ownership interval. -- At the hardened boundaries listed above, accepted identities and snapshots prevent runtime accessors or later mutation from splitting validation, execution, logging, and observation. -- Prompt protection, guards, final-result observation, and terminal stop each have only the one-way power their invariant requires. -- In-process subagents and workflow runs preserve readiness, terminal precedence, and disposal under provider callbacks, worker death, and racing owners. +### Costs and limits -### Costs and constraints +Scope-aware services still maintain global and identity-keyed maps, and operations must carry their real agent explicitly. Async create/resume and subagent start require callers to await ownership transfer and dispose returned handles. -The proof is not free: +The design trusts typed plugins in the same process. It does not defend against arbitrary casts, stateful getters, mutation that violates readonly contracts, or a plugin deliberately using ambient service access outside the supported composition API. -- Registries keep global and per-scope state, and every scoped dispatcher must preserve the operation subject through a proxy-shaped carrier. -- Programmatic create and resume require reservation capabilities, two-owner tracking, rollback state, ordered publication, and a shared quiescence promise. -- Acceptance boundaries copy, freeze, bind, or retain values according to their contract, increasing allocation and validation work. -- Owner-final behavior uses four explicit APIs instead of relying on ordinary listener ordering. -- Runtime invariants, generated dispatch checks, and focused Proxy/lifecycle/race tests remain necessary because TypeScript cannot enforce direct JavaScript dispatch or runtime reentrancy. - -The direct no-setup `ctx.agentLoop.create()` path remains synchronous for configuration and callers that already have complete options. Programmatic registry create/resume use the full unpublished transaction. - -### Limits of the proof - -The proof covers services and event families that explicitly adopt the agent-scope helpers. It does not make every service call scope-aware, strengthen the ordering contract of a custom agent registered outside AgentLoop, or force an arbitrary external subagent backend to reach quiescence after a workflow grace deadline. - -The [security and authority non-goal](2026-07-08-agent-scope-contexts.md#security-and-authority-are-explicit-non-goals) is part of the public contract. These mechanisms prove composition and ownership behavior inside one trusted process; they do not prove confinement or parent-to-child non-escalation. +The [security and authority non-goal](2026-07-08-agent-scope-contexts.md#security-and-authority-are-non-goals) remains fundamental. These mechanisms prove registration composition, publication, and lifetime ownership; they do not prove confinement or parent-to-child non-escalation. diff --git a/docs/rfc/implemented/feature/2026-06-15-code-mode.md b/docs/rfc/implemented/feature/2026-06-15-code-mode.md index be18e94ae0..16f42104be 100644 --- a/docs/rfc/implemented/feature/2026-06-15-code-mode.md +++ b/docs/rfc/implemented/feature/2026-06-15-code-mode.md @@ -24,11 +24,11 @@ Three decisions, each elaborated in its own section below: `ToolRegistry` gains a schemastery-validated config (`static Config`), its first: `mode: 'native' | 'code' | 'both'`, default `'native'`. A deployment flips it from `cordis.yml` (`tools: { mode: code }`) — no code edit, per the no-hardcoded-tunables convention. -**Wire tool list = the registry's contribution.** The registry feeds assembly through a mode-aware provider: `'native'` contributes every capability visible to that assembly scope, `'code'` contributes only `run_code`, and `'both'` contributes both. Because [`PromptAssembly.tools` is the single source the loop's request header snapshots](../../../../packages/core/system-prompt/src/index.ts), the presentation is logged and reconstructable. The reserved transport is not a capability: it lives outside global/scoped registration and restriction layers, cannot be registered or shadowed, and cannot be named by `ctx.tools.restrict()`. `systemPrompt.protect()` restores its canonical schema after the complete assembly waterfall, so listeners cannot strip, replace, duplicate, or fabricate it. The mode governs only the registry's contribution; a deployment that deliberately installs another direct `systemPrompt.tools()` provider still owns that provider's schemas. +**Wire tool list = the registry's contribution.** The registry feeds assembly through a mode-aware provider: `'native'` contributes every capability visible to that assembly scope, `'code'` contributes only `run_code`, and `'both'` contributes both. Because [`PromptAssembly.tools` is the single source the loop's request header snapshots](../../../../packages/core/system-prompt/src/index.ts), the presentation is logged and reconstructable. The reserved transport is not a capability: it lives outside global/scoped registration and restriction layers, cannot be registered or shadowed, and cannot be named by `ctx.tools.restrict()`. Its `ToolDefinition` declares `ownerFinal: true`, so the provider reports the name as final and assembly restores its canonical schema or canonical absence after the waterfall. The mode governs only the registry's contribution; a deployment that deliberately installs another direct `systemPrompt.tools()` provider still owns that provider's schemas. **Interaction with `toolOrder`, stated up front:** a configured `systemPrompt.toolOrder` naming native capabilities rejects every assembly under `mode: 'code'`, because those names are outside that mode's wire-validation universe. This is correct behavior, not a bug: a deployment using Code Mode updates its order config or drops it. -**The SDK prompt section.** Under `'code'` and `'both'` the registry registers one lazy prompt section (`tools:sdk`, in the 100–199 tool-guidance order band) whose thunk regenerates, for each assembly scope, a TypeScript declaration of every visible end-capability tool plus fixed usage instructions. It uses the same visibility resolver as lookup and execution, so scoped grants and shadows appear while restricted globals disappear; the reserved `run_code` transport itself is excluded. The thunk emits tools in lexicographic name order, so an unchanged visible set produces byte-identical text, and `systemPrompt.protect()` restores the canonical section after every assembly listener. Because that protection is global, it also reserves the `tools:sdk` registry name against scoped section shadows; otherwise scoped-over-global resolution could make a later shadow look canonical before restoration. +**The SDK prompt section.** Under `'code'` and `'both'` the registry registers one lazy prompt section (`tools:sdk`, in the 100–199 tool-guidance order band) whose thunk regenerates, for each assembly scope, a TypeScript declaration of every visible end-capability tool plus fixed usage instructions. It uses the same visibility resolver as lookup and execution, so scoped grants and shadows appear while restricted globals disappear; the reserved `run_code` transport itself is excluded. The thunk emits tools in lexicographic name order, so an unchanged visible set produces byte-identical text. The section declares `ownerFinal: true`, which restores its canonical contribution after every assembly listener and reserves the global name against scoped shadows. **Codegen.** A pure `jsonSchemaToTs(schema)` module inside `dsh-tools` (sibling of `json-schema.ts` — `schemas()` and the SDK are two projections of the same store) maps the JSON-Schema subset the `defineTool` DSL emits (object/string/number/boolean/array, `properties`, `required`, string `enum` → literal union, nested objects, array `items`, `description` → JSDoc) to a TS type literal. It is **total**: any construct outside that subset (`$ref`, `oneOf`/`anyOf`, `integer`, future MCP shapes, …) degrades to `unknown` without throwing. Because `ToolSchema.name` is an arbitrary string, the SDK is declared as one object constant — `declare const tools: { "some-mcp-tool"(args: …): Promise; bash(args: …): Promise; … }` — quoted keys make every name reachable with no sanitization or alias-collision logic. Typing is advisory (the runtime executes type-stripped JS); the instructions say so. diff --git a/docs/rfc/implemented/feature/2026-06-21-subagent-capability-seam.md b/docs/rfc/implemented/feature/2026-06-21-subagent-capability-seam.md index f26aa501ef..c24c6102d9 100644 --- a/docs/rfc/implemented/feature/2026-06-21-subagent-capability-seam.md +++ b/docs/rfc/implemented/feature/2026-06-21-subagent-capability-seam.md @@ -35,9 +35,9 @@ A new package group `packages/subagent/`: | `@deepseek-ai/dsh-subagent-mock` | support: a scripted provider for testing the seam through the real load path | | `@deepseek-ai/dsh-tool-subagent` | consumer: the model-facing `subagent` tool over `ctx.subagents` | -### The primitive: `start → SubagentRun` +### The primitive: async `start → SubagentRun` -A provider exposes `start(request) → SubagentRun`. The run carries `started` (the provider's publication/readiness promise), `result` (the terminal `SubagentResult`), `cancel()`, and `dispose()`. The transport-neutral verb is **`start`**; "spawn" is reserved for the in-process `dsh-subagent-spawn` backend's identity, not the service verb. The service's `start(name, request)` resolves the named provider, validates capabilities, delegates, and waits for `started` before emitting the paired `subagent/start` / `subagent/end`; an attempt that never establishes a child emits neither lifecycle event. For an in-process backend, readiness means the child is published in `ctx.agents`; for ACP it means the remote session exists. +A provider exposes `start(request) → Promise`. Promise fulfillment is the publication/readiness and provider-to-caller ownership boundary: for an in-process backend the child is already published in `ctx.agents`, and for ACP the remote session already exists. `SubagentStartRequest.signal` is the single cancellation channel before and after readiness; `SubagentRun` carries the terminal `result` and a `dispose()` method that cancels remaining work and awaits quiescence. The transport-neutral verb is **`start`**; "spawn" is reserved for the in-process `dsh-subagent-spawn` backend's identity, not the service verb. A rejected start cleans provider-owned partial resources and emits neither subagent lifecycle event. ### Two kinds of optional capability, discovered two ways @@ -54,7 +54,7 @@ Each subagent runs in its **own `Session`** (own id, `parentSession` lineage), p ### Synchronous collect (first cut) -The `dsh-tool-subagent` consumer awaits `run.result` and returns the child's final output as the tool result, blocking the parent's turn until the child finishes. It does so inside a `try/finally` that always `dispose()`s the run (no leaked idle child/session on any path), bridges `exec.signal` to `run.cancel()`, and maps a non-`completed` stop reason to an `isError` result rather than returning partial output as success. Steering (`sendMessage`) is part of the contract but **intentionally unused** this cut. +The `dsh-tool-subagent` consumer passes its execution signal into the start request, awaits the ready run's `result`, and returns the child's final output as the tool result, blocking the parent's turn until the child finishes. A `try/finally` always `dispose()`s the run, so no success, failure, or cancellation path leaks an idle child/session. A non-`completed` stop reason maps to an `isError` result rather than returning partial output as success. Steering (`sendMessage`) is part of the contract but intentionally unused in this consumer. ### Provider selection is config, not model-facing @@ -66,7 +66,7 @@ The seam is tested through the real cordis Loader / export path, not a hand-buil ## Consequences -- **Recursion.** Without a guard, an in-process child inherits the spawn tool and can spawn unboundedly. Depth-limit is an optional capability (the in-process backends enforce it; ACP advertises it off and rejects a `maxDepth` request); tool-filtering is likewise optional. Tool-filtering, when implemented, needs a `tools/pre-execute` deny in the child context — schema filtering alone is insufficient because a model can hallucinate a denied tool name. +- **Recursion.** Without a bound, an in-process child can see the delegation tool and recurse. The in-process backends implement the optional absolute depth limit and scoped live-global `toolFilter`; ACP advertises both capabilities off and rejects such a request. The [subagent composition-controls RFC](2026-07-12-subagent-persona-tool-filter-and-depth.md) owns their exact semantics and security limits. - **Blocking the parent turn.** Synchronous collect holds the parent's `runStep` open for the child's full duration. This is acceptable for the first cut; **background / poll / spill semantics are deferred to a future redesign that unifies long-running-tool handling across subagents AND bash** (a sub-agent and a long `bash` background task pose the same "the model started something slow, how does it collect later" problem, and should share one mechanism rather than each inventing its own). - **Live progress.** This cut surfaces only lifecycle + final result; a per-chunk child→parent update stream is deferred with the background redesign. - **ACP client surface.** Proxying `fs`/`terminal` from the ACP child back to the parent (a shared-workspace mode) is future work; the first cut advertises neither, so the child self-serves in its own process. diff --git a/docs/rfc/implemented/feature/2026-06-22-acp-subagent-backend.md b/docs/rfc/implemented/feature/2026-06-22-acp-subagent-backend.md index ec608177b4..518f9dedef 100644 --- a/docs/rfc/implemented/feature/2026-06-22-acp-subagent-backend.md +++ b/docs/rfc/implemented/feature/2026-06-22-acp-subagent-backend.md @@ -34,7 +34,7 @@ The child is a separate process, so it inherits an environment. Credential-shape Designed at every tier the backend touches, per the root AGENTS.md rule that a new capability shape names its coverage at every tier at plan time: -- **Keyless unit/integration** (`subagent-acp.spec.ts`): spawns a scripted mock ACP server subprocess (`tests/mock-acp-server.ts`) and drives it through the real backend over real ACP stdio. Covers: the prompt round-trip + output accumulation; every StopReason mapping; cancellation via `run.cancel()` and via the request signal; the already-aborted-before-start case; the cancel-races-ahead-of-newSession case; a torn-pipe-after-cancel (child crashes on cancel) settling `aborted`; permission auto-answer under both policies (including the allow-policy-no-allow-option fallback); a non-message update consumed but not accumulated; a nonexistent-command spawn failure settling `error`; HMR provider cleanup; and the namespace export shape. 100% per-file coverage. +- **Keyless unit/integration** (`subagent-acp.spec.ts`): spawns a scripted mock ACP server subprocess (`tests/mock-acp-server.ts`) and drives it through the real backend over real ACP stdio. Coverage includes the prompt round-trip and output accumulation; every StopReason mapping; cancellation through the required request signal and through disposal; already-aborted and cancel-races-ahead-of-newSession starts; a torn pipe after cancellation settling `aborted`; permission auto-answer under both policies; non-message updates; nonexistent-command startup failure with process reaping; provider HMR; and the namespace export shape. - **With-key e2e** (`subagent-acp.e2e.ts`): the harness drives ITSELF — the backend spawns the real `acp-agent` example process and a real model in that child answers a prompt (PONG) and does real file work (writes `proof.txt`, verified on disk). Self-skips without `DEEPSEEK_API_KEY`. This is the "talk to our own process" smoke and the out-of-process analogue of the in-process spawn e2e. - **Snapshot**: deferred as `TODO(acp-subagent-replay)`. An ACP child is a distinct replay shape — each child is its own PROCESS with its own single-agent replay (booted under `DSH_SNAPSHOT=replay` with its own sessions-root + fixture), unlike the in-process per-session keying that [the per-session replay RFC](../testing/2026-06-22-subagent-snapshot-replay.md) added. The keyless mock-server tests give deterministic coverage of the backend in the meantime; the snapshot follow-up would record the parent driving a real-but-replayed ACP child. diff --git a/docs/rfc/implemented/feature/2026-06-30-subagent-observe-enrich.md b/docs/rfc/implemented/feature/2026-06-30-subagent-observe-enrich.md index 17730458de..aa853edd24 100644 --- a/docs/rfc/implemented/feature/2026-06-30-subagent-observe-enrich.md +++ b/docs/rfc/implemented/feature/2026-06-30-subagent-observe-enrich.md @@ -6,13 +6,13 @@ Status: implemented The hooks subsystem ([interception seams RFC](2026-06-30-interception-seams.md)) lets a plugin observe and gate the agent at lifecycle points. Claude Code and Codex both expose **SubagentStart / SubagentStop** hooks, and CC's carry the subagent's final message. The harness already emits `subagent/start` and `subagent/end` lifecycle events ([the subagent capability-seam](2026-06-21-subagent-capability-seam.md)), but their payloads were minimal (`provider`, `id`, and on end `stopReason`) — not enough for a hooks bridge to report WHAT a subagent produced without separately reaching for the live run. -This RFC enriches the end payload. It is deliberately **observe-only**: no control-flow change, no waterfall, no `start()` restructure. A run-affecting subagent-stop decision (continuation, injection that changes the run) is a separate, larger redesign and stays out of scope. +This RFC enriches the end payload. It is deliberately **observe-only**: no control-flow change and no waterfall. A run-affecting subagent-stop decision (continuation, injection that changes the run) is a separate, larger redesign and stays out of scope. ## Decision -**Add `lastAssistantMessage` — the child's final output — to `SubagentRunEndInfo`.** On the settle path it is a DEEP CLONE of `SubagentResult.output` (so an observer sees WHAT the subagent produced without holding the run). On the REJECT path (an infrastructure fault where no `SubagentResult` was produced — the seam only knows `stopReason: 'error'`) it is absent. The clone is load-bearing for observe-only: the `subagent/end` emit fires from a detached `.then` registered *before* `start()` returns, i.e. before the caller's own `await run.result` continuation — handing listeners the same array reference would let a mutating listener corrupt the caller's `SubagentResult.output`. `structuredClone` makes the event a read-only view (a regression test mutates the event's array and asserts the caller's result is untouched); a clone failure is contained (logged, the event still fires without `lastAssistantMessage`) rather than becoming an unhandled rejection on the detached `.then`. +**Add `lastAssistantMessage` — the child's final output — to `SubagentRunEndInfo`.** On the settle path it is the readonly typed `SubagentResult.output`, so an observer sees what the child produced without holding the run. On an infrastructure rejection where no `SubagentResult` exists, it is absent and the event reports `stopReason: 'error'`. Providers and listeners are trusted same-process collaborators and honor the borrowed immutable payload contract. -Both events stay plain **`emit`s**. The service waits for `run.started` before firing `subagent/start`; an in-process listener can therefore reach the published child via `ctx.agents.get(info.id)` and `inject()` into it, while a remote provider need not have a local registry entry. It observes `run.result` immediately, snapshots the end payload before the caller can mutate it, and emits `subagent/end` only after start; readiness rejection emits neither event. The callbacks remain observe-only and per-listener containment keeps one bad subscriber from stranding a live run, surfacing as an unhandled rejection, or starving later listeners. +Both events stay plain **`emit`s**. Async `SubagentService.start()` attaches result observation to the ready provider run, emits `subagent/start`, and then returns the run; an in-process listener can therefore reach the published child via `ctx.agents.get(info.id)`, while a remote provider need not have a local registry entry. A rejected provider start emits neither event. The callbacks remain observe-only and per-listener containment keeps one bad subscriber from stranding a live run or starving later listeners. ## Alternatives considered diff --git a/docs/rfc/implemented/feature/2026-07-05-dynamic-workflows.md b/docs/rfc/implemented/feature/2026-07-05-dynamic-workflows.md index 47d9ec0473..ed718686ef 100644 --- a/docs/rfc/implemented/feature/2026-07-05-dynamic-workflows.md +++ b/docs/rfc/implemented/feature/2026-07-05-dynamic-workflows.md @@ -26,7 +26,7 @@ One deliberate strictness DIVERGENCE from CC: hook misuse — unknown or deferre **Why node:worker_threads**: one run uses one unpooled worker because a workflow run is already heavyweight relative to thread startup. The script runs in a vm context inside the worker, keeping the script-visible surface to the hook contract instead of exposing a bare worker realm, while `agent()` bridges by message-port RPC to I/O-bound child loops on the host. This keeps `start()` from blocking the host on the script's synchronous slice, makes the post-cancel deadline end in a real `worker.terminate()`, and gives cross-thread values a serialization boundary by construction. isolated-vm was rejected for its maintenance state, required `--no-node-snapshot` consumer flag on Node ≥ 20, and node-gyp fallback. -Host-side meta validation and body pre-parsing preserve the seam's synchronous errors, and private enum-keyed payload maps define the wire protocol. Readiness admission, the two child-cancellation channels, worker-death reaping, result precedence, and disposal quiescence preserve the subagent run contract across that wire; the [agent-scope runtime-design RFC](../architecture/2026-07-12-agent-scope-runtime-design.md#workflow-integration-preserves-the-subagent-contract) owns those race algorithms. Coverage uses an in-process `MessageChannel` for worker-side logic that main-process V8 coverage cannot see and separately proves the built `lib/worker.js`—a second tsdown entry sanctioned by the `"./worker"` subpath export—under plain Node in the built-bin smoke gate. +Host-side meta validation and body pre-parsing preserve the seam's synchronous errors, and private enum-keyed payload maps define the wire protocol. Pending async starts, published child records, one host cancellation signal, worker-death reaping, result precedence, and disposal quiescence preserve the subagent run contract across that wire; the [agent-scope runtime-design RFC](../architecture/2026-07-12-agent-scope-runtime-design.md#workflow-children-are-pending-starts-or-published-records) owns those race algorithms. Coverage uses an in-process `MessageChannel` for worker-side logic that main-process V8 coverage cannot see and separately proves the built `lib/worker.js`—a second tsdown entry sanctioned by the `"./worker"` subpath export—under plain Node in the built-bin smoke gate. **Meta as data, never evaluated**: the meta block reaches the seam as a plain JSON request field (the tool's schema-validated `meta` parameter) and the engine only shape-validates it, every violation named. This is a host-isolation invariant, not a convenience: evaluating a meta literal host-side — even one contractually "pure", in an empty timed vm context — hands script-controlled getters a host stack with no timeout the moment the result is READ, defeating the exact spin isolation the worker thread buys. @@ -42,7 +42,7 @@ A `workflow` tool mirroring `dsh-tool-subagent`'s synchronous shape: start, awai An output schema makes a schema-valid committed capture mandatory for successful child completion. The scoped runtime preserves the canonical capture tool and instruction, commits only a successful final outcome—including the enclosing `run_code` outcome for an SDK call—denies later side effects after capture becomes pending, and stops the child without another model step after commit. A validation failure remains a retryable tool error; clean completion without a committed capture settles as an error. -`StructuredOutputSchema` is the raw enforceable JSON-Schema subset in `dsh-tools` (single-string `type`, `properties`/`required`/`additionalProperties`, `items`, scalar `enum`/`const`), and unsupported keywords fail loudly because that wire data becomes the capture tool's parameters verbatim. The [agent-scope runtime-design RFC](../architecture/2026-07-12-agent-scope-runtime-design.md#structured-output-is-a-child-owned-terminal-protocol) owns the assembly, commit, guard, and terminal-stop correctness algorithms. +`StructuredOutputSchema` is the raw enforceable JSON-Schema subset in `dsh-tools` (single-string `type`, `properties`/`required`/`additionalProperties`, `items`, scalar `enum`/`const`), and unsupported keywords fail loudly because that wire data becomes the capture tool's parameters verbatim. The [agent-scope runtime-design RFC](../architecture/2026-07-12-agent-scope-runtime-design.md#structured-output-commits-only-authoritative-outcomes) owns the assembly, commit, guard, and terminal-stop correctness algorithms. ## Deferred (documented non-goals of this cut) diff --git a/docs/rfc/implemented/feature/2026-07-06-approval-seam.md b/docs/rfc/implemented/feature/2026-07-06-approval-seam.md index 0579bf2629..5d82226a96 100644 --- a/docs/rfc/implemented/feature/2026-07-06-approval-seam.md +++ b/docs/rfc/implemented/feature/2026-07-06-approval-seam.md @@ -49,11 +49,11 @@ The `escalation-rejected` twin ends in `{"outcome": "rejected"}` instead: nothin #### The seam: mechanism and policy split -After request validation and a successful `approval/asked` append, the answerer phase always resolves to a closed `ApprovalOutcome` — `allowed-once` / `rejected` / `cancelled` / `unavailable`. The service synchronously snapshots and shallow-freezes the accepted request before its first asynchronous boundary: scalar fields are copied while the agent and `AbortSignal` remain exact identity capabilities, so later caller mutation cannot redirect scope, payload, cancellation, or either audit event. The service dispatches the `approval/request` waterfall, races the captured signal (abort settles `cancelled`; a late answer is discarded, never double-audited), contains a throwing answerer as `unavailable`, normalizes a rogue non-vocabulary return to `unavailable`, and lands the log-only audit pair `approval/asked`/`approval/decided` (paired by the branded `ApprovalRequestId`) on the captured agent's captured session log. Request acceptance and either pre-commit audit append may still reject; returning a decision that could not be logged would violate the pair. Session owns post-commit observer containment, so a callback failure cannot turn an authoritative audit append into a rejected request or suppress the matching event. Grants are one-shot by definition: `allowed-once` authorizes the single asked-about action, never a class of future ones, and the service stores nothing between requests. `request()` also throws before appending anything when the agent's session has no open turn — the audit pair must be turn-enclosed, the turn being the durable log's commit/replay boundary (a bare event between turns is dropped as crash tail on reload); every ask path runs mid-turn already, and idle asks are a deferred design. +After request validation and a successful `approval/asked` append, the answerer phase always resolves to a closed `ApprovalOutcome` — `allowed-once` / `rejected` / `cancelled` / `unavailable`. `ApprovalRequest` is a readonly same-process contract, so the service borrows its routing identity and cancellation signal instead of copying the record or capturing a parallel callback bundle. It dispatches the `approval/request` waterfall, races the request signal (abort settles `cancelled`; a late answer is discarded, never double-audited), contains a throwing answerer as `unavailable`, normalizes a rogue non-vocabulary return to `unavailable`, and lands the log-only audit pair `approval/asked`/`approval/decided` (paired by the branded `ApprovalRequestId`) on the request agent's session log. Request acceptance and either pre-commit audit append may still reject; returning a decision that could not be logged would violate the pair. Session owns post-commit observer containment, so a callback failure cannot turn an authoritative audit append into a rejected request or suppress the matching event. Grants are one-shot by definition: `allowed-once` authorizes the single asked-about action, never a class of future ones, and the service stores nothing between requests. `request()` also throws before appending anything when the agent's session has no open turn — the audit pair must be turn-enclosed, the turn being the durable log's commit/replay boundary (a bare event between turns is dropped as crash tail on reload); every ask path runs mid-turn already, and idle asks are a deferred design. Answerers are the policy, and they are `approval/request` waterfall listeners. The waterfall buys exactly what the seam needs: with zero listeners the dispatch falls through to the caller-supplied default — `unavailable`, so fail-closed needs no configuration and no code in any deployment; a listener that recognizes the request's agent answers by returning an outcome without calling `next()` (the decision slot is single-occupancy, first answer wins — the same documented semantics as the `fs/write-intent` gate); a listener that does not recognize the agent MUST delegate via `next()` so another answerer or the default gets the question; and listeners dispose with their owning fiber, so an unloaded UI plugin degrades the next ask to `unavailable` instead of leaving a dangling channel. Registration order across sibling plugins is not load-order deterministic (the loader starts siblings concurrently), so a deployment composes ONE terminal answerer and reserves `prepend` listeners for decide-or-delegate gates. -`ApprovalRequest` carries the asking `agent` (routes the question; receives the audit events), the `toolName`, the optional exact `callId`, the asker's human-readable `reason`, and the optional `signal`. The caller owns this input record; `request()` owns its frozen acceptance snapshot. The vocabulary is deliberately self-contained — it names the tool-call by the `CallId` brand from `dsh-llm` and never imports `dsh-tools` — because `dsh-tools` depends on `dsh-user-approval` (the ask routing) and a `ToolCallView` import would close a package cycle. It deliberately does NOT carry tool arguments: a UI answerer attaches the prompt to the already-streamed tool call via `callId` instead of re-rendering the call. +`ApprovalRequest` carries the asking `agent` (routes the question; receives the audit events), the `toolName`, the optional exact `callId`, the asker's human-readable `reason`, and the optional `signal`. The caller retains ownership and honors the readonly contract for the duration of `request()`. The vocabulary is deliberately self-contained — it names the tool-call by the `CallId` brand from `dsh-llm` and never imports `dsh-tools` — because `dsh-tools` depends on `dsh-user-approval` (the ask routing) and a `ToolCallView` import would close a package cycle. It deliberately does NOT carry tool arguments: a UI answerer attaches the prompt to the already-streamed tool call via `callId` instead of re-rendering the call. #### Ask routing in dsh-tools @@ -79,7 +79,7 @@ One package, no cycles: `dsh-user-approval` peers on `cordis`, `dsh-session` (ev ### Testing -Unit tier: the service's outcome branches (fail-closed default, first-wins slot, delegation, containment, rogue-value normalization, abort-before and abort-during with late-answer discard, fresh ids, fiber-disposal degradation), accepted-request mutation across agent scopes, post-append observer throws on both audit events, and the policy tier (both values × dispatch/decide, a `'never'` decision unbypassable even by an answerer prepended AFTER the service, audit pair intact) in `dsh-user-approval`; the ask routing matrix (grant dispatches; three non-grant reasons pinned verbatim; unmounted and agent-less degrades; the registry's own exhaustiveness backstop against a non-conforming stand-in) in `dsh-tools`; the answerer (wire shape of the prompt, outcome mapping, unknown-option conservatism, foreign-agent and call-less delegation) driven through a real bridge + scripted client in `dsh-acp`. +Unit tier: the service's outcome branches (fail-closed default, first-wins slot, delegation, containment, rogue-value normalization, abort-before and abort-during with late-answer discard, fresh ids, fiber-disposal degradation), scoped routing, post-append observer throws on both audit events, and the policy tier (both values × dispatch/decide, a `'never'` decision unbypassable even by an answerer prepended AFTER the service, audit pair intact) in `dsh-user-approval`; the ask routing matrix (grant dispatches; three non-grant reasons pinned verbatim; unmounted and agent-less degrades; the registry's own exhaustiveness backstop against a non-conforming stand-in) in `dsh-tools`; the answerer (wire shape of the prompt, outcome mapping, unknown-option conservatism, foreign-agent and call-less delegation) driven through a real bridge + scripted client in `dsh-acp`. Snapshot tier: the harness accepts scripted permission answers (`permissionAnswers` in a scenario's `input.json`, consumed FIFO; an unscripted prompt answers `cancelled`, fail closed). The seam's wire is recorded end to end in the sandbox example's suite: both escalation branches drive `session/request_permission` through this seam over scripted answers (grant and rejection), and the recorded `mode-switching` scenario pins the `'never'` prompt sentence and the policy-switch notice ([the sandbox RFC](2026-07-06-sandbox.md) § Testing). @@ -105,7 +105,7 @@ The implemented contract is pinned by the suites in Testing: - With an ApprovalService and an answerer composed, a hook's `ask` reaches a human and `allowed-once` dispatches the tool; every other outcome denies with its distinct reason. - A `'never'` session auto-rejects every ask without prompting anyone, states the policy in its prompt, and narrates switches (the shared switching mechanics are pinned in [the sandbox RFC](2026-07-06-sandbox.md)). - Every unanswerable path fails closed to `unavailable`: no service, no listener, a foreign or agent-less request, a throwing answerer, a rogue return value, or a dead client connection. -- Every `request()` snapshots its routing identity and lands exactly one `approval/asked`/`approval/decided` pair on that agent's captured log, replayable and invisible to the model transcript; post-append observer failures cannot split the pair. +- Every `request()` routes through its readonly agent identity and lands exactly one `approval/asked`/`approval/decided` pair on that agent's log, replayable and invisible to the model transcript; post-append observer failures cannot split the pair. - Prompts route per-session through the bridge's ownership map; one session's prompt can never reach another session's editor. - A deployment with no ApprovalService emits no approval prompt or approval audit events and denies every `ask` request. diff --git a/docs/rfc/implemented/feature/2026-07-12-subagent-persona-tool-filter-and-depth.md b/docs/rfc/implemented/feature/2026-07-12-subagent-persona-tool-filter-and-depth.md new file mode 100644 index 0000000000..5b6eeb4ccb --- /dev/null +++ b/docs/rfc/implemented/feature/2026-07-12-subagent-persona-tool-filter-and-depth.md @@ -0,0 +1,92 @@ +# RFC: Configure subagent persona, tool visibility, and depth + +Status: implemented + +## Problem + +A reusable subagent provider answers how to run a child, but different delegation tools need different child behavior. One deployment may want a reviewer persona, a research-only tool set, or a hard recursion bound without creating a new provider for every combination. + +These controls affect the child's first model request and therefore cannot be installed after the child is visible. They also need honest provider support: an ACP backend cannot silently accept an in-process-only tool filter, and a filter must not be described as a security boundary when every plugin runs in the same trusted process. + +## Decision + +Subagent starts have three independent composition controls: `persona`, `toolFilter`, and `maxDepth`. A provider advertises support for each control, the service rejects unsupported requests before starting a run, and an in-process provider installs the requested composition while the child is still unpublished. + +The controls answer different questions: + +| Control | Question | Result | +|---|---|---| +| `persona` | What role instructions replace the deployment persona for this child? | A child-local prompt section shadows `deployment:persona` | +| `toolFilter` | Which deployment-global tools enter this child's visible tool view? | A scoped restriction filters globals before child-local tools are added | +| `maxDepth` | How deep may this delegation tree grow? | A start whose child depth exceeds the absolute cap is rejected | + +`dsh-tool-subagent` exposes the controls as plugin configuration and copies them into each request it creates. Direct `SubagentService` callers may choose them per request. The provider capability descriptor remains the source of truth for whether a backend can honor each field. + +### Persona is a scoped shadow + +The persona control changes one child without changing deployment-wide prompt assembly. During unpublished setup, an in-process provider registers a child-scoped section named `deployment:persona`; ordinary most-specific-wins resolution replaces the global section only in that child's assemblies. + +The value has the same strict template semantics as the deployment persona. Omitting it inherits the deployment section through the global layer; an explicit empty string shadows the global persona with an empty section. Parent and sibling personas never enter the child's flat scope. + +This uses the normal system-prompt registration mechanism rather than a second persona channel. The first prompt therefore sees the same named contribution that later prompts and prompt-inspection tools see. + +### Tool filtering is one live global-view rule + +The tool filter controls visibility and executable lookup together. An in-process provider installs `ToolRegistry.restrict()` in the child's scope before publication, and the registry's single resolver applies the same result to prompt schemas, lookup, execution, and Code Mode SDK generation. + +Resolution follows these rules: + +1. Each restriction applies `allow` before `deny` to the live deployment-global tool registry. +2. Multiple restrictions intersect, so every installed restriction must admit a global tool. +3. Child-scoped tools are added after global filtering and may shadow an admitted global tool. +4. Reserved `run_code` presentation and other scope-local protocol contributions are outside the global filter. + +Configuration fails loudly when a filter supplies neither `allow` nor `deny`, or names something outside the current global restrictable set, including a scope-local-only or reserved name. `allow: []` is valid and deliberately hides every global tool. These checks catch misspellings and prevent configuration from appearing effective when it cannot affect the named entry. + +The global registry remains live. A deny-only filter admits a later global name unless it explicitly denies that name; an allow-list excludes a later global name unless it explicitly allows that name. Removing a global tool removes it from every resolved view. These semantics preserve hot registration while making the difference between allow and deny explicit. + +### Depth is an absolute tree cap + +The depth limit bounds recursive delegation independently of tool visibility. A top-level agent has depth zero; an in-process child has its parent's validated depth plus one. `maxDepth` is an absolute non-negative safe integer, and a start rejects before child ownership begins when the derived child depth is greater than the cap. + +Every public entry validates the domain rather than relying on one model-facing configuration path. Negative values, fractions, negative zero, non-finite values, unsafe integers, malformed stored parent depth, and derived overflow all reject. Omitting the cap leaves depth unbounded by this mechanism. + +A deployment can combine depth and filtering. For example, it may keep the delegation tool visible at depth one but set `maxDepth: 1`, or deny the delegation tool entirely in children. Neither choice changes the provider's conversation-history behavior. + +### Capability gating keeps providers honest + +Capabilities separate a requested feature from a provider implementation. `SubagentCapabilities` advertises `persona`, `toolFilter`, and `depthLimit`; `SubagentService.start()` checks every present request field against those flags before calling the provider. + +This lets spawn and fork providers share the in-process implementation while external providers advertise only what they can enforce. A request never degrades silently: selecting an unsupported control produces `UNSUPPORTED_CAPABILITY`, and no run or lifecycle event exists. + +### Unpublished setup makes the first request correct + +All child-local composition is complete before the child becomes observable. The in-process provider supplies one setup callback to agent creation; that callback installs persona, tool restriction, and structured-output contributions in the child's scope. Only after setup succeeds does creation publish the session and agent and allow the driver to start. + +A setup failure rolls back the private child. No observer can acquire a child whose first prompt used the deployment persona or unfiltered tool set and whose later prompts use the requested configuration. + +## Visibility is not authority + +These controls compose trusted same-process behavior; they do not authorize it. `toolFilter` changes the child view resolved by the tool registry, but it does not create a parent-to-child grant lattice, require a child to be a subset of its parent, sandbox plugins, or prevent code with another Cordis context from calling services directly. + +In particular, a child-local tool is added after the global filter and may be absent from the parent's view. A deny-only child also sees later global tools not named by the deny-list. Those are deliberate live-composition semantics, not non-escalation guarantees. + +A security design would need a separate authority representation, propagation rule, and execution-time enforcement point. Creation-time grant snapshots, parent-subset grants, explicit future-grant APIs, and generic capability/output/termination tags are outside this feature. + +## Alternatives considered + +**Create one provider per persona or tool set.** This multiplies providers that share the same transport and lifecycle implementation, makes dynamic deployment configuration awkward, and still needs a recursion mechanism. Providers remain about execution transport; requests carry per-child composition. + +**Copy the parent's complete tool view.** Registration scope is flat by design, and lifetime ownership does not imply visibility inheritance. Copying a resolved view would also freeze dynamic global registrations and conflate composition with authority without defining either contract fully. + +**Snapshot allowed global tools at child creation.** A frozen allow-set makes future registration uniformly unavailable, but it changes hot-registration semantics and starts an authorization design. The implemented filter stays a live registry predicate and documents allow-versus-deny behavior directly. + +**Hide only tool schemas.** Presentation-only filtering lets the model execute a tool that the prompt says does not exist through Code Mode or a forged call. One resolver governs both presentation and execution instead. + +**Use only tool filtering to stop recursion.** Removing the delegation tool is useful but provider-specific and does not protect direct service callers or alternate delegation tools. Absolute depth is an independent structural bound. + +## Consequences + +Contributors can configure child role, visible global tools, and recursion without defining new providers. Capability checks fail before ownership starts, unpublished setup makes the first request consistent, and one tool resolver prevents presentation/execution drift. + +The cost is that deployments must understand live allow/deny behavior and the distinction between visibility and authority. Provider authors must advertise each supported control accurately, and in-process providers must install every requested contribution before publication. The controls deliberately do not solve security confinement or parent-to-child non-escalation. diff --git a/packages/AGENTS.md b/packages/AGENTS.md index ac26b926f7..4b7bed4d1c 100644 --- a/packages/AGENTS.md +++ b/packages/AGENTS.md @@ -5,6 +5,8 @@ This directory contains all `@deepseek-ai/dsh-*` harness packages. Repo-wide con - **Plugin export shape — namespace OR default, never both.** A *service* package exports the service class as `export default` (the Loader instantiates it). A *function/namespace* plugin exports `name` / `inject` / `Config` / `apply` as separate named exports and **must NOT add `export default`** — the cordis Loader's `unwrapExports` does `exports.default ?? exports`, so a stray default export collapses the module to the bare `apply` function and silently discards the `inject`/`name`/`Config` namespace, leaving the plugin with no injected services (it then throws `cannot get property … without inject` at load). See [docs/postmortem/0001](../docs/postmortem/0001-acp-default-export-drops-inject.md). - **Read an optional (non-injected) service via `ctx.get(name)`, not `ctx.`.** For a service a plugin reads opportunistically but deliberately leaves out of `static inject` (e.g. `AgentLoop` reading `sessionPersistence`), the `ctx.` property proxy resolves by an ancestor-only fiber walk that throws when the call arrives through a foreign traceable shadow (the service lives on a sibling fiber). `ctx.get(name)` is the topology-independent global-store lookup, strict by default (an inactive/absent backend reads as `undefined` — prefer it over the `ctx.get(name, false)` overload, which also skips the active-state check). Services that ARE in `static inject` resolve fine via `ctx.`. See [docs/postmortem/0001](../docs/postmortem/0001-acp-default-export-drops-inject.md). - **A plugin shipped via `cordis.yml` needs at least one test through the REAL Loader/export path** — hand-built `ctx.plugin({...})` mounts bypass `unwrapExports` and cannot catch a broken export shape. Full testing policy (tiers, with-key generosity, real-entry-path guards): [docs/testing.md](../docs/testing.md). +- **Typed same-process service and plugin calls are contracts, not serialization boundaries.** Prefer readonly borrowed values; materialize or defensively validate only at parser/config, queued, model/tool JSON, durable/file, worker, process, or wire boundaries. +- **Represent one asynchronous operation with one lifecycle controller or transaction.** Separate readiness, cancellation, disposal, reservation, or sentinel state requires an independent owner or settlement boundary; otherwise fold it while preserving rollback, callback containment, and quiescence. Naming notes: diff --git a/packages/cordis/tool-cordis/src/api-catalog.ts b/packages/cordis/tool-cordis/src/api-catalog.ts index 88b89f5d3b..3ee6d1081f 100644 --- a/packages/cordis/tool-cordis/src/api-catalog.ts +++ b/packages/cordis/tool-cordis/src/api-catalog.ts @@ -54,7 +54,7 @@ export interface TypeApiEntry { export const SERVICE_API: readonly ServiceApiEntry[] = [ { key: 'agentLoop', - summary: 'The agent-loop plugin (`ctx.agentLoop`): creates ReactLoopAgents, runs their loops, and registers them in `ctx.agents`.', + summary: 'Concrete ReactLoopAgent factory and driver service.', methods: [ 'create(id: AgentId, options: AgentOptions = {}, meta: Pick = {}): ReactLoopAgent', 'async createAgent(ownerCtx: Context, options: CreateAgentOptions): Promise', @@ -65,12 +65,11 @@ export const SERVICE_API: readonly ServiceApiEntry[] = [ key: 'agents', summary: 'Agent registry (`ctx.agents`): tracks live agents so UI, hook, and orchestrator plugins can find them without depending on the concrete loop package.', methods: [ - 'reserve(id: AgentId): AgentRegistrationReservation', 'setFactory(factory: AgentFactory): () => Promise | void', 'async create(options: CreateAgentOptions): Promise', 'async resume(options: ResumeAgentOptions): Promise', 'register(agent: Agent): () => Promise | void', - 'enter(agent: Agent, reservation?: AgentRegistrationReservation): () => void', + 'enter(agent: Agent): () => void', 'announce(agent: Agent): void', 'get(id: AgentId): Agent | undefined', 'list(): Agent[]', @@ -156,10 +155,9 @@ export const SERVICE_API: readonly ServiceApiEntry[] = [ key: 'sessions', summary: 'In-memory session store (`ctx.sessions`).', methods: [ - 'reserve(id: SessionId): SessionRegistrationReservation', 'create(id?: SessionId, options?: CreateSessionOptions): Session', 'prepare(id?: SessionId, options?: CreateSessionOptions): Session', - 'enter(session: Session, reservation?: SessionRegistrationReservation): () => void', + 'enter(session: Session): () => void', 'announce(session: Session): void', 'async flush(session: Session): Promise', 'get(id: SessionId): Session | undefined', @@ -179,22 +177,21 @@ export const SERVICE_API: readonly ServiceApiEntry[] = [ }, { key: 'subagents', - summary: 'The `subagents` service: a registry of named SubagentProviders and a capability-checked start surface.', + summary: 'Named provider registry and capability-checked start surface.', methods: [ 'registerProvider(provider: SubagentProvider): () => Promise | void', 'getProvider(name: string): SubagentProvider | undefined', 'list(): string[]', - 'start(name: string, request: SubagentStartRequest): SubagentRun', + 'async start(name: string, request: SubagentStartRequest): Promise', ], }, { key: 'systemPrompt', - summary: 'Registry service (`ctx.systemPrompt`): plugins contribute ordered text sections, tool-schema providers, named prompt variables, and owner-final contribution protections; the agent loop calls `assemble(context)` once per step.', + summary: 'Registry service (`ctx.systemPrompt`): plugins contribute ordered text sections, tool-schema providers, named prompt variables, and owner-final contributions; the agent loop calls `assemble(context)` once per step.', methods: [ 'section(section: PromptSection): () => Promise | void', 'tools(provider: (context: AssembleContext) => ToolProviderResult): () => Promise | void', 'variable(name: string, provider: (context: AssembleContext) => string | undefined): () => Promise | void', - 'protect(protection: PromptProtection): () => Promise | void', 'async assemble(context: AssembleContext = {}): Promise', ], }, @@ -205,10 +202,8 @@ export const SERVICE_API: readonly ServiceApiEntry[] = [ 'register(definition: ToolDefinition): () => Promise | void', 'restrict(filter: ToolRestriction): () => Promise | void', 'guard(guard: ToolGuard): () => Promise | void', - 'visible(scope?: ScopeKey): ToolDefinition[]', 'get(name: string, scope?: ScopeKey): ToolDefinition | undefined', 'schemas(scope?: ScopeKey): ToolSchema[]', - 'knownNames(scope?: ScopeKey): string[]', 'async execute(exec: ToolExecutionInput): Promise', ], }, @@ -389,25 +384,25 @@ export const EVENT_API: readonly EventApiEntry[] = [ name: 'subagent/end', mode: 'emit', signature: '\'subagent/end\'(this: Scoped, info: SubagentRunEndInfo): void', - summary: 'A started subagent run settled — emitted when SubagentRun.result resolves (any stop reason) or rejects (reported as `error`).', + summary: 'A ready child settled.', }, { name: 'subagent/provider-added', mode: 'emit', signature: '\'subagent/provider-added\'(provider: SubagentProvider): void', - summary: 'A provider became resolvable in the SubagentService registry.', + summary: 'A provider became resolvable in the registry.', }, { name: 'subagent/provider-removed', mode: 'emit', signature: '\'subagent/provider-removed\'(name: string): void', - summary: 'A provider left the registry (its plugin\'s fiber was disposed — an unload or an HMR reload).', + summary: 'A provider left the registry.', }, { name: 'subagent/start', mode: 'emit', signature: '\'subagent/start\'(this: Scoped, info: SubagentRunInfo): void', - summary: 'A subagent run started — emitted only after SubagentRun.started fulfills, when the provider has established a live child.', + summary: 'A provider established a ready child.', }, { name: 'system-prompt/assemble', @@ -419,7 +414,7 @@ export const EVENT_API: readonly EventApiEntry[] = [ name: 'system-prompt/change', mode: 'emit', signature: '\'system-prompt/change\'(): void', - summary: 'A section, tool provider, variable provider, or protection was registered or unregistered (the assembly inputs changed — possibly for one scope only).', + summary: 'A section, tool provider, or variable provider was registered or unregistered (the assembly inputs changed — possibly for one scope only).', }, { name: 'tools/change', @@ -436,7 +431,7 @@ export const EVENT_API: readonly EventApiEntry[] = [ { name: 'tools/post-execute', mode: 'waterfall', - signature: '\'tools/post-execute\'(this: Scoped, exec: ToolExecution, result: ToolExecutionResult, next: () => Promise): Promise', + signature: '\'tools/post-execute\'(this: Scoped, exec: ToolExecution, result: Readonly, next: () => Promise): Promise', summary: 'Waterfall AFTER a tool runs — where hook plugins inspect the result and accept it (optionally REPLACING the model-facing content, and/or attaching `additionalContext` for the next request) or block it with corrective `feedback` (Claude Code\'s `PostToolUse`).', }, { @@ -511,10 +506,6 @@ export const TYPE_API: readonly TypeApiEntry[] = [ name: 'AgentOptions', declaration: 'export interface AgentOptions {\n model?: string;\n}', }, - { - name: 'AgentRegistrationReservation', - declaration: 'export interface AgentRegistrationReservation {\n readonly id: AgentId;\n release(): void;\n}', - }, { name: 'AgentStatus', declaration: 'export type AgentStatus = \'idle\' | \'running\' | \'disposed\';', @@ -525,7 +516,7 @@ export const TYPE_API: readonly TypeApiEntry[] = [ }, { name: 'ApprovalRequest', - declaration: 'export interface ApprovalRequest {\n agent: Agent;\n toolName: string;\n callId?: CallId;\n reason?: string;\n signal?: AbortSignal;\n}', + declaration: 'export interface ApprovalRequest {\n readonly agent: Agent;\n readonly toolName: string;\n readonly callId?: CallId;\n readonly reason?: string;\n readonly signal?: AbortSignal;\n}', }, { name: 'AskUserQuestionAnswer', @@ -653,11 +644,11 @@ export const TYPE_API: readonly TypeApiEntry[] = [ }, { name: 'CreateAgentOptions', - declaration: 'export interface CreateAgentOptions {\n agentId: AgentId;\n sessionId: SessionId;\n meta?: {\n cwd?: string;\n parentSession?: SessionId;\n seedLength?: number;\n };\n seed?: SessionEvent[];\n agentOptions?: AgentOptions;\n setup?: (agentCtx: Context) => Promise | void;\n}', + declaration: 'export interface CreateAgentOptions {\n readonly agentId: AgentId;\n readonly sessionId: SessionId;\n readonly meta?: {\n readonly cwd?: string;\n readonly parentSession?: SessionId;\n readonly seedLength?: number;\n };\n readonly seed?: readonly SessionEvent[];\n readonly agentOptions?: AgentOptions;\n readonly signal?: AbortSignal;\n readonly setup?: (agentCtx: Context) => Promise | void;\n}', }, { name: 'CreateSessionOptions', - declaration: 'export interface CreateSessionOptions {\n seed?: SessionEvent[];\n meta?: {\n cwd?: string;\n parentSession?: SessionId;\n createdAt?: number;\n seedLength?: number;\n };\n}', + declaration: 'export interface CreateSessionOptions {\n readonly seed?: readonly SessionEvent[];\n readonly meta?: {\n readonly cwd?: string;\n readonly parentSession?: SessionId;\n readonly createdAt?: number;\n readonly seedLength?: number;\n };\n}', }, { name: 'DiffCallView', @@ -755,13 +746,9 @@ export const TYPE_API: readonly TypeApiEntry[] = [ name: 'PromptAssembly', declaration: 'export interface PromptAssembly {\n sections: AssembledSection[];\n tools: ToolSchema[];\n variables: Record;\n}', }, - { - name: 'PromptProtection', - declaration: 'export interface PromptProtection {\n sections?: readonly string[];\n tools?: readonly string[];\n}', - }, { name: 'PromptSection', - declaration: 'export interface PromptSection {\n name: string;\n order: number;\n text: string | ((context: AssembleContext) => string);\n}', + declaration: 'export interface PromptSection {\n readonly name: string;\n readonly order: number;\n readonly text: string | ((context: AssembleContext) => string);\n readonly ownerFinal?: boolean;\n}', }, { name: 'ReasoningBlock', @@ -769,7 +756,7 @@ export const TYPE_API: readonly TypeApiEntry[] = [ }, { name: 'ResumeAgentOptions', - declaration: 'export interface ResumeAgentOptions {\n agentId: AgentId;\n resumeSessionId: SessionId;\n agentOptions?: AgentOptions;\n setup?: (agentCtx: Context) => Promise | void;\n}', + declaration: 'export interface ResumeAgentOptions {\n readonly agentId: AgentId;\n readonly resumeSessionId: SessionId;\n readonly agentOptions?: AgentOptions;\n readonly signal?: AbortSignal;\n readonly setup?: (agentCtx: Context) => Promise | void;\n}', }, { name: 'SandboxEnforcement', @@ -809,23 +796,19 @@ export const TYPE_API: readonly TypeApiEntry[] = [ }, { name: 'SessionHeader', - declaration: 'export interface SessionHeader {\n version: number;\n id: SessionId;\n createdAt: number;\n cwd?: string;\n parentSession?: SessionId;\n seedLength?: number;\n}', + declaration: 'export interface SessionHeader {\n readonly version: number;\n readonly id: SessionId;\n readonly createdAt: number;\n readonly cwd?: string;\n readonly parentSession?: SessionId;\n readonly seedLength?: number;\n}', }, { name: 'SessionId', declaration: 'export type SessionId = Branded<\'SessionId\'>;', }, - { - name: 'SessionRegistrationReservation', - declaration: 'export interface SessionRegistrationReservation {\n readonly id: SessionId;\n prepare(options?: CreateSessionOptions): Session;\n release(): void;\n}', - }, { name: 'SkillCandidate', - declaration: 'export interface SkillCandidate extends SkillSummary {\n rank: number;\n locator: unknown;\n path?: string;\n metadata?: Record;\n}', + declaration: 'export interface SkillCandidate extends SkillSummary {\n readonly rank: number;\n readonly locator: unknown;\n readonly path?: string;\n readonly metadata?: Readonly>;\n}', }, { name: 'SkillDefinition', - declaration: 'export interface SkillDefinition extends SkillSummary {\n content: string;\n path?: string;\n metadata?: Record;\n}', + declaration: 'export interface SkillDefinition extends SkillSummary {\n readonly content: string;\n readonly path?: string;\n readonly metadata?: Readonly>;\n}', }, { name: 'SkillLookupOptions', @@ -833,15 +816,15 @@ export const TYPE_API: readonly TypeApiEntry[] = [ }, { name: 'SkillProvider', - declaration: 'export interface SkillProvider {\n name: string;\n list(options: SkillLookupOptions): Promise;\n get(candidate: SkillCandidate, options: SkillLookupOptions): Promise;\n}', + declaration: 'export interface SkillProvider {\n readonly name: string;\n readonly list: (options: SkillLookupOptions) => Promise;\n readonly get: (candidate: SkillCandidate, options: SkillLookupOptions) => Promise;\n}', }, { name: 'SkillRegistration', - declaration: 'export type SkillRegistration = Omit & {\n provider?: string;\n};', + declaration: 'export type SkillRegistration = Omit & {\n readonly provider?: string;\n};', }, { name: 'SkillResourceBase', - declaration: 'export type SkillResourceBase = {\n kind: \'directory\';\n path: string;\n} | {\n kind: \'url\';\n url: string;\n} | {\n kind: \'opaque\';\n description: string;\n};', + declaration: 'export type SkillResourceBase = {\n readonly kind: \'directory\';\n readonly path: string;\n} | {\n readonly kind: \'url\';\n readonly url: string;\n} | {\n readonly kind: \'opaque\';\n readonly description: string;\n};', }, { name: 'SkillSource', @@ -849,7 +832,7 @@ export const TYPE_API: readonly TypeApiEntry[] = [ }, { name: 'SkillSummary', - declaration: 'export interface SkillSummary {\n name: string;\n description: string;\n whenToUse?: string;\n disableModelInvocation?: boolean;\n source: SkillSource;\n provider: string;\n resourceBase?: SkillResourceBase;\n}', + declaration: 'export interface SkillSummary {\n readonly name: string;\n readonly description: string;\n readonly whenToUse?: string;\n readonly disableModelInvocation?: boolean;\n readonly source: SkillSource;\n readonly provider: string;\n readonly resourceBase?: SkillResourceBase;\n}', }, { name: 'StreamChunk', @@ -873,23 +856,23 @@ export const TYPE_API: readonly TypeApiEntry[] = [ }, { name: 'SubagentCapabilities', - declaration: 'export interface SubagentCapabilities {\n outputSchema: boolean;\n depthLimit: boolean;\n toolFilter: boolean;\n persona: boolean;\n}', + declaration: 'export interface SubagentCapabilities {\n readonly outputSchema: boolean;\n readonly depthLimit: boolean;\n readonly toolFilter: boolean;\n readonly persona: boolean;\n}', }, { name: 'SubagentProvider', - declaration: 'export interface SubagentProvider {\n readonly name: string;\n readonly capabilities: SubagentCapabilities;\n readonly inheritsParentContext: boolean;\n start(request: SubagentStartRequest): SubagentRun;\n}', + declaration: 'export interface SubagentProvider {\n readonly name: string;\n readonly capabilities: SubagentCapabilities;\n readonly inheritsParentContext: boolean;\n start(request: SubagentStartRequest): Promise;\n}', }, { name: 'SubagentResult', - declaration: 'export interface SubagentResult {\n output: ContentBlock[];\n structured?: unknown;\n stopReason: SubagentStopReason;\n}', + declaration: 'export interface SubagentResult {\n readonly output: ContentBlock[];\n readonly structured?: unknown;\n readonly stopReason: SubagentStopReason;\n}', }, { name: 'SubagentRun', - declaration: 'export interface SubagentRun {\n readonly id: AgentId;\n readonly started: Promise;\n readonly result: Promise;\n cancel(reason?: string): void;\n dispose(): Promise;\n sendMessage?(content: ContentBlock[]): void;\n resume?(content: ContentBlock[]): SubagentRun;\n}', + declaration: 'export interface SubagentRun {\n readonly id: AgentId;\n readonly result: Promise;\n dispose(): Promise;\n sendMessage?(content: ContentBlock[]): void;\n resume?(content: ContentBlock[]): Promise;\n}', }, { name: 'SubagentStartRequest', - declaration: 'export interface SubagentStartRequest {\n prompt: ContentBlock[];\n parent: Agent;\n signal?: AbortSignal;\n agentOptions?: AgentOptions;\n outputSchema?: StructuredOutputSchema;\n maxDepth?: number;\n toolFilter?: {\n allow?: string[];\n deny?: string[];\n };\n persona?: string;\n}', + declaration: 'export interface SubagentStartRequest {\n readonly prompt: ContentBlock[];\n readonly parent: Agent;\n readonly signal: AbortSignal;\n readonly agentOptions?: AgentOptions;\n readonly outputSchema?: StructuredOutputSchema;\n readonly maxDepth?: number;\n readonly toolFilter?: ToolRestriction;\n readonly persona?: string;\n}', }, { name: 'SubagentStopReason', @@ -937,7 +920,7 @@ export const TYPE_API: readonly TypeApiEntry[] = [ }, { name: 'ToolDefinition', - declaration: 'export interface ToolDefinition extends ToolSchema {\n execute(args: unknown, exec: ToolExecution): Promise;\n timeoutMs?: number;\n presentCall?(args: unknown): ToolCallView | undefined;\n presentResult?(args: unknown, result: ToolResult): ToolResultView | undefined;\n}', + declaration: 'export interface ToolDefinition extends ToolSchema {\n execute(args: unknown, exec: ToolExecution): Promise;\n timeoutMs?: number;\n readonly ownerFinal?: boolean;\n presentCall?(args: unknown): ToolCallView | undefined;\n presentResult?(args: unknown, result: ToolResult): ToolResultView | undefined;\n}', }, { name: 'ToolErrorInfo', @@ -961,7 +944,7 @@ export const TYPE_API: readonly TypeApiEntry[] = [ }, { name: 'ToolExecutionToken', - declaration: 'export interface ToolExecutionToken {\n readonly [toolExecutionTokenBrand]: true;\n}', + declaration: 'export type ToolExecutionToken = symbol & {\n readonly [toolExecutionTokenBrand]: true;\n};', }, { name: 'ToolGuard', @@ -969,11 +952,11 @@ export const TYPE_API: readonly TypeApiEntry[] = [ }, { name: 'ToolProviderResult', - declaration: 'export interface ToolProviderResult {\n schemas: ToolSchema[];\n knownNames?: readonly string[];\n}', + declaration: 'export interface ToolProviderResult {\n readonly schemas: readonly ToolSchema[];\n readonly knownNames?: readonly string[];\n readonly ownerFinalNames?: readonly string[];\n}', }, { name: 'ToolRestriction', - declaration: 'export interface ToolRestriction {\n allow?: string[];\n deny?: string[];\n}', + declaration: 'export interface ToolRestriction {\n readonly allow?: readonly string[];\n readonly deny?: readonly string[];\n}', }, { name: 'ToolResult', diff --git a/packages/core/agent-loop/README.md b/packages/core/agent-loop/README.md index 8f6b9604c5..871689008b 100644 --- a/packages/core/agent-loop/README.md +++ b/packages/core/agent-loop/README.md @@ -8,16 +8,18 @@ This is the only package in the harness that contains concrete loop logic. Every ### Public API -Lifecycle (scoped and dual-owned): programmatic creation and resume snapshot caller-owned identity/configuration data, obtain registry/store-owned capabilities for both unpublished IDs, mint `agent.ctx`, and install the ordered teardown skeleton before awaiting optional `setup`. `AgentFactory.createAgent(ownerCtx, options)` and `resume(ownerCtx, options)` receive caller ownership explicitly; the trace-bound `AgentLoop` receiver still supplies the dependency origin, so a caller that injects only `agents` can create an agent whose scope reaches the loop's `sessions`/`llm`/`tools`/`systemPrompt` surface. The caller owns cancellation and the returned handle, while AgentLoop remains a structural second owner because the live driver depends on that service surface: unloading the provider aborts pending load/setup, tears down live programmatic agents, and awaits the same quiescence and ID-release boundary before its dependencies disappear. +Creation and resume are one rollback-covered transaction: construct a private session, concrete agent, and scoped context; await optional setup; enter both registries; announce `session/created` then `agent/created`; emit `agent/session-start`; and only then start the driver. Setup receives the full scoped `Context` as trusted same-process composition code and must not drive the unpublished agent. Ordinary typed identity and option inputs are borrowed under their readonly contract, while seed events and session metadata are validated and snapshotted because they cross the durable session boundary. An optional `AbortSignal` cancels only load/setup/publication and is detached before the returned handle becomes visible. -The complete create transaction is factory-tracked before ID reservation/session validation, and both a factory placeholder and lifecycle-long caller sentinel exist before scope minting can reenter plugin lifecycle notifications. The caller sentinel adopts the exact reservation effects and always follows the memoized lifecycle boundary, including handle-first teardown followed by caller unload. Resume adds a load sentinel before persistence I/O; it waits for load settlement until `startOwned` synchronously returns a lifecycle/rollback disposer, then follows that disposer without a handoff gap. The ID capabilities reject competing `register`/`enter`/`prepare`/`create` calls and remain held through scope quiescence. A create hands one-read raw seed and metadata references synchronously to the session boundary, which rejects exotic shells and materializes accepted values in a single recursive pass; pre-cloning either value could incorrectly sanitize prototypes. Resume captures each loaded metadata field once. After setup resolves, the factory checks caller and provider liveness after constructing both registry entries but before the first announcement, after `session/created`, after `agent/created`, and again after `agent/session-start` before starting the driver, so synchronous getter- or listener-triggered teardown wins. A publication-wide barrier flips lifecycle liveness immediately but keeps both entries and `agent.ctx` intact until the current synchronous notification phase unwinds; only then does rollback revoke them. Registry/store entries claim IDs across caller-code commit windows, detach exact objects only, and reuse stable carriers for paired edges. Load/setup rejection or owner unload before announcement emits no creation edge; if teardown begins inside a creation or session-start listener, the already-started notifications are paired during rollback and no live or drivable publication survives. Teardown runs stop/drain (including outstanding idle-injection flushes) → unregister → detach session → unwind scope → release reservations. After quiescence, the caller sentinel and any resume-load sentinel disarm and remove their owner-fiber effects so a long-lived caller does not retain the completed agent and scope. Ordinary non-vetoing `agent/*` notifications go through `agentEvents(ctx, agent)`; the registry's paired disposal edge applies the same failure containment through its captured carrier. Per-step assembly goes through `assembleContextFor(agent)`, and the turn-end durability checkpoint goes through `ctx.sessions.flush(session)`. +The caller fiber and the AgentLoop provider are co-owners. `AgentFactory.createAgent(ownerCtx, options)` and `resume(ownerCtx, options)` receive caller ownership explicitly, while the factory keeps its own dependency context for `sessions`/`llm`/`tools`/`systemPrompt`; this lets a caller inject only `agents` without shrinking the new agent's service surface. Caller unload, handle disposal, or provider unload converge on one memoized quiescence boundary. Provider shutdown waits both resource teardown and the public create/resume wrapper that observed deactivation, so no continuation can publish after dependencies disappear. + +IDs are caller-chosen and assumed globally unique; accidental UUID collisions are outside the supported model. Two concurrent operations with the same agent or session id may both prepare, but the final `enter()` calls arbitrate publication and every loser rolls its private resources back. Each detach is bound to the exact entered object, so a stale disposer cannot remove a later same-id replacement. A detach requested during a synchronous creation notification waits for that dispatch to unwind, preserving created/disposed pairing. Teardown runs stop and drain (including outstanding idle-injection flushes) → detach agent → detach session → unwind scope; IDs become reusable at detach even if private scope cleanup is still finishing. Ordinary non-vetoing `agent/*` notifications go through `agentEvents(ctx, agent)`, per-step assembly goes through `assembleContextFor(agent)`, and turn-end durability checkpoints go through `ctx.sessions.flush(session)`. - `ctx.agentLoop.create(id: string, options?: AgentOptions, meta?: { cwd?: string }): ReactLoopAgent` — synchronous no-setup create, used directly by programs and by `cordis.yml`-configured agents. It creates a fresh per-run session id `${id}-session-` with optional metadata; the uuid avoids colliding with a prior durable log. Each call is a new session (a deliberate demo simplification — a real resume-or-create policy is a TODO). Disposed with the calling fiber. `AgentLoop` also implements the `AgentFactory` seam and registers itself via `ctx.agents.setFactory(this)`, so plugins create/resume agents through `ctx.agents` (the interface): -- `ctx.agents.create({ agentId, sessionId, meta?, seed?, agentOptions?, setup? }): Promise` — programmatic create on a caller-supplied `sessionId`, NOT `${id}-session`. It awaits the unpublished setup transaction before returning; `meta` carries cwd/lineage/seed-boundary metadata and `seed` reconstructs a forked child prefix after the session boundary validates and detaches each raw value in one pass. The resolved [`AgentHandle`](../agent/README.md) owns exact teardown. -- `ctx.agents.resume({ agentId, resumeSessionId, agentOptions?, setup? }): Promise` — load a persisted session via `ctx.sessionPersistence` ([session persistence](../../../docs/rfc/implemented/architecture/2026-06-14-session-persistence.md)), reconstruct its history, then await setup against a fresh unpublished agent scope before rollback-covered publication. The live session id is the resumed id; turn numbering and derived history continue from the loaded log. Requires a session-persistence backend (NOT hard-injected — non-persistent demos still work; `resume` rejects with a clear error when persistence is absent). Returns an `AgentHandle`. +- `ctx.agents.create({ agentId, sessionId, meta?, seed?, agentOptions?, setup?, signal? }): Promise` — programmatic create on a caller-supplied `sessionId`, NOT `${id}-session`. It awaits the unpublished setup transaction before returning; `meta` carries cwd/lineage/seed-boundary metadata and `seed` reconstructs a forked child prefix after the session boundary validates and snapshots the durable values. `signal` applies only until this promise settles. The resolved [`AgentHandle`](../agent/README.md) owns exact teardown. +- `ctx.agents.resume({ agentId, resumeSessionId, agentOptions?, setup?, signal? }): Promise` — load a persisted session via `ctx.sessionPersistence` ([session persistence](../../../docs/rfc/implemented/architecture/2026-06-14-session-persistence.md)), reconstruct its history, then await setup against a fresh unpublished agent scope before rollback-covered publication. The live session id is the resumed id; turn numbering and derived history continue from the loaded log. Requires a session-persistence backend (NOT hard-injected — non-persistent demos still work; `resume` rejects with a clear error when persistence is absent). `signal` is creation-only. Returns an `AgentHandle`. The config-driven `ctx.agentLoop.create()` path keeps its agent owned by the loop fiber (it discards the handle). For a programmatic agent, the handle holder is the only consumer-facing teardown capability; AgentLoop provider unload is the independent structural teardown edge, not another handle exposed to application code. @@ -38,13 +40,13 @@ interface Config { } ``` -Agents listed in config are auto-created at startup. `cwd` applies only to fresh config-created sessions; `resumeSessionId` keeps the persisted session header. There is no per-agent persona: the deployment persona is `dsh-system-prompt`'s own `persona` config, shared by every agent in the context. The plugin registers the built-in `model`/`cwd` prompt variables on `ctx.systemPrompt`, resolved per step from the `assemble({ agent })` context — runtime facts of the agents THIS loop drives, unlike the `harness:identity`/`deployment:persona` sections, which live on `dsh-system-prompt` so they survive a swapped loop plugin. +Agents listed in config are auto-created at startup. `cwd` applies only to fresh config-created sessions; `resumeSessionId` keeps the persisted session header. Config agents have no per-agent persona field: they use `dsh-system-prompt`'s deployment default, while programmatic factory callers can register an agent-scoped `deployment:persona` shadow in `setup`. The plugin registers the built-in `model`/`cwd` prompt variables on `ctx.systemPrompt`, resolved per step from the `assemble({ agent })` context — runtime facts of the agents THIS loop drives, unlike the `harness:identity` and default `deployment:persona` sections, which live on `dsh-system-prompt` so they survive a swapped loop plugin. ### Exported concrete class - `ReactLoopAgent` — the concrete `Agent` implementation. Its inbox is a JavaScript native-private field, and one prepared session can be claimed by only one concrete driver. Everything observable happens through session events and the `agent/*` event taxonomy. -`Inbox`, `runLoop`, and the instance-bound enable/start controls are package-internal. The package root does not export them, and the package exports map exposes no `./src/*` escape hatch; lifecycle owners create agents through `ctx.agents` rather than constructing or starting the driver internals. `ReactLoopAgent.send()` and running `steer()` materialize content plus resolved source once as detached, deeply frozen lossless JSON, then share that accepted record between `agent/queued` and the inbox; malformed data throws before either boundary. +`Inbox`, `runLoop`, and the instance-bound publication/start controls are package-internal. The package root does not export them, and the package exports map exposes no `./src/*` escape hatch; lifecycle owners create agents through `ctx.agents` rather than constructing or starting the driver internals. `ReactLoopAgent.send()` and running `steer()` materialize content plus resolved source once as detached, deeply frozen lossless JSON, then share that accepted record between `agent/queued` and the inbox; malformed data throws before either boundary. ### Loop lifecycle (`loop.ts`) @@ -91,7 +93,7 @@ forever: idle unless more queued ``` -Error containment: a throwing plugin ends the **turn**, never the loop. A malformed or throwing `agent/turn-stop` policy likewise fails the turn closed. A successful terminal stop stays authoritative through `turn/end` and `session/flush`, preventing their listeners from resurrecting steering through the late fallback. Dispose mid-turn emits `agent/status('disposed')` and ends with reason `disposed`. A step that hits the model's output-token ceiling makes the turn end `max-tokens` (the rule: any `max-tokens` step in the turn surfaces as `max-tokens`; `disposed`/`aborted`/`error` still take precedence) — distinct from a clean `completed` stop. +Error containment: a throwing plugin ends the **turn**, never the loop. A throwing `agent/turn-stop` policy likewise fails the turn closed. A successful terminal stop stays authoritative through `turn/end` and `session/flush`, preventing their listeners from resurrecting steering through the late fallback. Dispose mid-turn emits `agent/status('disposed')` and ends with reason `disposed`. A step that hits the model's output-token ceiling makes the turn end `max-tokens` (the rule: any `max-tokens` step in the turn surfaces as `max-tokens`; `disposed`/`aborted`/`error` still take precedence) — distinct from a clean `completed` stop. Cancellation: `agent.cancel()` is the single public stop primitive — it clears the queued + steering FIFOs, aborts the in-flight step, and drives a turn-scoped marker the driver checks at every point a turn could start or continue (right after the idle wait, after the `running` flip, before each step, and at the continuation gate) so a turn about to start is dropped. A cancelled turn ends `aborted`; a queued-but-not-started prompt never runs and cannot be batched into the cancelled turn. The marker is reset once per loop iteration, so a cancel governs exactly one turn and never leaks onto a later prompt. (The loop still aborts its own per-step `AbortController` directly on disposal and from `cancel()`; that controller is loop-internal, not a public verb.) diff --git a/packages/core/agent-loop/src/agent.ts b/packages/core/agent-loop/src/agent.ts index a449e6c907..ff66f942be 100644 --- a/packages/core/agent-loop/src/agent.ts +++ b/packages/core/agent-loop/src/agent.ts @@ -16,9 +16,6 @@ import { snapshotJsonValue, type Session } from '@deepseek-ai/dsh-session' import { Inbox, type InboxMessage } from './inbox.ts' import { isTurnOpen, lastTurnNumber, runLoop } from './loop.ts' -/** Agents whose rollback-covered publication enabled driving. */ -const driveEnabledAgents = new WeakSet() - /** Sessions already claimed by a concrete driver construction. */ const claimedDriverSessions = new WeakSet() @@ -28,12 +25,18 @@ const startDriver = Symbol('dsh.agent-loop.start-driver') /** Module-private quiescent stop, valid both before and after driver start. */ const stopDriver = Symbol('dsh.agent-loop.stop-driver') +/** Module-private context binding for the mutually referential agent scope. */ +const bindContext = Symbol('dsh.agent-loop.bind-context') + +/** Module-private publication marker. */ +const publishAgent = Symbol('dsh.agent-loop.publish-agent') + /** Factory-owned controls that can operate only on the agent created with them. */ export interface PreparedReactLoopAgent { /** The unpublished concrete agent. */ agent: ReactLoopAgent - /** Open its driving verbs at the rollback-covered publication boundary. */ - enableDrive(): void + /** Mark the agent public so teardown emits its status lifecycle. */ + markPublished(): void /** Stop the prepared instance even when publication has not started its loop. */ dispose(): Promise | void /** @@ -48,7 +51,7 @@ export interface PreparedReactLoopAgent { * Construct one concrete agent together with unforgeable, instance-bound * lifecycle controls. The package surface deliberately exposes neither source * subpaths nor this helper: setup code may identify the concrete class, but it - * cannot enable or start the factory's unpublished instance. + * cannot publish or start the factory's unpublished instance. * @param ctx - the agent-loop service context used for driving and events. * @param id - the concrete agent identity. * @param options - loop options for the agent. @@ -62,14 +65,11 @@ export function prepareReactLoopAgent( throw new Error(`session "${session.id}" already has a concrete agent driver`) } const agent = new ReactLoopAgent(ctx, id, options, session) - // Construction snapshots caller options and can throw. Claim only the fully - // initialized driver so the same prepared session remains retryable after a - // rejected caller value. claimedDriverSessions.add(session) const dispose = () => agent[stopDriver]() return { agent, - enableDrive: () => { driveEnabledAgents.add(agent) }, + markPublished: () => { agent[publishAgent]() }, dispose, startDriver: () => { agent[startDriver]() @@ -82,20 +82,12 @@ export function prepareReactLoopAgent( * Install the concrete agent's scope context exactly once. Construction and * scope minting are mutually referential (the scope key is the agent), so the * factory performs this one post-construction binding before setup receives - * the unpublished agent. The runtime slot is non-writable/non-configurable; - * TypeScript `readonly` alone would still let JavaScript redirect later - * registrations to another context. + * the unpublished agent. The module-private binding rejects a second bind. * @param agent - the unpublished concrete agent to bind. * @param ctx - its fully extended agent scope context. */ export function bindReactLoopAgentContext(agent: ReactLoopAgent, ctx: Context): void { - if (Object.hasOwn(agent, 'ctx')) throw new Error(`agent "${agent.id}" context is already bound`) - Object.defineProperty(agent, 'ctx', { - value: ctx, - enumerable: true, - writable: false, - configurable: false, - }) + agent[bindContext](ctx) } /** @@ -106,7 +98,7 @@ export function bindReactLoopAgentContext(agent: ReactLoopAgent, ctx: Context): * the agent/* event taxonomy — plugins never need this class. */ export class ReactLoopAgent implements Agent { - /** Queued + steering FIFOs; native-private so setup cannot bypass driving verbs. */ + /** Queued + steering FIFOs; native-private so callers cannot bypass the public driving verbs. */ readonly #inbox = new Inbox() /** @@ -117,12 +109,20 @@ export class ReactLoopAgent implements Agent { * context are mutually referential (the scope is keyed BY this agent), so * neither can exist strictly before the other. */ - declare readonly ctx: Context + private boundContext: Context | undefined + + /** The agent's scoped composition context, bound once by its factory. */ + get ctx(): Context { + if (this.boundContext === undefined) throw new Error(`agent "${this.id}" context is not bound`) + return this.boundContext + } private _status: AgentStatus = 'idle' private currentAbort: AbortController | undefined /** Whether runLoop has been installed into {@link done}. */ private driverStarted = false + /** Whether registry publication began and status disposal is externally visible. */ + private published = false /** * Turn-scoped cancel marker, set by {@link cancel} and read/cleared by the * driver loop (via the LoopHandle) at every point a turn could start or @@ -167,16 +167,6 @@ export class ReactLoopAgent implements Agent { public readonly options: AgentOptions, public readonly session: Session, ) { - const acceptedOptions = deepFreeze(structuredClone(options)) - // Pin the public ownership/identity bindings in the runtime object. A - // JavaScript caller can otherwise replace TS-readonly parameter properties - // after publication and split the registry, driver, session, and model - // configuration into different worlds. - Object.defineProperties(this, { - id: { value: id, enumerable: true, writable: false, configurable: false }, - options: { value: acceptedOptions, enumerable: true, writable: false, configurable: false }, - session: { value: session, enumerable: true, writable: false, configurable: false }, - }) const { promise, resolve } = Promise.withResolvers() this.disposed = promise this.resolveDisposed = resolve @@ -233,36 +223,24 @@ export class ReactLoopAgent implements Agent { if (this._status === 'disposed') throw new Error(`agent "${this.id}" is disposed`) } - /** Reject every driving verb while creation setup still owns the agent. */ - private assertDriveEnabled(action: string): void { - if (driveEnabledAgents.has(this)) return - throw new Error(`agent "${this.id}" cannot ${action} before creation setup completes`) - } - send(content: ContentBlock[], options?: SendOptions): void { - this.assertDriveEnabled('send') this.assertNotDisposed() const accepted = this.acceptInboxMessage(content, options) - // Materialization invokes caller getters, which may reenter handle disposal. - this.assertNotDisposed() this.#inbox.enqueue(accepted) - const info = deepFreeze({ source: accepted.source, steering: false }) + const info = { source: accepted.source, steering: false } as const agentEvents(this.loopCtx, this).emit('agent/queued', accepted.content, info) } steer(content: ContentBlock[], options?: SendOptions): void { - this.assertDriveEnabled('steer') this.assertNotDisposed() if (this._status !== 'running') { this.send(content, options); return } const accepted = this.acceptInboxMessage(content, options) - this.assertNotDisposed() this.#inbox.steer(accepted) - const info = deepFreeze({ source: accepted.source, steering: true }) + const info = { source: accepted.source, steering: true } as const agentEvents(this.loopCtx, this).emit('agent/queued', accepted.content, info) } inject(content: ContentBlock[], options?: SendOptions): void { - this.assertDriveEnabled('inject') this.assertNotDisposed() const source = this.resolveSource(options) if (isTurnOpen(this.session)) { @@ -323,7 +301,6 @@ export class ReactLoopAgent implements Agent { } cancel(reason?: string): void { - this.assertDriveEnabled('cancel') // Arm-gate: only mark a cancellation when there is actually work to cancel — // a running turn, an in-flight step, or queued/steering work. An idle cancel // with nothing pending is a true no-op; arming the marker then would wrongly @@ -382,6 +359,17 @@ export class ReactLoopAgent implements Agent { }) } + /** Bind the mutually referential scope context once. */ + private [bindContext](ctx: Context): void { + if (this.boundContext !== undefined) throw new Error(`agent "${this.id}" context is already bound`) + this.boundContext = ctx + } + + /** Mark that public lifecycle publication began. */ + private [publishAgent](): void { + this.published = true + } + /** * Start the driver loop. The prepared controller already owns its stable * disposer, so teardown can mark the agent disposed even in the narrow @@ -424,15 +412,15 @@ export class ReactLoopAgent implements Agent { this.settleIdleWaiters() this.currentAbort?.abort('disposed') // An unpublished rollback has no public status lifecycle to announce. - // Once driving is enabled, disposed is part of the agent/status contract. - if (driveEnabledAgents.has(this)) { + // Once publication begins, disposed is part of the agent/status contract. + if (this.published) { agentEvents(this.loopCtx, this).emit('agent/status', 'disposed') } } // Before runLoop starts there is normally nothing asynchronous to drain; // keep publication rollback synchronous so create() cannot throw while its // session/agent entries are still briefly live. A session-start listener - // may have used the newly enabled inject() surface, however, so preserve + // may have called inject(), however, so preserve // its durability checkpoint as a real quiescence boundary. if (!this.driverStarted && this.pendingIdleFlushes.size === 0) return return this.drainDriver() @@ -455,11 +443,7 @@ export class ReactLoopAgent implements Agent { } } -/** Render an arbitrary thrown value without allowing coercion to throw again. */ +/** Render an ordinary thrown value for the error event and log. */ function renderThrown(value: unknown): string { - try { - return value instanceof Error ? value.message : String(value) - } catch { - return '' - } + return value instanceof Error ? value.message : String(value) } diff --git a/packages/core/agent-loop/src/index.ts b/packages/core/agent-loop/src/index.ts index 8722751962..4ad4779176 100644 --- a/packages/core/agent-loop/src/index.ts +++ b/packages/core/agent-loop/src/index.ts @@ -1,109 +1,318 @@ /** - * THE concrete agent plugin: creates ReactLoopAgents, runs their loops, and - * registers them in ctx.agents. Deliberately thin — every behavior beyond - * "call the model, run the tools, repeat" belongs to plugins on the event - * taxonomy. + * Concrete agent-loop plugin: creates scoped ReactLoopAgents, publishes them + * through the agent/session registries, and owns their ordered teardown. * * @module @deepseek-ai/dsh-agent-loop */ -import { Context, CordisError, FiberState, Service, symbols } from 'cordis' +import { Context, FiberState, Service } from 'cordis' import { randomUUID } from 'node:crypto' import z from 'schemastery' import { createScope } from '@deepseek-ai/dsh-scope' import type { Scope } from '@deepseek-ai/dsh-scope' import { agentEvents } from '@deepseek-ai/dsh-agent' -import type { AgentFactory, AgentHandle, AgentId, AgentOptions, AgentRegistrationReservation, CreateAgentOptions, ResumeAgentOptions, SessionStartSource } from '@deepseek-ai/dsh-agent' +import type { + AgentFactory, + AgentHandle, + AgentId, + AgentOptions, + CreateAgentOptions, + ResumeAgentOptions, + SessionStartSource, +} from '@deepseek-ai/dsh-agent' import type {} from '@deepseek-ai/dsh-llm' -import { SessionId, type SessionHeader } from '@deepseek-ai/dsh-session' -import type { Session, SessionRegistrationReservation } from '@deepseek-ai/dsh-session' +import { SessionId } from '@deepseek-ai/dsh-session' +import type { Session, SessionHeader } from '@deepseek-ai/dsh-session' import type {} from '@deepseek-ai/dsh-system-prompt' import type {} from '@deepseek-ai/dsh-tools' import type { SessionPersistence } from '@deepseek-ai/dsh-session-persistence' -import { bindReactLoopAgentContext, prepareReactLoopAgent, ReactLoopAgent } from './agent.ts' +import { + bindReactLoopAgentContext, + prepareReactLoopAgent, + ReactLoopAgent, +} from './agent.ts' +import type { PreparedReactLoopAgent } from './agent.ts' export { ReactLoopAgent } from './agent.ts' -/** Both unpublished identity capabilities held by one factory transaction. */ -interface RegistrationReservations { - agent: AgentRegistrationReservation - session: SessionRegistrationReservation - release(): void -} - -/** A synchronously established ownership handoff plus its async publication result. */ -interface OwnedAgentStart { - result: Promise - dispose: () => Promise -} - -/** Internal carrier for a preparation error whose rollback still has to quiesce. */ -class LifecyclePreparationFailure extends Error { - constructor( - readonly reason: unknown, - readonly dispose: () => Promise, - ) { - super('agent lifecycle preparation failed', { cause: reason }) - this.name = 'LifecyclePreparationFailure' - } -} - -/** Stable construction-time state shared by every traceable AgentLoop receiver. */ -interface FactoryOwnership { - isActive(): boolean - track(dispose: () => Promise): () => void - dispose(): Promise -} - -/** Fiber states in which a concrete factory cannot safely serve dependencies. */ -const INACTIVE_FACTORY_STATES: ReadonlySet = new Set([ +/** Fiber states that cannot own or serve a new lifecycle. */ +const INACTIVE_STATES: ReadonlySet = new Set([ FiberState.UNLOADING, FiberState.DISPOSED, FiberState.FAILED, ]) -/** Build a tamper-resistant controller around one factory's private ledger. */ -function createFactoryOwnership(fiber: Context['fiber']): FactoryOwnership { - let accepting = true - const transactions = new Set<() => Promise>() - const isActive = (): boolean => accepting && !INACTIVE_FACTORY_STATES.has(fiber.state) - return Object.freeze({ - isActive, - track(dispose: () => Promise): () => void { - /* v8 ignore next -- every call site checks the same controller immediately - * before this synchronous, non-reentrant insertion; retain the guard as an invariant */ - if (!isActive()) throw new Error('agent loop is not active') - transactions.add(dispose) - return () => { transactions.delete(dispose) } - }, - async dispose(): Promise { - accepting = false - const disposers = [...transactions] - transactions.clear() - const results = await Promise.allSettled(disposers.map(dispose => Promise.resolve().then(dispose))) - /* v8 ignore next -- tracked lifecycle/load boundaries are deliberately - * infallible; keep reasons if that lower-level contract ever breaks */ - const errors = results.flatMap(result => result.status === 'rejected' ? [result.reason as unknown] : []) - /* v8 ignore next -- every tracked boundary is deliberately infallible; - * preserve an exact unexpected single failure as a defensive backstop */ - if (errors.length === 1) throw errors[0] - /* v8 ignore next -- multiple failures require multiple contract-breaking - * lifecycle disposers, but teardown must still retain every cause */ - if (errors.length > 1) throw new AggregateError(errors, 'agent loop transaction disposal failed') - }, - }) +/** Factory-level ownership of every preparing or live transaction. */ +class FactoryOwnership { + private accepting = true + private transactions = new Set() + + constructor(private readonly fiber: Context['fiber']) {} + + isActive(): boolean { + return this.accepting && !INACTIVE_STATES.has(this.fiber.state) + } + + track(transaction: AgentCreationTransaction): () => void { + this.transactions.add(transaction) + return () => { this.transactions.delete(transaction) } + } + + async dispose(): Promise { + this.accepting = false + const reason = new Error('agent loop is not active') + await Promise.all( + [...this.transactions].map(transaction => transaction.disposeForFactory(reason)), + ) + } } -/** Private ownership controllers keyed by the concrete, unproxied service. */ -const factoryOwnerships = new WeakMap() +/** Build the public cancellation error while preserving a caller-supplied cause. */ +function signalAbortError(id: AgentId, signal: AbortSignal): Error { + if (signal.reason instanceof Error) return signal.reason + return new Error(`agent "${id}" creation aborted`, { cause: signal.reason }) +} -/** Recover the stable controller when a Cordis trace proxy is the receiver. */ -function factoryOwnershipFor(loop: AgentLoop): FactoryOwnership { - const original = (loop as AgentLoop & { [symbols.original]?: AgentLoop })[symbols.original] ?? loop - // Installed immediately after Service construction, before AgentLoop starts - // any effect or config-driven transaction. - // eslint-disable-next-line @typescript-eslint/no-non-null-assertion - return factoryOwnerships.get(original)! +/** + * One create/resume transaction from caller ownership through unpublished + * setup, rollback-covered publication, and final quiescent teardown. + * + * The class deliberately owns the state machine in one place. Registries only + * arbitrate identity at their final `enter()` calls; before that point every + * resource is private to this transaction. + */ +class AgentCreationTransaction { + private active = true + private failure: Error | undefined + private readonly deactivation = Promise.withResolvers() + private readonly publication = Promise.withResolvers() + private readonly torndown = Promise.withResolvers() + private readonly wrapperCompletion = Promise.withResolvers() + private preparing: Promise | undefined + private driver: PreparedReactLoopAgent | undefined + private scope: Scope | undefined + private session: Session | undefined + private lifecycleDispose: (() => Promise | void) | undefined + private detachSession: (() => void) | undefined + private detachAgent: (() => void) | undefined + private publishing = false + private cleanupTask: Promise | undefined + private ownerFollowing = true + private readonly ownerDispose: () => Promise | void + private readonly untrackFactory: () => void + private readonly abortListener: (() => void) | undefined + readonly ownerAgent: Context['agent'] + readonly ownerFiber: Context['fiber'] + + constructor( + private readonly loopCtx: Context, + private readonly ownerCtx: Context, + private readonly ownership: FactoryOwnership, + readonly id: AgentId, + signal?: AbortSignal, + ) { + ownerCtx.fiber.assertActive() + this.ownerAgent = ownerCtx.agent + this.ownerFiber = ownerCtx.fiber + if (!ownership.isActive()) throw new Error('agent loop is not active') + this.ownerDispose = ownerCtx.effect(() => () => { + if (!this.ownerFollowing) return + return this.dispose(new Error(`agent "${id}" setup aborted: owner disposed during setup`)) + }, `agentLoop.owner(${id})`) + this.untrackFactory = ownership.track(this) + if (signal === undefined) { + this.abortListener = undefined + } else { + this.abortListener = () => { + /* v8 ignore next 3 -- transaction teardown contains callback/driver failures; rejection is a future-drift backstop. */ + void this.dispose(signalAbortError(id, signal)).catch((error: unknown) => { + this.loopCtx.logger.error(error) + }) + } + signal.addEventListener('abort', this.abortListener, { once: true }) + if (signal.aborted) this.deactivate(signalAbortError(id, signal)) + } + this.signal = signal + } + + private readonly signal: AbortSignal | undefined + + /** Whether caller, provider, and optional parent-agent ownership remain live. */ + isActive(): boolean { + return this.active + && this.ownership.isActive() + && this.ownerFiber.uid !== null + && !INACTIVE_STATES.has(this.ownerFiber.state) + && this.ownerAgent?.status !== 'disposed' + } + + /** Fail synchronously at every real lifecycle boundary after deactivation. */ + assertActive(): void { + if (this.isActive()) return + if (!this.ownership.isActive()) throw new Error('agent loop is not active') + throw this.failure ?? new Error(`agent "${this.id}" setup aborted: owner disposed during setup`) + } + + /** Race an external async operation against structural/signal deactivation. */ + async waitFor(operation: PromiseLike | T): Promise { + this.assertActive() + return await Promise.race([ + Promise.resolve(operation), + this.deactivation.promise.then(() => { + /* v8 ignore next -- deactivate() assigns failure before resolving deactivation. */ + throw this.failure ?? new Error(`agent "${this.id}" creation deactivated`) + }), + ]) + } + + /** Construct the driver and scope, then install their complete ordered lifecycle. */ + prepare(options: AgentOptions, session: Session): ReactLoopAgent { + this.assertActive() + const gate = Promise.withResolvers() + this.preparing = gate.promise + try { + this.session = session + const driver = prepareReactLoopAgent(this.loopCtx, this.id, options, session) + this.driver = driver + const agent = driver.agent + const scope = createScope(this.loopCtx, agent) + this.scope = scope + bindReactLoopAgentContext(agent, scope.ctx.extend({ agent })) + this.installLifecycle(scope, driver) + this.assertActive() + return agent + } catch (error: unknown) { + if (!this.isActive() && error instanceof Error && /inactive context/.test(error.message)) { + throw this.failure ?? this.disposalReason() + } + throw error + } finally { + gate.resolve() + this.preparing = undefined + } + } + + /** Register the exact scope disposer inside the ordered transaction effect. */ + private installLifecycle(scope: Scope, driver: PreparedReactLoopAgent): void { + this.lifecycleDispose = this.ownerCtx.effect(function* (this: AgentCreationTransaction) { + // First yielded, disposed last. + yield () => { this.finish() } + yield scope.rawDispose + yield () => { + this.detachSession?.() + this.detachSession = undefined + } + yield () => { + this.detachAgent?.() + this.detachAgent = undefined + } + // Last yielded, disposed first. + yield () => { + this.deactivate(this.disposalReason()) + if (this.publishing) { + return this.publication.promise.then(() => driver.dispose()) + } + return driver.dispose() + } + }.bind(this), `agentLoop.lifecycle(${this.id})`) + } + + /** Publish the exact prepared objects and start the driver. */ + publish(source: SessionStartSource): AgentHandle { + this.assertActive() + const driver = this.driver + /* v8 ignore next -- publish() is private and every caller invokes prepare() first. */ + if (driver === undefined) throw new Error(`agent "${this.id}" is not prepared`) + const agent = driver.agent + const session = this.session + /* v8 ignore next -- prepare() assigns the session before it can produce the driver above. */ + if (session === undefined) throw new Error(`agent "${this.id}" has no prepared session`) + this.publishing = true + try { + this.detachSession = agent.ctx.sessions.enter(session) + this.detachAgent = this.loopCtx.agents.enter(agent) + + agent.ctx.sessions.announce(session) + this.assertActive() + this.loopCtx.agents.announce(agent) + this.assertActive() + + driver.markPublished() + agentEvents(this.loopCtx, agent).emit('agent/session-start', source) + this.assertActive() + driver.startDriver() + return { agent, dispose: () => this.dispose() } + } finally { + this.publishing = false + this.publication.resolve() + } + } + + /** Mark the transaction inactive exactly once and wake load/setup races. */ + private deactivate(reason: Error): void { + if (!this.active) return + this.active = false + this.failure = reason + this.deactivation.resolve() + } + + /** Choose the structural cause when an owner/factory effect starts teardown first. */ + private disposalReason(): Error { + if (this.failure !== undefined) return this.failure + if (!this.ownership.isActive()) return new Error('agent loop is not active') + if (this.ownerFiber.uid === null || INACTIVE_STATES.has(this.ownerFiber.state) || this.ownerAgent?.status === 'disposed') { + return new Error(`agent "${this.id}" setup aborted: owner disposed during setup`) + } + return new Error(`agent "${this.id}" lifecycle disposed`) + } + + /** Complete ownership bookkeeping after every resource reached quiescence. */ + private finish(): void { + this.untrackFactory() + this.ownerFollowing = false + void this.ownerDispose() + this.torndown.resolve() + } + + /** + * Deactivate and quiesce this transaction. The promise is memoized because + * Cordis effect disposers are single-shot while handles promise shared + * quiescence to every racing owner. + */ + dispose(reason = new Error(`agent "${this.id}" lifecycle disposed`)): Promise { + this.deactivate(reason) + return (this.cleanupTask ??= (async () => { + if (this.preparing !== undefined) await this.preparing + if (this.lifecycleDispose !== undefined) { + await this.lifecycleDispose() + await this.torndown.promise + return + } + try { + await this.driver?.dispose() + } finally { + try { + await this.scope?.dispose() + } finally { + this.finish() + } + } + })()) + } + + /** Mark the public create/resume continuation settled and detach its creation-only signal. */ + finishWrapper(): void { + if (this.signal !== undefined && this.abortListener !== undefined) { + this.signal.removeEventListener('abort', this.abortListener) + } + this.wrapperCompletion.resolve() + } + + /** Factory shutdown joins both resource teardown and the public wrapper's deactivation continuation. */ + async disposeForFactory(reason: Error): Promise { + await this.dispose(reason) + await this.wrapperCompletion.promise + } } declare module 'cordis' { @@ -112,52 +321,24 @@ declare module 'cordis' { } } -/** - * Plugin config: the agents to create — or resume, via `resumeSessionId` — - * declaratively at startup, so a cordis.yml deployment needs no code. - */ +/** Plugin configuration for declarative startup agents. */ export interface Config { - /** Agents created from configuration at startup. */ + /** Agents created or resumed at plugin startup. */ agents: (AgentOptions & { - /** Agent id to register under; also seeds the fresh per-run session id (`${id}-session-`). */ + /** Registry identity for the live agent. */ id: AgentId - /** Optional workspace cwd for the config-created fresh session. */ + /** Optional workspace for a fresh session. */ cwd?: string - /** - * If set, the config agent RESUMES this persisted session id instead of - * starting a fresh `${id}-session-`. Sourced from an env var in - * cordis.yml (`resumeSessionId: !!js process.env.RESUME_SESSION_ID`), so a - * demo can continue a prior conversation without code changes. Requires a - * `dsh-session-persistence` backend; the resume is deferred until that - * service is available (via `ctx.inject`) and the loaded session's events - * seed the live session so history continues. - * - * The schema accepts a plain string at runtime (cordis.yml values are - * untyped); the brand is compile-time only — the config format is the - * boundary where an id enters, so the TYPE declares the brand here. - */ + /** Persisted session to resume instead of creating a fresh session. */ resumeSessionId?: SessionId })[] } -/** - * The agent-loop plugin (`ctx.agentLoop`): creates {@link ReactLoopAgent}s, runs - * their loops, and registers them in `ctx.agents`. Also implements the - * {@link AgentFactory} seam, so plugins create/resume agents through - * `ctx.agents` (the interface) without depending on this concrete package. - * - * The loop itself is deliberately thin — every behavior beyond "call the - * model, run the tools, repeat" belongs to plugins listening on the event - * taxonomy declared in @deepseek-ai/dsh-agent. - */ +/** Concrete ReactLoopAgent factory and driver service. */ export class AgentLoop extends Service implements AgentFactory { static inject = ['agents', 'sessions', 'llm', 'tools', 'systemPrompt'] - // The schema validates plain strings (cordis.yml config values are untyped at - // runtime); the {@link Config} TYPE declares the branded `id`/`resumeSessionId` - // because the config format is the boundary where an id enters. The brand is a - // zero-cost compile-time cast, so the runtime schema stays string-based and we - // assert the branded view once here — the single schema boundary. + /** Runtime schema for declarative agents. */ static Config = z.object({ agents: z.array(z.object({ id: z.string().required(), @@ -167,782 +348,143 @@ export class AgentLoop extends Service implements AgentFactory { })).default([]), }) as unknown as z + private readonly ownership: FactoryOwnership + /** Plain holder prevents Cordis from re-tracing the factory's dependency context through a caller shadow. */ + private readonly runtime: { ctx: Context } + constructor(ctx: Context, public config: Config) { super(ctx, 'agentLoop') - const factoryOwnership = createFactoryOwnership(ctx.fiber) - factoryOwnerships.set(this, factoryOwnership) - // Programmatic agents are caller-owned, but this implementation is their - // dependency provider too. Retain a second ownership edge so unloading the - // loop aborts unpublished work and drains every live lifecycle before its - // service surface disappears. - ctx.effect(() => () => factoryOwnership.dispose(), 'agentLoop.factoryTransactions()') - // Provide the agent-creation factory to the registry (effect-scoped: the - // slot is cleared on dispose). - ctx.effect(() => this.ctx.agents.setFactory(this), 'agentLoop.setFactory()') - // The prompt variables the shipped loop provides, registered once. The - // sections themselves (`harness:identity`, `deployment:persona`) belong to - // dsh-system-prompt — they must survive a swapped loop plugin — but - // `{{model}}`/`{{cwd}}` are runtime facts of the agents THIS loop drives: - // it assembles with `{ agent }` each step (loop.ts), and the variables - // project the agent's configured model and its session workspace from that - // context. A provider returns undefined when the fact is absent - // (renderPrompt then rejects a persona that claims it — fail loud). + this.ownership = new FactoryOwnership(ctx.fiber) + this.runtime = { ctx } + ctx.effect(() => () => this.ownership.dispose(), 'agentLoop.transactions()') + ctx.effect(() => ctx.agents.setFactory(this), 'agentLoop.setFactory()') ctx.systemPrompt.variable('model', context => context.agent?.options.model) ctx.systemPrompt.variable('cwd', context => context.agent?.session.header.cwd) + for (const { id, cwd, resumeSessionId, ...options } of config.agents) { - if (resumeSessionId !== undefined && resumeSessionId !== '') { - // Resume a prior session instead of starting fresh. resume() needs - // `ctx.sessionPersistence`, which may load AFTER this plugin (cordis.yml - // lists the backend later). `ctx.inject(['sessionPersistence'], cb)` - // runs `cb` with a child ctx once the service exists; the child reads - // the persistence and hands it to resumeWith (which uses this.ctx — the - // parent — for sessions/registry, all in AgentLoop's static inject). A - // failed resume is contained + logged: startup must not crash. - ctx.effect(() => { - const fiber = this.ctx.inject(['sessionPersistence'], (childCtx: Context) => { - void this.resumeWith(ctx, childCtx.sessionPersistence, { - agentId: id, - resumeSessionId, - agentOptions: options, - }) - .catch((error: unknown) => { - this.ctx.logger.warn(`agent "${id}": config-driven resume of "${resumeSessionId}" failed: ${String(error)}`) - }) - }) - // Return the EXACT child-fiber disposer. Cordis moves a returned - // effect into this labeled owner's teardown tree by function - // identity; a wrapper would leave the child as a concurrent sibling - // and could discard its async quiescence promise. - return fiber.dispose - }, `agentLoop.resume(${id})`) - } else { + if (resumeSessionId === undefined || resumeSessionId === '') { this.create(id, options, cwd === undefined ? {} : { cwd }) + continue } + ctx.effect(() => { + const fiber = ctx.inject(['sessionPersistence'], (childCtx: Context) => { + void this.resumeWith(ctx, childCtx.sessionPersistence, { + agentId: id, + resumeSessionId, + agentOptions: options, + }).catch((error: unknown) => { + ctx.logger.warn(`agent "${id}": config-driven resume of "${resumeSessionId}" failed: ${String(error)}`) + }) + }) + return fiber.dispose + }, `agentLoop.resume(${id})`) } } - /** Whether this concrete factory may begin or publish more work. */ - private factoryIsActive(): boolean { - return factoryOwnershipFor(this).isActive() - } - - /** Reject a call that raced the concrete loop's unload boundary. */ - private assertFactoryActive(): void { - if (this.factoryIsActive()) return - throw new Error('agent loop is not active') - } - - /** Add one memoized quiescence boundary to the factory's ownership set. */ - private trackFactoryTransaction(dispose: () => Promise): () => void { - return factoryOwnershipFor(this).track(dispose) - } - /** - * Config-driven create: an agent on a FRESH, non-colliding session id per run - * (`${id}-session-`). Used for `cordis.yml`-configured agents and as - * the shared core for the programmatic factory {@link createAgent}. - * - * Why a per-run id, not a fixed `${id}-session`: once a durable persistence - * backend is loaded, a fixed id collides on the second run — the backend - * refuses to re-create an id whose log already exists on disk (the SessionId - * is the identity). A fresh id means each run is a new session. - * - * TODO(demo): each run starting a brand-new session is fine for demos but is - * NOT real conversation continuity. A production config-driven agent needs a - * deliberate resume-or-create policy (resume the prior session if one exists, - * else start fresh) or an explicit caller-chosen session id — revisit when the - * UI/ACP path owns session selection. - * @param id - the agent id; also seeds the generated session id. - * @param options - loop options (model, limits, …); defaults applied per option. - * @param meta - optional session metadata for the fresh session. - * @returns the running agent, owned by the calling fiber (no handle). + * Create an agent on a fresh per-run session, owned by the accessing fiber. + * Constructor-driven config calls use the loop fiber itself. + * @param id - agent registry id. + * @param options - concrete loop options. + * @param meta - optional fresh-session workspace metadata. + * @returns the published running agent. */ create(id: AgentId, options: AgentOptions = {}, meta: Pick = {}): ReactLoopAgent { - const sessionId = SessionId(`${id}-session-${randomUUID()}`) - const reservations = this.reserve(id, sessionId) - // Config/programmatic path: prepare the session and let start() fold its - // lifecycle into the agent's composite effect (so a fiber unload tears the - // session + agent down as one ordered chain, capturing the loop's closing - // flush). The whole effect is owned by THIS fiber; no AgentHandle is needed. - let session: Session + const loopCtx = this.runtime.ctx + const transaction = new AgentCreationTransaction(loopCtx, this.ctx, this.ownership, id) try { - session = reservations.session.prepare({ meta }) + const sessionId = SessionId(`${id}-session-${randomUUID()}`) + const session = loopCtx.sessions.prepare(sessionId, { meta }) + const agent = transaction.prepare(options, session) + transaction.publish('startup') + return agent } catch (error: unknown) { - reservations.release() + void transaction.dispose(error instanceof Error ? error : new Error(String(error))) throw error + } finally { + transaction.finishWrapper() } - // start() accepts ownership of both reservation capabilities even when - // synchronous preparation fails; its rollback releases them at quiescence. - const { agent } = this.start(id, options, session, 'startup', reservations) - return agent } /** - * Programmatic factory create ({@link AgentFactory}): an agent on a - * caller-supplied `sessionId` (NOT `${id}-session`), with optional session - * metadata (validated `cwd`, lineage) and an optional `seed` event prefix. The - * ACP bridge uses this so the client-generated session id becomes the - * live/persisted session id; the in-process FORK subagent backend passes a - * `seed` (a balanced completed-turn prefix of the parent's log) so the child - * starts with the parent's context. Returns an {@link AgentHandle} the owner - * disposes to tear down exactly this agent. - * @param ownerCtx - the caller context that owns setup and the live lifecycle. - * @param options - agent id, caller-supplied session id, optional seed/meta, - * and agent options. - * @returns the handle whose dispose tears down exactly this agent. + * Create an owned agent on a caller-supplied session id. + * @param ownerCtx - caller context that structurally owns the transaction. + * @param options - identities, session seed/metadata, loop options, setup, and cancellation. + * @returns the published handle. */ async createAgent(ownerCtx: Context, options: CreateAgentOptions): Promise { - this.assertFactoryActive() - // Snapshot every caller-owned field before the first async setup boundary. - // The callback itself is an identity capability. Agent options detach here; - // seed and metadata stay raw only until sessions.prepare() synchronously - // reads, validates, and detaches them, so structuredClone cannot erase an - // exotic prototype before the session boundary sees it. - const agentId = options.agentId - const sessionId = options.sessionId - const setup = options.setup - const agentOptions = structuredClone(options.agentOptions ?? {}) - const seed = options.seed - const meta = options.meta - // Snapshot accessors can reenter plugin teardown; do not reserve identities - // after the dependency provider has begun unloading. - this.assertFactoryActive() - const { promise: transactionSettled, resolve: markTransactionSettled } = Promise.withResolvers() - const disposeCreateForFactory = (): Promise => transactionSettled - const untrackFactoryCreate = this.trackFactoryTransaction(disposeCreateForFactory) + const transaction = new AgentCreationTransaction( + this.runtime.ctx, + ownerCtx, + this.ownership, + options.agentId, + options.signal, + ) try { - const reservations = this.reserve(agentId, sessionId) - let lifecycleStarted = false - try { - const session = reservations.session.prepare({ - ...seed !== undefined ? { seed } : {}, - ...meta !== undefined ? { meta } : {}, - }) - // A seeded (forked) create is still a fresh start, NOT a resume. - lifecycleStarted = true - return await this.startOwned(ownerCtx, agentId, agentOptions, session, 'startup', reservations, setup).result - } finally { - // Once startOwned is invoked, even a preparation failure carries its - // own quiescent rollback boundary. Only failures before that handoff - // release directly here. - if (!lifecycleStarted) reservations.release() - } + const session = this.runtime.ctx.sessions.prepare(options.sessionId, { + ...options.seed === undefined ? {} : { seed: options.seed }, + ...options.meta === undefined ? {} : { meta: options.meta }, + }) + const agent = transaction.prepare(options.agentOptions ?? {}, session) + await transaction.waitFor(options.setup?.(agent.ctx)) + transaction.assertActive() + return transaction.publish('startup') + } catch (error: unknown) { + await transaction.dispose(error instanceof Error ? error : new Error(String(error))) + throw error } finally { - markTransactionSettled() - untrackFactoryCreate() + transaction.finishWrapper() } } /** - * Resume an agent on a persisted session ({@link AgentFactory}). Loads the - * session log + metadata via `ctx.sessionPersistence`, reconstructs the live - * session with the loaded events (so `lastTurnNumber`/`deriveMessages` - * continue), and starts a fresh agent on it. The live session id is the - * resumed id, NOT `${agentId}-session`. - * - * Requires `ctx.sessionPersistence`; rejects with a clear error if it is not - * configured. NOT hard-injected (that would make non-persistent demos pend - * forever) — callers that need resume (ACP) inject `sessionPersistence`, so - * by the time this runs the service exists. - * @param ownerCtx - the caller context that owns load, setup, and the live lifecycle. - * @param options - the persisted session id to reload, plus agent id/options. - * @returns the handle for the agent resumed on the reconstructed session. + * Resume an owned agent from the configured persistence service. + * @param ownerCtx - caller context that owns load, setup, and the live lifecycle. + * @param options - persisted identity, loop options, setup, and cancellation. + * @returns the published handle. */ async resume(ownerCtx: Context, options: ResumeAgentOptions): Promise { - this.assertFactoryActive() - // Read the service through `ctx.get('sessionPersistence')` — a direct - // global-store lookup keyed by the isolate symbol — NOT - // `this.ctx.sessionPersistence`. AgentLoop deliberately does NOT inject - // `sessionPersistence` (injecting it would pend non-persistent demos - // forever). The `ctx.` property proxy resolves a service by an - // ancestor-only walk of the current fiber's parent chain; from AgentLoop's - // own fiber (which lacks the inject) that walk never reaches the sibling - // backend fiber and throws "cannot get property … without inject". Worse, - // when the call arrives via a traceable shadow (e.g. the ACP bridge child - // fiber → `ctx.agents.resume()` → `this.factory.resume()`), the walk starts - // at the shadow's origin fiber and fails the same way. `ctx.get(name)` - // sidesteps the fiber walk entirely (a store lookup by the global isolate - // key), so resume works from any caller fiber. It is strict by default: a - // backend that is not ACTIVE (absent, or mid-teardown) reads as undefined - // and we reject below, rather than handing back an unusable handle. - const persistence = this.ctx.get('sessionPersistence') + const persistence = this.runtime.ctx.get('sessionPersistence') if (persistence === undefined) { throw new Error('cannot resume: session persistence is not configured (load a dsh-session-persistence backend)') } return this.resumeWith(ownerCtx, persistence, options) } - /** - * Resume against an EXPLICIT persistence handle. Factored out of {@link resume} - * so the config-driven path can pass the handle it obtained from a - * `ctx.inject(['sessionPersistence'], …)` child context: `this.ctx` (the - * service's own fiber) did not inject `sessionPersistence`, so reading it - * there from inside the inject child trips the cordis inject guard. The - * sessions store + registry are still read through `this.ctx` (both are in - * AgentLoop's static inject, so they resolve fine). - */ - private async resumeWith(ownerCtx: Context, persistence: SessionPersistence, options: ResumeAgentOptions): Promise { - // Persistence is an async trust boundary. Reserve, load, reconstruct, and - // publish only the identities/options accepted at entry—never fields - // reread from a caller-owned object after the await. - const agentId = options.agentId - const sessionId = options.resumeSessionId - const agentOptions = structuredClone(options.agentOptions ?? {}) - const setup = options.setup - // Caller-owned accessors above are a synchronous reentrancy boundary: one - // can begin factory unload while options are snapshotted. Re-check before - // installing either ownership sentinel, so a rejected transaction leaves - // no orphan effect or unresolved settlement promise. - this.assertFactoryActive() - const { promise: ownerDisposed, resolve: markOwnerDisposed } = Promise.withResolvers() - const { promise: transactionSettled, resolve: markTransactionSettled } = Promise.withResolvers() - let observingOwner = true - // Resume must observe its caller from BEFORE persistence I/O begins. The - // full agent lifecycle does not exist until load returns, so without this - // sentinel a never-settling backend outlives owner disposal and holds both - // public identities forever. The caller-bound effect retains the same owner - // later used by startOwned's lifecycle effect and adopts both reservation - // disposers before persistence I/O begins. - let lifecycleBoundary: (() => Promise) | undefined - let disposingForFactory: Promise | undefined - const disposeLoadForFactory = (): Promise => (disposingForFactory ??= (async () => { - markOwnerDisposed() - await transactionSettled - })()) - let untrackFactoryLoad: (() => void) | undefined - let disposeLoadSentinel: (() => Promise | void) | undefined - let loadSentinelRetired = false - const retireLoadSentinel = (): void => { - /* v8 ignore next -- every lifecycle/rollback boundary is memoized and - * invokes its after-quiescence hook once; retain idempotence defensively */ - if (loadSentinelRetired) return - // Disarm the follower before invoking its wrapper: retirement can happen - // from inside the lifecycle it used to follow, so recursing into that - // same boundary here would deadlock final teardown. - loadSentinelRetired = true - observingOwner = false - void disposeLoadSentinel?.() - } - let reservations: RegistrationReservations | undefined - let lifecycleStarted = false - try { - reservations = this.reserve(agentId, sessionId) - const ownedReservations = reservations - // Move both reservation effects under a sentinel BEFORE persistence I/O. - // Its first teardown stage either aborts/waits for the load transaction - // or follows the full lifecycle after handoff; only then do the exact - // reservation disposers run. They therefore cannot race ahead as owner - // siblings and reopen ids while load/setup/scope cleanup is still live. - disposeLoadSentinel = ownerCtx.effect(function* () { - // eslint-disable-next-line @typescript-eslint/unbound-method -- exact effect-disposer identity is the ownership contract - yield ownedReservations.agent.release - // eslint-disable-next-line @typescript-eslint/unbound-method -- exact effect-disposer identity is the ownership contract - yield ownedReservations.session.release - yield () => { - if (loadSentinelRetired) return - if (observingOwner) { - markOwnerDisposed() - return transactionSettled - } - return lifecycleBoundary?.() - } - }, `agentLoop.resumeLoad(${agentId})`) - untrackFactoryLoad = this.trackFactoryTransaction(disposeLoadForFactory) - try { - const loadTask = persistence.load(sessionId) - const { meta, events } = await Promise.race([ - loadTask, - ownerDisposed.then(() => { - throw new Error(`agent "${agentId}" resume aborted: owner disposed during persistence load`) - }), - ]) - // The backend is an async boundary too. Read each loaded header field - // once so a stateful implementation cannot pass a valid presence check - // and then substitute a different value during reconstruction. - const createdAt = meta.createdAt - const cwd = meta.cwd - const parentSession = meta.parentSession - const seedLength = meta.seedLength - // An out-of-band direct registry/session insertion can still race this - // service's reservation, so the public enter primitives re-check exact - // liveness at publication. - const session = reservations.session.prepare({ - seed: events, - meta: { - createdAt, - ...cwd !== undefined ? { cwd } : {}, - ...parentSession !== undefined ? { parentSession } : {}, - ...seedLength !== undefined ? { seedLength } : {}, - }, - }) - // startOwned synchronously returns either the complete lifecycle or a - // preparation-rollback boundary before its result reaches the first - // setup await. Retarget the lifecycle-long load sentinel to that disposer; - // ownership overlaps instead of creating a gap. - lifecycleStarted = true - const starting = this.startOwned( - ownerCtx, - agentId, - agentOptions, - session, - 'resume', - reservations, - setup, - retireLoadSentinel, - ) - lifecycleBoundary = starting.dispose - observingOwner = false - return await starting.result - } finally { - if (!lifecycleStarted) reservations.release() - } - } finally { - try { - // Manual handoff/removal must not return transactionSettled: awaiting - // that promise from inside this transaction would deadlock it. If the - // owner already triggered cleanup, this idempotent second disposal is a - // no-op and the owner's first cleanup remains parked on the shared - // settlement promise. - if (!lifecycleStarted) { - // Covers reserve succeeding but sentinel/factory tracking failing - // before the inner load transaction begins. - reservations?.release() - // A failed pre-lifecycle transaction has already released directly; - // retire the sentinel so it cannot remain as a stale owner effect. - retireLoadSentinel() - await disposeLoadSentinel?.() - } - } finally { - markTransactionSettled() - untrackFactoryLoad?.() - } - } - } - - /** Reserve both public identities in their owning registries. */ - private reserve(agentId: AgentId, sessionId: SessionId): RegistrationReservations { - const agent = this.ctx.agents.reserve(agentId) - try { - const session = this.ctx.sessions.reserve(sessionId) - return { - agent, - session, - release() { - // Both owner capabilities are independently idempotent, so the - // composite needs no second state machine of its own. - session.release() - agent.release() - }, - } - } catch (error: unknown) { - agent.release() - throw error - } - } - - /** - * Construct an unpublished agent and synchronously install its complete - * teardown skeleton before any setup await. A lifecycle-long caller sentinel and - * factory placeholder exist before driver/scope construction; the closures - * receive their session/registry/loop disposers only at publication, while - * the exact scope disposer is nested as soon as construction returns. Owner - * unload during preparation or setup therefore follows a real rollback - * boundary, flips liveness, and wins without late Cordis effect collection. - */ - private prepareLifecycle( + /** Resume through an explicit persistence handle used by the deferred config path. */ + private async resumeWith( ownerCtx: Context, - id: AgentId, - options: AgentOptions, - session: Session, - reservations: RegistrationReservations, - afterQuiescence?: () => void, - ): { - agent: ReactLoopAgent - active: () => boolean - deactivated: Promise - publish: (source: SessionStartSource) => void - disposeAgent: () => Promise - } { - // When creation is invoked through an agent scope (subagents), the owner - // agent's disposed status flips synchronously at handle teardown—earlier - // than Cordis reaches nested scope effects. Include that signal in the - // pre-publication liveness check so a same-turn parent dispose cannot race - // an already-fulfilled setup promise into briefly publishing a child. - let ownerAgent: Context['agent'] - let ownerFiber: Context['fiber'] - try { - this.assertFactoryActive() - ownerCtx.fiber.assertActive() - ownerAgent = ownerCtx.agent - ownerFiber = ownerCtx.fiber - } catch (error: unknown) { - reservations.release() - afterQuiescence?.() - const dispose = (): Promise => Promise.resolve() - throw new LifecyclePreparationFailure(error, dispose) - } - - // Establish BOTH ownership edges before driver preparation or scope - // minting can publish an internal lifecycle notification. The lifecycle-long - // caller sentinel also adopts the exact reservation effects: owner unload - // first waits for the memoized lifecycle boundary, then reaches those - // capabilities, so IDs cannot reopen while scope cleanup is still live. - const { promise: lifecycleReady, resolve: markLifecycleReady } - = Promise.withResolvers<() => Promise>() - const { promise: deactivated, resolve: markDeactivated } = Promise.withResolvers() - let ownerDisposed = false - const ownerIsDisposed = (): boolean => ownerDisposed - let ownerSentinelRetired = false - let disposeOwnerSentinel: () => Promise | void - try { - disposeOwnerSentinel = ownerCtx.effect(function* () { - // eslint-disable-next-line @typescript-eslint/unbound-method -- exact effect-disposer identity is the ownership contract - yield reservations.agent.release - // eslint-disable-next-line @typescript-eslint/unbound-method -- exact effect-disposer identity is the ownership contract - yield reservations.session.release - yield () => { - if (ownerSentinelRetired) return - ownerDisposed = true - markDeactivated() - return lifecycleReady.then(disposeLifecycle => disposeLifecycle()) - } - }, `agentLoop.ownerLifecycle(${id})`) - } catch (error: unknown) { - reservations.release() - afterQuiescence?.() - const dispose = (): Promise => Promise.resolve() - markLifecycleReady(dispose) - // The only callback-free effect-install failure is Cordis's inactive - // owner boundary; preserve the original value as cause for diagnostics. - const reportedError = new Error(`agent "${id}" setup aborted: owner disposed during setup`, { cause: error }) - throw new LifecyclePreparationFailure(reportedError, dispose) - } - let disposingForFactory: Promise | undefined - const disposeForFactory = (): Promise => (disposingForFactory ??= (async () => { - const disposeLifecycle = await lifecycleReady - await disposeLifecycle() - })()) - let untrackFactory: () => void - try { - untrackFactory = this.trackFactoryTransaction(disposeForFactory) - } catch (error: unknown) { - /* v8 ignore start -- no callback boundary exists between the active - * factory check, sentinel installation, and this synchronous ledger insert */ - let cleanupTask: Promise | undefined - const cleanup = (): Promise => (cleanupTask ??= Promise.resolve().then(() => { - reservations.release() - afterQuiescence?.() - })) - markLifecycleReady(cleanup) - void disposeOwnerSentinel() - throw new LifecyclePreparationFailure(error, cleanup) - /* v8 ignore stop */ - } - - let scope: Scope | undefined - let stopPrepared: (() => Promise | void) | undefined - let disposeAgent: (() => Promise) | undefined - try { - const driver = prepareReactLoopAgent(this.ctx, id, options, session) - stopPrepared = () => driver.dispose() - const { agent } = driver - scope = createScope(this.ctx, agent) - const lifecycleScope = scope - if (ownerIsDisposed() || !this.factoryIsActive() - || ownerFiber.state === FiberState.UNLOADING - || ownerFiber.state === FiberState.DISPOSED - || ownerFiber.state === FiberState.FAILED - || ownerAgent?.status === 'disposed') { - throw new Error(`agent "${id}" setup aborted: owner disposed during setup`) - } - bindReactLoopAgentContext(agent, lifecycleScope.ctx.extend({ agent })) - - let active = true - let detachSession: (() => void) | undefined - let detachAgent: (() => void) | undefined - const stop = stopPrepared - const { promise: torndown, resolve: markTorndown } = Promise.withResolvers() - const { promise: publicationSettled, resolve: markPublicationSettled } = Promise.withResolvers() - let publishing = false - - const dispose = ownerCtx.effect(function* () { - // First yielded, disposed last: every preceding teardown stage settled. - yield () => { - // Reservation ownership is part of lifecycle settlement: a factory - // unload that awaited this disposer may reuse both ids immediately. - reservations.release() - // Retire both follower effects only after quiescence reached this final - // stage. Their retired branches skip recursively disposing this same - // lifecycle while their exact reservation children are already inert. - ownerSentinelRetired = true - void disposeOwnerSentinel() - afterQuiescence?.() - untrackFactory() - markTorndown() - } - // Exact identity moves the scope fiber out of the owner's concurrent - // sibling list and into this ordered transaction. - yield lifecycleScope.rawDispose - yield () => { - detachSession?.() - detachSession = undefined - } - yield () => { - detachAgent?.() - detachAgent = undefined - } - // Last yielded, disposed first. Keep the pre-publication path - // synchronous: returning a Promise only after the loop actually began - // lets a failed announcement roll back registry/store before create's - // rejection is observed. - yield () => { - active = false - markDeactivated() - // A listener can begin owner teardown reentrantly. Flip liveness now - // so publish's next checkpoint aborts, but keep both registry entries - // and the scope intact until the current synchronous publication - // phase has unwound. - if (publishing) return publicationSettled.then(stop) - return stop() - } - }, 'agentLoop.lifecycle()') - - let disposing: Promise | undefined - disposeAgent = (): Promise => (disposing ??= (async () => { - await dispose() - await torndown - })()) - markLifecycleReady(disposeAgent) - - const isActive = (): boolean => active - && !ownerIsDisposed() - && this.factoryIsActive() - && ownerFiber.state !== FiberState.UNLOADING - && ownerFiber.state !== FiberState.DISPOSED - && ownerFiber.state !== FiberState.FAILED - && ownerAgent?.status !== 'disposed' - - const publish = (source: SessionStartSource): void => { - publishing = true - try { - /* v8 ignore next 3 -- both callers check active immediately before - * this callback-free synchronous publish entry */ - if (!isActive()) { - throw new Error(`agent "${id}" setup aborted: owner disposed during setup`) - } - // Publication is one synchronous, rollback-covered sequence. Setup has - // already completed, so its scoped listeners observe both announcements. - detachSession = agent.ctx.sessions.enter(session, reservations.session) - detachAgent = this.ctx.agents.enter(agent, reservations.agent) - // Both enter() calls capture stable dispatch carriers and therefore - // evaluate a caller-owned Context.filter. A getter can begin teardown; - // entries exist for rollback, but no creation edge may escape afterward. - if (!isActive()) { - throw new Error(`agent "${id}" setup aborted: owner disposed during setup`) - } - this.ctx.sessions.announce(session) - // Session listeners can dispose an owner. Finish that dispatch while - // both entries/scope remain live, then skip the agent edge entirely. - if (!isActive()) { - throw new Error(`agent "${id}" setup aborted: owner disposed during setup`) - } - this.ctx.agents.announce(agent) - // Creation listeners may synchronously dispose either owner. Cordis - // flips the relevant fiber state before it invokes nested effects, so - // re-check here and never unlock a driver after teardown began. - if (!isActive()) { - throw new Error(`agent "${id}" setup aborted: owner disposed during setup`) - } - // Setup is over and both entries are live. Open the driving surface just - // before session-start so its listeners retain their supported ability to - // inject/queue, while setup itself can never drive an unpublished agent. - driver.enableDrive() - agentEvents(this.ctx, agent).emit('agent/session-start', source) - // session-start is the final synchronous listener boundary before the - // loop begins. Teardown there must win just like teardown from either - // creation announcement; the prebuilt driver disposer makes rollback - // quiescent even though the loop never started. - if (!isActive()) { - throw new Error(`agent "${id}" setup aborted: owner disposed during setup`) - } - driver.startDriver() - } finally { - publishing = false - markPublicationSettled() - } - } - - return { - agent, - active: isActive, - deactivated, - publish, - disposeAgent, - } - } catch (error: unknown) { - // Preparation failed before startOwned received a lifecycle object. Give - // a factory unload that already captured the placeholder a real boundary, - // and retire the entry only after the minted scope (if any) is quiescent. - const failedScope = scope - const ownershipInactive = ownerIsDisposed() || ownerFiber.uid === null || !this.factoryIsActive() - || ownerFiber.state === FiberState.UNLOADING - || ownerFiber.state === FiberState.DISPOSED - || ownerFiber.state === FiberState.FAILED - || ownerAgent?.status === 'disposed' - const reportedError = ownershipInactive && error instanceof CordisError - ? new Error(`agent "${id}" setup aborted: owner disposed during setup`, { cause: error }) - : error - let fallbackTask: Promise | undefined - const cleanup = disposeAgent ?? (() => (fallbackTask ??= (async () => { - try { - await stopPrepared?.() - } finally { - try { - await failedScope?.dispose() - } finally { - // Factory and caller quiescence include the prepared driver, - // minted scope, and both unpublished identities even when the - // complete lifecycle effect could not be installed. - reservations.release() - afterQuiescence?.() - } - } - })())) - markLifecycleReady(cleanup) - const cleanupTask = cleanup() - // Retire the provisional owner edge. If owner unload already claimed it, - // this is an inert repeat and that first caller is following cleanupTask. - void disposeOwnerSentinel() - void cleanupTask.then( - untrackFactory, - /* v8 ignore next -- Scope.dispose is specified to contain child - * failures; preserve diagnostics if that lower-level contract breaks */ - (cleanupError: unknown) => { - untrackFactory() - try { - this.ctx.logger.error(new AggregateError([reportedError, cleanupError], 'agent lifecycle preparation and rollback failed')) - } catch { - // Only a logger-export failure is swallowed: the original - // preparation error is already propagating to the caller. - } - }, - ) - throw new LifecyclePreparationFailure(reportedError, cleanup) - } - } - - /** Publish a no-setup config agent synchronously. */ - private start( - id: AgentId, - options: AgentOptions, - session: Session, - source: SessionStartSource, - reservations: RegistrationReservations, - ): { agent: ReactLoopAgent; disposeAgent: () => Promise } { - let lifecycle: ReturnType - try { - lifecycle = this.prepareLifecycle(this.ctx, id, options, session, reservations) - } catch (error: unknown) { - /* v8 ignore next -- prepareLifecycle converts every failure into its - * rollback-bearing internal error before crossing this boundary */ - if (!(error instanceof LifecyclePreparationFailure)) throw error - void error.dispose() - throw error.reason - } - try { - lifecycle.publish(source) - return { agent: lifecycle.agent, disposeAgent: lifecycle.disposeAgent } - } catch (error: unknown) { - void lifecycle.disposeAgent() - throw error - } - } - - /** - * Build an {@link AgentHandle} for a PREPARED session + a fresh agent. The - * handle's `dispose()` runs the composite effect's disposer (see - * {@link start}) — which stops the loop, awaits its exit and outstanding - * idle-injection flushes, unregisters the agent, detaches the session, - * unwinds the scope, and releases both ids, in that order. Caller-fiber unload - * also invokes an independent sentinel that follows this memoized boundary, - * so handle-first and owner-first races honor the same ordering. - * - * `dispose()` is MEMOIZED: the underlying cordis effect disposer is - * single-shot (a second call returns immediately because the effect's epoch is - * already cleared, NOT awaiting the in-flight teardown), so concurrent/repeated - * `dispose()` calls would otherwise resolve before the first call's - * loop + flush quiescence boundary completed. Memoizing the promise makes - * every caller observe that SAME boundary, honoring the - * `AgentHandle.dispose(): Promise` contract (mirrors the ACP `quiesce()` - * helper). - */ - private startOwned( - ownerCtx: Context, - id: AgentId, options: AgentOptions, session: Session, source: SessionStartSource, - reservations: RegistrationReservations, - setup?: (agentCtx: Context) => Promise | void, - afterQuiescence?: () => void, - ): OwnedAgentStart { - let lifecycle: ReturnType - try { - lifecycle = this.prepareLifecycle(ownerCtx, id, options, session, reservations, afterQuiescence) - } catch (error: unknown) { - /* v8 ignore next 1 -- prepareLifecycle wraps every synchronous failure */ - if (!(error instanceof LifecyclePreparationFailure)) throw error - return { - dispose: error.dispose, - result: (async () => { - await error.dispose() - throw error.reason - })(), - } - } - return { - dispose: lifecycle.disposeAgent, - result: this.finishOwnedStart(lifecycle, id, source, setup), - } - } - - /** Await setup and publish after {@link startOwned} established ownership synchronously. */ - private async finishOwnedStart( - lifecycle: ReturnType, - id: AgentId, - source: SessionStartSource, - setup?: (agentCtx: Context) => Promise | void, + persistence: SessionPersistence, + options: ResumeAgentOptions, ): Promise { + const transaction = new AgentCreationTransaction( + this.runtime.ctx, + ownerCtx, + this.ownership, + options.agentId, + options.signal, + ) try { - // Scope minting emits Cordis's synchronous internal/plugin notification. - // A listener can unload either owner there; never run arbitrary setup in - // the already-doomed scope while the tracked disposer is catching up. - /* v8 ignore next 3 -- prepareLifecycle returns success only after its - * final synchronous liveness check; no callback runs before this line */ - if (!lifecycle.active()) { - throw new Error(`agent "${id}" setup aborted: owner disposed during setup`) - } - // The owner-disposal branch makes a never-settling setup unable to hold - // the transaction or its ID reservations forever. Promise.race installs - // rejection observation on setup even if owner disposal wins first. - const setupTask = Promise.resolve(setup?.(lifecycle.agent.ctx)) - await Promise.race([ - setupTask, - lifecycle.deactivated.then(() => { - throw new Error(`agent "${id}" setup aborted: owner disposed during setup`) - }), - ]) - // Cordis begins a fiber unload synchronously but invokes nested effect - // disposers from its next microtask. Give that already-started unload one - // checkpoint to deactivate this lifecycle before publication; otherwise - // an immediately fulfilled setup continuation can outrun its owner's - // same-turn dispose and briefly publish an already-doomed child. - await Promise.resolve() - if (!lifecycle.active()) { - throw new Error(`agent "${id}" setup aborted: owner disposed during setup`) - } - lifecycle.publish(source) - return { agent: lifecycle.agent, dispose: lifecycle.disposeAgent } + const loaded = await transaction.waitFor(persistence.load(options.resumeSessionId)) + transaction.assertActive() + const session = this.runtime.ctx.sessions.prepare(options.resumeSessionId, { + seed: loaded.events, + meta: { + createdAt: loaded.meta.createdAt, + ...loaded.meta.cwd === undefined ? {} : { cwd: loaded.meta.cwd }, + ...loaded.meta.parentSession === undefined ? {} : { parentSession: loaded.meta.parentSession }, + ...loaded.meta.seedLength === undefined ? {} : { seedLength: loaded.meta.seedLength }, + }, + }) + const agent = transaction.prepare(options.agentOptions ?? {}, session) + await transaction.waitFor(options.setup?.(agent.ctx)) + transaction.assertActive() + return transaction.publish('resume') } catch (error: unknown) { - await lifecycle.disposeAgent() + await transaction.dispose(error instanceof Error ? error : new Error(String(error))) throw error + } finally { + transaction.finishWrapper() } } } diff --git a/packages/core/agent-loop/src/loop.ts b/packages/core/agent-loop/src/loop.ts index 982fa02cab..7a7ca2e28c 100644 --- a/packages/core/agent-loop/src/loop.ts +++ b/packages/core/agent-loop/src/loop.ts @@ -11,7 +11,7 @@ import type { Context } from 'cordis' import type { FinishReason, GenerateOptions, LlmCallConfig, Message } from '@deepseek-ai/dsh-llm' import { BlockAssembler, HarnessError, deepFreeze } from '@deepseek-ai/dsh-llm' import { agentEvents, assembleContextFor } from '@deepseek-ai/dsh-agent' -import type { AgentEventDispatch, ContinuationDecision, ContinuationStop, HookContext, PromptDecision } from '@deepseek-ai/dsh-agent' +import type { AgentEventDispatch, ContinuationDecision, HookContext, PromptDecision } from '@deepseek-ai/dsh-agent' import { canonicalHeader } from '@deepseek-ai/dsh-session' import type { Session, TurnEndReason, TurnTrigger } from '@deepseek-ai/dsh-session' import { createTransmissionLog, recordRequestHeader } from './request-log.ts' @@ -36,20 +36,6 @@ function toError(error: unknown): CodedError { return error instanceof Error ? error : new HarnessError(String(error), 'UNKNOWN', { cause: error }) } -/** - * Validate the runtime result of the terminal-stop serial event. Event types - * protect TypeScript listeners, but JavaScript and casts can still return an - * arbitrary bail value; accepting one as an implicit stop would hide a broken - * policy plugin. - */ -function assertContinuationStop(value: unknown): asserts value is ContinuationStop | undefined { - if (value === undefined) return - const candidate = Object(value) as { action?: unknown } - if (candidate.action !== 'stop') { - throw new Error('agent/turn-stop returned an invalid result; expected { action: \'stop\' } or undefined') - } -} - /** * Map a model-call {@link FinishReason} to the step error it should raise, or * `undefined` when the step completed normally. @@ -644,8 +630,7 @@ async function runTurn( // no later listener or steering override can resurrect the turn. let terminalStop = false try { - const stop = await events.strictSerial('agent/turn-stop', turn) - assertContinuationStop(stop) + const stop = await events.serial('agent/turn-stop', turn) terminalStop = stop !== undefined } catch (error: unknown) { // A broken terminal policy is an ordinary continuation failure: fail diff --git a/packages/core/agent-loop/tests/agent.spec.ts b/packages/core/agent-loop/tests/agent.spec.ts index d132d862a8..7f65bbc5f3 100644 --- a/packages/core/agent-loop/tests/agent.spec.ts +++ b/packages/core/agent-loop/tests/agent.spec.ts @@ -49,30 +49,29 @@ function send(agent: ReactLoopAgent, text: string) { } describe('ReactLoopAgent', () => { - it('owns immutable runtime bindings for id, options, session, and scoped context', async () => { + it('rejects access before context binding and a second driver for one session', async () => { + const ctx = new Context() + await ctx.plugin(SessionStore) + const session = ctx.sessions.create(SessionId('exclusive-driver')) + const prepared = prepareReactLoopAgent(ctx, AgentId('first-driver'), { model: 'mock' }, session) + + expect(() => prepared.agent.ctx).toThrow('context is not bound') + expect(() => prepareReactLoopAgent(ctx, AgentId('second-driver'), { model: 'mock' }, session)) + .toThrow('already has a concrete agent driver') + + await prepared.dispose() + await ctx.fiber.dispose() + }) + + it('borrows caller options and binds its scoped context exactly once', async () => { const ctx = await harness(new MockAdapter([textResponse('unused')])) const options = { model: 'mock' } const agent = ctx.agentLoop.create(AgentId('owned-bindings'), options) - const acceptedSession = agent.session - const acceptedContext = agent.ctx - options.model = 'caller-mutated' - expect(agent.options).toEqual({ model: 'mock' }) - expect(Object.isFrozen(agent.options)).toBe(true) - expect(Reflect.set(agent, 'id', AgentId('redirected'))).toBe(false) - expect(Reflect.set(agent, 'options', { model: 'other' })).toBe(false) - expect(Reflect.set(agent, 'session', ctx.sessions.create(SessionId('other')))).toBe(false) - expect(Reflect.set(agent, 'ctx', new Context())).toBe(false) + expect(agent.options).toBe(options) expect(agent.id).toBe('owned-bindings') - expect(agent.session).toBe(acceptedSession) - expect(agent.ctx).toBe(acceptedContext) + expect(agent.session.id).toMatch(/^owned-bindings-session-/) expect(() => { bindReactLoopAgentContext(agent, new Context()) }).toThrow(/context is already bound/) - for (const name of ['id', 'options', 'session', 'ctx']) { - expect(Object.getOwnPropertyDescriptor(agent, name)).toMatchObject({ - configurable: false, - writable: false, - }) - } await ctx.fiber.dispose() }) @@ -223,23 +222,6 @@ describe('ReactLoopAgent', () => { warn.mockRestore() }) - it('idle inject() safely renders a hostile non-Error flush failure', async () => { - const ctx = await harness(new MockAdapter([textResponse('ok')])) - const hostile = { [Symbol.toPrimitive]() { throw new Error('no coercion') } } - ctx.on('session/flush', () => { throw hostile }) - const warn = vi.spyOn(ctx.logger, 'warn').mockImplementation(() => undefined) - const agent = ctx.agentLoop.create(AgentId('hostile-flush'), { model: 'mock' }) - const errors: string[] = [] - ctx.on('agent/error', (_a, _turn, _step, error) => void errors.push(error.message)) - - agent.inject([{ type: 'text', text: 'notice' }]) - await new Promise(r => setTimeout(r, 20)) - - expect(errors).toEqual(['']) - expect(warn).toHaveBeenCalledWith(expect.stringContaining('')) - warn.mockRestore() - }) - it('idle inject() with a non-serializable source opens no turn (nothing to close)', async () => { const adapter = new MockAdapter([textResponse('ok')]) const ctx = await harness(adapter) @@ -281,7 +263,7 @@ describe('ReactLoopAgent', () => { // Start the loop to get the disposer; the agent waits for messages // (idle, never-resolving cancel), so it will stay idle. - prepared.enableDrive() + prepared.markPublished() const dispose = prepared.startDriver() // First dispose @@ -309,24 +291,6 @@ describe('ReactLoopAgent', () => { await ctx.fiber.dispose() }) - it('does not claim a session when concrete-agent construction rejects options', async () => { - const ctx = new Context() - await ctx.plugin(SessionStore) - const session = ctx.sessions.create(SessionId('constructor-retry')) - const badOptions = { - get model(): string { - throw new Error('bad model getter') - }, - } - - expect(() => prepareReactLoopAgent(ctx, AgentId('bad-constructor'), badOptions, session)) - .toThrow('bad model getter') - const prepared = prepareReactLoopAgent(ctx, AgentId('constructor-retry'), { model: 'mock' }, session) - await prepared.dispose() - expect(prepared.agent.status).toBe('disposed') - await ctx.fiber.dispose() - }) - it('setting the same status does not emit agent/status again', async () => { const adapter = new MockAdapter([textResponse('ok')]) const ctx = await harness(adapter) @@ -417,7 +381,7 @@ describe('ReactLoopAgent', () => { const session = ctx.sessions.create(SessionId('bare')) const prepared = prepareReactLoopAgent(ctx, AgentId('bare'), { model: 'mock' }, session) const { agent } = prepared - prepared.enableDrive() + prepared.markPublished() const dispose = prepared.startDriver() agent.send([{ type: 'text', text: 'go' }]) await new Promise(r => setTimeout(r, 30)) diff --git a/packages/core/agent-loop/tests/loop.spec.ts b/packages/core/agent-loop/tests/loop.spec.ts index f1c3e9af9d..71be5b339e 100644 --- a/packages/core/agent-loop/tests/loop.spec.ts +++ b/packages/core/agent-loop/tests/loop.spec.ts @@ -272,7 +272,7 @@ describe('agent loop', () => { expect(result.data.meta).toBeUndefined() expect(result.data.content).toEqual([{ type: 'text', - text: 'Error: tools/execute must return a losslessly JSON-serializable ToolExecutionResult', + text: 'Error: tool result must be losslessly JSON-serializable', }]) } // The normalized failure was durably logged and fed back to the model; the diff --git a/packages/core/agent-loop/tests/resume.spec.ts b/packages/core/agent-loop/tests/resume.spec.ts index d739ff0ed3..93605008ec 100644 --- a/packages/core/agent-loop/tests/resume.spec.ts +++ b/packages/core/agent-loop/tests/resume.spec.ts @@ -5,7 +5,7 @@ import { tmpdir } from 'node:os' import { join } from 'node:path' import LlmService from '@deepseek-ai/dsh-llm' import SessionStore, { Session, SessionId } from '@deepseek-ai/dsh-session' -import type { SessionEvent, SessionHeader } from '@deepseek-ai/dsh-session' +import type { SessionEvent } from '@deepseek-ai/dsh-session' import SystemPrompt from '@deepseek-ai/dsh-system-prompt' import ToolRegistry from '@deepseek-ai/dsh-tools' import AgentRegistry, { AgentId } from '@deepseek-ai/dsh-agent' @@ -69,7 +69,29 @@ async function promptly(task: Promise): Promise { } } +/** Throw an arbitrary callback value to exercise the public unknown-error boundary. */ +function throwUnknown(value: unknown): never { + throw value +} + describe('the session-persistence RFC: AgentLoop factory create/resume', () => { + it('normalizes a non-Error resume publication failure for rollback and rethrows it', async () => { + const sessionId = SessionId('unknown-resume-failure-s') + const root = await persistSession(sessionId) + const ctx = await mountPersistentHarness(root, new MockAdapter([textResponse('next')])) + const failure = { source: 'resume' } + ctx.on('session/created', () => throwUnknown(failure)) + + await expect(ctx.agents.resume({ + agentId: AgentId('unknown-resume-failure'), + resumeSessionId: sessionId, + })).rejects.toBe(failure) + + expect(ctx.agents.get(AgentId('unknown-resume-failure'))).toBeUndefined() + expect(ctx.sessions.get(sessionId)).toBeUndefined() + await ctx.fiber.dispose() + }) + it('createAgent uses the caller-supplied sessionId (not ${id}-session)', async () => { const adapter = new MockAdapter([textResponse('hi')]) const { ctx } = await persistentHarness(adapter) @@ -99,22 +121,6 @@ describe('the session-persistence RFC: AgentLoop factory create/resume', () => { await ctx.fiber.dispose() }) - it('createAgent sends raw metadata to the session validator before cloning can sanitize it', async () => { - class ExoticMeta { - readonly cwd = '/accepted' - } - const { ctx } = await persistentHarness(new MockAdapter([textResponse('hi')])) - - await expect(ctx.agents.create({ - agentId: AgentId('exotic-meta-agent'), - sessionId: SessionId('exotic-meta-session'), - meta: new ExoticMeta(), - })).rejects.toThrow(/session metadata is not a plain JSON record/) - expect(ctx.agents.get(AgentId('exotic-meta-agent'))).toBeUndefined() - expect(ctx.sessions.get(SessionId('exotic-meta-session'))).toBeUndefined() - await ctx.fiber.dispose() - }) - it('resume of a session with no cwd carries an undefined cwd header', async () => { // Lifecycle 1: create a no-cwd session and run a turn. const adapter1 = new MockAdapter([textResponse('a')]) @@ -184,7 +190,7 @@ describe('the session-persistence RFC: AgentLoop factory create/resume', () => { order.push('session/created') }) ctx.on('agent/created', (agent) => { - expect(() => { agent.cancel('too early') }).toThrow(/cannot cancel before creation setup completes/) + expect(agent.status).toBe('idle') order.push('agent/created') }) ctx.on('agent/session-start', (agent) => { @@ -228,9 +234,9 @@ describe('the session-persistence RFC: AgentLoop factory create/resume', () => { await ctx.fiber.dispose() }) - it('successful resume disposal retires both caller ownership sentinels', async () => { - const sessionId = SessionId('resume-retired-sentinels-s') - const agentId = AgentId('resume-retired-sentinels') + it('successful resume disposal retires its caller-owned transaction effects', async () => { + const sessionId = SessionId('resume-retired-effects-s') + const agentId = AgentId('resume-retired-effects') const root = await persistSession(sessionId) const ctx = await mountPersistentHarness(root, new MockAdapter([textResponse('next')])) const handle = await ctx.agents.resume({ @@ -238,14 +244,14 @@ describe('the session-persistence RFC: AgentLoop factory create/resume', () => { resumeSessionId: sessionId, agentOptions: { model: 'mock' }, }) - const sentinelLabels = [ - `agentLoop.resumeLoad(${agentId})`, - `agentLoop.ownerLifecycle(${agentId})`, + const transactionLabels = [ + `agentLoop.owner(${agentId})`, + `agentLoop.lifecycle(${agentId})`, ] - expect(ctx.fiber.getEffects().map(effect => effect.label)).toEqual(expect.arrayContaining(sentinelLabels)) + expect(ctx.fiber.getEffects().map(effect => effect.label)).toEqual(expect.arrayContaining(transactionLabels)) await handle.dispose() - expect(ctx.fiber.getEffects().filter(effect => sentinelLabels.includes(effect.label))).toEqual([]) + expect(ctx.fiber.getEffects().filter(effect => transactionLabels.includes(effect.label))).toEqual([]) await ctx.fiber.dispose() }) @@ -346,14 +352,14 @@ describe('the session-persistence RFC: AgentLoop factory create/resume', () => { }, { inject: ['agents'] })) await loadStarted.promise - const rejection = expect(promptly(resuming)).rejects.toThrow(/owner disposed during persistence load/) + const rejection = expect(promptly(resuming)).rejects.toThrow(/owner disposed during setup/) await promptly(owner.dispose()) expect(published).toEqual([]) expect(ctx.agents.get(agentId)).toBeUndefined() expect(ctx.sessions.get(sessionId)).toBeUndefined() - // owner.dispose() itself awaited transaction settlement and reservation - // release: reuse the same identities BEFORE awaiting the resume rejection. + // owner.dispose() awaited transaction settlement, so the same identities + // can be reused before awaiting the public rejection. const retry = await promptly(ctx.agents.resume({ agentId, resumeSessionId: sessionId, agentOptions: { model: 'mock' } })) await rejection expect(loads).toBe(2) @@ -372,7 +378,7 @@ describe('the session-persistence RFC: AgentLoop factory create/resume', () => { await ctx.fiber.dispose() }) - it('AgentLoop unload aborts persistence load and awaits reservation release', async () => { + it('AgentLoop unload aborts persistence load and awaits wrapper settlement', async () => { const sessionId = SessionId('resume-load-factory-unload') const agentId = AgentId('resume-load-factory-race') const root = await persistSession(sessionId) @@ -400,18 +406,13 @@ describe('the session-persistence RFC: AgentLoop factory create/resume', () => { const resuming = ctx.agents.resume({ agentId, resumeSessionId: sessionId, agentOptions: { model: 'mock' } }) await loadStarted.promise - const rejection = expect(promptly(resuming)).rejects.toThrow(/owner disposed during persistence load/) + const rejection = expect(promptly(resuming)).rejects.toThrow(/agent loop is not active/) await promptly(loopFiber.dispose()) await rejection expect(published).toEqual([]) expect(ctx.agents.get(agentId)).toBeUndefined() expect(ctx.sessions.get(sessionId)).toBeUndefined() - const agentReservation = ctx.agents.reserve(agentId) - const sessionReservation = ctx.sessions.reserve(sessionId) - sessionReservation.release() - agentReservation.release() - lateLoad.resolve(structuredClone(snapshot)) await Promise.resolve() await Promise.resolve() @@ -419,83 +420,6 @@ describe('the session-persistence RFC: AgentLoop factory create/resume', () => { await ctx.fiber.dispose() }) - it('option snapshot reentrancy cannot install a resume sentinel after factory unload begins', async () => { - const sessionId = SessionId('resume-snapshot-factory-unload') - const agentId = AgentId('resume-snapshot-factory-race') - const root = await persistSession(sessionId) - const ctx = new Context() - await ctx.plugin(LlmService) - await ctx.plugin(SessionStore) - await ctx.plugin(SystemPrompt) - await ctx.plugin(ToolRegistry) - await ctx.plugin(AgentRegistry) - const loopFiber = await ctx.plugin(AgentLoop, { agents: [] }) - await ctx.plugin(SessionPersistenceJsonl, { root }) - ctx.llm.registerAdapter(['mock'], new MockAdapter([textResponse('next')])) - - let loads = 0 - const load = ctx.sessionPersistence.load.bind(ctx.sessionPersistence) - ctx.sessionPersistence.load = (id) => { - loads += 1 - return load(id) - } - const options = { - agentId, - resumeSessionId: sessionId, - get agentOptions() { - void loopFiber.dispose() - return { model: 'mock' } - }, - } - - await expect(ctx.agents.resume(options)).rejects.toThrow('agent loop is not active') - await loopFiber.dispose() - expect(loads).toBe(0) - expect(ctx.fiber.getEffects().filter(effect => effect.label === `agentLoop.resumeLoad(${agentId})`)).toEqual([]) - const agentReservation = ctx.agents.reserve(agentId) - const sessionReservation = ctx.sessions.reserve(sessionId) - sessionReservation.release() - agentReservation.release() - await ctx.fiber.dispose() - }) - - it('snapshots resume identities and agent options before persistence load', async () => { - const sessionId = SessionId('resume-snapshot-source') - const root = await persistSession(sessionId) - const ctx = await mountPersistentHarness(root, new MockAdapter([textResponse('next')])) - const loaded = await ctx.sessionPersistence.load(sessionId) - const loadGate = Promise.withResolvers() - ctx.sessionPersistence.load = () => loadGate.promise - - const occupied = await ctx.agents.create({ - agentId: AgentId('occupied-agent'), - sessionId: SessionId('occupied-session'), - agentOptions: { model: 'mock' }, - }) - const options = { - agentId: AgentId('accepted-agent'), - resumeSessionId: sessionId, - agentOptions: { model: 'mock' }, - } - const resuming = ctx.agents.resume(options) - - options.agentId = AgentId('occupied-agent') - options.resumeSessionId = SessionId('occupied-session') - options.agentOptions.model = 'mutated-model' - loadGate.resolve(structuredClone(loaded)) - - const resumed = await resuming - expect(resumed.agent.id).toBe(AgentId('accepted-agent')) - expect(resumed.agent.session.id).toBe(sessionId) - expect(resumed.agent.options.model).toBe('mock') - expect(ctx.agents.get(AgentId('occupied-agent'))).toBe(occupied.agent) - expect(ctx.sessions.get(SessionId('occupied-session'))).toBe(occupied.agent.session) - - await resumed.dispose() - await occupied.dispose() - await ctx.fiber.dispose() - }) - it('resume of a forked session preserves the parentSession lineage and seed boundary in the header', async () => { // Lifecycle 1: persist a FORKED session (carries parentSession + seedLength // in its header) by creating it with a complete-turn seed — the write path @@ -535,54 +459,6 @@ describe('the session-persistence RFC: AgentLoop factory create/resume', () => { await ctx2.fiber.dispose() }) - it('reads each loaded metadata field once before reconstructing a resumed session', async () => { - const sessionId = SessionId('resume-loaded-meta-once') - const root = await persistSession(sessionId) - const ctx = await mountPersistentHarness(root, new MockAdapter([textResponse('next')])) - const loaded = await ctx.sessionPersistence.load(sessionId) - const reads = { createdAt: 0, cwd: 0, parentSession: 0, seedLength: 0 } - const meta = Object.defineProperties({ - version: loaded.meta.version, - id: loaded.meta.id, - }, { - createdAt: { - enumerable: true, - get: () => { reads.createdAt += 1; return reads.createdAt === 1 ? loaded.meta.createdAt : 1n }, - }, - cwd: { - enumerable: true, - get: () => { reads.cwd += 1; return reads.cwd === 1 ? '/loaded' : 'relative' }, - }, - parentSession: { - enumerable: true, - get: () => { reads.parentSession += 1; return reads.parentSession === 1 ? SessionId('parent') : 1n }, - }, - seedLength: { - enumerable: true, - get: () => { reads.seedLength += 1; return reads.seedLength === 1 ? 0 : 1n }, - }, - }) as unknown as SessionHeader - ctx.sessionPersistence.load = () => Promise.resolve({ meta, events: loaded.events }) - - const resumed = await ctx.agents.resume({ - agentId: AgentId('resume-loaded-meta-once'), - resumeSessionId: sessionId, - agentOptions: { model: 'mock' }, - }) - - expect(reads).toEqual({ createdAt: 1, cwd: 1, parentSession: 1, seedLength: 1 }) - expect(resumed.agent.session.header).toEqual({ - version: loaded.meta.version, - id: sessionId, - createdAt: loaded.meta.createdAt, - cwd: '/loaded', - parentSession: 'parent', - seedLength: 0, - }) - await resumed.dispose() - await ctx.fiber.dispose() - }) - it('an idle inject() is flushed durably on its own (survives without explicit flush/dispose)', async () => { // Lifecycle 1: run a turn, then inject context while idle. The idle inject // wraps its context/message in a one-shot turn AND checkpoints it (the turn-enclosure RFC) diff --git a/packages/core/agent-loop/tests/review-fixes.spec.ts b/packages/core/agent-loop/tests/review-fixes.spec.ts index d4742da3e6..5898a82ee2 100644 --- a/packages/core/agent-loop/tests/review-fixes.spec.ts +++ b/packages/core/agent-loop/tests/review-fixes.spec.ts @@ -443,14 +443,12 @@ describe('MEDIUM: misc registry and config fixes', () => { const source = { kind: 'plugin' as const, plugin: 'accepted-source' } let notifiedContent: ContentBlock[] | undefined let notifiedSource: MessageSource | undefined - let notifiedInfoFrozen = false ctx.on('agent/queued', (subject, acceptedContent, info) => { if (subject !== agent || info.steering) return // Retain the exact notification references: cloning here would test the // listener's copy rather than the event/inbox ownership boundary. notifiedContent = acceptedContent notifiedSource = info.source - notifiedInfoFrozen = Object.isFrozen(info) }) agent.send(content, { source }) @@ -463,7 +461,6 @@ describe('MEDIUM: misc registry and config fixes', () => { expect(Object.isFrozen(notifiedContent)).toBe(true) expect(Object.isFrozen(notifiedContent?.[0])).toBe(true) expect(Object.isFrozen(notifiedSource)).toBe(true) - expect(notifiedInfoFrozen).toBe(true) const recorded = agent.session.events.flatMap(event => event.type === 'user/message' ? [event.data] : []) expect(recorded).toContainEqual({ content: [{ type: 'text', text: 'accepted-send' }], @@ -474,33 +471,6 @@ describe('MEDIUM: misc registry and config fixes', () => { expect(request).not.toContain('caller-mutated-send') }) - it('send() rechecks disposal after materializing caller getters', async () => { - const adapter = new MockAdapter([textResponse('unused')]) - const ctx = await harness(adapter) - const handle = await ctx.agents.create({ - agentId: AgentId('reentrant-send-dispose'), - sessionId: SessionId('reentrant-send-dispose-session'), - agentOptions: { model: 'mock' }, - }) - const { agent } = handle - let queued = 0 - ctx.on('agent/queued', subject => void (queued += Number(subject === agent))) - const content = [{ - type: 'text' as const, - get text() { - void handle.dispose() - return 'accepted-after-dispose' - }, - }] - - expect(() => { agent.send(content) }).toThrow(/agent "reentrant-send-dispose" is disposed/) - await handle.dispose() - - expect(queued).toBe(0) - expect(agent.session.events).toHaveLength(0) - expect(adapter.requests).toHaveLength(0) - }) - it('running steer() owns content and source before notification and delivery', async () => { const adapter = new MockAdapter([toolCallResponse('c1', 'gate', {}), textResponse('done')]) const ctx = await harness(adapter) @@ -519,12 +489,10 @@ describe('MEDIUM: misc registry and config fixes', () => { })) let notifiedContent: ContentBlock[] | undefined let notifiedSource: MessageSource | undefined - let notifiedInfoFrozen = false ctx.on('agent/queued', (subject, acceptedContent, info) => { if (subject !== agent || !info.steering) return notifiedContent = acceptedContent notifiedSource = info.source - notifiedInfoFrozen = Object.isFrozen(info) }) agent.send([{ type: 'text', text: 'start' }]) @@ -544,7 +512,6 @@ describe('MEDIUM: misc registry and config fixes', () => { expect(Object.isFrozen(notifiedContent)).toBe(true) expect(Object.isFrozen(notifiedContent?.[0])).toBe(true) expect(Object.isFrozen(notifiedSource)).toBe(true) - expect(notifiedInfoFrozen).toBe(true) const recorded = agent.session.events.flatMap(event => event.type === 'steering/message' ? [event.data] : []) expect(recorded).toContainEqual({ turn: 1, @@ -579,7 +546,7 @@ describe('MEDIUM: turn numbering continues across seeded (forked) sessions', () const seeded = ctx2.sessions.create(SessionId('forked'), { seed: [...agent.session.events] }) const prepared = prepareReactLoopAgent(ctx2, AgentId('forked-agent'), { model: 'mock' }, seeded) const forked = prepared.agent - prepared.enableDrive() + prepared.markPublished() ctx2.effect(() => prepared.startDriver()) const turns: number[] = [] diff --git a/packages/core/agent-loop/tests/scope-lifecycle.spec.ts b/packages/core/agent-loop/tests/scope-lifecycle.spec.ts index 275fdadb5a..2c478ad1f0 100644 --- a/packages/core/agent-loop/tests/scope-lifecycle.spec.ts +++ b/packages/core/agent-loop/tests/scope-lifecycle.spec.ts @@ -8,7 +8,6 @@ import AgentRegistry, { AgentId, agentEvents, assembleContextFor } from '@deepse import type { Agent } from '@deepseek-ai/dsh-agent' import { scopeOf } from '@deepseek-ai/dsh-scope' import AgentLoop, { ReactLoopAgent } from '@deepseek-ai/dsh-agent-loop' -import * as concreteAgentModule from '../src/agent.ts' import type { ContentBlock } from '@deepseek-ai/dsh-llm' import { MockAdapter, textResponse } from './mock-adapter.ts' @@ -41,18 +40,108 @@ function waitForIdle(ctx: Context, agent: ReactLoopAgent): Promise { const text = (t: string): ContentBlock[] => [{ type: 'text', text: t }] +/** Throw an arbitrary callback value to exercise the public unknown-error boundary. */ +function throwUnknown(value: unknown): never { + throw value +} + /** Invoke the exact lifecycle effect to exercise same-stack reentrant teardown. */ function disposeCurrentLifecycle(ownerCtx: Context): void { const lifecycle = [...ownerCtx.fiber._disposables] .find((dispose) => { const effect = (dispose as typeof dispose & { [symbols.effect]?: EffectMeta })[symbols.effect] - return effect?.label === 'agentLoop.lifecycle()' + return effect?.label.startsWith('agentLoop.lifecycle(') === true }) if (lifecycle === undefined) throw new Error('agent lifecycle effect not found') void lifecycle() } describe('agent scope lifecycle', () => { + it('rejects an already-aborted creation signal before publishing either identity', async () => { + const ctx = await harness() + const reason = new Error('cancelled before creation') + const controller = new AbortController() + controller.abort(reason) + + await expect(ctx.agents.create({ + agentId: AgentId('pre-aborted'), + sessionId: SessionId('pre-aborted-s'), + signal: controller.signal, + })).rejects.toBe(reason) + + expect(ctx.agents.get(AgentId('pre-aborted'))).toBeUndefined() + expect(ctx.sessions.get(SessionId('pre-aborted-s'))).toBeUndefined() + + const valueController = new AbortController() + valueController.abort('plain cancellation reason') + await expect(ctx.agents.create({ + agentId: AgentId('pre-aborted-value'), + sessionId: SessionId('pre-aborted-value-s'), + signal: valueController.signal, + })).rejects.toMatchObject({ + message: 'agent "pre-aborted-value" creation aborted', + cause: 'plain cancellation reason', + }) + + expect(ctx.agents.get(AgentId('pre-aborted-value'))).toBeUndefined() + expect(ctx.sessions.get(SessionId('pre-aborted-value-s'))).toBeUndefined() + await ctx.fiber.dispose() + }) + + it('joins cleanup when an abort lands reentrantly during scope preparation', async () => { + const ctx = await harness() + const reason = new Error('cancelled while preparing') + const controller = new AbortController() + let aborted = false + ctx.on('internal/plugin', (fiber) => { + if (aborted || fiber.name !== 'scope') return + aborted = true + controller.abort(reason) + }) + + await expect(ctx.agents.create({ + agentId: AgentId('prepare-abort'), + sessionId: SessionId('prepare-abort-s'), + signal: controller.signal, + })).rejects.toBe(reason) + + expect(ctx.agents.get(AgentId('prepare-abort'))).toBeUndefined() + expect(ctx.sessions.get(SessionId('prepare-abort-s'))).toBeUndefined() + await ctx.fiber.dispose() + }) + + it('normalizes non-Error create failures for rollback while rethrowing the original value', async () => { + const ctx = await harness() + let thrown: unknown + ctx.on('session/created', () => { + if (thrown === undefined) return + const value = thrown + thrown = undefined + throwUnknown(value) + }) + + const createFailure = { source: 'create' } + thrown = createFailure + let createCaught: unknown + try { + ctx.agentLoop.create(AgentId('unknown-create')) + } catch (error: unknown) { + createCaught = error + } + expect(createCaught).toBe(createFailure) + + const ownedFailure = { source: 'createAgent' } + thrown = ownedFailure + await expect(ctx.agents.create({ + agentId: AgentId('unknown-owned-create'), + sessionId: SessionId('unknown-owned-create-s'), + })).rejects.toBe(ownedFailure) + + expect(ctx.agents.get(AgentId('unknown-create'))).toBeUndefined() + expect(ctx.agents.get(AgentId('unknown-owned-create'))).toBeUndefined() + await ctx.fiber.dispose() + }) + it('wires agent.ctx: tagged with the agent, DX field set, ctx.agent safe elsewhere', async () => { const ctx = await harness() const agent = ctx.agentLoop.create(AgentId('a1'), { model: 'mock' }) @@ -167,11 +256,9 @@ describe('agent scope lifecycle', () => { expect(ctx.agents.get(AgentId('atomic'))).toBeUndefined() expect(ctx.sessions.get(SessionId('atomic-s'))).toBeUndefined() expect(order).toEqual(['setup:start']) - acceptedOptions.model = 'mutated while setup was pending' - gate.resolve(undefined) const handle = await creating - expect(handle.agent.options.model).toBe('mock') + expect(handle.agent.options).toBe(acceptedOptions) expect(order).toEqual([ 'setup:start', 'setup:end', @@ -184,90 +271,80 @@ describe('agent scope lifecycle', () => { await handle.dispose() }) - it('reserves agent and session ids across concurrent async setup', async () => { + it('lets the final enter arbitrate unsupported concurrent same-id creation and rolls the loser back', async () => { const ctx = await harness() const gate = Promise.withResolvers() + const bothStarted = Promise.withResolvers() + let started = 0 + const setup = async (): Promise => { + started += 1 + if (started === 2) bothStarted.resolve(undefined) + await gate.promise + } + const agentId = AgentId('concurrent-final-enter') const first = ctx.agents.create({ - agentId: AgentId('reserved'), - sessionId: SessionId('reserved-s'), + agentId, + sessionId: SessionId('concurrent-final-enter-a'), agentOptions: { model: 'mock' }, - setup: () => gate.promise, + setup, }) - - await expect(ctx.agents.create({ - agentId: AgentId('reserved'), - sessionId: SessionId('other-s'), + const second = ctx.agents.create({ + agentId, + sessionId: SessionId('concurrent-final-enter-b'), agentOptions: { model: 'mock' }, - })).rejects.toThrow(/already registered/) - await expect(ctx.agents.create({ - agentId: AgentId('other'), - sessionId: SessionId('reserved-s'), - agentOptions: { model: 'mock' }, - })).rejects.toThrow(/already exists/) + setup, + }) + await bothStarted.promise expect(ctx.agents.list()).toEqual([]) expect(ctx.sessions.list()).toEqual([]) gate.resolve(undefined) - const handle = await first - await handle.dispose() + const outcomes = await Promise.allSettled([first, second]) + const fulfilled = outcomes.filter((outcome): outcome is PromiseFulfilledResult> => outcome.status === 'fulfilled') + const rejected = outcomes.filter((outcome): outcome is PromiseRejectedResult => outcome.status === 'rejected') + expect(fulfilled).toHaveLength(1) + expect(rejected).toHaveLength(1) + expect(String(rejected[0]!.reason)).toMatch(/already registered/) + expect(ctx.agents.list()).toEqual([fulfilled[0]!.value.agent]) + expect(ctx.sessions.list()).toEqual([fulfilled[0]!.value.agent.session]) + + await fulfilled[0]!.value.dispose() + expect(ctx.agents.list()).toEqual([]) + expect(ctx.sessions.list()).toEqual([]) }) - it('makes setup-time publication structurally impossible through public stores', async () => { + it('uses signal only for creation: aborts pending setup but not a returned live handle', async () => { const ctx = await harness() - const lifecycle: string[] = [] - ctx.on('session/created', () => void lifecycle.push('session')) - ctx.on('agent/created', () => void lifecycle.push('agent')) - - const handle = await ctx.agents.create({ - agentId: AgentId('guarded-publication'), - sessionId: SessionId('guarded-publication-s'), + const pendingController = new AbortController() + const setupStarted = Promise.withResolvers() + const pending = ctx.agents.create({ + agentId: AgentId('signal-pending'), + sessionId: SessionId('signal-pending-s'), agentOptions: { model: 'mock' }, - setup: (agentCtx) => { - const agent = agentCtx.agent! - expect(() => agentCtx.agents.enter(agent)).toThrow(/reserved for unpublished creation/) - expect(() => agentCtx.agents.register(agent)).toThrow(/reserved for unpublished creation/) - expect(() => agentCtx.sessions.enter(agent.session)).toThrow(/reserved for unpublished creation/) - expect(() => agentCtx.sessions.prepare(agent.session.id)).toThrow(/reserved for unpublished creation/) - expect(() => agentCtx.sessions.create(agent.session.id)).toThrow(/reserved for unpublished creation/) - expect(lifecycle).toEqual([]) - expect(ctx.agents.get(agent.id)).toBeUndefined() - expect(ctx.sessions.get(agent.session.id)).toBeUndefined() + signal: pendingController.signal, + setup: async () => { + setupStarted.resolve(undefined) + await new Promise(() => {}) }, }) + await setupStarted.promise + pendingController.abort(new Error('cancel pending creation')) + await expect(pending).rejects.toThrow('cancel pending creation') + expect(ctx.agents.get(AgentId('signal-pending'))).toBeUndefined() + expect(ctx.sessions.get(SessionId('signal-pending-s'))).toBeUndefined() - expect(lifecycle).toEqual(['session', 'agent']) - expect(ctx.agents.get(handle.agent.id)).toBe(handle.agent) - expect(ctx.sessions.get(handle.agent.session.id)).toBe(handle.agent.session) - await handle.dispose() - }) - - it('structurally rejects every driving verb during setup', async () => { - const ctx = await harness() - const handle = await ctx.agents.create({ - agentId: AgentId('no-drive'), - sessionId: SessionId('no-drive-s'), + const liveController = new AbortController() + const live = await ctx.agents.create({ + agentId: AgentId('signal-live'), + sessionId: SessionId('signal-live-s'), agentOptions: { model: 'mock' }, - setup: (agentCtx) => { - const agent = agentCtx.agent! - // Even JavaScript or a cast to the exported concrete class cannot name - // a public start method. Driver startup is behind a module-private - // symbol used only by AgentLoop after rollback-covered publication. - expect(Reflect.get(agent as ReactLoopAgent, 'start')).toBeUndefined() - expect(Reflect.get(concreteAgentModule, 'enableAgentDrive')).toBeUndefined() - expect(Reflect.get(concreteAgentModule, 'startAgentDriver')).toBeUndefined() - expect(() => concreteAgentModule.prepareReactLoopAgent( - agentCtx, agent.id, agent.options, agent.session, - )).toThrow(/already has a concrete agent driver/) - expect(Reflect.get(agent as ReactLoopAgent, 'inbox')).toBeUndefined() - expect(() => { agent.send(text('queued too soon')) }).toThrow(/cannot send before creation setup completes/) - expect(() => { agent.steer(text('steered too soon')) }).toThrow(/cannot steer before creation setup completes/) - expect(() => { agent.inject(text('injected too soon')) }).toThrow(/cannot inject before creation setup completes/) - expect(() => { agent.cancel('cancel too soon') }).toThrow(/cannot cancel before creation setup completes/) - expect(agent.session.events).toEqual([]) - }, + signal: liveController.signal, }) - expect(handle.agent.session.events).toEqual([]) - await handle.dispose() + liveController.abort(new Error('too late')) + await Promise.resolve() + expect(ctx.agents.get(live.agent.id)).toBe(live.agent) + expect(live.agent.status).toBe('idle') + await live.dispose() }) it('owner unload aborts a pending setup and publishes nothing', async () => { @@ -347,16 +424,11 @@ describe('agent scope lifecycle', () => { await setupStarted.promise await loopFiber.dispose() - await expect(creating).rejects.toThrow(/owner disposed during setup/) + await expect(creating).rejects.toThrow(/agent loop is not active/) expect(published).toEqual([]) expect(ctx.agents.get(AgentId('factory-setup-race'))).toBeUndefined() expect(ctx.sessions.get(SessionId('factory-setup-race-s'))).toBeUndefined() - // Factory unload itself reached the reservation-release boundary. - const agentReservation = ctx.agents.reserve(AgentId('factory-setup-race')) - const sessionReservation = ctx.sessions.reserve(SessionId('factory-setup-race-s')) - sessionReservation.release() - agentReservation.release() gate.resolve(undefined) await ctx.fiber.dispose() }) @@ -377,16 +449,12 @@ describe('agent scope lifecycle', () => { agentOptions: { model: 'mock' }, setup: () => { setupCalls += 1 }, }) - await expect(creating).rejects.toThrow(/owner disposed during setup/) + await expect(creating).rejects.toThrow(/agent loop is not active/) await loopFiber.dispose() expect(setupCalls).toBe(0) expect(ctx.agents.get(AgentId('factory-scope-race'))).toBeUndefined() expect(ctx.sessions.get(SessionId('factory-scope-race-s'))).toBeUndefined() - const agentReservation = ctx.agents.reserve(AgentId('factory-scope-race')) - const sessionReservation = ctx.sessions.reserve(SessionId('factory-scope-race-s')) - sessionReservation.release() - agentReservation.release() await ctx.fiber.dispose() }) @@ -444,14 +512,14 @@ describe('agent scope lifecycle', () => { }) expect(() => ctx.agentLoop.create(AgentId('config-scope-race'), { model: 'mock' })) - .toThrow(/owner disposed during setup/) + .toThrow(/agent loop is not active/) await loopFiber.dispose() expect(ctx.agents.get(AgentId('config-scope-race'))).toBeUndefined() expect(ctx.sessions.list()).toHaveLength(sessionsBefore) await ctx.fiber.dispose() }) - it('synchronous create releases both reservations when session preparation fails', async () => { + it('synchronous create leaves no lifecycle state when session preparation fails', async () => { const ctx = await harness() const id = AgentId('config-prepare-failure') @@ -463,44 +531,7 @@ describe('agent scope lifecycle', () => { await ctx.fiber.dispose() }) - it('turns owner disposal from the caller association getter into a rollback boundary', async () => { - const ctx = await harness() - let creating!: ReturnType - let getterCalls = 0 - const creationStarted = Promise.withResolvers() - const owner = ctx.plugin(Object.assign((inner: Context) => { - Object.defineProperty(inner, 'agent', { - configurable: true, - get() { - getterCalls += 1 - void inner.fiber.dispose() - return undefined - }, - }) - creating = inner.agents.create({ - agentId: AgentId('association-dispose'), - sessionId: SessionId('association-dispose-s'), - agentOptions: { model: 'mock' }, - }) - creationStarted.resolve(undefined) - }, { inject: ['agents'] })) - - await creationStarted.promise - await expect(creating).rejects.toThrow(/owner disposed during setup/) - await owner - expect(getterCalls).toBe(1) - expect(ctx.agents.get(AgentId('association-dispose'))).toBeUndefined() - expect(ctx.sessions.get(SessionId('association-dispose-s'))).toBeUndefined() - const replacement = await ctx.agents.create({ - agentId: AgentId('association-dispose'), - sessionId: SessionId('association-dispose-s'), - agentOptions: { model: 'mock' }, - }) - await replacement.dispose() - await ctx.fiber.dispose() - }) - - it('factory unload awaits reservations when reentrant scope preparation throws', async () => { + it('factory unload awaits provisional cleanup when scope preparation throws', async () => { const { ctx, loopFiber } = await harnessWithLoop() let triggered = false ctx.on('internal/plugin', (fiber) => { @@ -519,36 +550,6 @@ describe('agent scope lifecycle', () => { expect(ctx.agents.get(AgentId('factory-scope-throw'))).toBeUndefined() expect(ctx.sessions.get(SessionId('factory-scope-throw-s'))).toBeUndefined() - const agentReservation = ctx.agents.reserve(AgentId('factory-scope-throw')) - const sessionReservation = ctx.sessions.reserve(SessionId('factory-scope-throw-s')) - sessionReservation.release() - agentReservation.release() - await ctx.fiber.dispose() - }) - - it('factory unload during session preparation awaits create reservation release', async () => { - const { ctx, loopFiber } = await harnessWithLoop() - let unloading!: Promise - const meta = { - get cwd() { - unloading = loopFiber.dispose() - return '/factory-unload' - }, - } - - const creating = ctx.agents.create({ - agentId: AgentId('factory-prepare-race'), - sessionId: SessionId('factory-prepare-race-s'), - agentOptions: { model: 'mock' }, - meta, - }) - await unloading - await expect(creating).rejects.toThrow('agent loop is not active') - - const agentReservation = ctx.agents.reserve(AgentId('factory-prepare-race')) - const sessionReservation = ctx.sessions.reserve(SessionId('factory-prepare-race-s')) - sessionReservation.release() - agentReservation.release() await ctx.fiber.dispose() }) @@ -566,14 +567,10 @@ describe('agent scope lifecycle', () => { expect(handle.agent.status).toBe('disposed') expect(ctx.agents.get(agentId)).toBeUndefined() expect(ctx.sessions.get(SessionId('factory-live-s'))).toBeUndefined() - expect(ctx.fiber.getEffects().filter(effect => effect.label === `agentLoop.ownerLifecycle(${agentId})`)).toEqual([]) + expect(ctx.fiber.getEffects().filter(effect => effect.label === `agentLoop.owner(${agentId})`)).toEqual([]) // The consumer handle shares the provider's completed quiescence boundary. await handle.dispose() - const agentReservation = ctx.agents.reserve(agentId) - const sessionReservation = ctx.sessions.reserve(SessionId('factory-live-s')) - sessionReservation.release() - agentReservation.release() await expect(loop.createAgent(ctx, { agentId: AgentId('factory-inactive'), sessionId: SessionId('factory-inactive-s'), @@ -647,7 +644,7 @@ describe('agent scope lifecycle', () => { }) }, { inject: ['agents'] })) - await expect(creating).rejects.toThrow(/owner disposed during setup/) + await expect(creating).rejects.toThrow(/lifecycle disposed/) await owner.dispose() expect(lifecycle).toEqual([ 'session-created:dispose', @@ -696,7 +693,7 @@ describe('agent scope lifecycle', () => { }) }, { inject: ['agents'] })) - await expect(creating).rejects.toThrow(/owner disposed during setup/) + await expect(creating).rejects.toThrow(/lifecycle disposed/) await owner.dispose() expect(lifecycle).toEqual([ 'session-created', @@ -711,52 +708,6 @@ describe('agent scope lifecycle', () => { await ctx.fiber.dispose() }) - it('rechecks owner liveness after carrier capture before the first creation edge', async () => { - const ctx = await harness() - let ownerCtx!: Context - const owner = await ctx.plugin(Object.assign((inner: Context) => { ownerCtx = inner }, { inject: ['agents'] })) - const agentId = AgentId('carrier-owner-race') - const sessionId = SessionId('carrier-owner-race-s') - const lifecycle: string[] = [] - let filterReads = 0 - ctx.on('session/created', (session) => { - if (session.id === sessionId) lifecycle.push('session-created') - }) - ctx.on('session/disposed', (session) => { - if (session.id === sessionId) lifecycle.push('session-disposed') - }) - ctx.on('agent/created', (agent) => { - if (agent.id === agentId) lifecycle.push('agent-created') - }) - ctx.on('agent/disposed', (agent) => { - if (agent.id === agentId) lifecycle.push('agent-disposed') - }) - - const creating = ownerCtx.agents.create({ - agentId, - sessionId, - agentOptions: { model: 'mock' }, - setup(agentCtx) { - Object.defineProperty(agentCtx.agent!.session, Context.filter, { - configurable: true, - get() { - filterReads += 1 - void owner.dispose() - return undefined - }, - }) - }, - }) - - await expect(creating).rejects.toThrow(/owner disposed during setup/) - await owner.dispose() - expect(filterReads).toBe(1) - expect(lifecycle).toEqual([]) - expect(ctx.agents.get(agentId)).toBeUndefined() - expect(ctx.sessions.get(sessionId)).toBeUndefined() - await ctx.fiber.dispose() - }) - it('rechecks caller liveness after creation listeners before unlocking the driver', async () => { const ctx = await harness() const starts: string[] = [] @@ -817,7 +768,7 @@ describe('agent scope lifecycle', () => { }) }, { inject: ['agents'] })) - await expect(creating).rejects.toThrow(/owner disposed during setup/) + await expect(creating).rejects.toThrow(/lifecycle disposed/) await owner.dispose() expect(announced.status).toBe('disposed') expect(statuses).toEqual(['disposed']) @@ -853,7 +804,7 @@ describe('agent scope lifecycle', () => { await retry.dispose() }) - it('rejects an exotic seed before publishing either reserved identity', async () => { + it('rejects an exotic durable seed before publishing either identity', async () => { const ctx = await harness() const published: string[] = [] ctx.on('session/created', () => { published.push('session') }) @@ -966,29 +917,6 @@ describe('agent scope lifecycle', () => { expect(heard).toEqual(['a1:2']) }) - it('a listener may drive the agent through its declared `this` (the carrier is method-transparent)', async () => { - // ds-review-bot regression: agent/* listeners are typed - // `this: Scoped`, and ReactLoopAgent's send/steer/cancel read the - // native-private #carrier — a proxy-receiver carrier made - // `this.send(...)` throw TypeError. The carrier binds methods to the real - // agent, so driving through the event `this` is a working supported shape. - const adapter = new MockAdapter([textResponse('first'), textResponse('second')]) - const ctx = await harness(adapter) - const agent = ctx.agentLoop.create(AgentId('a1'), { model: 'mock' }) - let followUpSent = false - ctx.on('agent/session-start', function (this: Agent) { - // Deliberately through `this`, not the args subject. - this.send(text('driven through this')) - followUpSent = true - }) - const second = ctx.agentLoop.create(AgentId('a2'), { model: 'mock' }) - expect(followUpSent).toBe(true) - await second.whenIdle() - // The send actually reached the loop: the prompt ran a turn. - expect(second.session.events.some(e => e.type === 'turn/start')).toBe(true) - await agent.whenIdle() - }) - it('owner unload honors the documented teardown order: unregistration AFTER the drain, before detach', async () => { const ctx = await harness() let handle!: Awaited> @@ -1044,18 +972,18 @@ describe('agent scope lifecycle', () => { await unload }) - it('successful handle disposal retires its caller ownership sentinel', async () => { + it('successful handle disposal retires its caller ownership effect', async () => { const ctx = await harness() - const agentId = AgentId('retired-owner-sentinel') + const agentId = AgentId('retired-owner-effect') const handle = await ctx.agents.create({ agentId, - sessionId: SessionId('retired-owner-sentinel-s'), + sessionId: SessionId('retired-owner-effect-s'), agentOptions: { model: 'mock' }, }) - expect(ctx.fiber.getEffects().map(effect => effect.label)).toContain(`agentLoop.ownerLifecycle(${agentId})`) + expect(ctx.fiber.getEffects().map(effect => effect.label)).toContain(`agentLoop.owner(${agentId})`) await handle.dispose() - expect(ctx.fiber.getEffects().filter(effect => effect.label === `agentLoop.ownerLifecycle(${agentId})`)).toEqual([]) + expect(ctx.fiber.getEffects().filter(effect => effect.label === `agentLoop.owner(${agentId})`)).toEqual([]) await ctx.fiber.dispose() }) @@ -1091,13 +1019,13 @@ describe('agent scope lifecycle', () => { await ctx.fiber.dispose() }) - it('retains both identity reservations until scope teardown reaches quiescence', async () => { + it('reopens ids after detach while the prior private scope finishes quiescing', async () => { const ctx = await harness() const gate = Promise.withResolvers() const cleanupStarted = Promise.withResolvers() const sessionDisposed = Promise.withResolvers() - const agentId = AgentId('quiescent-reservation') - const sessionId = SessionId('quiescent-reservation-s') + const agentId = AgentId('quiescent-reuse') + const sessionId = SessionId('quiescent-reuse-s') ctx.on('session/disposed', (session) => { if (session.id === sessionId) sessionDisposed.resolve(undefined) }) @@ -1117,12 +1045,12 @@ describe('agent scope lifecycle', () => { await Promise.all([sessionDisposed.promise, cleanupStarted.promise]) expect(ctx.agents.get(agentId)).toBeUndefined() expect(ctx.sessions.get(sessionId)).toBeUndefined() - await expect(ctx.agents.create({ agentId, sessionId, agentOptions: { model: 'mock' } })) - .rejects.toThrow(/reserved/) + const replacement = await ctx.agents.create({ agentId, sessionId, agentOptions: { model: 'mock' } }) + expect(ctx.agents.get(agentId)).toBe(replacement.agent) + expect(ctx.sessions.get(sessionId)).toBe(replacement.agent.session) gate.resolve(undefined) await disposing - const replacement = await ctx.agents.create({ agentId, sessionId, agentOptions: { model: 'mock' } }) await replacement.dispose() await ctx.fiber.dispose() }) diff --git a/packages/core/agent-loop/tests/turn-stop.spec.ts b/packages/core/agent-loop/tests/turn-stop.spec.ts index c4ead42040..c275fee88c 100644 --- a/packages/core/agent-loop/tests/turn-stop.spec.ts +++ b/packages/core/agent-loop/tests/turn-stop.spec.ts @@ -156,12 +156,9 @@ describe('agent/turn-stop', () => { expect(adapter.requests).toHaveLength(3) }) - it('fails throwing and malformed terminal policies closed while the driver survives', async () => { + it('fails a throwing terminal policy closed while the driver survives', async () => { const adapter = new MockAdapter([ textResponse('throwing policy'), - textResponse('malformed continue policy'), - textResponse('malformed false policy'), - textResponse('malformed null policy'), textResponse('healthy later turn'), ]) const ctx = await harness(adapter) @@ -179,21 +176,10 @@ describe('agent/turn-stop', () => { await send(agent, 'first') disposeThrowing() - for (const [index, malformed] of [ - { action: 'continue' }, - false, - null, - ].entries()) { - const disposeMalformed = agent.ctx.on('agent/turn-stop', () => malformed as unknown as ContinuationStop) - await send(agent, `malformed ${index}`) - disposeMalformed() - } - await send(agent, 'healthy') - expect(reasons.map(reason => reason.kind)).toEqual(['error', 'error', 'error', 'error', 'completed']) + expect(reasons.map(reason => reason.kind)).toEqual(['error', 'completed']) expect(errors).toContain('terminal policy exploded') - expect(errors).toContain("agent/turn-stop returned an invalid result; expected { action: 'stop' } or undefined") - expect(adapter.requests).toHaveLength(5) + expect(adapter.requests).toHaveLength(2) }) }) diff --git a/packages/core/agent/README.md b/packages/core/agent/README.md index 3663808847..290c5d90bf 100644 --- a/packages/core/agent/README.md +++ b/packages/core/agent/README.md @@ -8,20 +8,20 @@ Tracks live agents so UI, hook, and orchestrator plugins can find them without i ### Public API -The scoped-registration surface: `Agent.ctx` is the agent's scope context (`dsh-scope`, key = the agent) — register tools/sections/variables/listeners through it for that agent alone, all unwound on disposal. `agentEvents(ctx, agent)` is the fused dispatcher for ordinary agent-subject operations (carrier + injected subject in one move); its notification mode invokes every listener and contains both synchronous throws and returned-promise rejections. The registry lifecycle pair deliberately reuses the stable carrier captured before entry commit and applies the same per-listener containment directly. `assembleContextFor(agent)` builds the per-agent assembly context (`agent` + `scope` together). `CreateAgentOptions.setup(agentCtx)` and `ResumeAgentOptions.setup(agentCtx)` compose a fresh or resumed agent's scoped world while registry/store-owned reservation capabilities keep both identities unpublished; creation awaits setup and a same-turn owner-unload checkpoint before either creation notification or the first assembly. Setup composes, it never drives or publishes: driving verbs and ordinary agent/session insertion both reject until the owning publication boundary. +The scoped-registration surface: `Agent.ctx` is the agent's scope context (`dsh-scope`, key = the agent) — register tools/sections/variables/listeners through it for that agent alone, all unwound on disposal. `agentEvents(ctx, agent)` is the fused dispatcher for ordinary agent-subject operations (carrier + injected subject in one move); its notification mode invokes every listener and contains both synchronous throws and returned-promise rejections. The registry lifecycle pair reuses one stable routing carrier. `assembleContextFor(agent)` builds the per-agent assembly context (`agent` + `scope` together). `CreateAgentOptions.setup(agentCtx)` and `ResumeAgentOptions.setup(agentCtx)` compose a fresh or resumed agent's scoped world while both objects remain unpublished. Setup is trusted, composition-only same-process code: drive the agent only after creation resolves. - `ctx.agents.register(agent: Agent): () => Promise | void` — record an **already-constructed** agent. Disposed with the calling fiber. -- Advanced ordered lifecycle: `reserve(id)` returns an opaque unpublished-identity capability whose `release` is the exact owner effect disposer, allowing the factory to place ID release after scope quiescence instead of racing owner unload as a sibling. `enter(agent, reservation?): () => void` claims the ID across runtime pinning and stable lifecycle-carrier construction, then inserts without announcing; a Proxy trap or filter getter cannot reentrantly overwrite the commit. `announce(agent)` reuses that carrier and emits `agent/created` exactly once for the exact live entry, rejecting repeat or reentrant announcement. A detach requested synchronously by a creation listener is deferred until that dispatch unwinds, and every detach is exact-object guarded, so a later listener cannot observe inverted lifecycle edges and a stale capability cannot delete a replacement. While reserved, bare `register`/`enter` calls for the id reject, including from setup. The factory uses this split; ordinary plugins use `register()`. +- Advanced ordered lifecycle: `enter(agent): () => void` performs the authoritative ID collision check and inserts without announcing; `announce(agent)` emits `agent/created` exactly once. A detach requested synchronously by a creation listener is deferred until that dispatch unwinds, and every detach checks the captured entry object, so a stale capability cannot delete a later same-ID replacement. The async factory uses this split; ordinary plugins use `register()`. - `ctx.agents.get(id: AgentId): Agent | undefined` - `ctx.agents.list(): Agent[]` #### Factory seam (creation) -Agent *creation* is provided by the plugin implementing `AgentFactory` (`dsh-agent-loop`), registered via `setFactory`. This keeps creation on the `dsh-agent` interface so consumers (UI, the ACP bridge) program against `ctx.agents` without depending on the concrete loop package. The registry canonicalizes an already traced Service to its concrete target, captures and validates the factory's `createAgent` and `resume` callbacks once at registration, retains that target as their intentional receiver, and passes each call an explicit caller-bound `ownerCtx`; later method replacement cannot redirect a transaction, double tracing cannot break raw-identity state, and a plain non-Cordis factory receives enough context to implement caller ownership. +Agent *creation* is provided by the plugin implementing `AgentFactory` (`dsh-agent-loop`), registered via `setFactory`. This keeps creation on the `dsh-agent` interface so consumers (UI, the ACP bridge) program against `ctx.agents` without depending on the concrete loop package. The registry canonicalizes an already traced Service to its concrete target and re-traces each call through the caller's context; this avoids nested Cordis shadows while passing an explicit caller-bound `ownerCtx` to plain factories. - `ctx.agents.setFactory(factory: AgentFactory): () => Promise | void` — register the creation factory (the loop calls this on construction). Throws on a second factory; the slot clears on dispose. -- `ctx.agents.create(options: CreateAgentOptions): Promise` — snapshot caller-owned IDs/options/metadata and hand the one-read raw seed synchronously to the session boundary for one-pass lossless-JSON materialization, construct and await optional setup while unpublished, insert both session and agent, then recheck caller and factory liveness before the first creation announcement and after each later notification boundary. Only a still-live transaction opens `agent/session-start` and starts a new loop on the caller-supplied `sessionId`. Registry/store reservation capabilities block every competing public insertion across setup; seed rejection, setup rejection, caller unload, factory unload, or cancellation from a creation listener publishes no drivable agent. Publication is rollback-covered: if a creation listener throws, entries and scope unwind but effects of already-delivered notifications remain observable; any creation announcement that began is paired by `agent/disposed` or `session/disposed`. Rejects if no factory is registered. -- `ctx.agents.resume(options: ResumeAgentOptions): Promise` — snapshot caller-owned IDs/options, load a persisted session ([session persistence](../../../docs/rfc/implemented/architecture/2026-06-14-session-persistence.md)), mint a fresh agent scope, await optional setup while unpublished, then follow the same insert both → pre-announcement liveness check → session announcement → liveness check → agent announcement → liveness check → session-start → final liveness check → loop-start boundary. The IDs are reserved across persistence load, setup, and teardown quiescence; load/setup rejection, caller unload, or factory unload leaves no drivable or live publication, while any creation edge that already began is paired during rollback. Rejects if no factory is registered or session persistence is unconfigured. +- `ctx.agents.create(options: CreateAgentOptions): Promise` — create a session and agent, await optional setup while unpublished, then publish through final `SessionStore.enter()` and `AgentRegistry.enter()` checks. Concurrent same-ID creation is unsupported: more than one operation may prepare, but only one can enter; every loser rolls its private scope/session/driver back. An optional creation-only `signal` cancels unpublished setup and is detached before the handle is returned; later cancellation uses `handle.dispose()` or `agent.cancel()`. Publication is rollback-covered and every delivered creation edge is paired during rollback. Rejects if no factory is registered. +- `ctx.agents.resume(options: ResumeAgentOptions): Promise` — load a persisted session ([session persistence](../../../docs/rfc/implemented/architecture/2026-06-14-session-persistence.md)), mint a fresh unpublished agent scope, await optional setup, and use the same final-entry publication sequence. Its optional `signal` is likewise creation-only. Rejects if no factory is registered or session persistence is unconfigured. `AgentHandle = { agent: Agent; dispose(): Promise }`. The disposer is a **consumer capability** — no observer holding the bare registry entry can tear the agent down. The caller fiber and the registered factory provider are structural co-owners: caller unload enforces structured ownership, while factory unload must stop old instances because their scoped dependency surface belongs to that provider. `dispose()` from any owner reaches one memoized quiescence boundary: it stops the loop, `await`s its exit plus every outstanding idle-injection flush (not just the `disposed` status flip), unregisters the agent, removes its session from the store, and finally unwinds its scoped world. This order captures every agent-started `session/flush` before the session is detached and keeps scoped listeners alive through those checkpoints. `ctx.agents.get(id)` still returns a bare `Agent`; the ACP bridge and in-process subagent backends hold consumer handles, while config-created agents are already owned by the loop fiber. @@ -29,7 +29,7 @@ Agent *creation* is provided by the plugin implementing `AgentFactory` (`dsh-age `dsh-agent` declares the live `agent/*` coordination vocabulary so plugins do not depend on the concrete loop. Exact signatures, dispatch modes, scope-filtering rules, and payload contracts live in the generated [Cordis event catalog](../../../docs/cordis-catalog/events.md); the [architecture turn flow](../../../docs/architecture.md#turn-flow) shows their order relative to durable session events. -The lifecycle edges have two important local caveats. `agent/created` runs after scoped setup and after both session and agent registry entries exist, but concrete driving remains locked until the immediately following `agent/session-start`; that non-vetoing notification is the first supported startup injection point. `agent/disposed` always means the exact agent has left the registry. AgentLoop emits it after its driver is quiescent, while ordered teardown may still be detaching the session and unwinding the scope; custom agents registered directly own any stronger driver-ordering contract themselves. +The lifecycle edges have two important local caveats. `agent/created` runs after scoped setup and after both session and agent registry entries exist. Setup is trusted composition-only code; the immediately following non-vetoing `agent/session-start` notification is the first supported startup injection point. `agent/disposed` always means the exact agent has left the registry. AgentLoop emits it after its driver is quiescent, while ordered teardown may still be detaching the session and unwinding the scope; custom agents registered directly own any stronger driver-ordering contract themselves. Most interception points are cooperative waterfalls returning seam-specific decisions. `agent/pre-step` is a serial surface-mutation checkpoint, while `agent/turn-stop` is the owner-final exception: it runs after ordinary continuation and steering folding, and its terminal state remains through turn close and flush so steering from those later listeners cannot create an extra step or turn. Ordinary queued prompts remain intact. The full rationale is in the [agent-scope runtime-design RFC](../../../docs/rfc/implemented/architecture/2026-07-12-agent-scope-runtime-design.md#owner-final-policy-four-narrow-boundaries). diff --git a/packages/core/agent/src/dispatch.ts b/packages/core/agent/src/dispatch.ts index f8726c50cc..cbb641d271 100644 --- a/packages/core/agent/src/dispatch.ts +++ b/packages/core/agent/src/dispatch.ts @@ -6,8 +6,8 @@ * the first argument in one move, so a site cannot name a different subject. * The registry lifecycle pair is the deliberate exception: `enter()` captures * one stable carrier before commit and `announce()`/detach dispatch through it - * directly, preventing a mutable filter getter from changing or reentering the - * paired edges. The dev scoped-dispatch invariant checks both shapes. + * directly, so both lifecycle edges use the same routing identity. The dev + * scoped-dispatch invariant checks both shapes. * * @module @deepseek-ai/dsh-agent/dispatch */ @@ -61,16 +61,6 @@ export interface AgentEventDispatch { * @returns the serial chain's result (the first bail value, if any). */ serial(name: K, ...rest: Tail): Promise>> - /** - * Await listeners in order and return the first value other than `undefined`. - * Unlike Cordis `serial`, this does not silently treat `null` or `false` as - * abstentions. Use it for a runtime-validated public boundary whose declared - * abstention is exactly `undefined` (currently `agent/turn-stop`). - * @param name - the agent-subject event to dispatch. - * @param rest - the event's arguments after the injected agent. - * @returns the first non-undefined listener result, or undefined. - */ - strictSerial(name: K, ...rest: Tail): Promise>> /** * Around-middleware dispatch (Cordis `waterfall`) in the agent's scope. The * declared event parameters already end with the `next` callback, so `rest` @@ -110,10 +100,10 @@ export function agentEvents(ctx: Context, agent: Agent): AgentEventDispatch { try { const returned: unknown = callback(...args) void Promise.resolve(returned).catch((error: unknown) => { - ctx.logger.warn(`agent event "${name}" listener rejected: ${renderThrown(error)}`) + ctx.logger.warn(`agent event "${name}" listener rejected: ${String(error)}`) }) } catch (error: unknown) { - ctx.logger.warn(`agent event "${name}" listener threw: ${renderThrown(error)}`) + ctx.logger.warn(`agent event "${name}" listener threw: ${String(error)}`) } } }, @@ -122,22 +112,6 @@ export function agentEvents(ctx: Context, agent: Agent): AgentEventDispatch { const serial = ctx.serial as (thisArg: Scoped, name: string, ...args: unknown[]) => Promise return await serial(carrier, name, agent, ...rest) }, - strictSerial(name, ...rest) { - return (async (): Promise => { - // EventsService.dispatch applies the carrier filter and emits the same - // internal/dispatch instrumentation as ctx.serial, then mutates `args` - // down to the actual listener parameters. Invoke those callbacks in order - // ourselves so every non-undefined value reaches the caller's validator; - // Cordis serial would discard null/false before validation could see them. - const args: unknown[] = [carrier, name, agent, ...rest] - const callbacks = ctx.events.dispatch('serial', args) - for (const callback of callbacks) { - const result: unknown = await callback(...args) - if (result !== undefined) return result - } - return undefined - })() as Promise>> - }, waterfall(name, ...rest) { // eslint-disable-next-line @typescript-eslint/unbound-method -- the events mixin accessor returns a pre-bound function const waterfall = ctx.waterfall as (thisArg: Scoped, name: string, ...args: unknown[]) => never @@ -146,15 +120,6 @@ export function agentEvents(ctx: Context, agent: Agent): AgentEventDispatch { } } -/** Render an arbitrary thrown value without allowing coercion to throw again. */ -function renderThrown(value: unknown): string { - try { - return value instanceof Error ? `${value.name}: ${value.message}` : String(value) - } catch { - return '' - } -} - /** * The assembly context for one agent's prompt: the typed `agent` DX field and * the `scope` layer selector, set together (setting `agent` without `scope` diff --git a/packages/core/agent/src/index.ts b/packages/core/agent/src/index.ts index 5526006796..42b9d13e35 100644 --- a/packages/core/agent/src/index.ts +++ b/packages/core/agent/src/index.ts @@ -40,21 +40,20 @@ declare module 'cordis' { */ export interface CreateAgentOptions { /** The agent's id (the registry handle). */ - agentId: AgentId + readonly agentId: AgentId /** The live session's id (NOT derived from agentId). */ - sessionId: SessionId + readonly sessionId: SessionId /** * Session creation metadata: validated absolute `cwd`, `parentSession` * fork lineage, and the `seedLength` seed boundary. Mirrors the * `cwd`/`parentSession`/`seedLength` fields of * {@link CreateSessionOptions.meta} in dsh-session (the internal-only * `createdAt`, used when reconstructing a persisted session, is deliberately - * excluded — a factory caller never sets it). The factory reads this raw - * reference once and hands it synchronously to the session boundary, which - * rejects an exotic shell and captures each accepted field once before any - * asynchronous setup. + * excluded — a factory caller never sets it). This is durable session data, + * so the session boundary validates and snapshots it before asynchronous + * setup begins. */ - meta?: { cwd?: string; parentSession?: SessionId; seedLength?: number } + readonly meta?: { readonly cwd?: string; readonly parentSession?: SessionId; readonly seedLength?: number } /** * Seed events to reconstruct the child session's log from (the fork lineage * primitive). When present, the factory creates the session with this event @@ -64,12 +63,14 @@ export interface CreateAgentOptions { * from seq 0, carry only lossless-JSON data, and be balanced (no open * turn/step, no dangling tool-call), or the session constructor (and the * dev-mode invariants replay) reject it. The factory passes the raw seed to - * the synchronous one-pass validator/copier; it never pre-clones and thereby - * sanitizes exotic prototypes. Absent for a fresh (spawn) child. + * the session's durable validator/snapshot boundary. Absent for a fresh + * (spawn) child. */ - seed?: SessionEvent[] + readonly seed?: readonly SessionEvent[] /** Per-agent options (model, …). */ - agentOptions?: AgentOptions + readonly agentOptions?: AgentOptions + /** Optional creation-only cancellation signal; detached before the returned handle becomes visible. */ + readonly signal?: AbortSignal /** * Creation-time composition of the agent's scoped world. The factory awaits * setup after minting `agentCtx` but BEFORE inserting or announcing either @@ -80,11 +81,11 @@ export interface CreateAgentOptions { * first prompt assembly. A throw/rejection or owner disposal rolls the scope * back without publishing either id. * - * **Setup composes, it never drives**: calling `send`/`steer`/`inject` here - * would run an unpublished agent and violate the session-start boundary. - * Drive the agent only after the creation promise resolves. + * **Setup composes, it never drives**: the callback is trusted same-process + * code and receives the full scoped context, so this is a contract rather + * than a runtime restriction. Drive the agent only after creation resolves. */ - setup?: (agentCtx: Context) => Promise | void + readonly setup?: (agentCtx: Context) => Promise | void } /** @@ -93,21 +94,23 @@ export interface CreateAgentOptions { */ export interface ResumeAgentOptions { /** The agent's id (the registry handle). */ - agentId: AgentId + readonly agentId: AgentId /** The persisted session id to load and resume on. */ - resumeSessionId: SessionId + readonly resumeSessionId: SessionId /** Per-agent options (model, …). */ - agentOptions?: AgentOptions + readonly agentOptions?: AgentOptions + /** Optional creation-only cancellation signal for persistence load/setup; detached before return. */ + readonly signal?: AbortSignal /** * Resume-time composition of the agent's fresh scoped world. Persistence is * loaded first; the factory then mints `agentCtx` and awaits setup while the * reconstructed session and agent remain unpublished. The callback has the - * same composition-only contract as {@link CreateAgentOptions.setup}: all - * registrations exist before either creation announcement, driving verbs are - * unavailable until the session-start boundary, and rejection or owner - * disposal rolls the transaction back without publishing either id. + * same trusted composition-only contract as + * {@link CreateAgentOptions.setup}: all registrations exist before either + * creation announcement, and rejection or owner disposal rolls the + * transaction back without publishing either id. */ - setup?: (agentCtx: Context) => Promise | void + readonly setup?: (agentCtx: Context) => Promise | void } /** @@ -143,8 +146,8 @@ export interface AgentFactory { /** * Create a new agent on a caller-supplied session id. Async because creation * awaits unpublished setup, inserts both session and agent, emits their - * creation notifications in order, unlocks driving at - * `agent/session-start`, and only then starts the loop. The sequence is + * creation notifications in order, emits `agent/session-start`, and only + * then starts the loop. The sequence is * rollback-covered, but notifications delivered before a later listener * failure remain observable; every agent or session creation announcement * that began is paired by `agent/disposed` or `session/disposed` during @@ -163,8 +166,8 @@ export interface AgentFactory { * Load a persisted session and resume an agent on it. Async because it awaits * both `ctx.sessionPersistence.load` and the optional unpublished setup * transaction; must be called after that service exists (consumers inject - * `sessionPersistence`). Publication and drive unlocking follow the same - * ordered boundary as {@link createAgent}. + * `sessionPersistence`). Publication follows the same ordered boundary as + * {@link createAgent}. * @param ownerCtx - caller-bound context that owns load, setup, and the live handle. * @param options - persisted identity, configuration, and optional setup. * @returns the owned handle after setup, both announcements, and loop start complete. @@ -172,72 +175,22 @@ export interface AgentFactory { resume(ownerCtx: Context, options: ResumeAgentOptions): Promise } -/** One accepted factory target plus the callback identities captured at registration. */ -interface AcceptedAgentFactory { - target: AgentFactory - createAgent: AgentFactory['createAgent'] - resume: AgentFactory['resume'] -} - -/** Slot reservation while callback accessors are being captured. */ -const ACCEPTING_FACTORY = Symbol('accepting agent factory') - -/** Capture and validate the complete factory contract exactly once. */ -function acceptAgentFactory(factory: unknown): AcceptedAgentFactory { - if ((typeof factory !== 'object' && typeof factory !== 'function') || factory === null) { - throw new TypeError('agent factory must be a non-null object or function') - } - // A service read through ctx is already a Cordis trace proxy. Retaining that - // proxy and tracing it again for each create() caller produces two shadow - // layers; raw-identity state (AgentLoop's private ownership controller is - // one example) then unwraps only to the inner proxy instead of its service. - // Canonicalize the one framework-produced layer at acceptance and capture - // callbacks from the concrete target. Plain objects expose no original. - const original: unknown = Reflect.get(factory, symbols.original) - const target = ((typeof original === 'object' || typeof original === 'function') && original !== null) - ? original - : factory - const createAgent: unknown = Reflect.get(target, 'createAgent') - const resume: unknown = Reflect.get(target, 'resume') - if (typeof createAgent !== 'function') throw new TypeError('agent factory createAgent must be a function') - if (typeof resume !== 'function') throw new TypeError('agent factory resume must be a function') - return Object.freeze({ - target: target as AgentFactory, - createAgent: createAgent as AgentFactory['createAgent'], - resume: resume as AgentFactory['resume'], - }) -} - /** Thrown when create/resume is called before an agent factory is registered. */ const NO_FACTORY_MESSAGE = 'no agent factory registered (load an agent-loop plugin)' -/** Render an arbitrary thrown value without allowing coercion to throw again. */ -function renderThrown(value: unknown): string { - try { - return value instanceof Error ? `${value.name}: ${value.message}` : String(value) - } catch { - return '' - } +/** All mutable lifecycle state for one exact registry entry. */ +interface AgentEntry { + readonly id: AgentId + readonly agent: Agent + readonly carrier: Scoped + announced: boolean + announcing: boolean + detachRequested: boolean } -/** - * Unforgeable ownership handle for one unpublished agent id. The factory holds - * this object across asynchronous setup; while it is live, ordinary public - * registration of that id fails, so setup cannot publish the factory's agent - * (or a replacement with the same id) ahead of the transaction. Callers obtain - * handles only from {@link AgentRegistry.reserve}. - */ -export interface AgentRegistrationReservation { - /** The reserved registry id. */ - readonly id: AgentId - /** - * Release the unpublished reservation; idempotent. The registry also - * releases it automatically when the fiber that called `reserve` disposes. - * This function is that exact Cordis effect disposer, so an ordered - * lifecycle may yield it by identity and place release after quiescence. - * @returns nothing. - */ - release(): void +/** Plain holder prevents Cordis from tracing the factory field before the caller context is known. */ +interface FactorySlot { + readonly target: AgentFactory } /** @@ -248,22 +201,9 @@ export interface AgentRegistrationReservation { * {@link setFactory}. */ export class AgentRegistry extends Service { - private store = new Map() - /** Ids claimed across caller-code boundaries before their exact entry commits. */ - private enteringIds = new Set() - /** The one accepted registry key for each live agent; never reread caller state. */ - private acceptedIds = new WeakMap() - /** Unpublished identities held across factory setup/load transactions. */ - private reservations = new Map() - /** Entries whose `agent/created` announcement phase began. */ - private announced = new WeakSet() - /** Entries currently dispatching `agent/created`; detach waits for that dispatch to unwind. */ - private announcing = new WeakSet() - /** A detach requested reentrantly from `agent/created`. */ - private pendingDetach = new WeakSet() - /** Stable lifecycle dispatch carrier captured before an entry commits. */ - private carriers = new WeakMap>() - private factory: AcceptedAgentFactory | typeof ACCEPTING_FACTORY | undefined + private store = new Map() + private entries = new WeakMap() + private factory: FactorySlot | undefined constructor(ctx: Context) { super(ctx, 'agents') @@ -276,41 +216,13 @@ export class AgentRegistry extends Service { ctx.accessor('agent', { get: () => undefined }) } - /** - * Reserve an unpublished agent id. Registration through {@link register} or - * bare {@link enter} fails until the returned capability is released; the - * owning factory passes the exact capability back to `enter` at publication. - * This makes “setup cannot publish” structural rather than a cooperative - * convention, including attempts to register a different object under the - * reserved id. The reservation belongs to the calling fiber and is released - * automatically if that owner unloads before the transaction settles. - * @param id - the id the factory transaction will publish. - * @returns the opaque reservation capability. - * @throws if the id is malformed, live, or already reserved. - */ - reserve(id: AgentId): AgentRegistrationReservation { - if (typeof id !== 'string') throw new TypeError('agent id must be a string') - if (this.store.has(id) || this.reservations.has(id) || this.enteringIds.has(id)) { - throw new Error(`agent "${id}" is already registered or reserved`) - } - const rawRelease = (): void => { - this.reservations.delete(id) - } - // `release` is the exact effect disposer. A composite lifecycle can yield - // it by identity, moving automatic owner cleanup from a racing sibling to - // the transaction's final ordered position. - const release = this.ctx.effect(() => rawRelease, `agents.reserve(${id})`) - const reservation: AgentRegistrationReservation = Object.freeze({ id, release }) - this.reservations.set(id, reservation) - return reservation - } - /** * Register the agent-creation factory (the loop calls this on construction, - * effect-scoped). The registry captures both callback identities once and - * later invokes them against the retained target receiver. Throws if a - * factory is already registered. Returns the disposer; on dispose the - * factory slot is cleared. + * effect-scoped). A traced Cordis service is canonicalized to its concrete + * target; each create/resume call is then traced through that caller's + * context so ownership follows the caller without stacking proxy layers. + * Throws if a factory is already registered. Returns the disposer; on + * dispose the factory slot is cleared. * @param factory - the loop-owned factory {@link create}/{@link resume} delegate to. * @returns the disposer that clears the factory slot. The exact * Cordis effect disposer (single-shot): composite (generator) effects may @@ -319,17 +231,11 @@ export class AgentRegistry extends Service { setFactory(factory: AgentFactory): () => Promise | void { const dispose = this.ctx.effect(() => { if (this.factory !== undefined) throw new Error('an agent factory is already registered') - // Claim the slot before reading caller-controlled method accessors. A - // getter may synchronously re-enter setFactory(); it must observe the - // registration in progress instead of installing a nested factory that - // the outer call would silently overwrite. - this.factory = ACCEPTING_FACTORY - try { - this.factory = acceptAgentFactory(factory) - } catch (error: unknown) { - this.factory = undefined - throw error - } + // Avoid stacking two Cordis shadow layers when a caller passes a Service + // already read through a context. Calls are re-traced through their + // actual owner context below. + const target = (factory as AgentFactory & { [symbols.original]?: AgentFactory })[symbols.original] ?? factory + this.factory = { target } return () => { this.factory = undefined } }, 'agents.setFactory()') // The exact cordis effect disposer (the agents.register() convention): a @@ -339,11 +245,10 @@ export class AgentRegistry extends Service { return dispose } - /** Return the accepted factory, excluding absence and reentrant acceptance. */ - private requireFactory(): AcceptedAgentFactory { - const accepted = this.factory - if (accepted === undefined || accepted === ACCEPTING_FACTORY) throw new Error(NO_FACTORY_MESSAGE) - return accepted + /** Return the active creation factory. */ + private requireFactory(): FactorySlot { + if (this.factory === undefined) throw new Error(NO_FACTORY_MESSAGE) + return this.factory } /** @@ -356,14 +261,15 @@ export class AgentRegistry extends Service { * @returns the handle after setup, rollback-covered publication, and loop start complete. */ async create(options: CreateAgentOptions): Promise { - const accepted = this.requireFactory() const ownerCtx = this.ctx // Re-trace a Service-backed factory through the accessing context // explicitly. This preserves AgentLoop's dependency origin while binding // its effects to ownerCtx; plain factories receive ownerCtx as an explicit // capability and need no Cordis tracker magic. - const receiver = getTraceable(ownerCtx, accepted.target) - return Reflect.apply(accepted.createAgent, receiver, [ownerCtx, options]) + const { target } = this.requireFactory() + const receiver = getTraceable(ownerCtx, target) + // eslint-disable-next-line @typescript-eslint/unbound-method -- Reflect.apply intentionally supplies the caller-traced receiver + return Reflect.apply(target.createAgent, receiver, [ownerCtx, options]) } /** @@ -374,10 +280,11 @@ export class AgentRegistry extends Service { * @returns the handle after setup, rollback-covered publication, and loop start complete. */ async resume(options: ResumeAgentOptions): Promise { - const accepted = this.requireFactory() const ownerCtx = this.ctx - const receiver = getTraceable(ownerCtx, accepted.target) - return Reflect.apply(accepted.resume, receiver, [ownerCtx, options]) + const { target } = this.requireFactory() + const receiver = getTraceable(ownerCtx, target) + // eslint-disable-next-line @typescript-eslint/unbound-method -- Reflect.apply intentionally supplies the caller-traced receiver + return Reflect.apply(target.resume, receiver, [ownerCtx, options]) } /** @@ -413,74 +320,27 @@ export class AgentRegistry extends Service { * returned detach closure into its pre-installed composite teardown before * calling {@link announce}. Ordinary callers use {@link register}. * @param agent - the prepared, unpublished agent. - * @param reservation - the exact unpublished-id capability, when a factory - * reserved this id across setup. * @returns an idempotent closure that removes this exact entry and emits * `agent/disposed` with listener failures contained. When called from a * synchronous `agent/created` listener, removal and disposal wait until * that creation dispatch unwinds. */ - enter(agent: Agent, reservation?: AgentRegistrationReservation): () => void { + enter(agent: Agent): () => void { const id = agent.id - if (typeof id !== 'string') throw new TypeError('agent id must be a string') - const held = this.reservations.get(id) - if (reservation === undefined) { - if (held !== undefined) throw new Error(`agent "${id}" is reserved for unpublished creation`) - } else if (reservation.id !== id || held !== reservation) { - throw new Error(`agent "${id}" registration reservation is not active for this id`) + const carrier = scopeTarget(agent, agent) + // This is the authoritative collision boundary. Concurrent create/resume + // operations may both prepare, but only one exact entry can publish. + if (this.entries.has(agent) || this.store.has(id)) throw new Error(`agent "${id}" is already registered`) + const entry: AgentEntry = { + id, + agent, + carrier, + announced: false, + announcing: false, + detachRequested: false, } - if (this.acceptedIds.has(agent)) { - throw new Error(`agent "${id}" is already registered`) - } - if (this.store.has(id) || this.enteringIds.has(id)) { - throw new Error(`agent "${id}" is already registered`) - } - this.enteringIds.add(id) - let carrier: Scoped - try { - // Registration accepts ownership of the public identity contract. Pin an - // own data slot from the one captured value so a custom JavaScript Agent - // with a getter or writable field cannot later present a different id to - // event listeners while the registry still owns the accepted key. - try { - Object.defineProperty(agent, 'id', { - value: id, - enumerable: true, - writable: false, - configurable: false, - }) - } catch { - // Only the engine's property-definition failure is normalized; filter - // construction below retains its own precise failure. - throw new TypeError('agent id must be installable as a stable own property') - } - // Capture one carrier for the paired lifecycle edges. Constructing it - // reads a custom Agent's Context.filter and is therefore caller code; - // the id claim above makes a same-id reentrant enter lose deterministically. - carrier = scopeTarget(agent, agent) - } finally { - // Kept through the entire caller-code window; the final commit below is - // synchronous and callback-free. - this.enteringIds.delete(id) - } - const currentReservation = this.reservations.get(id) - if (reservation === undefined) { - /* v8 ignore next 2 -- reserve() rejects enteringIds, so no callback in - * carrier construction can install a new same-id reservation */ - if (currentReservation !== undefined) { - throw new Error(`agent "${id}" is reserved for unpublished creation`) - } - } else if (currentReservation !== reservation) { - throw new Error(`agent "${id}" registration reservation is not active for this id`) - } - /* v8 ignore next 2 -- the enteringIds claim blocks every public same-id - * commit until this callback-free final check has completed */ - if (this.acceptedIds.has(agent) || this.store.has(id)) { - throw new Error(`agent "${id}" is already registered`) - } - this.store.set(id, agent) - this.acceptedIds.set(agent, id) - this.carriers.set(agent, carrier) + this.store.set(id, entry) + this.entries.set(agent, entry) let entered = true const detach = (): void => { if (!entered) return @@ -490,49 +350,43 @@ export class AgentRegistry extends Service { // the advanced detach capability, so make that ordering structural: // visibility and the paired disposal are deferred until announce()'s // synchronous dispatch has unwound. - if (this.announcing.has(agent)) { - this.pendingDetach.add(agent) + if (entry.announcing) { + entry.detachRequested = true return } - this.detachEntered(agent, id) + this.detachEntered(entry) } return detach } /** Remove one exact entered agent and emit its paired disposal when announced. */ - private detachEntered(agent: Agent, id: AgentId): void { - this.pendingDetach.delete(agent) + private detachEntered(entry: AgentEntry): void { + entry.detachRequested = false // A stale capability can never delete a later same-id lifecycle. The - // commit claim prevents this mismatch in normal operation; retain the - // exact-object guard as the final identity boundary. - /* v8 ignore next 1 -- the commit claim makes replacement impossible; this - * remains the exact-identity backstop against future mutation paths */ - if (this.store.get(id) !== agent || this.acceptedIds.get(agent) !== id) return - this.store.delete(id) - this.acceptedIds.delete(agent) - const carrier = this.carriers.get(agent) - this.carriers.delete(agent) + // captured entry identity is the final boundary. + /* v8 ignore next -- enter() rejects replacement while this single-shot detach capability is live. */ + if (this.store.get(entry.id) !== entry) return + this.store.delete(entry.id) + this.entries.delete(entry.agent) // An insertion rolled back before announce was never externally created, // so emitting disposed would invent an impossible lifecycle edge. Marking // happens before the created emit: if a later created listener throws, // earlier listeners may already have observed it and must see disposal. - if (!this.announced.delete(agent)) return - /* v8 ignore next -- enter commits the carrier with the exact store entry */ - if (carrier === undefined) throw new Error(`agent "${id}" has no dispatch carrier`) - this.emitDisposed(agent, carrier, id) + if (!entry.announced) return + this.emitDisposed(entry) } /** Emit the paired disposal edge through the entry's stable carrier. */ - private emitDisposed(agent: Agent, carrier: Scoped, id: AgentId): void { - const args: unknown[] = [carrier, 'agent/disposed', agent] + private emitDisposed(entry: AgentEntry): void { + const args: unknown[] = [entry.carrier, 'agent/disposed', entry.agent] for (const callback of this.ctx.events.dispatch('emit', args)) { try { const returned: unknown = callback(...args) void Promise.resolve(returned).catch((error: unknown) => { - this.ctx.logger.warn(`agent "${id}": agent/disposed listener rejected: ${renderThrown(error)}`) + this.ctx.logger.warn(`agent "${entry.id}": agent/disposed listener rejected: ${String(error)}`) }) } catch (error: unknown) { - this.ctx.logger.warn(`agent "${id}": agent/disposed listener threw: ${renderThrown(error)}`) + this.ctx.logger.warn(`agent "${entry.id}": agent/disposed listener threw: ${String(error)}`) } } } @@ -545,21 +399,18 @@ export class AgentRegistry extends Service { * creation listener). */ announce(agent: Agent): void { - const id = this.acceptedIds.get(agent) - if (id === undefined || this.store.get(id) !== agent) { - throw new Error(`agent "${id ?? ''}" is not live in this registry`) + const entry = this.entries.get(agent) + if (entry === undefined || this.store.get(entry.id) !== entry) { + throw new Error(`agent "${agent.id}" is not live in this registry`) } - if (this.announced.has(agent) || this.announcing.has(agent)) { - throw new Error(`agent "${id}" was already announced`) + if (entry.announced || entry.announcing) { + throw new Error(`agent "${entry.id}" was already announced`) } - const carrier = this.carriers.get(agent) - /* v8 ignore next -- enter commits the carrier with the exact store entry */ - if (carrier === undefined) throw new Error(`agent "${id}" has no dispatch carrier`) // Mark before dispatch so a listener cannot recursively create a second // lifecycle edge; detach still pairs a partially delivered first edge. - this.announcing.add(agent) - this.announced.add(agent) - const args: unknown[] = [carrier, 'agent/created', agent] + entry.announcing = true + entry.announced = true + const args: unknown[] = [entry.carrier, 'agent/created', entry.agent] try { for (const callback of this.ctx.events.dispatch('emit', args)) { // A synchronous creation failure vetoes publication and rolls back. @@ -567,12 +418,12 @@ export class AgentRegistry extends Service { // observe and report it instead of leaking an unhandled rejection. const returned: unknown = callback(...args) void Promise.resolve(returned).catch((error: unknown) => { - this.ctx.logger.warn(`agent "${id}": agent/created listener rejected: ${renderThrown(error)}`) + this.ctx.logger.warn(`agent "${entry.id}": agent/created listener rejected: ${String(error)}`) }) } } finally { - this.announcing.delete(agent) - if (this.pendingDetach.has(agent)) this.detachEntered(agent, id) + entry.announcing = false + if (entry.detachRequested) this.detachEntered(entry) } } @@ -582,7 +433,7 @@ export class AgentRegistry extends Service { * @returns the agent, or undefined when no live agent has that id. */ get(id: AgentId): Agent | undefined { - return this.store.get(id) + return this.store.get(id)?.agent } /** @@ -590,7 +441,7 @@ export class AgentRegistry extends Service { * @returns a fresh array; mutating it does not affect the registry. */ list(): Agent[] { - return [...this.store.values()] + return [...this.store.values()].map(entry => entry.agent) } } diff --git a/packages/core/agent/src/types.ts b/packages/core/agent/src/types.ts index 419cc9d1fe..9350a5cc03 100644 --- a/packages/core/agent/src/types.ts +++ b/packages/core/agent/src/types.ts @@ -294,10 +294,10 @@ declare module 'cordis' { // ---- lifecycle (emit) ---- /** * An agent's fully composed scoped world was published in the - * {@link AgentRegistry}. Its session is already live in the session store, - * but concrete factories may keep driving verbs locked until the subsequent - * `agent/session-start` boundary; that event is the first supported place - * to inject or queue work during startup. A synchronous listener throw + * {@link AgentRegistry}. Its session is already live in the session store. + * Setup is composition-only by contract; the subsequent + * `agent/session-start` boundary is the first supported place to inject or + * queue startup work. A synchronous listener throw * vetoes publication and rollback emits the matching disposal edges; * returned-promise rejection is observed and logged but cannot * retroactively veto this synchronous boundary. A synchronous listener @@ -346,9 +346,7 @@ declare module 'cordis' { /** * A message entered the agent's inbox (queued or steering). Content and the * resolved source are the detached, deeply-frozen values retained by the - * inbox; the `info` wrapper is frozen too, so one listener cannot rewrite - * what another listener observes. `source` has defaults applied and is not - * the caller's raw options. + * inbox. `source` has defaults applied and is not the caller's raw options. * @param agent - the agent whose inbox received the message. * @param content - the accepted content blocks retained by the inbox. * @param info - the accepted source plus whether it entered as steering. diff --git a/packages/core/agent/tests/agent.spec.ts b/packages/core/agent/tests/agent.spec.ts index ab3202a297..cd40ba59e3 100644 --- a/packages/core/agent/tests/agent.spec.ts +++ b/packages/core/agent/tests/agent.spec.ts @@ -1,8 +1,8 @@ import { describe, expect, it } from 'vitest' import { Context, Service, symbols } from 'cordis' import { Session, SessionId } from '@deepseek-ai/dsh-session' -import AgentRegistry, { Agent, AgentId, agentEvents } from '@deepseek-ai/dsh-agent' -import type { AgentFactory, CreateAgentOptions, ResumeAgentOptions } from '@deepseek-ai/dsh-agent' +import AgentRegistry, { AgentId, agentEvents } from '@deepseek-ai/dsh-agent' +import type { Agent, AgentFactory, CreateAgentOptions, ResumeAgentOptions } from '@deepseek-ai/dsh-agent' function stubAgent(rawId: string): Agent { const id = AgentId(rawId) @@ -11,8 +11,6 @@ function stubAgent(rawId: string): Agent { options: {}, session: new Session(SessionId(`${id}-session`)), status: 'idle', - // A bare context stands in for the agent scope: registry tests never - // register through it, they only need the field present. ctx: new Context(), send() {}, steer() {}, @@ -23,382 +21,126 @@ function stubAgent(rawId: string): Agent { } describe('AgentRegistry', () => { - it('registers agents and emits created/disposed events', async () => { + it('registers exact entries, emits lifecycle events, and unregisters on owner disposal', async () => { const ctx = new Context() await ctx.plugin(AgentRegistry) - - const created: string[] = [] - const disposed: string[] = [] - ctx.on('agent/created', agent => void created.push(agent.id)) - ctx.on('agent/disposed', agent => void disposed.push(agent.id)) + const lifecycle: string[] = [] + ctx.on('agent/created', agent => void lifecycle.push(`created:${agent.id}`)) + ctx.on('agent/disposed', agent => void lifecycle.push(`disposed:${agent.id}`)) const agent = stubAgent('a1') const dispose = ctx.agents.register(agent) - expect(created).toEqual(['a1']) - expect(ctx.agents.get(AgentId('a1'))).toBe(agent) + expect(ctx.agents.get(agent.id)).toBe(agent) expect(ctx.agents.list()).toEqual([agent]) + expect(() => ctx.agents.register(stubAgent('a1'))).toThrow(/already registered/) await dispose() - expect(disposed).toEqual(['a1']) - expect(ctx.agents.get(AgentId('a1'))).toBeUndefined() + expect(ctx.agents.get(agent.id)).toBeUndefined() + expect(lifecycle).toEqual(['created:a1', 'disposed:a1']) }) - it('rejects duplicate ids and unregisters on fiber dispose (HMR safety)', async () => { + it('rolls an entry back and pairs a partially delivered creation when a listener throws', async () => { const ctx = new Context() await ctx.plugin(AgentRegistry) - ctx.agents.register(stubAgent('main')) - expect(() => ctx.agents.register(stubAgent('main'))).toThrow('already registered') + const lifecycle: string[] = [] + ctx.on('agent/created', agent => void lifecycle.push(`created:${agent.id}`)) + ctx.on('agent/created', () => { throw new Error('creation veto') }) + ctx.on('agent/disposed', agent => void lifecycle.push(`disposed:${agent.id}`)) - const fiber = await ctx.plugin(Object.assign((inner: Context) => { - inner.agents.register(stubAgent('scoped')) - }, { inject: ['agents'] })) - expect(ctx.agents.list().map(a => a.id)).toEqual(['main', 'scoped']) - - await fiber.dispose() - expect(ctx.agents.list().map(a => a.id)).toEqual(['main']) + expect(() => ctx.agents.register(stubAgent('vetoed'))).toThrow('creation veto') + expect(ctx.agents.get(AgentId('vetoed'))).toBeUndefined() + expect(lifecycle).toEqual(['created:vetoed', 'disposed:vetoed']) }) - it('rolls back the agent entry when an agent/created listener throws (P1-1)', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - - let threw = false - ctx.on('agent/created', () => { - if (!threw) { threw = true; throw new Error('boom created listener') } - }) - - // The throwing emit must roll the entry back, not leak it. - expect(() => ctx.agents.register(stubAgent('main'))).toThrow('boom created listener') - expect(ctx.agents.get(AgentId('main'))).toBeUndefined() // rolled back, not leaked - - // A subsequent listener-free register of the SAME id succeeds and is - // tracked exactly once (the duplicate-id check is not wedged). - const dispose = ctx.agents.register(stubAgent('main')) - expect(ctx.agents.list().map(a => a.id)).toEqual(['main']) - await dispose() - expect(ctx.agents.get(AgentId('main'))).toBeUndefined() - }) - - it('observes async agent/created rejection without rolling back or starving peers', async () => { + it('contains asynchronous creation rejection and every disposal-listener failure', async () => { const ctx = new Context() await ctx.plugin(AgentRegistry) const warnings: string[] = [] - ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn - const hostile = { [Symbol.toPrimitive]() { throw new Error('cannot stringify') } } const heard: string[] = [] - ctx.on('agent/created', () => Promise.reject(new Error('ordinary async failure')) as never) - // eslint-disable-next-line @typescript-eslint/prefer-promise-reject-errors -- hostile thrown values are the boundary under test - ctx.on('agent/created', () => Promise.reject(hostile) as never) - ctx.on('agent/created', (agent) => { heard.push(agent.id) }) + ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn + ctx.on('agent/created', () => Promise.reject(new Error('created async')) as never) + ctx.on('agent/disposed', () => { throw new Error('disposed sync') }) + ctx.on('agent/disposed', () => Promise.reject(new Error('disposed async')) as never) + ctx.on('agent/disposed', agent => void heard.push(agent.id)) - const agent = stubAgent('async-created') - const dispose = ctx.agents.register(agent) + const dispose = ctx.agents.register(stubAgent('contained')) await Promise.resolve() - await Promise.resolve() - - expect(ctx.agents.get(agent.id)).toBe(agent) - expect(heard).toEqual(['async-created']) - expect(warnings).toEqual([ - 'agent "async-created": agent/created listener rejected: Error: ordinary async failure', - 'agent "async-created": agent/created listener rejected: ', - ]) await dispose() + await Promise.resolve() + + expect(heard).toEqual(['contained']) + expect(warnings).toEqual([ + 'agent "contained": agent/created listener rejected: Error: created async', + 'agent "contained": agent/disposed listener threw: Error: disposed sync', + 'agent "contained": agent/disposed listener rejected: Error: disposed async', + ]) }) - it('splits insertion from announcement and makes the detach exact/idempotent', async () => { + it('separates entry from announcement and stale/idempotent detach cannot remove a replacement', async () => { const ctx = new Context() await ctx.plugin(AgentRegistry) - const created: Agent[] = [] - const disposed: Agent[] = [] - ctx.on('agent/created', agent => void created.push(agent)) - ctx.on('agent/disposed', agent => void disposed.push(agent)) + const lifecycle: string[] = [] + ctx.on('agent/created', agent => void lifecycle.push(`created:${agent.id}`)) + ctx.on('agent/disposed', agent => void lifecycle.push(`disposed:${agent.id}`)) const first = stubAgent('split') const detachFirst = ctx.agents.enter(first) - expect(ctx.agents.get(first.id)).toBe(first) - expect(created).toEqual([]) + expect(lifecycle).toEqual([]) ctx.agents.announce(first) - expect(created).toEqual([first]) + expect(() => { ctx.agents.announce(first) }).toThrow(/already announced/) detachFirst() detachFirst() - expect(disposed).toEqual([first]) const replacement = stubAgent('split') const detachReplacement = ctx.agents.enter(replacement) - // A stale repeated detach cannot remove the replacement. detachFirst() expect(ctx.agents.get(replacement.id)).toBe(replacement) expect(() => { ctx.agents.announce(first) }).toThrow(/not live/) detachReplacement() - // The replacement was inserted but never announced, so rollback produces - // no disposed-without-created notification. - expect(disposed).toEqual([first]) + expect(lifecycle).toEqual(['created:split', 'disposed:split']) }) - it('captures and pins one runtime id before insertion, announcement, and detach', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - const existing = stubAgent('occupied') - const disposeExisting = ctx.agents.register(existing) - const candidate = stubAgent('placeholder') - let reads = 0 - Object.defineProperty(candidate, 'id', { - configurable: true, - get() { - reads += 1 - return reads === 1 ? AgentId('accepted') : AgentId('occupied') - }, - }) - - const detach = ctx.agents.enter(candidate) - expect(reads).toBe(1) - expect(candidate.id).toBe('accepted') - expect(reads).toBe(1) - expect(Object.getOwnPropertyDescriptor(candidate, 'id')).toMatchObject({ - configurable: false, - writable: false, - value: 'accepted', - }) - expect(ctx.agents.get(AgentId('accepted'))).toBe(candidate) - expect(ctx.agents.get(AgentId('occupied'))).toBe(existing) - expect(() => ctx.agents.enter(candidate)).toThrow(/already registered/) - - ctx.agents.announce(candidate) - detach() - expect(ctx.agents.get(AgentId('accepted'))).toBeUndefined() - expect(ctx.agents.get(AgentId('occupied'))).toBe(existing) - await disposeExisting() - - expect(() => ctx.agents.enter({ ...stubAgent('bad'), id: 42 } as unknown as Agent)) - .toThrow(/id must be a string/) - const pinnedAccessor = stubAgent('pinned') - Object.defineProperty(pinnedAccessor, 'id', { - configurable: false, - get: () => AgentId('pinned'), - }) - expect(() => ctx.agents.enter(pinnedAccessor)).toThrow(/installable as a stable own property/) - }) - - it('claims an id across a Proxy defineProperty trap before committing the exact entry', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - const id = AgentId('reentrant-enter') - const nested = stubAgent(id) - let nestedError = '' - let attempted = false - const target = stubAgent(id) - const outer = new Proxy(target, { - defineProperty(inner, property, descriptor) { - if (property === 'id' && !attempted) { - attempted = true - try { - ctx.agents.enter(nested) - } catch (error: unknown) { - nestedError = String(error) - } - } - return Reflect.defineProperty(inner, property, descriptor) - }, - }) - - const detach = ctx.agents.enter(outer) - expect(nestedError).toMatch(/already registered/) - expect(ctx.agents.get(id)).toBe(outer) - detach() - expect(ctx.agents.get(id)).toBeUndefined() - }) - - it('captures one lifecycle carrier before commit so a filter getter cannot invert edges', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - const events: string[] = [] - const agent = stubAgent('reentrant-carrier') - let detach = (): void => {} - Object.defineProperty(agent, Context.filter, { - configurable: true, - get() { - events.push('filter-getter') - detach() - return undefined - }, - }) - detach = ctx.agents.enter(agent) - ctx.on('agent/created', () => { events.push('created') }) - ctx.on('agent/disposed', () => { events.push('disposed') }) - - ctx.agents.announce(agent) - expect(events).toEqual(['filter-getter', 'created']) - expect(ctx.agents.get(agent.id)).toBe(agent) - detach() - expect(events).toEqual(['filter-getter', 'created', 'disposed']) - expect(ctx.agents.get(agent.id)).toBeUndefined() - }) - - it('revalidates an exact reservation after carrier construction runs caller code', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - const reservation = ctx.agents.reserve(AgentId('released-during-enter')) - const agent = stubAgent('released-during-enter') - Object.defineProperty(agent, Context.filter, { - configurable: true, - get() { - reservation.release() - return undefined - }, - }) - - expect(() => ctx.agents.enter(agent, reservation)).toThrow(/reservation is not active/) - expect(ctx.agents.get(agent.id)).toBeUndefined() - }) - - it('observes an async agent/disposed rejection through the stable carrier', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - const warnings: string[] = [] - ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn - ctx.on('agent/disposed', () => Promise.reject(new Error('late disposal failure')) as never) - const agent = stubAgent('async-disposed') - const detach = ctx.agents.enter(agent) - ctx.agents.announce(agent) - - detach() - await Promise.resolve() - await Promise.resolve() - expect(warnings).toEqual([ - 'agent "async-disposed": agent/disposed listener rejected: Error: late disposal failure', - ]) - }) - - it('uses an opaque one-id reservation to gate unpublished factory insertion', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - const held = ctx.agents.reserve(AgentId('held')) - - expect(() => ctx.agents.reserve(AgentId('held'))).toThrow(/already registered or reserved/) - expect(() => ctx.agents.enter(stubAgent('held'))).toThrow(/reserved for unpublished creation/) - const other = ctx.agents.reserve(AgentId('other')) - expect(() => ctx.agents.enter(stubAgent('held'), other)).toThrow(/not active for this id/) - - const agent = stubAgent('held') - const detach = ctx.agents.enter(agent, held) - ctx.agents.announce(agent) - held.release() - held.release() - expect(ctx.agents.get(AgentId('held'))).toBe(agent) - expect(() => ctx.agents.reserve(AgentId('held'))).toThrow(/already registered or reserved/) - detach() - other.release() - - const expired = ctx.agents.reserve(AgentId('expired')) - expired.release() - expect(() => ctx.agents.enter(stubAgent('expired'), expired)).toThrow(/not active for this id/) - expect(() => ctx.agents.reserve(42 as unknown as AgentId)).toThrow(/id must be a string/) - }) - - it('owns reservations by the calling fiber and rolls back failed ownership registration', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - let held!: import('@deepseek-ai/dsh-agent').AgentRegistrationReservation - let scopedAgents!: AgentRegistry - const owner = await ctx.plugin(Object.assign((inner: Context) => { - scopedAgents = inner.agents - held = inner.agents.reserve(AgentId('fiber-held')) - }, { inject: ['agents'] })) - - expect(() => ctx.agents.reserve(AgentId('fiber-held'))).toThrow(/already registered or reserved/) - await owner.dispose() - const reused = ctx.agents.reserve(AgentId('fiber-held')) - reused.release() - held.release() // idempotent after the automatic owner-disposal release - - // A disposed tracker cannot own a new effect. The failed effect install - // must remove the map entry it tentatively reserved before propagating. - expect(() => scopedAgents.reserve(AgentId('inactive-owner'))).toThrow(/inactive context/) - const recovered = ctx.agents.reserve(AgentId('inactive-owner')) - recovered.release() - }) - - it('rejects direct and reentrant repeat announcements to preserve one lifecycle pair', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - let created = 0 - let disposed = 0 - let reentrantError = '' - ctx.on('agent/created', (agent) => { - created += 1 - try { - ctx.agents.announce(agent) - } catch (error: unknown) { - reentrantError = String(error) - } - }) - ctx.on('agent/disposed', () => { disposed += 1 }) - - const agent = stubAgent('once') - const detach = ctx.agents.enter(agent) - ctx.agents.announce(agent) - expect(reentrantError).toMatch(/already announced/) - expect(() => { ctx.agents.announce(agent) }).toThrow(/already announced/) - detach() - expect({ created, disposed }).toEqual({ created: 1, disposed: 1 }) - }) - - it('defers a reentrant detach until the creation dispatch unwinds', async () => { + it('defers detach requested by a creation listener until that dispatch unwinds', async () => { const ctx = new Context() await ctx.plugin(AgentRegistry) const order: string[] = [] - const agent = stubAgent('reentrant-detach') - const detach = ctx.agents.enter(agent) - - ctx.on('agent/created', (created) => { - order.push('created:first') + const agent = stubAgent('reentrant') + ctx.on('agent/created', () => { + order.push(`first:${ctx.agents.get(agent.id) === agent}`) detach() - expect(ctx.agents.get(created.id)).toBe(created) + order.push(`after-detach:${ctx.agents.get(agent.id) === agent}`) }) - ctx.on('agent/created', (created) => { - order.push('created:second') - expect(ctx.agents.get(created.id)).toBe(created) - }) - ctx.on('agent/disposed', (disposed) => { - order.push('disposed') - expect(ctx.agents.get(disposed.id)).toBeUndefined() - }) - + ctx.on('agent/created', () => void order.push(`second:${ctx.agents.get(agent.id) === agent}`)) + ctx.on('agent/disposed', () => void order.push('disposed')) + const detach = ctx.agents.enter(agent) ctx.agents.announce(agent) - - expect(order).toEqual(['created:first', 'created:second', 'disposed']) + expect(order).toEqual(['first:true', 'after-detach:true', 'second:true', 'disposed']) expect(ctx.agents.get(agent.id)).toBeUndefined() - detach() }) }) describe('agentEvents()', () => { - it('contains synchronous throws and returned-promise rejections per listener', async () => { + it('contains each synchronous throw and returned-promise rejection', async () => { const ctx = new Context() const warnings: string[] = [] - ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn - const agent = stubAgent('contained') const heard: string[] = [] - const hostile = { [Symbol.toPrimitive]() { throw new Error('cannot stringify') } } - - ctx.on('agent/status', () => { throw hostile }) + ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn + const agent = stubAgent('event') + ctx.on('agent/status', () => { throw new Error('sync listener') }) ctx.on('agent/status', () => Promise.reject(new Error('async listener')) as never) - ctx.on('agent/status', (_subject, status) => { heard.push(status) }) + ctx.on('agent/status', (_agent, status) => void heard.push(status)) - expect(() => { agentEvents(ctx, agent).emit('agent/status', 'running') }).not.toThrow() + agentEvents(ctx, agent).emit('agent/status', 'running') await Promise.resolve() - await Promise.resolve() - expect(heard).toEqual(['running']) expect(warnings).toEqual([ - 'agent event "agent/status" listener threw: ', + 'agent event "agent/status" listener threw: Error: sync listener', 'agent event "agent/status" listener rejected: Error: async listener', ]) }) }) describe('AgentRegistry factory seam', () => { - /** A stub AgentFactory that records calls and returns a stub agent. */ function stubFactory() { const calls: { create: Array<{ ownerCtx: Context; options: CreateAgentOptions }> @@ -409,198 +151,44 @@ describe('AgentRegistry factory seam', () => { calls.create.push({ ownerCtx, options }) return { agent: stubAgent(options.agentId), dispose: () => Promise.resolve() } }, - resume(ownerCtx, options) { + async resume(ownerCtx, options) { calls.resume.push({ ownerCtx, options }) - return Promise.resolve({ agent: stubAgent(options.agentId), dispose: () => Promise.resolve() }) + return { agent: stubAgent(options.agentId), dispose: () => Promise.resolve() } }, } return { factory, calls } } - it('create()/resume() throw when no factory is registered', async () => { + it('requires a factory and delegates through the calling context', async () => { const ctx = new Context() await ctx.plugin(AgentRegistry) await expect(ctx.agents.create({ agentId: AgentId('a'), sessionId: SessionId('s') })).rejects.toThrow(/no agent factory/) - await expect(ctx.agents.resume({ agentId: AgentId('a'), resumeSessionId: SessionId('s') })).rejects.toThrow(/no agent factory/) - }) - - it('setFactory registers a factory; create/resume delegate to it', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) const { factory, calls } = stubFactory() ctx.agents.setFactory(factory) - const created = await ctx.agents.create({ agentId: AgentId('c1'), sessionId: SessionId('sess-1'), meta: { cwd: '/w' } }) - expect(created.agent.id).toBe('c1') - expect(calls.create).toHaveLength(1) - expect(calls.create[0]!.ownerCtx.fiber).toBe(ctx.fiber) - expect(calls.create[0]!.options) - .toEqual({ agentId: AgentId('c1'), sessionId: SessionId('sess-1'), meta: { cwd: '/w' } }) - - const resumed = await ctx.agents.resume({ agentId: AgentId('r1'), resumeSessionId: SessionId('old-sess') }) - expect(resumed.agent.id).toBe('r1') - expect(calls.resume).toHaveLength(1) - expect(calls.resume[0]!.ownerCtx.fiber).toBe(ctx.fiber) - expect(calls.resume[0]!.options).toEqual({ agentId: AgentId('r1'), resumeSessionId: SessionId('old-sess') }) - }) - - it('passes the calling fiber to a plain factory for create and resume ownership', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - const { factory, calls } = stubFactory() - ctx.agents.setFactory(factory) let callerFiber: Context['fiber'] | undefined - - const owner = await ctx.plugin(Object.assign(async (inner: Context) => { + await ctx.plugin(Object.assign(async (inner: Context) => { callerFiber = inner.fiber - await inner.agents.create({ agentId: AgentId('owned-create'), sessionId: SessionId('owned-session') }) - await inner.agents.resume({ agentId: AgentId('owned-resume'), resumeSessionId: SessionId('persisted') }) + await inner.agents.create({ agentId: AgentId('create'), sessionId: SessionId('create-s') }) + await inner.agents.resume({ agentId: AgentId('resume'), resumeSessionId: SessionId('resume-s') }) }, { inject: ['agents'] })) + expect(calls.create[0]?.ownerCtx.fiber).toBe(callerFiber) + expect(calls.resume[0]?.ownerCtx.fiber).toBe(callerFiber) + }) - expect(calls.create[0]!.ownerCtx.fiber).toBe(callerFiber) - expect(calls.resume[0]!.ownerCtx.fiber).toBe(callerFiber) + it('rejects a second factory and clears the slot with its owner (HMR)', async () => { + const ctx = new Context() + await ctx.plugin(AgentRegistry) + const owner = await ctx.plugin(Object.assign((inner: Context) => { + inner.agents.setFactory(stubFactory().factory) + expect(() => inner.agents.setFactory(stubFactory().factory)).toThrow(/already registered/) + }, { inject: ['agents'] })) + await expect(ctx.agents.create({ agentId: AgentId('before'), sessionId: SessionId('before-s') })).resolves.toBeDefined() await owner.dispose() + await expect(ctx.agents.create({ agentId: AgentId('after'), sessionId: SessionId('after-s') })).rejects.toThrow(/no agent factory/) }) - it('captures factory callbacks once while retaining the intentional target receiver', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - const reads = { create: 0, resume: 0 } - const receivers: unknown[] = [] - const replacements: string[] = [] - const target = { label: 'accepted-target' } as { label: string } & AgentFactory - - Object.defineProperties(target, { - createAgent: { - configurable: true, - get() { - reads.create += 1 - return function (this: typeof target, _ownerCtx: Context, options: CreateAgentOptions) { - receivers.push(this) - return Promise.resolve({ agent: stubAgent(options.agentId), dispose: () => Promise.resolve() }) - } - }, - }, - resume: { - configurable: true, - get() { - reads.resume += 1 - return function (this: typeof target, _ownerCtx: Context, options: ResumeAgentOptions) { - receivers.push(this) - return Promise.resolve({ agent: stubAgent(options.agentId), dispose: () => Promise.resolve() }) - } - }, - }, - }) - - ctx.agents.setFactory(target) - Object.defineProperties(target, { - createAgent: { - value: () => { - replacements.push('create') - return Promise.resolve({ agent: stubAgent('replacement'), dispose: () => Promise.resolve() }) - }, - }, - resume: { - value: () => { - replacements.push('resume') - return Promise.resolve({ agent: stubAgent('replacement'), dispose: () => Promise.resolve() }) - }, - }, - }) - - await ctx.agents.create({ agentId: AgentId('captured-create'), sessionId: SessionId('captured-session') }) - await ctx.agents.resume({ agentId: AgentId('captured-resume'), resumeSessionId: SessionId('captured-persisted') }) - - expect(reads).toEqual({ create: 1, resume: 1 }) - expect(receivers).toEqual([target, target]) - expect(replacements).toEqual([]) - }) - - it('reserves the factory slot before reading reentrant callback accessors', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - const nested = stubFactory().factory - const reads: string[] = [] - const reentrantCreate: Promise[] = [] - const target = {} as AgentFactory - - Object.defineProperties(target, { - createAgent: { - get() { - reads.push('createAgent') - expect(() => ctx.agents.setFactory(nested)).toThrow(/already registered/) - reentrantCreate.push(ctx.agents.create({ - agentId: AgentId('during-acceptance'), - sessionId: SessionId('during-acceptance-session'), - })) - return (_ownerCtx: Context, options: CreateAgentOptions) => Promise.resolve({ - agent: stubAgent(options.agentId), - dispose: () => Promise.resolve(), - }) - }, - }, - resume: { - get() { - reads.push('resume') - return (_ownerCtx: Context, options: ResumeAgentOptions) => Promise.resolve({ - agent: stubAgent(options.agentId), - dispose: () => Promise.resolve(), - }) - }, - }, - }) - - ctx.agents.setFactory(target) - expect(reentrantCreate).toHaveLength(1) - await expect(Promise.all(reentrantCreate)).rejects.toThrow(/no agent factory/) - await expect(ctx.agents.create({ - agentId: AgentId('after-acceptance'), - sessionId: SessionId('after-acceptance-session'), - })).resolves.toMatchObject({ agent: { id: 'after-acceptance' } }) - expect(reads).toEqual(['createAgent', 'resume']) - }) - - it('validates the complete factory shape when accepting it', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - - expect(() => ctx.agents.setFactory(null as unknown as AgentFactory)).toThrow(/non-null object or function/) - expect(() => ctx.agents.setFactory(42 as unknown as AgentFactory)).toThrow(/non-null object or function/) - expect(() => ctx.agents.setFactory({ resume() { return Promise.resolve() } } as unknown as AgentFactory)) - .toThrow(/createAgent must be a function/) - expect(() => ctx.agents.setFactory({ createAgent() { return Promise.resolve() } } as unknown as AgentFactory)) - .toThrow(/resume must be a function/) - - const callable = Object.assign(() => undefined, stubFactory().factory) - const dispose = ctx.agents.setFactory(callable) - await expect(ctx.agents.create({ agentId: AgentId('callable'), sessionId: SessionId('callable-session') })) - .resolves.toBeDefined() - await dispose() - }) - - it('setFactory rejects a second factory', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - ctx.agents.setFactory(stubFactory().factory) - expect(() => ctx.agents.setFactory(stubFactory().factory)).toThrow(/already registered/) - }) - - it('disposing the setFactory fiber clears the factory (HMR safety)', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - let dispose!: () => Promise | void - const fiber = await ctx.plugin(Object.assign((inner: Context) => { - dispose = inner.agents.setFactory(stubFactory().factory) - }, { inject: ['agents'] })) - await expect(ctx.agents.create({ agentId: AgentId('a'), sessionId: SessionId('s') })).resolves.toBeDefined() - void dispose - await fiber.dispose() - // factory slot cleared → create throws again - await expect(ctx.agents.create({ agentId: AgentId('a2'), sessionId: SessionId('s2') })).rejects.toThrow(/no agent factory/) - }) - - it('canonicalizes an already traced Service factory before caller retracing', async () => { + it('canonicalizes an already traced Service before tracing it for the caller', async () => { const ctx = new Context() await ctx.plugin(AgentRegistry) const states = new WeakMap() @@ -609,71 +197,27 @@ describe('AgentRegistry factory seam', () => { super(inner, 'tracedFactory') states.set(this, []) } - private calls(): string[] { const original = (this as unknown as { [symbols.original]?: TracedFactory })[symbols.original] ?? this const calls = states.get(original) - if (calls === undefined) throw new Error('factory receiver did not canonicalize to the raw service') + if (calls === undefined) throw new Error('factory receiver was not canonicalized') return calls } - - createAgent(_ownerCtx: Context, options: CreateAgentOptions) { + async createAgent(_ownerCtx: Context, options: CreateAgentOptions) { this.calls().push('create') - return Promise.resolve({ agent: stubAgent(options.agentId), dispose: () => Promise.resolve() }) + return { agent: stubAgent(options.agentId), dispose: () => Promise.resolve() } } - - resume(_ownerCtx: Context, options: ResumeAgentOptions) { + async resume(_ownerCtx: Context, options: ResumeAgentOptions) { this.calls().push('resume') - return Promise.resolve({ agent: stubAgent(options.agentId), dispose: () => Promise.resolve() }) + return { agent: stubAgent(options.agentId), dispose: () => Promise.resolve() } } } await ctx.plugin(TracedFactory) const traced = (ctx as Context & { tracedFactory: TracedFactory }).tracedFactory ctx.agents.setFactory(traced) - - await ctx.agents.create({ agentId: AgentId('traced-create'), sessionId: SessionId('traced-session') }) - await ctx.agents.resume({ agentId: AgentId('traced-resume'), resumeSessionId: SessionId('traced-persisted') }) + await ctx.agents.create({ agentId: AgentId('create'), sessionId: SessionId('create-s') }) + await ctx.agents.resume({ agentId: AgentId('resume'), resumeSessionId: SessionId('resume-s') }) const raw = (traced as unknown as { [symbols.original]?: TracedFactory })[symbols.original] expect(states.get(raw!)).toEqual(['create', 'resume']) }) - - it('rolls back register and factory acceptance when their owner unloads reentrantly', async () => { - const ctx = new Context() - await ctx.plugin(AgentRegistry) - let ownerCtx!: Context - const owner = await ctx.plugin(Object.assign((inner: Context) => { ownerCtx = inner }, { inject: ['agents'] })) - const agent = stubAgent('register-unload-race') - ctx.on('agent/created', (created) => { - if (created === agent) void owner.dispose() - }) - - ownerCtx.agents.register(agent) - await owner.dispose() - expect(ctx.agents.get(agent.id)).toBeUndefined() - - let factoryOwnerCtx!: Context - const factoryOwner = await ctx.plugin(Object.assign((inner: Context) => { factoryOwnerCtx = inner }, { inject: ['agents'] })) - const target = {} as AgentFactory - Object.defineProperties(target, { - createAgent: { - get() { - void factoryOwner.dispose() - return (_inner: Context, options: CreateAgentOptions) => Promise.resolve({ - agent: stubAgent(options.agentId), - dispose: () => Promise.resolve(), - }) - }, - }, - resume: { - value: (_inner: Context, options: ResumeAgentOptions) => Promise.resolve({ - agent: stubAgent(options.agentId), - dispose: () => Promise.resolve(), - }), - }, - }) - factoryOwnerCtx.agents.setFactory(target) - await factoryOwner.dispose() - await expect(ctx.agents.create({ agentId: AgentId('after-owner'), sessionId: SessionId('after-owner-s') })) - .rejects.toThrow(/no agent factory/) - }) }) diff --git a/packages/core/scope/README.md b/packages/core/scope/README.md index 06d06ba59e..55a0591d0e 100644 --- a/packages/core/scope/README.md +++ b/packages/core/scope/README.md @@ -4,15 +4,14 @@ Scoped-context registration primitive. `createScope(ctx, key)` mints a Cordis co ## Public API -- `createScope(ctx: Context, key: ScopeKey): Scope` Mint a scope under `ctx`'s fiber. Usable synchronously (effect collection is uid-gated; service resolution falls through to the minting plugin's dependency surface). Throws on a primitive key, or when `ctx`'s fiber is disposing (`INACTIVE_EFFECT`). +- `createScope(ctx: Context, key: ScopeKey): Scope` Mint a scope under `ctx`'s fiber. Usable synchronously (effect collection is uid-gated; service resolution falls through to the minting plugin's dependency surface). The typed, same-process key is trusted; an inactive minting context still fails through Cordis (`INACTIVE_EFFECT`). - `Scope.ctx` The tagged context: registrations through it are scope-visible AND scope-lifetime. Derived contexts (an `extend`, a fiber mounted under it) inherit the tag; nested scopes shadow (nearest tag wins). - `Scope.rawDispose` The EXACT Cordis disposer for the backing fiber — a composite (generator) effect yields THIS function to nest the scope's teardown at that yield position (Cordis dedupes nested effects by function identity; yielding a wrapper leaves the scope disposing as a concurrent sibling). - `Scope.dispose(): Promise` Idempotent, shared quiescence boundary for every registration made through the scope. Racing/repeat calls await the same teardown, including when `rawDispose` invoked the underlying single-shot Cordis disposer first. - `scopeOf(ctx: Context): ScopeKey | undefined` The tag a context (or any context derived from it) carries; `undefined` = context-global. -- `scopeTarget(base: T, key: ScopeKey | undefined): Scoped` Build the dispatch `thisArg` for a scope-filtered event: capture and compose `base`'s own `Context.filter` with the scope predicate (untagged listener ⇒ admitted; tagged ⇒ admitted iff tag === key; `key === undefined` ⇒ untagged only). The captured base filter and the exposed composed filter are invoked through captured JavaScript primordials, and the composed filter's frozen invocation surface cannot be replaced or tampered with. The carrier uses a dedicated surrogate proxy target; ordinary property access, writes, own-key visibility, methods, invocation, and construction delegate to `base`, and callable carriers match the base's constructable/non-constructable shape. For non-overlay base-owned properties, descriptor queries preserve values and flags except that configurable is normalized to `true`, as required to report those properties through an extensible surrogate; defining through the carrier is therefore supported only with an explicit `configurable: true` descriptor, while an omitted or false flag is rejected before the base is touched. Listener `this` stays `base`-shaped. `{ global: true }` listeners bypass filtering (Cordis semantics). -- `Scoped` The compile-time carrier brand: scope-filtered events demand it as their `this` type, so dispatching with a bare subject is a compile error. +- `scopeTarget(base: T, key: ScopeKey | undefined): Scoped` Build the opaque dispatch `thisArg` for a scope-filtered event. It composes `base`'s existing `Context.filter` with the scope predicate (untagged listener ⇒ admitted; tagged ⇒ admitted iff tag === key; `key === undefined` ⇒ untagged only). The carrier contains routing state only; the real subject is carried by the event arguments. `{ global: true }` listeners bypass filtering (Cordis semantics). +- `Scoped` The compile-time opaque carrier brand: scope-filtered events demand it as their `this` type, so dispatching with a bare subject is a compile error. The type parameter records the subject type but does not expose its properties. - `isScopeCarrier(value)` / `carrierKeyOf(value)` Runtime carrier marks, used by the dev invariants to assert every scope-filtered dispatch carries a carrier keyed to the subject its arguments name. -- `scopeHost(ctx, services)` Test/tooling host that snapshots the requested service list before activation, fails loud with stable missing-service diagnostics, and whose shared `dispose()` waits for both the host fiber and every minted scope, including a child already tearing down through `rawDispose`. ## Design contract diff --git a/packages/core/scope/src/index.ts b/packages/core/scope/src/index.ts index 526dffbb51..e229243504 100644 --- a/packages/core/scope/src/index.ts +++ b/packages/core/scope/src/index.ts @@ -1,23 +1,6 @@ /** - * Scoped-context primitive: mint a Cordis context that TAGS everything - * registered through it with an opaque {@link ScopeKey}, and dispatch events so - * listeners registered through such a context fire only for their key's - * subject. Scope-aware registries (`ctx.tools`, `ctx.systemPrompt`) read the - * tag via {@link scopeOf} to file a registration in the right layer; the agent - * loop is the one scope MINTER today (one scope per live agent, key = the - * `Agent` object — see `Agent.ctx` in `@deepseek-ai/dsh-agent`), but the - * mechanism is key-agnostic by design so packages below the agent layer - * (`dsh-session`, `dsh-system-prompt`) can depend on it without a dependency - * cycle. - * - * Ownership and visibility derive from ONE fact — which context a registration - * went through: the scope's fiber owns the disposal (a `ctx.effect()`/ - * `ctx.on()`/registry call through the scoped context unwinds on - * {@link Scope.dispose}, because Cordis routes a service method's `this.ctx` - * to the ACCESSING context), and the tag decides who sees it. Splitting those - * two — an explicit `{ scope }` registration parameter — would let a caller - * express "visible to X, disposed with Y", which is almost always a bug; the - * scoped context makes it unrepresentable. + * Scoped-context primitive: mint a Cordis context that tags registrations with + * an opaque identity and build routing-only event carriers for that identity. * * @module @deepseek-ai/dsh-scope */ @@ -25,476 +8,109 @@ import type { Context, Fiber } from 'cordis' import { Context as CordisContext } from 'cordis' -// Capture the invocation primordials once. A carrier holder can reach the -// composed Context.filter function, so neither that function's mutable -// property surface nor a base filter's own `.call` may choose how listener- -// selection predicates are invoked. -const reflectApply = Reflect.apply -// eslint-disable-next-line @typescript-eslint/unbound-method -const functionCall = Function.prototype.call - -/** - * The identity a scope is keyed by. Opaque and compared by object identity — - * never inspected. The harness convention: a live `Agent` is the key of its - * own scope, so seam vocabularies that already carry the agent - * (`ToolExecution.agent`, `AssembleContext.scope`) name the layer directly. - */ +/** An opaque, identity-compared scope key. */ export type ScopeKey = object -/** The context tag {@link createScope} writes and {@link scopeOf} reads (module-private). */ +/** Context tag written by {@link createScope}. */ const kScope = Symbol('dsh.scope') -/** The carrier mark {@link scopeTarget} writes and {@link carrierKeyOf} reads (module-private). */ -const kCarrier = Symbol('dsh.scope.carrier') - declare const ScopedBrand: unique symbol /** - * A dispatch carrier built by {@link scopeTarget}: structurally the `base` it - * overlays, branded so scope-filtered events can DEMAND a carrier as their - * `this` type — passing a bare subject where a `Scoped` is required is a - * compile error, which is what makes "forgot the carrier" unrepresentable at - * dispatch sites. The brand is compile-time only; {@link isScopeCarrier} is - * the runtime counterpart (used by the dev invariants). + * A routing-only event receiver built by {@link scopeTarget}. The type + * parameter records the subject type for dispatch checking; the carrier does + * not expose the subject's properties. Event payloads carry the real subject. */ -export type Scoped = T & { readonly [ScopedBrand]: 'dsh.scope.carrier' } +export type Scoped = object & { readonly [ScopedBrand]: T } -/** - * A minted scope: the tagged context to register through, plus the disposers - * that unwind every registration made through it. - */ +/** The key associated with each carrier. Presence distinguishes an unkeyed carrier from a non-carrier. */ +const carrierKeys = new WeakMap() + +/** A minted registration scope and its quiescent disposal boundaries. */ export interface Scope { - /** - * The scoped context. Registrations through it are tagged with the scope's - * key (scope-aware registries file them in that key's layer; `ctx.on` - * listeners fire only for dispatches targeted at that key) and owned by the - * scope's fiber (disposed together on {@link dispose}). Contexts DERIVED - * from it — an `extend`, a fiber mounted under it — inherit the tag through - * the prototype chain. - */ + /** Context through which scope-owned registrations are made. */ ctx: Context - /** - * The EXACT disposer Cordis registered on the minting fiber for the scope's - * backing fiber. A composite (generator) effect that owns the scope's - * position in an ordered teardown must yield THIS function: Cordis dedupes a - * nested effect out of the parent's concurrent disposal list by function - * identity, so yielding a wrapper would leave the scope disposing as an - * unordered sibling. Callers outside a composite effect use {@link dispose}. - * @returns the backing fiber's teardown promise (undefined on a repeat call - * — Cordis effect disposers are single-shot). - */ + /** Exact Cordis disposer, used when nesting this scope in an ordered composite effect. */ rawDispose: () => Promise | void - /** - * Unwind the scope: dispose the backing fiber, running every collected - * registration disposer. Idempotent and always awaitable: repeat and racing - * calls share one completion even though the underlying Cordis disposer is - * single-shot and returns undefined after its first invocation. - * After disposal the scoped context is inert — a further registration - * through it throws Cordis's INACTIVE_EFFECT. - * @returns for the call that initiates teardown: resolves when every - * registration's disposer has settled. Every repeat/racing call awaits - * that same quiescence boundary, including when {@link rawDispose} claimed - * the underlying single-shot Cordis disposer first. - */ + /** Dispose every scope-owned registration; racing calls await the same completion. */ dispose(): Promise } -/** - * Dispose a Cordis fiber and await its lifecycle inertia even when some other - * caller claimed the single-shot raw disposer first. `Fiber.dispose()` returns - * `undefined` on a repeat call, but the fiber's `inertia` remains the - * authoritative promise while its async unload is running. - */ +/** Follow a Cordis fiber through asynchronous teardown even if its raw disposer was already claimed. */ async function quiesceFiber(fiber: Fiber): Promise { await Promise.resolve(fiber.dispose()) while (fiber.inertia !== undefined) await fiber.inertia } -/** - * The shared no-op plugin every scope fiber mounts: named so diagnostics read - * `scope` and shared so all scopes join ONE plugin runtime (Cordis deletes the - * runtime record when its last fiber disposes, so idle deployments carry no - * residue). - */ +/** Shared no-op plugin used as the backing scope fiber. */ function scope(): void {} /** - * Mint a registration scope for `key` under `ctx`. - * - * Mounts a runtime fiber (`ctx.plugin`) and tags a child of its context with - * `key`. The fiber is usable synchronously — Cordis activates it on a - * microtask, but effect collection is uid-gated (not state-gated) and service - * resolution falls through the pending fiber to the MINTING plugin's - * dependency surface, so a caller may register through {@link Scope.ctx} the - * moment this returns. - * - * Service resolution through the scoped context flows through the minting - * plugin's dependency chain (the fiber walk), regardless of what the eventual - * holder's own fiber injected — handing out the scoped context hands out that - * dependency surface; see `Agent.ctx` in `@deepseek-ai/dsh-agent` for the harness's - * contract. - * @param ctx - the context to mount the scope under; its fiber must be active - * (a disposing owner throws Cordis's INACTIVE_EFFECT), and its plugin's - * `inject` surface is what the scoped context resolves services against. - * @param key - the scope's identity ({@link ScopeKey}); must be an object - * (identity-compared), else this throws. - * @returns the tagged context plus its disposers ({@link Scope}). + * Mint a scope under `ctx`. The scoped context inherits the minting plugin's + * dependency surface and owns every registration made through it. + * @param ctx - active context whose dependency surface the scope inherits. + * @param key - opaque identity used for listener routing. + * @returns the scoped context and exact/shared disposal boundaries. */ export function createScope(ctx: Context, key: ScopeKey): Scope { - // Runtime guard behind the ScopeKey type: callers outside the typechecker - // (yml-configured plugins, JS consumers) can still pass a primitive. - // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition - if ((typeof key !== 'object' && typeof key !== 'function') || key === null) { - throw new TypeError('createScope: key must be a non-null object or function (scope keys are identity-compared)') - } const fiber = ctx.plugin(scope) const scoped: Context = fiber.ctx.extend({ [kScope]: key }) let disposing: Promise | undefined return { ctx: scoped, - // fiber.dispose IS the disposer Cordis pushed onto the minting fiber's - // disposable list — the identity a composite effect must yield (see - // Scope.rawDispose). rawDispose: fiber.dispose, - // Memoize the public boundary and explicitly follow fiber inertia: the raw - // disposer must remain the exact Cordis function for ordered composition, - // so it cannot itself be wrapped to record a raw-first invocation. dispose: () => (disposing ??= quiesceFiber(fiber)), } } /** - * Read the scope key a context is tagged with, or `undefined` for an untagged - * (context-global) context. Walks the prototype chain, so any context DERIVED - * from a scoped context — service shadows, `extend`s, fibers mounted under it - * — reads as that scope; with nested scopes the nearest tag wins. - * @param ctx - the context to inspect (typically a registry method's - * `this.ctx`, i.e. the ACCESSING context). - * @returns the key given to {@link createScope}, or `undefined` when the - * context is not derived from any scope. + * Read the nearest scope tag inherited by a context. + * @param ctx - context to inspect. + * @returns its scope key, or `undefined` for an unscoped context. */ export function scopeOf(ctx: Context): ScopeKey | undefined { - // A plain (possibly proxied) property read: symbols bypass the Cordis - // context proxy's service resolution, and Reflect walks the prototype chain. return (ctx as Context & { [kScope]?: ScopeKey })[kScope] } -/** Whether a callable has JavaScript's internal construction capability. */ -function isConstructable(value: (...args: unknown[]) => unknown): boolean { - try { - // A Proxy has [[Construct]] iff its target does. Its trap returns before - // the engine invokes `value` or reads `value.prototype`, so a hostile but - // constructable callable cannot be mistaken for a non-constructor. - Reflect.construct(new Proxy(value, { construct: () => ({}) }), []) - return true - } catch { - // The harmless outer trap leaves lack of [[Construct]] as the only failure. - return false - } -} - /** - * Build the dispatch carrier for a scope-filtered event: `base` overlaid with - * a `Context.filter` that admits a listener iff + * Build the routing receiver for a scope-filtered event. Untagged listeners + * remain global; tagged listeners run only when their key matches. A base + * Cordis filter is composed before the scope predicate. * - * - its registering context is UNTAGGED (a context-global listener — the - * compatibility default: plain plugin listeners see every subject), or - * - its tag IS `key` (a scoped listener seeing exactly its own subject), - * - * AND `base`'s own filter (a Cordis `Service`'s listener-filter check) also admits - * it. Both the captured base filter and the composed filter are invoked - * through captured JavaScript primordials, so mutating either function's - * public `.call` property cannot bypass either predicate. Dispatching with - * `key === undefined` — a subject-less dispatch, e.g. a tool call with no - * calling agent or a bare (agent-less) session's events — - * admits only untagged listeners: a scoped listener never fires for someone - * else's (or nobody's) subject. Listeners registered `{ global: true }` - * bypass all filtering (Cordis semantics). - * - * Use it as the `thisArg` of the dispatch: - * `ctx.waterfall(scopeTarget(this, exec.agent), 'tools/pre-execute', …)`. The - * carrier is a TRANSPARENT proxy over `base`: reads delegate with `base` as - * the receiver and retrieved methods are bound to `base`, so a listener may - * call subject methods through its `this` (`this.send(…)` on a - * `Scoped`) even when the subject uses native `#private` fields — a - * bare proxy receiver would throw on those. Identity is still not - * transparent: `this !== subject` and method identity varies per read; the - * subject always travels in the event's arguments. The returned carrier is - * branded {@link Scoped} and runtime-marked ({@link isScopeCarrier} / - * {@link carrierKeyOf}) so both the type system and the dev invariants can - * tell a carrier from a bare subject. Defining an ordinary property through - * the carrier is supported only when its descriptor explicitly says - * `configurable: true`; an omitted or false flag is rejected before touching - * `base`, because the extensible surrogate cannot truthfully report a new - * non-configurable base property. - * @param base - the object the event is dispatched on behalf of (the owning - * service, or the subject agent itself); its own `Context.filter` is - * preserved and composed. - * @param key - the subject's scope key, or `undefined` for a subject-less - * dispatch. - * @returns the carrier to pass as the dispatch `thisArg`. + * The receiver is deliberately opaque: listener code obtains the real subject + * from event arguments, never from `this`. + * @param base - subject or service whose existing Cordis filter is preserved. + * @param key - routed scope identity, or `undefined` for an unscoped subject. + * @returns an opaque dispatch carrier. */ export function scopeTarget(base: T, key: ScopeKey | undefined): Scoped { - const baseFilter: unknown = (base as { [CordisContext.filter]?: unknown })[CordisContext.filter] - if (baseFilter !== undefined && typeof baseFilter !== 'function') { - throw new TypeError('scope target Context.filter must be a function when present') + const baseFilter = (base as { [CordisContext.filter]?: (ctx: Context) => boolean })[CordisContext.filter] + const carrier = { + [CordisContext.filter](ctx: Context): boolean { + if (baseFilter !== undefined && !baseFilter.call(base, ctx)) return false + const tag = scopeOf(ctx) + return tag === undefined || tag === key + }, } - const filter = (ctx: Context): boolean => { - if (baseFilter && !reflectApply(functionCall, baseFilter, [base, ctx])) return false - const tag = scopeOf(ctx) - return tag === undefined || tag === key - } - // Cordis invokes a dispatch filter as `filter.call(thisArg, listenerCtx)`. - // Pin that property to the captured primordial, then freeze the callable so - // a carrier holder cannot replace it with an always-true scope bypass. - Object.defineProperty(filter, 'call', { - value: functionCall, - writable: false, - configurable: false, - }) - Object.freeze(filter) - const overlay: Record = { - [CordisContext.filter]: filter, - [kCarrier]: Object.freeze({ key }), - } - // Use a dedicated extensible proxy TARGET, never `base` itself. Proxy get - // invariants force a trap to return a base's non-configurable/non-writable - // own value verbatim; if a caller pinned Context.filter during or after - // construction, a base-target proxy would therefore silently replace the - // composed scope predicate with the caller's filter. The surrogate owns the - // two immutable overlay slots, so later descriptor changes on `base` cannot - // affect listener selection. It shares the base prototype and delegates ordinary - // reads/writes/keys to preserve the supported transparent shape. Callable - // targets use native bound built-ins so V8 contributes no user-code surface; - // the chosen built-in matches whether `base` has [[Construct]], and the traps - // below delegate the actual call/construction to `base`. - const callableBase = typeof base === 'function' - ? base as unknown as (...args: unknown[]) => unknown - : undefined - const constructable = callableBase !== undefined && isConstructable(callableBase) - const target: object = callableBase === undefined - ? {} - : constructable - ? Object.bind(undefined) - : Math.max.bind(undefined) - Reflect.setPrototypeOf(target, Reflect.getPrototypeOf(base)) - Object.defineProperties(target, { - [CordisContext.filter]: { - value: filter, - enumerable: false, - writable: false, - configurable: false, - }, - [kCarrier]: { - value: overlay[kCarrier], - enumerable: false, - writable: false, - configurable: false, - }, - }) - const carrier = new Proxy(target, { - get(target, prop) { - // The callable surrogate has engine-owned pinned properties (`prototype`, - // `caller`, …); honor those target invariants. For object carriers the - // only pinned target properties are the exact overlay values above. - const own = Reflect.getOwnPropertyDescriptor(target, prop) - const pinned = own !== undefined && own.configurable === false - && own.get === undefined && own.writable !== true - if (pinned) { - const value: unknown = Reflect.get(target, prop, target) - return value - } - const value: unknown = Reflect.get(base, prop, base) - if (typeof value !== 'function') return value - // `constructor` is looked up, never invoked as a subject method — keep - // the real one (withProps special-cases it the same way), so - // `carrier.constructor` still identifies the subject's class. - if (prop === 'constructor') return value - // `Function.prototype.bind` types as `any`; the value is structurally - // T[prop] and the trap's contract is untyped (`any`), so unknown is the - // honest safe return. - return value.bind(base) as unknown - }, - set(_target, prop, value) { - if (Object.hasOwn(overlay, prop)) return false - return Reflect.set(base, prop, value, base) - }, - has(_target, prop) { - // A Proxy may not hide a non-configurable target key. Configurable - // surrogate-only keys (bound-function name/length) are omitted; the - // base's own/inherited surface remains authoritative. - const own = Reflect.getOwnPropertyDescriptor(target, prop) - return own?.configurable === false || Reflect.has(base, prop) - }, - ownKeys(target) { - const requiredTargetKeys = Reflect.ownKeys(target).filter((prop) => { - return Reflect.getOwnPropertyDescriptor(target, prop)?.configurable === false - }) - return [...new Set([...requiredTargetKeys, ...Reflect.ownKeys(base)])] - }, - getOwnPropertyDescriptor(target, prop) { - const targetDescriptor = Reflect.getOwnPropertyDescriptor(target, prop) - if (targetDescriptor?.configurable === false) return targetDescriptor - const baseDescriptor = Reflect.getOwnPropertyDescriptor(base, prop) - if (baseDescriptor !== undefined) return { ...baseDescriptor, configurable: true } - // Configurable surrogate-only function metadata is intentionally hidden. - return undefined - }, - defineProperty(_target, prop, attributes) { - if (Object.hasOwn(overlay, prop) || attributes.configurable !== true) return false - return Reflect.defineProperty(base, prop, attributes) - }, - deleteProperty(_target, prop) { - if (Object.hasOwn(overlay, prop)) return false - return Reflect.deleteProperty(base, prop) - }, - preventExtensions() { - // Keeping the surrogate extensible is required for ownKeys to report - // caller-owned base fields that may change over the carrier's lifetime. - return false - }, - setPrototypeOf() { - // The carrier prototype and base delegation must not be split. - return false - }, - apply(_target, thisArg, args) { - const callable = callableBase as (...values: unknown[]) => unknown - const result: unknown = Reflect.apply(callable, thisArg, args) - return result - }, - construct(_target, args, newTarget) { - const constructor = callableBase as unknown as new (...values: unknown[]) => object - const result: unknown = Reflect.construct( - constructor, - args, - newTarget === carrier ? constructor : newTarget, - ) - return result as object - }, - }) - return carrier as Scoped + carrierKeys.set(carrier, key) + return carrier as unknown as Scoped } /** - * Whether `value` is a carrier built by {@link scopeTarget} — the runtime - * counterpart of the {@link Scoped} brand, used by the dev invariants to - * assert that a scope-filtered event was dispatched with a carrier and not a - * bare subject. - * @param value - the dispatch `thisArg` to test. - * @returns true iff `value` came from {@link scopeTarget}. + * Test whether a value is a scope carrier. + * @param value - dispatch receiver to inspect. + * @returns whether {@link scopeTarget} created it. */ export function isScopeCarrier(value: unknown): value is Scoped { - if ((typeof value !== 'object' && typeof value !== 'function') || value === null) return false - // A property read checks the immutable marker owned by the surrogate target. - return (value as { [kCarrier]?: { key: ScopeKey | undefined } })[kCarrier] !== undefined + return typeof value === 'object' && value !== null && carrierKeys.has(value) } /** - * The scope key a carrier was built for — `undefined` for a subject-less - * carrier, and also `undefined` for a non-carrier (pair with - * {@link isScopeCarrier} when the distinction matters). The dev invariants - * use it to assert the carrier's key IS the subject the event's arguments - * name. - * @param value - the dispatch `thisArg` to read. - * @returns the `key` given to {@link scopeTarget}, or `undefined`. + * Read a carrier's routing key. + * @param value - dispatch receiver to inspect. + * @returns the carrier key, or `undefined` for an unkeyed/non-carrier value. */ export function carrierKeyOf(value: unknown): ScopeKey | undefined { if (!isScopeCarrier(value)) return undefined - // Optional-prop cast: the guard proves the mark is present at runtime, but - // the Scoped<> brand carries no structural kCarrier member to narrow from. - return (value as { [kCarrier]?: { key: ScopeKey | undefined } })[kCarrier]?.key -} - -/** - * A test/tooling host for minting scopes: one mounted plugin whose `inject` - * list is the service surface every scope minted through it can reach. - */ -export interface ScopeHost { - /** - * Mint a scope under the host (see {@link createScope}); the scoped context - * resolves exactly the host's injected services. - * @param key - the scope's identity ({@link ScopeKey}). - * @returns the minted scope. - */ - mint(key: ScopeKey): Scope - /** - * Dispose the host fiber and with it every scope minted through it. - * Every racing/repeat caller observes the same completion, including when a - * child's raw disposer started before host disposal. - * @returns resolves when the host and every minted scope have reached - * quiescence. - */ - dispose(): Promise -} - -/** - * Mount a scope-minting host plugin that injects `services`, THE sanctioned - * way to mint scopes in tests (production scopes are minted by the agent - * loop). Exists because the naive spelling fails confusingly twice over: - * a plugin with no `inject` mints scopes whose service reads throw Cordis's - * cryptic `cannot get property … without inject`, and a plugin whose inject - * can never be satisfied RESOLVES its fiber await without ever running the - * callback — a silent no-op host. This helper fails LOUD instead: when the - * callback did not run, it names the absent services and disposes the host. - * The service list is copied before plugin activation so caller mutation - * across the await cannot change dependency resolution or diagnostics. - * @param ctx - the context to mount the host under. - * @param services - the service names scopes minted through this host reach - * (the host plugin's `inject` list). - * @returns the host (mint scopes, dispose them all at once). - * @throws when any of `services` is not available on `ctx` — named, not the - * Cordis dead end. - */ -export async function scopeHost(ctx: Context, services: string[]): Promise { - // The inject list crosses an await before missing-service diagnostics run. - // Detach it now so caller mutation cannot change either Cordis dependency - // resolution or the names reported by this helper. - const requiredServices = [...services] - let hostCtx: Context | undefined - // A named function statement (not Object.assign({name}) — Function.name is - // read-only) so diagnostics read `scopeHost`. - function scopeHostPlugin(inner: Context): void { hostCtx = inner } - const fiber = ctx.plugin(Object.assign(scopeHostPlugin, { inject: requiredServices })) - await fiber - if (hostCtx === undefined) { - // Dependency-pending: cordis resolves the await without running the - // callback. Name the absentees and unwind the pending fiber. - const missing = requiredServices.filter(name => ctx.get(name) === undefined) - await fiber.dispose() - /* v8 ignore next -- the '(unknown)' fallback is defensive: a pending - * fiber with zero absent services cannot occur (an all-present inject - * list runs the callback) */ - const named = missing.map(name => `"${name}"`).join(', ') || '(unknown)' - throw new Error(`scopeHost: service${missing.length === 1 ? '' : 's'} ${named} not available on this context — load the providing plugin(s) before minting scopes`) - } - const host = hostCtx - const scopes = new Set() - let disposing: Promise | undefined - const dispose = async (): Promise => { - // Start every boundary before awaiting any one of them. A child whose raw - // disposer already ran is still followed through Scope.dispose(); a child - // the host unload claims first is followed through the same fiber inertia. - const tasks = [quiesceFiber(fiber), ...[...scopes].map(scope => scope.dispose())] - const results = await Promise.allSettled(tasks) - scopes.clear() - const errors = results.flatMap(result => result.status === 'rejected' ? [result.reason as unknown] : []) - if (errors.length === 1) throw errors[0] - if (errors.length > 1) throw new AggregateError(errors, 'scopeHost: disposal failed') - } - return { - mint: (key: ScopeKey) => { - const minted = createScope(host, key) - let disposing: Promise | undefined - const tracked: Scope = { - ctx: minted.ctx, - // Preserve the exact Cordis identity: only the public shared boundary - // is wrapped to retire this child from the host's tracking set. - rawDispose: minted.rawDispose, - dispose: () => (disposing ??= minted.dispose().finally(() => { scopes.delete(tracked) })), - } - scopes.add(tracked) - return tracked - }, - dispose: () => (disposing ??= dispose()), - } + return carrierKeys.get(value) } diff --git a/packages/core/scope/tests/scope.spec.ts b/packages/core/scope/tests/scope.spec.ts index 7242aa942d..0b7bbef348 100644 --- a/packages/core/scope/tests/scope.spec.ts +++ b/packages/core/scope/tests/scope.spec.ts @@ -1,569 +1,155 @@ import { describe, expect, expectTypeOf, it } from 'vitest' import { Context } from 'cordis' -import { carrierKeyOf, createScope, isScopeCarrier, scopeHost, scopeOf, scopeTarget } from '@deepseek-ai/dsh-scope' -import type { Scope, ScopeKey, Scoped } from '@deepseek-ai/dsh-scope' +import { carrierKeyOf, createScope, isScopeCarrier, scopeOf, scopeTarget } from '@deepseek-ai/dsh-scope' +import type { Scope, Scoped } from '@deepseek-ai/dsh-scope' declare module 'cordis' { interface Events { /** - * Test-only event for exercising scope-filtered dispatch. + * Test-only event for scope-filtered dispatch. * @param value - opaque payload recorded by listeners. * @mode emit */ 'scope-test/ping'(value: string): void - /** - * Test-only waterfall for exercising carrier `this` shape. - * @param value - seed value listeners may wrap. - * @mode waterfall - */ - 'scope-test/echo'(value: string, next: () => string): string } } -/** Mount a host plugin and mint a scope inside it, returning both. */ +/** Mount a host plugin and mint a scope inside it. */ async function mintScope(ctx: Context, key: object): Promise { let scope!: Scope - await ctx.plugin((inner: Context) => { - scope = createScope(inner, key) - }) + await ctx.plugin((inner: Context) => { scope = createScope(inner, key) }) return scope } describe('createScope', () => { - it('rejects primitive keys but accepts callable objects (matching ScopeKey)', async () => { + it('tags contexts and derived contexts, with the nearest tag winning', async () => { const ctx = new Context() - // Typed through `unknown` so the ScopeKey type cannot argue the assertion - // away: this test exercises exactly the callers the typechecker misses. - const badKeys: unknown[] = ['k', null] - for (const bad of badKeys) { - expect(() => createScope(ctx, bad as ScopeKey)).toThrow(/must be a non-null object or function/) - } + const outerKey = { name: 'outer' } + const innerKey = { name: 'inner' } + const outer = await mintScope(ctx, outerKey) + const inner = createScope(outer.ctx, innerKey) - const callable = Object.assign(() => {}, { nameForTest: 'callable-key' }) - const scope = await mintScope(ctx, callable) - expect(scopeOf(scope.ctx)).toBe(callable) - await scope.dispose() - }) - - it('tags the scoped context, readable through derivations (nearest tag wins)', async () => { - const ctx = new Context() - const key = { name: 'a' } - const inner = { name: 'a.inner' } - const scope = await mintScope(ctx, key) - - expect(scopeOf(scope.ctx)).toBe(key) - // An extend of the scoped context inherits the tag through the prototype chain. - expect(scopeOf(scope.ctx.extend({}))).toBe(key) - // A plain context carries no tag. expect(scopeOf(ctx)).toBeUndefined() - // A fiber mounted UNDER the scoped context reads as that scope… - let mountedCtx!: Context - await scope.ctx.plugin((c: Context) => { mountedCtx = c }) - expect(scopeOf(mountedCtx)).toBe(key) - // …and a nested scope shadows the outer tag (nearest wins). - const nested = createScope(scope.ctx, inner) - expect(scopeOf(nested.ctx)).toBe(inner) + expect(scopeOf(outer.ctx)).toBe(outerKey) + expect(scopeOf(outer.ctx.extend({}))).toBe(outerKey) + expect(scopeOf(inner.ctx)).toBe(innerKey) + + await inner.dispose() + await outer.dispose() }) - it('is usable synchronously: registrations land before the fiber activates', async () => { + it('is usable synchronously before the backing fiber activates', async () => { const ctx = new Context() const events: string[] = [] + let scope!: Scope await ctx.plugin((inner: Context) => { - const scope = createScope(inner, { name: 'sync' }) - // Same tick as createScope — no await between mint and use. - scope.ctx.effect(() => () => void events.push('effect-disposed')) - scope.ctx.on('scope-test/ping', value => void events.push(`heard:${value}`)) + scope = createScope(inner, { name: 'sync' }) + scope.ctx.effect(() => () => void events.push('disposed')) events.push('registered') }) - ctx.emit(scopeTarget(ctx, undefined), 'scope-test/ping', 'nobody') expect(events).toEqual(['registered']) - }) - - it('dispose() unwinds registrations, is idempotent, and inerts the context', async () => { - const ctx = new Context() - const scope = await mintScope(ctx, { name: 'd' }) - const order: string[] = [] - scope.ctx.effect(() => () => void order.push('a')) - scope.ctx.effect(() => () => void order.push('b')) - await scope.dispose() - expect(order).toEqual(['b', 'a']) // LIFO within the scope fiber - - // Repeat dispose: the underlying cordis disposer returns undefined; the - // wrapper still resolves. - await expect(scope.dispose()).resolves.toBeUndefined() - // Registration through a disposed scope throws INACTIVE_EFFECT. - expect(() => scope.ctx.effect(() => () => {})).toThrow(/inactive context/) + expect(events).toEqual(['registered', 'disposed']) }) - it('dispose() follows a rawDispose-first race through async quiescence', async () => { + it('shares quiescence across repeat and raw-disposer-first calls', async () => { const ctx = new Context() - const scope = await mintScope(ctx, { name: 'raw-first' }) + const scope = await mintScope(ctx, { name: 'quiescence' }) const gate = Promise.withResolvers() - let cleanupFinished = false + let finished = false scope.ctx.effect(() => async () => { await gate.promise - cleanupFinished = true + finished = true }) const raw = Promise.resolve(scope.rawDispose()) - let publicSettled = false - const publicDispose = scope.dispose().then(() => { publicSettled = true }) + const publicDispose = scope.dispose() await Promise.resolve() - expect(publicSettled).toBe(false) - expect(cleanupFinished).toBe(false) - + expect(finished).toBe(false) gate.resolve(undefined) - await Promise.all([raw, publicDispose]) - expect(cleanupFinished).toBe(true) - await expect(scope.dispose()).resolves.toBeUndefined() + await Promise.all([raw, publicDispose, scope.dispose()]) + expect(finished).toBe(true) }) - it('rawDispose is the exact cordis disposer: yielding it nests the scope at its position', async () => { + it('exposes the exact raw disposer for ordered composite teardown', async () => { const ctx = new Context() const order: string[] = [] - let composite!: () => Promise | void + let dispose!: () => Promise | void await ctx.plugin((inner: Context) => { - composite = inner.effect(function* () { - yield () => void order.push('outermost') // disposed LAST + dispose = inner.effect(function* () { + yield () => void order.push('outer') const scope = createScope(inner, { name: 'nested' }) - scope.ctx.effect(() => () => void order.push('scope-registration')) - yield scope.rawDispose // disposed SECOND — nested by identity - yield () => void order.push('innermost') // disposed FIRST + scope.ctx.effect(() => () => void order.push('scope')) + yield scope.rawDispose + yield () => void order.push('inner') }) }) - await composite() - // The scope disposed exactly at its yield position (between the two - // neighbours), not as a concurrent sibling of the composite. - expect(order).toEqual(['innermost', 'scope-registration', 'outermost']) + await dispose() + expect(order).toEqual(['inner', 'scope', 'outer']) }) }) -describe('scopeTarget dispatch filtering', () => { - it('scoped listeners hear only their key; untagged listeners hear everything', async () => { +describe('scopeTarget', () => { + it('routes scoped listeners by key while untagged listeners remain global', async () => { const ctx = new Context() const keyA = { name: 'A' } const keyB = { name: 'B' } const scopeA = await mintScope(ctx, keyA) const scopeB = await mintScope(ctx, keyB) - const heard: string[] = [] ctx.on('scope-test/ping', value => void heard.push(`global:${value}`)) scopeA.ctx.on('scope-test/ping', value => void heard.push(`A:${value}`)) scopeB.ctx.on('scope-test/ping', value => void heard.push(`B:${value}`)) - ctx.emit(scopeTarget(ctx, keyA), 'scope-test/ping', 'to-A') - ctx.emit(scopeTarget(ctx, keyB), 'scope-test/ping', 'to-B') - ctx.emit(scopeTarget(ctx, undefined), 'scope-test/ping', 'to-nobody') + ctx.emit(scopeTarget(ctx, keyA), 'scope-test/ping', 'a') + ctx.emit(scopeTarget(ctx, keyB), 'scope-test/ping', 'b') + ctx.emit(scopeTarget(ctx, undefined), 'scope-test/ping', 'none') - expect(heard).toEqual([ - 'global:to-A', 'A:to-A', - 'global:to-B', 'B:to-B', - 'global:to-nobody', - ]) + expect(heard).toEqual(['global:a', 'A:a', 'global:b', 'B:b', 'global:none']) + await Promise.all([scopeA.dispose(), scopeB.dispose()]) }) - it('{ global: true } listeners bypass scope filtering entirely', async () => { + it('preserves a base Cordis filter and its receiver', async () => { const ctx = new Context() - const keyA = { name: 'A' } - const scopeA = await mintScope(ctx, keyA) - const heard: string[] = [] - scopeA.ctx.on('scope-test/ping', value => void heard.push(`escape:${value}`), { global: true }) - - ctx.emit(scopeTarget(ctx, { name: 'other' }), 'scope-test/ping', 'foreign') - ctx.emit(scopeTarget(ctx, undefined), 'scope-test/ping', 'nobody') - expect(heard).toEqual(['escape:foreign', 'escape:nobody']) - }) - - it("composes the base's own Context.filter (a rejecting base filter wins)", async () => { - const ctx = new Context() - const keyA = { name: 'A' } - const scopeA = await mintScope(ctx, keyA) + const key = { name: 'A' } + const scope = await mintScope(ctx, key) const heard: string[] = [] ctx.on('scope-test/ping', value => void heard.push(`global:${value}`)) - scopeA.ctx.on('scope-test/ping', value => void heard.push(`A:${value}`)) - - // A base whose own filter rejects every listener context: nothing fires, - // scoped or not — the scope predicate never overrides the base's veto. - const vetoBase = { [Context.filter]: () => false } - ctx.emit(scopeTarget(vetoBase, keyA), 'scope-test/ping', 'vetoed') - expect(heard).toEqual([]) - - // A base whose filter accepts delegates to the scope predicate, with the - // real base preserved as its `this` receiver. - let baseReceiverWasOpen = false - const openBase = { + scope.ctx.on('scope-test/ping', value => void heard.push(`A:${value}`)) + let receiverMatches = false + const base = { [Context.filter](this: object): boolean { - baseReceiverWasOpen = this === openBase - return true + receiverMatches = this === base + return false }, } - ctx.emit(scopeTarget(openBase, keyA), 'scope-test/ping', 'open') - expect(heard).toEqual(['global:open', 'A:open']) - expect(baseReceiverWasOpen).toBe(true) - // A function's public `.call` property is not its invocation semantics. - // An always-true replacement must not override the base predicate's veto. - const tamperedVeto = (): boolean => false - Object.defineProperty(tamperedVeto, 'call', { value: () => true }) - ctx.emit(scopeTarget({ [Context.filter]: tamperedVeto }, keyA), 'scope-test/ping', 'tampered-veto') - expect(heard).toEqual(['global:open', 'A:open']) + ctx.emit(scopeTarget(base, key), 'scope-test/ping', 'vetoed') + expect(heard).toEqual([]) + expect(receiverMatches).toBe(true) + await scope.dispose() }) - it('pins the exposed composed filter invocation so a carrier holder cannot bypass isolation', async () => { + it('{ global: true } listeners retain Cordis global-listener semantics', async () => { const ctx = new Context() - const keyA = { name: 'A' } - const keyB = { name: 'B' } - const scopeA = await mintScope(ctx, keyA) - const scopeB = await mintScope(ctx, keyB) + const scope = await mintScope(ctx, { name: 'A' }) const heard: string[] = [] - ctx.on('scope-test/ping', value => void heard.push(`global:${value}`)) - scopeA.ctx.on('scope-test/ping', value => void heard.push(`A:${value}`)) - scopeB.ctx.on('scope-test/ping', value => void heard.push(`B:${value}`)) - - const carrier = scopeTarget(ctx, keyA) - const exposedFilter: unknown = Reflect.get(carrier, Context.filter) - expect(typeof exposedFilter).toBe('function') - const filter = exposedFilter as ((ctx: Context) => boolean) & { call: (...args: unknown[]) => unknown } - const primordialCall: unknown = Reflect.get(Function.prototype, 'call') - expect(Object.getOwnPropertyDescriptor(filter, 'call')).toMatchObject({ - value: primordialCall, - writable: false, - configurable: false, - }) - expect(Object.isFrozen(filter)).toBe(true) - expect(Reflect.set(filter, 'call', () => true)).toBe(false) - expect(Reflect.defineProperty(filter, 'call', { value: () => true })).toBe(false) - - ctx.emit(carrier, 'scope-test/ping', 'still-A-only') - expect(heard).toEqual(['global:still-A-only', 'A:still-A-only']) + scope.ctx.on('scope-test/ping', value => void heard.push(value), { global: true }) + ctx.emit(scopeTarget(ctx, { name: 'other' }), 'scope-test/ping', 'foreign') + ctx.emit(scopeTarget(ctx, undefined), 'scope-test/ping', 'none') + expect(heard).toEqual(['foreign', 'none']) + await scope.dispose() }) - it('keeps listener `this` base-shaped through the carrier (waterfall)', async () => { - const ctx = new Context() - const base = { label: 'the-base' } - let seenLabel: string | undefined - ctx.on('scope-test/echo', function (this: { label: string }, value, next) { - seenLabel = this.label - return `${next()}+${value}` - }) - const result = ctx.waterfall(scopeTarget(base, undefined), 'scope-test/echo', 'v', () => 'seed') - expect(result).toBe('seed+v') - expect(seenLabel).toBe('the-base') - }) - - it('is transparent for subjects with native #private fields: methods and getters through the carrier reach the real object', () => { - // The ds-review-bot regression: cordis hands the carrier to listeners as - // `this` (typed Scoped), so subject method calls through it are a - // supported shape. A proxy that delegates with the PROXY as receiver - // (cordis withProps) throws TypeError on any native #private the method - // or getter touches; the carrier must delegate with the BASE as receiver - // and bind retrieved methods to it. - class Subject { - #count = 0 - bump(): number { return ++this.#count } - get count(): number { return this.#count } - } - const subject = new Subject() - const carrier = scopeTarget(subject, subject) - expect(carrier.bump()).toBe(1) // method call: bound to the base - expect(subject.count).toBe(1) // ...and it mutated the REAL object - expect(carrier.count).toBe(1) // getter: runs with the base as receiver - // The get trap returns the method already bound to the base; - // detachability IS the assertion. - // eslint-disable-next-line @typescript-eslint/unbound-method - const detached = carrier.bump - expect(detached()).toBe(2) - }) - - it('delegates the ordinary reflective surface while keeping overlays immutable', () => { - const frozenFn = (): string => 'frozen' - const base: { mutable: number; pinned: () => string; toString: () => string } = { - mutable: 0, - pinned: frozenFn, - toString: () => 'base-str', - } - Object.defineProperty(base, 'pinned', { value: frozenFn, writable: false, configurable: false }) - const carrier = scopeTarget(base, undefined) - carrier.mutable = 7 - expect(base.mutable).toBe(7) // sets land on the base, not a detached overlay - // The surrogate target frees reads from the base property's proxy - // invariant, so even a frozen own method can be safely bound to the base. - expect(carrier.pinned).not.toBe(frozenFn) - expect(carrier.pinned()).toBe('frozen') - // The overlay literal inherits Object.prototype; hasOwn (not `in`) keeps - // it from shadowing the subject's own prototype-surface members. - expect(String(carrier)).toBe('base-str') - expect('mutable' in carrier).toBe(true) - expect(Object.hasOwn(carrier, 'mutable')).toBe(true) - expect(Object.keys(carrier)).toEqual(['mutable', 'pinned', 'toString']) - Object.defineProperty(carrier, 'extra', { value: 1, configurable: true }) - expect((base as typeof base & { extra?: number }).extra).toBe(1) - expect(delete (carrier as typeof carrier & { extra?: number }).extra).toBe(true) - - // A non-configurable property cannot be reflected truthfully through the - // extensible surrogate. Reject before mutating the delegated base; an - // omitted `configurable` has JavaScript's false default and is rejected too. - expect(Reflect.defineProperty(carrier, 'sealed', { value: 1, configurable: false })).toBe(false) - expect(Object.hasOwn(base, 'sealed')).toBe(false) - expect(Reflect.defineProperty(carrier, 'default-sealed', { value: 2 })).toBe(false) - expect(Object.hasOwn(base, 'default-sealed')).toBe(false) - expect(Reflect.preventExtensions(carrier)).toBe(false) - expect(Reflect.setPrototypeOf(carrier, null)).toBe(false) - }) - - it('keeps isolation when the base filter is pinned before, during, or after construction', async () => { - const ctx = new Context() - const keyA = { name: 'A' } - const keyB = { name: 'B' } - const scopeA = await mintScope(ctx, keyA) - const scopeB = await mintScope(ctx, keyB) - const heard: string[] = [] - ctx.on('scope-test/ping', value => void heard.push(`global:${value}`)) - scopeA.ctx.on('scope-test/ping', value => void heard.push(`A:${value}`)) - scopeB.ctx.on('scope-test/ping', value => void heard.push(`B:${value}`)) - const pinnedFilter = (): boolean => true - - const pinnedData = {} - Object.defineProperty(pinnedData, Context.filter, { - value: pinnedFilter, - writable: false, - configurable: false, - }) - const pinnedCarrier = scopeTarget(pinnedData, keyA) - ctx.emit(pinnedCarrier, 'scope-test/ping', 'before') - - const duringRead = {} - Object.defineProperty(duringRead, Context.filter, { - configurable: true, - get() { - Object.defineProperty(duringRead, Context.filter, { - value: pinnedFilter, - writable: false, - configurable: false, - }) - return pinnedFilter - }, - }) - ctx.emit(scopeTarget(duringRead, keyA), 'scope-test/ping', 'during') - - const pinnedAfter = { [Context.filter]: pinnedFilter } - const afterCarrier = scopeTarget(pinnedAfter, keyA) - Object.defineProperty(pinnedAfter, Context.filter, { - value: pinnedFilter, - writable: false, - configurable: false, - }) - ctx.emit(afterCarrier, 'scope-test/ping', 'after') - - const pinnedGetterless = {} - Object.defineProperty(pinnedGetterless, Context.filter, { set(_value: unknown) {}, configurable: false }) - ctx.emit(scopeTarget(pinnedGetterless, keyA), 'scope-test/ping', 'getterless') - - expect(heard).toEqual([ - 'global:before', 'A:before', - 'global:during', 'A:during', - 'global:after', 'A:after', - 'global:getterless', 'A:getterless', - ]) - expect((pinnedCarrier as Record)[Context.filter]).not.toBe(pinnedFilter) - expect(Reflect.set(pinnedCarrier, Context.filter, pinnedFilter)).toBe(false) - expect(Reflect.defineProperty(pinnedCarrier, Context.filter, { value: pinnedFilter })).toBe(false) - expect(Reflect.deleteProperty(pinnedCarrier, Context.filter)).toBe(false) - - expect(() => scopeTarget({ [Context.filter]: 1 }, { name: 'A' })).toThrow( - /Context\.filter must be a function/, - ) - }) - - it('preserves callable and constructable bases', () => { - function Subject(this: { value?: number }, value: number): number { - if (new.target) { - this.value = value - return value - } - return value * 2 - } - const carrier = scopeTarget(Subject as typeof Subject & (new (value: number) => { value: number }), { - name: 'callable', - }) - - const called: unknown = Reflect.apply(carrier, { value: 0 }, [3]) - expect(called).toBe(6) - const instance = new carrier(4) - expect(instance).toBeInstanceOf(Subject) - expect(instance.value).toBe(4) - const prototypeDescriptor = Object.getOwnPropertyDescriptor(carrier, 'prototype') - const subjectPrototype: unknown = Reflect.get(Subject, 'prototype') - expect(prototypeDescriptor?.configurable).toBe(true) - expect(prototypeDescriptor?.value).toBe(subjectPrototype) - class Derived extends carrier {} - const derived = new Derived(5) - expect(derived).toBeInstanceOf(Derived) - expect(derived).toBeInstanceOf(Subject) - expect(derived.value).toBe(5) - expect(isScopeCarrier(carrier)).toBe(true) - }) - - it('matches non-constructable and bound-constructor function shapes', () => { - const arrow = (value: number): number => value + 1 - const arrowCarrier = scopeTarget(arrow, { name: 'arrow' }) - const arrowResult: unknown = Reflect.apply(arrowCarrier, undefined, [2]) - expect(arrowResult).toBe(3) - expect('prototype' in arrowCarrier).toBe(false) - expect(Object.getOwnPropertyDescriptor(arrowCarrier, 'prototype')).toBeUndefined() - expect(() => { Reflect.construct(arrowCarrier, []) }).toThrow(TypeError) - - class Subject { - constructor(readonly value: number) {} - } - const bound = Subject.bind(undefined, 7) - const boundCarrier = scopeTarget(bound, { name: 'bound-constructor' }) - expect('prototype' in boundCarrier).toBe(false) - expect(Object.getOwnPropertyDescriptor(boundCarrier, 'prototype')).toBeUndefined() - const instance = new boundCarrier() - expect(instance).toBeInstanceOf(Subject) - expect(instance.value).toBe(7) - }) - - it('detects construction without reading a hostile base prototype', () => { - class Subject { - constructor(readonly value: number) {} - } - let prototypeReads = 0 - const hostile = new Proxy(Subject, { - get(target, prop, receiver) { - if (prop === 'prototype') { - prototypeReads += 1 - throw new Error('hostile prototype getter') - } - return Reflect.get(target, prop, receiver) as unknown - }, - }) - - const carrier = scopeTarget(hostile, { name: 'hostile-constructor' }) - expect(prototypeReads).toBe(0) - const instance: unknown = Reflect.construct(carrier, [9], Subject) - expect(instance).toBeInstanceOf(Subject) - expect(instance).toMatchObject({ value: 9 }) - expect(prototypeReads).toBe(0) - }) - - it('keeps the real constructor: class identity survives the carrier', () => { - class Subject { work(): string { return 'w' } } - const subject = new Subject() - const carrier = scopeTarget(subject, subject) - // `constructor` is looked up, never invoked as a subject method — binding - // it would break `carrier.constructor === Subject` for no benefit. - expect(carrier.constructor).toBe(Subject) - }) -}) - -describe('carrier marks', () => { - it('isScopeCarrier / carrierKeyOf distinguish carriers, keys, and bare subjects', () => { - const base = { name: 'base' } + it('uses an opaque branded carrier with a separately tracked key', () => { const key = { name: 'key' } - const keyed = scopeTarget(base, key) - const subjectless = scopeTarget(base, undefined) - - expect(isScopeCarrier(keyed)).toBe(true) - expect(carrierKeyOf(keyed)).toBe(key) - expect(isScopeCarrier(subjectless)).toBe(true) - expect(carrierKeyOf(subjectless)).toBeUndefined() - - expect(isScopeCarrier(base)).toBe(false) - expect(carrierKeyOf(base)).toBeUndefined() - expect(isScopeCarrier(null)).toBe(false) - expect(isScopeCarrier('x')).toBe(false) - }) - - it('brands the carrier type (compile-time)', () => { - const base = { name: 'base' } - const carrier = scopeTarget(base, undefined) - expectTypeOf(carrier).toExtend>() - // A bare subject is NOT assignable where a carrier is demanded. - expectTypeOf(base).not.toExtend>() - }) -}) - -describe('scopeHost', () => { - it('mints scopes that reach the injected services; dispose unwinds them all', async () => { - const ctx = new Context() - ctx.provide('answers', { value: 42 }) - const host = await scopeHost(ctx, ['answers']) - const scope = host.mint({ name: 'a' }) - expect((scope.ctx as Context & { answers: { value: number } }).answers.value).toBe(42) - const order: string[] = [] - scope.ctx.effect(() => () => void order.push('scoped-disposed')) - await host.dispose() - expect(order).toEqual(['scoped-disposed']) - expect(() => scope.ctx.effect(() => () => {})).toThrow(/inactive context/) - }) - - it('dispose waits for a child whose raw disposer won the race', async () => { - const ctx = new Context() - ctx.provide('answers', { value: 42 }) - const host = await scopeHost(ctx, ['answers']) - const scope = host.mint({ name: 'raw-first-child' }) - const gate = Promise.withResolvers() - let cleanupFinished = false - scope.ctx.effect(() => async () => { - await gate.promise - cleanupFinished = true - }) - - const raw = Promise.resolve(scope.rawDispose()) - let hostSettled = false - const hostDispose = host.dispose().then(() => { hostSettled = true }) - await Promise.resolve() - expect(hostSettled).toBe(false) - - gate.resolve(undefined) - await Promise.all([raw, hostDispose]) - expect(cleanupFinished).toBe(true) - await expect(host.dispose()).resolves.toBeUndefined() - }) - - it('reaches every child before surfacing one or multiple disposal failures', async () => { - const oneCtx = new Context() - oneCtx.provide('answers', { value: 42 }) - const oneHost = await scopeHost(oneCtx, ['answers']) - const one = oneHost.mint({ name: 'one' }) - one.dispose = () => Promise.reject(new Error('one failed')) - await expect(oneHost.dispose()).rejects.toThrow('one failed') - - const manyCtx = new Context() - manyCtx.provide('answers', { value: 42 }) - const manyHost = await scopeHost(manyCtx, ['answers']) - const a = manyHost.mint({ name: 'a' }) - const b = manyHost.mint({ name: 'b' }) - a.dispose = () => Promise.reject(new Error('a failed')) - b.dispose = () => Promise.reject(new Error('b failed')) - await expect(manyHost.dispose()).rejects.toMatchObject({ - name: 'AggregateError', - message: 'scopeHost: disposal failed', - errors: [expect.objectContaining({ message: 'a failed' }), expect.objectContaining({ message: 'b failed' })], - }) - }) - - it('fails LOUD naming absent services instead of resolving as a silent no-op host', async () => { - const ctx = new Context() - await expect(scopeHost(ctx, ['tools', 'systemPrompt'])) - .rejects.toThrow('scopeHost: services "tools", "systemPrompt" not available') - }) - - it('snapshots missing-service diagnostics across the host activation await', async () => { - const ctx = new Context() - const services = ['tools', 'systemPrompt'] - const pending = scopeHost(ctx, services) - services.splice(0) - - await expect(pending) - .rejects.toThrow('scopeHost: services "tools", "systemPrompt" not available') - }) - - it('names a single absent service in the singular', async () => { - const ctx = new Context() - await expect(scopeHost(ctx, ['tools'])).rejects.toThrow('scopeHost: service "tools" not available') + const subject = { value: 1 } + const carrier = scopeTarget(subject, key) + expect(isScopeCarrier(carrier)).toBe(true) + expect(carrierKeyOf(carrier)).toBe(key) + expect(isScopeCarrier(subject)).toBe(false) + expect(carrierKeyOf(subject)).toBeUndefined() + expect('value' in carrier).toBe(false) + expectTypeOf(carrier).toEqualTypeOf>() }) }) diff --git a/packages/core/session/README.md b/packages/core/session/README.md index 085ebfda06..97b18b0504 100644 --- a/packages/core/session/README.md +++ b/packages/core/session/README.md @@ -8,7 +8,7 @@ Creates and holds event-sourced `Session` instances. Persistence is intentionall ### Public API -- `ctx.sessions.create(id?: SessionId, options?: { seed?: SessionEvent[]; meta?: { cwd?: string; parentSession?: SessionId; createdAt?: number; seedLength?: number } }): Session` — Create a session. `options.seed` replays/forks an existing event log: the constructor reads each array entry once, then recursively validates and copies every nested value in one pass so validation and storage cannot observe different getter results or erase an exotic prototype before checking it. `options.meta` attaches creation metadata (validated absolute `cwd`, `parentSession` lineage, seed boundary) as the immutable `SessionHeader`: the store rejects an exotic metadata shell, reads every accepted field once, and constructs a detached, deep-frozen header. The store fills `version`/`id` and defaults `createdAt` to now; a caller reconstructing a persisted session passes the original `createdAt` and persisted `seedLength` to preserve them. Disposed with the calling fiber. +- `ctx.sessions.create(id?: SessionId, options?: { seed?: readonly SessionEvent[]; meta?: { cwd?: string; parentSession?: SessionId; createdAt?: number; seedLength?: number } }): Session` — Create a session. The persistence/replay seed and resulting header are validated, detached, and deep-frozen at this durable boundary. The store fills `version`/`id` and defaults `createdAt` to now; a persisted reconstruction supplies the original `createdAt` and `seedLength`. Disposed with the calling fiber. - `ctx.sessions.flush(session: Session): Promise` Dispatch the awaited `session/flush` durability checkpoint with the carrier captured at enter — THE flush entry point (the loop's turn-end checkpoint and idle injection call it; never dispatch a raw `ctx.parallel`). Every captured listener starts, the call waits for all of them to settle, and a failure rejects only after the other listeners finish. Rejects a prepared, detached, or stale same-id object instead of inventing a subject-less carrier. - `ctx.sessions.fork(source, boundary?, childSessionId?): Session` — Resolve a live session object or id, select a seed through the inclusive `boundary` event seq (default: current last event), require that boundary to be `turn/end`, and create a live child session with lineage metadata. - `ctx.sessions.get(id: SessionId): Session | undefined` @@ -18,28 +18,27 @@ Creates and holds event-sourced `Session` instances. Persistence is intentionall `create()` covers the common case (the session is owned by the calling fiber). When a session must be torn down **in order with another resource** — so a final flush is captured before the store attachment and publication hooks are removed — `create()`'s self-contained effect is wrong, because a fiber unload disposes sibling effects *concurrently*. For that, split the lifecycle and fold it into the owner's single effect: -- `ctx.sessions.prepare(id?, options?): Session` — read `options.seed`/`options.meta` once, validate and detach the metadata/header, and construct the `Session` WITHOUT entering it into the store. Same options as `create`. -- `ctx.sessions.reserve(id): SessionRegistrationReservation` — hold an unpublished id under the calling fiber and construct its one owned Session through `reservation.prepare(options?)`. `release` is the exact owner effect disposer, so the agent lifecycle can adopt it and keep the ID reserved until scope cleanup quiesces. Until that release, bare `prepare`/`create`/`enter` calls for the id reject; the factory later presents the exact capability to `enter`, making setup-time publication structurally impossible without leaking an abandoned reservation across HMR disposal. -- `ctx.sessions.enter(session, reservation?): () => void` — claim the ID across caller-controlled filter/carrier evaluation, then install the module-private append publication hooks and add the exact session under its accepted key; a reentrant same-ID entry cannot be overwritten. Returns the idempotent, exact-object-guarded DETACH disposer, which clears publication, carrier, and accepted-key state without letting a stale capability delete a replacement. Does NOT emit `session/created` (the caller installs the disposer first, then calls `announce`, so a throwing listener rolls the attach back). It re-checks the id because public `prepare`/`enter` calls may be interleaved. A factory passes the opaque capability from `reserve(id)` so setup cannot enter the reserved session or publish a same-id replacement before the owning transaction. +- `ctx.sessions.prepare(id?, options?): Session` — validate durable seed/header data and construct the `Session` WITHOUT entering it into the store. Same options as `create`. +- `ctx.sessions.enter(session): () => void` — perform the authoritative ID collision check, install append publication state, and insert the exact session without announcing it. Returns an idempotent detach bound to the captured entry object, so a stale disposer cannot remove a later same-ID replacement. Concurrent same-ID preparation is allowed; only one final entry succeeds. - `ctx.sessions.announce(session): void` — begin the one allowed `session/created` announcement for an entered session; repeat and reentrant calls reject before dispatch. A detach requested synchronously by a creation listener is deferred until that dispatch unwinds, so another creation listener cannot observe `session/disposed` before its own `session/created` callback. Detach emits `session/disposed` exactly once, including rollback after a partially delivered creation notification; a never-announced entry emits neither edge. `dsh-agent-loop` is the canonical consumer: after unpublished agent setup it enters both session and agent before announcing either, then nests loop stop, agent removal, session detach, and scope unwind in one ordered lifecycle. The final flush therefore settles before this package detaches the session, whether teardown starts from an `AgentHandle` or owner-fiber unload. ### Live service events -The store pairs announced creation with disposal, publishes each append, and provides an awaited durability checkpoint. Before the log push it resolves the exact scoped `session/event` callback list, including development-time internal dispatch checks; substitution of the accepted session/event tuple rejects while the log is unchanged. The push is then the commit point, and callback throws or returned-promise rejections are logged and contained per observer. A committed append therefore returns normally, later observers still run, and teardown cannot interrupt an in-flight acceptance/publication boundary. Exact `session/*` signatures, modes, and scope-carrier behavior live in the generated [Cordis event catalog](../../../docs/cordis-catalog/events.md); the append-only payload vocabulary is separately generated into the [persistence catalog](../../../docs/persistence-catalog.md). Persistence consumers write behind from the append notification and drain on the store-owned flush entry point rather than dispatching the event directly. +The store pairs announced creation with disposal, publishes each append, and provides an awaited durability checkpoint. Before the log push it resolves the scoped `session/event` callback list. The push is the commit point; callback throws or returned-promise rejections are logged and contained per observer. A committed append therefore returns normally, later observers still run, and detach waits until publication unwinds. Exact `session/*` signatures, modes, and scope-carrier behavior live in the generated [Cordis event catalog](../../../docs/cordis-catalog/events.md); the append-only payload vocabulary is separately generated into the [persistence catalog](../../../docs/persistence-catalog.md). ### Class: `Session` Plain class (not a Cordis Service). Create via `ctx.sessions.create()`. -- `session.append(type, data, opts?): SessionEvent` — synchronous, never blocks on I/O. **Throws** if `data` or surface metadata is not losslessly JSON-serializable (BigInt, function, symbol, undefined, `-0`, non-finite number, circular ref, or an exotic object like Map/Set/Date/class instance). One recursive validate-and-copy pass reads each nested value exactly once and produces the detached value that enters the log, so validation and durability cannot diverge through a stateful getter or a prototype-erasing clone. The accepted event and every nested value are deep-frozen before publication; the returned event and observer notification share that immutable owned record. An entered session pins its attachment from materialization through observer delivery, rejects if a caller getter changes that attachment, and rejects a reentrant append until the outer callback list drains; these rules prevent an event from bypassing persistence or being delivered out of log order. The log push is the commit point: a synchronous observer throw or returned-promise rejection is logged per observer and cannot turn the committed append into a caller-visible failure or starve later observers. A third parameter `opts: SurfaceIntent` carries surface metadata: `surfaceOp` and `sourceEventSeqs` are each read once, then the former controls how the event enters the surface linked list and the latter records provenance. Runtime validation accepts only `'append'` or the exact `{ op: 'replace', start, end }` record with non-negative safe-integer bounds, and provenance must be an array of non-negative safe integers; non-surface events reject either field. The marker is **required** for the five `SurfaceEventType` events (every message-producing event must declare how it joins the surface) and rejected by the compiler for non-surface types. The contract is enforced two ways: the typed overload handles a specific event literal, AND runtime checks cover widened unions and raw seed/load logs so invalid metadata can never silently enter or disappear from `deriveMessages()`. -- `session.deriveMessages(): Message[]` — the LLM message history, CACHED: each surface node is projected exactly once, when first seen (O(new nodes) per call; a surface rewrite rebuilds via `surface.replaceGeneration`). Returns a fresh array snapshot per call over SHARED, deep-frozen `Message` objects — cloned once off the log at projection time, so a consumer can never mutate logged data (mutation throws). The surface is the single source of derived history — there is no raw-log fallback. -- `session.deriveEventMessage(event): Message | null` — the per-event projection `deriveMessages()` folds: one event's derived message (an unfrozen clone), or `null` when it produces none (a non-surface event, or an empty-content `assistant/message` hosting only usage). External reconstructors and the dev invariant fold the same function over a log prefix's surface, so no two paths can disagree about what a request's messages were (the reconstructability RFC). +- `session.append(type, data, opts?): SessionEvent` — synchronous, never blocks on I/O. At this durable boundary, data and surface metadata are lossless-JSON snapshotted and deep-frozen. For an attached session, a reentrant append during dispatch/observer publication rejects, and detach waits for that publication to unwind. Callbacks resolve before the log push; the push is the commit point, after which each observer failure is contained independently. Runtime surface validation covers widened unions and raw seed/load logs. +- `session.deriveMessages(): Message[]` — the LLM message history, CACHED: each surface node is projected exactly once, when first seen (O(new nodes) per call; a surface rewrite rebuilds via `surface.replaceGeneration`). Returns a fresh array per call over shared, deep-frozen `Message` objects. Each projection reuses the already deep-frozen content in its durable log event, so no second deep clone is needed and a consumer still cannot mutate logged data. The surface is the single source of derived history — there is no raw-log fallback. +- `session.deriveEventMessage(event): Message | null` — the per-event projection `deriveMessages()` folds: a fresh message wrapper that reuses the event's already frozen content, or `null` when the event produces none (a non-surface event, or an empty-content `assistant/message` hosting only usage). External reconstructors and the dev invariant fold the same function over a log prefix's surface, so no two paths can disagree about what a request's messages were (the reconstructability RFC). - `session.surface: SurfaceManager` — the derived surface, lazily rebuilt from `surfaceOp` markers in the log. Processes only new events (delta) on each access — the log is append-only, so prior events never change. `surface.replaceGeneration` is the rewrite signal: bumped by every folded `replace` and by `invalidate()`, never reset, so an incremental consumer comparing generations cannot be fooled. - `session.events` — a cached, frozen array snapshot over deep-frozen events. Repeated reads without an append return the same array; an append invalidates the cache and the next read returns a new snapshot, while earlier snapshots stay unchanged. Neither a cast nor a retained reference can push into the live log or rewrite an accepted event. -- `session.seq`, `session.id` — `id` is a non-writable, non-configurable runtime identity slot, not merely TypeScript-readonly. -- `session.header: SessionHeader` — detached, deep-frozen creation metadata (`version`, `id`, `createdAt`, optional `cwd`/`parentSession`/`seedLength`) published through a non-writable, non-configurable slot. Construction validates its lossless-JSON shape and requires the header id to match `session.id`, so a caller cannot later replace or mutate persistence routing or lineage. Kept out of the event log (a storage concern, not replayable state); a minimal header (stamped with the current `SESSION_FORMAT_VERSION`) is synthesized for bare `Session` construction. +- `session.seq`, `session.id` — current sequence and readonly typed identity. +- `session.header: SessionHeader` — detached, deep-frozen creation metadata (`version`, `id`, `createdAt`, optional `cwd`/`parentSession`/`seedLength`). Construction validates the durable record and requires its id to match `session.id`. ### Lossless JSON utilities diff --git a/packages/core/session/src/index.ts b/packages/core/session/src/index.ts index 3cd1c61451..6ce38bbffe 100644 --- a/packages/core/session/src/index.ts +++ b/packages/core/session/src/index.ts @@ -121,106 +121,40 @@ function renderTagged(tag: string, content: ContentBlock[], source: MessageSourc ] } -/** Reject a record shell that cloning or spreading would otherwise sanitize. */ -function assertPlainRecord(value: unknown, label: string): asserts value is Record { - if (value === null || typeof value !== 'object') { - throw new Error(`${label} is not a plain JSON record`) - } - const prototype = Object.getPrototypeOf(value) as unknown - if (prototype !== Object.prototype && prototype !== null) { - throw new Error(`${label} is not a plain JSON record`) - } -} - -/** Capture and validate the caller-owned fields that become a session header. */ -function snapshotSessionMeta(source: CreateSessionOptions['meta']): NonNullable { - if (source === undefined) return {} - assertPlainRecord(source, 'session metadata') - - // Read each accepted field exactly once. The metadata vocabulary is scalar, - // so this plain record is already detached from the caller; cloning the - // caller's shell first would erase a class prototype before validation. - const cwd = source.cwd - const parentSession = source.parentSession - const createdAt = source.createdAt - const seedLength = source.seedLength - const accepted = { - ...cwd !== undefined ? { cwd } : {}, - ...parentSession !== undefined ? { parentSession } : {}, - ...createdAt !== undefined ? { createdAt } : {}, - ...seedLength !== undefined ? { seedLength } : {}, - } - const snapshot = snapshotJsonValue(accepted) - if (snapshot === undefined) throw new Error('session metadata is not losslessly JSON-serializable') - if (snapshot.cwd !== undefined) { - if (typeof snapshot.cwd !== 'string') throw new Error('session cwd must be a string') - if (!isAbsolute(snapshot.cwd)) { - throw new Error(`session cwd must be an absolute path, got "${snapshot.cwd}"`) - } - } - if (snapshot.parentSession !== undefined && typeof snapshot.parentSession !== 'string') { - throw new Error('session parentSession must be a string') - } - if (snapshot.createdAt !== undefined - && (typeof snapshot.createdAt !== 'number' || !Number.isFinite(snapshot.createdAt))) { - throw new Error('session createdAt must be a finite number') - } - if (snapshot.seedLength !== undefined - && (typeof snapshot.seedLength !== 'number' || !Number.isSafeInteger(snapshot.seedLength) || snapshot.seedLength < 0)) { - throw new Error('session seedLength must be a non-negative safe integer') - } - return snapshot -} - /** Detach, validate, and freeze the creation metadata published by a session. */ function snapshotSessionHeader(id: SessionId, source?: SessionHeader): SessionHeader { - const input: SessionHeader = source === undefined + const input: unknown = source === undefined ? { version: SESSION_FORMAT_VERSION, id, createdAt: Date.now() } : source - assertPlainRecord(input, 'session header') - - // Capture each property once before validation. A stateful accessor therefore - // cannot present one identity or storage location to a check and publish a - // different one afterward. - const version = input.version - const headerId = input.id - const createdAt = input.createdAt - const cwd = input.cwd - const parentSession = input.parentSession - const seedLength = input.seedLength - const accepted = { - version, - id: headerId, - createdAt, - ...cwd !== undefined ? { cwd } : {}, - ...parentSession !== undefined ? { parentSession } : {}, - ...seedLength !== undefined ? { seedLength } : {}, - } - const snapshot = snapshotJsonValue(accepted) + const snapshot = snapshotJsonValue(input) if (snapshot === undefined) throw new Error('session header is not losslessly JSON-serializable') - if (snapshot.version !== SESSION_FORMAT_VERSION) { - throw new Error(`session header version must be ${SESSION_FORMAT_VERSION}, got ${String(snapshot.version)}`) + if (snapshot === null || typeof snapshot !== 'object' || Array.isArray(snapshot)) { + throw new Error('session header is not a plain JSON record') } - if (snapshot.id !== id) { - throw new Error(`session header id "${String(snapshot.id)}" does not match session id "${id}"`) + const record = snapshot as Record + if (record.version !== SESSION_FORMAT_VERSION) { + throw new Error(`session header version must be ${SESSION_FORMAT_VERSION}, got ${String(record.version)}`) } - if (typeof snapshot.createdAt !== 'number' || !Number.isFinite(snapshot.createdAt)) { + if (record.id !== id) { + throw new Error(`session header id "${String(record.id)}" does not match session id "${id}"`) + } + if (typeof record.createdAt !== 'number' || !Number.isFinite(record.createdAt)) { throw new Error('session header createdAt must be a finite number') } - if (snapshot.cwd !== undefined) { - if (typeof snapshot.cwd !== 'string') throw new Error('session header cwd must be a string') - if (!isAbsolute(snapshot.cwd)) { - throw new Error(`session header cwd must be an absolute path, got "${snapshot.cwd}"`) + if (record.cwd !== undefined) { + if (typeof record.cwd !== 'string') throw new Error('session header cwd must be a string') + if (!isAbsolute(record.cwd)) { + throw new Error(`session header cwd must be an absolute path, got "${record.cwd}"`) } } - if (snapshot.parentSession !== undefined && typeof snapshot.parentSession !== 'string') { + if (record.parentSession !== undefined && typeof record.parentSession !== 'string') { throw new Error('session header parentSession must be a string') } - if (snapshot.seedLength !== undefined - && (typeof snapshot.seedLength !== 'number' || !Number.isSafeInteger(snapshot.seedLength) || snapshot.seedLength < 0)) { + if (record.seedLength !== undefined + && (typeof record.seedLength !== 'number' || !Number.isSafeInteger(record.seedLength) || record.seedLength < 0)) { throw new Error('session header seedLength must be a non-negative safe integer') } - return deepFreeze(snapshot) + return deepFreeze(record as unknown as SessionHeader) } /** Validate the runtime shape of surface metadata after its JSON snapshot. */ @@ -275,25 +209,6 @@ function assertSessionEventEnvelope(value: Record, index: numbe } } -/** Render an arbitrary thrown value without allowing coercion to throw again. */ -function renderThrown(value: unknown): string { - try { - return value instanceof Error ? `${value.name}: ${value.message}` : String(value) - } catch { - return '' - } -} - -/** Best-effort reporting that cannot re-expose an already-contained failure. */ -function warnContained(ctx: Context, message: string): void { - try { - ctx.logger.warn(message) - } catch { - // contained: logger failure must not turn an observe-only callback failure - // back into a caller-visible error or an unhandled promise rejection. - } -} - type SessionCallback = (...args: unknown[]) => unknown /** Resolve one listener snapshot, including Cordis's internal dispatch checks. */ @@ -301,13 +216,6 @@ function collectSessionCallbacks(ctx: Context, args: unknown[]): SessionCallback return [...ctx.events.dispatch('emit', args)] as SessionCallback[] } -/** Reject pre-commit dispatch instrumentation that substituted accepted values. */ -function assertDispatchTuple(name: string, actual: unknown[], expected: unknown[]): void { - if (actual.length !== expected.length || actual.some((value, index) => value !== expected[index])) { - throw new Error(`${name} internal dispatch replaced the accepted callback tuple`) - } -} - /** Invoke one resolved observe-only listener snapshot with per-listener containment. */ function invokeContainedSessionObservers( ctx: Context, @@ -320,26 +228,29 @@ function invokeContainedSessionObservers( try { const returned: unknown = callback(...args) void Promise.resolve(returned).catch((error: unknown) => { - warnContained(ctx, `session "${id}": ${name} listener rejected: ${renderThrown(error)}`) + ctx.logger.warn(`session "${id}": ${name} listener rejected: ${String(error)}`) }) } catch (error: unknown) { - warnContained(ctx, `session "${id}": ${name} listener threw: ${renderThrown(error)}`) + ctx.logger.warn(`session "${id}": ${name} listener threw: ${String(error)}`) } } } -interface SessionAppendHooks { - /** Keep the store attachment live through acceptance and publication. */ - begin(): void - /** Resolve the exact observer list before commit; returns its contained publisher. */ - prepareObservation(event: SessionEvent): () => void - /** Release the attachment barrier and honor a deferred detach. */ - end(): void +/** All mutable lifecycle state for one exact store entry. */ +interface SessionEntry { + readonly id: SessionId + readonly session: Session + readonly carrier: Scoped + readonly emitCtx: Context + announced: boolean + announcing: boolean + appending: boolean + detachRequested: boolean + detach(): void } -const appendHooks = new WeakMap() -/** Identity token replaced on every store attachment or detachment. */ -const attachmentEpochs = new WeakMap() +/** Store attachment for the append path; module-private to keep Session store-agnostic publicly. */ +const attachments = new WeakMap() /** * An event-sourced session: an append-only log of {@link SessionEvent}s. @@ -349,8 +260,6 @@ const attachmentEpochs = new WeakMap() */ export class Session { private log: SessionEvent[] = [] - /** True throughout one event's materialization, validation, commit, and publication. */ - private appendInProgress = false /** * Derived surface — a cached linked list of message-producing events. @@ -377,7 +286,7 @@ export class Session { */ readonly header: SessionHeader - constructor(public readonly id: SessionId, seed?: SessionEvent[], header?: SessionHeader) { + constructor(public readonly id: SessionId, seed?: readonly SessionEvent[], header?: SessionHeader) { if (seed) { // Validate the seed to the SAME invariants `append` enforces, so a // replay/fork (`ctx.sessions.create(id, { seed })`) cannot construct a @@ -387,20 +296,9 @@ export class Session { // a bad seed would surface only later as a backend rejection or a silent // divergence between the live log and disk. this.log = Array.from(seed, (source, index) => { - // Spreading would erase a class instance's prototype. Reject an exotic - // event shell before that normalization can turn it into an apparently - // valid plain record; field values are still captured by the one spread - // below, so their accessors are not read twice. - assertPlainRecord(source, `seed event at index ${index}`) - // Read every enumerable event field once. Validation and snapshot - // construction must consume this same captured record: a stateful seed - // index or event getter cannot present one record to the checks and - // another to the durable log. - const event = { ...source } - // Materialize the complete accepted record in one recursive pass. A - // validate-then-structuredClone sequence would reread nested getters and - // could sanitize a class instance returned only to the clone. - const snapshot = snapshotJsonValue(event) + // The seed is a persistence/replay boundary: validate and detach the + // complete event in one lossless-JSON pass. + const snapshot = snapshotJsonValue(source) if (snapshot === undefined) { throw new Error(`seed event at index ${index} is not losslessly JSON-serializable`) } @@ -424,14 +322,6 @@ export class Session { }) } this.header = snapshotSessionHeader(id, header) - // TypeScript readonly prevents ordinary typed assignment only. Pin both - // public identity bindings at runtime too: setup/plugins receive the live - // Session object, and replacing either slot would split registry keys, - // persistence routing, and the already-validated header. - Object.defineProperties(this, { - id: { value: id, enumerable: true, writable: false, configurable: false }, - header: { value: this.header, enumerable: true, writable: false, configurable: false }, - }) } /** Cached immutable public snapshot of the private append-only log. */ @@ -490,88 +380,53 @@ export class Session { data: SessionEventMap[T], ...opts: T extends SurfaceEventType ? [opts: SurfaceIntent] : [] ): SessionEvent { - if (typeof type !== 'string') { - throw new TypeError('session event type must be a string') + const surfaceOpts: SurfaceIntent | undefined = opts[0] + const surfaceMetadata = { + ...surfaceOpts?.sourceEventSeqs === undefined ? {} : { sourceEventSeqs: surfaceOpts.sourceEventSeqs }, + ...surfaceOpts?.surfaceOp === undefined ? {} : { surfaceOp: surfaceOpts.surfaceOp }, } - if (this.appendInProgress) { - throw new Error('session append cannot reenter while another append is being accepted or published') + const dataSnapshot = snapshotJsonValue(data) + if (dataSnapshot === undefined) { + throw new Error(`session event "${type}" carries non-JSON-serializable data`) } - const hooks = appendHooks.get(this) - const attachmentEpoch = attachmentEpochs.get(this) - this.appendInProgress = true + const surfaceMetadataSnapshot = snapshotJsonValue(surfaceMetadata) + if (surfaceMetadataSnapshot === undefined) { + throw new Error(`session event "${type}" carries non-JSON-serializable surface metadata`) + } + assertSurfaceMetadataShape( + type, + (surfaceMetadataSnapshot as { surfaceOp?: unknown }).surfaceOp, + (surfaceMetadataSnapshot as { sourceEventSeqs?: unknown }).sourceEventSeqs, + ) + + const entry = attachments.get(this) + if (entry?.appending) { + throw new Error('session append cannot reenter while another append is being published') + } + if (entry !== undefined) entry.appending = true try { - // Start before reading caller-owned fields: a getter may request detach - // or try to append reentrantly. The attachment and sequence boundary stay - // stable until this exact acceptance attempt has either failed or reached - // every post-commit observer. - hooks?.begin() - const surfaceOpts: SurfaceIntent | undefined = opts[0] - const sourceEventSeqs = surfaceOpts?.sourceEventSeqs - const surfaceOp = surfaceOpts?.surfaceOp - // Surface-eligible events MUST carry a surfaceOp marker — the surface is the - // sole source of derived history, so a marker-less message event would be - // logged yet vanish from deriveMessages(). The typed `opts` overload makes - // the marker mandatory only when `T` is a SPECIFIC SurfaceEventType literal; - // when `T` widens to the SessionEventType union (a caller iterating raw - // events: `for (const e of log) append(e.type, e.data)`), the conditional - // rest collapses to optional and the compiler stops enforcing it. Re-check - // at runtime so that loophole can't silently drop history. - const surfaceMetadata = { - ...sourceEventSeqs !== undefined ? { sourceEventSeqs } : {}, - ...surfaceOp !== undefined ? { surfaceOp } : {}, - } - // The caller still owns the data and metadata objects and could mutate them - // after append. Materialize each accepted value exactly once while checking - // its JSON vocabulary, so the log cannot drift and a stateful getter cannot - // show one value to validation and another to a prototype-erasing clone. The - // returned event carries these SAME snapshots. - // - // Surface metadata accessors are read once into one plain record; the - // recursive snapshot then reads each nested value once as it copies it. - // Build the event shape with conditional surface fields via spreading. - // The result is cast through `unknown` because the conditional spreads - // produce an intersection type that the assignability checker can't - // narrow to a specific discriminated-union member when T is generic. - // This is a safe internal boundary: data and surface metadata are - // materialized below before the event enters the log. - const dataSnapshot = snapshotJsonValue(data) - if (dataSnapshot === undefined) { - throw new Error(`session event "${type}" carries non-JSON-serializable data`) - } - const surfaceMetadataSnapshot = snapshotJsonValue(surfaceMetadata) - if (surfaceMetadataSnapshot === undefined) { - throw new Error(`session event "${type}" carries non-JSON-serializable surface metadata`) - } - assertSurfaceMetadataShape( - type, - (surfaceMetadataSnapshot as { surfaceOp?: unknown }).surfaceOp, - (surfaceMetadataSnapshot as { sourceEventSeqs?: unknown }).sourceEventSeqs, - ) - if (appendHooks.get(this) !== hooks || attachmentEpochs.get(this) !== attachmentEpoch) { - throw new Error('session attachment changed while append input was being accepted') - } - const event = { + const event = deepFreeze({ type, seq: this.log.length, time: Date.now(), data: dataSnapshot, ...surfaceMetadataSnapshot, - } as unknown as SessionEvent - const acceptedEvent = deepFreeze(event) - // Resolve dispatch before the log push. Cordis runs internal/dispatch - // while producing this list; if instrumentation rejects the carrier, the - // append still fails before commit. The resolved callbacks themselves are - // observe-only and run with per-listener containment after the push. - const publish = hooks?.prepareObservation(acceptedEvent as unknown as SessionEvent) - this.log.push(acceptedEvent as unknown as SessionEvent) + } as unknown as SessionEvent) + let callbacks: SessionCallback[] | undefined + const callbackArgs: unknown[] = [this, event] + if (entry !== undefined) { + callbacks = collectSessionCallbacks(entry.emitCtx, [entry.carrier, 'session/event', ...callbackArgs]) + } + this.log.push(event as SessionEvent) this.eventsSnapshot = undefined - publish?.() - return acceptedEvent + if (callbacks !== undefined && entry !== undefined) { + invokeContainedSessionObservers(entry.emitCtx, 'session/event', entry.id, callbackArgs, callbacks) + } + return event } finally { - try { - hooks?.end() - } finally { - this.appendInProgress = false + if (entry !== undefined) { + entry.appending = false + if (entry.detachRequested && !entry.announcing) entry.detach() } } } @@ -621,10 +476,9 @@ export class Session { * call costs O(new nodes), and a surface rewrite (a `replace`; * {@link SurfaceManager.replaceGeneration}) rebuilds. The returned array is * a fresh snapshot per call (later appends never grow an array a caller - * already holds); the `Message` objects in it are SHARED and **deep-frozen** - * — cloned once off the log at projection time, so consumers can never - * mutate logged data, and mutation attempts throw instead of silently - * diverging replay from history. + * already holds); the `Message` objects in it are SHARED and **deep-frozen**. + * Their content reuses the already frozen durable event data, so the cache + * needs no second deep clone and consumers still cannot mutate the log. * @returns a fresh array of the shared, frozen derived history. */ deriveMessages(): Message[] { @@ -656,9 +510,10 @@ export class Session { * The per-node pure function {@link deriveMessages} folds over the surface; * an external reconstructor (or the dev invariant) folds the same function * over a log prefix's surface to rebuild the exact messages any request was - * built from (the reconstructability RFC). The returned `content` is - * deep-cloned off the logged event: the log is append-only by contract, so - * no live reference to logged data leaves this boundary. + * built from (the reconstructability RFC). The returned message wrapper is + * fresh; its content reuses the logged event's already deep-frozen durable + * data, so changing the wrapper cannot rewrite the log and changing content + * throws. * @param event - the event to project. * @returns the derived message, or null when the event produces none. */ @@ -669,29 +524,29 @@ export class Session { switch (event.type) { case 'user/message': { - return { role: 'user', content: structuredClone(event.data.content) } + return { role: 'user', content: event.data.content } } case 'assistant/message': { // Skip an empty-content assistant/message: it exists only to host a // max-tokens step's usage and must not inject a content-less assistant // turn into the provider transcript. if (event.data.content.length === 0) return null - return { role: 'assistant', content: structuredClone(event.data.content) } + return { role: 'assistant', content: event.data.content } } case 'tool/result': { const { callId, content, isError } = event.data return { role: 'user', - content: [{ type: 'tool-result', toolCallId: callId, content: structuredClone(content), isError }], + content: [{ type: 'tool-result', toolCallId: callId, content, isError }], } } case 'context/message': { const { content, source } = event.data - return { role: 'user', content: renderTagged('context', structuredClone(content), source) } + return { role: 'user', content: renderTagged('context', content, source) } } case 'steering/message': { const { content, source } = event.data - return { role: 'user', content: renderTagged('steering', structuredClone(content), source) } + return { role: 'user', content: renderTagged('steering', content, source) } } default: // A non-surface event (boundary, chunk, log-only record) projects to @@ -727,31 +582,6 @@ export class SessionForkError extends Error { } } -/** - * Unforgeable ownership handle for one unpublished session id. A factory keeps - * this capability across load/setup, preventing setup code from entering the - * prepared Session or publishing a replacement under the same id. Obtain it - * only from {@link SessionStore.reserve}. - */ -export interface SessionRegistrationReservation { - /** The reserved store id. */ - readonly id: SessionId - /** - * Construct the one Session owned by this reservation. - * @param options - seed events and creation metadata. - * @returns the still-unpublished Session. - */ - prepare(options?: CreateSessionOptions): Session - /** - * Release the unpublished reservation; idempotent. The store also releases - * it automatically when the fiber that called `reserve` disposes. This - * function is that exact Cordis effect disposer, so an ordered lifecycle may - * yield it by identity and place release after quiescence. - * @returns nothing. - */ - release(): void -} - /** * In-memory session store (`ctx.sessions`). * @@ -759,80 +589,13 @@ export interface SessionRegistrationReservation { * subscribe to `session/event` and flush on `session/flush` / dispose. */ export class SessionStore extends Service { - private store = new Map() - /** Ids claimed across caller-code boundaries before their exact entry commits. */ - private enteringIds = new Set() - /** The one accepted map key for each live session; never reread caller state. */ - private acceptedIds = new WeakMap() - /** Sessions whose creation announcement began and therefore require a pair. */ - private announced = new WeakSet() - /** Entries currently dispatching `session/created`; detach waits for dispatch to unwind. */ - private announcing = new WeakSet() - /** Entries accepting or publishing an append; detach waits for the boundary to unwind. */ - private appending = new WeakSet() - /** A detach requested reentrantly from creation or append publication. */ - private pendingDetach = new WeakSet() - /** Unpublished identities held across factory load/setup transactions. */ - private reservations = new Map() - /** The exact prepared object owned by each reservation capability. */ - private reservedSessions = new WeakMap() - /** - * Each live session's dispatch carrier, captured at {@link enter} from the - * ENTERING context's scope tag (an agent session is entered through - * `agent.ctx` ⇒ its events dispatch in that agent's scope; a bare session ⇒ - * subject-less carrier). WeakMap so a detached session drops its carrier - * with the entry. - */ - private carriers = new WeakMap>() + private store = new Map() private counter = 0 constructor(ctx: Context) { super(ctx, 'sessions') } - /** - * Reserve one unpublished session id across an asynchronous factory - * transaction. Bare `prepare`/`create`/`enter` calls for the id reject until - * release; the capability constructs exactly one Session and is passed back - * to {@link enter} at publication. The reservation belongs to the calling - * fiber, so owner unload releases an abandoned id automatically. - * @param id - the session id the transaction will publish. - * @returns the opaque reservation capability. - * @throws if the id is malformed, live, or already reserved. - */ - reserve(id: SessionId): SessionRegistrationReservation { - if (typeof id !== 'string') throw new TypeError('session id must be a string') - if (this.store.has(id) || this.reservations.has(id) || this.enteringIds.has(id)) { - throw new Error(`session "${id}" already exists or is reserved`) - } - let active = true - let prepared = false - const rawRelease = (): void => { - active = false - this.reservedSessions.delete(reservation) - this.reservations.delete(id) - } - // `release` is the exact effect disposer, so an ordered composite can - // adopt the automatic owner cleanup instead of racing it as a sibling. - const release = this.ctx.effect(() => rawRelease, `sessions.reserve(${id})`) - const reservation: SessionRegistrationReservation = Object.freeze({ - id, - prepare: (options?: CreateSessionOptions) => { - if (!active) { - throw new Error(`session "${id}" reservation is no longer active`) - } - if (prepared) throw new Error(`session "${id}" reservation already prepared a session`) - prepared = true - const session = this.prepareReserved(id, options, reservation) - this.reservedSessions.set(reservation, session) - return session - }, - release, - }) - this.reservations.set(id, reservation) - return reservation - } - /** * Create a session owned by the calling fiber: disposing that fiber stops * event notification and removes the session from the store. `options.seed` @@ -844,8 +607,8 @@ export class SessionStore extends Service { * For an agent whose session must be torn down IN ORDER with its loop (so the * loop's final flush is captured before the store attachment ends), do NOT use this * — fold the session lifecycle into the agent's own effect via - * {@link prepare} + {@link enter} + {@link announce} (see `dsh-agent-loop`'s - * `startOwned`). + * {@link prepare} + {@link enter} + {@link announce} (see + * `dsh-agent-loop`'s creation transaction). * * @param id - the session id; omitted, the store mints `session-`. * @param options - seed events and/or creation metadata for the header. @@ -884,40 +647,23 @@ export class SessionStore extends Service { * non-absolute path. */ prepare(id?: SessionId, options?: CreateSessionOptions): Session { - return this.prepareReserved(id, options) - } - - /** Shared prepare implementation, optionally authorized by a reservation. */ - private prepareReserved( - id?: SessionId, - options?: CreateSessionOptions, - reservation?: SessionRegistrationReservation, - ): Session { let sessionId: SessionId if (id === undefined) { do sessionId = SessionId(`session-${++this.counter}`) - while (this.store.has(sessionId) || this.reservations.has(sessionId)) + while (this.store.has(sessionId)) } else { sessionId = SessionId(id) } - if (typeof sessionId !== 'string') throw new TypeError('session id must be a string') - const held = this.reservations.get(sessionId) - if (reservation === undefined && held !== undefined) { - throw new Error(`session "${sessionId}" is reserved for unpublished creation`) - } if (this.store.has(sessionId)) throw new Error(`session "${sessionId}" already exists`) const seed = options?.seed - const meta = snapshotSessionMeta(options?.meta) - const cwd = meta.cwd - const parentSession = meta.parentSession - const seedLength = meta.seedLength + const meta = options?.meta const header: SessionHeader = { version: SESSION_FORMAT_VERSION, id: sessionId, - createdAt: meta.createdAt ?? Date.now(), - ...cwd !== undefined ? { cwd } : {}, - ...parentSession !== undefined ? { parentSession } : {}, - ...seedLength !== undefined ? { seedLength } : {}, + createdAt: meta?.createdAt ?? Date.now(), + ...meta?.cwd === undefined ? {} : { cwd: meta.cwd }, + ...meta?.parentSession === undefined ? {} : { parentSession: meta.parentSession }, + ...meta?.seedLength === undefined ? {} : { seedLength: meta.seedLength }, } return new Session(sessionId, seed, header) } @@ -939,77 +685,31 @@ export class SessionStore extends Service { * assume that. * * @param session - a {@link prepare}d session not yet in the store. - * @param reservation - the exact unpublished-id capability when a factory - * reserved this session across setup. * @returns the detach disposer (publication hooks + store removal). When called from * a synchronous `session/created` listener, removal and disposal wait until * that creation dispatch unwinds. * @throws if a session with this id is already in the store. */ - enter(session: Session, reservation?: SessionRegistrationReservation): () => void { + enter(session: Session): () => void { const id = session.id - if (typeof id !== 'string') throw new TypeError('session id must be a string') - const held = this.reservations.get(id) - if (reservation === undefined) { - if (held !== undefined) throw new Error(`session "${id}" is reserved for unpublished creation`) - } else if (reservation.id !== id || held !== reservation - || this.reservedSessions.get(reservation) !== session) { - throw new Error(`session "${id}" registration reservation does not own this prepared session`) - } - if (this.store.has(id) || this.enteringIds.has(id)) { - throw new Error(`session "${id}" already exists`) - } - if (appendHooks.has(session)) throw new Error(`session "${id}" is already attached to a store`) - this.enteringIds.add(id) - // The carrier is decided HERE, once, from the ENTERING context's scope tag - // (`this.ctx` is the caller's context — the tracker mechanism): every - // session/created|event|flush dispatch for this session uses it, so the - // session's whole event feed is scope-filtered consistently. The base is - // the session itself (scoped listeners' `this` is the session). - let carrier: Scoped - try { - carrier = scopeTarget(session, scopeOf(this.ctx)) - } finally { - this.enteringIds.delete(id) - } - const currentReservation = this.reservations.get(id) - if (reservation === undefined) { - /* v8 ignore next 2 -- reserve() rejects enteringIds, so carrier - * construction cannot install a new same-id reservation */ - if (currentReservation !== undefined) { - throw new Error(`session "${id}" is reserved for unpublished creation`) - } - } else if (currentReservation !== reservation - || this.reservedSessions.get(reservation) !== session) { - throw new Error(`session "${id}" registration reservation does not own this prepared session`) - } - /* v8 ignore next 1 -- enteringIds prevents a same-store commit during carrier construction */ + const carrier = scopeTarget(session, scopeOf(this.ctx)) + // This is the authoritative collision boundary after arbitrary unpublished + // preparation. Only one exact same-id transaction can publish. if (this.store.has(id)) throw new Error(`session "${id}" already exists`) - if (appendHooks.has(session)) throw new Error(`session "${id}" is already attached to a store`) - this.carriers.set(session, carrier) - const emitCtx = this.ctx - appendHooks.set(session, { - begin: () => { this.appending.add(session) }, - prepareObservation(event) { - // Cordis removes carrier/name in place and exposes the remaining array - // to internal/dispatch. Resolve with a throwaway array so an internal - // checker cannot replace the tuple later observers receive. - const dispatchArgs: unknown[] = [carrier, 'session/event', session, event] - const callbackArgs: unknown[] = [session, event] - const callbacks = collectSessionCallbacks(emitCtx, dispatchArgs) - assertDispatchTuple('session/event', dispatchArgs, callbackArgs) - return () => { invokeContainedSessionObservers(emitCtx, 'session/event', id, callbackArgs, callbacks) } - }, - end: () => { - this.appending.delete(session) - if (this.pendingDetach.has(session) && !this.announcing.has(session)) { - this.detachEntered(session, id, carrier) - } - }, - }) - attachmentEpochs.set(session, {}) - this.acceptedIds.set(session, id) - this.store.set(id, session) + if (attachments.has(session)) throw new Error(`session "${id}" is already attached to a store`) + const entry: SessionEntry = { + id, + session, + carrier, + emitCtx: this.ctx, + announced: false, + announcing: false, + appending: false, + detachRequested: false, + detach: () => { this.detachEntered(entry) }, + } + this.store.set(id, entry) + attachments.set(session, entry) let entered = true const detach = (): void => { if (!entered) return @@ -1017,30 +717,25 @@ export class SessionStore extends Service { // A lifecycle listener may own the advanced detach capability. Keep the // entry and its publication hooks live until synchronous creation or append // publication unwinds, then publish the paired disposal edge. - if (this.announcing.has(session) || this.appending.has(session)) { - this.pendingDetach.add(session) + if (entry.announcing || entry.appending) { + entry.detachRequested = true return } - this.detachEntered(session, id, carrier) + entry.detach() } return detach } /** Remove one exact entered session and emit its paired disposal when announced. */ - private detachEntered(session: Session, id: SessionId, carrier: Scoped): void { - this.pendingDetach.delete(session) + private detachEntered(entry: SessionEntry): void { + entry.detachRequested = false // A stale capability cannot remove observers or storage belonging to a // later same-id lifecycle. - /* v8 ignore next 1 -- the commit claim makes replacement impossible; this - * remains the exact-identity backstop against future mutation paths */ - if (this.store.get(id) !== session || this.acceptedIds.get(session) !== id) return - const wasAnnounced = this.announced.delete(session) - appendHooks.delete(session) - attachmentEpochs.set(session, {}) - this.acceptedIds.delete(session) - this.carriers.delete(session) - this.store.delete(id) - if (wasAnnounced) this.emitDisposed(session, carrier, id) + /* v8 ignore next -- enter() rejects replacement while this single-shot detach capability is live. */ + if (this.store.get(entry.id) !== entry) return + this.store.delete(entry.id) + attachments.delete(entry.session) + if (entry.announced) this.emitDisposed(entry) } /** Emit `session/created` exactly once for an {@link enter}ed session (with @@ -1051,20 +746,18 @@ export class SessionStore extends Service { * @throws if the session is not live or its announcement already began, * including a reentrant call from a creation listener. */ announce(session: Session): void { - const { carrier, id } = this.liveEntryFor(session) - if (this.announced.has(session)) { - throw new Error(`session "${id}" was already announced`) + const entry = this.liveEntryFor(session) + if (entry.announced || entry.announcing) { + throw new Error(`session "${entry.id}" was already announced`) } // Mark before emit: Cordis emit may deliver to earlier listeners and then // throw. Rollback must still pair that partial creation with disposal, and // a listener cannot recursively create a second lifecycle edge. - this.announced.add(session) - const dispatchArgs: unknown[] = [carrier, 'session/created', session] + entry.announced = true const callbackArgs: unknown[] = [session] - this.announcing.add(session) + entry.announcing = true try { - const callbacks = collectSessionCallbacks(this.ctx, dispatchArgs) - assertDispatchTuple('session/created', dispatchArgs, callbackArgs) + const callbacks = collectSessionCallbacks(this.ctx, [entry.carrier, 'session/created', session]) for (const callback of callbacks) { // Synchronous throws intentionally propagate and veto publication; the // yielded detach then emits the paired disposal edge. An async function @@ -1073,26 +766,23 @@ export class SessionStore extends Service { // of becoming unhandled. const returned: unknown = callback(...callbackArgs) void Promise.resolve(returned).catch((error: unknown) => { - warnContained(this.ctx, `session "${id}": session/created listener rejected: ${renderThrown(error)}`) + this.ctx.logger.warn(`session "${entry.id}": session/created listener rejected: ${String(error)}`) }) } } finally { - this.announcing.delete(session) - if (this.pendingDetach.has(session) && !this.appending.has(session)) { - this.detachEntered(session, id, carrier) - } + entry.announcing = false + if (entry.detachRequested && !entry.appending) entry.detach() } } /** Emit the paired teardown notification with per-listener containment. */ - private emitDisposed(session: Session, carrier: Scoped, id: SessionId): void { - const dispatchArgs: unknown[] = [carrier, 'session/disposed', session] - const callbackArgs: unknown[] = [session] + private emitDisposed(entry: SessionEntry): void { + const callbackArgs: unknown[] = [entry.session] try { - const callbacks = collectSessionCallbacks(this.ctx, dispatchArgs) - invokeContainedSessionObservers(this.ctx, 'session/disposed', id, callbackArgs, callbacks) + const callbacks = collectSessionCallbacks(this.ctx, [entry.carrier, 'session/disposed', entry.session]) + invokeContainedSessionObservers(this.ctx, 'session/disposed', entry.id, callbackArgs, callbacks) } catch (error: unknown) { - warnContained(this.ctx, `session "${id}": session/disposed dispatch threw: ${renderThrown(error)}`) + this.ctx.logger.warn(`session "${entry.id}": session/disposed dispatch threw: ${String(error)}`) } } @@ -1109,10 +799,8 @@ export class SessionStore extends Service { */ async flush(session: Session): Promise { const { carrier } = this.liveEntryFor(session) - const dispatchArgs: unknown[] = [carrier, 'session/flush', session] const callbackArgs: unknown[] = [session] - const callbacks = collectSessionCallbacks(this.ctx, dispatchArgs) - assertDispatchTuple('session/flush', dispatchArgs, callbackArgs) + const callbacks = collectSessionCallbacks(this.ctx, [carrier, 'session/flush', session]) const results = await Promise.allSettled(callbacks.map((callback) => { try { return callback(...callbackArgs) @@ -1127,21 +815,13 @@ export class SessionStore extends Service { if (failure !== undefined) throw failure.reason } - /** Return the exact live session's accepted id and carrier; detached/prepared objects reject. */ - private liveEntryFor(session: Session): { id: SessionId; carrier: Scoped } { - const id = this.acceptedIds.get(session) - if (id === undefined || this.store.get(id) !== session) { - throw new Error(`session "${id ?? session.id}" is not live in this store`) + /** Return the exact live entry; detached/prepared objects reject. */ + private liveEntryFor(session: Session): SessionEntry { + const entry = attachments.get(session) + if (entry === undefined || this.store.get(entry.id) !== entry) { + throw new Error(`session "${session.id}" is not live in this store`) } - const carrier = this.carriers.get(session) - // enter() installs store + carrier in one synchronous sequence; a live - // session without one is an internal invariant violation, never fallback - // to subject-less dispatch (that would silently cross scope boundaries). - /* v8 ignore next -- enter installs store and carrier in one synchronous sequence */ - if (carrier === undefined) { - throw new Error(`session "${id}" has no dispatch carrier`) - } - return { id, carrier } + return entry } /** @@ -1150,7 +830,7 @@ export class SessionStore extends Service { * @returns the session, or undefined when no live session has that id. */ get(id: SessionId): Session | undefined { - return this.store.get(id) + return this.store.get(id)?.session } /** @@ -1158,7 +838,7 @@ export class SessionStore extends Service { * @returns a fresh array; mutating it does not affect the store. */ list(): Session[] { - return [...this.store.values()] + return [...this.store.values()].map(entry => entry.session) } /** @@ -1228,7 +908,7 @@ export class SessionStore extends Service { ) } - return events.slice(0, boundary + 1).map(event => structuredClone(event)) + return events.slice(0, boundary + 1) } private _resolveForkSource(source: SessionForkSource): Session { diff --git a/packages/core/session/src/types.ts b/packages/core/session/src/types.ts index e564929ae0..c4838808c1 100644 --- a/packages/core/session/src/types.ts +++ b/packages/core/session/src/types.ts @@ -48,15 +48,15 @@ export interface SessionHeader { * session is created. A persistence backend rejects any other version on load * (no migration — see the constant). */ - version: number + readonly version: number /** The session's id (mirrors the {@link Session}'s id). */ - id: SessionId + readonly id: SessionId /** Unix epoch milliseconds when the session was created. */ - createdAt: number + readonly createdAt: number /** Absolute working directory the session was created in (if any). */ - cwd?: string + readonly cwd?: string /** The session this one was forked from (seed lineage), if any. */ - parentSession?: SessionId + readonly parentSession?: SessionId /** * How many leading events were INHERITED via a seed rather than produced by * this session — the seed boundary. Set when a fork seeds a child with a @@ -66,7 +66,7 @@ export interface SessionHeader { * harness can skip the inherited prefix when deriving the child's OWN script * (the seeded events are the parent's, not this child's model calls). */ - seedLength?: number + readonly seedLength?: number } /** @@ -76,7 +76,7 @@ export interface SessionHeader { */ export interface CreateSessionOptions { /** Events to seed the new session with (replay/fork). */ - seed?: SessionEvent[] + readonly seed?: readonly SessionEvent[] /** * Creation metadata. The store reads this plain record and each accepted * field once, then fills in `version`/`id` and defaults @@ -90,7 +90,12 @@ export interface CreateSessionOptions { * length, not the original boundary — the caller must pass the persisted * boundary back. A fresh fork passes its actual seeded-prefix length. */ - meta?: { cwd?: string; parentSession?: SessionId; createdAt?: number; seedLength?: number } + readonly meta?: { + readonly cwd?: string + readonly parentSession?: SessionId + readonly createdAt?: number + readonly seedLength?: number + } } /** diff --git a/packages/core/session/tests/derived-cache.spec.ts b/packages/core/session/tests/derived-cache.spec.ts index b416d15639..46015106c8 100644 --- a/packages/core/session/tests/derived-cache.spec.ts +++ b/packages/core/session/tests/derived-cache.spec.ts @@ -89,15 +89,15 @@ describe('Session.deriveEventMessage — the per-event projection', () => { expect(session.deriveEventMessage(event)).toEqual(session.deriveMessages().at(-1)) }) - it('clones content off the log: the projection never aliases the logged event', () => { + it('reuses the logged event\'s already frozen content', () => { const session = new Session(SessionId('per-event-clone')) session.append('turn/start', { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } }) const event = session.append('user/message', { content: [{ type: 'text', text: 'orig' }], source: { kind: 'user' } }, { surfaceOp: 'append' }) const message = session.deriveEventMessage(event)! - expect(message.content).not.toBe(event.data.content) - // deriveEventMessage returns an unfrozen clone (the cache freezes ITS - // copies); mutating it must not reach the log. - ;(message.content[0] as { text: string }).text = 'mutated' + expect(message.content).toBe(event.data.content) + expect(Object.isFrozen(message.content)).toBe(true) + expect(Object.isFrozen(message.content[0])).toBe(true) + expect(() => { (message.content[0] as { text: string }).text = 'mutated' }).toThrow() expect(session.deriveMessages().at(-1)!.content).toEqual([{ type: 'text', text: 'orig' }]) }) diff --git a/packages/core/session/tests/scoped.spec.ts b/packages/core/session/tests/scoped.spec.ts index 80cc08e629..7a5e617254 100644 --- a/packages/core/session/tests/scoped.spec.ts +++ b/packages/core/session/tests/scoped.spec.ts @@ -154,21 +154,6 @@ describe('sessions.flush()', () => { expect(flushed).toEqual([]) }) - it('rejects internal dispatch substitution before flush callbacks run', async () => { - const ctx = await mount() - const session = ctx.sessions.create() - const replacement = ctx.sessions.create() - const flushed: Session[] = [] - ctx.on('internal/dispatch', (_mode, name, args) => { - if (name === 'session/flush') args[0] = replacement - }) - ctx.on('session/flush', (candidate) => { flushed.push(candidate) }) - - await expect(ctx.sessions.flush(session)) - .rejects.toThrow('session/flush internal dispatch replaced the accepted callback tuple') - expect(flushed).toEqual([]) - }) - it('clears a detached carrier and rejects stale flushes', async () => { const ctx = await mount() const scope = await mintScope(ctx, 'owner') diff --git a/packages/core/session/tests/session.spec.ts b/packages/core/session/tests/session.spec.ts index 761f89ed2b..2204fc9027 100644 --- a/packages/core/session/tests/session.spec.ts +++ b/packages/core/session/tests/session.spec.ts @@ -129,16 +129,6 @@ describe('Session', () => { expect(session.events).toHaveLength(0) }) - it('rejects a non-string event type without retaining or freezing caller data', () => { - const session = new Session(SessionId('invalid-event-type')) - const type = { tag: 'caller-owned' } - const appendRaw = session.append.bind(session) as unknown as (type: unknown, data: unknown) => SessionEvent - - expect(() => appendRaw(type, {})).toThrow(/event type must be a string/) - expect(Object.isFrozen(type)).toBe(false) - expect(session.events).toEqual([]) - }) - it('rejects a surface-eligible append with no surfaceOp marker (runtime guard for the union-widening loophole)', () => { const session = new Session(SessionId('s5b')) session.append('turn/start', { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } }) @@ -282,7 +272,7 @@ describe('Session', () => { const seed: SessionEvent[] = [new SeedEvent()] expect(() => new Session(SessionId('seed-exotic-shell'), seed)) - .toThrow(/not a plain JSON record/) + .toThrow(/not losslessly JSON-serializable/) }) it('accepts a null-prototype seed event shell as a plain JSON record', () => { @@ -396,26 +386,6 @@ describe('Session', () => { expect(session.events).toEqual([event]) }) - it('reads surface metadata accessors once so a validated marker is logged', () => { - const session = new Session(SessionId('surface-intent-snapshot')) - let reads = 0 - const intent = { - get surfaceOp(): 'append' | undefined { - reads += 1 - return reads === 1 ? 'append' : undefined - }, - } - - const event = session.append( - 'user/message', - { content: [{ type: 'text', text: 'hello' }], source: { kind: 'user' } }, - intent as { surfaceOp: 'append' }, - ) - - expect(reads).toBe(1) - expect(event.surfaceOp).toBe('append') - }) - it('rejects non-JSON surface metadata before appending the event', () => { const session = new Session(SessionId('append-bad-metadata')) @@ -578,44 +548,10 @@ describe('Session', () => { expect(session.header).not.toBe(input) expect(Object.isFrozen(session.header)).toBe(true) expect(Reflect.set(session.header, 'cwd', '/published-mutated')).toBe(false) - expect(Reflect.set(session, 'id', SessionId('redirected'))).toBe(false) - expect(Reflect.set(session, 'header', input)).toBe(false) - expect(Object.getOwnPropertyDescriptor(session, 'id')).toMatchObject({ - configurable: false, - writable: false, - }) - expect(Object.getOwnPropertyDescriptor(session, 'header')).toMatchObject({ - configurable: false, - writable: false, - }) expect(session.id).toBe('header-owned') expect(session.header.cwd).toBe('/accepted') }) - it('reads each supplied header field once before validation and publication', () => { - const reads = { version: 0, id: 0, createdAt: 0, cwd: 0, parentSession: 0, seedLength: 0 } - const header = { - get version() { reads.version += 1; return reads.version === 1 ? SESSION_FORMAT_VERSION : 99 }, - get id() { reads.id += 1; return reads.id === 1 ? SessionId('header-once') : SessionId('drifted') }, - get createdAt() { reads.createdAt += 1; return reads.createdAt === 1 ? 123 : Number.NaN }, - get cwd() { reads.cwd += 1; return reads.cwd === 1 ? '/accepted' : 'relative' }, - get parentSession() { reads.parentSession += 1; return reads.parentSession === 1 ? SessionId('parent') : 1n }, - get seedLength() { reads.seedLength += 1; return reads.seedLength === 1 ? 0 : 1n }, - } as unknown as SessionHeader - - const session = new Session(SessionId('header-once'), undefined, header) - - expect(reads).toEqual({ version: 1, id: 1, createdAt: 1, cwd: 1, parentSession: 1, seedLength: 1 }) - expect(session.header).toEqual({ - version: SESSION_FORMAT_VERSION, - id: 'header-once', - createdAt: 123, - cwd: '/accepted', - parentSession: 'parent', - seedLength: 0, - }) - }) - it('rejects an exotic, non-JSON, or mismatched supplied header', () => { class ExoticHeader implements SessionHeader { readonly version = SESSION_FORMAT_VERSION @@ -624,7 +560,7 @@ describe('Session', () => { } expect(() => new Session(SessionId('header-invalid'), undefined, new ExoticHeader())) - .toThrow(/not a plain JSON record/) + .toThrow(/not losslessly JSON-serializable/) expect(() => new Session(SessionId('header-invalid'), undefined, { version: SESSION_FORMAT_VERSION, id: SessionId('header-invalid'), @@ -740,79 +676,6 @@ describe('SessionStore', () => { expect(ctx.sessions.get(SessionId('racy'))).toBe(live) }) - it('claims an id across Context.filter evaluation before committing the exact session', async () => { - const ctx = new Context() - await ctx.plugin(SessionStore) - const id = SessionId('reentrant-enter') - const nested = new Session(id) - const outer = new Session(id) - let nestedError = '' - let attempted = false - Object.defineProperty(outer, Context.filter, { - configurable: true, - get() { - if (!attempted) { - attempted = true - try { - ctx.sessions.enter(nested) - } catch (error: unknown) { - nestedError = String(error) - } - } - return undefined - }, - }) - - const detach = ctx.sessions.enter(outer) - expect(nestedError).toMatch(/already exists/) - expect(ctx.sessions.get(id)).toBe(outer) - detach() - expect(ctx.sessions.get(id)).toBeUndefined() - }) - - it('revalidates reservation ownership after carrier construction runs caller code', async () => { - const ctx = new Context() - await ctx.plugin(SessionStore) - const id = SessionId('released-during-enter') - const reservation = ctx.sessions.reserve(id) - const session = reservation.prepare() - Object.defineProperty(session, Context.filter, { - configurable: true, - get() { - reservation.release() - return undefined - }, - }) - - expect(() => ctx.sessions.enter(session, reservation)).toThrow(/does not own this prepared session/) - expect(ctx.sessions.get(id)).toBeUndefined() - }) - - it('rejects when carrier construction attaches the same session to another store', async () => { - const firstCtx = new Context() - const secondCtx = new Context() - await firstCtx.plugin(SessionStore) - await secondCtx.plugin(SessionStore) - const session = new Session(SessionId('cross-store-carrier')) - let attempted = false - let detachSecond = (): void => {} - Object.defineProperty(session, Context.filter, { - configurable: true, - get() { - if (!attempted) { - attempted = true - detachSecond = secondCtx.sessions.enter(session) - } - return undefined - }, - }) - - expect(() => firstCtx.sessions.enter(session)).toThrow(/already attached to a store/) - expect(firstCtx.sessions.get(session.id)).toBeUndefined() - expect(secondCtx.sessions.get(session.id)).toBe(session) - detachSecond() - }) - it('prepare() + enter() + announce() register a session and emit session/created', async () => { const ctx = new Context() await ctx.plugin(SessionStore) @@ -834,7 +697,7 @@ describe('SessionStore', () => { expect(ctx.sessions.get(SessionId('lifecycle'))).toBeUndefined() }) - it('captures the accepted id once and prevents simultaneous attachment to two stores', async () => { + it('prevents simultaneous attachment of one session object to two stores', async () => { const firstCtx = new Context() const secondCtx = new Context() await firstCtx.plugin(SessionStore) @@ -842,7 +705,6 @@ describe('SessionStore', () => { const session = new Session(SessionId('owned-key')) const detachFirst = firstCtx.sessions.enter(session) - expect(Reflect.set(session, 'id', SessionId('redirected'))).toBe(false) expect(() => secondCtx.sessions.enter(session)).toThrow(/already attached to a store/) expect(firstCtx.sessions.get(SessionId('owned-key'))).toBe(session) @@ -852,69 +714,6 @@ describe('SessionStore', () => { expect(secondCtx.sessions.get(SessionId('owned-key'))).toBe(session) detachSecond() - expect(() => firstCtx.sessions.enter({ id: 42 } as unknown as Session)).toThrow(/id must be a string/) - }) - - it('uses an opaque one-session reservation to gate unpublished factory insertion', async () => { - const ctx = new Context() - await ctx.plugin(SessionStore) - const held = ctx.sessions.reserve(SessionId('held-session')) - - expect(() => ctx.sessions.reserve(SessionId('held-session'))).toThrow(/already exists or is reserved/) - expect(() => ctx.sessions.prepare(SessionId('held-session'))).toThrow(/reserved for unpublished creation/) - expect(() => ctx.sessions.create(SessionId('held-session'))).toThrow(/reserved for unpublished creation/) - const session = held.prepare({ meta: { cwd: '/held' } }) - expect(() => held.prepare()).toThrow(/already prepared/) - expect(() => ctx.sessions.enter(session)).toThrow(/reserved for unpublished creation/) - - const other = ctx.sessions.reserve(SessionId('other-session')) - expect(() => ctx.sessions.enter(session, other)).toThrow(/does not own this prepared session/) - expect(() => ctx.sessions.enter(new Session(SessionId('held-session')), held)) - .toThrow(/does not own this prepared session/) - - const detach = ctx.sessions.enter(session, held) - ctx.sessions.announce(session) - held.release() - held.release() - expect(ctx.sessions.get(SessionId('held-session'))).toBe(session) - expect(() => ctx.sessions.reserve(SessionId('held-session'))).toThrow(/already exists or is reserved/) - detach() - other.release() - - const expired = ctx.sessions.reserve(SessionId('expired-session')) - expired.release() - expect(() => expired.prepare()).toThrow(/no longer active/) - expect(() => ctx.sessions.enter(new Session(SessionId('expired-session')), expired)) - .toThrow(/does not own this prepared session/) - expect(() => ctx.sessions.reserve(42 as unknown as SessionId)).toThrow(/id must be a string/) - expect(() => ctx.sessions.prepare(42 as unknown as SessionId)).toThrow(/id must be a string/) - - // Auto-generated ids skip unpublished reservations just as they skip live - // store entries; no hidden collision can be published later. - const firstAuto = ctx.sessions.reserve(SessionId('session-1')) - expect(ctx.sessions.prepare().id).toBe('session-2') - firstAuto.release() - }) - - it('owns reservations by the calling fiber and rolls back failed ownership registration', async () => { - const ctx = new Context() - await ctx.plugin(SessionStore) - let held!: import('@deepseek-ai/dsh-session').SessionRegistrationReservation - let scopedSessions!: SessionStore - const owner = await ctx.plugin(Object.assign((inner: Context) => { - scopedSessions = inner.sessions - held = inner.sessions.reserve(SessionId('fiber-held')) - }, { inject: ['sessions'] })) - - expect(() => ctx.sessions.reserve(SessionId('fiber-held'))).toThrow(/already exists or is reserved/) - await owner.dispose() - const reused = ctx.sessions.reserve(SessionId('fiber-held')) - reused.release() - held.release() // idempotent after the automatic owner-disposal release - - expect(() => scopedSessions.reserve(SessionId('inactive-owner'))).toThrow(/inactive context/) - const recovered = ctx.sessions.reserve(SessionId('inactive-owner')) - recovered.release() }) it('rejects direct and reentrant repeat announcements to preserve one lifecycle pair', async () => { @@ -1009,54 +808,14 @@ describe('SessionStore', () => { }) }) - it('reads session options and each metadata field once in prepare()', async () => { - const ctx = new Context() - await ctx.plugin(SessionStore) - const reads = { seed: 0, meta: 0, cwd: 0, parentSession: 0, createdAt: 0, seedLength: 0 } - const meta = { - get cwd() { reads.cwd += 1; return reads.cwd === 1 ? '/accepted' : 'relative' }, - get parentSession() { reads.parentSession += 1; return reads.parentSession === 1 ? SessionId('parent') : 1n }, - get createdAt() { reads.createdAt += 1; return reads.createdAt === 1 ? 123 : Number.NaN }, - get seedLength() { reads.seedLength += 1; return reads.seedLength === 1 ? 0 : 1n }, - } - const options = { - get seed() { reads.seed += 1; return reads.seed === 1 ? undefined : [] }, - get meta() { reads.meta += 1; return reads.meta === 1 ? meta : undefined }, - } as unknown as CreateSessionOptions - - const session = ctx.sessions.prepare(SessionId('metadata-once'), options) - - expect(reads).toEqual({ seed: 1, meta: 1, cwd: 1, parentSession: 1, createdAt: 1, seedLength: 1 }) - expect(session.header).toEqual({ - version: SESSION_FORMAT_VERSION, - id: 'metadata-once', - createdAt: 123, - cwd: '/accepted', - parentSession: 'parent', - seedLength: 0, - }) - }) - - it('rejects exotic metadata before cloning can erase its prototype', async () => { - class ExoticMeta { - readonly cwd = '/accepted' - } - const ctx = new Context() - await ctx.plugin(SessionStore) - - expect(() => ctx.sessions.prepare(SessionId('exotic-meta'), { meta: new ExoticMeta() })) - .toThrow(/session metadata is not a plain JSON record/) - }) - it('rejects non-JSON and invalid scalar session metadata', async () => { const ctx = new Context() await ctx.plugin(SessionStore) const cases: Array<{ meta: unknown; error: RegExp }> = [ - { meta: 1, error: /metadata is not a plain JSON record/ }, - { meta: { parentSession: 1n }, error: /metadata is not losslessly JSON-serializable/ }, - { meta: { cwd: 1 }, error: /session cwd must be a string/ }, - { meta: { parentSession: 1 }, error: /parentSession must be a string/ }, - { meta: { createdAt: '123' }, error: /createdAt must be a finite number/ }, + { meta: { parentSession: 1n }, error: /header is not losslessly JSON-serializable/ }, + { meta: { cwd: 1 }, error: /header cwd must be a string/ }, + { meta: { parentSession: 1 }, error: /header parentSession must be a string/ }, + { meta: { createdAt: '123' }, error: /header createdAt must be a finite number/ }, { meta: { seedLength: '1' }, error: /seedLength must be a non-negative safe integer/ }, { meta: { seedLength: 0.5 }, error: /seedLength must be a non-negative safe integer/ }, { meta: { seedLength: -1 }, error: /seedLength must be a non-negative safe integer/ }, @@ -1224,105 +983,6 @@ describe('SessionStore', () => { expect(observed).toEqual([]) }) - it('rejects prepend or append instrumentation that replaces the accepted observer tuple', async () => { - for (const prepend of [true, false]) { - const ctx = new Context() - await ctx.plugin(SessionStore) - const session = ctx.sessions.create(SessionId(`dispatch-tuple-${prepend}`)) - const replacementSession = new Session(SessionId('replacement')) - const replacementEvent = { - type: 'turn/end', - seq: 99, - time: 1, - data: { turn: 99, reason: { kind: 'completed' } }, - } as SessionEvent - const observed: Array<{ session: Session; event: SessionEvent }> = [] - let replace = true - ctx.on('internal/dispatch', (_mode, name, args) => { - if (name !== 'session/event' || !replace) return - args[0] = replacementSession - args[1] = replacementEvent - }, { prepend }) - ctx.on('session/event', (observedSession, event) => { - observed.push({ session: observedSession, event }) - }) - - expect(() => session.append('turn/start', { - turn: 1, - trigger: { kind: 'message', source: { kind: 'user' } }, - })).toThrow('session/event internal dispatch replaced the accepted callback tuple') - expect(session.events).toEqual([]) - expect(observed).toEqual([]) - - replace = false - const appended = session.append('turn/start', { - turn: 1, - trigger: { kind: 'message', source: { kind: 'user' } }, - }) - expect(session.events).toEqual([appended]) - expect(observed).toEqual([{ session, event: appended }]) - } - }) - - it('rejects if a bare session becomes attached while caller data is materialized', async () => { - const ctx = new Context() - await ctx.plugin(SessionStore) - const session = new Session(SessionId('attach-during-append')) - const observed: SessionEvent[] = [] - let sessionEventDispatches = 0 - let detach!: () => void - ctx.on('internal/dispatch', (_mode, name) => { - if (name === 'session/event') sessionEventDispatches += 1 - }) - ctx.on('session/event', (_observedSession, event) => { observed.push(event) }) - const data = { - get todos(): TodoItem[] { - detach = ctx.sessions.enter(session) - ctx.sessions.announce(session) - return [] - }, - } - - expect(() => session.append('todo/write', data)) - .toThrow('session attachment changed while append input was being accepted') - expect(ctx.sessions.get(session.id)).toBe(session) - expect(session.events).toEqual([]) - expect(sessionEventDispatches).toBe(0) - expect(observed).toEqual([]) - - const appended = session.append('todo/write', { todos: [] }) - expect(session.events).toEqual([appended]) - expect(sessionEventDispatches).toBe(1) - expect(observed).toEqual([appended]) - detach() - }) - - it('rejects a transient attach and detach while caller data is materialized', async () => { - const ctx = new Context() - await ctx.plugin(SessionStore) - const session = new Session(SessionId('attach-detach-during-append')) - const lifecycle: string[] = [] - const observed: SessionEvent[] = [] - ctx.on('session/created', () => { lifecycle.push('created') }) - ctx.on('session/disposed', () => { lifecycle.push('disposed') }) - ctx.on('session/event', (_observedSession, event) => { observed.push(event) }) - const data = { - get todos(): TodoItem[] { - const detach = ctx.sessions.enter(session) - ctx.sessions.announce(session) - detach() - return [] - }, - } - - expect(() => session.append('todo/write', data)) - .toThrow('session attachment changed while append input was being accepted') - expect(ctx.sessions.get(session.id)).toBeUndefined() - expect(session.events).toEqual([]) - expect(lifecycle).toEqual(['created', 'disposed']) - expect(observed).toEqual([]) - }) - it('contains a reentrant observer append without reordering later observers', async () => { const ctx = new Context() await ctx.plugin(SessionStore) @@ -1342,34 +1002,10 @@ describe('SessionStore', () => { expect(session.events).toEqual([appended]) expect(heard).toEqual([appended]) expect(warnings).toEqual([ - 'session "reentrant-observer": session/event listener threw: Error: session append cannot reenter while another append is being accepted or published', + 'session "reentrant-observer": session/event listener threw: Error: session append cannot reenter while another append is being published', ]) }) - it('keeps observer failures contained when warning output itself throws', async () => { - const ctx = new Context() - await ctx.plugin(SessionStore) - ctx.logger.warn = (() => { throw new Error('logger unavailable') }) as typeof ctx.logger.warn - const session = ctx.sessions.create(SessionId('throwing-logger')) - const heard: SessionEvent[] = [] - ctx.on('session/event', () => { throw new Error('sync observer') }) - ctx.on('session/event', () => Promise.reject(new Error('async observer')) as never) - ctx.on('session/event', (_observedSession, event) => { heard.push(event) }) - - let appended!: SessionEvent - expect(() => { - appended = session.append('turn/start', { - turn: 1, - trigger: { kind: 'message', source: { kind: 'user' } }, - }) - }).not.toThrow() - await Promise.resolve() - await Promise.resolve() - - expect(session.events).toEqual([appended]) - expect(heard).toEqual([appended]) - }) - it('defers detach through dispatch resolution, commit, and observer publication', async () => { const ctx = new Context() await ctx.plugin(SessionStore) @@ -1425,12 +1061,9 @@ describe('SessionStore', () => { await ctx.plugin(SessionStore) const warnings: string[] = [] ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn - const hostile = { [Symbol.toPrimitive]() { throw new Error('cannot stringify') } } - const printable = { toString: () => 'printable failure' } const heard: string[] = [] - ctx.on('session/disposed', () => { throw hostile }) + ctx.on('session/disposed', () => { throw new Error('sync disposed') }) ctx.on('session/disposed', () => Promise.reject(new Error('async disposed')) as never) - ctx.on('session/disposed', () => { throw printable }) ctx.on('session/disposed', (session) => { heard.push(session.id) }) const unannounced = ctx.sessions.prepare(SessionId('never-announced')) @@ -1447,8 +1080,7 @@ describe('SessionStore', () => { expect(heard).toEqual(['contained-disposal']) expect(warnings).toEqual([ - 'session "contained-disposal": session/disposed listener threw: ', - 'session "contained-disposal": session/disposed listener threw: printable failure', + 'session "contained-disposal": session/disposed listener threw: Error: sync disposed', 'session "contained-disposal": session/disposed listener rejected: Error: async disposed', ]) }) diff --git a/packages/core/system-prompt/README.md b/packages/core/system-prompt/README.md index 06d735e271..270a016cf3 100644 --- a/packages/core/system-prompt/README.md +++ b/packages/core/system-prompt/README.md @@ -1,34 +1,32 @@ # dsh-system-prompt -System prompt assembly registry. Plugins contribute ordered text sections, tool-schema providers, named prompt variables, and owner-final named protections; the agent loop calls `assemble(context)` once per step, and `renderPrompt(assembly)` is the full system prompt the model sees. The plugin registers the harness-owned openers itself — the static `harness:identity` section and the deployment's `deployment:persona` section — so they exist for every agent regardless of which loop plugin drives it. +System prompt assembly registry. Plugins contribute ordered text sections, tool-schema providers, and named prompt variables; contributions that implement required protocol may declare themselves owner-final. The agent loop calls `assemble(context)` once per step, and `renderPrompt(assembly)` is the full system prompt the model sees. The plugin registers the harness-owned openers itself — the static `harness:identity` section and the global default `deployment:persona` section — so they remain available regardless of which loop plugin drives an agent. An agent-scoped contribution with the same persona name shadows that default for its agent. ## Config | Key | Default | Meaning | |---|---|---| -| `persona` | `''` | The deployment persona: the ONE deployment-authored prompt fragment, rendered as the order-0 `deployment:persona` section and shared by every agent in the context (subagents included). A template — complete `{{…}}` groups are interpreted strictly against the registered variables (the shipped loop registers `{{model}}`/`{{cwd}}`), with no escape syntax for literal braces yet. Empty ⇒ the section is dropped at render. | +| `persona` | `''` | The global deployment-persona default: the ONE config-authored prompt fragment, rendered as the order-0 `deployment:persona` section unless an agent-scoped contribution shadows it. A template — complete `{{…}}` groups are interpreted strictly against the registered variables (the shipped loop registers `{{model}}`/`{{cwd}}`), with no escape syntax for literal braces yet. Empty ⇒ the section is dropped at render. | | `toolOrder` | — | Explicit model-facing tool order, as a list of `ToolSchema.name`s with one `''` rest entry (`TOOL_ORDER_REST`): listed tools take their listed position, unlisted tools land at the rest entry in lexicographic name order. Absent ⇒ plain lexicographic name order. Applied to the collected tools BEFORE the `system-prompt/assemble` waterfall — like the sections' `order` sort, it canonicalizes what the registry contributed (registration order is a plugin-load artifact), and a waterfall listener that mutates the list owns the determinism of what it emits. Misconfiguration fails loud: a list without exactly one rest entry, or with duplicates, throws at load; a listed name with no registered tool rejects every `assemble()`; a tool provider returning the reserved rest-entry name also rejects. Under the shipped loop the turn fails before any model request. Why a central list and not per-plugin weights: [Explicit model-facing tool order](../../../docs/rfc/implemented/feature/2026-07-06-explicit-tool-order.md). | ## Service: `SystemPrompt` (ctx key: `systemPrompt`) ### Public API -- `ctx.systemPrompt.section(section: PromptSection): () => Promise | void` Contribute a section. The registry reads `name`, `order`, and the text value/callback once, validates their fixed string/finite-number/string-or-function types, and stores only that accepted record; later caller-object mutation cannot rename or reshape it. The layer is the CALLING context's scope: `agent.ctx` contributes to that agent alone, SHADOWING a same-named global section there (the per-agent persona mechanism — a scoped `deployment:persona`). Duplicate names within one layer throw, and a globally protected section name cannot be shadowed. Disposed with the calling fiber. -- `ctx.systemPrompt.tools(provider: (context: AssembleContext) => ToolProviderResult): () => Promise | void` Contribute tool schemas, evaluated at each assembly with that assembly's context; a non-function provider rejects before effect storage. `ToolProviderResult` = `{ schemas, knownNames? }`: `schemas` is the post-restriction visible set for `context.scope`; `knownNames` (defaulting to the same captured schemas' names) is the pre-restriction universe `toolOrder` validates against. Assembly reads the result, each schema field, and the optional known-name list once before detaching them, rejects non-string schema names/descriptions or known names, and uses those same accepted strings for validation and the model-visible collection. A provider must not return a schema named `TOOL_ORDER_REST`. Scoped providers are consulted only for their scope's assemblies. Disposed with the calling fiber. -- `ctx.systemPrompt.variable(name: string, provider: (context) => string | undefined): () => Promise | void` Contribute a prompt variable, referenced from section text as `{{name}}`. The fixed string name and function provider types reject before effect storage. Scoped variables (via `agent.ctx`) shadow a same-named global for that agent. Duplicate-in-layer or unreferenceable names throw; `undefined` means "no value for this assembly". Disposed with the calling fiber. -- `ctx.systemPrompt.protect(protection: PromptProtection): () => Promise | void` Make named section/tool contributions owner-final after the assembly waterfall. Protection restores canonical registry/provider presence and definition; restored entries keep canonical order with one another and anchor before their first surviving later unprotected canonical neighbor (or at the end), without undoing listener reordering of unprotected entries. Canonical absence is owner-final too, so a mode-hidden tool cannot be fabricated by a listener. Calling through `agent.ctx` protects only that agent's assemblies. A global section protection additionally reserves its name against scoped shadows; registering either side of that conflict fails loudly instead of treating the shadow as canonical. Each optional field and array slot is read once, non-array fields or non-string names reject before effect storage, and the accepted arrays are deduplicated and frozen. Finalization materializes each waterfall-produced entry name once, so a stateful getter cannot evade canonical replacement. Empty protections throw, and disposal removes the protection. -- `ctx.systemPrompt.assemble(context?: AssembleContext): Promise` Assemble the prompt for one caller: the global layer merged with `context.scope`'s layer (scoped shadows global). Provider output becomes one coherent detached snapshot before `toolOrder` validation. Runs through the scope-filtered `system-prompt/assemble` waterfall, then restores protected contributions from the pre-waterfall canonical assembly. Rejects when a configured `toolOrder` names a tool outside the providers' `knownNames` universe (a restricted-away KNOWN tool is a normal absence), or when a provider returns the reserved rest-entry name. +- `ctx.systemPrompt.section(section: PromptSection): () => Promise | void` Contribute a section. The layer is the calling context's scope: `agent.ctx` contributes to that agent alone, shadowing a same-named global section there. Duplicate names within one layer and non-finite orders throw. `ownerFinal: true` restores this section's canonical definition after the complete waterfall and reserves a global section against scoped shadows. Disposed with the calling fiber. +- `ctx.systemPrompt.tools(provider: (context: AssembleContext) => ToolProviderResult): () => Promise | void` Contribute tool schemas, evaluated at each assembly with that assembly's context. `ToolProviderResult` = `{ schemas, knownNames?, ownerFinalNames? }`: `schemas` is the post-restriction visible set; `knownNames` is the pre-restriction universe used by `toolOrder`; `ownerFinalNames` makes those tools' canonical presence or absence survive the waterfall. A provider must not return a schema named `TOOL_ORDER_REST`. Scoped providers are consulted only for their scope's assemblies. Disposed with the calling fiber. +- `ctx.systemPrompt.variable(name: string, provider: (context) => string | undefined): () => Promise | void` Contribute a prompt variable, referenced from section text as `{{name}}`. Scoped variables shadow a same-named global for that agent. Duplicate-in-layer or unreferenceable names throw; `undefined` means "no value for this assembly". Disposed with the calling fiber. +- `ctx.systemPrompt.assemble(context?: AssembleContext): Promise` Assemble the prompt for one caller: the global layer merged with `context.scope`'s layer, with tool schemas detached before the transform seam. Runs through the scope-filtered `system-prompt/assemble` waterfall, then restores owner-final contributions from a private pre-waterfall snapshot. Restored entries keep canonical relative order without undoing listener reordering of ordinary entries. Rejects when a configured `toolOrder` names a tool outside the providers' `knownNames` universe, or when a provider returns the reserved rest-entry name. ### Live events -Prompt assembly is the scope-filtered transformable seam; registry change is the deliberately unfiltered notification that an assembly input changed, possibly for one scope. Exact signatures, dispatch modes, and filtering contracts live in the generated [Cordis event catalog](../../../docs/cordis-catalog/events.md). Named protections apply only after a successful assembly waterfall returns and are owned by the service rather than represented as another event listener. +Prompt assembly is the scope-filtered transformable seam; registry change is the deliberately unfiltered notification that an assembly input changed, possibly for one scope. Exact signatures, dispatch modes, and filtering contracts live in the generated [Cordis event catalog](../../../docs/cordis-catalog/events.md). Owner-final restoration applies only after a successful assembly waterfall returns. ### Key types - `AssembleContext` — what one `assemble()` call is FOR. Merge-extensible; declares `scope?: ScopeKey` (the layer selector) here, and `dsh-agent` declares `agent?: Agent` (the typed DX field — never set without `scope`; use `assembleContextFor(agent)`). Providers must tolerate absent fields (a bare `assemble()` carries an empty, scope-less context). -- `PromptSection` — `{ name, order, text: string | ((context) => string) }`. Sections are concatenated in ascending `order`. Order bands: `-100` is the harness identity, `0` the deployment persona (both registered by this plugin), tool guidance uses `100–199`; other negative orders also render before the persona. +- `PromptSection` — `{ name, order, text, ownerFinal? }`. Sections are concatenated in ascending `order`; `ownerFinal` is reserved for required protocol instructions. Order bands: `-100` is the harness identity, `0` the deployment persona, tool guidance uses `100–199`. - `PromptAssembly` — `{ sections: AssembledSection[], tools: ToolSchema[], variables: Record }`. Section texts arrive resolved but not yet interpolated; `variables` holds every registered variable resolved against the context. Tool schemas are part of the assembly by design: "what the model is told it can do" is one coherent thing, even though adapters transmit schemas as a separate wire field. -- `PromptProtection` — `{ sections?: readonly string[], tools?: readonly string[] }`. Named contributions whose canonical pre-waterfall state is restored after all listeners; protections compose by set union rather than callback order, and global section names are reserved against scoped shadows. - `renderPrompt(assembly)` — interpolates `{{variable}}` references in each section, drops empty sections, joins with blank lines. STRICT: an unknown reference (`Object.hasOwn` lookup — prototype names like `{{constructor}}` are unknown), a registered-but-valueless reference, a malformed complete `{{…}}` group, or a `{{` that opens no complete group while a `}}` still follows (`{{{model}}}`) throws — fail loud beats shipping a malformed prompt. A lone `{{` with no `}}` anywhere after it passes through verbatim; substituted values are never re-scanned. Merge-extensible: plugins can declare extra fields on `PromptAssembly` and `AssembleContext` via declaration merging. @@ -39,11 +37,11 @@ Merge-extensible: plugins can declare extra fields on `PromptAssembly` and `Asse - Variable providers: the agent loop registers `model` and `cwd`; any plugin can register the facts it owns (a future `date`, git state, …). - Tool schema providers: `ToolRegistry` registers itself as a tool provider automatically. - The `system-prompt/assemble` waterfall: mutate or replace the assembly per caller (dynamic tool filtering, extra variables). -- `systemPrompt.protect()`: reserve canonical section/tool contributions for invariants that ordinary waterfall listeners must not be able to remove or replace. +- Owner-final contributions: protocol owners declare finality on the section or tool contribution itself; there is no independent protection registry. ### What is NOT here -- Any deployment-authored prompt text outside config — the persona is this plugin's `persona` config, and every other section comes from the plugin that owns the fact. (The `harness:identity` line is deliberately a code literal: a harness fact, not a deployment choice; the `system-prompt/assemble` waterfall is the escape valve for a deployment that must drop it.) +- Any end-user prompt-editing API — this plugin owns the config-authored global persona default, creator plugins may register agent-scoped shadows during setup, and every other section comes from the plugin that owns the fact. (The `harness:identity` line is deliberately a code literal: a harness fact, not a deployment choice; the `system-prompt/assemble` waterfall is the escape valve for a deployment that must drop it.) - Prompt compaction (belongs on the `agent/pre-step` seam in `dsh-agent`). Design rationale: [the prompt-variables RFC](../../../docs/rfc/implemented/architecture/2026-07-05-prompt-variables-and-tool-guidance-ownership.md). diff --git a/packages/core/system-prompt/package.json b/packages/core/system-prompt/package.json index 69af28f3b5..120e10ef11 100644 --- a/packages/core/system-prompt/package.json +++ b/packages/core/system-prompt/package.json @@ -24,7 +24,6 @@ "peerDependencies": { "@deepseek-ai/dsh-llm": "^0.0.1", "@deepseek-ai/dsh-scope": "^0.0.1", - "@deepseek-ai/dsh-session": "^0.0.1", "cordis": "^4.0.0-rc.6" }, "dependencies": { @@ -33,7 +32,6 @@ "devDependencies": { "@deepseek-ai/dsh-llm": "workspace:^", "@deepseek-ai/dsh-scope": "workspace:^", - "@deepseek-ai/dsh-session": "workspace:^", "cordis": "^4.0.0-rc.6" } } diff --git a/packages/core/system-prompt/src/index.ts b/packages/core/system-prompt/src/index.ts index 40eddc7ba1..b1946c59e1 100644 --- a/packages/core/system-prompt/src/index.ts +++ b/packages/core/system-prompt/src/index.ts @@ -1,8 +1,8 @@ /** * System prompt assembly registry. Plugins contribute ordered text sections, - * tool schema providers, named prompt variables, and owner-final named - * protections; `assemble(context)` collates them through a waterfall that - * runs once per step, restores protected contributions, and `renderPrompt` + * tool schema providers, and named prompt variables; protocol contributions + * may declare themselves owner-final. `assemble(context)` collates them through a waterfall that + * runs once per step, restores owner-final contributions, and `renderPrompt` * interpolates `{{variable}}` references into the final text. * * The harness-owned prompt openers live here too: this plugin registers the @@ -18,7 +18,6 @@ import z from 'schemastery' import { scopeOf, scopeTarget } from '@deepseek-ai/dsh-scope' import type { ScopeKey, Scoped } from '@deepseek-ai/dsh-scope' import type { ToolSchema } from '@deepseek-ai/dsh-llm' -import { snapshotJsonValue } from '@deepseek-ai/dsh-session' declare module 'cordis' { interface Context { @@ -45,7 +44,7 @@ declare module 'cordis' { */ 'system-prompt/assemble'(this: Scoped, assembly: PromptAssembly, context: AssembleContext, next: () => Promise): Promise /** - * A section, tool provider, variable provider, or protection was registered + * A section, tool provider, or variable provider was registered * or unregistered (the assembly inputs changed — possibly for one scope * only). An UNFILTERED registry-subject notification, deliberately not * scope-filtered dispatch: a global change concerns every agent's next @@ -81,19 +80,25 @@ export interface AssembleContext { /** One contributed section of the system prompt (registry input). */ export interface PromptSection { /** Unique name — a duplicate registration throws (see {@link SystemPrompt.section}). */ - name: string + readonly name: string /** * Sections are concatenated in ascending order. Convention: `-100` is the * harness identity, `0` the deployment persona, tool guidance uses 100–199; * other negative orders also render before the persona. */ - order: number + readonly order: number /** * Static text or a provider evaluated at each assembly with that assembly's * {@link AssembleContext}. The text may reference `{{variable}}`s — they are * interpolated later, by {@link renderPrompt}. */ - text: string | ((context: AssembleContext) => string) + readonly text: string | ((context: AssembleContext) => string) + /** + * Whether this section's canonical presence and definition survive the + * complete assembly waterfall. Use this only for owner-required protocol + * instructions; ordinary sections remain transformable. + */ + readonly ownerFinal?: boolean } /** One section of an assembly: {@link PromptSection} with its text resolved. */ @@ -118,30 +123,15 @@ export interface AssembledSection { */ export interface ToolProviderResult { /** The schemas this provider contributes to THIS assembly. */ - schemas: ToolSchema[] + readonly schemas: readonly ToolSchema[] /** The pre-restriction name universe for config validation (defaults to `schemas`' names). */ - knownNames?: readonly string[] -} - -/** - * Canonical prompt contributions that survive the assembly waterfall. - * - * Protection is declarative by contribution name rather than an ordered - * callback: after every `system-prompt/assemble` listener has finished, the - * service restores each protected name to the exact presence and definition - * produced by its registries before the waterfall. Restored entries keep - * canonical order with one another and anchor before their first surviving - * later unprotected canonical neighbor (or at the end); the service does not - * undo a listener's reordering of unprotected entries. A name absent from that - * canonical assembly is removed from the result. This makes mode-dependent - * absence protectable too (for example, a native tool that intentionally stays - * off the wire in Code Mode). - */ -export interface PromptProtection { - /** Section names whose canonical presence and definition are restored after the waterfall. */ - sections?: readonly string[] - /** Tool names whose canonical presence and definition are restored after the waterfall. */ - tools?: readonly string[] + readonly knownNames?: readonly string[] + /** + * Tool names this provider owns finally. The names need not be present in + * `schemas`: naming a mode-hidden tool makes its canonical absence final, so + * an assembly listener cannot fabricate it onto the wire. + */ + readonly ownerFinalNames?: readonly string[] } /** @@ -234,76 +224,28 @@ function orderTools(tools: ToolSchema[], toolOrder: string[] | undefined, knownN name === TOOL_ORDER_REST ? rest : tools.filter(tool => tool.name === name)) } -/** Snapshot one waterfall-produced named entry with a stable, own data `name`. */ -function snapshotNamedEntry(entry: T): { entry: T; name: string } { - // Read the name exactly once before protection matching. The waterfall owns - // its output and may return accessor-backed records; retaining such an entry - // would let a getter answer "unprotected" during filtering and the protected - // name later when a consumer reads the final assembly. - const name = entry.name - const snapshot: Record = {} - Object.defineProperty(snapshot, 'name', { - value: name, - enumerable: true, - configurable: true, - writable: true, - }) - // Copy every other enumerable field once while deliberately skipping name. - // defineProperty keeps a literal "__proto__" extension field ordinary data. - for (const key of Object.keys(entry)) { - if (key === 'name') continue - Object.defineProperty(snapshot, key, { - value: (entry as unknown as Record)[key], - enumerable: true, - configurable: true, - writable: true, - }) - } - return { entry: snapshot as T, name } -} - -/** Restore protected named entries from `canonical`, anchored before their next unprotected canonical neighbor. */ -function restoreProtected( - canonical: readonly T[], result: readonly T[], protectedNames: ReadonlySet, +/** Restore owner-final entries from `canonical`, anchored before their next ordinary canonical neighbor. */ +function restoreOwnerFinal( + canonical: readonly T[], result: readonly T[], ownerFinalNames: ReadonlySet, ): T[] { - const restored = result - .map(snapshotNamedEntry) - .filter(record => !protectedNames.has(record.name)) + const restored = result.filter(entry => !ownerFinalNames.has(entry.name)) for (const [index, entry] of canonical.entries()) { - if (!protectedNames.has(entry.name)) continue + if (!ownerFinalNames.has(entry.name)) continue // Protected entries are inserted in canonical order. Anchor each one // before the first later UNPROTECTED canonical neighbor that survived the // waterfall; if none survived, it belongs at the end. Looking only at - // unprotected neighbors avoids reversing adjacent protected entries. + // ordinary neighbors avoids reversing adjacent owner-final entries. const following = new Set( canonical.slice(index + 1) - .filter(candidate => !protectedNames.has(candidate.name)) + .filter(candidate => !ownerFinalNames.has(candidate.name)) .map(candidate => candidate.name), ) const next = restored.findIndex(candidate => following.has(candidate.name)) - restored.splice(next < 0 ? restored.length : next, 0, { - entry: structuredClone(entry), - name: entry.name, - }) + // `canonical` is an owned snapshot made before the waterfall; no second + // clone is needed when moving its entries into the finalized assembly. + restored.splice(next < 0 ? restored.length : next, 0, entry) } - return restored.map(record => record.entry) -} - -/** Validate and detach one protection-name array without rereading an element. */ -function snapshotProtectionNames(value: unknown, field: 'sections' | 'tools'): readonly string[] { - if (!Array.isArray(value)) { - throw new TypeError(`systemPrompt.protect() ${field} must be an array of strings`) - } - const names: string[] = [] - const length = value.length - for (let index = 0; index < length; index += 1) { - const name: unknown = value[index] - if (typeof name !== 'string') { - throw new TypeError(`systemPrompt.protect() ${field} must be an array of strings`) - } - names.push(name) - } - return Object.freeze([...new Set(names)]) + return restored } /** Lexicographic (code-unit) name comparison — locale-independent, so the order is identical on every machine. */ @@ -423,7 +365,7 @@ function interpolate(section: AssembledSection, variables: Record ToolProviderResult)[] = [] private variableProviders = new Map string | undefined>() - private protections: PromptProtection[] = [] /** Per-scope layers (`@deepseek-ai/dsh-scope`); entries drop when a layer empties, so a disposed scope leaves no residue. */ private scopedSections = new Map() private scopedToolProviders = new Map ToolProviderResult)[]>() private scopedVariableProviders = new Map string | undefined>>() - private scopedProtections = new Map() private readonly toolOrder: string[] | undefined constructor(ctx: Context, public config: Config) { @@ -480,13 +420,11 @@ export class SystemPrompt extends Service { * scoped context (`agent.ctx`) contributes to that scope alone — and a * scoped section SHADOWS a same-named global section for that scope's * assemblies (most-specific-wins; this is how a per-agent persona overrides - * `deployment:persona`) unless that global name is protected: global - * protection reserves its section name against scoped shadows so the + * `deployment:persona`) unless that global contribution is owner-final: it + * reserves its section name against scoped shadows so the * registration owner—not a later scope—defines the canonical value. The - * registry reads `name`, `order`, and `text` once, validates their fixed - * string/finite-number/string-or-function types, and stores only that - * accepted record, so later caller-object mutation cannot rename or reshape - * a contribution. Throws if the SAME layer already has the name (a + * readonly typed contribution is borrowed until disposal; only the semantic + * finite-order rule is checked at runtime. Throws if the SAME layer already has the name (a * duplicate would silently double prompt text — e.g. a double-loaded tool * plugin; the global-duplicate message names `agent.ctx` as the per-agent * alternative). Removed when the calling fiber is disposed. Emits @@ -497,25 +435,20 @@ export class SystemPrompt extends Service { * yield it directly — exact identity nests the teardown in order. */ section(section: PromptSection): () => Promise | void { - const input: unknown = section - if (typeof input !== 'object' || input === null) { - throw new TypeError('systemPrompt.section() requires a section object') - } - const accepted = input as PromptSection - const name = accepted.name - const order = accepted.order - const text = accepted.text - if (typeof name !== 'string') throw new TypeError('prompt section name must be a string') - if (typeof order !== 'number' || !Number.isFinite(order)) { - throw new TypeError(`prompt section "${name}" order must be a finite number`) - } - if (typeof text !== 'string' && typeof text !== 'function') { - throw new TypeError(`prompt section "${name}" text must be a string or function`) + if (!Number.isFinite(section.order)) { + throw new TypeError(`prompt section "${section.name}" order must be a finite number`) } const scope = scopeOf(this.ctx) - const snapshot: PromptSection = { name, order, text } - if (scope !== undefined && this.protections.some(record => record.sections?.includes(snapshot.name))) { - throw new Error(`prompt section "${snapshot.name}" is globally protected and cannot be shadowed in an agent scope`) + if (scope !== undefined + && this.sections.some(global => global.name === section.name && global.ownerFinal === true)) { + throw new Error(`prompt section "${section.name}" is globally owner-final and cannot be shadowed in an agent scope`) + } + if (scope === undefined && section.ownerFinal === true) { + const hasScopedShadow = [...this.scopedSections.values()] + .some(layer => layer.some(scoped => scoped.name === section.name)) + if (hasScopedShadow) { + throw new Error(`owner-final prompt section "${section.name}" cannot be registered while a scoped shadow exists`) + } } const dispose = this.ctx.effect(function* (this: SystemPrompt) { const layer = scope === undefined @@ -525,18 +458,18 @@ export class SystemPrompt extends Service { this.scopedSections.set(scope, created) return created })() - if (layer.some(existing => existing.name === snapshot.name)) { + if (layer.some(existing => existing.name === section.name)) { throw new Error(scope === undefined - ? `prompt section "${snapshot.name}" is already registered (for a per-agent override, register through that agent's \`agent.ctx\` instead)` - : `prompt section "${snapshot.name}" is already registered in this scope`) + ? `prompt section "${section.name}" is already registered (for a per-agent override, register through that agent's \`agent.ctx\` instead)` + : `prompt section "${section.name}" is already registered in this scope`) } - layer.push(snapshot) + layer.push(section) // Yield the rollback BEFORE emitting `system-prompt/change`: a generator // effect collects each yielded disposer before the next step runs, so a // throwing change listener removes the section instead of leaking it into // every future assembly. yield () => { - const index = layer.indexOf(snapshot) + const index = layer.indexOf(section) /* v8 ignore next 3 -- defensive: section was registered, so indexOf is guaranteed >= 0 */ if (index >= 0) layer.splice(index, 1) if (scope !== undefined && layer.length === 0) this.scopedSections.delete(scope) @@ -560,8 +493,7 @@ export class SystemPrompt extends Service { * `schemas`/`knownNames` split). The layer is decided by the calling * context: a scoped provider (registered through `agent.ctx`) is consulted * only for that scope's assemblies. Removed when the calling fiber is - * disposed. A non-function provider is rejected before any effect is stored. - * A provider must not return a schema named + * disposed. A provider must not return a schema named * {@link TOOL_ORDER_REST}; that name is reserved for * {@link Config.toolOrder}'s rest entry and rejects the assembly. Emits * `system-prompt/change`. @@ -571,9 +503,6 @@ export class SystemPrompt extends Service { * yield it directly — exact identity nests the teardown in order. */ tools(provider: (context: AssembleContext) => ToolProviderResult): () => Promise | void { - if (typeof provider !== 'function') { - throw new TypeError('system prompt tool provider must be a function') - } const scope = scopeOf(this.ctx) const dispose = this.ctx.effect(function* (this: SystemPrompt) { const layer = scope === undefined @@ -611,8 +540,7 @@ export class SystemPrompt extends Service { * deployment must not claim facts it does not have). The layer is decided * by the calling context: a scoped variable (registered through * `agent.ctx`) resolves only for that scope's assemblies and SHADOWS a - * same-named global variable there. The fixed name and callback types are - * validated before effect storage. Throws on a name that does not match + * same-named global variable there. Throws on a name that does not match * `[a-z][a-z0-9_]*` (it could never be referenced) or one already registered * in the SAME layer. Removed when the calling fiber is disposed; emits * `system-prompt/change` on register/unregister. @@ -623,13 +551,8 @@ export class SystemPrompt extends Service { * yield it directly — exact identity nests the teardown in order. */ variable(name: string, provider: (context: AssembleContext) => string | undefined): () => Promise | void { - const inputName: unknown = name - if (typeof inputName !== 'string') throw new TypeError('prompt variable name must be a string') - if (!VARIABLE_NAME.test(inputName)) { - throw new Error(`invalid prompt variable name "${inputName}" (must match ${String(VARIABLE_NAME)})`) - } - if (typeof provider !== 'function') { - throw new TypeError(`prompt variable "${inputName}" provider must be a function`) + if (!VARIABLE_NAME.test(name)) { + throw new Error(`invalid prompt variable name "${name}" (must match ${String(VARIABLE_NAME)})`) } const scope = scopeOf(this.ctx) const dispose = this.ctx.effect(function* (this: SystemPrompt) { @@ -663,91 +586,6 @@ export class SystemPrompt extends Service { return dispose } - /** - * Protect named section/tool contributions from the assembly waterfall. - * The layer is decided by the calling context: a global protection applies - * to every assembly, while one registered through `agent.ctx` applies only - * to that agent's scope. The name's canonical registry/provider output is - * restored AFTER the whole waterfall, so listener registration order cannot - * strip, replace, duplicate, or fabricate it. Canonical absence is restored - * too: if the protected name is intentionally absent for an assembly, a - * listener-injected entry with that name is removed. Each optional field and - * array slot is read once; non-array fields or non-string names reject before - * effect storage, and the accepted deduplicated arrays are frozen. During - * finalization each waterfall-produced entry name is likewise read once into - * an owned data record, so a stateful getter cannot look unprotected during - * filtering and later impersonate a protected name. An empty protection - * throws because it cannot affect output. - * Removed with the calling fiber and emits `system-prompt/change` on - * registration/unregistration. A global section protection also reserves the - * name against scoped section shadows; registering protection when such a - * shadow already exists fails loudly instead of protecting the wrong owner. - * @param protection - section and/or tool names whose canonical presence and definitions are restored after the waterfall. - * @returns the exact Cordis effect disposer that removes the protection. - */ - protect(protection: PromptProtection): () => Promise | void { - const input: unknown = protection - if (typeof input !== 'object' || input === null) { - throw new TypeError('systemPrompt.protect() requires a protection object') - } - const accepted = input as PromptProtection - const inputSections = accepted.sections - const inputTools = accepted.tools - const sections = inputSections === undefined - ? undefined - : snapshotProtectionNames(inputSections, 'sections') - const tools = inputTools === undefined - ? undefined - : snapshotProtectionNames(inputTools, 'tools') - const scope = scopeOf(this.ctx) - const snapshot: PromptProtection = Object.freeze({ - ...sections !== undefined ? { sections } : {}, - ...tools !== undefined ? { tools } : {}, - }) - if ((snapshot.sections?.length ?? 0) === 0 && (snapshot.tools?.length ?? 0) === 0) { - throw new Error('systemPrompt.protect() requires at least one section or tool name') - } - if (scope === undefined && snapshot.sections !== undefined) { - const protectedSections = new Set(snapshot.sections) - const conflicts = [...this.scopedSections.values()] - .flatMap(layer => layer.filter(section => protectedSections.has(section.name)).map(section => section.name)) - if (conflicts.length > 0) { - throw new Error(`systemPrompt.protect() cannot globally protect section${conflicts.length > 1 ? 's' : ''} ${[...new Set(conflicts)].map(name => `"${name}"`).join(', ')} while scoped shadows are registered`) - } - } - const dispose = this.ctx.effect(function* (this: SystemPrompt) { - const layer = scope === undefined - ? this.protections - : this.scopedProtections.get(scope) ?? (() => { - const created: PromptProtection[] = [] - this.scopedProtections.set(scope, created) - return created - })() - layer.push(snapshot) - yield () => { - const index = layer.indexOf(snapshot) - /* v8 ignore next 3 -- defensive: protection was registered, so indexOf is guaranteed >= 0 */ - if (index >= 0) layer.splice(index, 1) - if (scope !== undefined && layer.length === 0) this.scopedProtections.delete(scope) - this.ctx.emit('system-prompt/change') - } - this.ctx.emit('system-prompt/change') - }.bind(this), 'systemPrompt.protect()') - return dispose - } - - /** Resolve the owner-final names registered for one assembly scope. */ - private protectedNames(scope: ScopeKey | undefined): { sections: Set; tools: Set } { - const records = [ - ...this.protections, - ...(scope === undefined ? [] : this.scopedProtections.get(scope)) ?? [], - ] - return { - sections: new Set(records.flatMap(record => record.sections ?? [])), - tools: new Set(records.flatMap(record => record.tools ?? [])), - } - } - /** * Assemble the current prompt for one caller: the global layer merged with * {@link AssembleContext.scope}'s layer (scoped sections/variables SHADOW @@ -760,14 +598,12 @@ export class SystemPrompt extends Service { * the providers' `knownNames` universe rejects the assembly, while a known * name restricted away for this scope is a normal absence), and every * visible variable resolved against `context` into `assembly.variables`. - * Each provider result and schema field is read once; those same captured - * names drive both `toolOrder` validation and the model-visible collection. - * Tool schemas are deep-cloned because adapters and request waterfalls may - * mutate schema objects. Runs through the `system-prompt/assemble` + * Tool schemas are detached because assembly waterfalls may mutate them. + * Runs through the `system-prompt/assemble` * waterfall, giving listeners the opportunity to mutate or replace the - * assembly, then restores every visible {@link PromptProtection} from the - * pre-waterfall canonical assembly. Like the sections' `order` sort, tool - * canonicalization happens on the initial assembly; unprotected listener + * assembly, then restores every contribution whose owner declared it final + * from the pre-waterfall canonical assembly. Like the sections' `order` sort, tool + * canonicalization happens on the initial assembly; ordinary listener * output owns its own determinism. Await the result before reading the * assembly values — waterfall listeners may be async. * Interpolation happens later, in {@link renderPrompt}. @@ -780,10 +616,6 @@ export class SystemPrompt extends Service { // (`assemble().catch(...)` would miss it). async assemble(context: AssembleContext = {}): Promise { const scope = context.scope - // Protection is a registry input too: snapshot which names are protected - // at assembly start. Registrations that land while an async waterfall is - // in flight affect the NEXT assembly, matching the other registries. - const protectedNames = this.protectedNames(scope) // Variables: global layer first, then the scope's layer OVERWRITES // same-named entries (shadowing — a per-agent value wins for that agent). const variables: Record = {} @@ -803,6 +635,11 @@ export class SystemPrompt extends Service { for (const section of (scope === undefined ? [] : this.scopedSections.get(scope)) ?? []) { sectionByName.set(section.name, section) } + const ownerFinalSections = new Set( + [...sectionByName.values()] + .filter(section => section.ownerFinal === true) + .map(section => section.name), + ) // Tools: consult the global providers plus the scope's, each with this // assembly's context. `schemas` are what the model may see (already // post-restriction, per provider); `knownNames` (defaulting to the @@ -815,45 +652,18 @@ export class SystemPrompt extends Service { ] const collected: ToolSchema[] = [] const knownNames = new Set() + const ownerFinalTools = new Set() for (const provider of providers) { const result = provider(context) - // One provider result snapshot: `schemas`, `knownNames`, and each schema - // field may be accessor-backed. The same captured names must drive both - // toolOrder validation and the model-visible collection. - const inputSchemas = result.schemas - const inputKnownNames = result.knownNames - const schemas = inputSchemas.map((tool, index): ToolSchema => { - const name = tool.name - const description = tool.description - const inputParameters = tool.parameters - if (typeof name !== 'string') { - throw new TypeError(`system prompt tool schema at index ${index} name must be a string`) - } - if (typeof description !== 'string') { - throw new TypeError(`system prompt tool "${name}" description must be a string`) - } - const parameters = snapshotJsonValue(inputParameters) - if (parameters === undefined) { - throw new TypeError(`system prompt tool "${name}" parameters must be losslessly JSON-serializable`) - } - return { name, description, parameters } - }) - let acceptedKnownNames: string[] - if (inputKnownNames === undefined) { - acceptedKnownNames = schemas.map(tool => tool.name) - } else { - if (!Array.isArray(inputKnownNames)) { - throw new TypeError('system prompt tool provider knownNames must be an array of strings') - } - acceptedKnownNames = Array.from(inputKnownNames, (name) => { - if (typeof name !== 'string') { - throw new TypeError('system prompt tool provider knownNames must be an array of strings') - } - return name - }) - } + const schemas = result.schemas.map(({ name, description, parameters }): ToolSchema => ({ + name, + description, + parameters: structuredClone(parameters), + })) + const acceptedKnownNames = result.knownNames ?? schemas.map(tool => tool.name) collected.push(...schemas) for (const name of acceptedKnownNames) knownNames.add(name) + for (const name of result.ownerFinalNames ?? []) ownerFinalTools.add(name) } const assembly: PromptAssembly = { sections: [...sectionByName.values()] @@ -866,11 +676,11 @@ export class SystemPrompt extends Service { tools: orderTools(collected, this.toolOrder, knownNames), variables, } - // Snapshot only the fields protection can restore. The waterfall receives + // Snapshot only the owner-final fields. The waterfall receives // `assembly` by reference and may mutate it or return a replacement; these // independent snapshots remain the authoritative registry product. - const canonicalSections = protectedNames.sections.size > 0 ? structuredClone(assembly.sections) : undefined - const canonicalTools = protectedNames.tools.size > 0 ? structuredClone(assembly.tools) : undefined + const canonicalSections = ownerFinalSections.size > 0 ? structuredClone(assembly.sections) : undefined + const canonicalTools = ownerFinalTools.size > 0 ? structuredClone(assembly.tools) : undefined const result = await this.ctx.waterfall( scopeTarget(this, scope), 'system-prompt/assemble', assembly, context, () => Promise.resolve(assembly), @@ -881,10 +691,10 @@ export class SystemPrompt extends Service { return { ...result, ...canonicalSections !== undefined - ? { sections: restoreProtected(canonicalSections, result.sections, protectedNames.sections) } + ? { sections: restoreOwnerFinal(canonicalSections, result.sections, ownerFinalSections) } : {}, ...canonicalTools !== undefined - ? { tools: restoreProtected(canonicalTools, result.tools, protectedNames.tools) } + ? { tools: restoreOwnerFinal(canonicalTools, result.tools, ownerFinalTools) } : {}, } } diff --git a/packages/core/system-prompt/tests/scoped.spec.ts b/packages/core/system-prompt/tests/scoped.spec.ts index d59e3a234a..329a66813f 100644 --- a/packages/core/system-prompt/tests/scoped.spec.ts +++ b/packages/core/system-prompt/tests/scoped.spec.ts @@ -63,19 +63,16 @@ describe('scoped sections', () => { expect(() => scope.ctx.systemPrompt.section({ name: 'y', order: 1, text: 'b' })).toThrow(/already registered in this scope/) }) - it.each([ - [['reserved'], 'section "reserved"'], - [['first', 'second'], 'sections "first", "second"'], - ])('rejects global protection added after scoped shadows (%j)', async (names, message) => { + it('rejects a global owner-final section added after a scoped shadow', async () => { const ctx = await mount() const scope = await mintScope(ctx, 'child') - for (const name of names) { - scope.ctx.systemPrompt.section({ name, order: 1, text: `scoped ${name}` }) - } + scope.ctx.systemPrompt.section({ name: 'reserved', order: 1, text: 'scoped reserved' }) - expect(() => ctx.systemPrompt.protect({ sections: names })).toThrow(message) + expect(() => ctx.systemPrompt.section({ + name: 'reserved', order: 1, text: 'global reserved', ownerFinal: true, + })).toThrow('owner-final prompt section "reserved"') expect(renderPrompt(await ctx.systemPrompt.assemble({ scope: scopeKeyOf(scope) }))) - .toContain(`scoped ${names[0]}`) + .toContain('scoped reserved') }) }) @@ -164,13 +161,18 @@ describe('scoped assemble dispatch', () => { expect(shaped).toHaveLength(1) }) - it('a scoped protection finalizes only its own assemblies and disappears with the scope', async () => { + it('scoped owner-final contributions finalize only their assemblies and disappear with the scope', async () => { const ctx = await mount() const scope = await mintScope(ctx, 'child') const key = scopeKeyOf(scope) ctx.systemPrompt.section({ name: 'required', order: 10, text: 'required' }) ctx.systemPrompt.tools(() => ({ schemas: [schema('required')] })) - scope.ctx.systemPrompt.protect({ sections: ['required'], tools: ['required'] }) + scope.ctx.systemPrompt.section({ + name: 'required', order: 10, text: 'scoped required', ownerFinal: true, + }) + scope.ctx.systemPrompt.tools(() => ({ + schemas: [schema('required')], ownerFinalNames: ['required'], + })) ctx.on('system-prompt/assemble', async (_assembly, _context, next) => { const result = await next() result.sections = result.sections.filter(section => section.name !== 'required') diff --git a/packages/core/system-prompt/tests/system-prompt.spec.ts b/packages/core/system-prompt/tests/system-prompt.spec.ts index c37ddd9c6c..3417364613 100644 --- a/packages/core/system-prompt/tests/system-prompt.spec.ts +++ b/packages/core/system-prompt/tests/system-prompt.spec.ts @@ -111,86 +111,14 @@ describe('SystemPrompt', () => { expect(contributed(assembly).map(s => s.text)).toEqual(['first']) }) - it('rejects malformed fixed registration fields before storing an effect', async () => { + it('rejects a non-finite section order', async () => { const ctx = new Context() await ctx.plugin(SystemPrompt) - const badName = { value: 'name' } - const badText = { value: 'text' } - - expect(() => ctx.systemPrompt.section(null as unknown as Parameters[0])) - .toThrow('requires a section object') - expect(() => ctx.systemPrompt.section(1 as unknown as Parameters[0])) - .toThrow('requires a section object') - expect(() => ctx.systemPrompt.section({ name: badName as unknown as string, order: 1, text: 'x' })) - .toThrow('prompt section name must be a string') - expect(() => ctx.systemPrompt.section({ name: 'bad-order', order: '1' as unknown as number, text: 'x' })) - .toThrow('order must be a finite number') expect(() => ctx.systemPrompt.section({ name: 'bad-order', order: Number.NaN, text: 'x' })) .toThrow('order must be a finite number') - expect(() => ctx.systemPrompt.section({ name: 'bad-text', order: 1, text: badText as unknown as string })) - .toThrow('text must be a string or function') - expect(() => ctx.systemPrompt.tools(1 as unknown as Parameters[0])) - .toThrow('tool provider must be a function') - expect(() => ctx.systemPrompt.variable({} as unknown as string, () => 'x')) - .toThrow('prompt variable name must be a string') - expect(() => ctx.systemPrompt.variable('valid', 1 as unknown as Parameters[1])) - .toThrow('provider must be a function') - expect(() => ctx.systemPrompt.protect(null as unknown as Parameters[0])) - .toThrow('requires a protection object') - expect(() => ctx.systemPrompt.protect(1 as unknown as Parameters[0])) - .toThrow('requires a protection object') - expect(() => ctx.systemPrompt.protect({ sections: 'x' as unknown as string[] })) - .toThrow('sections must be an array of strings') - expect(() => ctx.systemPrompt.protect({ tools: 'x' as unknown as string[] })) - .toThrow('tools must be an array of strings') - expect(() => ctx.systemPrompt.protect({ sections: ['ok', {} as unknown as string] })) - .toThrow('sections must be an array of strings') - expect(() => ctx.systemPrompt.protect({ tools: [{} as unknown as string] })) - .toThrow('tools must be an array of strings') - - expect(Object.isFrozen(badName)).toBe(false) - expect(Object.isFrozen(badText)).toBe(false) expect(contributed(await ctx.systemPrompt.assemble())).toEqual([]) }) - it('reads each section field and protection-name slot once at registration', async () => { - const ctx = new Context() - await ctx.plugin(SystemPrompt) - const reads = { name: 0, order: 0, text: 0, sections: 0, item: 0 } - const section = Object.defineProperties({}, { - name: { - enumerable: true, - get: () => (++reads.name === 1 ? 'stable' : 42), - }, - order: { - enumerable: true, - get: () => (++reads.order === 1 ? 10 : Number.NaN), - }, - text: { - enumerable: true, - get: () => (++reads.text === 1 ? 'stable text' : null), - }, - }) as unknown as Parameters[0] - const names = new Array(1) - Object.defineProperty(names, 0, { - enumerable: true, - get: () => (++reads.item === 1 ? 'stable' : 'drifted'), - }) - const protection = { - get sections(): string[] { - reads.sections += 1 - return reads.sections === 1 ? names : ['drifted'] - }, - } - - ctx.systemPrompt.section(section) - ctx.systemPrompt.protect(protection) - const assembly = await ctx.systemPrompt.assemble() - - expect(reads).toEqual({ name: 1, order: 1, text: 1, sections: 1, item: 1 }) - expect(assembly.sections).toContainEqual({ name: 'stable', order: 10, text: 'stable text' }) - }) - it('rolls back a section when a system-prompt/change listener throws (P1-1)', async () => { const ctx = new Context() await ctx.plugin(SystemPrompt) @@ -285,28 +213,21 @@ describe('SystemPrompt', () => { expect(assembly.sections).toHaveLength(0) }) - describe('canonical contribution protection', () => { - it('restores exact protected definitions after every listener, in canonical relative order', async () => { + describe('owner-final contributions', () => { + it('restores exact owner-final definitions after every listener, in canonical relative order', async () => { const ctx = new Context() await ctx.plugin(SystemPrompt) ctx.systemPrompt.section({ name: 'before', order: 10, text: 'before' }) - ctx.systemPrompt.section({ name: 'protected', order: 20, text: 'canonical section' }) + ctx.systemPrompt.section({ name: 'protected', order: 20, text: 'canonical section', ownerFinal: true }) ctx.systemPrompt.section({ name: 'after', order: 30, text: 'after' }) ctx.systemPrompt.tools(() => ({ schemas: [ { name: 'alpha', description: 'alpha', parameters: {} }, { name: 'protected', description: 'canonical tool', parameters: { type: 'object', properties: { answer: { type: 'number' } } } }, { name: 'zulu', description: 'zulu', parameters: {} }, - ] })) - const protection = { sections: ['protected'], tools: ['protected'] } - ctx.systemPrompt.protect(protection) - // Registration snapshots its arrays; caller mutation cannot change what - // the service makes authoritative. - protection.sections[0] = 'after' - protection.tools[0] = 'zulu' + ], ownerFinalNames: ['protected'] })) - // Registered AFTER the protection and prepended: it is outside every - // ordinary listener that existed when protect() ran, but service-level - // finalization still restores the canonical entries after it returns. + // Service-level finalization restores the canonical entries after the + // complete listener chain returns. ctx.on('system-prompt/assemble', async (_assembly, _context, next) => { const result = await next() return Object.freeze({ @@ -338,121 +259,18 @@ describe('SystemPrompt', () => { expect(assembly.tools.map(tool => tool.name)).toEqual(['alpha', 'protected', 'zulu']) }) - it('reads protection accessors once so the checked names are the protected names', async () => { + it('makes an owner-final tool\'s canonical absence survive the waterfall', async () => { const ctx = new Context() await ctx.plugin(SystemPrompt) - ctx.systemPrompt.section({ name: 'protected', order: 10, text: 'canonical' }) - let reads = 0 - const protection = { - get sections(): string[] { - reads += 1 - return reads === 1 ? ['protected'] : undefined as unknown as string[] - }, - } - ctx.systemPrompt.protect(protection) + ctx.systemPrompt.tools(() => ({ schemas: [], ownerFinalNames: ['mode-hidden'] })) ctx.on('system-prompt/assemble', async (_assembly, _context, next) => { const result = await next() - result.sections = result.sections.filter(section => section.name !== 'protected') - return result - }) - - const assembly = await ctx.systemPrompt.assemble() - - expect(reads).toBe(1) - expect(assembly.sections).toContainEqual({ name: 'protected', order: 10, text: 'canonical' }) - }) - - it('materializes waterfall entry names once before restoring protected definitions', async () => { - const ctx = new Context() - await ctx.plugin(SystemPrompt) - ctx.systemPrompt.section({ name: 'protected', order: 10, text: 'canonical section' }) - ctx.systemPrompt.tools(() => ({ schemas: [{ name: 'protected', description: 'canonical tool', parameters: {} }] })) - ctx.systemPrompt.protect({ sections: ['protected'], tools: ['protected'] }) - let sectionNameReads = 0 - let toolNameReads = 0 - const hostileSection = { - get name(): string { - sectionNameReads += 1 - return sectionNameReads === 1 ? 'impostor-section' : 'protected' - }, - order: 999, - text: 'listener section', - } - const hostileTool = { - get name(): string { - toolNameReads += 1 - return toolNameReads === 1 ? 'impostor-tool' : 'protected' - }, - description: 'listener tool', - parameters: {}, - } - ctx.on('system-prompt/assemble', async (_assembly, _context, next) => { - const result = await next() - result.sections = [ - ...result.sections.filter(section => section.name !== 'protected'), - hostileSection, - ] - result.tools = [ - ...result.tools.filter(tool => tool.name !== 'protected'), - hostileTool, - ] - return result - }) - - const assembly = await ctx.systemPrompt.assemble() - - expect(sectionNameReads).toBe(1) - expect(toolNameReads).toBe(1) - expect(assembly.sections.map(section => section.name)).toEqual([ - 'harness:identity', - 'deployment:persona', - 'impostor-section', - 'protected', - ]) - expect(assembly.tools.map(tool => tool.name)).toEqual(['impostor-tool', 'protected']) - }) - - it('protects canonical absence and rejects an empty protection', async () => { - const ctx = new Context() - await ctx.plugin(SystemPrompt) - // Separate registrations exercise the set-union contract: protections - // may name only sections or only tools and still compose. - ctx.systemPrompt.protect({ sections: ['mode-hidden'] }) - ctx.systemPrompt.protect({ tools: ['mode-hidden'] }) - ctx.on('system-prompt/assemble', async (_assembly, _context, next) => { - const result = await next() - result.sections.push({ name: 'mode-hidden', order: 100, text: 'fabricated' }) result.tools.push({ name: 'mode-hidden', description: 'fabricated', parameters: {} }) return result }) const assembly = await ctx.systemPrompt.assemble() - expect(assembly.sections.some(section => section.name === 'mode-hidden')).toBe(false) expect(assembly.tools.some(tool => tool.name === 'mode-hidden')).toBe(false) - expect(() => ctx.systemPrompt.protect({})).toThrow(/at least one section or tool name/) - expect(() => ctx.systemPrompt.protect({ sections: [], tools: [] })).toThrow(/at least one section or tool name/) - }) - - it('removes a protection with its contributing fiber (HMR safety)', async () => { - const ctx = new Context() - await ctx.plugin(SystemPrompt) - ctx.systemPrompt.section({ name: 'protected', order: 10, text: 'canonical' }) - ctx.on('system-prompt/assemble', async (_assembly, _context, next) => { - const result = await next() - result.sections = result.sections.filter(section => section.name !== 'protected') - return result - }) - let changes = 0 - ctx.on('system-prompt/change', () => { changes++ }) - const fiber = await ctx.plugin(Object.assign((inner: Context) => { - inner.systemPrompt.protect({ sections: ['protected'] }) - }, { inject: ['systemPrompt'] })) - - expect((await ctx.systemPrompt.assemble()).sections.some(section => section.name === 'protected')).toBe(true) - expect(changes).toBe(1) - await fiber.dispose() - expect((await ctx.systemPrompt.assemble()).sections.some(section => section.name === 'protected')).toBe(false) - expect(changes).toBe(2) }) }) diff --git a/packages/core/system-prompt/tests/tool-order.spec.ts b/packages/core/system-prompt/tests/tool-order.spec.ts index 088c6b70ed..16eff6e354 100644 --- a/packages/core/system-prompt/tests/tool-order.spec.ts +++ b/packages/core/system-prompt/tests/tool-order.spec.ts @@ -48,100 +48,6 @@ describe('SystemPrompt tool order', () => { expect(names(await ctx.systemPrompt.assemble())).toEqual(['todo_write', 'echo_a', 'echo_b', 'bash']) }) - it('reads provider schemas once so toolOrder validates the model-visible collection', async () => { - const ctx = await mount({ toolOrder: ['actual', TOOL_ORDER_REST] }) - let reads = 0 - ctx.systemPrompt.tools(() => ({ - get schemas(): ToolSchema[] { - reads += 1 - return reads === 1 ? [tool('actual')] : [tool('phantom')] - }, - })) - - const assembly = await ctx.systemPrompt.assemble() - - expect(reads).toBe(1) - expect(names(assembly)).toEqual(['actual']) - }) - - it('reads each provider schema field once before detaching it', async () => { - const ctx = await mount() - const accepted = { type: 'object', properties: { accepted: { type: 'string' } } } - let reads = 0 - const schema = { - name: 'stable', - description: 'stable', - get parameters(): object { - reads += 1 - return reads === 1 ? accepted : { type: 'object', properties: { drifted: { type: 'number' } } } - }, - } as ToolSchema - ctx.systemPrompt.tools(() => ({ schemas: [schema] })) - - const assembly = await ctx.systemPrompt.assemble() - - expect(reads).toBe(1) - expect(assembly.tools[0]?.parameters).toEqual(accepted) - }) - - it('rejects exotic provider parameters before model-visible assembly', async () => { - const ctx = await mount() - class ExoticParameters { - readonly type = 'object' - readonly properties = { value: { type: 'string' } } - } - ctx.systemPrompt.tools(() => ({ - schemas: [{ - name: 'exotic', - description: 'must not be sanitized', - parameters: new ExoticParameters() as unknown as ToolSchema['parameters'], - }], - })) - - await expect(ctx.systemPrompt.assemble()) - .rejects.toThrow(/parameters must be losslessly JSON-serializable/) - }) - - it('rejects malformed fixed provider fields without freezing caller objects', async () => { - const ctx = await mount() - const badName = { value: 'object-name' } - const badDescription = { value: 'object-description' } - ctx.systemPrompt.tools(() => ({ - schemas: [{ - name: badName as unknown as string, - description: 'bad name', - parameters: {}, - }], - })) - await expect(ctx.systemPrompt.assemble()).rejects.toThrow('name must be a string') - expect(Object.isFrozen(badName)).toBe(false) - - const descriptions = await mount() - descriptions.systemPrompt.tools(() => ({ - schemas: [{ - name: 'bad-description', - description: badDescription as unknown as string, - parameters: {}, - }], - })) - await expect(descriptions.systemPrompt.assemble()).rejects.toThrow('description must be a string') - expect(Object.isFrozen(badDescription)).toBe(false) - - const knownNames = await mount() - knownNames.systemPrompt.tools(() => ({ - schemas: [tool('valid')], - knownNames: [{} as unknown as string], - })) - await expect(knownNames.systemPrompt.assemble()).rejects.toThrow('knownNames must be an array of strings') - - const nonArrayKnownNames = await mount() - nonArrayKnownNames.systemPrompt.tools(() => ({ - schemas: [tool('valid')], - knownNames: 'valid' as unknown as string[], - })) - await expect(nonArrayKnownNames.systemPrompt.assemble()).rejects.toThrow('knownNames must be an array of strings') - }) - it('rejects the assembly when toolOrder names a tool that is not registered (misconfiguration blocks work)', async () => { const ctx = await mount({ toolOrder: ['todo_write', 'ghost', TOOL_ORDER_REST, 'wraith'] }) ctx.systemPrompt.tools(() => ({ schemas: [tool('bash'), tool('todo_write')] })) diff --git a/packages/core/system-prompt/tsconfig.json b/packages/core/system-prompt/tsconfig.json index a66ece4854..91e7bf1ba4 100644 --- a/packages/core/system-prompt/tsconfig.json +++ b/packages/core/system-prompt/tsconfig.json @@ -22,9 +22,6 @@ }, { "path": "../../core/scope" - }, - { - "path": "../../core/session" } ] } diff --git a/packages/core/tools/README.md b/packages/core/tools/README.md index 39874a41bd..a50e35a4b4 100644 --- a/packages/core/tools/README.md +++ b/packages/core/tools/README.md @@ -11,18 +11,16 @@ tools: mode: native # native (default) | code | both ``` -`native` contributes the calling agent's visible end capabilities as wire function definitions. Under `code`, this registry's canonical contribution is the reserved `run_code` transport plus the generated `tools:sdk` prompt section (see [Code Mode](#code-mode)); an assembly listener may still deliberately add unrelated schemas. `both` contributes the visible native definitions, `run_code`, and the SDK section. In non-native modes both infrastructure pieces are protected rather than filterable capabilities: restrictions and assembly listeners cannot remove them, a scoped section cannot shadow the globally protected `tools:sdk`, and registering, shadowing, or explicitly filtering `run_code` fails loudly. Non-native modes require a loaded `ctx.codeRuntime` with `language: 'typescript'`; a missing or mismatched runtime rejects every prompt assembly with an actionable error, and a `systemPrompt.toolOrder` naming tools the mode no longer contributes rejects the assembly the same way. +`native` contributes the calling agent's visible end capabilities as wire function definitions. Under `code`, this registry's canonical contribution is the reserved `run_code` transport plus the generated `tools:sdk` prompt section (see [Code Mode](#code-mode)); an assembly listener may still deliberately add unrelated schemas. `both` contributes the visible native definitions, `run_code`, and the SDK section. In non-native modes both infrastructure pieces are owner-final rather than filterable capabilities: restrictions and assembly listeners cannot remove them, a scoped section cannot shadow the globally owner-final `tools:sdk`, and registering, shadowing, or explicitly filtering `run_code` fails loudly. Non-native modes require a loaded `ctx.codeRuntime` with `language: 'typescript'`; a missing or mismatched runtime rejects every prompt assembly with an actionable error, and a `systemPrompt.toolOrder` naming tools the mode no longer contributes rejects the assembly the same way. ### Public API -- `ctx.tools.register(definition: ToolDefinition): () => Promise | void` Register a tool as a frozen snapshot. Every top-level caller field is read once into one coherent acceptance record; `name`/`description` must be strings and `timeoutMs`, when present, must be positive and finite before the snapshot can own them. Parameters are validated and detached by one recursive lossless-JSON traversal, so a stateful getter cannot show one value to a check and another to a prototype-erasing clone. Execute/presentation callbacks are bound once to the original definition as their method receiver, so later caller mutation cannot change the executable definition. The layer is the CALLING context's scope (`dsh-scope`): a plain plugin context registers globally; an agent's `agent.ctx` registers for that agent alone, SHADOWING a same-named global tool there (per-agent tool variants). Duplicate names within one layer throw; non-native modes also reject the reserved `run_code` transport name. Disposed with the calling fiber (= the agent, for scoped registrations). -- `ctx.tools.restrict(filter: ToolRestriction): () => Promise | void` Scoped-only (throws on a plain context): mask the GLOBAL end-capability surface for the calling agent — `allow` keeps only the listed tools, `deny` removes them; multiple restrictions intersect; scope-local registrations are merged after the global filter. The registry reads `allow`/`deny` once, so the values checked for an empty filter and unknown names are exactly the values enforced. A deny-list admits a later global tool unless it names that tool; an allow-list excludes later names; neither filters a later scope-local registration. The reserved `run_code` transport remains available automatically and cannot be named explicitly. Filter-value snapshot at registration, loud unknown-name validation, `restrict({})` rejects (the materialized-empty-config trap). This is live registration composition, not a parent-derived authority ceiling; see the [agent-scope security non-goal](../../../docs/rfc/implemented/architecture/2026-07-08-agent-scope-contexts.md#security-and-authority-are-explicit-non-goals). -- `ctx.tools.get(name: string, scope?: ScopeKey): ToolDefinition | undefined` Resolution as one scope sees it (shadowing applied; a restricted-away global reads as absent) — presenters pass the calling agent so the card matches what executed. Returned definitions are the registry's frozen snapshots. -- `ctx.tools.visible(scope?: ScopeKey): ToolDefinition[]` The canonical executable view — restricted global layer ∪ the scope's own layer, plus the reserved transport in non-native modes — feeding prompt assembly, `get`, and `execute`, so presentation and dispatch resolve the same frozen definitions. -- `ctx.tools.knownNames(scope?: ScopeKey): string[]` The PRE-restriction end-capability name universe `restrict` validates against: a typo fails loud while a restricted-away tool stays a normal absence. Presentation providers add reserved transport names separately when validating `toolOrder`. +- `ctx.tools.register(definition: ToolDefinition): () => Promise | void` Register a trusted typed same-process definition. The layer is the calling context's scope: a plain plugin context registers globally; an agent's `agent.ctx` registers for that agent alone, shadowing a same-named global tool there. Duplicate names within one layer throw; non-native modes also reject the reserved `run_code` transport name. `timeoutMs`, when present, must be positive and finite. `ownerFinal: true` makes the tool's canonical wire presence or absence survive prompt assembly listeners. Disposed with the calling fiber. +- `ctx.tools.restrict(filter: ToolRestriction): () => Promise | void` Scoped-only (throws on a plain context): mask the global end-capability surface for the calling agent — `allow` keeps only the listed global tools, `deny` removes them; multiple restrictions intersect; scope-local registrations are merged afterward. The readonly arrays compile once into private sets. Every listed name must exist in the current pre-restriction global registry; scope-local, unknown, and reserved `run_code` names fail loudly. A deny-list admits a later global tool unless it names that tool; an allow-list excludes later names; neither filters a later scope-local registration. `restrict({})` rejects. This is live registration composition, not a parent-derived authority ceiling; see the [agent-scope security non-goal](../../../docs/rfc/implemented/architecture/2026-07-08-agent-scope-contexts.md#security-and-authority-are-explicit-non-goals). +- `ctx.tools.get(name: string, scope?: ScopeKey): ToolDefinition | undefined` Resolution as one scope sees it (shadowing applied; a restricted-away global reads as absent) — presenters pass the calling agent so the card matches what executed. - `ctx.tools.schemas(scope?: ScopeKey): ToolSchema[]` Schemas of everything the scope can see (without the `execute` functions). The shipped tools' schemas are catalogued in [docs/tool-catalog.md](../../../docs/tool-catalog.md), generated by booting each tool plugin and harvesting this method (see [the tool-schema-catalog RFC](../../../docs/rfc/implemented/process/2026-07-02-tool-schema-catalog.md)). - `ctx.tools.guard(guard: ToolGuard): () => Promise | void` Register a monotonic synchronous execution guard after `tools/pre-execute`: returning a reason denies the call, while `undefined` leaves it unchanged. A plain-context guard applies globally; an `agent.ctx` guard applies only to that agent. Later waterfall listeners cannot turn a guard denial back into permission. Disposed with the calling fiber. -- `ctx.tools.execute(exec: ToolExecutionInput): Promise` Read each caller-owned top-level field once, require `callId` and `name` to yield strings, snapshot the single-use call into a pipeline-owned execution, assign its opaque correlation token, materialize `arguments` through one lossless-JSON traversal, deep-freeze them, and protect identity before running `tools/pre-execute` → guards → `tools/execute` → `tools/post-execute`; optional `signal` is the only operational field an around-dispatch wrapper may add, replace, or remove. After the required string correlation identity is captured, the same captured optional fields build the normalized error shell if a later accessor or validation fails, so policy, dispatch, routing, and `tools/result` cannot observe different caller values. Every top-level result field is likewise captured once and the complete result or post-decision is losslessly materialized before final observation. Invalid later input—including cloneable mutable exotics—and malformed or non-JSON listener/tool results normalize to `isError` outcomes rather than bypassing policy or failing later at the session log. A throwing accessor or non-string value in `callId` or `name` rejects before `tools/result` because no trustworthy result identity exists yet. +- `ctx.tools.execute(exec: ToolExecutionInput): Promise` Assign a fresh opaque correlation token, losslessly materialize and deep-freeze arguments once at the model/tool boundary, then run the call through `tools/pre-execute` → guards → `tools/execute` → `tools/post-execute`. Invalid arguments normalize through the same authoritative result path without reaching policy or the body. The final outcome is independently materialized and deep-frozen once before `tools/result`. Optional `signal` remains the operational field an around-dispatch wrapper may replace. ### Injected services @@ -34,12 +32,12 @@ The live registry pipeline has three transformable waterfalls followed by the ow ### Key types -- `ToolDefinition` — `ToolSchema` + `execute(args, exec): Promise` (the bare array is the model-facing content; the object form additionally attaches an opaque, JSON-serializable `meta` presentation payload persisted on the `tool/result` event and handed back to `presentResult`), plus optional `presentCall(args)` / `presentResult(args, result)` for tool-owned UI presentation (see below). It also carries an optional cooperative timeout budget `timeoutMs?: number` (ms) enforced by `@deepseek-ai/dsh-timeout-policy`, never sent to the model. Registration stores a frozen snapshot with detached JSON parameters and once-bound callback identities. -- `ToolExecutionInput` — the caller-supplied call description: `{ callId, name, arguments, agent?, parent?, signal? }`; `callId` and `name` must be strings, `arguments` must be losslessly JSON-serializable, and callers may pass an enclosing execution's opaque token as `parent` but never choose the new execution's own token. -- `ToolExecutionToken` — a frozen, property-free identity value assigned by the registry. It supports equality correlation only and exposes no live outer execution state. +- `ToolDefinition` — `ToolSchema` + `execute(args, exec)`, optional presentation callbacks, cooperative `timeoutMs`, and optional `ownerFinal`. `ownerFinal` is reserved for protocol tools such as `run_code` and structured-output capture whose canonical wire state must survive assembly listeners. +- `ToolExecutionInput` — the caller-supplied call description: `{ callId, name, arguments, agent?, parent?, signal? }`; callers may pass an enclosing execution's opaque token as `parent` but never choose the new execution's own token. +- `ToolExecutionToken` — a fresh branded `Symbol` assigned by the registry. It supports equality correlation only and never crosses a model, log, or worker boundary. - `ToolExecution` — the pipeline-owned call: immutable `{ token, callId, name, arguments, agent?, parent? }` identity plus optional operational `signal`, which an around wrapper may add, replace, remove, and restore. A nested call's `parent` is a `ToolExecutionToken`, not an execution object. -- `ToolExecutionResult` — losslessly JSON-serializable outcome: `{ callId, content, isError, error?, additionalContext?, meta? }`. The registry validates the complete post-policy value before final observation. On failure with a `HarnessError`, `error: { name, code }` carries the structured failure class alongside the model-facing text (the loop forwards it onto the `tool/result` session event for retry/sandbox plugins and replay). `additionalContext` (a `HookContext`) ferries any `tools/post-execute` context up to the loop, which buffers it and appends it as a `context/message` after all `tool/result`s in the step. `meta` is the tool's opaque presentation payload from a successful `execute` (the object return form); the loop forwards it onto the `tool/result` session event for result-card rendering. -- `PreToolDecision` — `{kind:'allow'}` | `{kind:'deny', reason}` | `{kind:'ask', reason?}`. The registry validates this exact union at runtime: a JavaScript/casted value with an unknown kind, a malformed reason, or extra fields fails closed as an `isError`; the tool body does not run and final observers still receive one result. Input rewrite (changing `arguments`) is deliberately NOT offered (it would desync the pre-execution audit/history/UI from what ran — its own proposed RFC); `ask` is serviced by [`ctx.approval`](../../ui/user-approval/README.md) when a deployment mounts it (`allowed-once` proceeds to dispatch; `rejected`/`cancelled`/`unavailable` deny with distinct reasons) and degrades to `deny` when none is mounted or the execution carries no agent. +- `ToolExecutionResult` — losslessly JSON-serializable outcome: `{ callId, content, isError, error?, additionalContext?, meta? }`. The registry materializes and freezes the complete post-policy value before final observation. On failure with a `HarnessError`, `error: { name, code }` carries the structured failure class alongside the model-facing text. +- `PreToolDecision` — `{kind:'allow'}` | `{kind:'deny', reason}` | `{kind:'ask', reason?}`. Input rewrite is deliberately not offered; `ask` is serviced by [`ctx.approval`](../../ui/user-approval/README.md) when mounted and otherwise degrades to deny. - `PostToolDecision` — `{kind:'accept', content?, additionalContext?}` (keep the call successful, optionally replacing the model-facing content) | `{kind:'block', feedback, additionalContext?}` (turn it into an `isError` whose content is the corrective feedback). Output replacement is clean because `tool/result` is logged AFTER `execute()` returns. - `ToolGuard` — `(execution) => string | undefined`; the returned string is a final monotonic denial reason evaluated after the reorderable pre-execute waterfall and before dispatch. - `ToolCallView` / `ToolResultView` — provider-neutral `card`-tagged render intents a tool returns from `presentCall` / `presentResult` to own how a UI renders ITS calls (see "Tool-owned UI presentation"). @@ -77,7 +75,7 @@ ctx.tools.register(defineTool({ })) ``` -The helper converts the author-facing `SchemaSpec` (with `required: true` as a per-property boolean) to standard JSON Schema for the wire format. Definition is a snapshot boundary: `defineTool` reads every top-level option once, detaches the schema, and derives both an independent wire schema and every later execute/presentation validation from that accepted snapshot. Stateful accessors or later caller mutation therefore cannot make the schema shown to the model disagree with the schema enforced at runtime. Raw JSON-Schema tool definitions (from MCP servers) are still accepted by the registry directly. +The helper converts the author-facing `SchemaSpec` (with `required: true` as a per-property boolean) to standard JSON Schema for the wire format and uses the same typed spec for execute/presentation validation. Raw JSON-Schema tool definitions (from MCP servers) are still accepted by the registry directly. A `defineTool` tool also **validates the model-generated arguments against its `SchemaSpec` before `execute` runs** (`validateArgs`). The model's JSON is untrusted — `InferArgs` is a compile-time claim, not a runtime guarantee — so on a mismatch (missing required key, wrong primitive, bad enum member, nested violation) the tool throws a `ToolArgsError` (`code: 'INVALID_ARGS'`); the registry turns it into an `isError` result whose text lists the violations, which the model sees and self-corrects from. Validation mirrors the JSON Schema conversion exactly: extra keys are allowed, `default` is not applied, and an `object`/`array` prop without `properties`/`items` only type-checks. Raw-registered tools (MCP) are **not** validated by the harness — they validate their own input. @@ -138,7 +136,7 @@ Under `mode: code` (or `both`) the registry turns the tool surface into a progra - **The dispatch bridge** (`run_code`'s execute): every binding call is JSON-normalized before dispatch (a value that does not survive — `BigInt`, circulars — rejects that one call, so the dispatched form and logged form are the same JSON value by construction), serialized through a per-run queue (even `Promise.all` executes underlying calls one at a time in submission order), given the outer execution's opaque token as `parent`, and run through the complete pre-execute → guards → execute → post-execute → result pipeline. A denial reaches the program as a binding rejection, and each sub-call is logged as a `tool/code-dispatch` session event with deterministic id `:code:`; `deriveMessages()` does not surface that event. Token correlation lets commit-style observers defer an inner success until the final `run_code` result without exposing the live outer execution; ordinary tool side effects are not rolled back. A sub-call's `additionalContext` is deliberately dropped because inserting it inside a running parent call would break tool-call/result adjacency. - **Settlement discipline**: the bridge owns a run-scoped abort that follows the outer signal in and fires when the run settles for any reason, so a budget expiry aborts an in-flight sub-tool instead of orphaning it; the bridge then drains its queue BEFORE returning, so every `tool/code-dispatch` lands inside the open turn. A failed run throws `CodeRunFailedError` (`code: 'CODE_RUN_FAILED'`, message = the failure kind + captured logs), which the pipeline converts to a structured `isError` the model self-corrects from. -The wire collapse is the registry's own contribution (`systemPrompt.tools()` is mode-aware), so the logged `request/header` records it for free. With no deliberate schema-adding assembly listener, `code` assembles exactly `[run_code]`, pinned by tests and the snapshot goldens; protection guarantees that `run_code` and `tools:sdk` remain present, not that unrelated listener additions are erased. Try it: `pnpm run demo:code-mode` ([the coding-agent example's Code Mode overlay](../../../examples/coding-agent/README.md#code-mode)); `pnpm run demo:code-mode acp` serves the same mode over ACP instead of the REPL. +The wire collapse is the registry's own contribution (`systemPrompt.tools()` is mode-aware), so the logged `request/header` records it for free. With no deliberate schema-adding assembly listener, `code` assembles exactly `[run_code]`, pinned by tests and the snapshot goldens; contribution-owned finality guarantees that `run_code` and `tools:sdk` remain present, not that unrelated listener additions are erased. Try it: `pnpm run demo:code-mode` ([the coding-agent example's Code Mode overlay](../../../examples/coding-agent/README.md#code-mode)); `pnpm run demo:code-mode acp` serves the same mode over ACP instead of the REPL. ### What is NOT here (TODO) diff --git a/packages/core/tools/src/code-mode.ts b/packages/core/tools/src/code-mode.ts index 323fbeed2b..702a403b03 100644 --- a/packages/core/tools/src/code-mode.ts +++ b/packages/core/tools/src/code-mode.ts @@ -151,6 +151,7 @@ function asRunCodeMeta(meta: unknown): RunCodeMeta | undefined { export function createRunCodeTool(registry: ToolRegistry, requireRuntime: () => CodeRuntime): ToolDefinition { return defineTool({ name: RUN_CODE_NAME, + ownerFinal: true, description: 'Execute a TypeScript program against the available tools. Write the BODY of an ' + 'async function (erasable syntax only; top-level `await` and `return` work) and ' diff --git a/packages/core/tools/src/index.ts b/packages/core/tools/src/index.ts index 4dd59efb8c..84a1ec3c51 100644 --- a/packages/core/tools/src/index.ts +++ b/packages/core/tools/src/index.ts @@ -91,9 +91,6 @@ declare module 'cordis' { * tool body never runs. Input rewrite is deliberately NOT offered here (see * {@link PreToolDecision}); `ask` is serviced by the `ctx.approval` seam * when one is mounted, and degrades to deny otherwise. - * The returned union is validated as an exact runtime shape before approval - * or guards run; a malformed JavaScript/casted decision fails closed as an - * `isError` result and the tool body never runs. * Scope-filtered dispatch (`@deepseek-ai/dsh-scope`) keys the carrier by `exec.agent`: a * listener registered through `agent.ctx` fires only for that agent's * calls, while a plain plugin listener fires for every call (including @@ -148,7 +145,7 @@ declare module 'cordis' { * @param result - the dispatch outcome a listener may accept, replace, or block. * @mode waterfall */ - 'tools/post-execute'(this: Scoped, exec: ToolExecution, result: ToolExecutionResult, next: () => Promise): Promise + 'tools/post-execute'(this: Scoped, exec: ToolExecution, result: Readonly, next: () => Promise): Promise /** * Awaited notification of the authoritative FINAL tool outcome, after the * complete pre/execute/post pipeline, final lossless-JSON validation, and @@ -202,6 +199,12 @@ export interface ToolDefinition extends ToolSchema { * cooperative implementation that can reach quiescence when the signal aborts. */ timeoutMs?: number + /** + * Whether this tool name's canonical wire presence or absence survives the + * complete system-prompt assembly waterfall. Reserved for protocol tools + * whose owner must retain the final definition. + */ + readonly ownerFinal?: boolean /** * Optional: how to present the PENDING state of one call in a UI, derived from * the call's `args` (parsed arguments, `unknown` — the tool validates/narrows @@ -239,23 +242,18 @@ export interface ToolResult { declare const toolExecutionTokenBrand: unique symbol -/** Tokens minted in this module; a cast or JavaScript object cannot forge membership. */ -const executionTokens = new WeakSet() - /** - * Opaque, immutable identity for one trip through the tool pipeline. Nested + * Opaque identity for one trip through the tool pipeline. Nested * transports carry the enclosing execution's token instead of its live object, * so observe-only result listeners can correlate calls without gaining a * mutation path into an outer around-dispatch wrapper. */ -export interface ToolExecutionToken { - readonly [toolExecutionTokenBrand]: true -} +export type ToolExecutionToken = symbol & { readonly [toolExecutionTokenBrand]: true } /** * Caller-supplied description of one tool call. {@link ToolRegistry.execute} - * snapshots this input into a pipeline-owned {@link ToolExecution}; callers do - * not choose the execution token. + * adds the registry-owned token to form a pipeline {@link ToolExecution}; + * callers do not choose that token. */ export interface ToolExecutionInput { readonly callId: CallId @@ -274,11 +272,11 @@ export interface ToolExecutionInput { } /** - * One pending tool call inside the registry pipeline. Call identity, the - * registry-assigned {@link token}, and a deep-frozen lossless-JSON snapshot of - * the parsed arguments are immutable from the first policy listener onward, - * while an around-dispatch wrapper may set, replace, or remove only `signal`. - * The registry freezes the complete object before `tools/result` observers run. + * One pending tool call inside the registry pipeline. Parsed arguments cross + * one lossless-JSON materialization boundary before policy and are deep-frozen; + * call identity and the registry-assigned {@link token} are readonly. An + * around-dispatch wrapper may set, replace, or remove `signal`. The registry + * freezes the complete object before `tools/result` observers run. */ export interface ToolExecution extends ToolExecutionInput { /** Registry-assigned identity shared with nested calls only as their opaque `parent` token. */ @@ -430,8 +428,8 @@ export interface Config { * `deny` removes the listed ones; both present = allow first, then deny. * Restrictions never touch scoped registrations — a tool registered through * the same scope is merged after the global filter (which is what keeps e.g. a - * structured-output capture tool alive under an allow-list). The filter values - * are snapshotted at registration, but resolution uses the live global registry: + * structured-output capture tool alive under an allow-list). The readonly + * filter values compile to private sets at registration, but resolution uses the live global registry: * a later global name passes a deny-only filter unless explicitly denied and * fails an allow-list unless explicitly allowed. The * reserved `run_code` presentation transport is likewise outside capability @@ -440,9 +438,27 @@ export interface Config { */ export interface ToolRestriction { /** Global tool names that stay visible; everything else is removed. */ - allow?: string[] + readonly allow?: readonly string[] /** Global tool names removed from visibility. */ - deny?: string[] + readonly deny?: readonly string[] +} + +/** One restriction compiled at registration for repeated live-global lookup. */ +interface CompiledToolRestriction { + readonly allow?: ReadonlySet + readonly deny?: ReadonlySet +} + +/** One scope's complete registry view, derived in a single layer traversal. */ +interface ToolView { + /** Visible definitions after restrictions, scoped shadowing, and transport insertion. */ + readonly visible: ReadonlyMap + /** Pre-restriction capability names used by prompt-order validation. */ + readonly knownNames: ReadonlySet + /** Current global names that a scoped restriction may name. */ + readonly restrictableNames: ReadonlySet + /** Canonical names whose wire presence or absence is owner-final. */ + readonly ownerFinalNames: ReadonlySet } /** @@ -475,7 +491,7 @@ interface ToolGuardRegistration { * that agent alone, disposed with the scope, and SHADOWING a global tool of * the same name for that agent (most-specific-wins; within one layer a * duplicate name still throws). {@link restrict} masks the global layer per - * scope. One visibility function ({@link visible}) feeds prompt assembly, + * scope. One private visibility resolver feeds prompt assembly, * {@link get}, and {@link execute} — and, under a non-native mode, the SDK * section and `run_code`'s bindings — so what the model is shown, what a * presenter renders, what a program can call, and what dispatches can never @@ -490,8 +506,8 @@ export class ToolRegistry extends Service { private global = new Map() private scoped = new Map>() - /** Snapshot-at-registration restriction filters, per scope (see {@link restrict}). */ - private restrictions = new Map() + /** Compiled restriction filters, per scope (see {@link restrict}). */ + private restrictions = new Map() /** Monotonic post-policy guards, split into global and per-agent layers. */ private globalGuards = new Set() private scopedGuards = new Map>() @@ -511,12 +527,13 @@ export class ToolRegistry extends Service { // the filterable global/scoped capability layers. this.codeTransport = this.mode === 'native' ? undefined - : deepFreeze(createRunCodeTool(this, () => this.requireCodeRuntime())) + : createRunCodeTool(this, () => this.requireCodeRuntime()) ctx.systemPrompt.tools(context => this.wireSchemas(context.scope)) if (this.mode !== 'native') { ctx.systemPrompt.section({ name: 'tools:sdk', order: SDK_SECTION_ORDER, + ownerFinal: true, // A lazy thunk over the live registry, per assembly CONTEXT: // regenerated at each assembly over the CALLING SCOPE's visible set // (scoped tools join, restricted globals vanish — the SDK declares @@ -529,11 +546,6 @@ export class ToolRegistry extends Service { return renderToolsSdk(this.schemas(context.scope).filter(schema => schema.name !== RUN_CODE_NAME)) }, }) - // These are presentation infrastructure, not optional end capabilities. - // Protect them at their owner: assembly listeners may still transform - // ordinary tools and prose, but cannot silently leave Code Mode without - // its only wire transport or the SDK that tells the model how to use it. - ctx.systemPrompt.protect({ sections: ['tools:sdk'], tools: [RUN_CODE_NAME] }) } } @@ -553,16 +565,24 @@ export class ToolRegistry extends Service { * `mode: 'code'` the universe is `[run_code]` and a `toolOrder` naming a * native tool is dead configuration that fails every assembly loud. Under * `mode: 'both'`, the provider adds the reserved transport to the - * capability-only {@link knownNames} universe for `toolOrder` validation. + * capability-only known-name universe for `toolOrder` validation. */ private wireSchemas(scope?: ScopeKey): ToolProviderResult { - if (this.mode === 'native') return { schemas: this.schemas(scope), knownNames: this.knownNames(scope) } - this.requireCodeRuntime() - const all = this.schemas(scope) - if (this.mode === 'code') { - return { schemas: all.filter(schema => schema.name === RUN_CODE_NAME), knownNames: [RUN_CODE_NAME] } + const view = this.view(scope) + const schemas = [...view.visible.values()].map(definition => this.schemaOf(definition, false)) + const ownerFinalNames = [...view.ownerFinalNames] + if (this.mode === 'native') { + return { schemas, knownNames: [...view.knownNames], ownerFinalNames } } - return { schemas: all, knownNames: [...this.knownNames(scope), RUN_CODE_NAME] } + this.requireCodeRuntime() + if (this.mode === 'code') { + return { + schemas: schemas.filter(schema => schema.name === RUN_CODE_NAME), + knownNames: [RUN_CODE_NAME], + ownerFinalNames, + } + } + return { schemas, knownNames: [...view.knownNames, RUN_CODE_NAME], ownerFinalNames } } /** @@ -593,13 +613,10 @@ export class ToolRegistry extends Service { * the shadowing feature, not an error; the global-duplicate message names * `agent.ctx` as the per-agent alternative), or if a non-native mode reserves * the `run_code` name for its presentation transport. The visible schema set - * flows into prompt assembly automatically. Registration materializes the JSON - * parameters in one pass, copies scalar fields, binds each callback once to the - * caller's definition as its method receiver, and freezes the stored snapshot; - * later mutation or callback replacement on the input object does not rewrite - * the registry. Every top-level field is read once into one coherent acceptance - * snapshot, so stateful accessors cannot make validation and storage use - * different values. Emits `tools/change` on register/unregister. + * flows into prompt assembly automatically. Definitions are trusted typed + * same-process contributions; JSON materialization happens when the schema or + * result reaches its model/log boundary. Emits `tools/change` on + * register/unregister. * @param definition - the tool's schema plus its execute (and optional * presentation) functions. * @returns the disposer that unregisters the tool. The exact @@ -608,78 +625,39 @@ export class ToolRegistry extends Service { */ register(definition: ToolDefinition): () => Promise | void { const scope = scopeOf(this.ctx) - // One coherent acceptance snapshot: a caller may expose fields through - // accessors, so every top-level value is read exactly once before any - // validation or binding. Checked parameters and stored parameters must be - // the same reference, and a callback cannot change between lookup/bind. const name = definition.name - const description = definition.description - const inputParameters = definition.parameters const timeoutMs = definition.timeoutMs - // eslint-disable-next-line @typescript-eslint/unbound-method - const inputExecute = definition.execute - // eslint-disable-next-line @typescript-eslint/unbound-method - const inputPresentCall = definition.presentCall - // eslint-disable-next-line @typescript-eslint/unbound-method - const inputPresentResult = definition.presentResult - // Reject malformed fixed fields before any caller-owned value can enter the - // frozen snapshot. In particular, a boxed string/object must not become a - // Map key or get recursively frozen as though it were a scalar. - if (typeof name !== 'string') throw new TypeError('tool name must be a string') - if (typeof description !== 'string') throw new TypeError(`tool "${name}" description must be a string`) if (timeoutMs !== undefined - && (typeof timeoutMs !== 'number' || !Number.isFinite(timeoutMs) || timeoutMs <= 0)) { + && (!Number.isFinite(timeoutMs) || timeoutMs <= 0)) { throw new TypeError(`tool "${name}" timeoutMs must be a positive finite number`) } - if (typeof inputExecute !== 'function') throw new TypeError(`tool "${name}" execute must be a function`) - if (inputPresentCall !== undefined && typeof inputPresentCall !== 'function') { - throw new TypeError(`tool "${name}" presentCall must be a function when provided`) - } - if (inputPresentResult !== undefined && typeof inputPresentResult !== 'function') { - throw new TypeError(`tool "${name}" presentResult must be a function when provided`) - } - const execute = inputExecute.bind(definition) - const presentCall = inputPresentCall?.bind(definition) - const presentResult = inputPresentResult?.bind(definition) - // A schema crosses the same model/log boundary as execution arguments. - // Validate and detach it in one traversal: validate-then-structuredClone - // would reread getters and could erase an exotic prototype returned only to - // the clone. A frozen Map is still mutable, so deepFreeze alone is not a - // sufficient registration boundary. - const parameters = snapshotJsonValue(inputParameters) - if (parameters === undefined) { - throw new TypeError('tool parameters must be losslessly JSON-serializable') - } - // Bind once so replacing a callback on the caller-owned definition after - // registration cannot change dispatch, while preserving the historical - // method receiver (`this === definition`) for callbacks that use it. - const snapshot: ToolDefinition = deepFreeze({ - name, - description, - parameters, - execute, - ...timeoutMs !== undefined ? { timeoutMs } : {}, - ...presentCall !== undefined ? { presentCall } : {}, - ...presentResult !== undefined ? { presentResult } : {}, - }) - if (this.codeTransport !== undefined && snapshot.name === RUN_CODE_NAME) { + if (this.codeTransport !== undefined && name === RUN_CODE_NAME) { throw new Error(`tool name "${RUN_CODE_NAME}" is reserved for the Code Mode presentation transport and cannot be registered or shadowed`) } + if (scope !== undefined && this.global.get(name)?.ownerFinal === true) { + throw new Error(`tool "${name}" is globally owner-final and cannot be shadowed in an agent scope`) + } + if (scope === undefined && definition.ownerFinal === true) { + const hasScopedShadow = [...this.scoped.values()].some(layer => layer.has(name)) + if (hasScopedShadow) { + throw new Error(`owner-final tool "${name}" cannot be registered while a scoped shadow exists`) + } + } const dispose = this.ctx.effect(function* (this: ToolRegistry) { const layer = scope === undefined ? this.global : this.layerFor(scope) - if (layer.has(snapshot.name)) { + if (layer.has(name)) { throw new Error(scope === undefined - ? `tool "${snapshot.name}" is already registered (for a per-agent variant, register through that agent's \`agent.ctx\` instead)` - : `tool "${snapshot.name}" is already registered in this scope`) + ? `tool "${name}" is already registered (for a per-agent variant, register through that agent's \`agent.ctx\` instead)` + : `tool "${name}" is already registered in this scope`) } - layer.set(snapshot.name, snapshot) + layer.set(name, definition) // Yield the rollback BEFORE emitting `tools/change`: a generator effect // collects each yielded disposer before the next step runs, so a throwing // `tools/change` listener removes the tool instead of leaking it (a leak // would wedge the duplicate-name check until restart). The duplicate // throw above fires before any mutation — it leaks nothing. yield () => { - layer.delete(snapshot.name) + layer.delete(name) // An emptied scope layer is dropped so a disposed scope leaves no // residue keyed by its (dead) key. if (scope !== undefined && layer.size === 0) this.scoped.delete(scope) @@ -701,16 +679,14 @@ export class ToolRegistry extends Service { * through a scoped context (`agent.ctx`) — restricting "everyone" is not a * thing (throw), and an empty filter (neither `allow` nor `deny`) is a no-op * that can only be a bug (throw — the materialized-empty-config trap). - * Validates every listed name against the scope's CURRENT pre-restriction - * name universe ({@link knownNames}) and throws on an unknown one (fail loud + * Validates every listed name against the CURRENT global end-capability + * universe and throws on an unknown or scope-local name (fail loud * beats a typo silently filtering nothing) — register restrictions after the * global tools they mask exist (the agent-creation `setup` window satisfies * this). A non-native mode's reserved `run_code` presentation transport is * not a filterable capability; naming it explicitly throws, while omitting - * it from an allow-list cannot remove it. `allow` and `deny` are each read - * once, then the filter VALUES are snapshotted at registration: the values - * checked are the values enforced, and later caller mutation of the arrays - * changes nothing. Resolution still uses the live global registry, so a later + * it from an allow-list cannot remove it. The readonly arrays are compiled to + * private sets at registration. Resolution still uses the live global registry, so a later * global name passes a deny-only filter unless named and fails an allow-list * unless named. Multiple restrictions compose by intersection. Scoped * registrations are merged after restrictions and therefore remain visible. @@ -726,36 +702,31 @@ export class ToolRegistry extends Service { if (scope === undefined) { throw new Error('tools.restrict() requires a scoped context (agent.ctx): a context-global restriction would mask every agent — deny the tool for the intended agent instead') } - // Read each caller-owned accessor once. The same values must decide - // whether the filter is meaningful AND become the enforced snapshot: a - // stateful getter must not pass the no-op check as `allow: []` and then - // disappear when the snapshot is built. const allow = filter.allow const deny = filter.deny if (allow === undefined && deny === undefined) { throw new Error('tools.restrict({}) is a no-op: pass `allow` and/or `deny` (an empty filter is almost always a materialized-empty-config bug)') } - // Snapshot BEFORE validation so what was checked is what is enforced. - const snapshot: ToolRestriction = { - ...allow !== undefined ? { allow: [...allow] } : {}, - ...deny !== undefined ? { deny: [...deny] } : {}, + const compiled: CompiledToolRestriction = { + ...allow !== undefined ? { allow: new Set(allow) } : {}, + ...deny !== undefined ? { deny: new Set(deny) } : {}, } if (this.codeTransport !== undefined - && [...snapshot.allow ?? [], ...snapshot.deny ?? []].includes(RUN_CODE_NAME)) { + && [...allow ?? [], ...deny ?? []].includes(RUN_CODE_NAME)) { throw new Error(`tools.restrict() cannot name reserved Code Mode presentation transport "${RUN_CODE_NAME}"; restrict end-capability tools instead`) } - const known = new Set(this.knownNames(scope)) - const unknown = [...snapshot.allow ?? [], ...snapshot.deny ?? []].filter(name => !known.has(name)) + const known = this.view(scope).restrictableNames + const unknown = [...allow ?? [], ...deny ?? []].filter(name => !known.has(name)) if (unknown.length > 0) { - throw new Error(`tools.restrict() names unknown tool${unknown.length > 1 ? 's' : ''} ${unknown.map(n => `"${n}"`).join(', ')}; known tools for this scope: ${[...known].sort().join(', ') || '(none)'}`) + throw new Error(`tools.restrict() names unknown global tool${unknown.length > 1 ? 's' : ''} ${unknown.map(n => `"${n}"`).join(', ')}; known global tools: ${[...known].sort().join(', ') || '(none)'}`) } const dispose = this.ctx.effect(function* (this: ToolRegistry) { const list = this.restrictions.get(scope) ?? [] this.restrictions.set(scope, list) - list.push(snapshot) + list.push(compiled) yield () => { - const index = list.indexOf(snapshot) - /* v8 ignore next 3 -- defensive: the snapshot was pushed, so indexOf is guaranteed >= 0 */ + const index = list.indexOf(compiled) + /* v8 ignore next 3 -- defensive: the compiled restriction was pushed, so indexOf is guaranteed >= 0 */ if (index >= 0) list.splice(index, 1) if (list.length === 0) this.restrictions.delete(scope) this.ctx.emit('tools/change') @@ -817,69 +788,69 @@ export class ToolRegistry extends Service { /** First monotonic denial from the global then matching scoped guard layers. */ private guardReason(exec: ToolExecution): string | undefined { - // Guards are policy, not another transform seam. The pipeline execution's - // identity and arguments are already protected; freeze a detached view so - // an untyped guard cannot replace the wrapper-mutable signal either. - const view: Readonly = Object.freeze({ ...exec }) for (const { guard } of this.globalGuards) { - const reason = guard(view) - if (reason !== undefined) return this.assertGuardReason(reason) + const reason = guard(exec) + if (reason !== undefined) return reason } if (exec.agent !== undefined) { for (const { guard } of this.scopedGuards.get(exec.agent) ?? []) { - const reason = guard(view) - if (reason !== undefined) return this.assertGuardReason(reason) + const reason = guard(exec) + if (reason !== undefined) return reason } } return undefined } - /** Runtime boundary for JavaScript/casted guards: only strings can deny. */ - private assertGuardReason(reason: unknown): string { - if (typeof reason !== 'string') { - throw new TypeError(`tools.guard() must return a denial string or undefined, got ${typeof reason}`) - } - return reason - } - /** Whether every restriction registered for `scope` admits the global tool `name` (intersection semantics). */ private admits(scope: ScopeKey | undefined, name: string): boolean { if (scope === undefined) return true const filters = this.restrictions.get(scope) if (!filters) return true return filters.every(filter => - (filter.allow === undefined || filter.allow.includes(name)) - && (filter.deny === undefined || !filter.deny.includes(name))) + (filter.allow === undefined || filter.allow.has(name)) + && (filter.deny === undefined || !filter.deny.has(name))) } /** - * THE visibility function — one resolution feeding prompt assembly, - * {@link get}, and {@link execute}: the global layer masked by the scope's - * restrictions, unioned with the scope's own layer, scoped shadowing global - * on a name conflict, then the non-native mode's reserved `run_code` - * presentation transport. No scope = the unrestricted global view. + * Resolve every registry fact one scope needs in one layer traversal. The + * visible map applies global restrictions, scoped shadowing, and the reserved + * presentation transport; the other sets retain the pre-restriction facts + * needed by restriction and prompt-order validation and owner-final restore. * @param scope - the viewing scope (the agent), or undefined for the global view. - * @returns the visible definitions (scoped shadows applied), in per-layer - * registration order, global layer first. + * @returns the complete derived view for that scope. */ - visible(scope?: ScopeKey): ToolDefinition[] { + private view(scope?: ScopeKey): ToolView { const layer = scope === undefined ? undefined : this.scoped.get(scope) - const result = new Map() + const visible = new Map() + const knownNames = new Set() + const restrictableNames = new Set() + const ownerFinalNames = new Set() for (const [name, definition] of this.global) { - if (this.admits(scope, name)) result.set(name, definition) + knownNames.add(name) + restrictableNames.add(name) + if (definition.ownerFinal === true) ownerFinalNames.add(name) + if (this.admits(scope, name)) visible.set(name, definition) } // Scoped layer second: same-name entries REPLACE (shadow) the global ones, // and scope-local registrations are never part of the global filter above. - for (const [name, definition] of layer ?? []) result.set(name, definition) + for (const [name, definition] of layer ?? []) { + knownNames.add(name) + if (definition.ownerFinal === true) ownerFinalNames.add(name) + visible.set(name, definition) + } // Presentation infrastructure is resolved last and outside capability // filtering. Registration rejects this reserved name, so this set is an // invariant assertion as well as protection against future layer changes. - if (this.codeTransport !== undefined) result.set(RUN_CODE_NAME, this.codeTransport) - return [...result.values()] + if (this.codeTransport !== undefined) { + visible.set(RUN_CODE_NAME, this.codeTransport) + // createRunCodeTool() owns this internal transport and always marks it owner-final. + ownerFinalNames.add(RUN_CODE_NAME) + } + return { visible, knownNames, restrictableNames, ownerFinalNames } } /** - * Look up a tool as one scope sees it ({@link visible} semantics: scoped + * Look up a tool as one scope sees it (scoped * shadows global; a restricted-away global reads as absent). Presenters pass * the calling agent so the rendered card matches the definition that * actually executed. @@ -888,11 +859,7 @@ export class ToolRegistry extends Service { * @returns the definition the scope resolves, or undefined when none is visible. */ get(name: string, scope?: ScopeKey): ToolDefinition | undefined { - if (name === RUN_CODE_NAME && this.codeTransport !== undefined) return this.codeTransport - const shadowed = scope === undefined ? undefined : this.scoped.get(scope)?.get(name) - if (shadowed) return shadowed - if (!this.admits(scope, name)) return undefined - return this.global.get(name) + return this.view(scope).visible.get(name) } /** @@ -908,30 +875,17 @@ export class ToolRegistry extends Service { * @returns one deep-cloned schema per visible tool. */ schemas(scope?: ScopeKey): ToolSchema[] { - return this.visible(scope).map(({ name, description, parameters }): ToolSchema => ({ - name, - description, - parameters: structuredClone(parameters), - })) + return [...this.view(scope).visible.values()].map(definition => this.schemaOf(definition, true)) } - /** - * The PRE-restriction END-CAPABILITY name universe for `scope`: every global - * name plus the scope's own layer, ignoring restrictions. This is the set - * `restrict()` validates against, so a typo fails loud while a - * restricted-away tool remains a normal, non-erroneous absence. Reserved - * presentation transports are deliberately absent: `restrict()` rejects - * naming one, while {@link wireSchemas} adds it to the separate `toolOrder` - * validation universe when its presentation mode contributes it. - * @param scope - the viewing scope (the agent); omitted = global names only. - * @returns the known names, deduplicated. - */ - knownNames(scope?: ScopeKey): string[] { - const names = new Set(this.global.keys()) - if (scope !== undefined) { - for (const name of this.scoped.get(scope)?.keys() ?? []) names.add(name) + /** Project one definition onto the model-facing schema fields. */ + private schemaOf(definition: ToolDefinition, detachParameters: boolean): ToolSchema { + const { name, description, parameters } = definition + return { + name, + description, + parameters: detachParameters ? structuredClone(parameters) : parameters, } - return [...names] } /** @@ -952,129 +906,54 @@ export class ToolRegistry extends Service { * the final observe-only notification, the authoritative outcome is * materialized as a detached lossless-JSON snapshot; an invalid outcome is * normalized to an error. - * A malformed runtime/casted `tools/pre-execute` decision likewise normalizes - * to an error before approval, guards, or the tool body. - * Caller-owned arguments are validated and detached in one recursive - * lossless-JSON traversal; a violation normalizes to an error before policy - * or dispatch. - * @param exec - the single-use call input; every top-level field is read once. - * `callId` and `name` must each yield a string before that identity snapshot - * is protected and policy begins. - * @returns the final result after every waterfall. Once the required string - * `callId` and `name` correlation identity has been captured, later accessor, - * validation, listener, and tool failures resolve as `isError` results rather - * than rejections. A throwing accessor or non-string value in either identity - * field rejects because no trustworthy result correlation exists yet. + * @param exec - the typed same-process call input. The registry assigns its + * correlation token before policy begins. + * @returns the materialized final result after every waterfall; listener and + * tool failures resolve as `isError` results rather than rejections. */ async execute(exec: ToolExecutionInput): Promise { - // callId/name are the minimum correlation identity needed to construct a - // result at all. Capture each once, then validate the captured scalar before - // anything can treat it as a trustworthy identity. A JavaScript/casted - // caller that supplies another type rejects at this outer boundary: an error - // result carrying the same malformed value would not satisfy the correlation - // contract and might itself fail lossless-JSON materialization. Every other - // caller-controlled accessor is read once INSIDE the normalization boundary; - // if one throws, the error shell uses the fields captured before it and never - // rereads the hostile record. + const token = createExecutionToken() const callId = exec.callId const name = exec.name - if (typeof callId !== 'string') { - throw new TypeError('tool execution callId must be a string') + const agent = exec.agent + const parent = exec.parent + const signal = exec.signal + const base = { + token, + callId, + name, + ...agent !== undefined ? { agent } : {}, + ...parent !== undefined ? { parent } : {}, + ...signal !== undefined ? { signal } : {}, } - if (typeof name !== 'string') { - throw new TypeError('tool execution name must be a string') - } - let agent: Agent | undefined - let parent: ToolExecutionToken | undefined - let signal: AbortSignal | undefined let execution: ToolExecution try { - agent = exec.agent - parent = exec.parent - signal = exec.signal - const args = exec.arguments - const input: Readonly = Object.freeze({ - callId, - name, - arguments: args, - ...agent !== undefined ? { agent } : {}, - ...parent !== undefined ? { parent } : {}, - ...signal !== undefined ? { signal } : {}, - }) - execution = this.prepareExecution(input) + const detached = snapshotJsonValue(exec.arguments) + if (detached === undefined) { + throw new TypeError('tool execution arguments must be losslessly JSON-serializable') + } + execution = { + ...base, + arguments: deepFreeze(detached), + } } catch (error: unknown) { - // Contract-violating arguments outside the lossless-JSON vocabulary cannot - // enter a pipeline whose logged and executed forms must agree. Still - // publish one scoped final outcome, using an immutable identity shell, so - // result observers retain their every-call guarantee without seeing the - // invalid value. - execution = Object.freeze({ - token: createExecutionToken(), - callId, - name, - arguments: undefined, - ...agent !== undefined ? { agent } : {}, - ...isExecutionToken(parent) ? { parent } : {}, - ...signal !== undefined ? { signal } : {}, - }) - const result = toolErrorResult(execution.callId, error) + execution = { ...base, arguments: undefined } + const result = this.materializeFinalResult(toolErrorResult(callId, error)) await this.notifyResult(execution, result) return result } let result: ToolExecutionResult try { - // Materialize the authoritative FINAL result, not merely the tool body's - // intermediate return. Post-policy may replace content or attach context, - // and every one of these fields is session-bound. Reject anything outside - // the lossless-JSON vocabulary before the observe-only `tools/result` - // commit point sees success. - result = this.snapshotExecutionResult(execution, await this.executePipeline(execution)) + result = this.materializeFinalResult(await this.executePipeline(execution)) } catch (error: unknown) { // Outer backstop: a throwing pre/post-execute listener, guard, or the // waterfall machinery becomes an isError result, never a turn failure. - result = toolErrorResult(execution.callId, error) + result = this.materializeFinalResult(toolErrorResult(execution.callId, error)) } await this.notifyResult(execution, result) return result } - /** Snapshot one call into a shared pipeline object with immutable identity and mutable cancellation. */ - private prepareExecution(input: Readonly): ToolExecution { - if (input.parent !== undefined && !isExecutionToken(input.parent)) { - throw new TypeError('tool execution parent must be a registry-minted opaque token') - } - const args = snapshotJsonValue(input.arguments) - if (args === undefined) { - throw new TypeError('tool execution arguments must be losslessly JSON-serializable') - } - const execution: ToolExecution = { - token: createExecutionToken(), - callId: input.callId, - name: input.name, - arguments: deepFreeze(args), - ...input.agent !== undefined ? { agent: input.agent } : {}, - ...input.parent !== undefined ? { parent: input.parent } : {}, - ...input.signal !== undefined ? { signal: input.signal } : {}, - } - Object.defineProperties(execution, { - token: { value: execution.token, enumerable: true, writable: false, configurable: false }, - callId: { value: execution.callId, enumerable: true, writable: false, configurable: false }, - name: { value: execution.name, enumerable: true, writable: false, configurable: false }, - arguments: { value: execution.arguments, enumerable: true, writable: false, configurable: false }, - agent: { value: input.agent, enumerable: true, writable: false, configurable: false }, - parent: { value: input.parent, enumerable: true, writable: false, configurable: false }, - }) - if (input.signal !== undefined) { - Object.defineProperty(execution, 'signal', { - value: input.signal, - enumerable: true, - writable: true, - configurable: true, - }) - } - return execution - } - /** Run the transformable pipeline; {@link execute} owns final normalization and notification. */ private async executePipeline(exec: ToolExecution): Promise { // --- Gate: tools/pre-execute. An `ask` resolves through the optional @@ -1082,10 +961,10 @@ export class ToolRegistry extends Service { // carrier keys dispatch by exec.agent, so an `agent.ctx` listener gates only // its own agent's calls (agent-less calls are subject-less). const carrier = scopeTarget(this, exec.agent) - const gate = this.snapshotPreDecision(await this.ctx.waterfall( + const gate = await this.ctx.waterfall( carrier, 'tools/pre-execute', exec, () => Promise.resolve({ kind: 'allow' }), - )) + ) const decision = gate.kind === 'ask' ? await this.serviceAsk(exec, gate) : gate const denialReason = decision.kind === 'allow' ? this.guardReason(exec) @@ -1109,7 +988,7 @@ export class ToolRegistry extends Service { // before delegating and inspect the normalized result after. Dispatched with the // same carrier as the gate, so an `agent.ctx` wrapper wraps only its own // agent's calls. --- - const result = this.snapshotExecutionResult(exec, await this.ctx.waterfall( + const result = await this.ctx.waterfall( carrier, 'tools/execute', exec, async (): Promise => { try { @@ -1130,64 +1009,25 @@ export class ToolRegistry extends Service { return toolErrorResult(exec.callId, error) } }, - )) + ) + if (result.callId !== exec.callId) { + throw new TypeError(`tools/execute returned callId "${String(result.callId)}" for authoritative call "${exec.callId}"`) + } return await this.postExecute(exec, result) } - /** Validate and detach the extensible gate's decision before any grant can dispatch. */ - private snapshotPreDecision(value: unknown): PreToolDecision { - if (typeof value !== 'object' || value === null || Array.isArray(value)) { - throw new TypeError('tools/pre-execute must return a PreToolDecision object') - } - const decision = value as { kind?: unknown; reason?: unknown } - const keys = Reflect.ownKeys(decision) - const hasExactKeys = (...expected: string[]): boolean => - keys.length === expected.length && expected.every(key => Object.hasOwn(decision, key)) - switch (decision.kind) { - case 'allow': - if (!hasExactKeys('kind')) { - throw new TypeError('tools/pre-execute allow decision must contain only kind') - } - return { kind: 'allow' } - case 'deny': { - const reason = decision.reason - if (!hasExactKeys('kind', 'reason') || typeof reason !== 'string') { - throw new TypeError('tools/pre-execute deny decision must contain only kind and a string reason') - } - return { kind: 'deny', reason } - } - case 'ask': { - const reason = decision.reason - if (!(hasExactKeys('kind') || hasExactKeys('kind', 'reason')) - || (reason !== undefined && typeof reason !== 'string')) { - throw new TypeError('tools/pre-execute ask decision must contain only kind and an optional string reason') - } - return { kind: 'ask', ...reason !== undefined ? { reason } : {} } - } - default: - throw new TypeError('tools/pre-execute must return an allow, deny, or ask decision') - } - } - /** Notify final-result observers without giving them a mutation/error channel into the outcome. */ private async notifyResult(exec: ToolExecution, result: ToolExecutionResult): Promise { // The pipeline is over: freeze the remaining mutable signal slot so every // observer sees the SAME WeakMap-keyable execution without a mutation race. Object.freeze(exec) - // Materialize once more at the observer boundary so every listener receives - // the same detached result even when an internal error path constructed it. - const detached = snapshotJsonValue(result) - if (detached === undefined) { - throw new TypeError('tool result notification must be losslessly JSON-serializable') - } - const snapshot = deepFreeze(detached) const callbacks = this.ctx.events.dispatch('parallel', [ - scopeTarget(this, exec.agent), 'tools/result', exec, snapshot, + scopeTarget(this, exec.agent), 'tools/result', exec, result, ]) await Promise.all(callbacks.map(async (callback) => { try { - await callback(exec, snapshot) + await callback(exec, result) } catch (error: unknown) { this.ctx.logger.warn(`tool "${exec.name}" (${exec.callId}): tools/result observer failed: ${errorMessage(error)}`) } @@ -1241,112 +1081,40 @@ export class ToolRegistry extends Service { * Runs inside `execute`'s outer try/catch (a throwing listener → isError). */ private async postExecute(exec: ToolExecution, result: ToolExecutionResult): Promise { - // Snapshot the protected outcome BEFORE the waterfall. A listener receives - // the same `result` reference, so a post-waterfall read of `result.callId`/ - // `.isError`/`.error` could carry a listener's mutation — violating the - // authoritative-call-id requirement and the "preserve the dispatched - // isError/error" contract. The decision is the ONLY sanctioned channel for a - // listener to change the outcome (block, or accept-with-replacement); the - // call id is always the authoritative `exec.callId`. The one-pass snapshot - // protects nested content, error, and meta from in-place listener mutation. - const dispatched = this.snapshotExecutionResult(exec, result) - const decision = snapshotJsonValue(await this.ctx.waterfall( + const decision = await this.ctx.waterfall( scopeTarget(this, exec.agent), 'tools/post-execute', exec, result, () => Promise.resolve({ kind: 'accept' }), - )) - if (decision === undefined) { - throw new TypeError('tools/post-execute must return a losslessly JSON-serializable decision') - } - this.assertPostDecision(decision) + ) const additionalContext = decision.additionalContext if (decision.kind === 'block') { return { - callId: dispatched.callId, + callId: result.callId, content: decision.feedback, isError: true, ...additionalContext ? { additionalContext } : {}, } } - // accept: replace content if supplied, preserve the dispatched isError/error. + // Accept: replace content if supplied and preserve the dispatched outcome. return { - ...dispatched, + ...result, ...decision.content ? { content: decision.content } : {}, ...additionalContext ? { additionalContext } : {}, } } - /** Validate and detach an around-dispatch result before policy can observe or mutate it. */ - private snapshotExecutionResult(exec: ToolExecution, value: unknown): ToolExecutionResult { - if (typeof value !== 'object' || value === null) { - throw new TypeError('tools/execute must return a ToolExecutionResult object') - } - const result = value as Partial - // Capture the provider/listener-owned result exactly once. The same values - // must pass shape/correlation checks and become the detached final outcome; - // a stateful accessor cannot validate one result and publish another. - const callId = result.callId - const content = result.content - const isError = result.isError - const error = result.error - const additionalContext = result.additionalContext - const meta = result.meta - if (!Array.isArray(content) || typeof isError !== 'boolean') { - throw new TypeError('tools/execute must return a ToolExecutionResult with content[] and boolean isError') - } - if (callId !== exec.callId) { - throw new TypeError(`tools/execute returned callId "${String(callId)}" for authoritative call "${exec.callId}"`) - } - const candidate = { - callId: exec.callId, - content, - isError, - ...error !== undefined ? { error } : {}, - ...additionalContext !== undefined ? { additionalContext } : {}, - ...meta !== undefined ? { meta } : {}, - } - // One traversal both validates and detaches the accepted result. A separate - // check followed by structuredClone would reread getters and could sanitize - // a class instance into an apparently valid plain record. - const snapshot = snapshotJsonValue(candidate) - if (snapshot === undefined) { - throw new TypeError('tools/execute must return a losslessly JSON-serializable ToolExecutionResult') - } - return snapshot - } - - /** Reject malformed JavaScript/casted post decisions at the public event boundary. */ - private assertPostDecision(value: unknown): asserts value is PostToolDecision { - if (typeof value !== 'object' || value === null) { - throw new TypeError('tools/post-execute must return a PostToolDecision object') - } - const decision = value as Partial - switch (decision.kind) { - case 'accept': - if (decision.content !== undefined && !Array.isArray(decision.content)) { - throw new TypeError('tools/post-execute accept content must be an array') - } - return - case 'block': - if (!Array.isArray(decision.feedback)) { - throw new TypeError('tools/post-execute block feedback must be an array') - } - return - default: - throw new TypeError('tools/post-execute must return an accept or block decision') + /** Materialize the authoritative commit outcome once, immediately before `tools/result`. */ + private materializeFinalResult(result: ToolExecutionResult): ToolExecutionResult { + const detached = snapshotJsonValue(result) + if (detached === undefined) { + throw new TypeError('tool result must be losslessly JSON-serializable') } + return deepFreeze(detached) } } -/** Mint a frozen, property-free correlation token whose identity is its value. */ +/** Mint a same-process correlation token whose identity is its value. */ function createExecutionToken(): ToolExecutionToken { - const token = Object.freeze(Object.create(null)) as ToolExecutionToken - executionTokens.add(token) - return token -} - -/** Runtime counterpart of the opaque token type, including `undefined` input. */ -function isExecutionToken(value: unknown): value is ToolExecutionToken { - return typeof value === 'object' && value !== null && executionTokens.has(value) + return Symbol('dsh.tool.execution') as ToolExecutionToken } function toolErrorResult(callId: ToolExecution['callId'], error: unknown): ToolExecutionResult { diff --git a/packages/core/tools/src/schema.ts b/packages/core/tools/src/schema.ts index add9c29e61..dc33a97299 100644 --- a/packages/core/tools/src/schema.ts +++ b/packages/core/tools/src/schema.ts @@ -20,7 +20,6 @@ */ import { assertNever, HarnessError } from '@deepseek-ai/dsh-llm' -import { snapshotJsonValue } from '@deepseek-ai/dsh-session' import type { ToolDefinition, ToolExecuteReturn, ToolExecution, ToolResult } from './index.ts' import type { ToolCallView, ToolResultView } from './presentation.ts' @@ -288,21 +287,23 @@ export function validateArgs(spec: SchemaSpec, args: unknown): string[] { /** Options for {@link defineTool}. */ export interface DefineToolOptions { /** Tool name (must be unique). */ - name: string + readonly name: string /** Human-readable description sent to the model. */ - description: string + readonly description: string /** * Parameter schema using the per-property-required DSL. Converted to * standard JSON Schema at runtime. */ - parameters: S + readonly parameters: S /** * Optional cooperative tool-call timeout budget in milliseconds. When given it * must be a positive finite number; it is attached to the produced * {@link ToolDefinition} for `@deepseek-ai/dsh-timeout-policy` to enforce and * is never sent to the model. */ - timeoutMs?: number + readonly timeoutMs?: number + /** Make this protocol tool's canonical wire presence or absence owner-final. */ + readonly ownerFinal?: boolean /** * Tool execution function. `args` is typed as {@link InferArgs} — zero * casts needed. Returns either a bare {@link ContentBlock}`[]` (model-facing @@ -355,11 +356,6 @@ export interface DefineToolOptions { * by `ToolRegistry.register()` directly — `defineTool` is sugar for * first-party plugin authors. * - * Definition is an acceptance boundary: every top-level option is read once, - * and the parameter spec is detached before either the wire schema or the - * runtime validators are built. Later mutation of the caller's options or - * schema therefore cannot make the model-visible schema disagree with execute - * or presentation validation. * @param options - the tool's name, description, typed parameter schema, * execute body, and optional presenters. * @returns a registry-ready {@link ToolDefinition}: its `execute` validates the @@ -369,13 +365,6 @@ export interface DefineToolOptions { * args). */ export function defineTool(options: DefineToolOptions): ToolDefinition { - // Capture every caller-owned top-level field before inspecting any nested - // schema value. Accessors may be stateful, so validation, presentation, and - // the returned definition must all derive from this one accepted record. - const name = options.name - const description = options.description - const inputParameters = options.parameters - const timeoutMs = options.timeoutMs // Object-literal execute methods don't use `this`; the reference is safe. // eslint-disable-next-line @typescript-eslint/unbound-method const userExecute = options.execute @@ -383,31 +372,21 @@ export function defineTool(options: DefineToolOptions): const userPresentCall = options.presentCall // eslint-disable-next-line @typescript-eslint/unbound-method const userPresentResult = options.presentResult - if (timeoutMs !== undefined && (!Number.isFinite(timeoutMs) || timeoutMs <= 0)) { - throw new Error(`defineTool(${name}): timeoutMs must be a positive finite number`) - } - // The internal SchemaSpec and public wire schema must not share mutable - // subobjects. Each is materialized through the lossless one-pass boundary; - // structuredClone alone could sanitize an exotic default or nested getter. - const parameterSpec = snapshotJsonValue(inputParameters) - if (parameterSpec === undefined) { - throw new Error(`defineTool(${name}): parameters must be losslessly JSON-serializable`) - } - const wireParameters = snapshotJsonValue(schemaSpecToJsonSchema(parameterSpec)) - if (wireParameters === undefined) { - throw new Error(`defineTool(${name}): generated parameters must be losslessly JSON-serializable`) + if (options.timeoutMs !== undefined && (!Number.isFinite(options.timeoutMs) || options.timeoutMs <= 0)) { + throw new Error(`defineTool(${options.name}): timeoutMs must be a positive finite number`) } const tool: ToolDefinition = { - name, - description, - parameters: wireParameters as unknown as Record, - ...(timeoutMs !== undefined ? { timeoutMs } : {}), + name: options.name, + description: options.description, + parameters: schemaSpecToJsonSchema(options.parameters) as unknown as Record, + ...(options.timeoutMs !== undefined ? { timeoutMs: options.timeoutMs } : {}), + ...(options.ownerFinal === true ? { ownerFinal: true } : {}), async execute(args: unknown, exec: ToolExecution): Promise { // Validate the model-generated args before the typed body runs. On // mismatch we throw ToolArgsError; the registry turns it into an // isError result so the model can self-correct. After this guard, the // cast to InferArgs reflects the validated shape. - const violations = validateArgs(parameterSpec, args) + const violations = validateArgs(options.parameters, args) if (violations.length > 0) throw new ToolArgsError(violations) return userExecute(args as InferArgs, exec) }, @@ -418,13 +397,13 @@ export function defineTool(options: DefineToolOptions): // than the hard `ToolArgsError` the execute path raises. if (userPresentCall) { tool.presentCall = (args: unknown): ToolCallView | undefined => { - if (validateArgs(parameterSpec, args).length > 0) return undefined + if (validateArgs(options.parameters, args).length > 0) return undefined return userPresentCall(args as InferArgs) } } if (userPresentResult) { tool.presentResult = (args: unknown, result: ToolResult): ToolResultView | undefined => { - if (validateArgs(parameterSpec, args).length > 0) return undefined + if (validateArgs(options.parameters, args).length > 0) return undefined return userPresentResult(args as InferArgs, result) } } diff --git a/packages/core/tools/tests/code-mode.spec.ts b/packages/core/tools/tests/code-mode.spec.ts index da5725f10e..83dedbdb00 100644 --- a/packages/core/tools/tests/code-mode.spec.ts +++ b/packages/core/tools/tests/code-mode.spec.ts @@ -215,41 +215,24 @@ describe('mode-aware wire contribution', () => { expect(() => scope.ctx.tools.register(impostor)).toThrow(/reserved for the Code Mode presentation transport/) expect(() => ctx.tools.register(impostor)).toThrow(/reserved for the Code Mode presentation transport/) expect(() => scope.ctx.systemPrompt.section({ name: 'tools:sdk', order: -999, text: 'malicious SDK' })) - .toThrow(/globally protected and cannot be shadowed/) + .toThrow(/globally owner-final and cannot be shadowed/) expect(() => scope.ctx.tools.restrict({ allow: [RUN_CODE_NAME] })).toThrow(/cannot name reserved Code Mode presentation transport/) expect(() => scope.ctx.tools.restrict({ deny: [RUN_CODE_NAME] })).toThrow(/cannot name reserved Code Mode presentation transport/) - const transport = ctx.tools.get(RUN_CODE_NAME)! - expect(Object.isFrozen(transport)).toBe(true) - expect(Object.isFrozen(transport.parameters)).toBe(true) - expect(() => { transport.name = 'mutated_transport' }).toThrow(TypeError) - - const mutableSection = { name: 'scoped-note', order: 149, text: 'safe note' } - scope.ctx.systemPrompt.section(mutableSection) - mutableSection.name = 'tools:sdk' - mutableSection.text = 'mutated SDK' - const mutableTool = defineTool({ + scope.ctx.systemPrompt.section({ name: 'scoped-note', order: 149, text: 'safe note' }) + scope.ctx.tools.register(defineTool({ name: 'scoped_safe', description: 'Safe scoped tool.', parameters: {}, execute: () => Promise.resolve([{ type: 'text' as const, text: 'safe' }]), - }) - scope.ctx.tools.register(mutableTool) - mutableTool.name = RUN_CODE_NAME - mutableTool.description = 'Mutated transport impostor.' - const stored = ctx.tools.get('scoped_safe', agent)! - expect(Object.isFrozen(stored)).toBe(true) - expect(Object.isFrozen(stored.parameters)).toBe(true) - expect(() => { stored.name = RUN_CODE_NAME }).toThrow(TypeError) + })) const assembly = await systemPrompt.assemble({ scope: agent }) const transports = assembly.tools.filter(tool => tool.name === RUN_CODE_NAME) expect(transports).toHaveLength(1) expect(transports[0]?.description).toContain('Execute a TypeScript program') - expect(assembly.sections.find(section => section.name === 'tools:sdk')?.text).not.toContain('mutated SDK') expect(assembly.sections.find(section => section.name === 'scoped-note')?.text).toBe('safe note') expect(assembly.sections.find(section => section.name === 'tools:sdk')?.text).toContain('scoped_safe(args:') expect(ctx.tools.get(RUN_CODE_NAME, agent)).toBe(ctx.tools.get(RUN_CODE_NAME)) - expect(ctx.tools.knownNames(agent)).not.toContain(RUN_CODE_NAME) const result = await runCode(ctx, 'return 1', { agent }) expect(result.content).toEqual([{ type: 'text', text: '(run_code completed with no output)' }]) }) @@ -262,7 +245,6 @@ describe('mode-aware wire contribution', () => { registerEcho(ctx) const { agent } = await mintAgentScope(ctx) - expect(ctx.tools.knownNames(agent)).toEqual(['echo']) const assembly = await systemPrompt.assemble({ scope: agent }) expect(assembly.tools.map(tool => tool.name)).toEqual(mode === 'code' ? [RUN_CODE_NAME] diff --git a/packages/core/tools/tests/scoped.spec.ts b/packages/core/tools/tests/scoped.spec.ts index 40671ae6fc..991d420054 100644 --- a/packages/core/tools/tests/scoped.spec.ts +++ b/packages/core/tools/tests/scoped.spec.ts @@ -4,7 +4,7 @@ import { createScope } from '@deepseek-ai/dsh-scope' import type { Scope } from '@deepseek-ai/dsh-scope' import SystemPrompt from '@deepseek-ai/dsh-system-prompt' import ToolRegistry from '@deepseek-ai/dsh-tools' -import type { PreToolDecision, ToolDefinition, ToolExecution, ToolExecutionInput, ToolExecutionToken, ToolRestriction } from '@deepseek-ai/dsh-tools' +import type { PreToolDecision, ToolDefinition, ToolExecution, ToolExecutionInput, ToolExecutionToken } from '@deepseek-ai/dsh-tools' import type { Agent, AgentId } from '@deepseek-ai/dsh-agent' import { CallId } from '@deepseek-ai/dsh-llm' import type { ContentBlock } from '@deepseek-ai/dsh-llm' @@ -89,6 +89,35 @@ describe('scoped tool registration', () => { expect(() => scope.ctx.tools.register(tool('y'))).toThrow(/already registered in this scope/) }) + it('rejects either registration order between a global owner-final tool and a scoped shadow', async () => { + const first = await mount() + const { scope: firstScope } = await mintAgentScope(first, 'first') + first.tools.register({ ...tool('reserved'), ownerFinal: true }) + expect(() => firstScope.ctx.tools.register(tool('reserved'))) + .toThrow(/globally owner-final and cannot be shadowed/) + + const second = await mount() + const { scope: secondScope } = await mintAgentScope(second, 'second') + secondScope.ctx.tools.register(tool('reserved')) + expect(() => second.tools.register({ ...tool('reserved'), ownerFinal: true })) + .toThrow(/owner-final tool "reserved" cannot be registered while a scoped shadow exists/) + }) + + it('restores global and scoped owner-final tools removed by assembly middleware', async () => { + const ctx = await mount() + const { scope, key } = await mintAgentScope(ctx, 'owner-final') + ctx.tools.register({ ...tool('required'), ownerFinal: true }) + scope.ctx.tools.register({ ...tool('scoped-required'), ownerFinal: true }) + ctx.on('system-prompt/assemble', async assembly => ({ + ...assembly, + tools: assembly.tools.filter(schema => !schema.name.includes('required')), + })) + + expect((await ctx.systemPrompt.assemble()).tools.map(schema => schema.name)).toContain('required') + expect((await ctx.systemPrompt.assemble({ scope: key })).tools.map(schema => schema.name)) + .toEqual(expect.arrayContaining(['required', 'scoped-required'])) + }) + it('disposing the scope unwinds its registrations and leaves no residue', async () => { const ctx = await mount() const { scope, key } = await mintAgentScope(ctx, 'a') @@ -96,7 +125,7 @@ describe('scoped tool registration', () => { expect(ctx.tools.get('mine', key)).toBeDefined() await scope.dispose() expect(ctx.tools.get('mine', key)).toBeUndefined() - expect(ctx.tools.knownNames(key)).toEqual([]) + expect(ctx.tools.schemas(key)).toEqual([]) }) }) @@ -153,7 +182,7 @@ describe('restrict()', () => { expect(ctx.tools.schemas(key).map(t => t.name).sort()).toEqual(['a', 'c']) }) - it('snapshots the filter at registration (caller mutation changes nothing)', async () => { + it('compiles the readonly filter values at registration', async () => { const ctx = await mount() const { scope, key } = await mintAgentScope(ctx, 'a') ctx.tools.register(tool('a')) @@ -164,38 +193,21 @@ describe('restrict()', () => { expect(ctx.tools.schemas(key).map(t => t.name)).toEqual(['b']) }) - it('reads restriction accessors once so the checked filter is the enforced filter', async () => { - const ctx = await mount() - const { scope, key } = await mintAgentScope(ctx, 'a') - ctx.tools.register(tool('global')) - let allowReads = 0 - const filter = { - get allow(): string[] | undefined { - allowReads += 1 - return allowReads === 1 ? [] : undefined - }, - } as ToolRestriction - - scope.ctx.tools.restrict(filter) - - expect(allowReads).toBe(1) - expect(ctx.tools.schemas(key)).toEqual([]) - expect(await run(ctx, 'global', key)).toBe('Error: unknown tool "global"') - }) - - it('fails loud on an unscoped call, an empty filter, and unknown names', async () => { + it('fails loud on an unscoped call, an empty filter, and non-global names', async () => { const ctx = await mount() const { scope } = await mintAgentScope(ctx, 'a') ctx.tools.register(tool('real')) + scope.ctx.tools.register(tool('local')) expect(() => ctx.tools.restrict({ deny: ['real'] })).toThrow(/requires a scoped context/) expect(() => scope.ctx.tools.restrict({})).toThrow(/no-op/) - expect(() => scope.ctx.tools.restrict({ allow: ['reall'] })).toThrow(/unknown tool "reall"; known tools for this scope: real/) - expect(() => scope.ctx.tools.restrict({ deny: ['ghost', 'wraith'] })).toThrow(/unknown tools "ghost", "wraith"/) + expect(() => scope.ctx.tools.restrict({ allow: ['local'] })).toThrow(/unknown global tool "local"/) + expect(() => scope.ctx.tools.restrict({ allow: ['reall'] })).toThrow(/unknown global tool "reall"; known global tools: real/) + expect(() => scope.ctx.tools.restrict({ deny: ['ghost', 'wraith'] })).toThrow(/unknown global tools "ghost", "wraith"/) const emptyCtx = await mount() const { scope: emptyScope } = await mintAgentScope(emptyCtx, 'empty') expect(() => emptyScope.ctx.tools.restrict({ deny: ['ghost'] })) - .toThrow(/known tools for this scope: \(none\)/) + .toThrow(/known global tools: \(none\)/) }) }) @@ -230,9 +242,8 @@ describe('scoped execution dispatch', () => { return Promise.resolve([{ type: 'text', text: 'ran:t' }]) }, }) - let guardViewFrozen = false const guard = (execution: Readonly): string => { - guardViewFrozen = Object.isFrozen(execution) && Object.isFrozen(execution.arguments) + expect(Object.isFrozen(execution.arguments)).toBe(true) return 'terminal policy' } const liftFirst = scope.ctx.tools.guard(guard) @@ -243,7 +254,6 @@ describe('scoped execution dispatch', () => { scope.ctx.on('tools/pre-execute', () => Promise.resolve({ kind: 'allow' }), { prepend: true }) expect(await run(ctx, 't', key)).toBe('Error: terminal policy') - expect(guardViewFrozen).toBe(true) expect(await run(ctx, 't', other)).toBe('ran:t') expect(bodyCalls).toBe(1) @@ -271,13 +281,14 @@ describe('scoped execution dispatch', () => { expect(bodyCalls).toBe(0) }) - it('protects call identity before policy and dispatch while leaving only signal mutable', async () => { + it('shares one token and materialized argument value across the pipeline', async () => { const ctx = await mount() const { scope, key } = await mintAgentScope(ctx, 'a') let safeCalls = 0 let dangerCalls = 0 let scopedResults = 0 let safeArguments: unknown + const tokens = new Set() ctx.tools.register({ ...tool('safe'), execute: (args) => { @@ -295,17 +306,16 @@ describe('scoped execution dispatch', () => { }) scope.ctx.tools.guard(exec => exec.name === 'danger' ? 'danger denied' : undefined) ctx.on('tools/pre-execute', (exec, next) => { - expect(Reflect.set(exec, 'agent', undefined)).toBe(false) - expect(Reflect.set(exec, 'name', 'safe')).toBe(false) - expect(Reflect.set(exec.arguments as object, 'injected', true)).toBe(false) + tokens.add(exec.token) + expect(Object.isFrozen(exec.arguments)).toBe(true) return next() }) ctx.on('tools/execute', (exec, next) => { - expect(Reflect.set(exec, 'name', 'danger')).toBe(false) + tokens.add(exec.token) return next() }) ctx.on('tools/post-execute', (exec, _result, next) => { - expect(Reflect.set(exec, 'agent', undefined)).toBe(false) + tokens.add(exec.token) return next() }) scope.ctx.on('tools/result', () => { scopedResults += 1 }) @@ -323,6 +333,8 @@ describe('scoped execution dispatch', () => { expect(safeArguments).not.toBe(callerArguments) expect(Object.isFrozen(safeArguments)).toBe(true) expect(callerArguments).toEqual({ source: true }) + // One token for danger and one shared by every phase of safe. + expect(tokens.size).toBe(2) expect({ safeCalls, dangerCalls, scopedResults }).toEqual({ safeCalls: 1, dangerCalls: 0, @@ -395,25 +407,6 @@ describe('scoped execution dispatch', () => { expect(callerArguments.invalid).toBeTypeOf('function') }) - it('rejects a forged mutable parent token without exposing it to final observers', async () => { - const ctx = await mount() - ctx.tools.register(tool('t')) - const forged = { mutable: true } as unknown as ToolExecutionToken - let observedParent: ToolExecutionToken | undefined = forged - ctx.on('tools/result', (exec) => { observedParent = exec.parent }) - - const result = await ctx.tools.execute({ - callId: CallId('forged-parent'), name: 't', arguments: {}, parent: forged, - }) - - expect(result.isError).toBe(true) - expect(result.content).toEqual([{ - type: 'text', text: 'Error: tool execution parent must be a registry-minted opaque token', - }]) - expect(observedParent).toBeUndefined() - expect(Object.isFrozen(forged)).toBe(false) - }) - it('reads a stateful parent accessor once before policy, dispatch, and result observation', async () => { const ctx = await mount() const observed: (ToolExecutionToken | undefined)[] = [] diff --git a/packages/core/tools/tests/tools.spec.ts b/packages/core/tools/tests/tools.spec.ts index 1b83a8f633..4c52ba7910 100644 --- a/packages/core/tools/tests/tools.spec.ts +++ b/packages/core/tools/tests/tools.spec.ts @@ -6,8 +6,8 @@ import type { Agent } from '@deepseek-ai/dsh-agent' import ApprovalService, { type ApprovalOutcome, type ApprovalRequest } from '@deepseek-ai/dsh-user-approval' import ToolRegistry, { defineTool, schemaSpecToJsonSchema, validateArgs, ToolArgsError, ToolNotFoundError, - type DefineToolOptions, type InferArgs, type SchemaSpec, type PreToolDecision, type PostToolDecision, - type ToolDefinition, type ToolExecution, type ToolExecutionInput, type ToolExecutionResult, type ToolGuard, + type InferArgs, type SchemaSpec, type PreToolDecision, type PostToolDecision, + type ToolExecution, type ToolExecutionResult, } from '@deepseek-ai/dsh-tools' async function setup() { @@ -83,95 +83,6 @@ describe('ToolRegistry', () => { expect(result).toEqual({ callId: CallId('c1'), content: [{ type: 'text', text: 'hi' }], isError: false }) }) - it.each([ - { field: 'callId', value: 1n }, - { field: 'callId', value: 123 }, - { field: 'name', value: 1n }, - { field: 'name', value: 123 }, - ] as const)('rejects a non-string $field before final observation', async ({ field, value }) => { - const ctx = await setup() - let observed = 0 - ctx.on('tools/result', () => { observed += 1 }) - const input: Record = { - callId: CallId('valid-call'), - name: 'missing', - arguments: {}, - } - input[field] = value - - await expect(ctx.tools.execute(input as unknown as ToolExecutionInput)) - .rejects.toThrow(`tool execution ${field} must be a string`) - expect(observed).toBe(0) - }) - - it('reads correlation accessors once and normalizes a later hostile accessor', async () => { - const ctx = await setup() - const reads = { callId: 0, name: 0, arguments: 0 } - let observed: { callId: unknown; name: unknown; isError: boolean } | undefined - ctx.on('tools/result', (exec, result) => { - observed = { callId: exec.callId, name: exec.name, isError: result.isError } - }) - const input = Object.defineProperties({}, { - callId: { - enumerable: true, - get: () => { - reads.callId += 1 - if (reads.callId > 1) throw new Error('callId reread') - return CallId('one-read-call') - }, - }, - name: { - enumerable: true, - get: () => { - reads.name += 1 - if (reads.name > 1) throw new Error('name reread') - return 'missing' - }, - }, - arguments: { - enumerable: true, - get: () => { - reads.arguments += 1 - throw new Error('arguments accessor broke') - }, - }, - }) as unknown as ToolExecutionInput - - const result = await ctx.tools.execute(input) - - expect(reads).toEqual({ callId: 1, name: 1, arguments: 1 }) - expect(result).toMatchObject({ callId: CallId('one-read-call'), isError: true }) - expect(result.content[0]).toMatchObject({ text: 'Error: arguments accessor broke' }) - expect(observed).toEqual({ callId: CallId('one-read-call'), name: 'missing', isError: true }) - }) - - it('reads callId once before a hostile name accessor rejects correlation', async () => { - const ctx = await setup() - const reads = { callId: 0, name: 0 } - let observed = 0 - ctx.on('tools/result', () => { observed += 1 }) - const input = Object.defineProperties({ arguments: {} }, { - callId: { - enumerable: true, - get: () => { - reads.callId += 1 - return CallId('hostile-name') - }, - }, - name: { - enumerable: true, - get: () => { - reads.name += 1 - throw new Error('name accessor broke') - }, - }, - }) as unknown as ToolExecutionInput - - await expect(ctx.tools.execute(input)).rejects.toThrow('name accessor broke') - expect(reads).toEqual({ callId: 1, name: 1 }) - expect(observed).toBe(0) - }) - it('threads a tool-attached meta (object return form) onto the result', async () => { const ctx = await setup() ctx.tools.register({ @@ -224,77 +135,6 @@ describe('ToolRegistry', () => { expect(observedError).toBe(true) }) - it('reads each result value once so later getter drift cannot change the snapshot', async () => { - const ctx = await setup() - ctx.tools.register(echoTool) - let reads = 0 - const hostileBlock = Object.defineProperty({ type: 'text' }, 'text', { - enumerable: true, - get: () => ++reads === 1 ? 'safe' : new Map([['mutable', true]]), - }) - ctx.on('tools/execute', async exec => ({ - callId: exec.callId, - content: [hostileBlock], - isError: false, - }) as unknown as ToolExecutionResult) - - const result = await ctx.tools.execute({ - callId: CallId('unstable-result'), name: 'echo', arguments: {}, - }) - - expect(reads).toBe(1) - expect(result).toEqual({ - callId: CallId('unstable-result'), - content: [{ type: 'text', text: 'safe' }], - isError: false, - }) - }) - - it('reads every top-level execution result field once before validation', async () => { - const ctx = await setup() - ctx.tools.register(echoTool) - const reads = { callId: 0, content: 0, isError: 0, error: 0, additionalContext: 0, meta: 0 } - ctx.on('tools/execute', async exec => Object.defineProperties({}, { - callId: { enumerable: true, get: () => { reads.callId += 1; return reads.callId === 1 ? exec.callId : CallId('drifted') } }, - content: { enumerable: true, get: () => { reads.content += 1; return reads.content === 1 ? [{ type: 'text', text: 'accepted' }] : [] } }, - isError: { enumerable: true, get: () => { reads.isError += 1; return reads.isError !== 1 } }, - error: { enumerable: true, get: () => { reads.error += 1; return undefined } }, - additionalContext: { enumerable: true, get: () => { reads.additionalContext += 1; return undefined } }, - meta: { enumerable: true, get: () => { reads.meta += 1; return undefined } }, - }) as ToolExecutionResult) - - const result = await ctx.tools.execute({ - callId: CallId('one-read-result'), name: 'echo', arguments: {}, - }) - - expect(reads).toEqual({ callId: 1, content: 1, isError: 1, error: 1, additionalContext: 1, meta: 1 }) - expect(result).toEqual({ - callId: CallId('one-read-result'), - content: [{ type: 'text', text: 'accepted' }], - isError: false, - }) - }) - - it('rejects an exotic nested result before its prototype can be sanitized', async () => { - const ctx = await setup() - ctx.tools.register(echoTool) - class ExoticText { readonly value = 'not text' } - ctx.on('tools/execute', exec => Promise.resolve({ - callId: exec.callId, - content: [{ type: 'text', text: new ExoticText() }], - isError: false, - } as unknown as ToolExecutionResult)) - - const result = await ctx.tools.execute({ - callId: CallId('exotic-result'), name: 'echo', arguments: {}, - }) - - expect(result.isError).toBe(true) - expect(result.content).toEqual([{ - type: 'text', text: 'Error: tools/execute must return a losslessly JSON-serializable ToolExecutionResult', - }]) - }) - it('returns isError results for unknown tools and throwing tools', async () => { const ctx = await setup() ctx.tools.register({ @@ -362,85 +202,6 @@ describe('ToolRegistry', () => { expect(result.content[0]).toMatchObject({ text: 'Error: denied by policy' }) }) - it.each([ - { - name: 'non-object decision', - replacement: null, - message: 'tools/pre-execute must return a PreToolDecision object', - }, - { - name: 'unknown decision kind', - replacement: { kind: 'permit' }, - message: 'tools/pre-execute must return an allow, deny, or ask decision', - }, - { - name: 'allow decision carrying extra fields', - replacement: { kind: 'allow', reason: 'smuggled' }, - message: 'tools/pre-execute allow decision must contain only kind', - }, - { - name: 'deny decision without a reason', - replacement: { kind: 'deny' }, - message: 'tools/pre-execute deny decision must contain only kind and a string reason', - }, - { - name: 'deny decision with a non-string reason', - replacement: { kind: 'deny', reason: 42 }, - message: 'tools/pre-execute deny decision must contain only kind and a string reason', - }, - { - name: 'ask decision with a non-string reason', - replacement: { kind: 'ask', reason: true }, - message: 'tools/pre-execute ask decision must contain only kind and an optional string reason', - }, - { - name: 'ask decision carrying extra fields', - replacement: { kind: 'ask', cache: true }, - message: 'tools/pre-execute ask decision must contain only kind and an optional string reason', - }, - ])('fails closed on a malformed tools/pre-execute $name', async ({ replacement, message }) => { - const ctx = await setup() - let bodyCalls = 0 - const observed: ToolExecutionResult[] = [] - ctx.tools.register({ - ...echoTool, - async execute() { - bodyCalls += 1 - return [] - }, - }) - ctx.on('tools/pre-execute', async () => replacement as unknown as PreToolDecision) - ctx.on('tools/result', (_exec, result) => { observed.push(result) }) - - const result = await ctx.tools.execute({ - callId: CallId('malformed-pre'), name: 'echo', arguments: {}, - }) - - expect(result.isError).toBe(true) - expect(result.content[0]).toMatchObject({ text: `Error: ${message}` }) - expect(bodyCalls).toBe(0) - expect(observed).toEqual([result]) - }) - - it('rejects a JavaScript guard that returns an async/non-string decision', async () => { - const ctx = await setup() - let bodyCalls = 0 - ctx.tools.register({ - ...echoTool, - async execute() { - bodyCalls += 1 - return [] - }, - }) - ctx.tools.guard((() => Promise.resolve('late denial')) as unknown as ToolGuard) - - const result = await ctx.tools.execute({ callId: CallId('bad-guard'), name: 'echo', arguments: {} }) - expect(result.isError).toBe(true) - expect(result.content[0]?.type === 'text' && result.content[0].text) - .toContain('tools.guard() must return') - expect(bodyCalls).toBe(0) - }) - it('an ask decision degrades to deny when no approval seam is mounted', async () => { const ctx = await setup() ctx.tools.register(echoTool) @@ -617,55 +378,6 @@ describe('ToolRegistry', () => { expect(result.additionalContext).toMatchObject({ content: [{ text: 'fyi' }], source: { kind: 'plugin', plugin: 'test' } }) }) - it('a post-execute listener cannot mutate any nested part of the dispatched result', async () => { - // The decision is the ONLY sanctioned channel to change the outcome. A - // listener that reaches in and mutates the passed result reference (flipping - // isError, rewriting callId, attaching a bogus error) must NOT affect what - // execute() returns — the registry snapshots the authoritative fields before - // the waterfall and rebuilds from the snapshot + decision. - const ctx = await setup() - ctx.tools.register(echoTool) - ctx.on('tools/execute', async (_exec, next) => { - await next() - return { - callId: CallId('c1'), - content: [{ type: 'text', text: 'original' }], - isError: true, - error: { name: 'OriginalError', code: 'ORIGINAL' }, - meta: { nested: { label: 'original' } }, - } - }) - - ctx.on('tools/post-execute', async (_exec, result, next) => { - const mutable = result as { - callId: string - isError: boolean - error?: { name: string; code: string } - content: { type: 'text'; text: string }[] - meta?: { nested: { label: string } } - } - mutable.callId = 'hijacked' - mutable.isError = false - if (mutable.error) { - mutable.error.name = 'Evil' - mutable.error.code = 'EVIL' - } - mutable.content[0]!.text = 'MUTATED' - mutable.content.push({ type: 'text', text: 'INJECTED' }) // in-place array mutation - if (mutable.meta) mutable.meta.nested.label = 'MUTATED' - return next() // delegate to the default accept — no decision-level override - }) - - const result = await ctx.tools.execute({ callId: CallId('c1'), name: 'echo', arguments: { text: 'hi' } }) - expect(result.callId).toBe(CallId('c1')) // authoritative exec.callId, not 'hijacked' - expect(result.isError).toBe(true) - expect(result.error).toEqual({ name: 'OriginalError', code: 'ORIGINAL' }) - expect(result.content).toHaveLength(1) // the in-place push did not leak in - expect(result.content[0]).toMatchObject({ text: 'original' }) - expect(result.content.some(b => (b as { text?: string }).text === 'INJECTED')).toBe(false) - expect(result.meta).toEqual({ nested: { label: 'original' } }) - }) - it('composes pre + post waterfalls around dispatch (sandbox-wrap pattern)', async () => { const ctx = await setup() ctx.tools.register(echoTool) @@ -844,115 +556,20 @@ describe('ToolRegistry', () => { }) }) - it('normalizes malformed tools/execute results instead of treating them as success', async () => { + it('normalizes a tools/execute result with the wrong call id', async () => { const ctx = await setup() ctx.tools.register(echoTool) - let observedError: boolean | undefined - ctx.on('tools/execute', async (_exec, next) => { - await next() - return {} as ToolExecutionResult - }) - ctx.on('tools/result', (_exec, result) => { observedError = result.isError }) - - const result = await ctx.tools.execute({ callId: CallId('malformed'), name: 'echo', arguments: {} }) - expect(result.isError).toBe(true) - expect(result.content[0]).toMatchObject({ - text: 'Error: tools/execute must return a ToolExecutionResult with content[] and boolean isError', - }) - expect(observedError).toBe(true) - }) - - it('rejects non-JSON data at the defensive final-result notification boundary', async () => { - const ctx = await setup() - ctx.tools.register(echoTool) - let execution: ToolExecution | undefined - ctx.on('tools/execute', async (exec, next) => { - execution = exec - return next() - }) - await ctx.tools.execute({ callId: CallId('capture-execution'), name: 'echo', arguments: {} }) - if (execution === undefined) throw new Error('test fixture did not capture the execution') - - const internal = ctx.tools as unknown as { - notifyResult(exec: ToolExecution, result: ToolExecutionResult): Promise - } - const invalid = { - callId: CallId('capture-execution'), - content: new Map() as unknown as ToolExecutionResult['content'], - isError: false, - } - await expect(internal.notifyResult(execution, invalid)) - .rejects.toThrow('tool result notification must be losslessly JSON-serializable') - }) - - it.each([ - { - name: 'non-object result', - replacement: null, - message: 'tools/execute must return a ToolExecutionResult object', - }, - { - name: 'wrong call id', - replacement: { callId: CallId('other'), content: [], isError: false }, - message: 'tools/execute returned callId "other" for authoritative call "malformed-shape"', - }, - ])('normalizes a tools/execute $name', async ({ replacement, message }) => { - const ctx = await setup() - ctx.tools.register(echoTool) - ctx.on('tools/execute', async () => replacement as ToolExecutionResult) + ctx.on('tools/execute', async () => ({ callId: CallId('other'), content: [], isError: false })) const result = await ctx.tools.execute({ callId: CallId('malformed-shape'), name: 'echo', arguments: {}, }) expect(result.isError).toBe(true) - expect(result.content[0]).toMatchObject({ text: `Error: ${message}` }) - }) - - it('normalizes malformed tools/post-execute decisions', async () => { - const ctx = await setup() - ctx.tools.register(echoTool) - ctx.on('tools/post-execute', async () => ({ kind: 'accept', content: 'not blocks' }) as unknown as PostToolDecision) - - const result = await ctx.tools.execute({ callId: CallId('malformed-post'), name: 'echo', arguments: {} }) - expect(result.isError).toBe(true) expect(result.content[0]).toMatchObject({ - text: 'Error: tools/post-execute accept content must be an array', + text: 'Error: tools/execute returned callId "other" for authoritative call "malformed-shape"', }) }) - it.each([ - { - name: 'non-object decision', - replacement: null, - message: 'tools/post-execute must return a PostToolDecision object', - }, - { - name: 'block without feedback blocks', - replacement: { kind: 'block', feedback: 'not blocks' }, - message: 'tools/post-execute block feedback must be an array', - }, - { - name: 'unknown decision kind', - replacement: { kind: 'defer' }, - message: 'tools/post-execute must return an accept or block decision', - }, - { - name: 'non-JSON decision', - replacement: { kind: 'accept', content: new Map() }, - message: 'tools/post-execute must return a losslessly JSON-serializable decision', - }, - ])('normalizes a tools/post-execute $name', async ({ replacement, message }) => { - const ctx = await setup() - ctx.tools.register(echoTool) - ctx.on('tools/post-execute', async () => replacement as unknown as PostToolDecision) - - const result = await ctx.tools.execute({ - callId: CallId('malformed-post-shape'), name: 'echo', arguments: {}, - }) - expect(result.isError).toBe(true) - expect(result.content[0]).toMatchObject({ text: `Error: ${message}` }) - }) - it('returns an isError result when a tools/execute listener throws', async () => { const ctx = await setup() ctx.tools.register(echoTool) @@ -1030,109 +647,12 @@ describe('ToolRegistry', () => { }]) }) - it.each([ - ['Map', new Map([['mutable', true]])], - ['class instance', new (class Parameters { value = 1 })()], - ])('rejects cloneable non-JSON tool parameters (%s) without registry residue', async (_kind, parameters) => { + it('rejects a non-positive or non-finite registration timeout', async () => { const ctx = await setup() - const definition = { - ...echoTool, - name: 'invalid-parameters', - parameters, - } as unknown as typeof echoTool - - expect(() => ctx.tools.register(definition)).toThrow( - 'tool parameters must be losslessly JSON-serializable', - ) - expect(ctx.tools.get('invalid-parameters')).toBeUndefined() - }) - - it('reads nested tool parameters once into the accepted snapshot', async () => { - const ctx = await setup() - let reads = 0 - const parameters = Object.defineProperty({}, 'properties', { - enumerable: true, - get: () => ++reads === 1 ? {} : new Map([['mutable', true]]), - }) - - expect(() => ctx.tools.register({ - ...echoTool, - name: 'unstable-parameters', - parameters, - })).not.toThrow() - expect(reads).toBe(1) - expect(ctx.tools.get('unstable-parameters')?.parameters).toEqual({ properties: {} }) - }) - - it('reads a top-level parameters accessor once so validation and storage use one value', async () => { - const ctx = await setup() - const accepted = { type: 'object', properties: { accepted: { type: 'string' } } } - class DriftedParameters { - readonly type = 'object' - readonly properties = { drifted: { type: 'number' } } - } - let reads = 0 - const definition = { ...echoTool, name: 'top-level-parameters' } - Object.defineProperty(definition, 'parameters', { - enumerable: true, - get: () => { - reads += 1 - return reads === 1 ? accepted : new DriftedParameters() - }, - }) - - ctx.tools.register(definition) - - expect(reads).toBe(1) - expect(ctx.tools.get('top-level-parameters')?.parameters).toEqual(accepted) - }) - - it('rejects malformed fixed definition fields without freezing caller objects', async () => { - const ctx = await setup() - const badName = { value: 'object-name' } - const badDescription = { value: 'object-description' } - const badTimeout = { value: 100 } - - expect(() => ctx.tools.register({ ...echoTool, name: badName as unknown as string })) - .toThrow('tool name must be a string') - expect(() => ctx.tools.register({ ...echoTool, name: 'bad-description', description: badDescription as unknown as string })) - .toThrow('description must be a string') - expect(() => ctx.tools.register({ ...echoTool, name: 'bad-timeout', timeoutMs: badTimeout as unknown as number })) - .toThrow('timeoutMs must be a positive finite number') expect(() => ctx.tools.register({ ...echoTool, name: 'zero-timeout', timeoutMs: 0 })) .toThrow('timeoutMs must be a positive finite number') - expect(() => ctx.tools.register({ ...echoTool, name: 'bad-execute', execute: { bind() {} } as unknown as typeof echoTool.execute })) - .toThrow('execute must be a function') - expect(() => ctx.tools.register({ ...echoTool, name: 'bad-present-call', presentCall: 1 as unknown as NonNullable })) - .toThrow('presentCall must be a function') - expect(() => ctx.tools.register({ ...echoTool, name: 'bad-present-result', presentResult: 1 as unknown as NonNullable })) - .toThrow('presentResult must be a function') - expect(Object.isFrozen(badName)).toBe(false) - expect(Object.isFrozen(badDescription)).toBe(false) - expect(Object.isFrozen(badTimeout)).toBe(false) - expect(ctx.tools.schemas()).toEqual([]) - }) - - it('snapshots callbacks while preserving their registration-time method receiver', async () => { - const ctx = await setup() - const receivers: object[] = [] - const definition = { - ...echoTool, - name: 'callback-snapshot', - async execute() { - receivers.push(this) - return [{ type: 'text' as const, text: 'original' }] - }, - } - ctx.tools.register(definition) - definition.execute = async () => [{ type: 'text' as const, text: 'replacement' }] - - const result = await ctx.tools.execute({ - callId: CallId('callback-snapshot'), name: definition.name, arguments: {}, - }) - - expect(receivers).toEqual([definition]) - expect(result.content).toEqual([{ type: 'text', text: 'original' }]) + expect(() => ctx.tools.register({ ...echoTool, name: 'infinite-timeout', timeoutMs: Number.POSITIVE_INFINITY })) + .toThrow('timeoutMs must be a positive finite number') }) it('rejects duplicate names and unregisters on fiber dispose (HMR safety)', async () => { @@ -1305,101 +825,6 @@ describe('defineTool / schema DSL', () => { expect(result.content).toEqual([{ type: 'text', text: 'HELLO' }]) }) - it('reads defineTool options once and keeps wire and runtime schemas on one detached snapshot', async () => { - const accepted: SchemaSpec = { value: { type: 'string', required: true, enum: ['accepted'] } } - const drifted: SchemaSpec = { count: { type: 'number', required: true } } - const reads = { - name: 0, - description: 0, - parameters: 0, - timeoutMs: 0, - execute: 0, - presentCall: 0, - presentResult: 0, - } - const options = {} as DefineToolOptions - Object.defineProperties(options, { - name: { enumerable: true, get: () => { reads.name += 1; return reads.name === 1 ? 'accepted' : 'drifted' } }, - description: { enumerable: true, get: () => { reads.description += 1; return reads.description === 1 ? 'accepted description' : 'drifted description' } }, - parameters: { enumerable: true, get: () => { reads.parameters += 1; return reads.parameters === 1 ? accepted : drifted } }, - timeoutMs: { enumerable: true, get: () => { reads.timeoutMs += 1; return reads.timeoutMs === 1 ? 250 : 0 } }, - execute: { - enumerable: true, - get: () => { - reads.execute += 1 - return (args: Record) => Promise.resolve([{ type: 'text' as const, text: String(args['value']) }]) - }, - }, - presentCall: { - enumerable: true, - get: () => { - reads.presentCall += 1 - return (args: Record) => ({ card: 'generic' as const, title: String(args['value']) }) - }, - }, - presentResult: { - enumerable: true, - get: () => { - reads.presentResult += 1 - return (args: Record) => ({ card: 'generic' as const, title: String(args['value']) }) - }, - }, - }) - - const tool = defineTool(options) - accepted.value!.type = 'number' - accepted.value!.enum!.push('mutated') - - expect(tool).toMatchObject({ - name: 'accepted', - description: 'accepted description', - timeoutMs: 250, - parameters: { - type: 'object', - properties: { value: { type: 'string', enum: ['accepted'] } }, - required: ['value'], - }, - }) - await expect(tool.execute({ value: 'accepted' }, {} as ToolExecution)) - .resolves.toEqual([{ type: 'text', text: 'accepted' }]) - expect(tool.presentCall?.({ value: 'accepted' })).toEqual({ card: 'generic', title: 'accepted' }) - expect(tool.presentResult?.( - { value: 'accepted' }, - { content: [], isError: false }, - )).toEqual({ card: 'generic', title: 'accepted' }) - expect(reads).toEqual({ - name: 1, - description: 1, - parameters: 1, - timeoutMs: 1, - execute: 1, - presentCall: 1, - presentResult: 1, - }) - }) - - it('rejects an exotic defineTool schema before it can be normalized for the wire', () => { - class ExoticDefault { readonly value = 'not JSON' } - - expect(() => defineTool({ - name: 'exotic-schema', - description: 'must reject exotic defaults', - parameters: { - value: { type: 'string', default: new ExoticDefault() }, - }, - execute: () => Promise.resolve([]), - })).toThrow(/parameters must be losslessly JSON-serializable/) - }) - - it('rejects a malformed defineTool spec whose generated wire schema is not JSON', () => { - expect(() => defineTool({ - name: 'malformed-schema', - description: 'missing property type', - parameters: { value: {} } as unknown as SchemaSpec, - execute: () => Promise.resolve([]), - })).toThrow(/generated parameters must be losslessly JSON-serializable/) - }) - it('type-level: InferArgs maps required properties to non-optional', () => { // Compile-time check: if this compiles, InferArgs is correct. // args.a is string (required), args.b is number|undefined (optional). diff --git a/packages/session-persistence/session-persistence-jsonl/tests/jsonl.spec.ts b/packages/session-persistence/session-persistence-jsonl/tests/jsonl.spec.ts index 6eed63a2f4..757ba03aac 100644 --- a/packages/session-persistence/session-persistence-jsonl/tests/jsonl.spec.ts +++ b/packages/session-persistence/session-persistence-jsonl/tests/jsonl.spec.ts @@ -4,7 +4,7 @@ import { appendFile, mkdtemp, mkdir, rm, readFile, writeFile, readdir, stat } fr import { tmpdir } from 'node:os' import { join } from 'node:path' import SessionStore, { SessionId } from '@deepseek-ai/dsh-session' -import type { Session, SessionEvent } from '@deepseek-ai/dsh-session' +import type { Session, SessionEvent, SessionHeader } from '@deepseek-ai/dsh-session' import SessionPersistenceJsonl from '@deepseek-ai/dsh-session-persistence-jsonl' import { encodeSegment, logPath, scanLog, sessionDir } from '../src/format.ts' import { runPersistenceContract, meta, oneTurnLog, appendLog } from '../../session-persistence/tests/contract.ts' @@ -13,6 +13,13 @@ import { runCoordinatorContract, type CoordinatorFixture } from '../../session-p let root: string const dirs: string[] = [] +type MutableSessionHeader = { -readonly [K in keyof SessionHeader]: SessionHeader[K] } + +/** Test-only mutable view used to verify that backends detach returned/caller metadata. */ +function mutableHeader(header: SessionHeader): MutableSessionHeader { + return header +} + async function freshRoot(): Promise { const dir = await mkdtemp(join(tmpdir(), 'dsh-jsonl-')) dirs.push(dir) @@ -255,7 +262,7 @@ describe('SessionPersistenceJsonl: durability and crash semantics', () => { const loaded = await ctx.sessionPersistence.load(m.id) // A consumer mutates the returned meta's cwd. The backend's stored pathing // metadata must be unaffected, so a later append still finds the right log. - loaded.meta.cwd = '/evil' + mutableHeader(loaded.meta).cwd = '/evil' await ctx.sessionPersistence.append(m.id, [ { type: 'turn/start', seq: 6, time: 9, data: { turn: 2, trigger: { kind: 'message', source: { kind: 'user' } } } }, { type: 'turn/end', seq: 7, time: 10, data: { turn: 2, reason: { kind: 'completed' } } }, @@ -421,7 +428,7 @@ describe('SessionPersistenceJsonl: edge cases', () => { const m = meta('create-snap', '/orig') const p = ctx.sessionPersistence.create(m) // Mutate the caller's meta object immediately after calling create. - m.cwd = '/mutated' + mutableHeader(m).cwd = '/mutated' await p await ctx.sessionPersistence.append(SessionId('create-snap'), oneTurnLog()) // The log materialized under the ORIGINAL cwd, not the mutated one. diff --git a/packages/skill/skill/README.md b/packages/skill/skill/README.md index b035acf767..ebfac17e2e 100644 --- a/packages/skill/skill/README.md +++ b/packages/skill/skill/README.md @@ -8,28 +8,28 @@ This package owns the `ctx.skills` interface. It does not know whether skills co ### Public API -- `ctx.skills.registerProvider(provider): () => Promise | void` Registers a provider by unique `provider.name`. Duplicate provider names throw, and `runtime` is reserved for `ctx.skills.register(...)`. The registry snapshots the name and callback identities at registration, so replacing those fields later cannot change lookup or HMR cleanup; callbacks remain bound to the original provider object and can still read its mutable state. The registration is effect-scoped and HMR-safe, and the exact Cordis disposer supports ordered composite teardown. -- `ctx.skills.list({ cwd?, signal? })` Snapshots the lookup options, then returns detached model-invocable summaries for the current workspace, merged across providers and sorted by name. -- `ctx.skills.get(name, { cwd?, signal? })` Uses one lookup-options snapshot to select and load the winner, rechecks cancellation after discovery or a cache hit, races provider loading against the same signal, then returns a detached full definition, including disabled-for-model skills. -- `ctx.skills.register(skill): () => Promise | void` Registers a detached runtime embedded skill. Same-name runtime registrations are first-wins: a duplicate logs a warning and gets a no-op disposer. Successful registrations return the exact Cordis disposer for ordered composite teardown. +- `ctx.skills.registerProvider(provider): () => Promise | void` Registers a readonly provider by unique `provider.name`. Duplicate provider names throw, and `runtime` is reserved for `ctx.skills.register(...)`. The registry borrows the provider object and invokes its methods directly. The registration is effect-scoped and HMR-safe, and the exact Cordis disposer supports ordered composite teardown. +- `ctx.skills.list({ cwd?, signal? })` Borrows the readonly lookup options, then returns model-invocable summaries for the current workspace, merged across providers and sorted by name. +- `ctx.skills.get(name, { cwd?, signal? })` Uses the same readonly options and winning candidate for discovery and loading, rechecks cancellation after discovery or a cache hit, races provider loading against the signal, validates the loaded definition, then returns it, including disabled-for-model skills. +- `ctx.skills.register(skill): () => Promise | void` Registers a readonly runtime embedded skill, adding `provider: "runtime"` when omitted. Same-name runtime registrations are first-wins: a duplicate logs a warning and gets a no-op disposer. Successful registrations return the exact Cordis disposer for ordered composite teardown. ### Config | Field | Default | Meaning | |---|---|---| -| `collectCacheMaxEntries` | `128` | Maximum completed cwd/provider catalog snapshots kept in memory. | +| `collectCacheMaxEntries` | `128` | Maximum completed cwd/provider catalogs kept in memory. | ## Provider Contract -A provider registers synchronously from its `apply()` and returns `SkillCandidate[]` from `list(options)` when discovery is requested. Registration copies `name` and binds the current `list` and `get` methods once; replacing those fields on the caller-owned object later does not rewrite the live registry entry, and disposal always removes the original name. Remote setup, authentication, and discovery belong in the awaited `list()` call rather than plugin registration. Providers should stop promptly when `options.signal` aborts; the registry also stops awaiting uncooperative discovery and loading work so agent cancellation cannot hang prefix composition or skill loading. +A provider registers synchronously from its `apply()` and returns `readonly SkillCandidate[]` from `list(options)` when discovery is requested. The provider, lookup options, candidates, and loaded definitions are readonly same-process contracts: the registry borrows them rather than cloning, freezing, or rebinding callbacks. Remote setup, authentication, and discovery belong in the awaited `list()` call rather than plugin registration. Providers should stop promptly when `options.signal` aborts; the registry also stops awaiting uncooperative discovery and loading work so agent cancellation cannot hang prefix composition or skill loading. -Each public lookup captures `cwd` and the abort-signal identity once before cache or provider work, and providers receive that frozen lookup record. The registry reads each returned candidate once, validates that snapshot, and detaches its resource metadata before caching it. The winning provider receives another detached candidate in `get(candidate, options)`, while `candidate.locator` preserves the exact provider-owned identity originally returned by `list()`; a local provider can therefore use a file-path handle while a remote provider can use a URL, id, or version token. A loaded definition is detached again before it reaches the caller. +The registry validates parsed provider candidates before caching them and validates loaded definitions before returning them. The winning provider receives the exact candidate and opaque `locator` identity it returned from `list()`; a local provider can therefore use a file-path handle while a remote provider can use a URL, id, or version token. Callers and providers must honor the readonly contract after handing values to the registry. -The registry validates fixed provider, candidate, runtime-registration, and loaded-definition fields before detachment: names/descriptions/content use their declared string types, ranks are finite numbers, and `disableModelInvocation` is boolean when present. Caller-owned objects masquerading as scalars are rejected without being frozen. Candidate contract violations fail fast because the provider plugin is malformed; a provider `list()` rejection is treated as a transient source failure, logged, skipped for that request, and not cached. Only completed, registry-owned catalogs are cached, and a provider/runtime revision change during discovery discards the stale result and retries. Duplicate skill names are resolved first-wins by `rank`, provider registration order, then the provider's own local order. The final summary list is sorted by skill `name` for deterministic consumers. +Parsed candidate and loaded-definition fields are validated at the provider boundary: names/descriptions/content use their declared string types, ranks are finite numbers, and `disableModelInvocation` is boolean when present. Candidate contract violations fail fast because the provider or its parser is malformed; a provider `list()` rejection is treated as a transient source failure, logged, skipped for that request, and not cached. Only completed catalogs are cached, and a provider/runtime revision change during discovery discards the stale result and retries. Duplicate skill names are resolved first-wins by `rank`, provider registration order, then the provider's own local order. The final summary list is sorted by skill `name` for deterministic consumers. ## Runtime Skills -`ctx.skills.register(...)` is a convenience for embedded runtime skills. Runtime skills use rank `250`: project providers can override them, while they override the shipped local provider's custom and user roots. Runtime registration detaches the accepted definition and nested resource metadata; later mutation of the registration object or a returned list/get value cannot rewrite the live skill. Registration is also first-wins within runtime contributions, so a duplicate contribution cannot remove the active one through its disposer. +`ctx.skills.register(...)` is a convenience for embedded runtime skills. Runtime skills use rank `250`: project providers can override them, while they override the shipped local provider's custom and user roots. Runtime definitions and nested resource metadata are borrowed readonly; the service only materializes the top-level definition needed to supply the default `provider`. Registration is first-wins within runtime contributions, so a duplicate contribution cannot remove the active one through its disposer. ## Consumer boundary diff --git a/packages/skill/skill/src/index.ts b/packages/skill/skill/src/index.ts index f06788cde7..f2ece2379f 100644 --- a/packages/skill/skill/src/index.ts +++ b/packages/skill/skill/src/index.ts @@ -32,56 +32,56 @@ export type SkillSource = 'project-dsh' | 'project-agents' | 'runtime' | 'user-d /** Optional provider-specific base used by loaded skill bodies to resolve relative resources. */ export type SkillResourceBase = - | { kind: 'directory'; path: string } - | { kind: 'url'; url: string } - | { kind: 'opaque'; description: string } + | { readonly kind: 'directory'; readonly path: string } + | { readonly kind: 'url'; readonly url: string } + | { readonly kind: 'opaque'; readonly description: string } /** Model-visible skill metadata returned by `ctx.skills.list()` and rendered into request guidance. */ export interface SkillSummary { /** Kebab-case identifier used with the `skill` tool. */ - name: string + readonly name: string /** Short routing description shown to the model. */ - description: string + readonly description: string /** Optional extra routing guidance shown to the model. */ - whenToUse?: string + readonly whenToUse?: string /** Whether the skill is hidden from model listings while remaining loadable by trusted callers. */ - disableModelInvocation?: boolean + readonly disableModelInvocation?: boolean /** Discovery source that produced this winning skill. */ - source: SkillSource + readonly source: SkillSource /** Provider that owns this skill body. */ - provider: string + readonly provider: string /** Provider-specific base for relative resources. */ - resourceBase?: SkillResourceBase + readonly resourceBase?: SkillResourceBase } /** Provider catalog entry used by the registry to merge and later load skills. */ export interface SkillCandidate extends SkillSummary { /** Lower ranks win duplicate skill names before provider registration order is considered. */ - rank: number + readonly rank: number /** Opaque provider-owned handle passed back to `provider.get()`. */ - locator: unknown + readonly locator: unknown /** Absolute file path when the provider has one. */ - path?: string + readonly path?: string /** Parsed optional metadata object from provider-specific skill frontmatter. */ - metadata?: Record + readonly metadata?: Readonly> } /** Complete parsed skill definition, including the body loaded by `ctx.skills.get()`. */ export interface SkillDefinition extends SkillSummary { /** Markdown instruction body after any provider-specific metadata removal. */ - content: string + readonly content: string /** Absolute file path when the skill came from disk. */ - path?: string + readonly path?: string /** Parsed optional metadata object from frontmatter. */ - metadata?: Record + readonly metadata?: Readonly> } /** Runtime skill contribution accepted by `ctx.skills.register()`. */ -export type SkillRegistration = Omit & { provider?: string } +export type SkillRegistration = Omit & { readonly provider?: string } /** Caller context used for cwd-sensitive and abortable provider work. */ export interface SkillLookupOptions { - /** Workspace selector captured at lookup entry; providers receive a read-only snapshot. */ + /** Workspace selector for the current lookup. */ readonly cwd?: string | undefined /** Abort discovery or loading work for the current caller. */ readonly signal?: AbortSignal | undefined @@ -90,7 +90,7 @@ export interface SkillLookupOptions { /** Provider interface for one source of skills, such as local directories or a remote registry. */ export interface SkillProvider { /** Unique provider name in the `ctx.skills` registry. */ - name: string + readonly name: string /** * List available skill candidates for the current lookup context. Provider * plugins register synchronously during `apply()`; remote initialization, @@ -99,21 +99,20 @@ export interface SkillProvider { * @param options - lookup options; `cwd` selects workspace-sensitive skills and `signal` cancels work. * @returns provider candidates with precedence ranks and opaque locators. */ - list(options: SkillLookupOptions): Promise + readonly list: (options: SkillLookupOptions) => Promise /** * Load a complete skill body for a previously listed candidate. - * @param candidate - a detached snapshot of the winning candidate; its opaque - * `locator` retains the exact identity originally returned by this provider. + * @param candidate - the winning candidate originally returned by this provider. * @param options - lookup options; `cwd` selects workspace-sensitive skills and `signal` cancels work. * @returns the full skill body, or `undefined` if it is no longer loadable. */ - get(candidate: SkillCandidate, options: SkillLookupOptions): Promise + readonly get: (candidate: SkillCandidate, options: SkillLookupOptions) => Promise } /** Skill registry configuration. */ export interface Config { - /** Maximum number of completed cwd/provider catalog snapshots kept in memory. */ - collectCacheMaxEntries?: number + /** Maximum number of completed cwd/provider catalogs kept in memory. */ + readonly collectCacheMaxEntries?: number } declare module 'cordis' { @@ -163,7 +162,7 @@ export class SkillService extends Service { private readonly collectCacheMaxEntries: number private readonly providers = new Map() - private readonly runtime = new Map() + private readonly runtime = new Map() private readonly collectCache = new Map() private providerRevision = 0 private nextProviderOrder = 0 @@ -179,53 +178,38 @@ export class SkillService extends Service { * Register a skill provider synchronously during the provider plugin's * `apply()`. Throws if another provider already owns the same provider name, * including the reserved runtime provider name. Providers that need remote - * initialization do that work inside `list()` after registration. The name - * and callback identities are snapshotted at registration, so later - * replacement of those fields cannot change the registry key, dispatch - * callbacks, or HMR cleanup identity. Bound callbacks retain the original - * provider object as their receiver, so provider-owned mutable state remains - * live. Effect-scoped and HMR-safe: disposing the caller's fiber unregisters - * the provider and invalidates cached catalogs. + * initialization do that work inside `list()` after registration. Providers + * are readonly same-process registrations: the registry borrows the provider + * object and invokes its methods directly. Effect-scoped and HMR-safe: + * disposing the caller's fiber unregisters the provider and invalidates + * cached catalogs. * @param provider - the provider to register by `provider.name`. * @returns the exact Cordis effect disposer that unregisters this provider; * composite effects may yield it directly to preserve teardown ordering. */ registerProvider(provider: SkillProvider): () => Promise | void { - // Snapshot the registration contract before entering the effect. The - // callback binding preserves the historical method receiver while making - // replacement of `provider.list`/`provider.get` after registration inert. - // In particular, cleanup must never re-read caller-owned `provider.name`: - // an HMR host may mutate or reuse that object before its old fiber unloads. const name = provider.name - // eslint-disable-next-line @typescript-eslint/unbound-method - const inputList = provider.list - // eslint-disable-next-line @typescript-eslint/unbound-method - const inputGet = provider.get - if (typeof name !== 'string') throw new TypeError('skill provider name must be a string') - if (typeof inputList !== 'function') throw new TypeError(`skill provider "${name}" list must be a function`) - if (typeof inputGet !== 'function') throw new TypeError(`skill provider "${name}" get must be a function`) - const snapshot: SkillProvider = Object.freeze({ - name, - list: inputList.bind(provider), - get: inputGet.bind(provider), - }) - const dispose = this.ctx.effect(function* (this: SkillService) { - if (snapshot.name === RUNTIME_PROVIDER) { - throw new Error(`"${RUNTIME_PROVIDER}" is reserved for runtime skill registrations`) - } - if (this.providers.has(snapshot.name)) { - throw new Error(`a skill provider named "${snapshot.name}" is already registered`) - } - this.providers.set(snapshot.name, { provider: snapshot, order: this.nextProviderOrder }) - this.nextProviderOrder += 1 - this.invalidateCache() + if (name === RUNTIME_PROVIDER) { + throw new Error(`"${RUNTIME_PROVIDER}" is reserved for runtime skill registrations`) + } + if (this.providers.has(name)) { + throw new Error(`a skill provider named "${name}" is already registered`) + } + const providers = this.providers + const ctx = this.ctx + const order = this.nextProviderOrder + const invalidateCache = (): void => { this.invalidateCache() } + this.nextProviderOrder += 1 + const dispose = ctx.effect(function* () { + providers.set(name, { provider, order }) + invalidateCache() yield () => { - this.providers.delete(snapshot.name) - this.invalidateCache() - this.ctx.emit('skill/provider-removed', snapshot.name) + providers.delete(name) + invalidateCache() + ctx.emit('skill/provider-removed', name) } - this.ctx.emit('skill/provider-added', snapshot) - }.bind(this), 'skills.registerProvider()') + ctx.emit('skill/provider-added', provider) + }, 'skills.registerProvider()') return dispose } @@ -233,44 +217,46 @@ export class SkillService extends Service { * Register a runtime skill contribution. Runtime registrations are treated as * embedded provider entries with project-over-user priority. Same-name runtime * registrations are first-wins: a duplicate logs a warning and gets a no-op - * disposer so it cannot remove the active contribution. The registry detaches - * the accepted definition, including nested resource metadata, so later caller - * mutation cannot rewrite the live contribution. + * disposer so it cannot remove the active contribution. Runtime definitions + * are readonly same-process registrations; the registry borrows their nested + * resource metadata. * @param skill - the complete skill definition to expose for discovery. * @returns the exact Cordis effect disposer that removes this runtime * contribution and invalidates caches; composite effects may yield it * directly to preserve teardown ordering. */ register(skill: SkillRegistration): () => Promise | void { - const normalized = normalizeRuntimeSkill(skill) - const existing = this.runtime.get(normalized.name) + validateRuntimeSkill(skill) + const existing = this.runtime.get(skill.name) if (existing !== undefined) { - this.ctx.logger.warn(`runtime skill "${normalized.name}" ignored because it is already registered`) + this.ctx.logger.warn(`runtime skill "${skill.name}" ignored because it is already registered`) return () => {} } - const dispose = this.ctx.effect(function* (this: SkillService) { - this.runtime.set(normalized.name, normalized) - this.runtimeRevision += 1 - this.invalidateCache() + const runtime = this.runtime + const updateRevision = (): void => { this.runtimeRevision += 1 } + const invalidateCache = (): void => { this.invalidateCache() } + const dispose = this.ctx.effect(function* () { + runtime.set(skill.name, skill) + updateRevision() + invalidateCache() yield () => { - this.runtime.delete(normalized.name) - this.runtimeRevision += 1 - this.invalidateCache() + runtime.delete(skill.name) + updateRevision() + invalidateCache() } - }.bind(this), 'skills.register()') + }, 'skills.register()') return dispose } /** - * List model-invocable skill summaries for a workspace. The lookup options are - * snapshotted before discovery, and every returned summary is detached from the - * cached provider catalog. + * List model-invocable skill summaries for a workspace. Lookup options and + * provider candidates are readonly same-process values borrowed throughout + * discovery. * @param options - lookup options; `cwd` selects project roots and `signal` cancels discovery. * @returns sorted summaries, excluding skills disabled for model invocation. */ async list(options: SkillLookupOptions = {}): Promise { - const accepted = snapshotLookupOptions(options) - return (await this.collect(accepted)) + return (await this.collect(options)) .map(entry => entry.candidate) .filter(skill => skill.disableModelInvocation !== true) .map(toSummary) @@ -278,10 +264,10 @@ export class SkillService extends Service { } /** - * Load one full skill definition by name. One lookup-options snapshot selects - * and loads the winner; the provider receives detached candidate metadata with - * its opaque locator identity preserved, and the returned definition is also - * detached from provider-owned data. Cancellation is rechecked after catalog + * Load one full skill definition by name. The provider receives the winning + * candidate it returned during discovery, including its opaque locator, and + * the registry returns the provider's definition after validating it. + * Cancellation is rechecked after catalog * selection (including a cache hit), and provider loading is raced against the * same signal so an uncooperative provider cannot hang the caller. * @param name - kebab-case skill name. @@ -290,16 +276,17 @@ export class SkillService extends Service { */ async get(name: string, options: SkillLookupOptions = {}): Promise { if (!isSkillName(name)) return undefined - const accepted = snapshotLookupOptions(options) - const collected = await this.collect(accepted) - throwIfAborted(accepted.signal) + const collected = await this.collect(options) + throwIfAborted(options.signal) const match = collected.find(entry => entry.candidate.name === name) if (match === undefined) return undefined const definition = await waitWithAbort( - match.provider.get(copyCandidate(match.candidate), accepted), - accepted.signal, + match.provider.get(match.candidate, options), + options.signal, ) - return definition === undefined ? undefined : snapshotDefinition(definition) + if (definition === undefined) return undefined + validateDefinition(definition) + return definition } private async collect(options: SkillLookupOptions): Promise { @@ -358,21 +345,22 @@ export class SkillService extends Service { } for (const { provider, order } of [...this.providers.values()]) { let localOrder = 0 - let listed: SkillCandidate[] | undefined + let output: unknown try { - listed = await waitWithAbort(provider.list(options), options.signal) + output = await waitWithAbort(provider.list(options), options.signal) } catch (error) { if (options.signal?.aborted === true) throw toError(options.signal.reason) cacheable = false this.ctx.logger.warn(`skill provider "${provider.name}" skipped: ${errorMessage(error)}`) } - if (listed === undefined) continue - if (!Array.isArray(listed)) { + if (output === undefined) continue + if (!Array.isArray(output)) { throw new TypeError(`skill provider "${provider.name}" list() must return an array`) } + const listed = output as readonly SkillCandidate[] for (const candidate of listed) { - const snapshot = snapshotCandidate(candidate, provider.name) - candidates.push({ candidate: snapshot, provider, providerOrder: order, localOrder }) + validateCandidate(candidate, provider.name) + candidates.push({ candidate, provider, providerOrder: order, localOrder }) localOrder += 1 } } @@ -392,14 +380,20 @@ const RUNTIME_SKILL_PROVIDER: SkillProvider = { return Promise.resolve([]) }, get(candidate) { - const skill = candidate.locator as SkillDefinition - return Promise.resolve({ ...skill }) + const skill = candidate.locator as SkillRegistration + return Promise.resolve({ ...skill, provider: skill.provider ?? RUNTIME_PROVIDER }) }, } -function runtimeCandidate(skill: SkillDefinition): SkillCandidate { +function runtimeCandidate(skill: SkillRegistration): SkillCandidate { return { - ...toSummary(skill), + name: skill.name, + description: skill.description, + ...skill.whenToUse !== undefined ? { whenToUse: skill.whenToUse } : {}, + ...skill.disableModelInvocation !== undefined ? { disableModelInvocation: skill.disableModelInvocation } : {}, + source: skill.source, + provider: skill.provider ?? RUNTIME_PROVIDER, + ...skill.resourceBase !== undefined ? { resourceBase: skill.resourceBase } : {}, rank: RUNTIME_RANK, locator: skill, ...skill.path !== undefined ? { path: skill.path } : {}, @@ -407,50 +401,6 @@ function runtimeCandidate(skill: SkillDefinition): SkillCandidate { } } -/** Read provider candidate data once and detach it while preserving its opaque locator identity. */ -function copyCandidate(candidate: SkillCandidate, providerName?: string): SkillCandidate { - const name = candidate.name - const description = candidate.description - const whenToUse = candidate.whenToUse - const disableModelInvocation = candidate.disableModelInvocation - const source = candidate.source - const provider = candidate.provider - const resourceBase = candidate.resourceBase - const rank = candidate.rank - const locator = candidate.locator - const path = candidate.path - const metadata = candidate.metadata - const accepted: SkillCandidate = { - name, - description, - ...whenToUse !== undefined ? { whenToUse } : {}, - ...disableModelInvocation !== undefined ? { disableModelInvocation } : {}, - source, - provider, - ...resourceBase !== undefined ? { resourceBase } : {}, - rank, - // `locator` is the one deliberately provider-owned capability in a - // candidate. Its exact identity must round-trip back to provider.get(). - locator, - ...path !== undefined ? { path } : {}, - ...metadata !== undefined ? { metadata } : {}, - } - // Validate the exact scalar snapshot before cloning nested data. This keeps a - // malformed candidate's provider-contract error from being masked by an - // unrelated DataCloneError in its metadata. - if (providerName !== undefined) validateCandidate(accepted, providerName) - return { - ...accepted, - ...resourceBase !== undefined ? { resourceBase: structuredClone(resourceBase) } : {}, - ...metadata !== undefined ? { metadata: structuredClone(metadata) } : {}, - } -} - -/** Normalize one provider result into the registry-owned catalog snapshot. */ -function snapshotCandidate(candidate: SkillCandidate, providerName: string): SkillCandidate { - return copyCandidate(candidate, providerName) -} - function validateCandidate(candidate: SkillCandidate, providerName: string): void { if (typeof candidate.name !== 'string') { throw new TypeError(`skill provider "${providerName}" returned a non-string skill name`) @@ -487,58 +437,21 @@ function validateCandidate(candidate: SkillCandidate, providerName: string): voi } } -function normalizeRuntimeSkill(skill: SkillRegistration): SkillDefinition { - // Read every caller-owned top-level field once so validation and storage use - // one coherent definition even when JavaScript accessors are involved. - const name = skill.name - const description = skill.description - const whenToUse = skill.whenToUse - const disableModelInvocation = skill.disableModelInvocation - const source = skill.source - const inputProvider = skill.provider - const provider = inputProvider === undefined ? RUNTIME_PROVIDER : inputProvider - const resourceBase = skill.resourceBase - const content = skill.content - const path = skill.path - const metadata = skill.metadata - if (typeof name !== 'string') throw new TypeError('runtime skill name must be a string') - if (!SKILL_NAME.test(name)) throw new Error(`invalid skill name "${name}"`) - if (typeof description !== 'string') throw new TypeError(`skill "${name}" description must be a string`) - if (description.length === 0) throw new Error(`skill "${name}" requires a description`) - if (disableModelInvocation !== undefined && typeof disableModelInvocation !== 'boolean') { - throw new TypeError(`skill "${name}" disableModelInvocation must be a boolean`) - } - if (whenToUse !== undefined && typeof whenToUse !== 'string') throw new TypeError(`skill "${name}" whenToUse must be a string`) - if (typeof source !== 'string') throw new TypeError(`skill "${name}" source must be a string`) - if (typeof provider !== 'string') throw new TypeError(`skill "${name}" provider must be a string`) - if (typeof content !== 'string') throw new TypeError(`skill "${name}" content must be a string`) - if (path !== undefined && typeof path !== 'string') throw new TypeError(`skill "${name}" path must be a string`) - return { - name, - description, - ...whenToUse !== undefined ? { whenToUse } : {}, - ...disableModelInvocation !== undefined ? { disableModelInvocation } : {}, - source, - provider, - ...resourceBase !== undefined ? { resourceBase: structuredClone(resourceBase) } : {}, - content, - ...path !== undefined ? { path } : {}, - ...metadata !== undefined ? { metadata: structuredClone(metadata) } : {}, - } +function validateRuntimeSkill(skill: SkillRegistration): void { + if (!SKILL_NAME.test(skill.name)) throw new Error(`invalid skill name "${skill.name}"`) + if (skill.description.length === 0) throw new Error(`skill "${skill.name}" requires a description`) } -/** Detach a provider-loaded definition before it crosses back to the caller. */ -function snapshotDefinition(skill: SkillDefinition): SkillDefinition { +/** Validate a definition loaded from a provider-controlled parser or remote source. */ +function validateDefinition(skill: SkillDefinition): void { const name = skill.name const description = skill.description const whenToUse = skill.whenToUse const disableModelInvocation = skill.disableModelInvocation const source = skill.source const provider = skill.provider - const resourceBase = skill.resourceBase const content = skill.content const path = skill.path - const metadata = skill.metadata if (typeof name !== 'string') throw new TypeError('loaded skill name must be a string') if (!SKILL_NAME.test(name)) throw new Error(`loaded skill has invalid name "${name}"`) if (typeof description !== 'string') throw new TypeError(`loaded skill "${name}" description must be a string`) @@ -551,18 +464,6 @@ function snapshotDefinition(skill: SkillDefinition): SkillDefinition { if (typeof provider !== 'string') throw new TypeError(`loaded skill "${name}" provider must be a string`) if (typeof content !== 'string') throw new TypeError(`loaded skill "${name}" content must be a string`) if (path !== undefined && typeof path !== 'string') throw new TypeError(`loaded skill "${name}" path must be a string`) - return { - name, - description, - ...whenToUse !== undefined ? { whenToUse } : {}, - ...disableModelInvocation !== undefined ? { disableModelInvocation } : {}, - source, - provider, - ...resourceBase !== undefined ? { resourceBase: structuredClone(resourceBase) } : {}, - content, - ...path !== undefined ? { path } : {}, - ...metadata !== undefined ? { metadata: structuredClone(metadata) } : {}, - } } function toSummary(skill: SkillDefinition | SkillCandidate): SkillSummary { @@ -574,7 +475,7 @@ function toSummary(skill: SkillDefinition | SkillCandidate): SkillSummary { ...disableModelInvocation !== undefined ? { disableModelInvocation } : {}, source, provider, - ...resourceBase !== undefined ? { resourceBase: structuredClone(resourceBase) } : {}, + ...resourceBase !== undefined ? { resourceBase } : {}, } } @@ -604,16 +505,6 @@ function collectCacheKey(options: SkillLookupOptions, providerRevision: number, return JSON.stringify({ cwd: options.cwd, providerRevision, runtimeRevision }) } -/** Capture one lookup identity before any provider or cache async boundary. */ -function snapshotLookupOptions(options: SkillLookupOptions): Readonly { - const cwd = options.cwd - const signal = options.signal - return Object.freeze({ - ...cwd !== undefined ? { cwd } : {}, - ...signal !== undefined ? { signal } : {}, - }) -} - function waitWithAbort(promise: Promise, signal: AbortSignal | undefined): Promise { if (signal === undefined) return promise throwIfAborted(signal) @@ -636,7 +527,6 @@ function waitWithAbort(promise: Promise, signal: AbortSignal | undefined): reject(toError(error)) }, ) - if (signal.aborted) onAbort() }) } diff --git a/packages/skill/skill/tests/skill.spec.ts b/packages/skill/skill/tests/skill.spec.ts index b47bdb6431..90bf1ea983 100644 --- a/packages/skill/skill/tests/skill.spec.ts +++ b/packages/skill/skill/tests/skill.spec.ts @@ -107,85 +107,9 @@ describe('SkillService registry', () => { expect((await ctx.skills.list()).map(skill => skill.name)).toEqual(['same-rank-skill', 'shadowed']) }) - it('snapshots a provider registration so caller mutation cannot corrupt HMR cleanup', async () => { + it('validates parsed candidate fields', async () => { const ctx = new Context() await ctx.plugin(SkillService) - const candidate: SkillCandidate = { - name: 'stable-skill', - description: 'Stable skill', - provider: 'stable-provider', - source: 'test', - rank: 1, - locator: 'original', - } - const originalList = vi.fn(() => Promise.resolve([candidate])) - const originalGet = vi.fn((listed: SkillCandidate) => Promise.resolve({ - ...listed, - content: 'Original body.', - })) - const provider: SkillProvider = { - name: 'stable-provider', - list: originalList, - get: originalGet, - } - const added: SkillProvider[] = [] - const removed: string[] = [] - ctx.on('skill/provider-added', (registered) => { added.push(registered) }) - ctx.on('skill/provider-removed', (name) => { removed.push(name) }) - const owner = await ctx.plugin({ - name: 'mutable-provider-owner', - inject: ['skills'], - apply(pluginCtx: Context) { - pluginCtx.skills.registerProvider(provider) - }, - }) - - provider.name = 'mutated-provider' - const replacementList = vi.fn(() => Promise.resolve([])) - const replacementGet = vi.fn(() => Promise.resolve(undefined)) - provider.list = replacementList - provider.get = replacementGet - - expect(added).toHaveLength(1) - expect(added[0]).not.toBe(provider) - expect(added[0]?.name).toBe('stable-provider') - expect(Object.isFrozen(added[0])).toBe(true) - expect((await ctx.skills.list()).map(skill => skill.name)).toEqual(['stable-skill']) - expect((await ctx.skills.get('stable-skill'))?.content).toBe('Original body.') - expect(originalList).toHaveBeenCalledOnce() - expect(originalGet).toHaveBeenCalledOnce() - expect(replacementList).not.toHaveBeenCalled() - expect(replacementGet).not.toHaveBeenCalled() - - await owner.dispose() - expect(removed).toEqual(['stable-provider']) - expect(await ctx.skills.list()).toEqual([]) - const replacement = new MemoryProvider([]) - Object.defineProperty(replacement, 'name', { value: 'stable-provider' }) - expect(() => ctx.skills.registerProvider(replacement)).not.toThrow() - }) - - it('rejects malformed provider and candidate scalar fields without freezing caller objects', async () => { - const ctx = new Context() - await ctx.plugin(SkillService) - const badProviderName = { value: 'object-provider' } - expect(() => ctx.skills.registerProvider({ - name: badProviderName as unknown as string, - list: () => Promise.resolve([]), - get: () => Promise.resolve(undefined), - })).toThrow('skill provider name must be a string') - expect(Object.isFrozen(badProviderName)).toBe(false) - expect(() => ctx.skills.registerProvider({ - name: 'bad-list', - list: { bind() {} } as unknown as SkillProvider['list'], - get: () => Promise.resolve(undefined), - })).toThrow('list must be a function') - expect(() => ctx.skills.registerProvider({ - name: 'bad-get', - list: () => Promise.resolve([]), - get: { bind() {} } as unknown as SkillProvider['get'], - })).toThrow('get must be a function') - const badDescription = { value: 'object-description' } ctx.skills.registerProvider({ name: 'bad-candidate', @@ -198,7 +122,6 @@ describe('SkillService registry', () => { get: () => Promise.resolve(undefined), }) await expect(ctx.skills.list()).rejects.toThrow('non-string description') - expect(Object.isFrozen(badDescription)).toBe(false) const badBoolean = new Context() await badBoolean.plugin(SkillService) @@ -258,46 +181,37 @@ describe('SkillService registry', () => { } }) - it('snapshots lookup options before asynchronous discovery and loading', async () => { + it('borrows the exact lookup options through discovery and loading', async () => { const ctx = new Context() await ctx.plugin(SkillService) - let release: (() => void) | undefined - const gate = new Promise((resolve) => { release = resolve }) - const listCwds: (string | undefined)[] = [] - const getCwds: (string | undefined)[] = [] + const options: SkillLookupOptions = { cwd: '/workspace/a' } + let listedWith: SkillLookupOptions | undefined + let loadedWith: SkillLookupOptions | undefined + const candidate: SkillCandidate = { + name: 'skill-a', + description: 'Skill A', + provider: 'contextual', + source: 'test', + rank: 1, + locator: 'skill-a', + } ctx.skills.registerProvider({ name: 'contextual', - async list(options) { - listCwds.push(options.cwd) - await gate - const name = options.cwd === '/workspace/a' ? 'skill-a' : 'skill-b' - return [ - { name, description: name, provider: 'contextual', source: 'test', rank: 1, locator: name }, - { name: 'vanished', description: 'Vanished', provider: 'contextual', source: 'test', rank: 2, locator: 'vanished' }, - ] + async list(received) { + listedWith = received + return [candidate] }, - async get(candidate, options) { - getCwds.push(options.cwd) - if (candidate.name === 'vanished') return undefined - return { ...candidate, content: `${options.cwd}:${candidate.name}` } + async get(received, lookup) { + expect(received).toBe(candidate) + loadedWith = lookup + return { ...received, content: 'Skill A body.' } }, }) - const listOptions: { cwd: string | undefined } = { cwd: '/workspace/a' } - const pending = ctx.skills.list(listOptions) - listOptions.cwd = '/workspace/b' - release?.() - - expect((await pending).map(skill => skill.name)).toEqual(['skill-a', 'vanished']) - expect((await ctx.skills.list({ cwd: '/workspace/a' })).map(skill => skill.name)).toEqual(['skill-a', 'vanished']) - expect(listCwds).toEqual(['/workspace/a']) - - const getOptions: { cwd: string | undefined } = { cwd: '/workspace/a' } - const loading = ctx.skills.get('skill-a', getOptions) - getOptions.cwd = '/workspace/b' - expect((await loading)?.content).toBe('/workspace/a:skill-a') - expect(await ctx.skills.get('vanished', { cwd: '/workspace/a' })).toBeUndefined() - expect(getCwds).toEqual(['/workspace/a', '/workspace/a']) + expect((await ctx.skills.list(options)).map(skill => skill.name)).toEqual(['skill-a']) + expect(await ctx.skills.get('skill-a', options)).toMatchObject({ content: 'Skill A body.' }) + expect(listedWith).toBe(options) + expect(loadedWith).toBe(options) }) it('rechecks cancellation after cached discovery before provider loading', async () => { @@ -402,7 +316,7 @@ describe('SkillService registry', () => { expect(settled).toBe('aborted') }) - it('detaches cached candidates and loaded definitions while preserving locator identity', async () => { + it('borrows cached candidates and loaded definitions from the provider', async () => { const ctx = new Context() await ctx.plugin(SkillService) const locator = { id: 'provider-owned' } @@ -445,50 +359,27 @@ describe('SkillService registry', () => { }, }) - const first = await ctx.skills.list() - candidate.name = 'Bad_Name' - candidate.description = '' - if (candidate.resourceBase?.kind === 'opaque') candidate.resourceBase.description = 'mutated candidate' - if (candidate.metadata) candidate.metadata.owner = 'mutated candidate' - if (first[0]?.resourceBase?.kind === 'opaque') first[0].resourceBase.description = 'mutated summary' - - const second = await ctx.skills.list() - expect(second).toEqual([expect.objectContaining({ + const listed = await ctx.skills.list() + expect(listed).toEqual([expect.objectContaining({ name: 'stable-skill', description: 'Stable description', resourceBase: { kind: 'opaque', description: 'candidate resources' }, })]) + expect(listed[0]?.resourceBase).toBe(candidate.resourceBase) expect(listCalls).toBe(1) const loaded = await ctx.skills.get('stable-skill') - expect(received).not.toBe(candidate) + expect(received).toBe(candidate) expect(received?.locator).toBe(locator) - expect(received).toMatchObject({ - name: 'stable-skill', - description: 'Stable description', - resourceBase: { kind: 'opaque', description: 'candidate resources' }, - metadata: { owner: 'candidate' }, - }) - expect(loaded).not.toBe(definition) - if (loaded?.resourceBase?.kind === 'opaque') loaded.resourceBase.description = 'mutated definition output' - if (loaded?.metadata) loaded.metadata.owner = 'mutated definition output' - - expect(await ctx.skills.get('stable-skill')).toMatchObject({ - resourceBase: { kind: 'opaque', description: 'definition resources' }, - metadata: { owner: 'definition' }, - }) - expect(definition).toMatchObject({ - resourceBase: { kind: 'opaque', description: 'definition resources' }, - metadata: { owner: 'definition' }, - }) + expect(loaded).toBe(definition) }) - it('detaches runtime registrations and every public resource view', async () => { + it('preserves readonly runtime resource identities while adding the default provider', async () => { const ctx = new Context() await ctx.plugin(SkillService) const resourceBase = { kind: 'opaque' as const, description: 'runtime resources' } const metadata = { owner: 'runtime' } - ctx.skills.register({ + const registration = { name: 'runtime-skill', description: 'Runtime', whenToUse: 'When runtime data is needed.', @@ -497,101 +388,20 @@ describe('SkillService registry', () => { resourceBase, metadata, content: 'Runtime body.', - }) + } + ctx.skills.register(registration) ctx.skills.register({ name: 'z-runtime', description: 'Second runtime skill', source: 'runtime', content: 'Second runtime body.', }) - resourceBase.description = 'mutated registration' - metadata.owner = 'mutated registration' - const listed = await ctx.skills.list() const loaded = await ctx.skills.get('runtime-skill') - expect(listed[0]?.resourceBase).toEqual({ kind: 'opaque', description: 'runtime resources' }) - expect(loaded?.metadata).toEqual({ owner: 'runtime' }) - if (listed[0]?.resourceBase?.kind === 'opaque') listed[0].resourceBase.description = 'mutated list output' - if (loaded?.resourceBase?.kind === 'opaque') loaded.resourceBase.description = 'mutated get output' - if (loaded?.metadata) loaded.metadata.owner = 'mutated get output' - - expect((await ctx.skills.list())[0]?.resourceBase).toEqual({ kind: 'opaque', description: 'runtime resources' }) - expect(await ctx.skills.get('runtime-skill')).toMatchObject({ - resourceBase: { kind: 'opaque', description: 'runtime resources' }, - metadata: { owner: 'runtime' }, - }) - }) - - it('rejects malformed runtime and loaded-definition scalar fields without freezing them', async () => { - const ctx = new Context() - await ctx.plugin(SkillService) - const runtimeDescription = { value: 'runtime-description' } - expect(() => ctx.skills.register({ - name: 'bad-runtime', - description: runtimeDescription as unknown as string, - source: 'runtime', - content: 'body', - })).toThrow('description must be a string') - expect(Object.isFrozen(runtimeDescription)).toBe(false) - expect(() => ctx.skills.register({ - name: 'bad-runtime-boolean', - description: 'Runtime', - disableModelInvocation: 'false' as unknown as boolean, - source: 'runtime', - content: 'body', - })).toThrow('disableModelInvocation must be a boolean') - expect(() => ctx.skills.register({ - name: 'bad-runtime-provider', - description: 'Runtime', - source: 'runtime', - provider: null as unknown as string, - content: 'body', - })).toThrow('provider must be a string') - - const loadedContent = { value: 'loaded-content' } - ctx.skills.registerProvider({ - name: 'bad-definition', - list: () => Promise.resolve([{ - name: 'bad-definition', - description: 'Candidate', - provider: 'bad-definition', - source: 'test', - rank: 1, - locator: 'bad-definition', - }]), - get: candidate => Promise.resolve({ - ...candidate, - content: loadedContent as unknown as string, - }), - }) - await expect(ctx.skills.get('bad-definition')).rejects.toThrow('content must be a string') - expect(Object.isFrozen(loadedContent)).toBe(false) - }) - - it('rejects every other malformed runtime scalar', async () => { - const ctx = new Context() - await ctx.plugin(SkillService) - type Registration = Parameters[0] - const valid: Registration = { - name: 'runtime-validation', - description: 'Runtime validation', - whenToUse: 'Use this runtime skill.', - disableModelInvocation: false, - source: 'runtime', - provider: 'runtime-validation', - content: 'Runtime body.', - path: '/skills/runtime-validation/SKILL.md', - } - const cases: { patch: Partial; expected: string }[] = [ - { patch: { name: { value: 'runtime' } as unknown as string }, expected: 'runtime skill name must be a string' }, - { patch: { whenToUse: 1 as unknown as string }, expected: 'whenToUse must be a string' }, - { patch: { source: { value: 'source' } as unknown as string }, expected: 'source must be a string' }, - { patch: { content: { value: 'content' } as unknown as string }, expected: 'content must be a string' }, - { patch: { path: 1 as unknown as string }, expected: 'path must be a string' }, - ] - for (const { patch, expected } of cases) { - expect(() => ctx.skills.register({ ...valid, ...patch })).toThrow(expected) - } + expect(listed[0]?.resourceBase).toBe(resourceBase) + expect(loaded?.resourceBase).toBe(resourceBase) + expect(loaded?.metadata).toBe(metadata) + expect(loaded?.provider).toBe('runtime') }) it('rejects every malformed scalar in provider-loaded definitions', async () => { @@ -853,39 +663,6 @@ describe('SkillService registry', () => { expect(settled).toBe('aborted') }) - it('does not miss an abort racing listener installation', async () => { - const ctx = new Context() - await ctx.plugin(SkillService) - const reason = new Error('racing abort') - let aborted = false - const signal = { - get aborted() { - return aborted - }, - reason, - throwIfAborted() { - if (aborted) throw reason - }, - addEventListener(_type: string, listener: () => void) { - aborted = true - listener() - }, - removeEventListener() {}, - } as unknown as AbortSignal - ctx.skills.registerProvider({ - name: 'racing-abort', - list() { - return Promise.reject(new Error('late provider failure')) - }, - async get() { - return undefined - }, - }) - - await expect(ctx.skills.list({ signal })).rejects.toBe(reason) - await Promise.resolve() - }) - it('rejects invalid runtime skill registrations and ignores duplicates', async () => { const ctx = new Context() await ctx.plugin(SkillService) diff --git a/packages/subagent/subagent-acp/README.md b/packages/subagent/subagent-acp/README.md index cb51986915..c949685fcc 100644 --- a/packages/subagent/subagent-acp/README.md +++ b/packages/subagent/subagent-acp/README.md @@ -1,32 +1,31 @@ # @deepseek-ai/dsh-subagent-acp -The out-of-process **ACP subagent backend**: runs each child agent in a spawned subprocess, driven over the [Agent Client Protocol](https://github.com/zed-industries/agent-client-protocol) (ACP) as the *client*. Registers a `SubagentProvider` on `ctx.subagents` (the [subagent seam](../subagent)), alongside the in-process [`-spawn`](../subagent-spawn)/[`-fork`](../subagent-fork) backends — multiple backends coexist by name. +The ACP provider runs each subagent in a fresh subprocess and drives it as an Agent Client Protocol client. It is the out-of-process alternative to spawn and fork: the child has its own runtime, session, model configuration, and tools. -It is the direction-inverted twin of the server-side bridge in [`@deepseek-ai/dsh-acp`](../../ui/acp): that package is the ACP *agent* (it answers `initialize`/`newSession`/`prompt`); this one is the ACP *client* (it *calls* them and implements the `sessionUpdate`/`requestPermission` callbacks). Point the configured command at the `acp-agent` example to "talk to our own process". +## Start and ownership -## What it does +`start(request)` performs `spawn` → ACP `initialize` → `newSession` before it fulfills. Fulfillment therefore means a remote session is ready and ownership has transferred to the caller. A spawn, initialization, new-session, or pre-publication cancellation failure rejects only after the subprocess has been reaped. -`start(request)` spawns the configured command, wraps its stdio in an ACP `ClientSideConnection`, and drives one session: `initialize` → `newSession` → `prompt`. `run.started` resolves after `newSession` publishes the remote session and rejects when initialization fails or cancellation wins first; the service emits no start/end pair for a child that never became live. The child's streamed `agent_message_chunk` text becomes the `SubagentResult.output`; the prompt's terminal `StopReason` maps to the stop reason. `dispose()` kills the subprocess and awaits its exit. +After publication, the provider sends the prompt and collects streamed `agent_message_chunk` text into `SubagentResult.output`. A prompt/transport failure resolves with `stopReason: 'error'`, or `aborted` when the required request signal or disposal requested cancellation. -**Fresh process per run.** Each `start` spawns a new child, runs exactly one ACP session, and disposes it. Persistent-process pooling is a future optimization (see the RFC). +`dispose()` is idempotent. It removes the signal listener, requests ACP cancellation when possible, closes stdin, waits `disposeEofGraceMs`, escalates to SIGTERM, waits `disposeGraceMs`, and finally uses SIGKILL if necessary. Every run uses a fresh process; process pooling is not implemented. -Unlike the in-process backends, the child does NOT share this cordis context — it is a separate process with its own session, model client, and tools. So this backend: -- injects only `subagents` (no `ctx.agents`); -- advertises NO start-time capabilities (an out-of-process child can't enforce the parent's depth/tool-filter); -- ignores `request.parent`. +## Capabilities and context -## Config +ACP advertises no start-time capabilities because this process cannot enforce the remote child's depth, tool filter, persona, or structured-output runtime. It also reports `inheritsParentContext: false`: the remote session starts fresh and ignores `request.parent` beyond the seam's required attribution field. -| Key | Type | Default | Notes | -|---|---|---|---| -| `providerName` | string | `acp` | Registry name on `ctx.subagents`. | -| `command` | string | — (required) | The executable to spawn for each run (the child ACP agent). | -| `args` | string[] | `[]` | Arguments passed to `command`. | -| `cwd` | string | process cwd | Working directory for the child process and its ACP session. | -| `permission` | `'allow' \| 'reject'` | `reject` | How to auto-answer the child's `session/request_permission` prompts. `reject` declines every prompt (answer `cancelled`); `allow` approves via the first allow-shaped option. The first cut surfaces no prompt to a human. | -| `env` | Record | `{}` | Extra env vars for the child (e.g. its own `DEEPSEEK_API_KEY`). Forwarded on top of a credential-scrubbed copy of the parent env, so an explicit key reaches the child while ambient secrets do not leak implicitly. | -| `disposeEofGraceMs` | number | `6000` | Dispose ladder tier 1: how long the child gets to quiesce on its own after stdin EOF (flush persistence, tear down its nested subprocesses) before SIGTERM. | -| `disposeGraceMs` | number | `3000` | Dispose ladder tier 2: grace between SIGTERM and the SIGKILL escalation. | +## Configuration + +| Key | Default | Meaning | +|---|---|---| +| `providerName` | `acp` | Registry name on `ctx.subagents`. | +| `command` | required | Executable spawned for each run. | +| `args` | `[]` | Command arguments. | +| `cwd` | process cwd | Child process and ACP session working directory. | +| `permission` | `reject` | Auto-answer permission requests by rejecting or choosing the first allow-shaped option. | +| `env` | `{}` | Explicit child environment layered over a credential-scrubbed parent environment. | +| `disposeEofGraceMs` | `6000` | Grace after stdin EOF before SIGTERM. | +| `disposeGraceMs` | `3000` | Grace after SIGTERM before SIGKILL. | ```yaml - id: subagent-acp @@ -40,32 +39,20 @@ Unlike the in-process backends, the child does NOT share this cordis context — DEEPSEEK_API_KEY: !!js process.env.DEEPSEEK_API_KEY ``` -## StopReason mapping +## Stop-reason mapping -ACP `StopReason` → harness `SubagentStopReason`: - -| ACP | harness | +| ACP | Harness | |---|---| | `end_turn` | `completed` | | `max_tokens` | `max-tokens` | | `refusal` | `refusal` | | `cancelled` | `aborted` | -| `max_turn_requests` | `error` (no clean equivalent; the task did not finish) | -| _(unknown)_ | `error` | +| `max_turn_requests` or unknown | `error` | -A spawn/transport/RPC failure resolves `error` (or `aborted` if a cancel was requested) — `result` never rejects on a child-level failure, per the seam contract. +## Process boundary -## Environment scrub +The child environment is built by [`buildChildEnv`](../subagent-subprocess/README.md): credential-shaped ambient variables are removed, then explicit `config.env` values are applied. The ACP wire is the real serialization boundary; same-process subagent values are not defensively cloned. -The child env is built by [`buildChildEnv` from `@deepseek-ai/dsh-subagent-subprocess`](../subagent-subprocess/README.md) — the ambient env minus credential-shaped vars, with `config.env` layered on top after the scrub; the pattern and full semantics live there. For this backend that means the parent harness's own secrets never leak into the spawned agent implicitly, while the child's OWN `DEEPSEEK_API_KEY` is supplied deliberately via `config.env` and survives. +The package has no default export. Cordis loader unwrapping would otherwise hide the named `inject` metadata; see [postmortem 0001](../../../docs/postmortem/0001-acp-default-export-drops-inject.md). -## Testing - -- **Keyless** (`subagent-acp.spec.ts`): spawns a scripted mock ACP server subprocess (`tests/mock-acp-server.ts`) and drives it through the real backend over real ACP stdio — connection setup, client callbacks, the prompt round-trip, stop-reason mapping, cancellation (including the early-cancel race and a torn-pipe-after-cancel), permission auto-answer, and quiescent disposal. No model, no key. -- **With-key e2e** (`subagent-acp.e2e.ts`): the harness drives ITSELF — the backend spawns the real `acp-agent` example process and a real model in that child answers a prompt and does real file work (verified on disk). Self-skips without `DEEPSEEK_API_KEY`. - -`TODO(acp-subagent-replay)`: snapshot-tier coverage of an ACP child is a separate replay shape (each child is its own PROCESS with its own single-agent replay, distinct from the in-process per-session keying), deferred — see the RFC. - -## Plugin export shape - -Named `name` / `inject` / `Config` / `apply`, with **no default export**: the cordis Loader's `unwrapExports` does `exports.default ?? exports`, so a stray default would collapse the module to the bare function and drop the `inject` namespace (see [docs/postmortem/0001](../../../docs/postmortem/0001-acp-default-export-drops-inject.md)). +Keyless tests drive a scripted ACP subprocess over real stdio. The with-key e2e drives the repository's real ACP agent and self-skips without `DEEPSEEK_API_KEY`. diff --git a/packages/subagent/subagent-acp/src/run.ts b/packages/subagent/subagent-acp/src/run.ts index 410f7ad7bf..5222906a2d 100644 --- a/packages/subagent/subagent-acp/src/run.ts +++ b/packages/subagent/subagent-acp/src/run.ts @@ -180,35 +180,19 @@ function toError(value: unknown): Error { * and drives one session: `initialize` → `newSession` → `prompt`. The accumulated * `agent_message_chunk` text is the result output; the prompt's terminal * `StopReason` maps to the stop reason. `result` never REJECTS on a child-level - * failure (a spawn/transport/RPC error resolves with `stopReason: 'error'`), per - * the seam contract. `cancel()` sends `session/cancel`; `dispose()` kills the - * subprocess and awaits its exit (quiescent teardown). - * @param request - the start request; the driver consumes `prompt` and `signal` - * (an already-aborted signal yields an inert `aborted` run with no spawn). + * failure after publication resolves with `stopReason: 'error'`. A spawn, + * initialize, new-session, or pre-publication cancellation failure instead + * rejects only after the process has been reaped. `dispose()` requests ACP + * cancellation, then kills and reaps the subprocess. + * @param request - the start request; its signal is the cancellation channel. * @param spec - the resolved spawn spec: command/args/cwd, env, permission * policy, dispose graces, and the optional error sink. - * @returns the live run handle for the child subprocess. + * @returns the ready run handle for the child subprocess. */ -export function startAcpRun(request: SubagentStartRequest, spec: AcpRunSpec): SubagentRun { +export async function startAcpRun(request: SubagentStartRequest, spec: AcpRunSpec): Promise { const id = AgentId(randomUUID()) - // A request already aborted before it starts never spawns the child at all — - // return an inert run that settled `aborted`, rather than launching the - // configured binary just to tear it down. `dispose`/`cancel` are no-ops. - if (request.signal?.aborted) { - const started = Promise.reject(new Error('subagent request was aborted before the ACP child started')) - // The result is derived from the same boundary so the readiness rejection - // is observed even when this provider is driven directly rather than - // through SubagentService. - const result: Promise = started.catch(() => ({ output: [], stopReason: 'aborted' })) - return { - id, - started, - result, - cancel(_reason?: string): void { /* nothing was started */ }, - dispose(): Promise { return Promise.resolve() }, - } - } + if (request.signal.aborted) throw new Error('subagent request was aborted before the ACP child started') // Spawn the child ACP agent. stdin = ACP request channel, stdout = ACP // response channel, stderr = INHERIT so the child's diagnostics surface on the @@ -225,9 +209,18 @@ export function startAcpRun(request: SubagentStartRequest, spec: AcpRunSpec): Su // `error` like any child failure. const spawnFailed = spawnFailure(child) + // One memoized quiescence transaction is shared by startup rollback and the + // published run's disposer. Once start fulfills, only the holder can invoke + // it; before fulfillment the provider invokes it on every failure path. + let processDisposal: Promise | undefined + const disposeProcess = (): Promise => (processDisposal ??= disposeChildProcess(child, { + disposeEofGraceMs: spec.disposeEofGraceMs, + disposeGraceMs: spec.disposeGraceMs, + })) + // Accumulate the child's streamed assistant text — the SubagentResult output. const output: string[] = [] - // `cancelled` records that a cancel was requested (signal or cancel()), so a + // `cancelled` records that the required signal or disposal requested cancel, so a // run torn down before the prompt resolves settles `aborted` rather than the // generic error mapping. Held on a mutable object so the async closures that // set it (the abort listener) and the IIFE that reads it don't fight TS's @@ -271,7 +264,7 @@ export function startAcpRun(request: SubagentStartRequest, spec: AcpRunSpec): Su // Resolves when a cancel is requested, so `result` can settle `aborted` even // if the child never cooperates with `session/cancel` (it ignores the notify, // or the prompt wedges). The result path races this against the ACP drive: the - // FIRST to settle wins, so `cancel()` always honors the contract (`result` + // FIRST to settle wins, so signal/dispose cancellation always honors the contract (`result` // settles `aborted`) without waiting on a non-cooperative child. `dispose` // still kills the process and reaps it; this only unblocks `result`. The // executor runs synchronously, so `signalCancelSettled` is assigned before the @@ -279,6 +272,7 @@ export function startAcpRun(request: SubagentStartRequest, spec: AcpRunSpec): Su let signalCancelSettled!: () => void const cancelSettled = new Promise((resolve) => { signalCancelSettled = resolve }) const requestCancel = (): void => { + if (flags.cancelled) return flags.cancelled = true signalCancelSettled() // Best-effort: tell the child to cancel the in-flight turn. Swallows a @@ -293,7 +287,7 @@ export function startAcpRun(request: SubagentStartRequest, spec: AcpRunSpec): Su if (sessionId !== undefined) void conn.cancel({ sessionId }).catch(() => { /* child gone / no session */ }) } const onAbort = (): void => { requestCancel() } - request.signal?.addEventListener('abort', onAbort, { once: true }) + request.signal.addEventListener('abort', onAbort, { once: true }) // The accumulated child text as harness ContentBlocks (empty array when the // child streamed nothing). Read at every return so a partial answer survives @@ -303,47 +297,41 @@ export function startAcpRun(request: SubagentStartRequest, spec: AcpRunSpec): Su return text.length > 0 ? [{ type: 'text', text }] : [] } - // A provider is "started" only once the remote child has completed ACP - // initialization and published a session. SubagentService gates its - // `subagent/start` notification on this boundary, just as the in-process - // provider gates it on local Agent publication. Failure or cancellation - // before this point rejects readiness and therefore produces no paired - // lifecycle events for a child that never became live. - const started: Promise = Promise.race([ - (async (): Promise => { - await conn.initialize({ - protocolVersion: PROTOCOL_VERSION, - // Advertise NO optional client capabilities (no fs, no terminal): the - // child self-serves in its own process. - clientCapabilities: {}, - }) - const session = await conn.newSession({ cwd: spec.cwd, mcpServers: [] }) - sessionId = session.sessionId - if (flags.cancelled) throw new Error('subagent cancelled before the ACP session started') - })(), - spawnFailed.then((err): never => { throw err }), - cancelSettled.then((): never => { throw new Error('subagent cancelled before the ACP session started') }), - ]) + // Establish the remote session before publishing a handle. Any failure owns + // the still-private process and therefore reaps it before rejecting. + try { + await Promise.race([ + (async (): Promise => { + await conn.initialize({ + protocolVersion: PROTOCOL_VERSION, + // Advertise NO optional client capabilities (no fs, no terminal): the + // child self-serves in its own process. + clientCapabilities: {}, + }) + const session = await conn.newSession({ cwd: spec.cwd, mcpServers: [] }) + sessionId = session.sessionId + if (flags.cancelled) throw new Error('subagent cancelled before the ACP session started') + })(), + spawnFailed.then((err): never => { throw err }), + cancelSettled.then((): never => { throw new Error('subagent cancelled before the ACP session started') }), + ]) + } catch (error: unknown) { + request.signal.removeEventListener('abort', onAbort) + await disposeProcess() + if (flags.cancelled) throw new Error('subagent request was aborted before the ACP child started') + throw toError(error) + } const result: Promise = (async (): Promise => { try { - // Readiness is the initialize → newSession phase above. Awaiting the SAME - // promise immediately observes its rejection even without the service, - // and guarantees the prompt phase never starts before the provider can - // truthfully announce a live child. - await started - - // Race two post-start outcomes, first to settle wins: + // Race two post-publication outcomes, first to settle wins: // - prompt: the normal remote turn; // - cancelSettled: a cancel was requested — settle `aborted` immediately // rather than waiting on a child that may ignore `session/cancel` or - // wedge the prompt (the `cancel()` contract: `result` settles `aborted`). - // A spawn error can only precede readiness and is already one arm of - // `started`; after `newSession` succeeds, transport/process failure rejects - // the in-flight prompt RPC through the connection. + // wedge the prompt (`result` settles `aborted`). After `newSession` + // succeeds, transport/process failure rejects the in-flight prompt RPC. const prompt = async (): Promise => { - // `started` cannot fulfill without assigning the session id; the cast - // records that local invariant without an unreachable defensive arm. + // The startup phase cannot fulfill without assigning the session id. const promptResult = await conn.prompt({ sessionId: sessionId as string, prompt: toAcpPrompt(request.prompt) }) return { output: collectOutput(), stopReason: acpStopReason(promptResult.stopReason) } } @@ -352,11 +340,15 @@ export function startAcpRun(request: SubagentStartRequest, spec: AcpRunSpec): Su cancelSettled.then((): SubagentResult => ({ output: collectOutput(), stopReason: 'aborted' })), ]) } catch (error: unknown) { + // A deterministic cancellation resolves `cancelSettled` before its + // best-effort ACP cancel can reject the prompt. This fallback is only for + // a process/pipe rejection already queued when the abort event fires; its + // first-outcome ordering cannot be forced without a timing-dependent test. + /* v8 ignore next */ if (flags.cancelled) return { output: collectOutput(), stopReason: 'aborted' } // The seam contract: result resolves (never rejects) on a child-level - // failure. A cancellation is recognized by the flag above even when it - // wins during readiness; every other rejection is a genuine child-level - // error — initialize/newSession/prompt transport/RPC failure or ENOENT. + // failure. Startup failures were already rejected before publication; + // every rejection here is a prompt transport/RPC failure. // Flatten to `error` and surface the original via onError so a real fault // is preserved rather than silently lost. try { @@ -367,18 +359,19 @@ export function startAcpRun(request: SubagentStartRequest, spec: AcpRunSpec): Su // The child-level failure being reported still settles as `error`. } return { output: collectOutput(), stopReason: 'error' } + } finally { + request.signal.removeEventListener('abort', onAbort) } })() + let disposal: Promise | undefined return { id, - started, result, - cancel(_reason?: string): void { + dispose(): Promise { + if (disposal !== undefined) return disposal + request.signal.removeEventListener('abort', onAbort) requestCancel() - }, - async dispose(): Promise { - request.signal?.removeEventListener('abort', onAbort) // Quiescent teardown via the shared ladder (stdin EOF → SIGTERM → // SIGKILL, awaiting the actual exit). For THIS child the EOF tier is the // one that matters: our acp-agent has NO SIGTERM handler in a normal @@ -388,10 +381,8 @@ export function startAcpRun(request: SubagentStartRequest, spec: AcpRunSpec): Su // turn/end BEFORE that post-turn flush lands, so the child still has // durable work owed when dispose runs (hence the wide EOF grace; see // DEFAULT_DISPOSE_EOF_GRACE_MS). - await disposeChildProcess(child, { - disposeEofGraceMs: spec.disposeEofGraceMs, - disposeGraceMs: spec.disposeGraceMs, - }) + disposal = disposeProcess() + return disposal }, } } diff --git a/packages/subagent/subagent-acp/tests/mock-acp-server.ts b/packages/subagent/subagent-acp/tests/mock-acp-server.ts index 9cfeac1f44..fb200f3505 100644 --- a/packages/subagent/subagent-acp/tests/mock-acp-server.ts +++ b/packages/subagent/subagent-acp/tests/mock-acp-server.ts @@ -68,6 +68,7 @@ const WANT_PERMISSION = process.env.MOCK_PERMISSION === '1' const NO_ALLOW = process.env.MOCK_NO_ALLOW === '1' const THOUGHT = process.env.MOCK_THOUGHT === '1' const CRASH_ON_CANCEL = process.env.MOCK_CRASH_ON_CANCEL === '1' +const CRASH_ON_PROMPT = process.env.MOCK_CRASH_ON_PROMPT === '1' const IGNORE_CANCEL = process.env.MOCK_IGNORE_CANCEL === '1' const READY_FILE = process.env.MOCK_READY_FILE const FLUSH_ON_EOF = process.env.MOCK_FLUSH_ON_EOF @@ -105,6 +106,7 @@ function makeAgent(conn: AgentSideConnection): Agent { return Promise.resolve() }, async prompt(params: PromptRequest): Promise { + if (CRASH_ON_PROMPT) process.exit(1) if (WANT_PERMISSION) { // Ask the client to approve before answering; honor its decision. Under // MOCK_NO_ALLOW the only options are reject-shaped, so an `allow`-policy @@ -225,4 +227,3 @@ if (process.env.MOCK_IGNORE_EOF === '1') { setInterval(() => { /* stay alive past EOF until SIGTERM */ }, 1000) if (READY_FILE !== undefined) writeFileSync(READY_FILE, 'ignore-eof-armed') } - diff --git a/packages/subagent/subagent-acp/tests/subagent-acp.e2e.ts b/packages/subagent/subagent-acp/tests/subagent-acp.e2e.ts index 826ef198dd..7814dfaca3 100644 --- a/packages/subagent/subagent-acp/tests/subagent-acp.e2e.ts +++ b/packages/subagent/subagent-acp/tests/subagent-acp.e2e.ts @@ -60,9 +60,10 @@ describe.skipIf(!process.env.DEEPSEEK_API_KEY)('ACP backend with-key e2e (drive }, }) - const run = ctx.subagents.start('acp', { + const run = await ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'Reply with exactly the word PONG and nothing else. Do not use any tools.' }], parent: fakeParent, + signal: new AbortController().signal, }) const result = await run.result await run.dispose() @@ -93,11 +94,12 @@ describe.skipIf(!process.env.DEEPSEEK_API_KEY)('ACP backend with-key e2e (drive }, }) - const run = ctx.subagents.start('acp', { + const run = await ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'Use the bash tool to write the text ACP_CHILD_WAS_HERE into a file named proof.txt ' + 'in the current directory. Then reply DONE.' }], parent: fakeParent, + signal: new AbortController().signal, }) const result = await run.result await run.dispose() diff --git a/packages/subagent/subagent-acp/tests/subagent-acp.spec.ts b/packages/subagent/subagent-acp/tests/subagent-acp.spec.ts index e25d194c98..eed3cfe1ed 100644 --- a/packages/subagent/subagent-acp/tests/subagent-acp.spec.ts +++ b/packages/subagent/subagent-acp/tests/subagent-acp.spec.ts @@ -27,6 +27,10 @@ const repoTsconfig = fileURLToPath(new URL('../../../../tsconfig.json', import.m /** A throwaway parent Agent — the ACP backend ignores it, but the seam requires one. */ const fakeParent = { id: 'parent', session: { header: {} } } as unknown as Agent +function request(text = 'p', signal = new AbortController().signal) { + return { prompt: [{ type: 'text' as const, text }], parent: fakeParent, signal } +} + interface SetupEnv { /** Mock-server scripting env: MOCK_TEXT / MOCK_STOP / MOCK_HANG / MOCK_PERMISSION. */ [key: string]: string @@ -119,16 +123,18 @@ describe('buildChildEnv', () => { describe('dsh-subagent-acp', () => { it('drives a child process to completion and returns its streamed output', async () => { const ctx = await setup({ MOCK_TEXT: 'hello from acp child', MOCK_STOP: 'end_turn' }) - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'do X' }], parent: fakeParent }) + const run = await ctx.subagents.start('acp', request('do X')) const result = await run.result expect(result.stopReason).toBe('completed') expect(text(result.output)).toBe('hello from acp child') - await run.dispose() + const disposal = run.dispose() + expect(run.dispose()).toBe(disposal) + await disposal }) it('maps a max_tokens stop reason', async () => { const ctx = await setup({ MOCK_TEXT: 'cut off', MOCK_STOP: 'max_tokens' }) - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) + const run = await ctx.subagents.start('acp', request()) const result = await run.result expect(result.stopReason).toBe('max-tokens') await run.dispose() @@ -136,22 +142,23 @@ describe('dsh-subagent-acp', () => { it('maps a refusal stop reason', async () => { const ctx = await setup({ MOCK_TEXT: '', MOCK_STOP: 'refusal' }) - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) + const run = await ctx.subagents.start('acp', request()) const result = await run.result expect(result.stopReason).toBe('refusal') await run.dispose() }) - it('cancelling a running child settles aborted (session/cancel via run.cancel)', async () => { + it('aborting the required signal cancels a running child', async () => { const tmp = mkdtempSync(join(tmpdir(), 'acp-cancel-')) const readyFile = join(tmp, 'ready') try { const ctx = await setup({ MOCK_TEXT: 'partial', MOCK_HANG: '1', MOCK_READY_FILE: readyFile }) - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) + const controller = new AbortController() + const run = await ctx.subagents.start('acp', request('p', controller.signal)) // Wait until the child's prompt is in flight (condition, not a sleep), // then cancel — so we exercise the mid-run session/cancel path. await waitForFile(readyFile) - run.cancel('test') + controller.abort('test') const result = await run.result expect(result.stopReason).toBe('aborted') await run.dispose() @@ -160,7 +167,7 @@ describe('dsh-subagent-acp', () => { } }) - it('settles aborted WITHOUT spawning the child when the signal is already aborted', async () => { + it('rejects WITHOUT spawning the child when the signal is already aborted', async () => { // A pre-aborted request must not even launch the configured binary. Point // the command at one that would create a sentinel file if it ever ran, and // assert the sentinel never appears. @@ -169,17 +176,11 @@ describe('dsh-subagent-acp', () => { try { const controller = new AbortController() controller.abort() - const run = startAcpRun( - { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent, signal: controller.signal }, + await expect(startAcpRun( + request('p', controller.signal), // `touch ` — runs only if the process is actually spawned. { command: 'touch', args: [sentinel], cwd: tmp, permission: 'reject', env: {}, disposeEofGraceMs: DEFAULT_DISPOSE_EOF_GRACE_MS, disposeGraceMs: DEFAULT_DISPOSE_GRACE_MS }, - ) - const result = await run.result - expect(result.stopReason).toBe('aborted') - expect(result.output).toEqual([]) - // cancel/dispose on the inert run are safe no-ops. - run.cancel('noop') - await run.dispose() + )).rejects.toThrow('aborted before the ACP child started') // The binary was never launched — no sentinel. expect(existsSync(sentinel)).toBe(false) } finally { @@ -206,7 +207,7 @@ describe('dsh-subagent-acp', () => { disposeEofGraceMs: 150, disposeGraceMs: 150, } - const run = startAcpRun({ prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }, spec) + const run = await startAcpRun(request(), spec) // Wait until the child has BOOTED AND ARMED THE TRAP (a condition, not a // sleep) — otherwise SIGTERM races the trap install and the default handler // terminates the child, never exercising the escalation. @@ -253,7 +254,7 @@ describe('dsh-subagent-acp', () => { disposeEofGraceMs: 2000, disposeGraceMs: 50, } - const run = startAcpRun({ prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }, spec) + const run = await startAcpRun(request(), spec) // Wait until the child is fully booted with its prompt in flight (its ACP // stdin reader is attached), so dispose's stdin EOF reaches a live child. await waitForFile(ready) @@ -290,7 +291,7 @@ describe('dsh-subagent-acp', () => { disposeEofGraceMs: 150, disposeGraceMs: 2000, } - const run = startAcpRun({ prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }, spec) + const run = await startAcpRun(request(), spec) await waitForFile(ready) // Bound it so a hang fails loud rather than stalling the suite. await expect(Promise.race([ @@ -305,7 +306,7 @@ describe('dsh-subagent-acp', () => { } }) - it('honors a cancel that races AHEAD of newSession (no session id yet) without running the prompt', async () => { + it('rejects after cleanup when the signal aborts during newSession', async () => { // Gate the child at newSession: it signals `ready` and blocks until `go`. // We cancel WHILE newSession is pending (sessionId still undefined, so the // backend cannot send session/cancel) — the `cancelled` flag alone must @@ -315,14 +316,12 @@ describe('dsh-subagent-acp', () => { const go = join(tmp, 'go') try { const ctx = await setup({ MOCK_NEWSESSION_READY: ready, MOCK_NEWSESSION_GO: go, MOCK_TEXT: 'should not run' }) - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) + const controller = new AbortController() + const starting = ctx.subagents.start('acp', request('p', controller.signal)) await waitForFile(ready) // newSession is now in flight, sessionId undefined - run.cancel('early') // sets cancelled; cannot send session/cancel yet + controller.abort('early') writeFileSync(go, 'go') // let newSession resolve - const result = await run.result - expect(result.stopReason).toBe('aborted') - expect(result.output).toEqual([]) - await run.dispose() + await expect(starting).rejects.toThrow('aborted before the ACP child started') } finally { rmSync(tmp, { recursive: true, force: true }) } @@ -334,7 +333,7 @@ describe('dsh-subagent-acp', () => { try { const controller = new AbortController() const ctx = await setup({ MOCK_TEXT: 'partial', MOCK_HANG: '1', MOCK_READY_FILE: readyFile }) - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent, signal: controller.signal }) + const run = await ctx.subagents.start('acp', request('p', controller.signal)) await waitForFile(readyFile) controller.abort() const result = await run.result @@ -347,7 +346,7 @@ describe('dsh-subagent-acp', () => { it('auto-rejects a permission prompt by default (child settles cancelled→aborted)', async () => { const ctx = await setup({ MOCK_TEXT: 'x', MOCK_PERMISSION: '1' }, 'reject') - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) + const run = await ctx.subagents.start('acp', request()) const result = await run.result // The child asked permission, the backend rejected, the child returned cancelled. expect(result.stopReason).toBe('aborted') @@ -356,7 +355,7 @@ describe('dsh-subagent-acp', () => { it('auto-approves a permission prompt under the allow policy', async () => { const ctx = await setup({ MOCK_TEXT: 'approved answer', MOCK_PERMISSION: '1', MOCK_STOP: 'end_turn' }, 'allow') - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) + const run = await ctx.subagents.start('acp', request()) const result = await run.result expect(result.stopReason).toBe('completed') expect(text(result.output)).toBe('approved answer') @@ -367,7 +366,7 @@ describe('dsh-subagent-acp', () => { // The child asks permission but offers ONLY reject-shaped options, so an // allow-policy client finds nothing to select and must answer cancelled. const ctx = await setup({ MOCK_PERMISSION: '1', MOCK_NO_ALLOW: '1' }, 'allow') - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) + const run = await ctx.subagents.start('acp', request()) const result = await run.result expect(result.stopReason).toBe('aborted') await run.dispose() @@ -377,7 +376,7 @@ describe('dsh-subagent-acp', () => { // The child streams an agent_thought_chunk before its answer; the backend // must consume it but NOT include it in the result output. const ctx = await setup({ MOCK_THOUGHT: '1', MOCK_TEXT: 'final answer', MOCK_STOP: 'end_turn' }) - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) + const run = await ctx.subagents.start('acp', request()) const result = await run.result expect(result.stopReason).toBe('completed') // Only the message text, NOT the thought. @@ -385,18 +384,11 @@ describe('dsh-subagent-acp', () => { await run.dispose() }) - it('resolves error (not reject) when the spawn command does not exist', async () => { - // Direct startAcpRun with NO onError sink — the catch must still flatten the - // spawn failure to `error` (the onError call is optional, covering the - // absent-sink branch). - const run = startAcpRun( - { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }, + it('rejects a spawn failure after provider-owned cleanup', async () => { + await expect(startAcpRun( + request(), { command: '/nonexistent/acp-agent-binary', args: [], cwd: process.cwd(), permission: 'reject', env: {}, disposeEofGraceMs: DEFAULT_DISPOSE_EOF_GRACE_MS, disposeGraceMs: DEFAULT_DISPOSE_GRACE_MS }, - ) - const result = await run.result - // The seam contract: a child-level failure resolves error, never rejects. - expect(result.stopReason).toBe('error') - await run.dispose() + )).rejects.toThrow() }) it('plugin-config dispose graces reach the run (SIGKILL escalation through the provider)', async () => { @@ -418,7 +410,7 @@ describe('dsh-subagent-acp', () => { disposeEofGraceMs: 150, disposeGraceMs: 150, }) - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) + const run = await ctx.subagents.start('acp', request()) await waitForFile(ready) await expect(Promise.race([ run.dispose(), @@ -440,7 +432,7 @@ describe('dsh-subagent-acp', () => { } }) - it('resolves error via the provider (real load path) when the command does not exist', async () => { + it('rejects a startup failure via the provider load path', async () => { const ctx = new Context() await ctx.plugin(SubagentService) await ctx.plugin(acp, { @@ -450,26 +442,23 @@ describe('dsh-subagent-acp', () => { permission: 'reject', env: {}, }) - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) - const result = await run.result - expect(result.stopReason).toBe('error') - await run.dispose() + await expect(ctx.subagents.start('acp', request())).rejects.toThrow() }) it('reports a flattened child failure through onError (preserved, not silently lost)', async () => { // The seam forbids `result` rejecting, so a child-level failure is flattened // to a stop reason — onError must still surface the original error so a real - // fault is logged, not swallowed. A nonexistent command triggers the spawn - // failure path; the spy records the error + the chosen stop reason. + // fault is logged, not swallowed. The child exits after its session is + // published but while prompt is in flight. const errors: { message: string; stopReason: string }[] = [] - const run = startAcpRun( - { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }, + const run = await startAcpRun( + request(), { - command: '/nonexistent/acp-agent-binary', - args: [], + command: process.execPath, + args: ['--import', tsxLoader, mockServer], cwd: process.cwd(), permission: 'reject', - env: {}, + env: { MOCK_CRASH_ON_PROMPT: '1', TSX_TSCONFIG_PATH: repoTsconfig }, disposeEofGraceMs: DEFAULT_DISPOSE_EOF_GRACE_MS, disposeGraceMs: DEFAULT_DISPOSE_GRACE_MS, onError: (error, stopReason) => { errors.push({ message: error.message, stopReason }) }, @@ -483,18 +472,31 @@ describe('dsh-subagent-acp', () => { await run.dispose() }) + it('logs a flattened child failure through the registered provider', async () => { + const ctx = await setup({ MOCK_CRASH_ON_PROMPT: '1' }) + const warnings: string[] = [] + ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn + const run = await ctx.subagents.start('acp', request()) + const result = await run.result + expect(result.stopReason).toBe('error') + expect(warnings).toEqual([ + expect.stringContaining('subagent-acp "acp": child run failed (error):'), + ]) + await run.dispose() + }) + it('resolves error (never rejects) even when the onError sink itself throws', async () => { // onError is a caller-supplied callback boundary: its own exception must be // contained, or it would reject `result` and break the seam's "result never // rejects" contract that the flattening above exists to uphold. - const run = startAcpRun( - { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }, + const run = await startAcpRun( + request(), { - command: '/nonexistent/acp-agent-binary', - args: [], + command: process.execPath, + args: ['--import', tsxLoader, mockServer], cwd: process.cwd(), permission: 'reject', - env: {}, + env: { MOCK_CRASH_ON_PROMPT: '1', TSX_TSCONFIG_PATH: repoTsconfig }, disposeEofGraceMs: DEFAULT_DISPOSE_EOF_GRACE_MS, disposeGraceMs: DEFAULT_DISPOSE_GRACE_MS, onError: () => { throw new Error('sink boom') }, @@ -514,9 +516,10 @@ describe('dsh-subagent-acp', () => { const ready = join(tmp, 'ready') try { const ctx = await setup({ MOCK_TEXT: 'partial', MOCK_HANG: '1', MOCK_CRASH_ON_CANCEL: '1', MOCK_READY_FILE: ready }) - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) + const controller = new AbortController() + const run = await ctx.subagents.start('acp', request('p', controller.signal)) await waitForFile(ready) - run.cancel('crash it') + controller.abort('crash it') const result = await run.result expect(result.stopReason).toBe('aborted') await run.dispose() @@ -525,8 +528,8 @@ describe('dsh-subagent-acp', () => { } }) - it('settles aborted on cancel even when the child IGNORES session/cancel (non-cooperative)', async () => { - // The contract: run.cancel() → result settles `aborted`. A child that hangs + it('settles aborted on signal even when the child IGNORES session/cancel', async () => { + // The signal contract requires `result` to settle `aborted`. A child that hangs // its prompt AND ignores session/cancel must not wedge the parent — the // backend's own cancel-settle path resolves `aborted` without the child's // cooperation, and dispose() still reaps the process. @@ -534,9 +537,10 @@ describe('dsh-subagent-acp', () => { const ready = join(tmp, 'ready') try { const ctx = await setup({ MOCK_TEXT: 'partial', MOCK_HANG: '1', MOCK_IGNORE_CANCEL: '1', MOCK_READY_FILE: ready }) - const run = ctx.subagents.start('acp', { prompt: [{ type: 'text', text: 'p' }], parent: fakeParent }) + const controller = new AbortController() + const run = await ctx.subagents.start('acp', request('p', controller.signal)) await waitForFile(ready) - run.cancel('test') + controller.abort('test') // Bound it: a regression (cancel only notifies the child, which ignores it) // would hang result forever — fail loud instead of stalling the suite. const result = await Promise.race([ diff --git a/packages/subagent/subagent-fork/README.md b/packages/subagent/subagent-fork/README.md index 2aef651839..bf90ecdf52 100644 --- a/packages/subagent/subagent-fork/README.md +++ b/packages/subagent/subagent-fork/README.md @@ -1,23 +1,23 @@ # @deepseek-ai/dsh-subagent-fork -The in-process **fork** subagent backend: a [`SubagentProvider`](../subagent/README.md) that runs each child as a child [`Agent`](../../core/agent) **seeded with a prefix of the parent's session log** instead of starting with an empty conversation. The seed affects conversation history only. Tool registrations and restrictions follow the child's fresh flat scope; no parent/child authority relation is defined. Fork shares the run driver (`startInProcessRun`) with [`dsh-subagent-spawn`](../subagent-spawn/README.md); the only difference is the seed. The shared `run.started` boundary resolves only after the seeded child is published, so `subagent/start` observers see a live registry entry. +The fork provider creates an in-process child seeded with the parent's completed conversation turns. It shares all run mechanics with spawn; the session seed is the only behavioral difference. -## The seed boundary (the crux) +## Seed boundary -At the moment a subagent tool's `execute` runs, the parent's CURRENT turn is open and unbalanced: the log holds the `assistant/message` carrying this spawn's tool-call and the dangling `tool/call` with no `tool/result` yet. Seeding that raw prefix would give the child an open turn that the session constructor and the dev-mode [invariants](../../support/invariants) replay **reject**. +The parent's current tool-calling turn is still open when a subagent starts: its log contains the assistant tool call but not the matching tool result or `turn/end`. Copying that raw log would give the child an invalid, unbalanced session. -So the fork seeds only the **balanced completed-turn prefix** — the parent's log up to and including its last `turn/end`, excluding the in-flight turn entirely (`completedTurnPrefix`). Because the live log keeps `seq === index`, the slice is contiguous-from-0 and a valid seed. A parent on its very first (not-yet-complete) turn forks an *empty* seed — i.e. effectively a fresh child. +Fork therefore uses `completedTurnPrefix(parent.session.events)`: the contiguous prefix ending at the last `turn/end`. The child sees all completed parent turns and none of the in-flight turn. If the parent has not completed a turn yet, the seed is empty and the child behaves like a fresh spawn. -The seam this rides on: `CreateAgentOptions.seed` (added on `dsh-agent`, threaded through `AgentLoop.createAgent` → `ctx.sessions.prepare({ seed })`), the same primitive `resume` uses. +The seed transfers conversation history only. The child still receives a fresh flat registration scope; it does not inherit the parent's tool restrictions or authority. -## Capabilities +## Start and capabilities -`{ outputSchema: true, depthLimit: true, toolFilter: true, persona: true }` — identical to spawn because the shared driver owns depth, model, persona, tool-filter, and structured-output behavior. +`start(request)` passes the completed-turn seed to [`startInProcessRun`](../subagent-inprocess/README.md) and awaits child publication. The shared driver owns cancellation, depth, customization, result reading, and disposal. + +Fork advertises `{ outputSchema: true, depthLimit: true, toolFilter: true, persona: true }`, identical to spawn. ## Config | Key | Meaning | |---|---| | `providerName` | Registry name on `ctx.subagents` (default `fork`). | - -See [`dsh-subagent-spawn`](../subagent-spawn/README.md) for the run lifecycle, model inheritance, and depth tracking — all shared. diff --git a/packages/subagent/subagent-fork/src/index.ts b/packages/subagent/subagent-fork/src/index.ts index 3b9243aa2e..ebf64e7b70 100644 --- a/packages/subagent/subagent-fork/src/index.ts +++ b/packages/subagent/subagent-fork/src/index.ts @@ -72,11 +72,11 @@ class ForkProvider implements SubagentProvider { // Context contract: a forked child IS seeded with the parent's completed-turn prefix. readonly inheritsParentContext = true - constructor(readonly name: string, private readonly ctx: Context) {} + constructor(readonly name: string) {} start(request: SubagentStartRequest) { const seed = completedTurnPrefix(request.parent) - return startInProcessRun(this.ctx, request, { + return startInProcessRun(request, { // Only pass a seed when there's a completed turn to inherit; an empty seed // is equivalent to a fresh child, so omit it to keep the session unseeded. ...seed.length > 0 ? { seed } : {}, @@ -85,5 +85,5 @@ class ForkProvider implements SubagentProvider { } export function apply(ctx: Context, config: Config): void { - ctx.subagents.registerProvider(new ForkProvider(config.providerName, ctx)) + ctx.subagents.registerProvider(new ForkProvider(config.providerName)) } diff --git a/packages/subagent/subagent-fork/tests/multi-subagent.spec.ts b/packages/subagent/subagent-fork/tests/multi-subagent.spec.ts index 1f932fbaf9..8060a77fd4 100644 --- a/packages/subagent/subagent-fork/tests/multi-subagent.spec.ts +++ b/packages/subagent/subagent-fork/tests/multi-subagent.spec.ts @@ -7,13 +7,17 @@ import ToolRegistry from '@deepseek-ai/dsh-tools' import AgentRegistry, { AgentId } from '@deepseek-ai/dsh-agent' import AgentLoop from '@deepseek-ai/dsh-agent-loop' import * as Invariants from '@deepseek-ai/dsh-invariants' -import SubagentService from '@deepseek-ai/dsh-subagent' +import SubagentService, { type SubagentStartRequest } from '@deepseek-ai/dsh-subagent' import * as Spawn from '@deepseek-ai/dsh-subagent-spawn' import { MockAdapter, textResponse } from '../../../core/agent-loop/tests/mock-adapter.ts' import * as fork from '../src/index.ts' type Script = ConstructorParameters[0] +function start(ctx: Context, provider: string, request: Omit & { signal?: AbortSignal }) { + return ctx.subagents.start(provider, { signal: request.signal ?? new AbortController().signal, ...request }) +} + /** * The two in-process backends coexist on one context: the SAME parent agent * delegates to a `spawn` child (fresh) and a `fork` child (seeded with its log), @@ -62,13 +66,13 @@ describe('multi-subagent coexistence (spawn + fork on one context)', () => { const parentPrefixLen = parent.session.events.length // Delegate to a fresh spawn child. - const spawnRun = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'spawn task' }], parent }) + const spawnRun = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'spawn task' }], parent }) const spawnResult = await spawnRun.result expect(spawnResult.stopReason).toBe('completed') expect(text(spawnResult.output)).toBe('spawn child reply') // Delegate to a fork child (seeded with the parent's turn-1 prefix). - const forkRun = ctx.subagents.start('fork', { prompt: [{ type: 'text', text: 'fork task' }], parent }) + const forkRun = await start(ctx, 'fork', { prompt: [{ type: 'text', text: 'fork task' }], parent }) const forkResult = await forkRun.result expect(forkResult.stopReason).toBe('completed') expect(text(forkResult.output)).toBe('fork child reply') diff --git a/packages/subagent/subagent-fork/tests/subagent-fork.spec.ts b/packages/subagent/subagent-fork/tests/subagent-fork.spec.ts index 66234a012c..e089cebe5f 100644 --- a/packages/subagent/subagent-fork/tests/subagent-fork.spec.ts +++ b/packages/subagent/subagent-fork/tests/subagent-fork.spec.ts @@ -8,7 +8,7 @@ import ToolRegistry from '@deepseek-ai/dsh-tools' import AgentRegistry, { AgentId } from '@deepseek-ai/dsh-agent' import AgentLoop from '@deepseek-ai/dsh-agent-loop' import * as Invariants from '@deepseek-ai/dsh-invariants' -import SubagentService from '@deepseek-ai/dsh-subagent' +import SubagentService, { type SubagentStartRequest } from '@deepseek-ai/dsh-subagent' import { MockAdapter, textResponse, toolCallResponse } from '../../../core/agent-loop/tests/mock-adapter.ts' import type { StreamChunk } from '@deepseek-ai/dsh-llm' import * as fork from '../src/index.ts' @@ -17,6 +17,10 @@ import { completedTurnPrefix } from '../src/index.ts' type Script = ConstructorParameters[0] +function start(ctx: Context, provider: string, request: Omit & { signal?: AbortSignal }) { + return ctx.subagents.start(provider, { signal: request.signal ?? new AbortController().signal, ...request }) +} + /** A bare `stop` finish that streams no content → the turn ends `completed` * with NO `assistant/message` of its own. */ const emptyStop: StreamChunk[] = [{ type: 'finish', reason: { kind: 'stop' } }] @@ -77,9 +81,9 @@ describe('dsh-subagent-fork', () => { if (info.provider === 'fork') childAtStart = ctx.agents.get(info.id) }) - const run = ctx.subagents.start('fork', { prompt: [{ type: 'text', text: 'child q' }], parent }) + const starting = start(ctx, 'fork', { prompt: [{ type: 'text', text: 'child q' }], parent }) expect(childAtStart).toBeUndefined() - await run.started + const run = await starting expect(childAtStart).toBe(ctx.agents.get(run.id)) expect(childAtStart?.id).toBe(run.id) @@ -92,7 +96,7 @@ describe('dsh-subagent-fork', () => { // the seed → the child runs fresh. Exercises the `seed.length > 0` false arm. const { ctx, parent } = await setup([textResponse('fresh child')]) expect(completedTurnPrefix(parent)).toEqual([]) - const run = ctx.subagents.start('fork', { prompt: [{ type: 'text', text: 'child q' }], parent }) + const run = await start(ctx, 'fork', { prompt: [{ type: 'text', text: 'child q' }], parent }) const result = await run.result expect(result.stopReason).toBe('completed') expect(text(result.output)).toBe('fresh child') @@ -110,7 +114,7 @@ describe('dsh-subagent-fork', () => { await parent.whenIdle() const parentPrefixLen = parent.session.events.length - const run = ctx.subagents.start('fork', { prompt: [{ type: 'text', text: 'child question' }], parent }) + const run = await start(ctx, 'fork', { prompt: [{ type: 'text', text: 'child question' }], parent }) const result = await run.result expect(result.stopReason).toBe('completed') expect(text(result.output)).toBe('child answer') @@ -143,7 +147,7 @@ describe('dsh-subagent-fork', () => { await new Promise(r => setTimeout(r, 20)) // let the hanging turn open // Forking now must NOT throw (the open second turn is excluded from the seed). - const run = ctx.subagents.start('fork', { prompt: [{ type: 'text', text: 'child q' }], parent }) + const run = await start(ctx, 'fork', { prompt: [{ type: 'text', text: 'child q' }], parent }) const result = await run.result expect(result.stopReason).toBe('completed') expect(text(result.output)).toBe('child') @@ -165,7 +169,7 @@ describe('dsh-subagent-fork', () => { ]) parent.send([{ type: 'text', text: 'warm up' }]) await parent.whenIdle() - const run = ctx.subagents.start('fork', { + const run = await start(ctx, 'fork', { prompt: [{ type: 'text', text: 'report structured' }], parent, outputSchema: { type: 'object', properties: { answer: { type: 'number' } }, required: ['answer'] }, @@ -189,7 +193,7 @@ describe('dsh-subagent-fork', () => { parent.send([{ type: 'text', text: 'parent question' }]) await parent.whenIdle() - const run = ctx.subagents.start('fork', { prompt: [{ type: 'text', text: 'child question' }], parent }) + const run = await start(ctx, 'fork', { prompt: [{ type: 'text', text: 'child question' }], parent }) const result = await run.result // The child completed its own (empty) turn — completed, but with NO output // borrowed from the seeded parent prefix. diff --git a/packages/subagent/subagent-inprocess/README.md b/packages/subagent/subagent-inprocess/README.md index 0980d85aa9..6077a139cd 100644 --- a/packages/subagent/subagent-inprocess/README.md +++ b/packages/subagent/subagent-inprocess/README.md @@ -1,41 +1,41 @@ # @deepseek-ai/dsh-subagent-inprocess -The shared **in-process subagent run driver**. A library with no provider or import-time registration that the in-process backends — [spawn](../subagent-spawn/README.md) (a fresh child) and [fork](../subagent-fork/README.md) (a child seeded with a prefix of the parent's log) — both build on. Each accepted run installs one provider-owned cleanup effect. The backends are thin shells that differ ONLY in the session seed they pass; everything downstream lives here, so neither backend depends on the other. +This package is the shared run driver for the two in-process providers. Spawn passes no session seed; fork passes the parent's completed-turn prefix. Everything else—depth, child creation, optional child customization, result reading, cancellation, and disposal—has one implementation here. -## What it exports +## Start contract -### `startInProcessRun(ctx, request, options): SubagentRun` +`startInProcessRun(request, options): Promise` fulfills only after the child is published in `ctx.agents`. A rejected start has already quiesced the agent factory's unpublished creation transaction, so the caller never receives a half-created handle. -Runs a child as a child [`Agent`](../../core/agent) on the same cordis context (`ctx.agents`): +The driver follows this sequence: -1. reads every public request and seed field once before asynchronous owner setup: the parent and signal remain identity capabilities, while tool filter, seed, agent options, output schema, and prompt are each materialized by the shared one-pass lossless-JSON snapshot. It rejects malformed `request.maxDepth` and `persona` values, validates the parent's `subagentDepth`, computes child depth = `depthOf(parent) + 1`, rejects a child depth outside the safe-integer domain with `RangeError`, rejects a defined `maxDepth` cap breach with `SubagentDepthError`, reports an invalid schema as `OutputSchemaError`, and derives both the child prefix and `seedLength` from the same detached seed; -2. first installs provider ownership, then attaches the request abort listener and creates one run-owner Cordis fiber under `parent.ctx`; an already-unloading provider therefore leaves no child or orphaned listener. Async child creation goes through that fiber's `ctx.agents` service with fresh IDs, lineage/seed, inherited model, and an unpublished setup transaction for persona, tool restriction, and structured output. Parent teardown, provider teardown, manual `run.dispose()`, and cancellation before readiness all dispose this exact node, preventing publication after it becomes inactive and sharing the same quiescence boundary. `startInProcessRun` still returns its `SubagentRun` immediately: `run.started` resolves only after `ctx.agents.create()` has published the child and rejects when pre-readiness cancellation rolls the transaction back; -3. drives the one-shot: `child.send(prompt)` then `await child.whenIdle()` (ordering matters — `send` enqueues synchronously, so `whenIdle` observes the queued work and resolves on the child's `running → idle` transition, never before the turn starts); there is deliberately NO re-prompt for a structured child that finished cleanly without calling `structured_output` — the shortfall maps to an `error` result for the parent; -4. reads the result, scoped to the child's OWN events (everything at or after `seedLength`, so a seeded child that produced no message of its own never returns the seeded parent's last message): the last `assistant/message` content (deep-cloned — the log is frozen) and the last `turn/end.reason` mapped to a `SubagentStopReason`. A structured run surfaces the captured value as `result.structured`; a structured child that finished cleanly WITHOUT ever capturing settles `error` (a clean finish without the demanded result is a failure, not a success with a missing field). +1. Validate the parent depth and optional absolute `maxDepth`, then derive child depth as parent depth plus one. +2. Call `parent.ctx.agents.create` directly, passing the required request signal into the factory's creation transaction. +3. During that transaction's unpublished setup window, install the requested persona, tool restriction, and structured-output runtime. +4. Publish the child, retain the returned `AgentHandle`, and drive one task with `child.send(prompt)` followed by `child.whenIdle()`. +5. Read the child's own last assistant message and terminal turn reason, excluding any fork seed. -`SubagentService` waits for `run.started` before emitting `subagent/start`, so a synchronous start observer can resolve the published child with `ctx.agents.get(run.id)`; the result driver awaits the same boundary before sending the prompt. An attempt that never publishes rejects readiness and emits no false start/end pair; its result reports a deliberate cancel/dispose as `aborted` and propagates an infrastructure fault. `dispose()` awaits creation or rollback and then delegates to `AgentHandle.dispose()` (stop and drain → remove agent → detach session → unwind scope). Before readiness, `cancel()` deactivates the creation owner: before creation notification begins, no agent/session lifecycle edge escapes; if cancellation is triggered synchronously by a creation observer, every begun edge is paired by rollback and the driver never unlocks or starts. After readiness, cancellation reaches the live child immediately. Either path records the cancellation, so a cancel landing before any `turn/end` settles `aborted`, honoring the cancel contract rather than the generic no-turn `error`. +The child gets the parent's working-directory/session lineage and inherits the parent model unless `request.agentOptions` overrides it. It gets a fresh flat registration scope: parent ownership does not import parent tool restrictions or establish an authority subset. -The child receives a fresh flat registration scope. Its `toolFilter` masks the live global tool layer and scope-local registrations are merged afterward; parent ownership does not import the parent's tool restrictions or establish an authority subset. The [agent-scope RFC](../../../docs/rfc/implemented/architecture/2026-07-08-agent-scope-contexts.md#security-and-authority-are-explicit-non-goals) owns that explicit non-goal. +## Cancellation and ownership -### `InProcessRunOptions` +The required request signal covers both startup and the live run. Before publication, `AgentCreationTransaction` observes it, rolls back, and rejects. The factory detaches that creation-only listener before returning; the driver immediately checks the signal once more before installing a minimal live-run listener, closing the handoff race. After publication, abort cancels the child. -`{ seed?: SessionEvent[] }` — the optional child-session seed: absent for spawn, or the parent's balanced completed-turn prefix for fork. +After fulfillment, the caller owns the run. Provider-plugin unload does not revoke it. `dispose()` removes the live abort listener, records cancellation, and delegates to the returned `AgentHandle.dispose()`, whose memoized quiescence transaction stops the loop, removes the agent and session, and unwinds scoped registrations. Cancellation owns every non-completed in-flight outcome and reports `aborted`; an already-completed turn remains completed. -### Structured output (package-internal runtime) +## Spawn and fork inputs -`attachStructuredRuntime(childCtx, schema)` registers the run's whole enforcement surface as SCOPED registrations on the child's `agent.ctx` — riding the child's fiber (a backend hot-reload mid-run cannot unregister anything; a disposed child leaves no residue) and visible to that child alone (two concurrent structured runs never interact; no placeholder schema, no strip-for-everyone-else, no refcounted global state): +`InProcessRunOptions` is `{ seed?: SessionEvent[] }`. Spawn omits it. Fork supplies a balanced completed-turn prefix and records its length so the result reader never mistakes a seeded parent message for child output. -- the `structured_output` capture tool with the run's REAL schema as its registered `parameters`, validating each call (`validateStructuredValue`) — violations become an `INVALID_ARGS` isError the model retries in-turn; a valid call STAGES the value in a `WeakMap` keyed by that call's `ToolExecution` object; -- the calling instruction as an ordinary order-190 scoped prompt section (the demand travels with the tool, as prompt state of exactly one agent); -- a scoped `systemPrompt.protect()` registration making the capture instruction and schema canonical after the complete assembly waterfall. Canonical absence is protected too: pure Code Mode removes `structured_output` from the wire and declares it through the SDK. The tool registry separately owns and protects its `tools:sdk` section and reserved `run_code` transport; protection guarantees those named contributions, while unrelated listener-added schemas remain the listener's responsibility. The loop logs the finalized assembly as the step's `request/header`, so the demand remains reconstructable; -- a scoped `tools/result` observer as the commit point: it promotes a staged value only when that same execution's immutable, JSON-safe authoritative result after the complete pre-execute → guards → execute → post-execute pipeline succeeds. For a Code Mode SDK sub-dispatch, the child's opaque `parent` token matches the enclosing `run_code` execution's registry-assigned `token`, so promotion waits for that outer final result without exposing its live object; a runtime failure or post-policy block discards the value. Execution-object identity prevents call-id reuse or another execution from reaching the stage; -- a scoped monotonic `tools.guard()` denial for every call arriving after capture. Guards run after the extensible pre-execute waterfall and cannot return allow, so terminal means terminal within the step regardless of listener order; -- a scoped `agent/turn-stop` terminal policy stopping the child's turn once its output is captured. It runs after ordinary continuation and steering folding, and its terminal state survives turn close and flush, so later listeners cannot leak steering into another step or turn; ordinary queued prompts remain intact. +`depthOf(agent)` reads `AgentOptions.subagentDepth`, treating absence as top-level depth zero and rejecting malformed stored values. `SubagentDepthError` reports an attempted child depth above `maxDepth`; an unrepresentable depth above the safe-integer domain is a `RangeError`. -### `depthOf(agent): number` +## Structured output -Delegation depth rides on a merge-extensible `AgentOptions.subagentDepth` field (0 for a top-level agent, parent + 1 for a child), so a nested spawn reads its parent's depth from `parent.options.subagentDepth`. `depthOf` treats only `undefined` as absent (top-level depth 0) and rejects every malformed present value instead of letting it disable the cap comparison. +`attachStructuredRuntime(childCtx, schema)` installs the whole contract in the child's scope: -### `SubagentDepthError` +- A `structured_output` tool registered with the requested schema validates and stages the model's value. +- An order-190 system-prompt section tells the child that the tool call is the terminal answer. +- Both contributions use `ownerFinal: true`, so their owners control their final presence after prompt/tool assembly while unrelated contributions remain extensible. +- A `tools/result` observer commits a staged value only after that execution's authoritative final tool result succeeds, including the enclosing `run_code` result for Code Mode sub-dispatch. +- A monotonic tool guard blocks later calls after capture, and `agent/turn-stop` ends the turn after the structured result commits. -Thrown by `startInProcessRun` when a spawn would exceed the request's defined `maxDepth` cap; carries `attemptedDepth` and `maxDepth`. A valid parent at `Number.MAX_SAFE_INTEGER` instead produces `RangeError`, because its child depth cannot be represented within the stored safe-integer domain even when `maxDepth` is omitted. +A clean turn that never commits the required structured value reports `error`; the driver does not re-prompt. All registrations ride the child fiber and disappear with it. diff --git a/packages/subagent/subagent-inprocess/src/index.ts b/packages/subagent/subagent-inprocess/src/index.ts index 3ccb08fda9..f6de7200cf 100644 --- a/packages/subagent/subagent-inprocess/src/index.ts +++ b/packages/subagent/subagent-inprocess/src/index.ts @@ -1,26 +1,17 @@ /** - * The shared in-process subagent run driver: run a child as a child - * {@link Agent} on the SAME cordis context (`ctx.agents`) — the cheapest - * transport, reusing the agent factory's quiescent {@link AgentHandle} - * teardown. The concrete in-process backends are thin shells over this driver, - * differing ONLY in the `seed` they pass (a fresh child vs. a child seeded with - * a prefix of the parent's log); everything downstream — drive the child, read - * its final output, map the stop reason, dispose — is identical and lives here. - * - * This package declares no provider and performs no import-time registration; - * it is a library the backend packages depend on, so neither backend needs to - * know about the other. Each accepted run does install one provider-owned - * effect for structured-concurrency cleanup. + * Shared driver for in-process subagent providers. The agent factory's + * creation transaction owns unpublished setup and rollback; after publication + * the returned AgentHandle is the one quiescent lifecycle owner held by the + * provider's caller. * * @module @deepseek-ai/dsh-subagent-inprocess */ import { randomUUID } from 'node:crypto' -import type { Context, Fiber } from 'cordis' -import { AgentId, type Agent, type AgentHandle, type AgentOptions } from '@deepseek-ai/dsh-agent' -import { SessionId, snapshotJsonValue, type SessionEvent, type TurnEndReason } from '@deepseek-ai/dsh-session' +import type { Context } from 'cordis' +import { AgentId, type Agent, type AgentOptions } from '@deepseek-ai/dsh-agent' +import { SessionId, type SessionEvent, type TurnEndReason } from '@deepseek-ai/dsh-session' import type { ContentBlock } from '@deepseek-ai/dsh-llm' -import { assertSupportedOutputSchema, OutputSchemaError } from '@deepseek-ai/dsh-tools' import { assertSubagentMaxDepth } from '@deepseek-ai/dsh-subagent' import type { SubagentResult, SubagentRun, SubagentStartRequest, SubagentStopReason } from '@deepseek-ai/dsh-subagent' import { @@ -28,9 +19,6 @@ import { type StructuredAttachment, } from './structured.ts' -// The runtime itself (attach) is package-internal: runs attach it inside -// startInProcessRun's setup window, and no other package drives it. Only the -// model-facing vocabulary is public. export { STRUCTURED_OUTPUT_TOOL, STRUCTURED_OUTPUT_INSTRUCTION, @@ -38,24 +26,15 @@ export { declare module '@deepseek-ai/dsh-agent' { interface AgentOptions { - /** - * The agent's delegation depth in the subagent tree — 0 for a top-level - * (config/ACP-created) agent, parent depth + 1 for a subagent. Set by the - * in-process backends on every child they create so a nested spawn reads its - * parent's depth from `parent.options.subagentDepth` and the `depthLimit` - * capability can cap the tree. When present it is a non-negative safe - * integer. Merge-extensible field (the seam owns it; the loop neither sets - * nor reads it). - */ + /** Delegation depth: zero for a top-level agent and parent depth + 1 for a child. */ subagentDepth?: number } } /** - * Read an agent's delegation depth (absent ⇒ a top-level agent, depth 0), - * rejecting a malformed stored value instead of letting it disable comparison. - * @param agent - the agent whose options may carry `subagentDepth`. - * @returns 0 for a top-level agent, its parent's depth + 1 for a subagent. + * Read an agent's delegation depth, treating absence as top-level depth zero. + * @param agent - the agent whose options carry the depth. + * @returns its non-negative safe-integer depth. */ export function depthOf(agent: Agent): number { const depth = agent.options.subagentDepth @@ -66,7 +45,7 @@ export function depthOf(agent: Agent): number { return depth } -/** Thrown when a spawn would exceed the request's `maxDepth` cap. */ +/** Thrown when starting a child would exceed the requested depth cap. */ export class SubagentDepthError extends Error { constructor(public readonly attemptedDepth: number, public readonly maxDepth: number) { super(`subagent depth ${attemptedDepth} exceeds maxDepth ${maxDepth}`) @@ -74,7 +53,7 @@ export class SubagentDepthError extends Error { } } -/** Map a session `turn/end` reason to a {@link SubagentStopReason}. */ +/** Map a session turn outcome to the subagent seam's terminal vocabulary. */ function toStopReason(reason: TurnEndReason | undefined): SubagentStopReason { switch (reason?.kind) { case 'completed': @@ -83,9 +62,6 @@ function toStopReason(reason: TurnEndReason | undefined): SubagentStopReason { return 'max-tokens' case 'aborted': return 'aborted' - // `disposed` (torn down mid-turn) and `interrupted` (crash-closed) both mean - // the turn did not finish cleanly; surface them as a generic failure rather - // than a clean completion. A missing reason (no turn ran) is also an error. case 'error': case 'disposed': case 'interrupted': @@ -94,329 +70,120 @@ function toStopReason(reason: TurnEndReason | undefined): SubagentStopReason { } } -/** Extra inputs the spawn/fork backends supply to {@link startInProcessRun}. */ +/** Extra inputs the spawn and fork providers supply to the shared driver. */ export interface InProcessRunOptions { - /** - * The child session's seed: a balanced, contiguous-from-0 prefix of the - * parent's log (FORK), or `undefined` for a fresh child (SPAWN). - */ + /** Completed-turn seed for fork, or undefined for a fresh spawn. */ readonly seed?: SessionEvent[] } -/** Dispose a run-owner fiber and follow an already-started unload to quiescence. */ -async function quiesceFiber(fiber: Fiber): Promise { - await Promise.resolve(fiber.dispose()) - while (fiber.inertia !== undefined) await fiber.inertia +/** Error used when cancellation wins before the child publication boundary. */ +function prePublicationAbort(): Error { + return new Error('subagent request was aborted before child publication') } /** - * Start an in-process child agent for `request` and return a {@link SubagentRun}. - * - * Drives the child as a one-shot: `send(prompt)` then `whenIdle()` (the ordering - * matters — `send` enqueues synchronously, so `whenIdle` observes the queued - * work and resolves only on the child's `running → idle` transition, never - * before the turn starts). The final `assistant/message` is the result output, - * the matching `turn/end.reason` the stop reason. `dispose()` delegates to the - * factory's {@link AgentHandle.dispose} (stop loop → await quiescence → remove - * session). `cancel()` cancels a published child's in-flight turn; before - * readiness it instead deactivates the unpublished run-owner transaction, so - * `started` rejects, no agent/session lifecycle is published, and `result` - * resolves `aborted`. - * - * Throws {@link SubagentDepthError} before creating anything when the child's - * depth (parent depth + 1) would exceed `request.maxDepth`, and throws a - * `RangeError` when a valid parent depth has no safe-integer successor. - * @param ctx - the provider context that owns the live run as a second - * structured-concurrency boundary alongside the parent agent. - * @param request - the start request (prompt, parent, signal, per-child options). - * @param options - the backend's optional child-session seed. - * @returns the live run handle for the child agent. + * Establish and drive one in-process child. Fulfillment means the agent is + * already published in the registry; rejection means the agent factory's + * creation transaction and any partially-created child have reached quiescence. + * @param request - the trusted typed start request, including its required signal. + * @param options - the optional fork seed. + * @returns a ready holder-owned run. */ -export function startInProcessRun( - ctx: Context, +export async function startInProcessRun( request: SubagentStartRequest, options: InProcessRunOptions, -): SubagentRun { - // Capture every top-level field once. Parent/signal are identity capabilities; - // every data value is materialized below before asynchronous owner setup. +): Promise { + assertSubagentMaxDepth(request.maxDepth) + if (request.signal.aborted) throw prePublicationAbort() const parent = request.parent - const signal = request.signal - const persona = request.persona - const inputToolFilter = request.toolFilter - const inputMaxDepth = request.maxDepth - const inputSchema = request.outputSchema - const inputPrompt = request.prompt - const inputAgentOptions = request.agentOptions - const inputSeed = options.seed - assertSubagentMaxDepth(inputMaxDepth) - if (persona !== undefined && typeof persona !== 'string') { - throw new TypeError('subagent persona must be a string') - } - const toolFilter = inputToolFilter === undefined ? undefined : snapshotJsonValue(inputToolFilter) - if (inputToolFilter !== undefined && toolFilter === undefined) { - throw new TypeError('subagent tool filter must be losslessly JSON-serializable') - } - const seed = inputSeed === undefined ? undefined : snapshotJsonValue(inputSeed) - if (inputSeed !== undefined && seed === undefined) { - throw new TypeError('subagent seed must be losslessly JSON-serializable') - } const childDepth = depthOf(parent) + 1 if (!Number.isSafeInteger(childDepth)) { throw new RangeError('subagent child depth exceeds the safe-integer range') } - if (inputMaxDepth !== undefined && childDepth > inputMaxDepth) { - throw new SubagentDepthError(childDepth, inputMaxDepth) - } - const requestedAgentOptions = inputAgentOptions === undefined - ? {} - : snapshotJsonValue(inputAgentOptions) - if (requestedAgentOptions === undefined) { - throw new TypeError('subagent agent options must be losslessly JSON-serializable') - } - // Materialize, then assert, the schema subset BEFORE any child exists. The - // single traversal rejects non-JSON data without rereading accessors; the - // detached value then pins assertion, model-visible parameters, and runtime - // validation to one provider-owned schema. Contract failures stay typed as - // OutputSchemaError rather than leaking a materialization detail. - const schema = inputSchema === undefined ? undefined : snapshotJsonValue(inputSchema) - if (inputSchema !== undefined && schema === undefined) { - throw new OutputSchemaError(['schema annotation must be JSON data; the complete schema must be losslessly JSON-serializable']) - } - if (schema !== undefined) assertSupportedOutputSchema(schema) - // The accepted request owns a value snapshot, not the caller's mutable - // content array. Use the same one-pass boundary Session.append enforces before - // any child exists so later mutation cannot change what is logged or sent. - const prompt = snapshotJsonValue(inputPrompt) - if (prompt === undefined) { - throw new TypeError('subagent prompt must be losslessly JSON-serializable') + if (request.maxDepth !== undefined && childDepth > request.maxDepth) { + throw new SubagentDepthError(childDepth, request.maxDepth) } const childId = AgentId(randomUUID()) - // The child's OWN events begin after the seed (fork seeds the parent's - // completed-turn prefix; spawn seeds nothing). `readResult` scopes to this - // boundary so a child that produces no message of its own never returns the - // SEEDED parent's last assistant message as its result. - const seedLength = seed?.length ?? 0 + const seedLength = options.seed?.length ?? 0 const parentHeader = parent.session.header - // Inherit the parent's model by default (a child with no model cannot run); - // an explicit `request.agentOptions.model` overrides it. The deployment - // persona needs no inheritance (a context-wide section both render); a - // per-child `request.persona` becomes a SCOPED section of the same name in - // the setup below, shadowing the deployment's for this child alone. const parentModel = parent.options.model - const agentOptions = snapshotJsonValue({ + const agentOptions: AgentOptions = { ...parentModel !== undefined ? { model: parentModel } : {}, - ...requestedAgentOptions, + ...request.agentOptions, subagentDepth: childDepth, - }) - if (agentOptions === undefined) { - throw new TypeError('subagent agent options must be losslessly JSON-serializable') } - // The child's scoped world, composed in the factory's unpublished setup - // window. The factory awaits it before inserting or announcing the child, so - // a throw/rejection exposes neither id and every first assembly sees it: - // - persona: a scoped `deployment:persona` section shadowing the global one; - // - toolFilter: a scoped restrict() masking the global tool surface - // (loud unknown-name validation lives in the registry); - // - outputSchema: the structured runtime, attached as scoped registrations. let structured: StructuredAttachment | undefined const setup = (childCtx: Context): void => { - if (persona !== undefined) { - childCtx.systemPrompt.section({ name: 'deployment:persona', order: 0, text: persona }) + if (request.persona !== undefined) { + childCtx.systemPrompt.section({ name: 'deployment:persona', order: 0, text: request.persona }) } - if (toolFilter !== undefined) { - childCtx.tools.restrict(toolFilter) - } - if (schema !== undefined) { - structured = attachStructuredRuntime(childCtx, schema) + if (request.toolFilter !== undefined) childCtx.tools.restrict(request.toolFilter) + if (request.outputSchema !== undefined) { + structured = attachStructuredRuntime(childCtx, request.outputSchema) } } - // Bridge the request's abort signal to the child (the consumer also bridges - // its own exec.signal, but a backend-level bridge keeps the contract local). - // Install it after provider ownership succeeds but BEFORE awaiting creation, - // so an inactive provider cannot leave an orphaned listener and abort/dispose - // during async setup is still recorded and applied the moment a child exists. - // `cancelled` records that a cancel was requested at all. Before readiness, - // cancellation deactivates the unpublished run-owner transaction so the - // factory cannot publish an agent or session. After readiness, it cancels the - // live child. Either path settles as `aborted` (honoring the cancel contract) - // rather than falling through to the no-turn `error` mapping. - let cancelled = false - // An accessor, not an inline read: `cancelled` mutates from closures (the - // abort listener, run.cancel), which control-flow narrowing cannot see — an - // inline read at the result mapping would narrow to the initializer. - const isCancelled = (): boolean => cancelled - let child: Agent | undefined - let handle: AgentHandle | undefined - - // One run-owned Cordis fiber is the common ownership node. Install the - // provider effect FIRST: a start racing an already-unloading provider fails - // before it can mint anything under the parent. The owner fiber is then - // nested under the parent scope, and the provider/run handle both dispose - // this exact fiber. AgentFactory binds its lifecycle to `ownerCtx`, so any of - // the three owners moves the fiber out of ACTIVE synchronously and setup - // cannot publish afterward. - let ownerCtx: Context | undefined - function subagentRunOwner(inner: Context): void { ownerCtx = inner } - let ownerFiber: (Fiber & PromiseLike) | undefined - let ownerSetupError: unknown - let ownerDisposing: Promise | undefined - const disposeOwner = (): Promise => { - if (ownerDisposing !== undefined) return ownerDisposing - // An already-aborted request is observed before the owner fiber is minted. - // Do not memoize that no-op: the post-plugin cancellation check below must - // still be able to claim and deactivate the real fiber. - if (ownerFiber === undefined) return Promise.resolve() - ownerDisposing = quiesceFiber(ownerFiber) - // Pre-readiness cancellation is synchronous fire-and-forget at the public - // `cancel()` boundary. Observe a teardown rejection here; dispose() still - // awaits the same memoized promise and reports it to an explicit caller. - void ownerDisposing.catch(() => undefined) - return ownerDisposing - } - const requestCancel = (reason: string): void => { - cancelled = true - if (child === undefined) { - if (ownerFiber !== undefined) void disposeOwner() - return - } - child.cancel(reason) - } - const onAbort = (): void => { requestCancel('subagent cancelled') } - const unlinkProvider = ctx.effect(() => () => { - requestCancel('subagent provider disposed') - return disposeOwner() - }, 'subagent-inprocess.run()') - signal?.addEventListener('abort', onAbort, { once: true }) - if (signal?.aborted) requestCancel('subagent cancelled') - try { - ownerFiber = parent.ctx.plugin(Object.assign(subagentRunOwner, { - inject: ['agents', 'sessions', 'llm', 'tools', 'systemPrompt'], - })) - // `signal.aborted` is checked before this fiber exists. Once it does, make - // that recorded cancellation effective immediately; awaiting creation must - // observe an inactive owner instead of reaching the publication boundary. - if (isCancelled()) void disposeOwner() - } catch (error: unknown) { - ownerSetupError = error + const flags = { cancelled: false } + const handle = await parent.ctx.agents.create({ + agentId: childId, + sessionId: SessionId(randomUUID()), + meta: { + ...parentHeader.cwd !== undefined ? { cwd: parentHeader.cwd } : {}, + parentSession: parentHeader.id, + ...seedLength > 0 ? { seedLength } : {}, + }, + ...options.seed !== undefined ? { seed: options.seed } : {}, + agentOptions, + signal: request.signal, + setup, + }) + const child = handle.agent + // Agent creation detaches its creation-only abort listener before returning. + // Close the narrow handoff race before installing the live-run listener. + // Static analysis does not model the abort that may land between the + // factory's listener detachment and this continuation. + // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition + if (request.signal.aborted) { + flags.cancelled = true + await handle.dispose() + throw prePublicationAbort() } - const creation: Promise = (async () => { - if (ownerSetupError !== undefined) { - throw ownerSetupError instanceof Error - ? ownerSetupError - : new Error('subagent run owner setup failed with a non-Error value', { cause: ownerSetupError }) - } - await ownerFiber - if (ownerCtx === undefined) { - throw new Error('subagent run owner became inactive before child creation') - } - // Invoke the factory THROUGH the parent scope. Cordis binds the factory's - // lifecycle effect to the accessing context, so parent ownership exists - // before persistence/setup and publication—not as a fallible link added - // after the child is already visible. A disposed parent therefore rejects - // before any session/agent notification, and disposal during async setup - // wins the unpublished transaction. - const created = await ownerCtx.agents.create({ - agentId: childId, - sessionId: SessionId(randomUUID()), - meta: { - ...parentHeader.cwd !== undefined ? { cwd: parentHeader.cwd } : {}, - parentSession: parentHeader.id, - ...seedLength > 0 ? { seedLength } : {}, - }, - ...seed !== undefined ? { seed } : {}, - agentOptions, - setup, - }) - handle = created - child = created.agent - return created.agent - })() - - // Provider readiness is a distinct lifecycle boundary from accepting the - // request. It resolves only after the factory has published the child and - // returned its handle, so SubagentService can emit `subagent/start` while - // `ctx.agents.get(childId)` is guaranteed to resolve. The result path awaits - // THIS SAME promise immediately, which also observes a readiness rejection - // when the driver is invoked directly rather than through SubagentService. - const started: Promise = creation.then(() => undefined) + const onAbort = (): void => { + flags.cancelled = true + child.cancel('subagent request aborted') + } + request.signal.addEventListener('abort', onAbort, { once: true }) const result: Promise = (async () => { try { - let liveChild: Agent - try { - await started - // `creation` assigns `child` before it fulfills, and `started` is its - // direct fulfillment projection. The cast records that local invariant - // without manufacturing an unreachable runtime branch. - liveChild = child as Agent - } catch (error: unknown) { - if (isCancelled()) return { output: [], stopReason: 'aborted' } - throw error instanceof Error ? error : new Error('subagent child creation failed with a non-Error value', { cause: error }) - } - liveChild.send(prompt) - await liveChild.whenIdle() - // Deliberately NO re-prompt when a structured child finishes cleanly - // without calling structured_output: readResult maps that to `error` — - // the shortfall goes to the parent instead of buying extra model turns. - return readResult(liveChild, seedLength, isCancelled(), structured ? { captured: structured.captured() } : undefined) + child.send(request.prompt) + await child.whenIdle() + return readResult( + child, + seedLength, + flags.cancelled, + structured ? { captured: structured.captured() } : undefined, + ) } finally { - signal?.removeEventListener('abort', onAbort) + request.signal.removeEventListener('abort', onAbort) } })() - let disposing: Promise | undefined return { id: childId, - started, result, - cancel(reason?: string): void { - requestCancel(reason ?? 'subagent cancelled') - }, - async dispose(): Promise { - return (disposing ??= (async () => { - signal?.removeEventListener('abort', onAbort) - requestCancel('subagent disposed during creation') - // Removing provider ownership and disposing the common run-owner fiber - // are the same quiescence transaction; parent disposal may already have - // claimed it, in which case disposeOwner follows fiber inertia. - await unlinkProvider() - try { - await creation - } catch { - // Creation rollback already reached quiescence; there is no handle - // left to dispose, and dispose must not mask result's infrastructure - // rejection with the same error from a finally block. - return - } - await disposeOwner() - await handle?.dispose() - })()) + dispose(): Promise { + request.signal.removeEventListener('abort', onAbort) + flags.cancelled = true + return handle.dispose() }, } } -/** - * Read a settled child's terminal result from its session log, scoped to the - * child's OWN events (everything at or after `seedLength` — fork seeds the - * parent's completed-turn prefix, so a child that produced no message of its - * own must NOT return the seeded parent's last assistant message). The output - * is the child's last `assistant/message` content (deep-cloned — the log is - * frozen); the stop reason is the child's last `turn/end` reason mapped to a - * {@link SubagentStopReason}. When `cancelled` is set but no `turn/end` was - * logged (a cancel landed in the pre-turn window, before any turn ran), the - * run settles `aborted` per the {@link SubagentRun.cancel} contract rather than - * the generic no-turn `error`. - * - * A structured run (`structured` present) additionally reports the captured - * value on {@link SubagentResult.structured}. A structured child that finished - * CLEANLY without ever capturing (the nudges ran out) settles `error` — a clean - * finish without the demanded structured result is a failure, not a success - * with a missing field; a non-`completed` reason keeps its own honest mapping. - */ +/** Read one settled child's result from events after its optional fork seed. */ function readResult( child: Agent, seedLength: number, @@ -424,17 +191,20 @@ function readResult( structured?: { captured?: { value: unknown } | undefined }, ): SubagentResult { const own = child.session.events.slice(seedLength) - const lastMessage = own.findLast((e): e is SessionEvent<'assistant/message'> => e.type === 'assistant/message') - const lastEnd = own.findLast((e): e is SessionEvent<'turn/end'> => e.type === 'turn/end') - const output: ContentBlock[] = lastMessage ? structuredClone(lastMessage.data.content) : [] - const stopReason: SubagentStopReason = lastEnd === undefined && cancelled + const lastMessage = own.findLast((event): event is SessionEvent<'assistant/message'> => event.type === 'assistant/message') + const lastEnd = own.findLast((event): event is SessionEvent<'turn/end'> => event.type === 'turn/end') + const output: ContentBlock[] = lastMessage?.data.content ?? [] + const recorded = toStopReason(lastEnd?.data.reason) + // Disposal can tear the owner down before the loop records its ordinary + // `aborted` end, yielding `disposed` instead. A requested cancellation owns + // every non-completed in-flight outcome; a turn already completed stays so. + const stopReason: SubagentStopReason = cancelled && recorded !== 'completed' ? 'aborted' - : toStopReason(lastEnd?.data.reason) - if (structured) { - if (structured.captured) return { output, structured: structured.captured.value, stopReason } - // No capture on a cleanly-completed turn: an ERROR when the run was left - // to finish (the nudges ran out), but ABORTED when a cancel is why the - // nudging stopped — the cancel contract outranks the schema shortfall. + : recorded + if (structured !== undefined) { + if (structured.captured !== undefined) { + return { output, structured: structured.captured.value, stopReason } + } if (stopReason === 'completed') return { output, stopReason: cancelled ? 'aborted' : 'error' } } return { output, stopReason } diff --git a/packages/subagent/subagent-inprocess/src/structured.ts b/packages/subagent/subagent-inprocess/src/structured.ts index d4267ae009..f81d82ef62 100644 --- a/packages/subagent/subagent-inprocess/src/structured.ts +++ b/packages/subagent/subagent-inprocess/src/structured.ts @@ -16,12 +16,12 @@ * * The child scope's registrations enforce the contract: * - * - `systemPrompt.protect()` declaratively protects the capture tool and its - * instruction. The service restores their canonical pre-waterfall state + * - `ownerFinal: true` on the capture tool and instruction declares that the + * owning registrations control their final presence. Prompt assembly restores their canonical state * after EVERY assembly listener. Canonical absence is protected too: pure * Code Mode keeps `structured_output` in the SDK only and never grows a - * second native wire tool. Code Mode's owner independently protects its SDK - * and `run_code` transport. The loop logs the finalized assembly as the + * second native wire tool. Code Mode independently declares its SDK section + * and `run_code` transport owner-final. The loop logs the finalized assembly as the * request header, so the demand is reconstructable log state, never a * wire-only mutation. * - `agent/turn-stop` (serial, scoped): stop the child's turn once its output @@ -79,7 +79,7 @@ export interface StructuredAttachment { * agent-creation `setup` window with the child's scope context — every * registration rides the child's fiber and unwinds with the child. * @param childCtx - the child agent's scope context (`setup`'s argument). - * @param schema - the detached, already-asserted schema subset to enforce (see + * @param schema - the trusted, already-asserted schema subset to enforce (see * `assertSupportedOutputSchema` in dsh-tools). * @returns the attachment handle (read `captured()` after the child settles). */ @@ -110,15 +110,16 @@ export function attachStructuredRuntime(childCtx: Context, schema: StructuredOut childCtx.tools.register({ ...schemaEntry, + ownerFinal: true, execute(args: unknown, exec: ToolExecution): Promise { const violations = validateStructuredValue(schema, args) // ToolArgsError → isError result with INVALID_ARGS: the model retries // within the same turn, exactly like a schema-validated defineTool call. if (violations.length > 0) throw new ToolArgsError(violations) // Two-phase commit, keyed by THIS execution: later transformable - // waterfalls may still turn the success into an error. Snapshot the - // validated value independently of the already-frozen pipeline arguments. - staged.set(exec, { value: structuredClone(args) }) + // waterfalls may still turn the success into an error. ToolRegistry has + // already frozen model-bound arguments at the actual input boundary. + staged.set(exec, { value: args }) return Promise.resolve([{ type: 'text', text: 'Structured output recorded.' }]) }, }) @@ -127,16 +128,7 @@ export function attachStructuredRuntime(childCtx: Context, schema: StructuredOut name: `tool:${STRUCTURED_OUTPUT_TOOL}`, order: 190, text: STRUCTURED_OUTPUT_INSTRUCTION, - }) - - // Service-owned finalization, not waterfall ordering. The canonical - // assembly determines both presence and absence: native/both modes restore - // the capture schema on the wire, while pure Code Mode removes any injected - // native entry. ToolRegistry's own protection independently restores the SDK - // section and run_code transport that carry the same schema. - childCtx.systemPrompt.protect({ - sections: [`tool:${STRUCTURED_OUTPUT_TOOL}`], - tools: [STRUCTURED_OUTPUT_TOOL], + ownerFinal: true, }) // Stop the child's turn once its output is captured. This monotonic serial diff --git a/packages/subagent/subagent-inprocess/tests/structured.spec.ts b/packages/subagent/subagent-inprocess/tests/structured.spec.ts index 54ac968f4b..f989d727c8 100644 --- a/packages/subagent/subagent-inprocess/tests/structured.spec.ts +++ b/packages/subagent/subagent-inprocess/tests/structured.spec.ts @@ -65,7 +65,7 @@ async function setup(script: Script, options: SetupOptions = {}) { name: 'spawn', capabilities: { outputSchema: true, depthLimit: true, toolFilter: false, persona: false }, inheritsParentContext: false, - start: (request: SubagentStartRequest) => startInProcessRun(ctx, request, {}), + start: (request: SubagentStartRequest) => startInProcessRun(request, {}), }) ctx.llm.registerAdapter(['mock'], adapter) const parent = ctx.agentLoop.create(AgentId('parent'), { model: 'mock' }) @@ -73,7 +73,13 @@ async function setup(script: Script, options: SetupOptions = {}) { } function structuredRequest(parent: SubagentStartRequest['parent'], extra?: Partial): SubagentStartRequest { - return { prompt: [{ type: 'text', text: 'produce the answer' }], parent, outputSchema: SCHEMA, ...extra } + return { + prompt: [{ type: 'text', text: 'produce the answer' }], + parent, + signal: new AbortController().signal, + outputSchema: SCHEMA, + ...extra, + } } /** The tool names of one recorded model request. */ @@ -86,7 +92,7 @@ describe('in-process structured output', () => { const { ctx, parent } = await setup([ toolCallResponse('c1', STRUCTURED_OUTPUT_TOOL, { answer: 42, note: 'done' }), ]) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(result.stopReason).toBe('completed') expect(result.structured).toEqual({ answer: 42, note: 'done' }) @@ -98,7 +104,7 @@ describe('in-process structured output', () => { toolCallResponse('c1', STRUCTURED_OUTPUT_TOOL, { answer: 1 }), textResponse('MUST NOT BE CONSUMED'), ]) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) await run.result // Default continuation would run a second step after the tool call; the // structured runtime's turn-continuation veto stops the turn instead. @@ -129,7 +135,7 @@ describe('in-process structured output', () => { return Promise.resolve([{ type: 'text', text: 'ran' }]) }, }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(result.stopReason).toBe('completed') expect(result.structured).toEqual({ answer: 5 }) @@ -157,7 +163,7 @@ describe('in-process structured output', () => { return Promise.resolve([{ type: 'text', text: 'ran' }]) }, }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) // Registered after the child and prepended: this listener returns allow // after every downstream pre-execute decision. The service-owned guard // runs after the waterfall and can only deny, so the body still cannot run. @@ -194,7 +200,7 @@ describe('in-process structured output', () => { return Promise.resolve([{ type: 'text', text: 'ran' }]) }, }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result // The call ran BEFORE captured was set: the deny gate only guards the // window after the terminal answer landed. @@ -203,46 +209,20 @@ describe('in-process structured output', () => { await run.dispose() }) - it('snapshots the schema at start(): caller mutation after start cannot drift enforcement', async () => { - const mutable: StructuredOutputSchema = { - type: 'object', - properties: { answer: { type: 'number' } }, - required: ['answer'], - additionalProperties: false, - } - const pristine = structuredClone(mutable) - const { ctx, parent, adapter } = await setup([ - toolCallResponse('c1', STRUCTURED_OUTPUT_TOOL, { answer: 3 }), - ]) - const run = ctx.subagents.start('spawn', structuredRequest(parent, { outputSchema: mutable })) - // Mutate the caller's object AFTER start() returned but before the child's - // first request assembles: with a live reference this would reach both the - // model-visible parameters and validateStructuredValue. - ;(mutable.properties as Record).answer = { type: 'string' } - const result = await run.result - expect(result.structured).toEqual({ answer: 3 }) - // The child's request carried the PRISTINE schema, not the mutated one. - const childRequest = adapter.requests.at(-1) - const captureTool = (childRequest?.tools ?? []).find(tool => tool.name === STRUCTURED_OUTPUT_TOOL) - expect(captureTool?.parameters).toEqual(pristine) - await run.dispose() - }) - it('a later-prepended continuation wrapper cannot resurrect a captured turn', async () => { const { ctx, parent, adapter } = await setup([ toolCallResponse('c1', STRUCTURED_OUTPUT_TOOL, { answer: 7 }), textResponse('MUST NOT BE CONSUMED'), ]) ctx.on('agent/turn-continuation', () => Promise.resolve({ action: 'stop' })) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) let wrapperInstalled = false - // Register this observer only after start() returns. The child session-start - // boundary is after its unpublished setup attached structured output but - // before the loop can run; install a prepended wrapper there. It awaits the + // Register before the ready-only start. The child session-start boundary is + // after unpublished setup attached structured output but before the loop + // can run. The wrapper awaits the // explicit downstream stop above, then overwrites that result with continue. // The later terminal checkpoint still wins. ctx.on('agent/session-start', (child) => { - if (child.id !== run.id) return + if (child === parent) return wrapperInstalled = true child.ctx.on('agent/turn-continuation', async (_subject, _turn, _decision, next): Promise => { const downstream = await next() @@ -250,6 +230,7 @@ describe('in-process structured output', () => { return { action: 'continue' } }, { prepend: true }) }) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(wrapperInstalled).toBe(true) expect(result.structured).toEqual({ answer: 7 }) @@ -268,7 +249,7 @@ describe('in-process structured output', () => { // would turn the stop back into continue. The terminal checkpoint runs // afterwards and discards that steering. ctx.on('agent/turn-continuation', () => Promise.resolve({ action: 'stop' })) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) ctx.on('agent/session-start', (child) => { if (child.id !== run.id) return child.ctx.on('agent/turn-continuation', async (subject, _turn, _decision, next): Promise => { @@ -294,7 +275,7 @@ describe('in-process structured output', () => { toolCallResponse('c1', STRUCTURED_OUTPUT_TOOL, { answer: 'not-a-number' }), toolCallResponse('c2', STRUCTURED_OUTPUT_TOOL, { answer: 7 }), ]) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(result.structured).toEqual({ answer: 7 }) expect(result.stopReason).toBe('completed') @@ -311,7 +292,7 @@ describe('in-process structured output', () => { textResponse('here is my answer in prose'), textResponse('MUST NOT BE CONSUMED'), ]) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(result.stopReason).toBe('error') expect(result.structured).toBeUndefined() @@ -325,7 +306,7 @@ describe('in-process structured output', () => { it('an errored child keeps its honest error result (no capture expected)', async () => { // Script exhaustion on the first call → the child turn errors. const { ctx, parent, adapter } = await setup([]) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(result.stopReason).toBe('error') expect(adapter.requests.length).toBe(1) @@ -334,12 +315,13 @@ describe('in-process structured output', () => { it('a cancel landing after a clean capture-less turn settles aborted, not error', async () => { const { ctx, parent } = await setup([textResponse('prose, no capture')]) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const controller = new AbortController() + const run = await ctx.subagents.start('spawn', structuredRequest(parent, { signal: controller.signal })) // Cancel synchronously inside the turn's end recording: the cancel // contract outranks the schema shortfall, so the result maps to aborted. ctx.on('session/event', (session, event) => { const child = ctx.agents.get(run.id) - if (session === child?.session && event.type === 'turn/end') run.cancel('cancelled at turn end') + if (session === child?.session && event.type === 'turn/end') controller.abort('cancelled at turn end') }) const result = await run.result expect(result.stopReason).toBe('aborted') @@ -348,20 +330,18 @@ describe('in-process structured output', () => { it('rejects a schema outside the subset loud, before any child exists', async () => { const { ctx, parent } = await setup([]) - expect(() => ctx.subagents.start('spawn', structuredRequest(parent, { + await expect(ctx.subagents.start('spawn', structuredRequest(parent, { outputSchema: { type: 'object', oneOf: [] } as unknown as StructuredOutputSchema, - }))).toThrow(/unsupported output schema/) + }))).rejects.toThrow(/unsupported output schema/) expect(ctx.agents.get(AgentId('parent'))).toBeDefined() }) - it('a schema carrying non-JSON values fails as OutputSchemaError, never as a raw clone error', async () => { + it('a schema carrying non-JSON values fails as OutputSchemaError at the validation boundary', async () => { const { ctx, parent } = await setup([]) - // Assertion runs BEFORE the defensive structuredClone: a function-valued - // annotation must surface as the subset violation it is, not escape as - // structuredClone's DataCloneError. - expect(() => ctx.subagents.start('spawn', structuredRequest(parent, { + // Semantic assertion runs before provider startup. + await expect(ctx.subagents.start('spawn', structuredRequest(parent, { outputSchema: { type: 'object', default: () => {} } as unknown as StructuredOutputSchema, - }))).toThrow(/unsupported output schema.*annotation must be JSON data/) + }))).rejects.toThrow(/unsupported output schema.*annotation must be JSON data/) }) it('a post-execute BLOCK on the capture call denies the capture: log and result agree on failure', async () => { @@ -377,7 +357,7 @@ describe('in-process structured output', () => { } return next() }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result // No capture was committed: the run reports the schema shortfall... expect(result.structured).toBeUndefined() @@ -403,7 +383,7 @@ describe('in-process structured output', () => { } return next() }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(result.stopReason).toBe('completed') expect(result.structured).toEqual({ answer: 8 }) @@ -415,7 +395,7 @@ describe('in-process structured output', () => { toolCallResponse('c1', STRUCTURED_OUTPUT_TOOL, { answer: 8 }), textResponse('capture was rejected'), ]) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) // Registered after attachment and prepended, so it wraps every listener // the child installed. It delegates first, then converts the apparent // capture success into the pipeline's authoritative failure. @@ -442,7 +422,7 @@ describe('in-process structured output', () => { // replace it (AgentOptions has no prompt field — the instruction is // per-request wire state added by the final-request listener). ctx.systemPrompt.section({ name: 'test:persona', order: 10, text: 'You are a counter.' }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) await run.result const childRequest = adapter.requests.at(-1)! expect(childRequest.system).toContain('You are a counter.') @@ -463,7 +443,7 @@ describe('in-process structured output', () => { return { logs: [], value: 'captured' } }, }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) // This listener is registered after the child's protection and prepended. // Service finalization still restores the stripped transport and prompt @@ -507,7 +487,7 @@ describe('in-process structured output', () => { } as never }, }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(result.structured).toBeUndefined() @@ -536,7 +516,7 @@ describe('in-process structured output', () => { ctx.on('tools/post-execute', (exec, _result, next) => exec.name === RUN_CODE_NAME ? Promise.resolve({ kind: 'block' as const, feedback: [{ type: 'text' as const, text: 'outer blocked' }] }) : next()) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(result.structured).toBeUndefined() @@ -553,7 +533,7 @@ describe('in-process structured output', () => { parent.send([{ type: 'text', text: 'hello' }]) await parent.whenIdle() expect(adapter.requests[0]!.system ?? '').not.toContain(STRUCTURED_OUTPUT_INSTRUCTION) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) await run.result // The loop always assembles a base prompt (the harness identity section), // so the instruction APPENDS — never replaces. @@ -584,7 +564,7 @@ describe('in-process structured output', () => { await parent.whenIdle() expect(toolNames(adapter.requests[0]!)).not.toContain(STRUCTURED_OUTPUT_TOOL) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) await run.result const childRequest = adapter.requests[1]! expect(toolNames(childRequest)).toContain(STRUCTURED_OUTPUT_TOOL) @@ -617,8 +597,8 @@ describe('in-process structured output', () => { return toolCallResponse('c2', STRUCTURED_OUTPUT_TOOL, args) }, ]) - const runA = ctx.subagents.start('spawn', structuredRequest(parent)) - const runB = ctx.subagents.start('spawn', structuredRequest(parent, { outputSchema: otherSchema })) + const runA = await ctx.subagents.start('spawn', structuredRequest(parent)) + const runB = await ctx.subagents.start('spawn', structuredRequest(parent, { outputSchema: otherSchema })) const [a, b] = await Promise.all([runA.result, runB.result]) expect(a.structured).toEqual({ answer: 1 }) expect(b.structured).toEqual({ verdict: 'real' }) @@ -647,7 +627,7 @@ describe('in-process structured output', () => { variables: { ...replaced.variables }, } }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(result.structured).toEqual({ answer: 5 }) const entries = adapter.requests[0]!.tools!.filter(tool => tool.name === STRUCTURED_OUTPUT_TOOL) @@ -672,7 +652,7 @@ describe('in-process structured output', () => { variables: { ...replaced.variables }, } }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(result.structured).toEqual({ answer: 5 }) const entry = adapter.requests[0]!.tools!.find(tool => tool.name === STRUCTURED_OUTPUT_TOOL) @@ -697,7 +677,7 @@ describe('in-process structured output', () => { execute: () => Promise.resolve([{ type: 'text', text: 'x' }]), }) ctx.systemPrompt.section({ name: 'after-band', order: 200, text: 'AFTER-BAND' }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) await run.result const request = adapter.requests[0]! const names = toolNames(request) @@ -731,7 +711,7 @@ describe('in-process structured output', () => { variables: { ...replaced.variables }, } }) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) const result = await run.result expect(result.structured).toEqual({ answer: 3 }) const request = adapter.requests[0]! @@ -759,7 +739,7 @@ describe('in-process structured output', () => { toolCallResponse('c1', STRUCTURED_OUTPUT_TOOL, { answer: 4 }), ]) expect(ctx.tools.get(STRUCTURED_OUTPUT_TOOL)).toBeUndefined() - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) // A backend hot-reload mid-run must not unregister the capture tool out // from under the live child: the registration rides the CHILD's fiber. await disposeProvider() @@ -800,7 +780,7 @@ describe('in-process structured output', () => { const { ctx, parent } = await setup([ toolCallResponse('c1', STRUCTURED_OUTPUT_TOOL, { answer: 1 }), ]) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) // A prepended post-execute listener blocks the first capture without // delegating. The final-result notification discards that execution's // stage when it observes the error. @@ -841,7 +821,7 @@ describe('in-process structured output', () => { const { ctx, parent } = await setup([ toolCallResponse('c1', STRUCTURED_OUTPUT_TOOL, { answer: 1 }), ]) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) // Block the first capture after its body stages a value. Its final error // discards that execution's stage. let blocks = 1 @@ -879,7 +859,7 @@ describe('in-process structured output', () => { const { ctx, parent } = await setup([ toolCallResponse('c1', STRUCTURED_OUTPUT_TOOL, { answer: 1 }), ]) - const run = ctx.subagents.start('spawn', structuredRequest(parent)) + const run = await ctx.subagents.start('spawn', structuredRequest(parent)) // Discard the first capture's stage via a final post-execute block. let blocks = 1 ctx.on('tools/post-execute', (exec, _result, next) => { diff --git a/packages/subagent/subagent-inprocess/tests/subagent-inprocess.spec.ts b/packages/subagent/subagent-inprocess/tests/subagent-inprocess.spec.ts index 8f710237ab..0005246523 100644 --- a/packages/subagent/subagent-inprocess/tests/subagent-inprocess.spec.ts +++ b/packages/subagent/subagent-inprocess/tests/subagent-inprocess.spec.ts @@ -1,25 +1,18 @@ -import { describe, expect, it, vi } from 'vitest' -import { Context, type Fiber } from 'cordis' +import { describe, expect, it } from 'vitest' +import { Context } from 'cordis' import LlmService from '@deepseek-ai/dsh-llm' -import SessionStore, { type SessionEvent } from '@deepseek-ai/dsh-session' +import SessionStore from '@deepseek-ai/dsh-session' import SystemPrompt from '@deepseek-ai/dsh-system-prompt' import ToolRegistry from '@deepseek-ai/dsh-tools' import AgentRegistry, { AgentId, type Agent } from '@deepseek-ai/dsh-agent' import AgentLoop from '@deepseek-ai/dsh-agent-loop' import * as Invariants from '@deepseek-ai/dsh-invariants' -import SubagentService, { type SubagentStartRequest } from '@deepseek-ai/dsh-subagent' +import SubagentService from '@deepseek-ai/dsh-subagent' import { MockAdapter, textResponse } from '../../../core/agent-loop/tests/mock-adapter.ts' -import { depthOf, type InProcessRunOptions, SubagentDepthError, startInProcessRun } from '../src/index.ts' +import { depthOf, SubagentDepthError, startInProcessRun } from '../src/index.ts' type Script = ConstructorParameters[0] -/** - * Drives the shared in-process run driver DIRECTLY (no provider package), so the - * driver's own contract — depth read/cap, the one-shot drive, the result read — - * is covered independently of which backend (spawn/fork) calls it. The only - * mocked boundary is the model; the real agent loop, SubagentService, and - * dsh-invariants are mounted, so a malformed child session log fails the test. - */ async function setup(script: Script) { const ctx = new Context() await ctx.plugin(LlmService) @@ -35,393 +28,127 @@ async function setup(script: Script) { return { ctx, parent } } -function text(blocks: { type: string; text?: string }[]): string { - return blocks.filter(b => b.type === 'text').map(b => b.text).join('') +function request(parent: Agent, signal = new AbortController().signal) { + return { prompt: [{ type: 'text' as const, text: 'child task' }], parent, signal } +} + +function text(blocks: readonly { type: string; text?: string }[]): string { + return blocks.filter(block => block.type === 'text').map(block => block.text).join('') } describe('depthOf', () => { - it('reads 0 for an agent with no subagentDepth, the set value otherwise', async () => { + it('reads zero for a top-level agent and an explicit child depth', async () => { const { parent } = await setup([]) expect(depthOf(parent)).toBe(0) - const withDepth = { options: { subagentDepth: 3 } } as unknown as Agent - expect(depthOf(withDepth)).toBe(3) + expect(depthOf({ options: { subagentDepth: 3 } } as unknown as Agent)).toBe(3) }) - it.each([ - { label: 'null', value: null as unknown as number }, - { label: 'a string', value: '1' as unknown as number }, - { label: 'NaN', value: Number.NaN }, - { label: 'positive infinity', value: Number.POSITIVE_INFINITY }, - { label: 'negative infinity', value: Number.NEGATIVE_INFINITY }, - { label: 'a fraction', value: 1.5 }, - { label: 'a negative integer', value: -1 }, - { label: 'negative zero', value: -0 }, - { label: 'an unsafe integer', value: Number.MAX_SAFE_INTEGER + 1 }, - ])('rejects subagentDepth=$label', ({ value }) => { - const agent = { options: { subagentDepth: value } } as unknown as Agent - expect(() => depthOf(agent)).toThrow('agent subagentDepth must be a non-negative safe integer') + it.each([Number.NaN, 1.5, -1, -0, Number.MAX_SAFE_INTEGER + 1])('rejects malformed depth %s', (value) => { + expect(() => depthOf({ options: { subagentDepth: value } } as unknown as Agent)) + .toThrow('non-negative safe integer') }) }) describe('startInProcessRun', () => { - it.each([ - { label: 'null', value: null as unknown as number }, - { label: 'a string', value: '1' as unknown as number }, - { label: 'NaN', value: Number.NaN }, - { label: 'positive infinity', value: Number.POSITIVE_INFINITY }, - { label: 'negative infinity', value: Number.NEGATIVE_INFINITY }, - { label: 'a fraction', value: 1.5 }, - { label: 'a negative integer', value: -1 }, - { label: 'negative zero', value: -0 }, - { label: 'an unsafe integer', value: Number.MAX_SAFE_INTEGER + 1 }, - ])('rejects maxDepth=$label before acquiring run ownership', async ({ value }) => { - const { ctx, parent } = await setup([]) - expect(() => startInProcessRun(ctx, { - prompt: [{ type: 'text', text: 'must never start' }], - parent, - maxDepth: value, - }, {})).toThrow('subagent maxDepth must be a non-negative safe integer') - }) - - it('rejects a non-string persona before acquiring run ownership', async () => { - const { ctx, parent } = await setup([]) - expect(() => startInProcessRun(ctx, { - prompt: [{ type: 'text', text: 'must never start' }], - parent, - persona: 42 as unknown as string, - }, {})).toThrow('subagent persona must be a string') - }) - - it('rejects a child depth with no safe-integer representation before acquiring run ownership', async () => { - const { ctx } = await setup([]) - const parent = { - options: { subagentDepth: Number.MAX_SAFE_INTEGER }, - } as unknown as Agent - - expect(() => startInProcessRun(ctx, { - prompt: [{ type: 'text', text: 'must never start' }], - parent, - }, {})).toThrow(RangeError) - }) - - it('rejects a non-JSON prompt before acquiring any run ownership', async () => { - const { ctx, parent } = await setup([]) - expect(() => startInProcessRun(ctx, { - prompt: [{ type: 'text', text: Number.NaN as unknown as string }], - parent, - }, {})).toThrow('subagent prompt must be losslessly JSON-serializable') - }) - - it('reads each prompt value once before asynchronous child creation', async () => { - const { ctx, parent } = await setup([]) - let reads = 0 - const prompt = [{ - type: 'text' as const, - get text(): string { - reads += 1 - return reads === 1 ? 'valid during pre-check' : Number.NaN as unknown as string - }, - }] - - const run = startInProcessRun(ctx, { prompt, parent }, {}) - expect(reads).toBe(1) - await run.dispose() - }) - - it('reads each public request and seed option field once', async () => { - const { ctx, parent } = await setup([]) - const reads = { prompt: 0, toolFilter: 0, maxDepth: 0, outputSchema: 0, agentOptions: 0, persona: 0, seed: 0 } - const request = Object.defineProperties({ parent }, { - prompt: { enumerable: true, get: () => { reads.prompt += 1; return [{ type: 'text', text: 'accepted' }] } }, - toolFilter: { enumerable: true, get: () => { reads.toolFilter += 1; return reads.toolFilter === 1 ? undefined : { deny: ['ghost'] } } }, - maxDepth: { enumerable: true, get: () => { reads.maxDepth += 1; return reads.maxDepth === 1 ? undefined : 0 } }, - outputSchema: { enumerable: true, get: () => { reads.outputSchema += 1; return undefined } }, - agentOptions: { enumerable: true, get: () => { reads.agentOptions += 1; return {} } }, - persona: { enumerable: true, get: () => { reads.persona += 1; return undefined } }, - }) as unknown as SubagentStartRequest - const options = Object.defineProperty({}, 'seed', { - enumerable: true, - get: () => { reads.seed += 1; return reads.seed === 1 ? undefined : [] }, - }) as InProcessRunOptions - - const run = startInProcessRun(ctx, request, options) - - expect(reads).toEqual({ prompt: 1, toolFilter: 1, maxDepth: 1, outputSchema: 1, agentOptions: 1, persona: 1, seed: 1 }) - await run.dispose() - }) - - it('rejects an exotic seed before asynchronous owner setup can sanitize it', async () => { - const { ctx, parent } = await setup([]) - class ExoticSeedEvent { - readonly type = 'turn/start' - readonly seq = 0 - readonly time = 1 - readonly data = { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } } - } - - expect(() => startInProcessRun(ctx, { - prompt: [{ type: 'text', text: 'accepted' }], - parent, - }, { seed: [new ExoticSeedEvent()] as unknown as SessionEvent[] })) - .toThrow(/subagent seed must be losslessly JSON-serializable/) - }) - - it.each([ - { - label: 'tool filter', - overrides: { toolFilter: { deny: [Number.NaN as unknown as string] } }, - message: 'subagent tool filter must be losslessly JSON-serializable', - }, - { - label: 'agent options', - overrides: { agentOptions: { model: Number.NaN as unknown as string } }, - message: 'subagent agent options must be losslessly JSON-serializable', - }, - { - label: 'output schema', - overrides: { - outputSchema: { - type: 'object', - properties: { answer: { type: Number.NaN } }, - } as unknown as NonNullable, - }, - message: 'schema annotation must be JSON data', - }, - ])('rejects non-JSON $label before asynchronous child creation', async ({ overrides, message }) => { - const { ctx, parent } = await setup([]) - - expect(() => startInProcessRun(ctx, { - prompt: [{ type: 'text', text: 'accepted' }], - parent, - ...overrides, - }, {})).toThrow(message) - }) - - it('rejects a non-JSON model inherited from the parent before child creation', async () => { - const { ctx, parent } = await setup([]) - const invalidParent = { - options: { ...parent.options, model: Number.NaN as unknown as string }, - session: parent.session, - } as unknown as Agent - - expect(() => startInProcessRun(ctx, { - prompt: [{ type: 'text', text: 'accepted' }], - parent: invalidParent, - }, {})).toThrow('subagent agent options must be losslessly JSON-serializable') - }) - - it('rejects when the run-owner fiber settles without installing its context', async () => { - const { ctx, parent } = await setup([]) - function inertOwner(): void {} - const inertFiber = ctx.plugin(inertOwner) - await inertFiber - const parentWithoutOwnerContext = { - options: parent.options, - session: parent.session, - ctx: { plugin: () => inertFiber }, - } as unknown as Agent - - const run = startInProcessRun(ctx, { - prompt: [{ type: 'text', text: 'must never start' }], - parent: parentWithoutOwnerContext, - }, {}) - await expect(run.result).rejects.toThrow('subagent run owner became inactive before child creation') - await run.dispose() - }) - - it('normalizes a non-Error thrown while installing the run-owner fiber', async () => { - const { ctx, parent } = await setup([]) - const setupFailure = 'non-Error owner setup failure' - const parentWithFailingOwnerSetup = { - options: parent.options, - session: parent.session, - ctx: { plugin: () => { throw setupFailure } }, - } as unknown as Agent - - const run = startInProcessRun(ctx, { - prompt: [{ type: 'text', text: 'must never start' }], - parent: parentWithFailingOwnerSetup, - }, {}) - await expect(run.result).rejects.toMatchObject({ - message: 'subagent run owner setup failed with a non-Error value', - cause: setupFailure, - }) - await run.dispose() - }) - - it('normalizes a non-Error rejected by asynchronous child creation', async () => { - const { ctx, parent } = await setup([]) - const creationFailure = 'non-Error child creation failure' - function inertOwner(): void {} - const ownerFiber = ctx.plugin(inertOwner) - await ownerFiber - const rejectWithNonError = (): Promise => { - // Deliberately violate the promise contract to exercise boundary normalization. - // eslint-disable-next-line @typescript-eslint/prefer-promise-reject-errors - return Promise.reject(creationFailure) - } - const rejectingOwnerCtx = { - agents: { create: rejectWithNonError }, - } as unknown as Context - const parentWithRejectingFactory = { - options: parent.options, - session: parent.session, - ctx: { - plugin(plugin: (inner: Context) => void) { - plugin(rejectingOwnerCtx) - return ownerFiber - }, - }, - } as unknown as Agent - - const run = startInProcessRun(ctx, { - prompt: [{ type: 'text', text: 'must never start' }], - parent: parentWithRejectingFactory, - }, {}) - await expect(run.result).rejects.toMatchObject({ - message: 'subagent child creation failed with a non-Error value', - cause: creationFailure, - }) - await run.dispose() - }) - - it('follows owner-fiber inertia when raw teardown was already in flight', async () => { - const { ctx, parent } = await setup([]) - const gate = Promise.withResolvers() - let inertia: Promise | undefined = gate.promise - const fakeFiber = { - dispose: vi.fn(() => undefined), - get inertia() { return inertia }, - } as unknown as Fiber & PromiseLike - const rejectingOwnerCtx = { - agents: { create: () => Promise.reject(new Error('creation stopped by teardown')) }, - } as unknown as Context - const parentWithDisposingOwner = { - options: parent.options, - session: parent.session, - ctx: { - plugin(plugin: (inner: Context) => void) { - plugin(rejectingOwnerCtx) - return fakeFiber - }, - }, - } as unknown as Agent - const run = startInProcessRun(ctx, { - prompt: [{ type: 'text', text: 'must never start' }], - parent: parentWithDisposingOwner, - }, {}) - - let settled = false - const disposing = run.dispose().then(() => { settled = true }) - await Promise.resolve() - await Promise.resolve() - expect(fakeFiber.dispose).toHaveBeenCalledOnce() - expect(settled).toBe(false) - - inertia = undefined - gate.resolve(undefined) - await disposing - await expect(run.result).resolves.toEqual({ output: [], stopReason: 'aborted' }) - }) - - it('observes detached pre-readiness teardown failure and reports it to explicit dispose', async () => { - const { ctx, parent } = await setup([]) - function inertOwner(): void {} - const ownerFiber = ctx.plugin(inertOwner) - await ownerFiber - const disposeFailure = new Error('owner dispose exploded') - const disposeSpy = vi.spyOn(ownerFiber, 'dispose').mockImplementation(() => { throw disposeFailure }) - const rejectingOwnerCtx = { - agents: { create: () => Promise.reject(new Error('creation stopped by cancellation')) }, - } as unknown as Context - const parentWithFailingTeardown = { - options: parent.options, - session: parent.session, - ctx: { - plugin(plugin: (inner: Context) => void) { - plugin(rejectingOwnerCtx) - return ownerFiber - }, - }, - } as unknown as Agent - const run = startInProcessRun(ctx, { - prompt: [{ type: 'text', text: 'must never start' }], - parent: parentWithFailingTeardown, - }, {}) - - run.cancel('cancel before readiness') - await expect(run.result).resolves.toEqual({ output: [], stopReason: 'aborted' }) - await expect(run.dispose()).rejects.toBe(disposeFailure) - disposeSpy.mockRestore() - await ownerFiber.dispose() - }) - - it('does not attach an abort listener when provider ownership is already inactive', async () => { - const { ctx, parent } = await setup([]) - let providerCtx: Context | undefined - function provider(inner: Context): void { providerCtx = inner } - const providerFiber = await ctx.plugin(provider) - await providerFiber.dispose() - if (providerCtx === undefined) throw new Error('provider context was not captured') - const inactiveProviderCtx = providerCtx - - const controller = new AbortController() - const addListener = vi.spyOn(controller.signal, 'addEventListener') - expect(() => startInProcessRun(inactiveProviderCtx, { - prompt: [{ type: 'text', text: 'must never start' }], - parent, - signal: controller.signal, - }, {})).toThrow(/inactive context/) - expect(addListener).not.toHaveBeenCalled() - }) - - it('drives a fresh child (no seed) to completion and returns its output', async () => { - const { ctx, parent } = await setup([textResponse('driver child answer')]) - const run = startInProcessRun(ctx, { prompt: [{ type: 'text', text: 'do X' }], parent }, {}) + it('returns only after publication, drives a fresh child, and disposes it', async () => { + const { ctx, parent } = await setup([textResponse('driver answer')]) + const run = await startInProcessRun(request(parent), {}) + expect(ctx.agents.get(run.id)).toBeDefined() const result = await run.result expect(result.stopReason).toBe('completed') - expect(text(result.output)).toBe('driver child answer') + expect(text(result.output)).toBe('driver answer') expect(depthOf(ctx.agents.get(run.id)!)).toBe(1) await run.dispose() - }) - - it('snapshots the prompt before asynchronous child creation', async () => { - const { ctx, parent } = await setup([textResponse('done')]) - const prompt = [{ type: 'text' as const, text: 'original prompt' }] - const run = startInProcessRun(ctx, { prompt, parent }, {}) - - prompt[0]!.text = 'mutated after start' - prompt.push({ type: 'text', text: 'also injected' }) - await run.result - - const child = ctx.agents.get(run.id)! - const userMessage = child.session.events.find(event => event.type === 'user/message') - expect(userMessage?.type === 'user/message' && userMessage.data.content) - .toEqual([{ type: 'text', text: 'original prompt' }]) await run.dispose() + expect(ctx.agents.get(run.id)).toBeUndefined() }) - it('throws SubagentDepthError when the child would exceed maxDepth', async () => { - const { ctx, parent } = await setup([]) - expect(() => startInProcessRun(ctx, { prompt: [{ type: 'text', text: 'p' }], parent, maxDepth: 0 }, {})) - .toThrow(SubagentDepthError) - }) - - it('seeds the child session when a seed is supplied', async () => { - // Drive the parent through one real turn, then seed the child with that - // completed-turn prefix — the child must SEE the parent's history but its - // result is scoped to its OWN events (not the seeded parent message). - const { ctx, parent } = await setup([textResponse('parent turn'), textResponse('seeded child reply')]) - parent.send([{ type: 'text', text: 'parent q' }]) + it('seeds a forked child but reads only the child-owned output', async () => { + const { ctx, parent } = await setup([textResponse('parent answer'), textResponse('child answer')]) + parent.send([{ type: 'text', text: 'parent question' }]) await parent.whenIdle() const seed = parent.session.events.slice() - const run = startInProcessRun(ctx, { prompt: [{ type: 'text', text: 'child q' }], parent }, { seed }) + const run = await startInProcessRun(request(parent), { seed }) const result = await run.result - expect(result.stopReason).toBe('completed') - expect(text(result.output)).toBe('seeded child reply') + expect(text(result.output)).toBe('child answer') const child = ctx.agents.get(run.id)! - // The child inherited the parent's prefix. - expect(child.session.events.slice(0, seed.length).some(e => e.type === 'user/message')).toBe(true) + expect(child.session.header.seedLength).toBe(seed.length) + expect(child.session.events.slice(0, seed.length)).toEqual(seed) await run.dispose() }) + + it('rejects invalid and exceeded depth before publication', async () => { + const { parent } = await setup([]) + await expect(startInProcessRun({ ...request(parent), maxDepth: -1 }, {})) + .rejects.toThrow('non-negative safe integer') + await expect(startInProcessRun({ ...request(parent), maxDepth: 0 }, {})) + .rejects.toBeInstanceOf(SubagentDepthError) + const maxParent = { options: { subagentDepth: Number.MAX_SAFE_INTEGER } } as unknown as Agent + await expect(startInProcessRun(request(maxParent), {})).rejects.toBeInstanceOf(RangeError) + }) + + it('rejects an already-aborted request without publishing a child', async () => { + const { ctx, parent } = await setup([]) + const beforeAgents = ctx.agents.list().length + const beforeSessions = ctx.sessions.list().length + const controller = new AbortController() + controller.abort('too late') + await expect(startInProcessRun(request(parent, controller.signal), {})) + .rejects.toThrow('aborted before child publication') + expect(ctx.agents.list()).toHaveLength(beforeAgents) + expect(ctx.sessions.list()).toHaveLength(beforeSessions) + }) + + it('uses the request signal after publication and dispose as cancellation paths', async () => { + const { parent } = await setup(['hang', 'hang']) + const controller = new AbortController() + const signalled = await startInProcessRun(request(parent, controller.signal), {}) + await new Promise(resolve => setTimeout(resolve, 30)) + controller.abort('stop child') + await expect(signalled.result).resolves.toMatchObject({ stopReason: 'aborted' }) + await signalled.dispose() + + const disposed = await startInProcessRun(request(parent), {}) + await new Promise(resolve => setTimeout(resolve, 30)) + await disposed.dispose() + await expect(disposed.result).resolves.toMatchObject({ stopReason: 'aborted' }) + }) + + it('cleans a failed unpublished setup before rejecting', async () => { + const { ctx, parent } = await setup([]) + const beforeAgents = ctx.agents.list().length + const beforeSessions = ctx.sessions.list().length + await expect(startInProcessRun({ + ...request(parent), + toolFilter: { deny: ['unknown-tool'] }, + }, {})).rejects.toThrow('unknown global tool') + expect(ctx.agents.list()).toHaveLength(beforeAgents) + expect(ctx.sessions.list()).toHaveLength(beforeSessions) + }) + + it('closes the abort handoff after the factory detaches its creation listener', async () => { + const { ctx, parent } = await setup([]) + const controller = new AbortController() + const beforeAgents = ctx.agents.list().length + const beforeSessions = ctx.sessions.list().length + const parentWithAbortAtHandoff = { + options: parent.options, + session: parent.session, + ctx: { + agents: { + create: async (options: Parameters[0]) => { + const handle = await ctx.agents.create(options) + // `create()` has detached its creation-only listener, but the + // provider continuation has not installed its live-run listener. + controller.abort('handoff race') + return handle + }, + }, + }, + } as unknown as Agent + await expect(startInProcessRun(request(parentWithAbortAtHandoff, controller.signal), {})) + .rejects.toThrow('aborted before child publication') + expect(ctx.agents.list()).toHaveLength(beforeAgents) + expect(ctx.sessions.list()).toHaveLength(beforeSessions) + }) }) diff --git a/packages/subagent/subagent-spawn/README.md b/packages/subagent/subagent-spawn/README.md index 07c237b303..3c0f739ecf 100644 --- a/packages/subagent/subagent-spawn/README.md +++ b/packages/subagent/subagent-spawn/README.md @@ -1,16 +1,16 @@ # @deepseek-ai/dsh-subagent-spawn -The in-process **spawn** subagent backend: a [`SubagentProvider`](../subagent/README.md) that runs each child as a **fresh** child [`Agent`](../../core/agent) on the same cordis context (`ctx.agents`) — its own session, its own (or the parent's) model, zero inherited conversation. The cheapest transport, reusing the agent factory's quiescent [`AgentHandle`](../../core/agent) teardown. +The spawn provider creates a fresh child `Agent` in the current process. The child has its own session, sees no parent conversation history, and reuses the host's agent factory and LLM/tool services. -The run mechanics live in the shared [`@deepseek-ai/dsh-subagent-inprocess`](../subagent-inprocess/README.md) driver (`startInProcessRun`); this backend just passes **no seed** (a fresh child). The [fork](../subagent-fork/README.md) backend is an independent peer over the same driver — neither knows about the other. +## Behavior -## What it does +`start(request)` delegates to [`startInProcessRun`](../subagent-inprocess/README.md) with no seed and awaits publication before returning. The child receives parent working-directory/session lineage and inherits the parent model unless overridden, but starts with an empty conversation. -`start(request)` delegates to `startInProcessRun(ctx, request, {})` with no seed: a fresh child agent with the parent's `cwd`/`parentSession` lineage and (by default) the parent's model. The driver creates one run-owner fiber under `parent.ctx`; parent teardown, this provider's teardown, manual disposal, and cancellation before readiness all converge there before child publication. Its `run.started` boundary resolves only after the fresh child is published, so `subagent/start` observers see a live registry entry; a same-tick cancel deactivates the unpublished transaction instead, rejects readiness, resolves the result as `aborted`, and emits no agent/session or subagent lifecycle. See the [driver README](../subagent-inprocess/README.md) for the full lifecycle (depth check, one-shot drive, result read, dispose). +The shared driver owns depth checking, persona and tool-filter setup, structured output, required-signal cancellation, one-shot execution, result reading, and quiescent disposal. A startup rejection leaves no published child; provider unload after fulfillment does not revoke the holder-owned run. ## Capabilities -`{ outputSchema: true, depthLimit: true, toolFilter: true, persona: true }`. It constructs the child, so it enforces a recursion cap and composes the child's persona, global-tool restriction, and [structured runtime](../subagent-inprocess/README.md) inside the agent-creation setup window. At apply it registers this one named provider on `ctx.subagents`; per-run contributions belong to each child's scope. +Spawn advertises `{ outputSchema: true, depthLimit: true, toolFilter: true, persona: true }` because it controls the child's creation window and can enforce all four features. ## Config diff --git a/packages/subagent/subagent-spawn/src/index.ts b/packages/subagent/subagent-spawn/src/index.ts index 835cf34aae..6d2d8f3a11 100644 --- a/packages/subagent/subagent-spawn/src/index.ts +++ b/packages/subagent/subagent-spawn/src/index.ts @@ -53,16 +53,16 @@ class SpawnProvider implements SubagentProvider { // Context contract: a spawned child starts fresh — it never sees the parent conversation. readonly inheritsParentContext = false - constructor(readonly name: string, private readonly ctx: Context) {} + constructor(readonly name: string) {} start(request: SubagentStartRequest) { // Fresh child: no seed. The shared driver mints ids, stamps cwd/lineage/ // depth, drives the one-shot (including the structured capture when the // request carries an outputSchema), and maps the result. - return startInProcessRun(this.ctx, request, {}) + return startInProcessRun(request, {}) } } export function apply(ctx: Context, config: Config): void { - ctx.subagents.registerProvider(new SpawnProvider(config.providerName, ctx)) + ctx.subagents.registerProvider(new SpawnProvider(config.providerName)) } diff --git a/packages/subagent/subagent-spawn/tests/subagent-spawn.spec.ts b/packages/subagent/subagent-spawn/tests/subagent-spawn.spec.ts index f00eddeb25..9e4e489882 100644 --- a/packages/subagent/subagent-spawn/tests/subagent-spawn.spec.ts +++ b/packages/subagent/subagent-spawn/tests/subagent-spawn.spec.ts @@ -9,7 +9,7 @@ import AgentRegistry, { AgentId } from '@deepseek-ai/dsh-agent' import { SessionId } from '@deepseek-ai/dsh-session' import AgentLoop from '@deepseek-ai/dsh-agent-loop' import * as Invariants from '@deepseek-ai/dsh-invariants' -import SubagentService from '@deepseek-ai/dsh-subagent' +import SubagentService, { type SubagentStartRequest } from '@deepseek-ai/dsh-subagent' import { MockAdapter, maxTokensResponse, textResponse, toolCallResponse } from '../../../core/agent-loop/tests/mock-adapter.ts' import * as spawn from '../src/index.ts' import { depthOf, STRUCTURED_OUTPUT_TOOL, SubagentDepthError } from '@deepseek-ai/dsh-subagent-inprocess' @@ -44,11 +44,15 @@ function text(blocks: { type: string; text?: string }[]): string { return blocks.filter(b => b.type === 'text').map(b => b.text).join('') } +function start(ctx: Context, provider: string, request: Omit & { signal?: AbortSignal }) { + return ctx.subagents.start(provider, { signal: request.signal ?? new AbortController().signal, ...request }) +} + describe('dsh-subagent-spawn', () => { it('runs a fresh child to completion and returns its final assistant output', async () => { // One model call for the child: a plain text answer. const { ctx, parent } = await setup([textResponse('child answer')]) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'do X' }], parent }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'do X' }], parent }) const result = await run.result expect(result.stopReason).toBe('completed') expect(text(result.output)).toBe('child answer') @@ -62,11 +66,11 @@ describe('dsh-subagent-spawn', () => { if (info.provider === 'spawn') childAtStart = ctx.agents.get(info.id) }) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'do X' }], parent }) + const starting = start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'do X' }], parent }) // Creation is asynchronous; no lifecycle claim is made while the child is // still inside its unpublished setup transaction. expect(childAtStart).toBeUndefined() - await run.started + const run = await starting expect(childAtStart).toBe(ctx.agents.get(run.id)) expect(childAtStart?.id).toBe(run.id) @@ -76,7 +80,7 @@ describe('dsh-subagent-spawn', () => { it('gives the child its OWN session (not the parent\'s), with parentSession lineage', async () => { const { ctx, parent } = await setup([textResponse('hi')]) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) await run.result const child = ctx.agents.get(run.id)! expect(child.session.header.id).not.toBe(parent.session.header.id) @@ -92,7 +96,7 @@ describe('dsh-subagent-spawn', () => { const parentEventCount = parent.session.events.length expect(parentEventCount).toBeGreaterThan(0) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'child prompt' }], parent }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'child prompt' }], parent }) await run.result const child = ctx.agents.get(run.id)! // The child's first user/message is its OWN prompt, not the parent's history. @@ -103,7 +107,7 @@ describe('dsh-subagent-spawn', () => { it('disposes the child to quiescence (agent removed from the registry)', async () => { const { ctx, parent } = await setup([textResponse('x')]) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) await run.result expect(ctx.agents.get(run.id)).toBeDefined() await run.dispose() @@ -114,7 +118,7 @@ describe('dsh-subagent-spawn', () => { it('stamps child depth = parent depth + 1 (via the merged AgentOptions field)', async () => { const { ctx, parent } = await setup([textResponse('x')]) expect(depthOf(parent)).toBe(0) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) await run.result const child = ctx.agents.get(run.id)! expect(depthOf(child)).toBe(1) @@ -124,13 +128,13 @@ describe('dsh-subagent-spawn', () => { it('refuses to spawn past maxDepth (depthLimit capability)', async () => { const { ctx, parent } = await setup([]) // parent is depth 0, child would be depth 1 — cap at 0 forbids any child. - expect(() => ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent, maxDepth: 0 })) - .toThrow(SubagentDepthError) + await expect(start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent, maxDepth: 0 })) + .rejects.toThrow(SubagentDepthError) }) it('maps a child that hit its token ceiling to stopReason "max-tokens"', async () => { const { ctx, parent } = await setup([maxTokensResponse('cut off')]) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) const result = await run.result expect(result.stopReason).toBe('max-tokens') await run.dispose() @@ -140,14 +144,14 @@ describe('dsh-subagent-spawn', () => { // Empty script: the child's first model call throws "script exhausted", the // turn ends `error`, and there is no assistant/message → empty output. const { ctx, parent } = await setup([]) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) const result = await run.result expect(result.stopReason).toBe('error') expect(result.output).toEqual([]) await run.dispose() }) - it('settles aborted (without running the child) when the request signal is ALREADY aborted', async () => { + it('rejects without publishing when the request signal is already aborted', async () => { // Regression: a signal aborted BEFORE the run starts never fires an `abort` // event, so the listener can't catch it. The driver must check the // already-aborted case up front and settle `aborted` without running the @@ -156,15 +160,12 @@ describe('dsh-subagent-spawn', () => { const controller = new AbortController() controller.abort() const { ctx, parent } = await setup([]) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent, signal: controller.signal }) - const result = await run.result - expect(result.stopReason).toBe('aborted') - expect(result.output).toEqual([]) - await run.dispose() + await expect(start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent, signal: controller.signal })) + .rejects.toThrow('aborted before child publication') }) - it('same-tick cancellation rejects readiness and prevents child publication', async () => { - // Regression: cancellation before readiness used to set a flag but let the + it('same-tick cancellation rejects start and prevents child publication', async () => { + // Regression: cancellation before publication used to set a flag but let the // async factory publish a child anyway, so `started` fulfilled and lifecycle // observers saw an agent for an attempt the caller had already cancelled. // The empty script also proves no model turn can run. @@ -177,14 +178,12 @@ describe('dsh-subagent-spawn', () => { ctx.on('agent/session-start', () => void published.push('agent/session-start')) ctx.on('subagent/start', () => void published.push('subagent/start')) ctx.on('subagent/end', () => void published.push('subagent/end')) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) - run.cancel('early') + const controller = new AbortController() + const starting = start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent, signal: controller.signal }) + controller.abort('early') - await expect(run.started).rejects.toThrow() - await expect(run.result).resolves.toEqual({ output: [], stopReason: 'aborted' }) - await run.dispose() + await expect(starting).rejects.toThrow() await Promise.resolve() - expect(ctx.agents.get(run.id)).toBeUndefined() expect(ctx.agents.list()).toHaveLength(beforeAgents) expect(ctx.sessions.list()).toHaveLength(beforeSessions) expect(published).toEqual([]) @@ -192,10 +191,9 @@ describe('dsh-subagent-spawn', () => { it('a cancel from agent/queued maps a no-turn child log to aborted', async () => { const { ctx, parent } = await setup([]) - ctx.on('agent/queued', (agent) => { - if (agent.id === run.id) run.cancel('queued-window') - }) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) + const controller = new AbortController() + ctx.on('agent/queued', () => { controller.abort('queued-window') }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent, signal: controller.signal }) const result = await run.result expect(result).toMatchObject({ stopReason: 'aborted', output: [] }) const child = ctx.agents.get(run.id)! @@ -203,30 +201,11 @@ describe('dsh-subagent-spawn', () => { await run.dispose() }) - it('dispose during async child creation waits for rollback and leaves no orphan', async () => { - const { ctx, parent } = await setup([]) - const beforeAgents = ctx.agents.list().length - const beforeSessions = ctx.sessions.list().length - const published: string[] = [] - ctx.on('session/created', () => void published.push('session/created')) - ctx.on('agent/created', () => void published.push('agent/created')) - ctx.on('agent/session-start', () => void published.push('agent/session-start')) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) - - // Same tick: the factory has reserved ids and entered its async setup - // transaction, but has not published the child yet. - await run.dispose() - await expect(run.result).resolves.toMatchObject({ stopReason: 'aborted', output: [] }) - expect(ctx.agents.list()).toHaveLength(beforeAgents) - expect(ctx.sessions.list()).toHaveLength(beforeSessions) - expect(published).toEqual([]) - }) - it('cancelling a running child settles the run as aborted (the abort bridge + cancel())', async () => { // 'hang' makes the child's model stream one chunk then wait until aborted. const controller = new AbortController() const { ctx, parent } = await setup(['hang']) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent, signal: controller.signal }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent, signal: controller.signal }) // Let the child's turn start, then abort via the request signal (the // backend bridges it to child.cancel()). await new Promise(r => setTimeout(r, 30)) @@ -236,29 +215,18 @@ describe('dsh-subagent-spawn', () => { await run.dispose() }) - it('run.cancel() also cancels the child directly', async () => { + it('dispose cancels the child and reaches quiescence', async () => { const { ctx, parent } = await setup(['hang']) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) await new Promise(r => setTimeout(r, 30)) - run.cancel('test cancel') + await run.dispose() const result = await run.result expect(result.stopReason).toBe('aborted') - await run.dispose() - }) - - it('run.cancel() with no reason uses the default cancel reason', async () => { - const { ctx, parent } = await setup(['hang']) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) - await new Promise(r => setTimeout(r, 30)) - run.cancel() - const result = await run.result - expect(result.stopReason).toBe('aborted') - await run.dispose() }) it('does not expose the optional runtime methods (sendMessage/resume) in this cut', async () => { const { ctx, parent } = await setup([textResponse('x')]) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent }) expect('sendMessage' in run).toBe(false) expect('resume' in run).toBe(false) await run.result @@ -274,7 +242,7 @@ describe('dsh-subagent-spawn', () => { meta: { cwd: '/tmp/parent-workspace' }, agentOptions: { model: 'mock' }, }) - const run = ctx.subagents.start('spawn', { prompt: [{ type: 'text', text: 'p' }], parent: parentHandle.agent }) + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent: parentHandle.agent }) await run.result const child = ctx.agents.get(run.id)! expect(child.session.header.cwd).toBe('/tmp/parent-workspace') @@ -291,7 +259,7 @@ describe('dsh-subagent-spawn', () => { agentOptions: {}, }) // The request supplies the child's model explicitly. - const run = ctx.subagents.start('spawn', { + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'p' }], parent: parentHandle.agent, agentOptions: { model: 'mock' }, @@ -323,7 +291,7 @@ describe('dsh-subagent-spawn', () => { const { ctx, parent } = await setup([ toolCallResponse('c1', STRUCTURED_OUTPUT_TOOL, { answer: 42 }), ]) - const run = ctx.subagents.start('spawn', { + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'produce the answer' }], parent, outputSchema: { type: 'object', properties: { answer: { type: 'number' } }, required: ['answer'] }, @@ -336,7 +304,7 @@ describe('dsh-subagent-spawn', () => { await run.dispose() }) - it('a backend unload mid-structured-run settles the run and releases the runtime', async () => { + it('a backend unload does not revoke an accepted holder-owned run', async () => { // Rebuild the stack by hand so we hold the backend's fiber. const ctx = new Context() const adapter = new MockAdapter(['hang']) @@ -351,52 +319,27 @@ describe('dsh-subagent-spawn', () => { const fiber = await ctx.plugin(spawn, { providerName: 'spawn' }) ctx.llm.registerAdapter(['mock'], adapter) const parent = ctx.agentLoop.create(AgentId('parent'), { model: 'mock' }) - const run = ctx.subagents.start('spawn', { + const controller = new AbortController() + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'q' }], parent, + signal: controller.signal, outputSchema: { type: 'object', properties: { a: { type: 'number' } } }, }) - // Let the child's step start streaming, then unload the backend. The - // backend owns the child agent, so the unload tears the child down and - // the run settles — releasing its own runtime acquisition on the way out. + // Provider removal prevents new starts but the returned run belongs to its + // holder and remains live. await new Promise(resolve => setTimeout(resolve, 30)) await fiber.dispose() + expect(ctx.subagents.getProvider('spawn')).toBeUndefined() + expect(ctx.agents.get(run.id)).toBeDefined() + controller.abort('test complete') const result = await run.result - expect(result.stopReason).toBe('error') + expect(result.stopReason).toBe('aborted') expect(ctx.tools.get(STRUCTURED_OUTPUT_TOOL)).toBeUndefined() await run.dispose() }) - it('a backend unload during child creation prevents every publication notification', async () => { - const ctx = new Context() - await ctx.plugin(LlmService) - await ctx.plugin(SessionStore) - await ctx.plugin(SystemPrompt) - await ctx.plugin(ToolRegistry) - await ctx.plugin(AgentRegistry) - await ctx.plugin(Invariants) - await ctx.plugin(AgentLoop, { agents: [] }) - await ctx.plugin(SubagentService) - const fiber = await ctx.plugin(spawn, { providerName: 'spawn' }) - ctx.llm.registerAdapter(['mock'], new MockAdapter([])) - const parent = ctx.agentLoop.create(AgentId('parent'), { model: 'mock' }) - const published: string[] = [] - ctx.on('session/created', () => void published.push('session/created')) - ctx.on('agent/created', () => void published.push('agent/created')) - ctx.on('agent/session-start', () => void published.push('agent/session-start')) - - const run = ctx.subagents.start('spawn', { - prompt: [{ type: 'text', text: 'must never run' }], parent, - }) - await fiber.dispose() - await run.result.catch(() => undefined) - await run.dispose() - - expect(ctx.agents.get(run.id)).toBeUndefined() - expect(published).toEqual([]) - }) - - it('a start racing an already-unloading backend cannot mint a run-owner fiber', async () => { + it('a start racing an already-unloading backend cannot begin child creation', async () => { const ctx = new Context() await ctx.plugin(LlmService) await ctx.plugin(SessionStore) @@ -413,10 +356,10 @@ describe('dsh-subagent-spawn', () => { ctx.on('agent/created', () => void published.push('agent/created')) const unloading = fiber.dispose() - expect(() => ctx.subagents.start('spawn', { - prompt: [{ type: 'text', text: 'must never start' }], parent, - })).toThrow(/inactive context/) await unloading + await expect(start(ctx, 'spawn', { + prompt: [{ type: 'text', text: 'must never start' }], parent, + })).rejects.toThrow(/no subagent provider/) expect(parent.ctx.fiber.getEffects()).toHaveLength(parentEffects) expect(published).toEqual([]) @@ -443,7 +386,7 @@ describe('dsh-subagent-spawn', () => { parent.send([{ type: 'text', text: 'hi' }]) await parent.whenIdle() - const run = ctx.subagents.start('spawn', { + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'do X' }], parent, persona: 'You are the tersest test runner.', @@ -466,7 +409,7 @@ describe('dsh-subagent-spawn', () => { name: 'forbidden_tool', description: 'global', parameters: {}, execute: () => Promise.resolve([{ type: 'text', text: 'ran' }]), }) - const run = ctx.subagents.start('spawn', { + const run = await start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'do X' }], parent, toolFilter: { deny: ['forbidden_tool'] }, @@ -486,13 +429,11 @@ describe('dsh-subagent-spawn', () => { it('an unknown toolFilter name fails the spawn loudly with no orphaned child', async () => { const { ctx, parent } = await setup([]) const before = ctx.agents.list().length - const run = ctx.subagents.start('spawn', { + await expect(start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'do X' }], parent, toolFilter: { deny: ['no_such_tool'] }, - }) - await expect(run.result).rejects.toThrow(/unknown tool "no_such_tool"/) - await run.dispose() + })).rejects.toThrow(/unknown global tool "no_such_tool"/) expect(ctx.agents.list().length).toBe(before) }) }) @@ -512,12 +453,10 @@ describe('dsh-subagent-spawn', () => { ctx.on('session/created', () => void published.push('session/created')) ctx.on('agent/created', () => void published.push('agent/created')) ctx.on('agent/session-start', () => void published.push('agent/session-start')) - const run = ctx.subagents.start('spawn', { + await expect(start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'do X' }], parent: parentHandle.agent, - }) - await expect(run.result).rejects.toThrow(/inactive context/) - await run.dispose() + })).rejects.toThrow(/inactive context/) expect(ctx.agents.list().length).toBe(before) expect(ctx.sessions.list()).toHaveLength(sessionsBefore) expect(published).toEqual([]) @@ -535,18 +474,16 @@ describe('dsh-subagent-spawn', () => { ctx.on('agent/created', () => void published.push('agent/created')) ctx.on('agent/session-start', () => void published.push('agent/session-start')) - const run = ctx.subagents.start('spawn', { + const starting = start(ctx, 'spawn', { prompt: [{ type: 'text', text: 'must never run' }], parent: parentHandle.agent, }) - // The factory has entered its awaited unpublished setup transaction. Parent - // ownership was installed before that await, so disposal wins without an + // The factory has entered its awaited unpublished setup transaction. The + // parent context owns that transaction, so disposal wins without an // observer ever seeing the child. await parentHandle.dispose() - await expect(run.result).rejects.toThrow(/owner disposed during setup|inactive context/) - await run.dispose() + await expect(starting).rejects.toThrow(/owner disposed during setup|inactive context/) - expect(ctx.agents.get(run.id)).toBeUndefined() expect(published).toEqual([]) }) }) diff --git a/packages/subagent/subagent/README.md b/packages/subagent/subagent/README.md index 6470b64ee5..413bbbdafc 100644 --- a/packages/subagent/subagent/README.md +++ b/packages/subagent/subagent/README.md @@ -1,44 +1,61 @@ # @deepseek-ai/dsh-subagent -The **subagent seam**: an abstract `SubagentService` (`ctx.subagents`) for an agent delegating work to another agent. A *subagent* is a child agent; a `SubagentProvider` is one transport for running it. +The subagent seam lets one agent delegate work to a child through a named provider. Callers use one service API (`ctx.subagents`); providers decide whether the child runs in this process, in another process, or through a future transport. -This package is the interface third of the capability seam, split so each concern evolves (and swaps) independently: +## Package roles + +The family separates the stable interface from implementations and model-facing tools: | Package | Role | |---|---| -| `@deepseek-ai/dsh-subagent` (this) | the interface: registry service + vocabulary types | -| `@deepseek-ai/dsh-subagent-spawn` | an implementation: fresh in-process child | -| `@deepseek-ai/dsh-subagent-fork` | an implementation: in-process child seeded from the parent's log | -| `@deepseek-ai/dsh-subagent-acp` | an implementation: ACP client driving another process | -| `@deepseek-ai/dsh-tool-subagent` | the model-facing tool over `ctx.subagents` | +| `@deepseek-ai/dsh-subagent` | Provider registry, request/result types, and lifecycle events. | +| `@deepseek-ai/dsh-subagent-spawn` | Fresh in-process child. | +| `@deepseek-ai/dsh-subagent-fork` | In-process child seeded with completed parent turns. | +| `@deepseek-ai/dsh-subagent-acp` | Fresh out-of-process ACP child. | +| `@deepseek-ai/dsh-tool-subagent` | Model-facing tool over one configured provider. | -Unlike the bash seam (one executor per context, second load throws), **multiple providers coexist** here. Each registers under a unique name and a caller picks one by name — the shape mirrors the LLM adapter registry (`LlmService.registerAdapter`), not the single-service bash executor. This is the requirement that rules out the bash shape: an agent may want an in-process child for a cheap subtask and an out-of-process ACP child for an isolated one, in the same runtime. +Multiple providers may coexist under different names. This lets a deployment expose, for example, a cheap in-process child and an isolated ACP child without changing the service contract. -## Service API (`ctx.subagents`) +## Service API -| Member | Semantics | +`SubagentService` has four main operations: + +| Member | Meaning | |---|---| -| `registerProvider(provider)` | Read and validate the name, capability object and four boolean flags, `inheritsParentContext`, and `start` callback exactly once, then register a frozen acceptance snapshot under the accepted name. Malformed fixed fields fail loud before registration; later caller mutation cannot change registry behavior or HMR cleanup, while `start` stays bound to the original provider receiver. Throws `SubagentError('DUPLICATE_PROVIDER')` on a name clash. Effect-scoped (HMR-safe); returns the disposer. | -| `assertSubagentMaxDepth(value)` | Shared runtime boundary for recursion caps. Accepts absence or a non-negative safe integer; rejects fractions, non-finite numbers, negative values, negative zero, and unsafe integers. The service, direct in-process driver, and model-facing config adapter all use it. | -| `getProvider(name)` | Look up the frozen registry snapshot (`undefined` if absent). | -| `list()` | Registered provider names (insertion order). | -| `start(name, request)` | Resolve the provider (`NO_PROVIDER` if absent), read every caller field once into one acceptance snapshot, validate every requested START-TIME capability and scalar value before any child is created, and materialize prompt/schema/options/filter data through a single-pass lossless-JSON snapshot before delegating to `provider.start`. Acquire the provider run's disposer before reading the rest of its handle, then return a frozen service-owned wrapper whose fields are captured once, whose methods remain bound to the provider handle, and whose `result` is one detached, deeply frozen normalization shared by the caller and telemetry. The wrapper claims its shared disposal promise before invoking raw provider code, so synchronous reentry and ordinary repeats join one provider call; a raw disposer that directly returns that same reentrant wrapper promise is rejected as a cyclic provider contract instead of hanging forever. Malformed handle access/binding starts rollback before the synchronous fault escapes; malformed terminal data rejects only after rollback reaches quiescence. Emit `subagent/start` only after `run.started` fulfills and the paired `subagent/end` after that started run settles; a pre-publication readiness rejection emits neither. | +| `registerProvider(provider)` | Register one trusted same-process implementation by name. Registration is effect-scoped; removing it prevents new starts but does not revoke runs already returned to callers. Duplicate names fail loud. | +| `getProvider(name)` | Return the provider, or `undefined` when absent. | +| `list()` | Return provider names in insertion order. | +| `start(name, request)` | Validate requested capabilities and semantic values, then await the provider until a real child is ready. Fulfillment returns a holder-owned `SubagentRun`; rejection means the provider has already cleaned every partial startup resource. | -## Capabilities: two kinds, discovered two ways +`SubagentStartRequest.signal` is required and is the canonical cancellation channel. An abort before publication makes `start()` reject after rollback; an abort after publication cancels the live child. The request may also select a model, require structured output, cap delegation depth, restrict child tools, or set a child persona. -- **Start-time features** (`outputSchema`, `depthLimit`, `toolFilter/persona`) are a static `provider.capabilities` descriptor, checked by the service BEFORE a run exists. A request that needs one the provider lacks is **rejected loud** (`UNSUPPORTED_CAPABILITY`), never accepted-then-ignored. -- **Runtime features** (steering, resume) are **optional methods** on `SubagentRun` (`sendMessage?`, `resume?`). The method's presence IS the capability; TS narrowing is the discovery mechanism — a consumer cannot call an absent method without narrowing first, so there is no silent degradation path. +Same-process requests, descriptors, results, and event payloads are trusted typed values borrowed as immutable. The service does not clone or freeze them; serialization and hostile-input validation belong at actual process, worker, persistence, and model boundaries. -Beside `capabilities` sits one DESCRIPTIVE fact: `provider.inheritsParentContext` — whether a child sees the parent conversation (`fork`: true — seeded with the completed-turn prefix; `spawn`/`acp`: false). “Context” here means conversation history only; it says nothing about tool registrations, injected services, or authority inheritance. The service validates that the descriptor is a boolean but does not interpret or enforce its meaning; the model-facing consumer (`dsh-tool-subagent`) derives truthful tool wording from it. +## Capabilities -## Run lifecycle +Start-time features are advertised in `provider.capabilities` because the service must reject an unsupported request before child creation: -`provider.start(request)` returns a provider-owned `SubagentRun`; `SubagentService.start` captures that handle once and returns a frozen service-owned wrapper with `started` (the publication/readiness promise), a normalized `result` (the terminal outcome), bound `cancel()` and `dispose()`, and bound optional runtime methods. `started` resolves only after the provider has established a real child and rejects if the attempt fails or is cancelled first. `result` resolves with one detached, deeply frozen `SubagentResult` (`output`, optional `structured`, `stopReason`) that the service and caller share — it does **not** reject on a child-level failure (a model/transport failure resolves with `stopReason: 'error'`), but malformed provider data rejects as an infrastructure contract fault. The consumer maps a non-`completed` reason to an `isError` tool result and MUST `dispose()` on every path (success, error, abort) to reach child quiescence and avoid leaking an idle child / session. +- `outputSchema` — enforce a structured final result. +- `depthLimit` — enforce `maxDepth`. +- `toolFilter` — apply the requested child tool restriction. +- `persona` — apply a per-child persona. -The service also announces provider lifecycle: `subagent/provider-added` (the frozen registry snapshot) fires after a registration and `subagent/provider-removed` (the accepted name) after an unregistration, so a consumer deriving state from a named provider (the model-facing tool wording) mirrors registry membership instead of assuming load order — the cordis Loader starts sibling plugins concurrently, so "listed earlier" does not mean "registered earlier". Run lifecycle is gated by provider readiness: the service captures the provider handle's public fields once, `subagent/start` (payload `SubagentRunInfo`) fires only after the accepted `started` promise fulfills, and `subagent/end` (payload `SubagentRunEndInfo`) uses the same accepted id and normalized result; readiness rejection emits neither. For spawn/fork, the start listener can resolve the published child via `ctx.agents.get(info.id)`; a remote provider need not have a local registry entry. Both events are **observe-only** plain emits whose service-owned payloads are deeply frozen before per-listener dispatch. A synchronous listener throw or returned-promise rejection is logged per listener; later listeners still run synchronously, and async listeners remain concurrent fire-and-forget. The service observes the normalized `result` immediately even while readiness is pending and buffers that end payload until start has fired; a malformed provider result rejects the returned result promise and becomes contained `error` telemetry, a rejection cannot become an unhandled detached promise, start always precedes end, and one listener cannot corrupt either the caller or later listeners. `subagent/end` carries the same frozen output as `lastAssistantMessage` on a valid settle path and omits it on infrastructure or result-contract failure. Any run-affecting decision is out of scope for this observe-only surface. +Runtime features are optional methods on `SubagentRun`: `sendMessage?` steers a live child, while `resume?` asynchronously creates a continuation run. Method presence is the capability check. -## Scope (first cut) +`inheritsParentContext` is descriptive rather than enforceable. It says only whether the child sees completed parent conversation history (`fork` does; `spawn` and ACP do not), not whether it inherits tools, services, or authority. -The consumer collects **synchronously**: it starts a run and awaits `result`. Steering (`sendMessage`) is part of the contract but intentionally unused. Background, poll, and spill semantics are outside this seam; long-running-tool handling is shared work across subagents and bash. See the RFC: [docs/rfc/implemented/feature/2026-06-21-subagent-capability-seam.md](../../../docs/rfc/implemented/feature/2026-06-21-subagent-capability-seam.md). +## Ownership and lifecycle -See `src/types.ts` for the full contracts. +`provider.start(request): Promise` is the ownership-transfer boundary. Before fulfillment, the provider owns setup and must cancel, roll back, and quiesce partial resources on every failure. After fulfillment, the caller owns the run and must call `dispose()` on every path. + +`SubagentRun.result` resolves to `{ output, structured?, stopReason }`. Child-level failures resolve with a non-`completed` reason; only an infrastructure fault that the seam cannot represent may reject. `dispose()` is idempotent, cancels remaining work, and waits for the child resources to quiesce. + +The service emits `subagent/start` only after `start()` has fulfilled. It attaches the result observer before that synchronous notification, so even an already-settled child still produces `subagent/start` before `subagent/end`. In-process start observers can resolve the published child through `ctx.agents.get(info.id)`; remote providers need not publish a local agent. + +Run events are scoped to the delegating parent. Every listener is independently contained: a synchronous throw or rejected returned promise is logged without starving peer listeners or changing the run. + +Provider additions and removals also emit `subagent/provider-added` and `subagent/provider-removed`. Consumers such as the model-facing tool use those events because Cordis may load sibling plugins concurrently; configuration order does not prove registration order. + +## Collection model + +The current model-facing tool collects synchronously: it awaits the child result and disposes the run before returning. Background collection and polling remain outside this seam. See the [capability-seam RFC](../../../docs/rfc/implemented/feature/2026-06-21-subagent-capability-seam.md) and `src/types.ts` for the complete contracts. diff --git a/packages/subagent/subagent/package.json b/packages/subagent/subagent/package.json index cf1ebf9485..eb0dbf8da0 100644 --- a/packages/subagent/subagent/package.json +++ b/packages/subagent/subagent/package.json @@ -25,7 +25,6 @@ "@deepseek-ai/dsh-agent": "^0.0.1", "@deepseek-ai/dsh-llm": "^0.0.1", "@deepseek-ai/dsh-scope": "^0.0.1", - "@deepseek-ai/dsh-session": "^0.0.1", "@deepseek-ai/dsh-tools": "^0.0.1", "cordis": "^4.0.0-rc.6" }, @@ -33,7 +32,6 @@ "@deepseek-ai/dsh-agent": "workspace:^", "@deepseek-ai/dsh-llm": "workspace:^", "@deepseek-ai/dsh-scope": "workspace:^", - "@deepseek-ai/dsh-session": "workspace:^", "@deepseek-ai/dsh-tools": "workspace:^", "cordis": "^4.0.0-rc.6" } diff --git a/packages/subagent/subagent/src/index.ts b/packages/subagent/subagent/src/index.ts index ba44354358..6332a0a458 100644 --- a/packages/subagent/subagent/src/index.ts +++ b/packages/subagent/subagent/src/index.ts @@ -1,44 +1,23 @@ /** * The subagent seam (`ctx.subagents`): a named-provider registry plus a - * capability-validating `start` surface. A subagent is an agent delegating - * work to another agent; a {@link SubagentProvider} is one transport for - * running that child (in-process spawn/fork, ACP to another process, and — - * later — A2A, the Codex app-server, the Claude Code Agent SDK). + * capability-validating asynchronous start surface. Providers establish a + * child before returning its run, so fulfillment is the single publication and + * ownership-transfer boundary. * - * Unlike the bash seam (one executor per context, second load throws), MULTIPLE - * providers coexist here: each registers under a unique name and a caller picks - * one by name. The shape mirrors the LLM adapter registry - * (`LlmService.registerAdapter`), not the single-service bash executor. - * - * This package is the INTERFACE third of the capability seam. Implementations - * (`@deepseek-ai/dsh-subagent-spawn`, `-fork`, `-acp`) and the model-facing - * consumer (`@deepseek-ai/dsh-tool-subagent`) are separate packages. - * - * Scope (first cut): the consumer collects synchronously — it starts a run and - * awaits {@link SubagentRun.result}. Steering ({@link SubagentRun.sendMessage}) - * is part of the contract but intentionally unused; background / poll / spill - * semantics are deferred to a future redesign that unifies long-running-tool - * handling across subagents and bash. - * - * The `subagent/start` / `subagent/end` lifecycle events carry an OBSERVE-ONLY - * payload; `subagent/end` additionally carries the child's `lastAssistantMessage` - * — see `docs/rfc/implemented/feature/2026-06-30-subagent-observe-enrich.md`. - * FIXME(subagent-continuation): a control-flow `subagent/end` (an awaited - * waterfall returning a stop/continue decision, like the other interception - * seams) would require reshaping this emit into a waterfall, awaiting listeners - * before settling, and a `resume` capability on the in-process provider — part - * of the deferred background/steering redesign, NOT this observe-only cut. + * Same-process providers are trusted typed collaborators. Requests, provider + * descriptors, results, and lifecycle payloads are borrowed immutable values; + * serialization and hostile-input validation belong at real process, worker, + * persistence, and model boundaries. * * @module @deepseek-ai/dsh-subagent */ import { Context, Service } from 'cordis' import { scopeTarget } from '@deepseek-ai/dsh-scope' -import { assertSupportedOutputSchema, OutputSchemaError } from '@deepseek-ai/dsh-tools' import type { Scoped } from '@deepseek-ai/dsh-scope' -import { deepFreeze, HarnessError } from '@deepseek-ai/dsh-llm' +import { assertSupportedOutputSchema } from '@deepseek-ai/dsh-tools' +import { HarnessError } from '@deepseek-ai/dsh-llm' import type { ContentBlock } from '@deepseek-ai/dsh-llm' -import { snapshotJsonValue } from '@deepseek-ai/dsh-session' import type { Agent, AgentId } from '@deepseek-ai/dsh-agent' import type { SubagentCapabilities, @@ -60,10 +39,6 @@ export type { /** * Reject a recursion cap that cannot represent an exact delegation depth. - * Undefined means the caller did not request a cap and is accepted. The - * service, direct in-process driver, and model-facing config adapter share this - * boundary so no entry path can turn a fractional or non-finite value into an - * ineffective limit. * @param maxDepth - the optional runtime value to validate. */ export function assertSubagentMaxDepth(maxDepth: unknown): void { @@ -84,88 +59,59 @@ declare module 'cordis' { interface Events { /** - * A provider became resolvable in the {@link SubagentService} registry. - * Consumers that derive state from a named provider (e.g. the model-facing - * tool wording in `dsh-tool-subagent`) react HERE instead of assuming load - * order — the cordis Loader starts sibling plugins concurrently, so - * "listed earlier in cordis.yml" does not mean "registered earlier". - * @param provider - the registry's frozen acceptance snapshot of the provider. + * A provider became resolvable in the registry. + * @param provider - the registered provider. * @mode emit */ 'subagent/provider-added'(provider: SubagentProvider): void /** - * A provider left the registry (its plugin's fiber was disposed — an - * unload or an HMR reload). Consumers holding provider-derived state drop - * it here; a reload re-fires `subagent/provider-added` with the fresh - * provider. Delivered with per-listener containment: a throwing - * subscriber is logged, never starves later subscribers, and never - * disrupts the provider's teardown. - * @param name - the registry name that no longer resolves. + * A provider left the registry. Accepted runs remain holder-owned. + * @param name - the provider name that no longer resolves. * @mode emit */ 'subagent/provider-removed'(name: string): void /** - * A subagent run started — emitted only after {@link SubagentRun.started} - * fulfills, when the provider has established a live child. For an - * in-process provider, `ctx.agents.get(info.id)` is therefore guaranteed to - * resolve during this notification. A readiness rejection emits neither - * lifecycle event; every emitted start is paired with - * {@link Events['subagent/end']}. - * Scope-filtered dispatch (`@deepseek-ai/dsh-scope`): the carrier is keyed - * by the DELEGATING PARENT — a listener registered through the parent's - * `agent.ctx` observes only its own delegations; a plain plugin listener - * observes every run. - * @param info - which provider started which child agent. + * A provider established a ready child. For in-process providers, + * `ctx.agents.get(info.id)` resolves during this notification. + * Scope-filtered dispatch keys the carrier by the delegating parent, so a + * parent-scoped listener observes only its own delegations. Paired with + * `subagent/end`. + * @param info - the provider and ready child identity. * @mode emit */ 'subagent/start'(this: Scoped, info: SubagentRunInfo): void /** - * A started subagent run settled — emitted when {@link SubagentRun.result} - * resolves (any stop reason) or rejects (reported as `error`). Paired with - * {@link Events['subagent/start']}; a run whose readiness rejected emits - * neither event. - * Scope-filtered dispatch (`@deepseek-ai/dsh-scope`): the carrier is keyed - * by the DELEGATING PARENT — a listener registered through the parent's - * `agent.ctx` observes only its own delegations; a plain plugin listener - * observes every run. - * @param info - the run identity plus stop reason and final output. + * A ready child settled. Scope-filtered dispatch uses the same delegating + * parent carrier as `subagent/start`, so the lifecycle pair reaches the + * same scoped audience. + * @param info - the run identity and terminal outcome. * @mode emit */ 'subagent/end'(this: Scoped, info: SubagentRunEndInfo): void } } -/** Deep-frozen, observe-only identifying detail for a started subagent run. */ +/** Observe-only identifying detail for a ready subagent run. */ export interface SubagentRunInfo { - /** The provider that started the run. */ - provider: string + /** The provider that established the run. */ + readonly provider: string /** The child agent's id. */ - id: AgentId + readonly id: AgentId } -/** Deep-frozen, observe-only outcome detail for a settled subagent run. */ +/** Observe-only outcome detail for a settled subagent run. */ export interface SubagentRunEndInfo { /** The provider that ran it. */ - provider: string + readonly provider: string /** The child agent's id. */ - id: AgentId + readonly id: AgentId /** The terminal stop reason. */ - stopReason: SubagentResult['stopReason'] - /** - * The child's final assistant output ({@link SubagentResult.output}), carried - * onto the end event so an observer sees WHAT the subagent produced without - * holding the run. Absent when the run rejected at the infrastructure level - * (no {@link SubagentResult} was produced — the seam only knows `stopReason: - * 'error'`). - */ - lastAssistantMessage?: ContentBlock[] + readonly stopReason: SubagentResult['stopReason'] + /** The child's final assistant output, absent on infrastructure rejection. */ + readonly lastAssistantMessage?: ContentBlock[] } -/** - * Typed error for subagent-seam failures. Extends {@link HarnessError}, so the - * `code` string (`DUPLICATE_PROVIDER`, `NO_PROVIDER`, `UNSUPPORTED_CAPABILITY`) - * is shared, machine-routable taxonomy. - */ +/** Typed error for provider lookup, registration, and capability failures. */ export class SubagentError extends HarnessError { constructor(message: string, code: string, options?: ErrorOptions) { super(message, code, options) @@ -173,10 +119,7 @@ export class SubagentError extends HarnessError { } } -/** - * The `subagents` service: a registry of named {@link SubagentProvider}s and a - * capability-checked {@link start} surface. - */ +/** Named provider registry and capability-checked start surface. */ export class SubagentService extends Service { private providers = new Map() @@ -185,451 +128,88 @@ export class SubagentService extends Service { } /** - * Register a provider under its `provider.name`. Throws {@link SubagentError} - * (`DUPLICATE_PROVIDER`) if the name is already taken. The registry snapshots - * the name, static descriptors, and `start` callback identity at acceptance; - * every fixed field and capability flag is read once and validated before - * registration, so malformed provider objects fail loud without entering the - * registry. Later caller mutation cannot change lookup, capability validation, - * consumer wording, dispatch, or HMR cleanup. The callback remains bound to - * the original provider object, so provider-owned mutable state stays live. - * Effect-scoped: disposed with the calling fiber (HMR-safe). Emits - * `subagent/provider-added` after the registration and - * `subagent/provider-removed` on unregistration, so consumers can mirror - * provider lifecycle instead of assuming load order. - * @param provider - the provider; its `name` is the registry key. - * @returns the disposer that unregisters the provider. The exact - * Cordis effect disposer (single-shot): composite (generator) effects may - * yield it directly — exact identity nests the teardown in order. + * Register a provider under its name. Registration is effect-scoped and HMR + * safe; removing a provider blocks new starts but does not revoke runs that + * were already returned to their holders. + * @param provider - the trusted provider implementation. + * @returns the exact Cordis effect disposer. */ registerProvider(provider: SubagentProvider): () => Promise | void { - // Snapshot the accepted registration contract before entering the effect. - // Cleanup must never re-read caller-owned `provider.name`: an HMR host may - // mutate or reuse the provider object before its old fiber unloads. Binding - // preserves the provider method's receiver while making replacement of the - // public callback field after registration inert. - const name: unknown = provider.name - const inputCapabilities: unknown = provider.capabilities - const inheritsParentContext: unknown = provider.inheritsParentContext - // eslint-disable-next-line @typescript-eslint/unbound-method - const inputStart: unknown = provider.start - if (typeof name !== 'string') { - throw new TypeError('subagent provider name must be a string') - } - if (inputCapabilities === null || typeof inputCapabilities !== 'object' || Array.isArray(inputCapabilities)) { - throw new TypeError(`subagent provider "${name}" capabilities must be an object`) - } - const inputCapabilityFields = inputCapabilities as Record - const outputSchema = inputCapabilityFields.outputSchema - const depthLimit = inputCapabilityFields.depthLimit - const toolFilter = inputCapabilityFields.toolFilter - const persona = inputCapabilityFields.persona - for (const [capability, value] of [ - ['outputSchema', outputSchema], - ['depthLimit', depthLimit], - ['toolFilter', toolFilter], - ['persona', persona], - ] as const) { - if (typeof value !== 'boolean') { - throw new TypeError(`subagent provider "${name}" capability "${capability}" must be a boolean`) + const name = provider.name + return this.ctx.effect(function* (this: SubagentService) { + if (this.providers.has(name)) { + throw new SubagentError(`a subagent provider named "${name}" is already registered`, 'DUPLICATE_PROVIDER') } - } - if (typeof inheritsParentContext !== 'boolean') { - throw new TypeError(`subagent provider "${name}" inheritsParentContext must be a boolean`) - } - if (typeof inputStart !== 'function') { - throw new TypeError(`subagent provider "${name}" start must be a function`) - } - const capabilities: SubagentCapabilities = Object.freeze({ - outputSchema: outputSchema as boolean, - depthLimit: depthLimit as boolean, - toolFilter: toolFilter as boolean, - persona: persona as boolean, - }) - const snapshot: SubagentProvider = Object.freeze({ - name, - capabilities, - inheritsParentContext, - start: Function.prototype.bind.call(inputStart, provider) as SubagentProvider['start'], - }) - const dispose = this.ctx.effect(function* (this: SubagentService) { - if (this.providers.has(snapshot.name)) { - throw new SubagentError(`a subagent provider named "${snapshot.name}" is already registered`, 'DUPLICATE_PROVIDER') - } - this.providers.set(snapshot.name, snapshot) - // Yield the rollback BEFORE emitting `subagent/provider-added`: a - // throwing added-listener then unregisters the provider (and announces - // the removal) instead of leaking it into the registry. The removal - // announcement itself is contained PER LISTENER ({@link emitLifecycle}): - // it runs inside this disposer, where a propagating subscriber would - // disrupt the backend fiber's teardown and starve later mirrors. + this.providers.set(name, provider) yield () => { - this.providers.delete(snapshot.name) - this.emitLifecycle('subagent/provider-removed', snapshot.name) + this.providers.delete(name) + this.emitLifecycle('subagent/provider-removed', name) } - this.ctx.emit('subagent/provider-added', snapshot) + // A throwing added-listener unwinds the yielded rollback, matching the + // repository's fail-loud registration semantics. + this.ctx.emit('subagent/provider-added', provider) }.bind(this), 'subagents.registerProvider()') - // The EXACT cordis effect disposer, not a wrapper: a composite (generator) - // effect that owns a teardown ORDER must be able to yield THIS function — - // cordis nests a disposer out of the fiber's concurrent sibling list by - // exact function identity, so a wrapper would silently break the nesting - // (the agents.register() lesson). Fire-and-forget callers may still - // discard the (always-resolved) promise. - return dispose } /** - * Look up the registry's frozen provider snapshot by its accepted name - * (`undefined` if absent). - * @param name - the provider name accepted at registration. - * @returns the frozen acceptance snapshot, or undefined when the name is unknown. + * Look up a provider by name. + * @param name - the provider name. + * @returns the provider, or undefined when absent. */ getProvider(name: string): SubagentProvider | undefined { return this.providers.get(name) } /** - * The names of all registered providers (insertion order). - * @returns the registered provider names. + * List registered provider names in insertion order. + * @returns the registered names. */ list(): string[] { return [...this.providers.keys()] } /** - * Start a subagent run on the named provider. Resolves the provider (throws - * `NO_PROVIDER` if absent), reads the caller request once into a coherent - * acceptance snapshot, validates every requested START-TIME capability - * against {@link SubagentProvider.capabilities} (throws `UNSUPPORTED_CAPABILITY` - * for the first unmet one — fail loud, before any child is created), then - * validates the request's scalar values, materializes model-bound data in one - * lossless-JSON traversal, and delegates the detached request to - * {@link SubagentProvider.start}. The returned handle is a service-owned, - * frozen wrapper: provider fields are captured once, methods stay bound to the - * provider handle, and `result` resolves to one detached, deeply frozen value - * shared by the caller and lifecycle telemetry. Once a provider returns a - * callable disposer, malformed handle access/binding starts rollback before - * the synchronous fault escapes; malformed terminal data rejects only after - * that same memoized disposal reaches quiescence. Emits `subagent/start` / - * `subagent/end` only after the run's readiness boundary fulfills. A provider - * that fails before establishing a child emits neither event. - * @param name - the provider to run on. - * @param request - the child's prompt, capabilities, and options. - * @returns the live run (its `result` resolves when the child settles). + * Establish a ready child on the named provider. Capability and semantic + * checks run before delegation. Provider ownership lasts until its promise + * fulfills; a rejection therefore has no run for the caller to dispose and + * emits no run lifecycle events. + * @param name - the provider to use. + * @param request - child prompt, parent, signal, and optional capabilities. + * @returns the ready holder-owned run. */ - start(name: string, request: SubagentStartRequest): SubagentRun { + async start(name: string, request: SubagentStartRequest): Promise { const provider = this.providers.get(name) - if (!provider) { + if (provider === undefined) { throw new SubagentError(`no subagent provider registered for "${name}"`, 'NO_PROVIDER') } - // Read every top-level field exactly once before capability checks or - // detachment. A stateful accessor must not look absent to validation and then - // appear in the provider request (or vice versa). - const input = this.snapshotStartRequest(request) - const parent = input.parent - this.assertCapabilities(provider, input) - assertSubagentMaxDepth(input.maxDepth) - if (input.persona !== undefined && typeof input.persona !== 'string') { - throw new TypeError('subagent persona must be a string') - } - // Model/session-bound values are validated and detached in a single - // recursive pass. A check followed by structuredClone would reread getters - // and could erase an exotic prototype returned only to the clone. - const prompt = snapshotJsonValue(input.prompt) - if (prompt === undefined) { - throw new TypeError('subagent prompt must be losslessly JSON-serializable') - } - const outputSchema = input.outputSchema === undefined - ? undefined - : snapshotJsonValue(input.outputSchema) - if (input.outputSchema !== undefined && outputSchema === undefined) { - throw new OutputSchemaError(['schema annotation must be JSON data; the complete schema must be losslessly JSON-serializable']) - } - if (outputSchema !== undefined) assertSupportedOutputSchema(outputSchema) - const agentOptions = input.agentOptions === undefined - ? undefined - : snapshotJsonValue(input.agentOptions) - if (input.agentOptions !== undefined && agentOptions === undefined) { - throw new TypeError('subagent agent options must be losslessly JSON-serializable') - } - const toolFilter = input.toolFilter === undefined - ? undefined - : snapshotJsonValue(input.toolFilter) - if (input.toolFilter !== undefined && toolFilter === undefined) { - throw new TypeError('subagent tool filter must be losslessly JSON-serializable') - } + this.assertCapabilities(provider, request) + assertSubagentMaxDepth(request.maxDepth) + if (request.outputSchema !== undefined) assertSupportedOutputSchema(request.outputSchema) - // Detach every data field before crossing into a provider. Parent/signal - // are live identity capabilities and stay exact; the mutable request record - // and its arrays/objects are never retained, so every backend (including an - // async out-of-process one) observes the request accepted at start. - const accepted: SubagentStartRequest = { - prompt, - parent, - ...input.signal !== undefined ? { signal: input.signal } : {}, - ...agentOptions !== undefined ? { agentOptions } : {}, - ...outputSchema !== undefined ? { outputSchema } : {}, - ...input.maxDepth !== undefined ? { maxDepth: input.maxDepth } : {}, - ...toolFilter !== undefined ? { toolFilter } : {}, - ...input.persona !== undefined ? { persona: input.persona } : {}, - } - const providerRun: unknown = provider.start(accepted) - if (providerRun === null || (typeof providerRun !== 'object' && typeof providerRun !== 'function')) { - throw new TypeError(`subagent provider "${name}" start must return a SubagentRun object`) - } - const acceptedRun = providerRun as SubagentRun - // Acquire the one rollback capability BEFORE touching any other provider-run - // field. Once start() returned a handle, the service owns an accepted live - // attempt; a hostile later accessor or bind must not make that attempt - // unreachable. The wrapper also memoizes provider disposal, so automatic - // rollback and a racing caller join one quiescence transaction even if a - // contract-violating provider forgot to make its own method idempotent. - // eslint-disable-next-line @typescript-eslint/unbound-method - const inputDispose = acceptedRun.dispose - if (typeof inputDispose !== 'function') { - throw new TypeError(`subagent provider "${name}" run dispose must be a function`) - } - let disposal: Promise | undefined - const dispose = (): Promise => { - if (disposal === undefined) { - // Claim the shared transaction before invoking provider code: a raw - // disposer can synchronously reenter this wrapper through a reference - // retained by its caller, and both calls must join one provider call. - const claimed = Promise.withResolvers() - disposal = claimed.promise - try { - // Invoke through the captured callable without reading its public - // `bind`/`length`/`name` properties. Disposal is the recovery - // capability itself; hostile function metadata must not prevent the - // seam from exercising it when a later handle field is malformed. - const returned: unknown = Reflect.apply(inputDispose, acceptedRun, []) - // A raw disposer can reenter the service wrapper and directly return - // that same shared promise. Awaiting it here would make the promise - // depend on itself forever; reject the cyclic provider contract loud. - if (returned === claimed.promise) { - claimed.reject(new TypeError(`subagent provider "${name}" run dispose returned its own wrapper disposal promise`)) - return disposal - } - void Promise.resolve(returned).then( - () => { claimed.resolve(undefined) }, - (error: unknown) => { claimed.reject(error) }, - ) - } catch (error: unknown) { - claimed.reject(error instanceof Error - ? error - : new Error('subagent provider run dispose threw a non-Error value', { cause: error })) - } - } - return disposal - } - // Provider-owned run objects can be accessor-backed too. Capture every - // public field exactly once, bind methods to the provider's original handle, - // and expose only this service-owned wrapper. The normalized result promise - // is also the one lifecycle telemetry observes, so the caller and observers - // cannot receive different values from stateful accessors. - try { - const id = acceptedRun.id - if (typeof id !== 'string') { - throw new TypeError(`subagent provider "${name}" run id must be a string`) - } - const started = acceptedRun.started - if (!(started instanceof Promise)) { - throw new TypeError(`subagent provider "${name}" run started must be a Promise`) - } - // Observe each accepted provider promise before reading the next hostile - // field. A later accessor/validation failure prevents a wrapper from being - // returned, but must not leave an already-rejected provider promise - // unhandled while rollback proceeds. - void started.catch(() => undefined) - const providerResult = acceptedRun.result - if (!(providerResult instanceof Promise)) { - throw new TypeError(`subagent provider "${name}" run result must be a Promise`) - } - void providerResult.catch(() => undefined) - // eslint-disable-next-line @typescript-eslint/unbound-method - const inputCancel = acceptedRun.cancel - if (typeof inputCancel !== 'function') { - throw new TypeError(`subagent provider "${name}" run cancel must be a function`) - } - // eslint-disable-next-line @typescript-eslint/unbound-method - const inputSendMessage = acceptedRun.sendMessage - if (inputSendMessage !== undefined && typeof inputSendMessage !== 'function') { - throw new TypeError(`subagent provider "${name}" run sendMessage must be a function when provided`) - } - // eslint-disable-next-line @typescript-eslint/unbound-method - const inputResume = acceptedRun.resume - if (inputResume !== undefined && typeof inputResume !== 'function') { - throw new TypeError(`subagent provider "${name}" run resume must be a function when provided`) - } - const cancel = Function.prototype.bind.call(inputCancel, acceptedRun) as SubagentRun['cancel'] - const sendMessage = inputSendMessage === undefined - ? undefined - : Function.prototype.bind.call(inputSendMessage, acceptedRun) as NonNullable - const resume = inputResume === undefined - ? undefined - : Function.prototype.bind.call(inputResume, acceptedRun) as NonNullable - const result = providerResult.then(async (value) => { - try { - return this.snapshotRunResult(value) - } catch (error: unknown) { - // A malformed terminal value is an infrastructure contract fault. The - // result rejects only after the accepted provider attempt has reached - // quiescence, so a caller cannot lose the only cleanup handle by merely - // observing the normalization failure. - await this.rollbackProviderRun(name, dispose) - throw error - } - }) - const run: SubagentRun = Object.freeze({ - id, - started, - result, - cancel, - dispose, - ...sendMessage === undefined - ? {} - : { sendMessage }, - ...resume === undefined - ? {} - : { resume }, - }) - - // Observe result settlement IMMEDIATELY, before waiting on readiness. A - // provider may fail both promises in the same turn; deferring the rejection - // handler until `started` fulfilled would leave `result` transiently - // unhandled. The settled event is buffered until start has been announced, - // preserving start → end order even for an already-settled scripted run. - let readiness: 'pending' | 'started' | 'failed' = 'pending' - let pendingEnd: SubagentRunEndInfo | undefined - const deliverEnd = (info: SubagentRunEndInfo): void => { - if (readiness === 'started') this.emitLifecycle('subagent/end', info, parent) - else if (readiness === 'pending') pendingEnd = info - // A pre-publication readiness failure has no lifecycle pair; result - // remains observable by the run's consumer, but telemetry must not claim - // that a child started. - } - void result.then( - (value) => { - deliverEnd({ - provider: name, - id, - stopReason: value.stopReason, - lastAssistantMessage: value.output, - }) - }, - () => { deliverEnd({ provider: name, id, stopReason: 'error' }) }, - ) - - // Readiness is the publication boundary owned by the provider. For - // in-process runs, fulfillment means the agent registry already contains - // `run.id`; for ACP it means the remote session exists. Emit start with - // per-listener containment, then flush an outcome that settled unusually - // early. A readiness rejection is handled here and deliberately emits no - // false start/end pair; the result path above remains independently handled. - void started.then( - () => { - readiness = 'started' - this.emitLifecycle('subagent/start', { provider: name, id }, parent) - if (pendingEnd !== undefined) { - const info = pendingEnd - pendingEnd = undefined - this.emitLifecycle('subagent/end', info, parent) - } - }, - () => { - readiness = 'failed' - pendingEnd = undefined - }, - ) - return run - } catch (error: unknown) { - // start() has already transferred a live attempt to the seam. Begin - // rollback synchronously before surfacing the malformed-handle failure; - // the contained cleanup promise prevents either a resource leak or an - // unhandled rejection even though this API cannot synchronously await it. - void this.rollbackProviderRun(name, dispose) - throw error - } - } - - /** Dispose one malformed provider attempt without letting cleanup mask the contract fault. */ - private async rollbackProviderRun(providerName: string, dispose: () => Promise): Promise { - try { - await dispose() - } catch (error: unknown) { - this.ctx.logger.warn(`subagent provider "${providerName}" malformed run rollback failed: ${renderThrown(error)}`) - } - } - - /** Normalize one provider result into the immutable seam value. */ - private snapshotRunResult(value: SubagentResult): SubagentResult { - // Capture every provider-owned field once before validation. In particular, - // lifecycle telemetry must not reread accessors after the caller receives - // the result and observe a different terminal outcome. - const output = value.output - const structured = value.structured - const stopReason = value.stopReason - if (!Array.isArray(output)) { - throw new TypeError('subagent result output must be an array') - } - if (typeof stopReason !== 'string') { - throw new TypeError('subagent result stopReason must be a string') - } - const accepted: SubagentResult = { - output, - ...structured === undefined ? {} : { structured }, - stopReason, - } - const snapshot = snapshotJsonValue(accepted) - if (snapshot === undefined) { - throw new TypeError('subagent result must be losslessly JSON-serializable') - } - return deepFreeze(snapshot) - } - - /** Read one coherent caller request into immutable data properties. */ - private snapshotStartRequest(request: SubagentStartRequest): Readonly { - const prompt = request.prompt const parent = request.parent - const signal = request.signal - const agentOptions = request.agentOptions - const outputSchema = request.outputSchema - const maxDepth = request.maxDepth - const toolFilter = request.toolFilter - const persona = request.persona - return Object.freeze({ - prompt, - parent, - ...signal !== undefined ? { signal } : {}, - ...agentOptions !== undefined ? { agentOptions } : {}, - ...outputSchema !== undefined ? { outputSchema } : {}, - ...maxDepth !== undefined ? { maxDepth } : {}, - ...toolFilter !== undefined ? { toolFilter } : {}, - ...persona !== undefined ? { persona } : {}, - }) + const run = await provider.start(request) + // Attach the terminal observer before dispatching start. Promise reactions + // still run after this synchronous start emission, preserving start → end. + void run.result.then( + (result) => { + this.emitLifecycle('subagent/end', { + provider: name, + id: run.id, + stopReason: result.stopReason, + lastAssistantMessage: result.output, + }, parent) + }, + () => { + this.emitLifecycle('subagent/end', { provider: name, id: run.id, stopReason: 'error' }, parent) + }, + ) + this.emitLifecycle('subagent/start', { provider: name, id: run.id }, parent) + return run } /** - * Emit a `subagent/*` lifecycle event with PER-LISTENER containment: dispatch - * each subscriber individually and log (never propagate) either a synchronous - * throw or a returned-promise rejection, so one bad subscriber can neither - * strand the already-live run, surface as an unhandled rejection on the - * detached settle hook, NOR starve the listeners registered after it. Async - * listeners remain concurrent fire-and-forget; dispatch does not await or - * serialize them. A single try/catch around `ctx.emit` would not do the - * last part — cordis `emit` runs listeners in a `.map(cb => cb())` that halts - * on the first throw — so this resolves the listener callbacks via - * `ctx.events.dispatch` and contains each call, the same guarantee - * `BashExecutor.notifyTaskDone` gives its own listener set. - * - * `subagent/provider-removed` routes through here too: it fires inside the - * provider registration's DISPOSER, where a propagating listener would - * disrupt the backend fiber's teardown (dispose must reach quiescence) and a - * starved later listener would leave a mirror consumer (`dsh-tool-subagent`) - * holding a tool for a provider that no longer exists. `subagent/provider-added` - * deliberately does NOT: it fires at registration time, where a throwing - * listener unwinds the yielded rollback — the same fail-loud register-time - * semantics as the system-prompt registries. + * Emit lifecycle events with per-listener synchronous and asynchronous + * exception containment. Payloads are borrowed immutable values. */ private emitLifecycle(name: 'subagent/start', info: SubagentRunInfo, parent: Agent): void private emitLifecycle(name: 'subagent/end', info: SubagentRunEndInfo, parent: Agent): void @@ -639,21 +219,12 @@ export class SubagentService extends Service { info: SubagentRunInfo | SubagentRunEndInfo | string, parent?: Agent, ): void { - // Run lifecycle events dispatch in the DELEGATING PARENT's scope (a - // parent-scoped listener observes only its own delegations); the - // provider-removed registry notification stays unfiltered. The carrier is - // args[0] of the dispatch call, exactly as cordis' own emit spells it. - const acceptedInfo = typeof info === 'string' ? info : deepFreeze(info) const dispatchArgs: unknown[] = parent === undefined - ? [name, acceptedInfo] - : [scopeTarget(this, parent), name, acceptedInfo] + ? [name, info] + : [scopeTarget(this, parent), name, info] for (const callback of this.ctx.events.dispatch('emit', dispatchArgs)) { try { - const returned: unknown = callback(acceptedInfo) - // Plain emits remain fire-and-forget and every callback is still invoked - // synchronously in this loop. Observe a returned promise independently so - // an async listener rejection is contained without serializing listeners - // or delaying provider/run lifecycle. + const returned: unknown = callback(info) void Promise.resolve(returned).catch((error: unknown) => { this.ctx.logger.warn(`subagent: ${name} listener rejected: ${renderThrown(error)}`) }) @@ -663,11 +234,7 @@ export class SubagentService extends Service { } } - /** - * Reject a request that needs a start-time capability the provider lacks. - * Each optional request field maps to one {@link SubagentCapabilities} flag; - * the first unmet one throws `UNSUPPORTED_CAPABILITY`. - */ + /** Reject the first requested capability that the provider lacks. */ private assertCapabilities(provider: SubagentProvider, request: SubagentStartRequest): void { const needs: { when: boolean; cap: keyof SubagentCapabilities }[] = [ { when: request.outputSchema !== undefined, cap: 'outputSchema' }, @@ -686,7 +253,7 @@ export class SubagentService extends Service { } } -/** Render an arbitrary thrown value without allowing coercion to throw again. */ +/** Render any listener-thrown value without letting coercion escape containment. */ function renderThrown(value: unknown): string { try { return value instanceof Error ? `${value.name}: ${value.message}` : String(value) diff --git a/packages/subagent/subagent/src/types.ts b/packages/subagent/subagent/src/types.ts index 42f55b300d..f08d8a0622 100644 --- a/packages/subagent/subagent/src/types.ts +++ b/packages/subagent/subagent/src/types.ts @@ -8,7 +8,7 @@ import type { Agent, AgentId, AgentOptions } from '@deepseek-ai/dsh-agent' import type { ContentBlock } from '@deepseek-ai/dsh-llm' -import type { StructuredOutputSchema } from '@deepseek-ai/dsh-tools' +import type { StructuredOutputSchema, ToolRestriction } from '@deepseek-ai/dsh-tools' /** * Which START-TIME features a provider supports. Checked by the service @@ -24,13 +24,13 @@ import type { StructuredOutputSchema } from '@deepseek-ai/dsh-tools' */ export interface SubagentCapabilities { /** Honor {@link SubagentStartRequest.outputSchema} (structured final output). */ - outputSchema: boolean + readonly outputSchema: boolean /** Enforce {@link SubagentStartRequest.maxDepth} (recursion cap). */ - depthLimit: boolean + readonly depthLimit: boolean /** Enforce {@link SubagentStartRequest.toolFilter} (child tool scoping). */ - toolFilter: boolean + readonly toolFilter: boolean /** Honor {@link SubagentStartRequest.persona} (a per-child persona). */ - persona: boolean + readonly persona: boolean } /** @@ -41,22 +41,24 @@ export interface SubagentCapabilities { */ export interface SubagentStartRequest { /** The task/prompt for the child agent (a user message in the child session). */ - prompt: ContentBlock[] + readonly prompt: ContentBlock[] /** * The spawning ("parent") agent — the one whose tool call started this * subagent. REQUIRED: in-process backends read `parent.session.header` for * the working directory, the `parentSession` lineage to stamp on the child, * and the parent's delegation depth. Out-of-process backends (ACP) ignore it. */ - parent: Agent + readonly parent: Agent /** * Cancellation signal from the spawning context (the tool's `exec.signal`). - * A provider that honors it aborts the child when the signal fires; the - * consumer also bridges it to {@link SubagentRun.cancel} explicitly. + * This is the canonical cancellation channel both before and after startup: + * a provider rejects `start()` after cleaning partial resources when it + * fires before publication, and cancels a published child when it fires + * afterward. */ - signal?: AbortSignal + readonly signal: AbortSignal /** Per-child agent options (model, system prompt). */ - agentOptions?: AgentOptions + readonly agentOptions?: AgentOptions /** * Optional structured-output schema — an object-rooted JSON Schema within the * enforced subset (see `assertSupportedOutputSchema` in dsh-tools; a schema @@ -67,14 +69,14 @@ export interface SubagentStartRequest { * data — a caller holding foreign-realm data materializes it first. * Requesting it against a provider that lacks the capability is rejected at start. */ - outputSchema?: StructuredOutputSchema + readonly outputSchema?: StructuredOutputSchema /** * Optional absolute delegation-depth cap for the child being started: its * computed depth must be less than or equal to this non-negative safe * integer. Requires {@link SubagentCapabilities.depthLimit}; rejected at * start otherwise. */ - maxDepth?: number + readonly maxDepth?: number /** * Optional child tool scoping. Requires {@link SubagentCapabilities.toolFilter}; * rejected at start otherwise. In-process backends apply it as a scoped @@ -82,7 +84,7 @@ export interface SubagentStartRequest { * from the child's prompt AND refuse to execute (one visibility), with loud * unknown-name validation. */ - toolFilter?: { allow?: string[]; deny?: string[] } + readonly toolFilter?: ToolRestriction /** * Optional per-child persona. Requires {@link SubagentCapabilities.persona}; * rejected at start otherwise. In-process backends register it as a scoped @@ -90,7 +92,7 @@ export interface SubagentStartRequest { * persona for this child alone — same template semantics as the deployment * persona (strict `{{…}}` interpolation against the registered variables). */ - persona?: string + readonly persona?: string } /** @@ -102,7 +104,7 @@ export interface SubagentStartRequest { export interface SubagentStopReasonMap { /** The child finished its turn normally. */ completed: 'completed' - /** The run was cancelled (parent signal, explicit `cancel()`, or peer cancel). */ + /** The run was cancelled by its request signal or by disposal. */ aborted: 'aborted' /** The child failed (model error, transport error). */ error: 'error' @@ -120,7 +122,7 @@ export type SubagentStopReason = SubagentStopReasonMap[keyof SubagentStopReasonM */ export interface SubagentResult { /** The child's final assistant output (the last assistant message's content). */ - output: ContentBlock[] + readonly output: ContentBlock[] /** * The structured result after a requested `outputSchema` was successfully * satisfied. Requesting a schema does not guarantee presence: a provider can @@ -128,32 +130,24 @@ export interface SubagentResult { * valid capture. Shape is validated against the request schema by the * provider; `unknown` here because the seam is schema-agnostic. */ - structured?: unknown + readonly structured?: unknown /** Why the run ended. A non-`completed` reason means `output` may be partial. */ - stopReason: SubagentStopReason + readonly stopReason: SubagentStopReason } /** * A live subagent run: a handle the consumer holds while a child executes. - * Returned by {@link SubagentProvider.start} (via the service). The consumer - * awaits {@link result}, may {@link cancel} mid-flight, and MUST {@link dispose} - * on every path to reach child quiescence (no leaked idle child / session). + * Returned by {@link SubagentProvider.start} (via the service) only after the + * child is ready. The consumer awaits {@link result} and MUST {@link dispose} + * on every path to cancel any remaining work and reach child quiescence. * * {@link sendMessage} and {@link resume} are OPTIONAL: a provider that supports * the runtime capability defines the method; one that doesn't omits it. The * presence of the method IS the capability — narrow before calling. */ export interface SubagentRun { - /** The child agent's id (local in-process runs publish it in `ctx.agents`; remote transports need not). */ + /** The child agent's id (local in-process runs are already published in `ctx.agents`; remote transports need not publish locally). */ readonly id: AgentId - /** - * The provider's publication/readiness boundary. Resolves only after a real - * child is established: an in-process agent is live in `ctx.agents`, or a - * remote transport has created its child session. Rejects when the attempt - * fails or is cancelled before that boundary. The service emits the paired - * `subagent/start`/`subagent/end` lifecycle only after this fulfills. - */ - readonly started: Promise /** * Resolves with the child's terminal {@link SubagentResult} when the run * settles. Does NOT reject on a child-level failure — a model/transport @@ -162,12 +156,10 @@ export interface SubagentRun { * cannot represent as a stop reason. */ readonly result: Promise - /** Request cancellation of the in-flight run; {@link result} settles `aborted`. */ - cancel(reason?: string): void /** - * Reach child quiescence and release the run's resources (in-process: dispose - * the owned agent handle and remove its session; ACP: kill the subprocess). - * Idempotent; awaits the child actually stopping, not merely requesting it. + * Cancel remaining work, reach child quiescence, and release the run's + * resources (in-process: dispose the owned agent and remove its session; + * ACP: kill and reap the subprocess). Idempotent. */ dispose(): Promise /** @@ -179,7 +171,7 @@ export interface SubagentRun { * OPTIONAL (resume capability): send a follow-up task to a settled child, * continuing its session, and return a fresh run for the continuation. */ - resume?(content: ContentBlock[]): SubagentRun + resume?(content: ContentBlock[]): Promise } /** @@ -187,8 +179,8 @@ export interface SubagentRun { * spawn/fork, ACP to another process, …). Implementations register under a * unique name via {@link SubagentService.registerProvider}; multiple providers * coexist in one context (unlike the single-implementation bash seam). The - * service freezes the public descriptor and callback identity at registration; - * the captured `start` remains bound to the original provider receiver. + * Providers are trusted same-process implementations; callers treat their + * descriptors and returned values as borrowed immutable data. */ export interface SubagentProvider { /** Unique registry name (e.g. `spawn`, `fork`, `acp`). */ @@ -208,12 +200,12 @@ export interface SubagentProvider { */ readonly inheritsParentContext: boolean /** - * Start preparing a child run and return its handle synchronously. The + * Establish a child and return its handle only after publication. The * service has already validated that every requested start-time capability * is supported, so an implementation may assume e.g. `request.maxDepth` is - * honorable when present. The returned {@link SubagentRun.started} must mark - * the real publication/readiness boundary; the result path must observe that - * promise immediately so a pre-start rejection cannot become unhandled. + * honorable when present. If setup fails or `request.signal` aborts before + * fulfillment, the provider owns and cleans all partial resources before this + * promise rejects. Ownership transfers to the caller only on fulfillment. */ - start(request: SubagentStartRequest): SubagentRun + start(request: SubagentStartRequest): Promise } diff --git a/packages/subagent/subagent/tests/service.spec.ts b/packages/subagent/subagent/tests/service.spec.ts index 182ee7ac8b..23c2caf445 100644 --- a/packages/subagent/subagent/tests/service.spec.ts +++ b/packages/subagent/subagent/tests/service.spec.ts @@ -5,6 +5,7 @@ import { HarnessError } from '@deepseek-ai/dsh-llm' import { carrierKeyOf } from '@deepseek-ai/dsh-scope' import SubagentService, { SubagentError, + assertSubagentMaxDepth, type SubagentCapabilities, type SubagentProvider, type SubagentResult, @@ -12,1350 +13,218 @@ import SubagentService, { type SubagentStartRequest, } from '@deepseek-ai/dsh-subagent' -/** A minimal parent Agent stand-in — the service only reads `parent.id`. */ function fakeParent(id = 'parent-1'): Agent { return { id: AgentId(id) } as unknown as Agent } -const ALL_CAPS: SubagentCapabilities = { outputSchema: true, depthLimit: true, toolFilter: true, persona: false } +const ALL_CAPS: SubagentCapabilities = { outputSchema: true, depthLimit: true, toolFilter: true, persona: true } const NO_CAPS: SubagentCapabilities = { outputSchema: false, depthLimit: false, toolFilter: false, persona: false } -/** A scripted provider whose run settles immediately with a fixed result. */ +function baseRequest(overrides: Partial = {}): SubagentStartRequest { + return { + prompt: [{ type: 'text', text: 'do a thing' }], + parent: fakeParent(), + signal: new AbortController().signal, + ...overrides, + } +} + class StubProvider implements SubagentProvider { - startCount = 0 readonly inheritsParentContext = false + startCount = 0 + constructor( readonly name: string, readonly capabilities: SubagentCapabilities = ALL_CAPS, - private readonly result: SubagentResult = { output: [{ type: 'text', text: 'ok' }], stopReason: 'completed' }, + private readonly outcome: SubagentResult = { + output: [{ type: 'text', text: 'ok' }], + stopReason: 'completed', + }, ) {} - start(request: SubagentStartRequest): SubagentRun { - this.startCount++ + async start(request: SubagentStartRequest): Promise { + this.startCount += 1 return { id: AgentId(`child:${this.name}:${request.parent.id}`), - started: Promise.resolve(), - result: Promise.resolve(this.result), - cancel() {}, + result: Promise.resolve(this.outcome), async dispose() {}, } } } -function baseRequest(overrides: Partial = {}): SubagentStartRequest { - return { prompt: [{ type: 'text', text: 'do a thing' }], parent: fakeParent(), ...overrides } +async function service(): Promise<{ ctx: Context; subagents: SubagentService }> { + const ctx = new Context() + await ctx.plugin(SubagentService) + return { ctx, subagents: ctx.subagents } } describe('SubagentService', () => { - it('announces provider lifecycle: added on register, removed on dispose', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) + it('registers, lists, looks up, starts, and removes providers', async () => { + const { ctx, subagents } = await service() const added: string[] = [] const removed: string[] = [] ctx.on('subagent/provider-added', provider => void added.push(provider.name)) ctx.on('subagent/provider-removed', name => void removed.push(name)) - - const dispose = ctx.subagents.registerProvider(new StubProvider('alpha')) - expect(added).toEqual(['alpha']) - expect(removed).toEqual([]) - - await dispose() - expect(removed).toEqual(['alpha']) - }) - - it('rolls back the registration when a provider-added listener throws', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - let threw = false - const off = ctx.on('subagent/provider-added', () => { - if (!threw) { threw = true; throw new Error('boom added listener') } - }) - - expect(() => ctx.subagents.registerProvider(new StubProvider('alpha'))).toThrow('boom added listener') - expect(ctx.subagents.getProvider('alpha')).toBeUndefined() // nothing leaked - - off() - ctx.subagents.registerProvider(new StubProvider('alpha')) - expect(ctx.subagents.getProvider('alpha')).toBeDefined() - }) - - it('contains a throwing provider-removed listener: later mirrors still hear it, teardown completes', async () => { - // provider-removed fires inside the registration's DISPOSER, so a - // propagating listener would disrupt the backend's teardown; and cordis - // emit halts on the first throw, so an uncontained one would starve every - // mirror registered after it (a stale model-facing tool). Both are - // prevented by per-listener containment. - const ctx = new Context() - await ctx.plugin(SubagentService) - const warnings: string[] = [] - ctx.logger.warn = ((message: unknown) => void warnings.push(String(message))) as typeof ctx.logger.warn - ctx.on('subagent/provider-removed', () => { throw new Error('boom removed listener') }) - const heard: string[] = [] - ctx.on('subagent/provider-removed', name => void heard.push(name)) - - const dispose = ctx.subagents.registerProvider(new StubProvider('alpha')) - expect(() => void dispose()).not.toThrow() - expect(heard).toEqual(['alpha']) // the listener AFTER the thrower still ran - expect(ctx.subagents.getProvider('alpha')).toBeUndefined() // teardown reached quiescence - expect(warnings.some(w => w.includes('boom removed listener'))).toBe(true) - }) - - it('registers a provider and starts a run on it by name', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) const provider = new StubProvider('alpha') - ctx.subagents.registerProvider(provider) - expect(ctx.subagents.list()).toEqual(['alpha']) - expect(ctx.subagents.getProvider('alpha')).toMatchObject({ name: 'alpha' }) - - const run = ctx.subagents.start('alpha', baseRequest()) - expect(provider.startCount).toBe(1) - await expect(run.result).resolves.toMatchObject({ stopReason: 'completed' }) - }) - - it('lets multiple providers coexist (the defining requirement)', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider(new StubProvider('spawn')) - ctx.subagents.registerProvider(new StubProvider('acp')) - - expect(ctx.subagents.list()).toEqual(['spawn', 'acp']) - expect(ctx.subagents.getProvider('spawn')).toBeDefined() - expect(ctx.subagents.getProvider('acp')).toBeDefined() - }) - - it('throws NO_PROVIDER when starting on an unregistered name', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - try { - ctx.subagents.start('missing', baseRequest()) - expect.fail('expected NO_PROVIDER') - } catch (error: unknown) { - expect(error).toBeInstanceOf(SubagentError) - expect((error as SubagentError).code).toBe('NO_PROVIDER') - } - }) - - it('rejects duplicate provider names with DUPLICATE_PROVIDER', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider(new StubProvider('dup')) - try { - ctx.subagents.registerProvider(new StubProvider('dup')) - expect.fail('expected DUPLICATE_PROVIDER') - } catch (error: unknown) { - expect(error).toBeInstanceOf(SubagentError) - expect((error as SubagentError).code).toBe('DUPLICATE_PROVIDER') - } - }) - - it('claims wrapper disposal before a raw provider disposer can reenter it', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const observed: { reentrant?: Promise } = {} - const providerDispose = vi.fn(() => { - observed.reentrant = run.dispose() - return Promise.resolve() - }) - ctx.subagents.registerProvider({ - name: 'dispose-reentry', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ - id: AgentId('dispose-reentry-child'), - started: Promise.resolve(), - result: Promise.resolve({ output: [], stopReason: 'completed' }), - cancel() {}, - dispose: providerDispose, - }), - }) - const run = ctx.subagents.start('dispose-reentry', baseRequest()) - - const disposal = run.dispose() - - expect(observed.reentrant).toBe(disposal) - await disposal - expect(providerDispose).toHaveBeenCalledOnce() - }) - - it('rejects a raw disposer that directly returns its reentrant wrapper promise instead of hanging', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const providerDispose = vi.fn(() => run.dispose()) - ctx.subagents.registerProvider({ - name: 'dispose-self-cycle', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ - id: AgentId('dispose-self-cycle-child'), - started: Promise.resolve(), - result: Promise.resolve({ output: [], stopReason: 'completed' }), - cancel() {}, - dispose: providerDispose, - }), - }) - const run = ctx.subagents.start('dispose-self-cycle', baseRequest()) - - await expect(run.dispose()).rejects.toThrow('run dispose returned its own wrapper disposal promise') - expect(providerDispose).toHaveBeenCalledOnce() - }) - - it.each([ - { label: 'a non-string name', patch: { name: 42 }, message: 'name must be a string' }, - { label: 'null capabilities', patch: { capabilities: null }, message: 'capabilities must be an object' }, - { label: 'primitive capabilities', patch: { capabilities: 42 }, message: 'capabilities must be an object' }, - { label: 'array capabilities', patch: { capabilities: [] }, message: 'capabilities must be an object' }, - { - label: 'a non-boolean outputSchema capability', - patch: { capabilities: { ...NO_CAPS, outputSchema: 'yes' } }, - message: 'capability "outputSchema" must be a boolean', - }, - { - label: 'a non-boolean depthLimit capability', - patch: { capabilities: { ...NO_CAPS, depthLimit: 'yes' } }, - message: 'capability "depthLimit" must be a boolean', - }, - { - label: 'a non-boolean toolFilter capability', - patch: { capabilities: { ...NO_CAPS, toolFilter: 'yes' } }, - message: 'capability "toolFilter" must be a boolean', - }, - { - label: 'a non-boolean persona capability', - patch: { capabilities: { ...NO_CAPS, persona: 'yes' } }, - message: 'capability "persona" must be a boolean', - }, - { label: 'a non-boolean conversation-history descriptor', patch: { inheritsParentContext: 'yes' }, message: 'inheritsParentContext must be a boolean' }, - { label: 'a non-callable start field', patch: { start: 42 }, message: 'start must be a function' }, - ])('rejects a provider registration with $label before entering the registry', async ({ patch, message }) => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const provider = Object.assign(new StubProvider('invalid'), patch) - - expect(() => ctx.subagents.registerProvider(provider as unknown as SubagentProvider)).toThrow(message) - expect(ctx.subagents.list()).toEqual([]) - expect(Object.isFrozen(provider)).toBe(false) - }) - - it('reads every registration field once and binds the accepted start callback to the provider', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const reads = { - name: 0, - capabilities: 0, - outputSchema: 0, - depthLimit: 0, - toolFilter: 0, - persona: 0, - inheritsParentContext: 0, - start: 0, - } - const capabilities = Object.defineProperties({}, { - outputSchema: { enumerable: true, get: () => { reads.outputSchema += 1; return false } }, - depthLimit: { enumerable: true, get: () => { reads.depthLimit += 1; return false } }, - toolFilter: { enumerable: true, get: () => { reads.toolFilter += 1; return false } }, - persona: { enumerable: true, get: () => { reads.persona += 1; return false } }, - }) as SubagentCapabilities - const acceptedStart = function (this: SubagentProvider, request: SubagentStartRequest): SubagentRun { - expect(this).toBe(provider) - return { - id: AgentId(`one-read:${request.parent.id}`), - started: Promise.resolve(), - result: Promise.resolve({ output: [], stopReason: 'completed' }), - cancel() {}, - async dispose() {}, - } - } - const provider = Object.defineProperties({}, { - name: { enumerable: true, get: () => { reads.name += 1; return 'one-read' } }, - capabilities: { enumerable: true, get: () => { reads.capabilities += 1; return capabilities } }, - inheritsParentContext: { enumerable: true, get: () => { reads.inheritsParentContext += 1; return false } }, - start: { enumerable: true, get: () => { reads.start += 1; return acceptedStart } }, - }) as SubagentProvider - - ctx.subagents.registerProvider(provider) - await expect(ctx.subagents.start('one-read', baseRequest()).result).resolves.toMatchObject({ stopReason: 'completed' }) - expect(reads).toEqual({ - name: 1, - capabilities: 1, - outputSchema: 1, - depthLimit: 1, - toolFilter: 1, - persona: 1, - inheritsParentContext: 1, - start: 1, - }) - }) - - it('unregisters a provider when its owning fiber is disposed (HMR safety)', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - - const fiber = await ctx.plugin(Object.assign((inner: Context) => { - inner.subagents.registerProvider(new StubProvider('scoped')) - }, { inject: ['subagents'] })) - expect(ctx.subagents.list()).toEqual(['scoped']) - - await fiber.dispose() - expect(ctx.subagents.list()).toEqual([]) - }) - - it('snapshots a provider registration so caller mutation cannot corrupt dispatch or HMR cleanup', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const capabilities: SubagentCapabilities = { - outputSchema: true, - depthLimit: true, - toolFilter: true, - persona: true, - } - const provider = new StubProvider('stable', capabilities) - let capabilityReads = 0 - let capabilityValue = capabilities - Object.defineProperty(provider, 'capabilities', { - configurable: true, - get: () => { - capabilityReads += 1 - return capabilityValue - }, - set: (value: SubagentCapabilities) => { capabilityValue = value }, - }) - const added: SubagentProvider[] = [] - const removed: string[] = [] - ctx.on('subagent/provider-added', registered => void added.push(registered)) - ctx.on('subagent/provider-removed', name => void removed.push(name)) - const owner = await ctx.plugin({ - name: 'mutable-provider-owner', - inject: ['subagents'], - apply(pluginCtx: Context) { - pluginCtx.subagents.registerProvider(provider) - }, - }) - expect(capabilityReads).toBe(1) - const accepted = ctx.subagents.getProvider('stable') - - const mutable = provider as unknown as { - name: string - capabilities: SubagentCapabilities - inheritsParentContext: boolean - start: SubagentProvider['start'] - } - mutable.name = 'mutated' - capabilities.outputSchema = false - capabilities.depthLimit = false - capabilities.toolFilter = false - capabilities.persona = false - mutable.capabilities = NO_CAPS - mutable.inheritsParentContext = true - const replacementStart = vi.fn((_request: SubagentStartRequest): SubagentRun => { - throw new Error('replacement start must not run') - }) - mutable.start = replacementStart - - expect(added).toEqual([accepted]) - expect(accepted).not.toBe(provider) - expect(Object.isFrozen(accepted)).toBe(true) - expect(Object.isFrozen(accepted?.capabilities)).toBe(true) - expect(accepted).toMatchObject({ - name: 'stable', - capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: true }, - inheritsParentContext: false, - }) - expect(ctx.subagents.list()).toEqual(['stable']) - expect(ctx.subagents.getProvider('mutated')).toBeUndefined() - - const controller = new AbortController() - const run = ctx.subagents.start('stable', baseRequest({ - signal: controller.signal, - agentOptions: { model: 'mock' }, - outputSchema: { type: 'object', properties: { answer: { type: 'string' } } }, - maxDepth: 2, - toolFilter: { deny: ['bash'] }, - persona: 'reviewer', - })) + const dispose = subagents.registerProvider(provider) + expect(subagents.list()).toEqual(['alpha']) + expect(subagents.getProvider('alpha')).toBe(provider) + const run = await subagents.start('alpha', baseRequest()) await expect(run.result).resolves.toMatchObject({ stopReason: 'completed' }) expect(provider.startCount).toBe(1) - expect(replacementStart).not.toHaveBeenCalled() - await owner.dispose() - expect(removed).toEqual(['stable']) - expect(ctx.subagents.list()).toEqual([]) - expect(() => ctx.subagents.registerProvider(new StubProvider('stable'))).not.toThrow() - }) - - it('re-registers a name after its prior registration is disposed (not wedged)', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - - const dispose = ctx.subagents.registerProvider(new StubProvider('reuse')) - expect(ctx.subagents.list()).toEqual(['reuse']) await dispose() - expect(ctx.subagents.list()).toEqual([]) - - const disposeAgain = ctx.subagents.registerProvider(new StubProvider('reuse')) - expect(ctx.subagents.list()).toEqual(['reuse']) - await disposeAgain() - expect(ctx.subagents.list()).toEqual([]) + expect(added).toEqual(['alpha']) + expect(removed).toEqual(['alpha']) + expect(subagents.getProvider('alpha')).toBeUndefined() }) - describe('start-time capability validation (fail loud, before any child)', () => { - it.each([ - { field: 'outputSchema', request: baseRequest({ outputSchema: { type: 'object', properties: { x: { type: 'string' } } } }) }, - { field: 'maxDepth', request: baseRequest({ maxDepth: 2 }) }, - { field: 'toolFilter', request: baseRequest({ toolFilter: { deny: ['bash'] } }) }, - ])('rejects $field against a provider that lacks the capability — before start() runs', ({ request }) => { - const ctx = new Context() - return ctx.plugin(SubagentService).then(() => { - const provider = new StubProvider('weak', NO_CAPS) - ctx.subagents.registerProvider(provider) - try { - ctx.subagents.start('weak', request) - expect.fail('expected UNSUPPORTED_CAPABILITY') - } catch (error: unknown) { - expect(error).toBeInstanceOf(SubagentError) - expect((error as SubagentError).code).toBe('UNSUPPORTED_CAPABILITY') - } - // The child was never started — the check is pre-spawn. - expect(provider.startCount).toBe(0) - }) - }) - - it('allows a capability request when the provider supports it', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const provider = new StubProvider('strong', ALL_CAPS) - ctx.subagents.registerProvider(provider) - ctx.subagents.start('strong', baseRequest({ outputSchema: { type: 'object', properties: { x: { type: 'string' } } }, maxDepth: 1 })) - expect(provider.startCount).toBe(1) - }) - - it.each([ - { label: 'null', value: null as unknown as number }, - { label: 'a string', value: '1' as unknown as number }, - { label: 'NaN', value: Number.NaN }, - { label: 'positive infinity', value: Number.POSITIVE_INFINITY }, - { label: 'negative infinity', value: Number.NEGATIVE_INFINITY }, - { label: 'a fraction', value: 1.5 }, - { label: 'a negative integer', value: -1 }, - { label: 'negative zero', value: -0 }, - { label: 'an unsafe integer', value: Number.MAX_SAFE_INTEGER + 1 }, - ])('rejects maxDepth=$label before the provider starts', async ({ value }) => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const provider = new StubProvider('invalid-depth', ALL_CAPS) - ctx.subagents.registerProvider(provider) - - expect(() => ctx.subagents.start('invalid-depth', baseRequest({ maxDepth: value }))) - .toThrow('subagent maxDepth must be a non-negative safe integer') - expect(provider.startCount).toBe(0) - }) - - it('rejects a non-string persona before the provider starts', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const provider = new StubProvider('invalid-persona', { ...ALL_CAPS, persona: true }) - ctx.subagents.registerProvider(provider) - - expect(() => ctx.subagents.start('invalid-persona', baseRequest({ - persona: 42 as unknown as string, - }))).toThrow('subagent persona must be a string') - expect(provider.startCount).toBe(0) - }) - - it('reads an optional capability accessor once so it cannot appear after validation', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - let accepted: SubagentStartRequest | undefined - const provider: SubagentProvider = { - name: 'weak-getter', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: (request) => { - accepted = request - return { - id: AgentId('weak-getter-child'), - started: Promise.resolve(), - result: Promise.resolve({ output: [], stopReason: 'completed' }), - cancel() {}, - async dispose() {}, - } - }, - } - ctx.subagents.registerProvider(provider) - let reads = 0 - const request = baseRequest() - Object.defineProperty(request, 'toolFilter', { - enumerable: true, - get: () => { - reads += 1 - return reads === 1 ? undefined : { deny: ['bash'] } - }, - }) - - ctx.subagents.start('weak-getter', request) - - expect(reads).toBe(1) - expect(accepted?.toolFilter).toBeUndefined() - }) + it('rolls registration back when provider-added throws', async () => { + const { ctx, subagents } = await service() + ctx.on('subagent/provider-added', () => { throw new Error('added boom') }) + expect(() => { subagents.registerProvider(new StubProvider('alpha')) }).toThrow('added boom') + expect(subagents.getProvider('alpha')).toBeUndefined() }) - it('rejects an exotic public prompt before the provider starts', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const provider = new StubProvider('prompt-boundary') - ctx.subagents.registerProvider(provider) - class ExoticTextBlock { - readonly type = 'text' - readonly text = 'hello' - } + it('rejects duplicate and absent provider names with typed errors', async () => { + const { subagents } = await service() + subagents.registerProvider(new StubProvider('dup')) + expect(() => { subagents.registerProvider(new StubProvider('dup')) }) + .toThrow(expect.objectContaining({ code: 'DUPLICATE_PROVIDER' })) + await expect(subagents.start('missing', baseRequest())) + .rejects.toMatchObject({ code: 'NO_PROVIDER' }) + }) - expect(() => ctx.subagents.start('prompt-boundary', baseRequest({ - prompt: [new ExoticTextBlock()] as unknown as SubagentStartRequest['prompt'], - }))).toThrow('subagent prompt must be losslessly JSON-serializable') + it.each([ + ['outputSchema', { outputSchema: { type: 'object', properties: {} } }], + ['depthLimit', { maxDepth: 1 }], + ['toolFilter', { toolFilter: { deny: ['bash'] } }], + ['persona', { persona: 'reviewer' }], + ] as const)('rejects unsupported %s before provider startup', async (_capability, override) => { + const { subagents } = await service() + const provider = new StubProvider('weak', NO_CAPS) + subagents.registerProvider(provider) + await expect(subagents.start('weak', baseRequest(override))) + .rejects.toMatchObject({ code: 'UNSUPPORTED_CAPABILITY' }) expect(provider.startCount).toBe(0) }) - it.each([ - { - label: 'agent options', - overrides: { agentOptions: { model: Number.NaN as unknown as string } }, - message: 'subagent agent options must be losslessly JSON-serializable', - }, - { - label: 'tool filter', - overrides: { toolFilter: { deny: [Number.NaN as unknown as string] } }, - message: 'subagent tool filter must be losslessly JSON-serializable', - }, - { - label: 'output schema', - overrides: { - outputSchema: { - type: 'object', - properties: { answer: { type: Number.NaN } }, - } as unknown as NonNullable, - }, - message: 'schema annotation must be JSON data', - }, - ])('rejects non-JSON $label before the provider starts', async ({ overrides, message }) => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const provider = new StubProvider('invalid-request-data', ALL_CAPS) - ctx.subagents.registerProvider(provider) - - expect(() => ctx.subagents.start('invalid-request-data', baseRequest(overrides))) - .toThrow(message) + it('validates depth and schema semantics before provider startup', async () => { + const { subagents } = await service() + const provider = new StubProvider('strong') + subagents.registerProvider(provider) + await expect(subagents.start('strong', baseRequest({ maxDepth: -1 }))) + .rejects.toThrow('non-negative safe integer') + await expect(subagents.start('strong', baseRequest({ outputSchema: { type: 'string' } as never }))) + .rejects.toThrow() expect(provider.startCount).toBe(0) + expect(() => { assertSubagentMaxDepth(undefined) }).not.toThrow() }) - it('reads each nested prompt value once into the provider snapshot', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const provider = new StubProvider('unstable-prompt') - ctx.subagents.registerProvider(provider) - let reads = 0 - const block = Object.defineProperties({}, { - type: { enumerable: true, value: 'text' }, - text: { - enumerable: true, - get: () => { - reads += 1 - return reads === 1 ? 'hello' : new Map([['not', 'json']]) - }, - }, - }) - - expect(() => ctx.subagents.start('unstable-prompt', baseRequest({ - prompt: [block] as unknown as SubagentStartRequest['prompt'], - }))).not.toThrow() - expect(reads).toBe(1) - expect(provider.startCount).toBe(1) - }) - - it('emits subagent/start then subagent/end around a run', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider(new StubProvider('events')) - - const started = vi.fn() - const ended = vi.fn() - ctx.on('subagent/start', started) - ctx.on('subagent/end', ended) - - const run = ctx.subagents.start('events', baseRequest()) - await run.started - expect(started).toHaveBeenCalledWith(expect.objectContaining({ provider: 'events', id: run.id })) - - await run.result - // `subagent/end` fires from a `.then` on the result — let the microtask run. - await Promise.resolve() - expect(ended).toHaveBeenCalledWith(expect.objectContaining({ provider: 'events', id: run.id, stopReason: 'completed' })) - }) - - it('captures a provider run once and gives callers and telemetry one normalized result', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const reads = { - id: 0, - started: 0, - result: 0, - cancel: 0, - sendMessage: 0, - dispose: 0, - resume: 0, - output: 0, - structured: 0, - stopReason: 0, - } - const methodReceivers: string[] = [] - const providerResult = Object.defineProperties({}, { - output: { - enumerable: true, - get: () => { - reads.output += 1 - return reads.output === 1 - ? [{ type: 'text', text: 'accepted output' }] - : [{ type: 'text', text: 'drifted output' }] - }, - }, - structured: { - enumerable: true, - get: () => { - reads.structured += 1 - return { verdict: reads.structured === 1 ? 'accepted' : 'drifted' } - }, - }, - stopReason: { - enumerable: true, - get: () => { - reads.stopReason += 1 - return reads.stopReason === 1 ? 'completed' : 'error' - }, - }, - }) as SubagentResult - const providerRun = Object.defineProperties({}, { - id: { - enumerable: true, - get: () => { - reads.id += 1 - return AgentId(reads.id === 1 ? 'accepted-child' : 'drifted-child') - }, - }, - started: { - enumerable: true, - get: () => { - reads.started += 1 - if (reads.started !== 1) throw new Error('started reread') - return Promise.resolve() - }, - }, - result: { - enumerable: true, - get: () => { - reads.result += 1 - if (reads.result !== 1) throw new Error('result reread') - return Promise.resolve(providerResult) - }, - }, - cancel: { - enumerable: true, - get: () => { - reads.cancel += 1 - if (reads.cancel !== 1) throw new Error('cancel reread') - return function (this: SubagentRun): void { - expect(this).toBe(providerRun) - methodReceivers.push('cancel') - } - }, - }, - sendMessage: { - enumerable: true, - get: () => { - reads.sendMessage += 1 - if (reads.sendMessage !== 1) throw new Error('sendMessage reread') - return function (this: SubagentRun): void { - expect(this).toBe(providerRun) - methodReceivers.push('sendMessage') - } - }, - }, - dispose: { - enumerable: true, - get: () => { - reads.dispose += 1 - if (reads.dispose !== 1) throw new Error('dispose reread') - return async function (this: SubagentRun): Promise { - expect(this).toBe(providerRun) - methodReceivers.push('dispose') - } - }, - }, - resume: { - enumerable: true, - get: () => { - reads.resume += 1 - if (reads.resume !== 1) throw new Error('resume reread') - return function (this: SubagentRun): SubagentRun { - expect(this).toBe(providerRun) - methodReceivers.push('resume') - return providerRun - } - }, - }, - }) as SubagentRun - ctx.subagents.registerProvider({ - name: 'stateful-run', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => providerRun, - }) - const ended = vi.fn() - ctx.on('subagent/end', ended) - - const run = ctx.subagents.start('stateful-run', baseRequest()) - expect(Object.is(run, providerRun)).toBe(false) - expect(Object.isFrozen(run)).toBe(true) - run.cancel() - run.sendMessage?.([]) - expect(Object.is(run.resume?.([]), providerRun)).toBe(true) - await run.dispose() - const result = await run.result - await run.started - await Promise.resolve() - - expect(reads).toEqual({ - id: 1, - started: 1, - result: 1, - cancel: 1, - sendMessage: 1, - dispose: 1, - resume: 1, - output: 1, - structured: 1, - stopReason: 1, - }) - expect(methodReceivers).toEqual(['cancel', 'sendMessage', 'resume', 'dispose']) - expect(result).toEqual({ - output: [{ type: 'text', text: 'accepted output' }], - structured: { verdict: 'accepted' }, - stopReason: 'completed', - }) - expect(Object.isFrozen(result)).toBe(true) - expect(Object.isFrozen(result.output)).toBe(true) - expect(ended).toHaveBeenCalledWith({ - provider: 'stateful-run', - id: 'accepted-child', - stopReason: 'completed', - lastAssistantMessage: [{ type: 'text', text: 'accepted output' }], - }) - }) - - it.each([ - { label: 'a non-string id', field: 'id', value: 42, message: 'run id must be a string' }, - { label: 'a non-Promise started field', field: 'started', value: undefined, message: 'run started must be a Promise' }, - { label: 'a non-Promise result field', field: 'result', value: undefined, message: 'run result must be a Promise' }, - { label: 'a non-callable cancel field', field: 'cancel', value: undefined, message: 'run cancel must be a function' }, - { label: 'a non-callable sendMessage field', field: 'sendMessage', value: 42, message: 'run sendMessage must be a function' }, - { label: 'a non-callable resume field', field: 'resume', value: 42, message: 'run resume must be a function' }, - ])('rolls back a provider run with $label', async ({ field, value, message }) => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const providerDispose = vi.fn(async () => {}) - const providerRun = { - id: AgentId('invalid-handle-child'), - started: Promise.resolve(), - result: Promise.resolve({ output: [], stopReason: 'completed' } satisfies SubagentResult), - cancel() {}, - dispose: providerDispose, - [field]: value, - } as unknown as SubagentRun - ctx.subagents.registerProvider({ - name: 'invalid-handle', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => providerRun, - }) - - expect(() => ctx.subagents.start('invalid-handle', baseRequest())).toThrow(message) - expect(providerDispose).toHaveBeenCalledOnce() - }) - - it.each([ - { label: 'null', value: null }, - { label: 'a primitive', value: 42 }, - ])('rejects $label returned by provider.start before reading a disposer', async ({ value }) => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider({ - name: 'invalid-run-shell', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => value as unknown as SubagentRun, - }) - - expect(() => ctx.subagents.start('invalid-run-shell', baseRequest())).toThrow('must return a SubagentRun object') - }) - - it('rejects a run without a callable disposer before accepting ownership', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider({ - name: 'invalid-dispose', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ dispose: 42 }) as unknown as SubagentRun, - }) - - expect(() => ctx.subagents.start('invalid-dispose', baseRequest())).toThrow('run dispose must be a function') - }) - - it('observes accepted provider promises when a later handle field is malformed', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const providerDispose = vi.fn(async () => {}) - ctx.subagents.registerProvider({ - name: 'rejected-malformed-handle', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ - id: AgentId('rejected-malformed-child'), - started: Promise.reject(new Error('readiness already rejected')), - result: Promise.reject(new Error('result already rejected')), - cancel: 42, - dispose: providerDispose, - }) as unknown as SubagentRun, - }) - - expect(() => ctx.subagents.start('rejected-malformed-handle', baseRequest())).toThrow('run cancel must be a function') - expect(providerDispose).toHaveBeenCalledOnce() - // Let both provider rejections run: the seam's immediate observers keep - // them from surfacing as unhandled after no wrapper was returned. - await Promise.resolve() - }) - - it('starts rollback before surfacing a hostile run accessor failure', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const disposalGate = Promise.withResolvers() - const order: string[] = [] - const providerRun = Object.defineProperties({}, { - dispose: { - get: () => { - order.push('dispose:get') - return async function (this: SubagentRun): Promise { - expect(this).toBe(providerRun) - order.push('dispose:call') - await disposalGate.promise - order.push('dispose:quiescent') - } - }, - }, - id: { get: () => { order.push('id:get'); return AgentId('hostile-handle-child') } }, - started: { get: () => { order.push('started:get'); return Promise.resolve() } }, - result: { get: () => { order.push('result:get'); throw new Error('result accessor exploded') } }, - }) as SubagentRun - ctx.subagents.registerProvider({ - name: 'hostile-handle', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => providerRun, - }) - - expect(() => ctx.subagents.start('hostile-handle', baseRequest())).toThrow('result accessor exploded') - expect(order).toEqual(['dispose:get', 'id:get', 'started:get', 'result:get', 'dispose:call']) - disposalGate.resolve(undefined) - await vi.waitFor(() => { expect(order).toContain('dispose:quiescent') }) - }) - - it('rolls back when binding a hostile optional run method fails', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const providerDispose = vi.fn(async () => {}) - const hostileCancel = new Proxy(() => {}, { - get(_target, property) { - if (property === 'length') throw new Error('cancel bind exploded') - return undefined - }, - }) - ctx.subagents.registerProvider({ - name: 'hostile-bind', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ - id: AgentId('hostile-bind-child'), - started: Promise.resolve(), - result: Promise.resolve({ output: [], stopReason: 'completed' }), - cancel: hostileCancel, - dispose: providerDispose, - }), - }) - - expect(() => ctx.subagents.start('hostile-bind', baseRequest())).toThrow('cancel bind exploded') - expect(providerDispose).toHaveBeenCalledOnce() - }) - - it.each([ - { label: 'an Error', thrown: new Error('cleanup exploded'), warning: 'cleanup exploded' }, - { label: 'a non-Error value', thrown: 'naked cleanup fault', warning: 'dispose threw a non-Error value' }, - ])('contains rollback failure from $label while preserving the malformed-handle fault', async ({ thrown, warning }) => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const warnings: string[] = [] - ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn - ctx.subagents.registerProvider({ - name: 'rollback-failure', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ - id: 42, - started: Promise.resolve(), - result: Promise.resolve({ output: [], stopReason: 'completed' }), - cancel() {}, - dispose: () => { - // Deliberately violate the seam contract to exercise normalization. - throw thrown - }, - }) as unknown as SubagentRun, - }) - - expect(() => ctx.subagents.start('rollback-failure', baseRequest())).toThrow('run id must be a string') - await vi.waitFor(() => { expect(warnings.some(message => message.includes(warning))).toBe(true) }) - }) - - it('waits for provider readiness and observes an early result rejection without reordering lifecycle', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const readiness = Promise.withResolvers() - ctx.subagents.registerProvider({ - name: 'delayed-start', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ - id: AgentId('delayed-child'), - started: readiness.promise, - // Already rejected: SubagentService must attach its result handler in - // the same synchronous start() call, before awaiting readiness. - result: Promise.reject(new Error('early infrastructure fault')), - cancel() {}, - async dispose() {}, - }), - }) - const lifecycle: string[] = [] - ctx.on('subagent/start', () => void lifecycle.push('start')) - ctx.on('subagent/end', info => void lifecycle.push(`end:${info.stopReason}`)) - - const run = ctx.subagents.start('delayed-start', baseRequest()) - await expect(run.result).rejects.toThrow('early infrastructure fault') - expect(lifecycle).toEqual([]) - - readiness.resolve(undefined) - await run.started - expect(lifecycle).toEqual(['start', 'end:error']) - }) - - it('emits no lifecycle pair when readiness rejects before a child exists', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const readiness = Promise.withResolvers() + it('publishes lifecycle only after async provider start and keeps parent scope', async () => { + const { ctx, subagents } = await service() + const ready = Promise.withResolvers() const result = Promise.withResolvers() - ctx.subagents.registerProvider({ - name: 'never-started', + subagents.registerProvider({ + name: 'deferred', capabilities: NO_CAPS, inheritsParentContext: false, - start: () => ({ - id: AgentId('never-started-child'), - started: readiness.promise, - result: result.promise, - cancel() {}, - async dispose() {}, - }), + start: () => ready.promise, + }) + const parent = fakeParent('delegator') + const events: string[] = [] + const keys: unknown[] = [] + ctx.on('subagent/start', function () { events.push('start'); keys.push(carrierKeyOf(this)) }) + ctx.on('subagent/end', function () { events.push('end'); keys.push(carrierKeyOf(this)) }) + + const starting = subagents.start('deferred', baseRequest({ parent })) + await Promise.resolve() + expect(events).toEqual([]) + ready.resolve({ id: AgentId('child'), result: result.promise, async dispose() {} }) + const run = await starting + expect(events).toEqual(['start']) + result.resolve({ output: [{ type: 'text', text: 'answer' }], stopReason: 'completed' }) + await run.result + await Promise.resolve() + expect(events).toEqual(['start', 'end']) + expect(keys).toEqual([parent, parent]) + }) + + it('emits no run lifecycle when provider startup rejects', async () => { + const { ctx, subagents } = await service() + subagents.registerProvider({ + name: 'failed', + capabilities: NO_CAPS, + inheritsParentContext: false, + start: async () => { throw new Error('setup rolled back') }, }) const lifecycle = vi.fn() ctx.on('subagent/start', lifecycle) ctx.on('subagent/end', lifecycle) - - const run = ctx.subagents.start('never-started', baseRequest()) - readiness.reject(new Error('publication rolled back')) - await expect(run.started).rejects.toThrow('publication rolled back') - result.resolve({ output: [], stopReason: 'aborted' }) - await run.result - await Promise.resolve() + await expect(subagents.start('failed', baseRequest())).rejects.toThrow('setup rolled back') expect(lifecycle).not.toHaveBeenCalled() }) - it('pins start and end to the parent accepted at start despite caller mutation', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const gate = Promise.withResolvers() - let acceptedRequest: SubagentStartRequest | undefined - ctx.subagents.registerProvider({ - name: 'deferred', + it('emits an enriched end event and maps result rejection to error telemetry', async () => { + const { ctx, subagents } = await service() + const completed = new StubProvider('completed', NO_CAPS, { + output: [{ type: 'text', text: 'answer' }], + stopReason: 'completed', + }) + subagents.registerProvider(completed) + const ended = vi.fn() + ctx.on('subagent/end', ended) + const run = await subagents.start('completed', baseRequest()) + await run.result + await Promise.resolve() + expect(ended).toHaveBeenCalledWith(expect.objectContaining({ + provider: 'completed', + lastAssistantMessage: [{ type: 'text', text: 'answer' }], + stopReason: 'completed', + })) + + const failure = Promise.withResolvers() + subagents.registerProvider({ + name: 'infra', capabilities: NO_CAPS, inheritsParentContext: false, - start: (accepted) => { - acceptedRequest = accepted - return { - id: AgentId('deferred-child'), - started: Promise.resolve(), - result: gate.promise, - cancel() {}, - async dispose() {}, - } + async start() { + return { id: AgentId('infra-child'), result: failure.promise, async dispose() {} } }, }) - const accepted = fakeParent('accepted-parent') - const replacement = fakeParent('replacement-parent') - const keys: unknown[] = [] - ctx.on('subagent/start', function () { keys.push(carrierKeyOf(this)) }) - ctx.on('subagent/end', function () { keys.push(carrierKeyOf(this)) }) - const request = baseRequest({ parent: accepted }) - - const run = ctx.subagents.start('deferred', request) - request.parent = replacement - request.prompt[0] = { type: 'text', text: 'mutated prompt' } - expect(acceptedRequest?.parent).toBe(accepted) - expect(acceptedRequest?.prompt).toEqual([{ type: 'text', text: 'do a thing' }]) - expect(acceptedRequest?.prompt).not.toBe(request.prompt) - gate.resolve({ output: [], stopReason: 'completed' }) - await run.result + const failedRun = await subagents.start('infra', baseRequest()) + failure.reject(new Error('transport')) + await expect(failedRun.result).rejects.toThrow('transport') await Promise.resolve() - - expect(keys).toEqual([accepted, accepted]) + expect(ended).toHaveBeenCalledWith(expect.objectContaining({ provider: 'infra', stopReason: 'error' })) }) - it('carries lastAssistantMessage (the child output) onto the end event', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider(new StubProvider( - 'enriched', - ALL_CAPS, - { output: [{ type: 'text', text: 'the child answer' }], stopReason: 'completed' }, - )) - - const started = vi.fn() - const ended = vi.fn() - ctx.on('subagent/start', started) - ctx.on('subagent/end', ended) - - const run = ctx.subagents.start('enriched', baseRequest()) - await run.started - expect(started).toHaveBeenCalledWith(expect.objectContaining({ provider: 'enriched', id: run.id })) - - await run.result - await Promise.resolve() - expect(ended).toHaveBeenCalledWith(expect.objectContaining({ - provider: 'enriched', - id: run.id, - stopReason: 'completed', - lastAssistantMessage: [{ type: 'text', text: 'the child answer' }], - })) - }) - - it('observe-only: a mutating subagent/end listener cannot corrupt the caller or later listeners', async () => { - // The subagent/end emit fires from a detached `.then` registered before - // start() returns — i.e. BEFORE the caller's own `await run.result` - // continuation. If the event shared the result.output reference, a mutating - // listener would change the SubagentResult the caller consumes or the value - // a later observer sees. The service freezes one normalized result and the - // lifecycle payload before dispatching either public surface. - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider(new StubProvider( - 'clone', - ALL_CAPS, - { output: [{ type: 'text', text: 'original' }], stopReason: 'completed' }, - )) - - ctx.on('subagent/end', (info) => { - // A hostile/buggy listener reaches in and mutates the event's array. - const blocks = info.lastAssistantMessage - if (blocks?.[0]?.type === 'text') blocks[0].text = 'HIJACKED' - blocks?.push({ type: 'text', text: 'injected' }) - }) - const later = vi.fn() - ctx.on('subagent/end', later) - - const run = ctx.subagents.start('clone', baseRequest()) - const result = await run.result - await Promise.resolve() // let the detached settle hook (and its listener) run - // The caller and the listener after the mutator both retain the accepted value. - expect(result.output).toEqual([{ type: 'text', text: 'original' }]) - expect(Object.isFrozen(result.output)).toBe(true) - expect(later).toHaveBeenCalledWith(expect.objectContaining({ - stopReason: 'completed', - lastAssistantMessage: [{ type: 'text', text: 'original' }], - })) - const laterInfo = later.mock.calls[0]![0] as Record - expect(Object.isFrozen(laterInfo)).toBe(true) - expect(Object.isFrozen(laterInfo.lastAssistantMessage)).toBe(true) - }) - - it('omits lastAssistantMessage on the reject path (no SubagentResult was produced)', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider({ - name: 'rej', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ - id: AgentId('rej-child'), - started: Promise.resolve(), - result: Promise.reject(new Error('infra fault')), - cancel() {}, - dispose: async () => {}, - }), - }) - - const ended = vi.fn() - ctx.on('subagent/end', ended) - const run = ctx.subagents.start('rej', baseRequest()) - await run.result.catch(() => {}) - await Promise.resolve() - - const endInfo = ended.mock.calls[0]![0] as Record - expect(endInfo.stopReason).toBe('error') - expect('lastAssistantMessage' in endInfo).toBe(false) // no output exists on reject - }) - - it('rejects an invalid provider output and maps the contract fault to error telemetry', async () => { - // A function is outside the lossless JSON vocabulary. The service-owned - // result promise rejects instead of exposing the malformed provider value; - // its already-attached lifecycle observer maps that infrastructure fault to - // error telemetry without producing an unhandled rejection. - const ctx = new Context() - await ctx.plugin(SubagentService) - const nonJsonOutput = [{ type: 'text', text: 'x', evil: () => 0 }] as unknown as SubagentResult['output'] - const disposalGate = Promise.withResolvers() - const providerDispose = vi.fn(async () => { await disposalGate.promise }) - ctx.subagents.registerProvider({ - name: 'unclone', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ - id: AgentId('unclone-child'), - started: Promise.resolve(), - result: Promise.resolve({ output: nonJsonOutput, stopReason: 'completed' } as SubagentResult), - cancel() {}, - dispose: providerDispose, - }), - }) - - const ended = vi.fn() - ctx.on('subagent/end', ended) - const run = ctx.subagents.start('unclone', baseRequest()) - let resultSettled = false - void run.result.catch(() => { resultSettled = true }) - await vi.waitFor(() => { expect(providerDispose).toHaveBeenCalledOnce() }) - expect(resultSettled).toBe(false) - disposalGate.resolve(undefined) - await expect(run.result).rejects.toThrow('subagent result must be losslessly JSON-serializable') - await run.dispose() - await Promise.resolve() - - expect(providerDispose).toHaveBeenCalledOnce() - const endInfo = ended.mock.calls[0]![0] as Record - expect(endInfo.stopReason).toBe('error') - expect('lastAssistantMessage' in endInfo).toBe(false) - }) - - it.each([ - { - label: 'a non-array output', - value: { output: { type: 'text', text: 'not an array' }, stopReason: 'completed' }, - message: 'subagent result output must be an array', - }, - { - label: 'a non-string stopReason', - value: { output: [], stopReason: 42 }, - message: 'subagent result stopReason must be a string', - }, - ])('rejects a provider result with $label', async ({ value, message }) => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider({ - name: 'invalid-shape', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ - id: AgentId('invalid-shape-child'), - started: Promise.resolve(), - result: Promise.resolve(value as unknown as SubagentResult), - cancel() {}, - async dispose() {}, - }), - }) - const ended = vi.fn() - ctx.on('subagent/end', ended) - - const run = ctx.subagents.start('invalid-shape', baseRequest()) - await expect(run.result).rejects.toThrow(message) - await Promise.resolve() - - expect(ended).toHaveBeenCalledWith(expect.objectContaining({ - provider: 'invalid-shape', - id: 'invalid-shape-child', - stopReason: 'error', - })) - }) - - it('emits subagent/end with stopReason "error" when the run result promise rejects', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - // A provider whose run.result REJECTS (an infrastructure fault — the seam - // contract says child-level failures resolve with stopReason 'error', but a - // rejection is still surfaced as an 'error' telemetry event). - ctx.subagents.registerProvider({ - name: 'rejecter', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ - id: AgentId('rej-child'), - started: Promise.resolve(), - result: Promise.reject(new Error('infra fault')), - cancel() {}, - dispose: async () => {}, - }), - }) - - const ended = vi.fn() - ctx.on('subagent/end', ended) - const run = ctx.subagents.start('rejecter', baseRequest()) - // Observe (and swallow) the rejection the consumer would see, then let the - // detached `.then` settle the telemetry emit. - await run.result.catch(() => {}) - await Promise.resolve() - expect(ended).toHaveBeenCalledWith(expect.objectContaining({ provider: 'rejecter', id: run.id, stopReason: 'error' })) - }) - - it('contains a throwing subagent/start listener per-listener: a later listener still observes the event and start() returns the run', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider(new StubProvider('contain')) - // Two listeners; the FIRST throws. Per-listener containment means the second - // must STILL run (a single try/catch around ctx.emit would let the first - // throw halt the dispatch and starve the second — the round-2 regression). - const second = vi.fn() - ctx.on('subagent/start', () => { throw new Error('bad start listener') }) - ctx.on('subagent/start', second) - - const run = ctx.subagents.start('contain', baseRequest()) - expect(run.id).toBeDefined() - await run.started - expect(second).toHaveBeenCalledWith(expect.objectContaining({ provider: 'contain', id: run.id })) - await expect(run.result).resolves.toMatchObject({ stopReason: 'completed' }) - }) - - it('contains asynchronous lifecycle-listener rejections without serializing later listeners', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) + it('contains synchronous and asynchronous lifecycle observer failures', async () => { + const { ctx, subagents } = await service() const warnings: string[] = [] - ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn - const laterStart = vi.fn() - const laterEnd = vi.fn() - const laterRemoved = vi.fn() - const asyncStart = (async () => { await Promise.resolve(); throw new Error('async start listener') }) as unknown as () => void - const asyncEnd = (async () => { await Promise.resolve(); throw new Error('async end listener') }) as unknown as () => void - const asyncRemoved = (async () => { await Promise.resolve(); throw new Error('async removed listener') }) as unknown as () => void - ctx.on('subagent/start', asyncStart) - ctx.on('subagent/start', laterStart) - ctx.on('subagent/end', asyncEnd) - ctx.on('subagent/end', laterEnd) - ctx.on('subagent/provider-removed', asyncRemoved) - ctx.on('subagent/provider-removed', laterRemoved) - const unregister = ctx.subagents.registerProvider(new StubProvider('async-listeners')) + ctx.logger.warn = ((message: unknown) => void warnings.push(String(message))) as typeof ctx.logger.warn + const heard: string[] = [] + ctx.on('subagent/provider-removed', () => { throw new Error('sync boom') }) + // Runtime listeners may return thenables even though the declaration's observable result is void. + // eslint-disable-next-line @typescript-eslint/no-misused-promises -- exercises rejected-listener containment + ctx.on('subagent/provider-removed', async () => { throw new Error('async boom') }) + ctx.on('subagent/provider-removed', () => { throw { toString: () => { throw new Error('coercion') } } }) + ctx.on('subagent/provider-removed', name => void heard.push(name)) + const dispose = subagents.registerProvider(new StubProvider('contained')) - const run = ctx.subagents.start('async-listeners', baseRequest()) - await run.started - expect(laterStart).toHaveBeenCalledOnce() - await run.result - await vi.waitFor(() => { - expect(laterEnd).toHaveBeenCalledOnce() - expect(warnings.some(message => message.includes('async start listener'))).toBe(true) - expect(warnings.some(message => message.includes('async end listener'))).toBe(true) - }) - - await unregister() - expect(laterRemoved).toHaveBeenCalledWith('async-listeners') - await vi.waitFor(() => { - expect(warnings.some(message => message.includes('async removed listener'))).toBe(true) - }) - }) - - it('contains a listener whose thrown value cannot be stringified', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider(new StubProvider('hostile-listener')) - const warnings: string[] = [] - ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn - const hostile = { - [Symbol.toPrimitive]() { throw new Error('render failed') }, - } - const second = vi.fn() - ctx.on('subagent/start', () => { throw hostile }) - ctx.on('subagent/start', second) - - const run = ctx.subagents.start('hostile-listener', baseRequest()) - await run.started - - expect(second).toHaveBeenCalledOnce() + await dispose() + await Promise.resolve() + expect(heard).toEqual(['contained']) + expect(warnings.some(message => message.includes('sync boom'))).toBe(true) + expect(warnings.some(message => message.includes('async boom'))).toBe(true) expect(warnings.some(message => message.includes(''))).toBe(true) - await run.result }) - it('rejects a throwing provider result accessor and maps it to error telemetry', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider({ - name: 'hostile-result', - capabilities: NO_CAPS, - inheritsParentContext: false, - start: () => ({ - id: AgentId('hostile-result-child'), - started: Promise.resolve(), - result: Promise.resolve({ - output: [], - get stopReason(): 'completed' { throw new Error('stop reason exploded') }, - }), - cancel() {}, - async dispose() {}, - }), - }) - const ended = vi.fn() - ctx.on('subagent/end', ended) - - const run = ctx.subagents.start('hostile-result', baseRequest()) - await run.started - await expect(run.result).rejects.toThrow('stop reason exploded') - await Promise.resolve() - - expect(ended).toHaveBeenCalledWith(expect.objectContaining({ - provider: 'hostile-result', - id: 'hostile-result-child', - stopReason: 'error', - })) - }) - - it('contains a throwing subagent/end listener per-listener: a later listener still observes the settle, no unhandled rejection', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - ctx.subagents.registerProvider(new StubProvider('contain-end')) - const second = vi.fn() - ctx.on('subagent/end', () => { throw new Error('bad end listener') }) - ctx.on('subagent/end', second) - - const run = ctx.subagents.start('contain-end', baseRequest()) - await run.result - // Let the detached `.then` + the contained emit run. - await Promise.resolve() - await Promise.resolve() - expect(second).toHaveBeenCalledWith(expect.objectContaining({ provider: 'contain-end', id: run.id, stopReason: 'completed' })) - }) - - it('SubagentError extends the shared HarnessError base', () => { - const err = new SubagentError('boom', 'NO_PROVIDER') - expect(err).toBeInstanceOf(HarnessError) - expect(err.name).toBe('SubagentError') - expect(err.code).toBe('NO_PROVIDER') + it('SubagentError participates in the harness error taxonomy', () => { + const error = new SubagentError('boom', 'NO_PROVIDER') + expect(error).toBeInstanceOf(HarnessError) + expect(error.name).toBe('SubagentError') + expect(error.code).toBe('NO_PROVIDER') }) }) diff --git a/packages/subagent/subagent/tsconfig.json b/packages/subagent/subagent/tsconfig.json index 3d3c9be6a9..f93f929241 100644 --- a/packages/subagent/subagent/tsconfig.json +++ b/packages/subagent/subagent/tsconfig.json @@ -17,9 +17,6 @@ { "path": "../../core/agent" }, - { - "path": "../../core/session" - }, { "path": "../../llm/llm" }, diff --git a/packages/subagent/tool-subagent/README.md b/packages/subagent/tool-subagent/README.md index 2163d4c884..fbd01a0cb2 100644 --- a/packages/subagent/tool-subagent/README.md +++ b/packages/subagent/tool-subagent/README.md @@ -1,28 +1,28 @@ # @deepseek-ai/dsh-tool-subagent -The model-facing `subagent` tool: delegate a self-contained task to a child agent and return its final output. Pure schema + lifecycle shaping over the [`ctx.subagents`](../subagent/README.md) provider registry — an in-process, ACP, or future A2A backend swaps in without changing what the model sees. +The `subagent` tool lets the model delegate one self-contained task and collect the child's final output. It is a thin consumer of `ctx.subagents`; changing the configured provider changes the transport without changing the model-facing execution contract. -## Provider selection is config, not model-facing +## Provider selection -This plugin binds to **exactly one** provider (`Config.provider`). The model sees only `{ description, prompt }` — there is no provider/type parameter in the schema. To expose more than one transport, load the plugin more than once, each bound to a different provider **and a distinct `toolName`** (the tool registry rejects a duplicate name, so a second load that kept the default `subagent` name would throw). Keeping selection in config (not the schema) is the deliberate split: the *service* holds a multi-provider registry; the *tool* picks one. +Each plugin instance binds to exactly one provider. The model sees `{ description, prompt }`, not a provider selector. To expose multiple transports, load the plugin multiple times with distinct `toolName` values. -## The description states the provider's conversation-history descriptor +The description is derived from `provider.inheritsParentContext`: spawn and ACP tell the model to provide a standalone prompt, while fork says the child already sees completed conversation turns. The plugin follows `subagent/provider-added` and `subagent/provider-removed`, so concurrent Cordis plugin loading does not create a registration-order dependency. -The tool description and the `prompt` parameter description are DERIVED from the bound provider's `inheritsParentContext` (`providerWording`): a fresh-conversation provider (spawn, ACP) gets the standalone-prompt wording ("it does not see this conversation"), while fork tells the model the child is seeded with the conversation's completed turns and its prompt should state only what is new. The descriptor concerns conversation history only, not tool or authority inheritance. Because the description is fixed at tool registration, the tool **mirrors the provider's lifecycle** (`subagent/provider-added`/`-removed`): it registers when the bound provider is (or becomes) available and unregisters when the provider goes away — no load-order requirement (the cordis Loader starts sibling plugins concurrently, so "listed first" never guaranteed "registered first"), and an HMR reload of the backend re-derives the wording from the fresh provider. While the provider is absent the tool simply does not exist (a `ctx.logger` note records the wait; a typo'd provider name shows up as a tool that never materializes). +## Lifecycle -| Config key | Meaning | +`execute` passes the tool execution's abort signal directly as the required `SubagentStartRequest.signal`, awaits `ctx.subagents.start(...)`, then awaits `run.result` inside a `try/finally` that always calls `run.dispose()`. The same signal therefore covers startup and live execution, while disposal guarantees quiescence on success, failure, and abort. + +A non-`completed` stop reason becomes an `isError` tool result; partial child output is never reported as success. The current tool blocks the parent turn until collection finishes; background and polling modes are deferred. + +## Config + +| Key | Meaning | |---|---| -| `provider` (required) | The `ctx.subagents` provider name to start runs on (`spawn`, `fork`, `acp`, …). | -| `toolName` | The model-facing tool name to register (default `subagent`). Set a distinct value per load when exposing multiple providers, e.g. `subagent` + `subagent_acp`. | -| `agentOptions` | Default per-child `{ model? }` applied to every spawned child. | -| `persona` | Per-child persona that shadows the deployment persona; requires the provider's `persona` capability. | -| `toolFilter` | Per-child `{ allow?, deny? }` restriction over global tools; requires the provider's `toolFilter` capability. | -| `maxDepth` | Maximum absolute delegation-tree depth for every child this tool starts; a non-negative safe integer validated when this plugin loads. Requires the provider's `depthLimit` capability. | +| `provider` | Required `ctx.subagents` provider name. | +| `toolName` | Model-facing tool name (default `subagent`). Must be unique per plugin instance. | +| `agentOptions` | Default child agent options, currently including `model`. | +| `persona` | Per-child persona; requires provider `persona` capability. | +| `toolFilter` | Per-child global-tool restriction; requires provider `toolFilter` capability. | +| `maxDepth` | Absolute delegation-depth cap; requires provider `depthLimit` capability. | -`toolFilter` uses [`ToolRegistry.restrict()`](../../core/tools/README.md)'s live global-view semantics and is not a parent-derived authority ceiling; see the [agent-scope security non-goal](../../../docs/rfc/implemented/architecture/2026-07-08-agent-scope-contexts.md#security-and-authority-are-explicit-non-goals). - -## Lifecycle (synchronous collect) - -`execute` starts a run on the configured provider and **awaits `run.result` inside a `try/finally` that always `dispose()`s the run** — the owned child agent/session is torn down on every path (success, error, abort), never leaked. The tool's abort signal (`exec.signal`) is bridged to `run.cancel()`. A non-`completed` stop reason (aborted/error/max-tokens/refusal) maps to an `isError` tool result rather than returning partial output as success. - -Background / poll collection is deferred (see the [RFC](../../../docs/rfc/implemented/feature/2026-06-21-subagent-capability-seam.md)); this cut blocks the parent turn until the child finishes. +`toolFilter` changes the child's visible global tool layer; it is not a parent-derived authority ceiling. See the [agent-scope security non-goal](../../../docs/rfc/implemented/architecture/2026-07-08-agent-scope-contexts.md#security-and-authority-are-explicit-non-goals). diff --git a/packages/subagent/tool-subagent/src/index.ts b/packages/subagent/tool-subagent/src/index.ts index 25bd5e3acf..322a76e825 100644 --- a/packages/subagent/tool-subagent/src/index.ts +++ b/packages/subagent/tool-subagent/src/index.ts @@ -242,24 +242,14 @@ export function apply(ctx: Context, config: Config): void { const request: SubagentStartRequest = { prompt: [{ type: 'text', text: args.prompt }], parent, - ...exec.signal ? { signal: exec.signal } : {}, + signal: exec.signal ?? new AbortController().signal, ...config.agentOptions !== undefined ? { agentOptions: config.agentOptions } : {}, ...config.persona !== undefined ? { persona: config.persona } : {}, ...config.toolFilter !== undefined ? { toolFilter: config.toolFilter } : {}, ...config.maxDepth !== undefined ? { maxDepth: config.maxDepth } : {}, } - const run: SubagentRun = ctx.subagents.start(config.provider, request) - - // Bridge the tool's abort signal to the run: if the parent step is - // aborted while the child is in flight, cancel the child too. - const onAbort = (): void => { run.cancel('parent step aborted') } - exec.signal?.addEventListener('abort', onAbort, { once: true }) - // `addEventListener` does NOT fire for a signal already aborted before this - // line, so a step cancelled before the tool ran would never reach the - // child. Cancel explicitly in that case — the bridge must honor an - // already-aborted signal, not lean on each provider re-checking it. - if (exec.signal?.aborted) run.cancel('parent step aborted') + const run: SubagentRun = await ctx.subagents.start(config.provider, request) try { const result = await run.result @@ -271,7 +261,6 @@ export function apply(ctx: Context, config: Config): void { } return [{ type: 'text', text: outputText(result.output) }] } finally { - exec.signal?.removeEventListener('abort', onAbort) // Always reach child quiescence — never leak a live idle child/session. await run.dispose() } diff --git a/packages/subagent/tool-subagent/tests/tool-subagent.spec.ts b/packages/subagent/tool-subagent/tests/tool-subagent.spec.ts index 701a4dc667..5830513cf4 100644 --- a/packages/subagent/tool-subagent/tests/tool-subagent.spec.ts +++ b/packages/subagent/tool-subagent/tests/tool-subagent.spec.ts @@ -113,11 +113,9 @@ describe('dsh-tool-subagent', () => { name: 'weird', capabilities: { outputSchema: false, depthLimit: false, toolFilter: false, persona: false }, inheritsParentContext: false, - start: () => ({ + start: async () => ({ id: AgentId('weird-child'), - started: Promise.resolve(), result: Promise.resolve({ output: [{ type: 'text', text: 'partial' }], stopReason: 'frobnicated' as never }), - cancel() {}, dispose: async () => {}, }), }) @@ -140,13 +138,11 @@ describe('dsh-tool-subagent', () => { name: 'capture', capabilities: { outputSchema: false, depthLimit: false, toolFilter: false, persona: false }, inheritsParentContext: false, - start: (request) => { + start: async (request) => { seen = request return { id: AgentId('capture-child'), - started: Promise.resolve(), result: Promise.resolve({ output: [{ type: 'text', text: 'ok' }], stopReason: 'completed' as const }), - cancel() {}, dispose: async () => {}, } }, @@ -171,13 +167,11 @@ describe('dsh-tool-subagent', () => { name: 'bare', capabilities: { outputSchema: false, depthLimit: false, toolFilter: false, persona: false }, inheritsParentContext: false, - start: (request) => { + start: async (request) => { seen = request return { id: AgentId('bare-child'), - started: Promise.resolve(), result: Promise.resolve({ output: [{ type: 'text', text: 'ok' }], stopReason: 'completed' as const }), - cancel() {}, dispose: async () => {}, } }, @@ -302,11 +296,9 @@ describe('dsh-tool-subagent', () => { name: 'spy', capabilities: { outputSchema: false, depthLimit: false, toolFilter: false, persona: false }, inheritsParentContext: false, - start: () => ({ + start: async () => ({ id: AgentId('spy-child'), - started: Promise.resolve(), result: Promise.resolve({ output: [{ type: 'text', text: 'ok' }], stopReason: 'completed' as const }), - cancel() {}, dispose: async () => void disposed(), }), }) @@ -326,11 +318,9 @@ describe('dsh-tool-subagent', () => { name: 'spy', capabilities: { outputSchema: false, depthLimit: false, toolFilter: false, persona: false }, inheritsParentContext: false, - start: () => ({ + start: async () => ({ id: AgentId('spy-child'), - started: Promise.resolve(), result: Promise.resolve({ output: [], stopReason: 'error' as const }), - cancel() {}, dispose: async () => void disposed(), }), }) @@ -341,7 +331,7 @@ describe('dsh-tool-subagent', () => { expect(disposed).toHaveBeenCalledTimes(1) }) - it('bridges the tool abort signal to run.cancel()', async () => { + it('passes the tool abort signal as the provider cancellation channel', async () => { const cancelled = vi.fn() const ctx = new Context() await ctx.plugin(SystemPrompt) @@ -351,17 +341,17 @@ describe('dsh-tool-subagent', () => { name: 'spy', capabilities: { outputSchema: false, depthLimit: false, toolFilter: false, persona: false }, inheritsParentContext: false, - start: () => { + start: async (request) => { + if (request.signal.aborted) throw new Error('start aborted') let resolveResult: (r: { output: never[]; stopReason: 'aborted' }) => void const result = new Promise<{ output: never[]; stopReason: 'aborted' }>((res) => { resolveResult = res }) + request.signal.addEventListener('abort', () => { + cancelled() + resolveResult({ output: [], stopReason: 'aborted' }) + }, { once: true }) return { id: AgentId('spy-child'), - started: Promise.resolve(), result, - cancel: () => { - cancelled() - resolveResult({ output: [], stopReason: 'aborted' }) - }, dispose: async () => {}, } }, @@ -370,12 +360,7 @@ describe('dsh-tool-subagent', () => { const controller = new AbortController() const pending = callSubagent(ctx, { description: 'd', prompt: 'p' }, { signal: controller.signal }) - // Abort AFTER the tool body has had a chance to register its abort listener - // (ctx.tools.execute now awaits the tools/pre-execute waterfall before the - // body runs, so the listener is not registered synchronously). A few - // microtask turns let execute() reach `addEventListener('abort')`, so this - // exercises the LIVE onAbort bridge — distinct from the already-aborted - // sync path the next test covers. + // Let provider.start install its listener before aborting. await Promise.resolve() await Promise.resolve() controller.abort() @@ -384,13 +369,8 @@ describe('dsh-tool-subagent', () => { expect(result.isError).toBe(true) }) - it('cancels the run when the tool signal is ALREADY aborted before execute (no missed abort)', async () => { - // `addEventListener('abort')` does not fire for a signal already aborted - // before the listener is added, so a step cancelled before the tool ran - // would never reach the child unless the bridge re-checks `signal.aborted`. - // A provider that leans only on the abort EVENT (this spy never inspects - // request.signal) proves the bridge itself must cancel. - const cancelled = vi.fn() + it('passes an already-aborted signal so provider startup rejects', async () => { + const sawAborted = vi.fn() const ctx = new Context() await ctx.plugin(SystemPrompt) await ctx.plugin(ToolRegistry) @@ -399,19 +379,9 @@ describe('dsh-tool-subagent', () => { name: 'spy', capabilities: { outputSchema: false, depthLimit: false, toolFilter: false, persona: false }, inheritsParentContext: false, - start: () => { - let resolveResult: (r: { output: never[]; stopReason: 'aborted' }) => void - const result = new Promise<{ output: never[]; stopReason: 'aborted' }>((res) => { resolveResult = res }) - return { - id: AgentId('spy-child'), - started: Promise.resolve(), - result, - cancel: () => { - cancelled() - resolveResult({ output: [], stopReason: 'aborted' }) - }, - dispose: async () => {}, - } + start: async (request) => { + if (request.signal.aborted) sawAborted() + throw new Error('start aborted') }, }) await ctx.plugin(tool, { provider: 'spy' }) @@ -419,7 +389,7 @@ describe('dsh-tool-subagent', () => { const controller = new AbortController() controller.abort() // already aborted BEFORE the tool runs const result = await callSubagent(ctx, { description: 'd', prompt: 'p' }, { signal: controller.signal }) - expect(cancelled).toHaveBeenCalledTimes(1) + expect(sawAborted).toHaveBeenCalledTimes(1) expect(result.isError).toBe(true) }) @@ -469,13 +439,11 @@ describe('dsh-tool-subagent', () => { name: 'capture2', capabilities: { outputSchema: false, depthLimit: true, toolFilter: true, persona: true }, inheritsParentContext: false, - start: (request) => { + start: async (request) => { seen = request return { id: AgentId('capture2-child'), - started: Promise.resolve(), result: Promise.resolve({ output: [{ type: 'text', text: 'ok' }], stopReason: 'completed' as const }), - cancel() {}, dispose: async () => {}, } }, @@ -519,7 +487,7 @@ describe('dsh-tool-subagent', () => { }) it('a partial toolFilter (deny only) does not materialize an empty allow-list (deny-all trap)', async () => { - let seen: { toolFilter?: { allow?: string[]; deny?: string[] } } | undefined + let seen: { toolFilter?: { readonly allow?: readonly string[]; readonly deny?: readonly string[] } } | undefined const ctx = new Context() await ctx.plugin(SystemPrompt) await ctx.plugin(ToolRegistry) @@ -528,13 +496,11 @@ describe('dsh-tool-subagent', () => { name: 'capture3', capabilities: { outputSchema: false, depthLimit: false, toolFilter: true, persona: false }, inheritsParentContext: false, - start: (request) => { + start: async (request) => { seen = request return { id: AgentId('capture3-child'), - started: Promise.resolve(), result: Promise.resolve({ output: [{ type: 'text', text: 'ok' }], stopReason: 'completed' as const }), - cancel() {}, dispose: async () => {}, } }, @@ -559,13 +525,11 @@ describe('dsh-tool-subagent', () => { name: 'capture4', capabilities: { outputSchema: false, depthLimit: false, toolFilter: false, persona: false }, inheritsParentContext: false, - start: (request) => { + start: async (request) => { seen = request return { id: AgentId('capture4-child'), - started: Promise.resolve(), result: Promise.resolve({ output: [{ type: 'text', text: 'ok' }], stopReason: 'completed' as const }), - cancel() {}, dispose: async () => {}, } }, diff --git a/packages/support/invariants/package.json b/packages/support/invariants/package.json index ba95eea315..588c658857 100644 --- a/packages/support/invariants/package.json +++ b/packages/support/invariants/package.json @@ -26,8 +26,6 @@ "@deepseek-ai/dsh-llm": "^0.0.1", "@deepseek-ai/dsh-scope": "^0.0.1", "@deepseek-ai/dsh-session": "^0.0.1", - "@deepseek-ai/dsh-system-prompt": "^0.0.1", - "@deepseek-ai/dsh-tools": "^0.0.1", "cordis": "^4.0.0-rc.6" }, "devDependencies": { @@ -35,8 +33,6 @@ "@deepseek-ai/dsh-llm": "workspace:^", "@deepseek-ai/dsh-scope": "workspace:^", "@deepseek-ai/dsh-session": "workspace:^", - "@deepseek-ai/dsh-system-prompt": "workspace:^", - "@deepseek-ai/dsh-tools": "workspace:^", "cordis": "^4.0.0-rc.6" } } diff --git a/packages/support/invariants/src/index.ts b/packages/support/invariants/src/index.ts index 20396c4f5d..f5ef6d2b4a 100644 --- a/packages/support/invariants/src/index.ts +++ b/packages/support/invariants/src/index.ts @@ -19,8 +19,6 @@ import type { Context } from 'cordis' import { carrierKeyOf, isScopeCarrier } from '@deepseek-ai/dsh-scope' -import type { AssembleContext } from '@deepseek-ai/dsh-system-prompt' -import type { ToolExecution } from '@deepseek-ai/dsh-tools' import { assertNever, HarnessError } from '@deepseek-ai/dsh-llm' import type { CallId, GenerateOptions } from '@deepseek-ai/dsh-llm' import type { Agent, AgentStatus } from '@deepseek-ai/dsh-agent' @@ -92,6 +90,12 @@ interface AgentSubject { agent: Agent } +/** Structural subject fields used without coupling this dev plugin to owning services. */ +interface ScopedSubjectFields { + agent?: Agent + scope?: object +} + /** Assert that a step-scoped event names the currently open turn and step. */ function requireOpenStep(trace: SessionTrace, kind: string, turn: number, step: number): void { if (trace.openTurn !== turn || trace.openStep !== step) { @@ -412,32 +416,6 @@ export function apply(ctx: Context): void { lastStatus.set(agent, status) }, { global: true }) - // --- Setup-drives invariant --------------------------------------------- - // - // CreateAgentOptions.setup COMPOSES the agent's scoped world; it must not - // DRIVE the agent. ReactLoopAgent rejects every driving verb structurally - // until rollback-covered publication reaches the session-start boundary; - // this event-level invariant remains the cross-implementation backstop for - // alternate Agent implementations and raw session writes. A turn/start - // candidate before agent/session-start is rejected by internal/dispatch, - // before Session commits it. Sessions of agents that exist BEFORE this - // plugin applies are marked started (their ordering is unknowable after the - // fact — never a false positive on HMR). `agents` is read via ctx.get (a - // strict, optional store lookup) rather than injected: the invariants plugin - // must load in harnesses that carry no agent registry at all (bare session - // tests), where this check simply never trips. - const sessionStarted = new WeakSet() - for (const agent of ctx.get('agents')?.list() ?? []) sessionStarted.add(agent.session) - const assertSessionStartedBeforeTurn = (session: Session, event: SessionEvent): void => { - if (event.type !== 'turn/start' || sessionStarted.has(session)) return - const owner = ctx.get('agents')?.list().find(agent => agent.session === session) - if (owner === undefined) return - throw new InvariantError( - `agent "${owner.id}": a turn opened before agent/session-start fired — ` - + 'CreateAgentOptions.setup composes the scoped world, it must not drive the agent ' - + '(send/steer/inject belong after creation returns)') - } - // --- Scoped-dispatch invariants (the agent-scoping seam) --------------- // // Every scope-filtered event family must dispatch with a scope carrier @@ -466,11 +444,11 @@ export function apply(ctx: Context): void { 'agent/turn-stop': args => args[0], 'agent/error': args => args[0], 'approval/request': args => (args[0] as AgentSubject).agent, - 'tools/pre-execute': args => (args[0] as ToolExecution).agent, - 'tools/execute': args => (args[0] as ToolExecution).agent, - 'tools/post-execute': args => (args[0] as ToolExecution).agent, - 'tools/result': args => (args[0] as ToolExecution).agent, - 'system-prompt/assemble': args => (args[1] as AssembleContext).scope, + 'tools/pre-execute': args => (args[0] as ScopedSubjectFields).agent, + 'tools/execute': args => (args[0] as ScopedSubjectFields).agent, + 'tools/post-execute': args => (args[0] as ScopedSubjectFields).agent, + 'tools/result': args => (args[0] as ScopedSubjectFields).agent, + 'system-prompt/assemble': args => (args[1] as ScopedSubjectFields).scope, 'session/created': null, 'session/disposed': null, 'session/event': null, @@ -491,33 +469,16 @@ export function apply(ctx: Context): void { `"${name}" was dispatched with a scope carrier keyed to a DIFFERENT subject than its arguments name — ` + 'the carrier key and the event\'s subject must be the same object (use agentEvents(ctx, agent))') } - if (name === 'agent/session-start') { - // Mark before product listeners run: a prepended session-start listener is - // explicitly allowed to inject the first turn's context synchronously. - sessionStarted.add((args[0] as Agent).session) - } if (name === 'session/event') { const [session, event] = args as [Session, SessionEvent] const trace = traceFor(session) const transition = validateEvent(trace, event) - assertSessionStartedBeforeTurn(session, event) // The exact event identity reaches the contained post-commit listener. // A later internal/dispatch listener may still veto; because validation // is pure, abandoning this weakly keyed transition does not advance the // committed trace or retain the session. stagedTransitions.set(event, { session, trace, transition }) } - // The assembly context must never carry the agent DX field without the - // scope layer selector: the assembly would silently miss the agent's - // scoped sections/tools (use assembleContextFor(agent)). - if (name === 'system-prompt/assemble') { - const context = args[1] as AssembleContext - if (context.agent !== undefined && context.scope !== context.agent) { - throw new InvariantError( - 'an assembly context carries `agent` without `scope` (or with a mismatched scope) — ' - + 'use assembleContextFor(agent) so the assembly resolves the agent\'s scoped layer') - } - } }, { global: true }) // Request-reconstruction cross-check (the reconstructability RFC): a diff --git a/packages/support/invariants/tests/invariants.spec.ts b/packages/support/invariants/tests/invariants.spec.ts index efbf1dd978..5c26b1f4d7 100644 --- a/packages/support/invariants/tests/invariants.spec.ts +++ b/packages/support/invariants/tests/invariants.spec.ts @@ -79,34 +79,6 @@ describe('session-log invariants', () => { expect(session.events.map(event => event.type)).toEqual(['turn/start', 'turn/end']) }) - it('does not stage a substituted candidate from prepended internal instrumentation', async () => { - const { ctx } = await setup() - const session = ctx.sessions.create(SessionId('dispatch-substitution-rollback')) - let substitute = true - const replacement = { - type: 'turn/start', - seq: 0, - time: 1, - data: { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } }, - } as const - ctx.on('internal/dispatch', (_mode, name, args) => { - if (name !== 'session/event' || !substitute) return - substitute = false - args[1] = replacement - }, { prepend: true }) - - expect(() => session.append('turn/start', { - turn: 1, - trigger: { kind: 'message', source: { kind: 'user' } }, - })).toThrow('session/event internal dispatch replaced the accepted callback tuple') - expect(session.events).toEqual([]) - - expect(() => { - session.append('turn/start', { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } }) - session.append('turn/end', { turn: 1, reason: { kind: 'completed' } }) - }).not.toThrow() - }) - it('applies the committed transition after a prepended observer throws', async () => { const { ctx } = await setup() const warnings: string[] = [] @@ -921,62 +893,4 @@ describe('scoped-dispatch invariants', () => { .not.toThrow() }) - it('rejects an assembly context carrying agent without scope', async () => { - const ctx = await scopedCtx() - const agent = { id: 'a1' } as unknown as Agent - const base = { name: 'systemPrompt' } - const assembly = { sections: [], tools: [], variables: {} } - const bad = { agent } - expect(() => { - // The carrier base stands in for the SystemPrompt service (the declared `this`); the invariant only reads the carrier marks. - void ctx.waterfall(scopeTarget(base, undefined) as never, 'system-prompt/assemble', assembly as never, bad as never, () => Promise.resolve(assembly as never)) - }).toThrow(/agent.*without.*scope|assembleContextFor/) - const good = { agent, scope: agent } - expect(() => { - void ctx.waterfall(scopeTarget(base, agent) as never, 'system-prompt/assemble', assembly as never, good as never, () => Promise.resolve(assembly as never)) - }).not.toThrow() - }) - - it('backstops alternate agents that open a turn before agent/session-start', async () => { - const ctx = await scopedCtx() - // A live agent whose session is in the store but whose session-start has - // not fired: appending turn/start must throw the teaching error. - const session = ctx.sessions.create(SessionId('drive-s')) - const agent = { id: 'driver', session } as unknown as Agent - // Provide a minimal agents lookup: the invariant reads ctx.get('agents'). - const registryStub = { list: () => [agent] } - ctx.root.provide('agents', registryStub as never) - expect(() => { - session.append('turn/start', { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } }) - }).toThrow(/turn opened before agent\/session-start/) - expect(session.events).toEqual([]) - // The internal boundary marks the session before even a prepended product - // listener runs, so the supported session-start injection pattern can open - // and close its one-shot context turn synchronously. - ctx.on('agent/session-start', () => { - session.append('turn/start', { turn: 1, trigger: { kind: 'injection', source: { kind: 'plugin', plugin: 'test' } } }) - session.append('turn/end', { turn: 1, reason: { kind: 'completed' } }) - }, { prepend: true }) - expect(() => { - ctx.emit(scopeTarget(agent, agent), 'agent/session-start', agent, 'startup') - }).not.toThrow() - expect(session.events.map(event => event.type)).toEqual(['turn/start', 'turn/end']) - expect(() => { - session.append('turn/start', { turn: 2, trigger: { kind: 'message', source: { kind: 'user' } } }) - }).not.toThrow() - }) - - it('marks sessions of agents that predate the plugin as started (HMR re-apply safety)', async () => { - const ctx = new Context() - await ctx.plugin(SessionStore) - const session = ctx.sessions.create(SessionId('pre-s')) - const agent = { id: 'pre', session } as unknown as Agent - ctx.root.provide('agents', { list: () => [agent] } as never) - // Invariants apply AFTER the agent exists: its ordering is unknowable, so - // a turn opening without an observed session-start must NOT false-positive. - await ctx.plugin(Invariants) - expect(() => { - session.append('turn/start', { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } }) - }).not.toThrow() - }) }) diff --git a/packages/support/invariants/tsconfig.json b/packages/support/invariants/tsconfig.json index 88944f779e..cd17f67d7c 100644 --- a/packages/support/invariants/tsconfig.json +++ b/packages/support/invariants/tsconfig.json @@ -25,12 +25,6 @@ }, { "path": "../../core/scope" - }, - { - "path": "../../core/system-prompt" - }, - { - "path": "../../core/tools" } ] } diff --git a/packages/support/subagent-mock/README.md b/packages/support/subagent-mock/README.md index 6114bdf2c9..b8d9fbba44 100644 --- a/packages/support/subagent-mock/README.md +++ b/packages/support/subagent-mock/README.md @@ -2,7 +2,7 @@ A scripted `SubagentProvider` for testing the [subagent seam](../../subagent/subagent/README.md) without a model or a real child agent — the subagent analog of [`dsh-llm-replay`](../llm-replay/README.md). -It lets a test drive `ctx.subagents` and the model-facing `dsh-tool-subagent` through the **real cordis Loader / export path**, exercising provider registration, start-time capability validation, the run lifecycle (`result` / `cancel` / `dispose`), and the structured-output branch — all deterministically and keylessly. +It lets a test drive `ctx.subagents` and the model-facing `dsh-tool-subagent` through the real Cordis loader/export path, exercising provider registration, async start, start-time capability validation, required-signal cancellation, `result`, `dispose`, and structured output deterministically and keylessly. ## Usage @@ -13,8 +13,8 @@ Load it as a plugin (functional shape: `name`/`inject`/`Config`/`apply`, no defa | `name` | `mock` | Registry name to register the provider under. | | `reply` | `mock subagent reply` | The scripted child's final answer text. | | `stopReason` | `completed` | The stop reason `result` settles with. | -| `capabilities` | all `true` | Which start-time capabilities (`outputSchema`/`depthLimit`/`toolFilter`) the provider advertises. | +| `capabilities` | all `true` | Which start-time capabilities (`outputSchema`, `depthLimit`, `toolFilter`, and `persona`) the provider advertises. | | `inheritsParentContext` | `false` | Conversation-history descriptor: `false` means fresh, while `true` exercises seeded/fork wording. It says nothing about tool, service, scope, or authority inheritance. | | `structured` | `{ reply }` | Structured value surfaced when a request carries an `outputSchema` and the capability is on. | -A `cancel()` issued before `result` settles flips the stop reason to `aborted`, so the cancellation path is observable. +Aborting the required request signal or disposing before `result` settles flips the stop reason to `aborted`, so both holder-facing cancellation paths are observable. diff --git a/packages/support/subagent-mock/src/index.ts b/packages/support/subagent-mock/src/index.ts index 554889db71..abc1a6c5e9 100644 --- a/packages/support/subagent-mock/src/index.ts +++ b/packages/support/subagent-mock/src/index.ts @@ -29,11 +29,10 @@ const STOP_REASONS = ['completed', 'aborted', 'error', 'max-tokens', 'refusal'] const DEFAULT_CAPS: SubagentCapabilities = { outputSchema: true, depthLimit: true, toolFilter: true, persona: true } /** - * A scripted provider: every {@link start} returns a run whose `result` - * resolves on a microtask with the configured reply (and a structured value - * when the request asked for one and the capability is on). `dispose` is a - * no-op; a `cancel()` before the result settles flips the stop reason to - * `aborted`, so the cancellation path is observable in a test. + * A scripted provider: every {@link start} returns a ready run whose `result` + * resolves on the next task with the configured reply (and a structured value + * when the request asked for one and the capability is on). The required + * signal and `dispose()` both flip an unsettled result to `aborted`. */ class MockSubagentProvider implements SubagentProvider { readonly capabilities: SubagentCapabilities @@ -47,12 +46,22 @@ class MockSubagentProvider implements SubagentProvider { this.inheritsParentContext = config.inheritsParentContext ?? false } - start(request: SubagentStartRequest): SubagentRun { + async start(request: SubagentStartRequest): Promise { + if (request.signal.aborted) throw new Error('mock subagent start aborted before publication') const reply = this.config.reply ?? 'mock subagent reply' const output: ContentBlock[] = [{ type: 'text', text: reply }] const wantsStructured = request.outputSchema !== undefined && this.capabilities.outputSchema const baseStop: SubagentStopReason = this.config.stopReason ?? 'completed' - let cancelled = false + const flags = { cancelled: false } + const onAbort = (): void => { flags.cancelled = true } + request.signal.addEventListener('abort', onAbort, { once: true }) + // Make publication genuinely asynchronous so a same-turn abort is still + // a provider-owned startup failure rather than a returned live run. + await Promise.resolve() + if (flags.cancelled) { + request.signal.removeEventListener('abort', onAbort) + throw new Error('mock subagent start aborted before publication') + } // A deterministic child id derived from the parent — no clock/random (both // banned in deterministic paths here, and unnecessary for a scripted run). @@ -60,21 +69,22 @@ class MockSubagentProvider implements SubagentProvider { const resultFor = (): SubagentResult => ({ output, - structured: wantsStructured ? (this.config.structured ?? { reply }) : undefined, - stopReason: cancelled ? 'aborted' : baseStop, + ...wantsStructured ? { structured: this.config.structured ?? { reply } } : {}, + stopReason: flags.cancelled ? 'aborted' : baseStop, }) + const result = new Promise((resolve) => { + setTimeout(() => { resolve(resultFor()) }, 0) + }).finally(() => { + request.signal.removeEventListener('abort', onAbort) + }) return { id, - // A scripted run has no asynchronous publication phase; it is ready as - // soon as the provider returns the handle. - started: Promise.resolve(), - result: Promise.resolve().then(resultFor), - cancel() { - cancelled = true - }, - async dispose() { - // Scripted run holds no resources — nothing to await. + result, + dispose(): Promise { + flags.cancelled = true + request.signal.removeEventListener('abort', onAbort) + return Promise.resolve() }, } } diff --git a/packages/support/subagent-mock/tests/subagent-mock.spec.ts b/packages/support/subagent-mock/tests/subagent-mock.spec.ts index 8a96cc4141..7fd93c62bc 100644 --- a/packages/support/subagent-mock/tests/subagent-mock.spec.ts +++ b/packages/support/subagent-mock/tests/subagent-mock.spec.ts @@ -11,7 +11,7 @@ function fakeParent(id = 'parent-1'): Agent { } function baseRequest(over: Partial = {}): SubagentStartRequest { - return { prompt: [{ type: 'text', text: 'task' }], parent: fakeParent(), ...over } + return { prompt: [{ type: 'text', text: 'task' }], parent: fakeParent(), signal: new AbortController().signal, ...over } } async function mount(config: Partial = {}): Promise { @@ -26,12 +26,13 @@ describe('dsh-subagent-mock', () => { const ctx = await mount({ reply: 'hello from mock' }) expect(ctx.subagents.list()).toEqual(['mock']) - const run = ctx.subagents.start('mock', baseRequest()) + const run = await ctx.subagents.start('mock', baseRequest()) await expect(run.result).resolves.toEqual({ output: [{ type: 'text', text: 'hello from mock' }], structured: undefined, stopReason: 'completed', }) + await run.dispose() }) it('registers under a configurable name', async () => { @@ -41,13 +42,13 @@ describe('dsh-subagent-mock', () => { it('surfaces a structured result when the request carries an outputSchema', async () => { const ctx = await mount({ reply: 'r', structured: { answer: 42 } }) - const run = ctx.subagents.start('mock', baseRequest({ outputSchema: { type: 'object', properties: { answer: { type: 'number' } } } })) + const run = await ctx.subagents.start('mock', baseRequest({ outputSchema: { type: 'object', properties: { answer: { type: 'number' } } } })) await expect(run.result).resolves.toMatchObject({ structured: { answer: 42 } }) }) it('defaults structured output to { reply } when outputSchema is requested but no structured value is configured', async () => { const ctx = await mount({ reply: 'fallback reply' }) - const run = ctx.subagents.start('mock', baseRequest({ outputSchema: { type: 'object', properties: { answer: { type: 'number' } } } })) + const run = await ctx.subagents.start('mock', baseRequest({ outputSchema: { type: 'object', properties: { answer: { type: 'number' } } } })) await expect(run.result).resolves.toMatchObject({ structured: { reply: 'fallback reply' } }) }) @@ -56,24 +57,44 @@ describe('dsh-subagent-mock', () => { // The service rejects an outputSchema request against a no-cap provider, so // the structured path is only reachable when the cap is on; with it off and // no schema requested, the result has no structured field. - const run = ctx.subagents.start('mock', baseRequest()) + const run = await ctx.subagents.start('mock', baseRequest()) const result = await run.result expect(result).not.toHaveProperty('structured') }) it('honors a configured stop reason', async () => { const ctx = await mount({ stopReason: 'refusal' }) - const run = ctx.subagents.start('mock', baseRequest()) + const run = await ctx.subagents.start('mock', baseRequest()) await expect(run.result).resolves.toMatchObject({ stopReason: 'refusal' }) }) - it('flips the stop reason to aborted when cancelled before the result settles', async () => { + it('flips the stop reason to aborted when the signal fires before the result settles', async () => { const ctx = await mount() - const run = ctx.subagents.start('mock', baseRequest()) - run.cancel() + const controller = new AbortController() + const run = await ctx.subagents.start('mock', baseRequest({ signal: controller.signal })) + controller.abort() await expect(run.result).resolves.toMatchObject({ stopReason: 'aborted' }) }) + it('rejects an already-aborted request before starting publication', async () => { + const ctx = await mount() + const controller = new AbortController() + controller.abort() + + await expect(ctx.subagents.start('mock', baseRequest({ signal: controller.signal }))) + .rejects.toThrow('mock subagent start aborted before publication') + }) + + it('rejects when cancellation wins the asynchronous publication handoff', async () => { + const ctx = await mount() + const controller = new AbortController() + const pending = ctx.subagents.start('mock', baseRequest({ signal: controller.signal })) + + controller.abort() + + await expect(pending).rejects.toThrow('mock subagent start aborted before publication') + }) + it('unregisters the provider when the owning fiber is disposed (HMR safety)', async () => { const ctx = new Context() await ctx.plugin(SubagentService) diff --git a/packages/ui/acp/README.md b/packages/ui/acp/README.md index f5734e1f10..efd5e5247a 100644 --- a/packages/ui/acp/README.md +++ b/packages/ui/acp/README.md @@ -36,7 +36,7 @@ The `initialize` handshake reports a fixed server identity (`agentInfo: { name: ## Multi-session -The bridge multiplexes N sessions over one connection. Live sessions are held in a `Map` (forward) with a `WeakMap` reverse map so `agent/*` events — which carry only the `Agent` — demux in O(1). Every `session/event` and `agent/status` is routed strictly to its owning record, so concurrent sessions never cross-settle or interleave their `session/update` notifications. State is per session: one in-flight prompt each, `session/cancel` aborts and settles only its own agent/prompt, and disposal drains every live session in parallel to quiescence. Permission prompts follow the same ownership: the `approval/request` answerer resolves the owning session through the reverse map and prompts only there. +The bridge multiplexes N sessions over one connection. Live sessions are held in a `Map` (forward) with a `WeakMap` reverse map so agent-scoped approval events demultiplex in O(1). Every `session/event` is routed strictly to its owning record, so concurrent sessions never cross-settle or interleave their `session/update` notifications. State is per session: one in-flight prompt each, `session/cancel` aborts and settles only its own agent/prompt, and disposal drains every live session in parallel to quiescence. Permission prompts follow the same ownership: the `approval/request` answerer resolves the owning session through the reverse map and prompts only there. ## Session config options @@ -71,7 +71,7 @@ When the client does NOT advertise the capability, none of the `_meta`/terminal ## Settle-exactly-once -A `session/prompt` resolves (or rejects) exactly once, keyed off the canonical session log (the `session/event` stream). One listener captures the prompt's owning turn from the log's `turn/start` and settles on the matching `turn/end` — the durable boundary event (`closeTurn` appends it unconditionally; there is no `agent/*` turn mirror). A prompt settles only on ITS OWN turn (`inflight.turn === turn/end.turn`), so a stale `turn/end` for a previously-cancelled turn whose end arrives late can never settle the wrong prompt. A turn that ends `error` REJECTS the RPC with an internal error carrying the failure message (ACP has no error stop reason); every other reason resolves via the codec. Session contains post-commit observer failures per listener, so another subscriber cannot starve the bridge. As defensive cross-seam reconciliation, an `agent/status` handler checks the log whenever the agent reaches `idle`/`disposed` with a prompt still pending, settling from the owning turn's `turn/end` or as `cancelled` if teardown left no clean boundary. An empty/whitespace prompt is rejected up front — it would queue no work, so no turn would start and the RPC would hang. +A `session/prompt` resolves or rejects exactly once from the canonical `session/event` stream. The listener captures the prompt's owning turn from `turn/start` and settles in a `finally` block when the matching `turn/end` is appended, so a presentation/streaming failure cannot strand the RPC after the durable terminal event exists. Correlation by turn id prevents a late end from a cancelled prompt from settling its successor. A turn ending in `error` rejects the RPC with an internal error carrying the failure message because ACP has no error stop reason; every other reason resolves through the codec. An empty or whitespace-only prompt is rejected before enqueue because it would start no turn and otherwise leave the RPC pending. ## Permission prompts diff --git a/packages/ui/acp/src/index.ts b/packages/ui/acp/src/index.ts index 5cd24c2c73..5a5ce9e51f 100644 --- a/packages/ui/acp/src/index.ts +++ b/packages/ui/acp/src/index.ts @@ -71,7 +71,7 @@ import { } from '@agentclientprotocol/sdk' import type { ContentBlock } from '@deepseek-ai/dsh-llm' import { assertNever, CallId } from '@deepseek-ai/dsh-llm' -import type { Agent, AgentStatus } from '@deepseek-ai/dsh-agent' +import type { Agent } from '@deepseek-ai/dsh-agent' import { AgentId } from '@deepseek-ai/dsh-agent' import { SessionId } from '@deepseek-ai/dsh-session' import { SANDBOX_MODES, effectiveSandboxMode, setSandboxMode } from '@deepseek-ai/dsh-bash' @@ -297,7 +297,8 @@ interface SessionRecord { /** * The in-flight `session/prompt`, or `undefined` when none is pending. A * prompt resolves with a {@link StopReason} or rejects with an Error (a - * turn that ended in failure). Settled exactly once via {@link settlePrompt}. + * turn that ended in failure). Settled exactly once by its matching + * `turn/end`, direct cancellation, or teardown. * * `turn` is the loop turn number this prompt owns, captured from the log's * `turn/start` after `send()`. Until then it is `undefined` (the turn has not @@ -307,17 +308,11 @@ interface SessionRecord { * wrong prompt. A direct cancel/dispose settle clears the whole in-flight slot, * so a later stale `turn/end` finds no pending prompt. * - * `logWatermark` is the session log length at the moment the prompt was - * installed (before `send()`). Defensive settle-from-log reconciliation uses - * it to infer the owning `turn/start` if status reaches idle/disposed before - * live correlation settled the prompt: the prompt owns the FIRST message - * `turn/start` appended at or after this watermark. */ inflight: { resolve: (reason: StopReason) => void reject: (error: Error) => void turn: number | undefined - logWatermark: number } | undefined /** * Config switches accepted while the session was IDLE, not yet anchored in @@ -336,12 +331,9 @@ interface SessionRecord { /** * Drive the in-flight prompt's settle from the harness event stream. The bridge - * settles off the durable log: the `turn/end` session event on the - * `session/event` feed for the prompt's own turn, with idle/disposed status as - * defensive log reconciliation (docs/defensive-patterns.md "honor cross-seam - * contracts on BOTH sides"). Session contains post-commit observer failures, - * so peers cannot starve this feed. The first settlement path clears the slot, - * making every later signal a no-op. + * settles off the durable `turn/end` event for the prompt's own turn. Session + * contains post-commit observers independently, and this listener performs + * correlation in a `finally` so presentation failure cannot starve settlement. */ export function apply(ctx: Context, config: AcpConfig): void { // Capture the injected services NOW, during apply(), while we are inside this @@ -499,77 +491,24 @@ export function apply(ctx: Context, config: AcpConfig): void { ctx.on('session/event', (session, event: SessionEvent) => { const rec = sessions.get(session.header.id) if (rec === undefined) return - streamSessionEventUpdate(rec.sessionId, event, notify, rec.presenter, { - enabled: rec.terminalEnabled, - cwd: session.header.cwd, - }, { includeUserMessages: false }) - const inflight = rec.inflight - if (inflight === undefined) return - if (event.type === 'turn/start') { - // Tag the in-flight prompt with its owning turn — but ONLY a - // `message`-triggered turn (the kind a `send()` prompt produces). A turn - // a plugin opens between prompt-install and the prompt's own turn (an idle - // `agent.inject()` writes a one-shot `injection`-triggered turn) must NOT - // be mistaken for the prompt's turn, or its turn/end would settle the RPC - // early. The first message turn at/after install owns the prompt - // (`turn === undefined` guard); the loop batches queued messages into one - // turn, so there is exactly one. - if (inflight.turn === undefined && event.data.trigger.kind === 'message') { - inflight.turn = event.data.turn + try { + streamSessionEventUpdate(rec.sessionId, event, notify, rec.presenter, { + enabled: rec.terminalEnabled, + cwd: session.header.cwd, + }, { includeUserMessages: false }) + } finally { + const inflight = rec.inflight + if (inflight !== undefined && event.type === 'turn/start') { + // The first message-triggered turn after prompt installation owns the + // prompt; injection-triggered turns must not settle it early. + if (inflight.turn === undefined && event.data.trigger.kind === 'message') { + inflight.turn = event.data.turn + } + } else if (inflight !== undefined && event.type === 'turn/end' && inflight.turn === event.data.turn) { + rec.inflight = undefined + settleFromTurnEnd(inflight, event.data.reason) } - return } - // Settle only on the OWNING turn's end. - if (event.type !== 'turn/end' || inflight.turn !== event.data.turn) return - rec.inflight = undefined - settleFromTurnEnd(inflight, event.data.reason) - }) - - // Defensive settle fallback: when the agent reaches idle/disposed while a - // prompt is still pending, reconcile against the canonical log. Determine - // the owning turn from live capture or the first message turn after the - // install-time watermark, then settle from its turn/end; if no clean owning - // turn exists, settle cancelled. The slot is cleared first, so this cannot - // double-settle against the live session/event path. - const settleFromLog = (rec: SessionRecord): void => { - const inflight = rec.inflight - if (inflight === undefined) return - const events = rec.agent.session.events - // The owning turn number: the captured one, or inferred from the log as the - // first MESSAGE-triggered turn opened at/after the watermark. The filter matches the live - // capture: a one-shot `injection` turn a plugin may open between - // prompt-install and the prompt's turn is NOT the prompt's turn. Undefined - // only if no message turn ever started for this prompt. - const owningTurn = inflight.turn ?? events.slice(inflight.logWatermark).find( - (e): e is Extract => - e.type === 'turn/start' && e.data.trigger.kind === 'message', - )?.data.turn - // The owning turn's end in the log. If `owningTurn` is undefined (no turn - // ever started for this prompt — a torn-down-before-turn case that quiesce's - // direct settle normally pre-empts), no `turn/end` matches (turn numbers are - // >= 1) and `findLast` returns undefined, falling through to cancelled. - const end = events.findLast( - (e): e is Extract => - e.type === 'turn/end' && e.data.turn === owningTurn, - ) - rec.inflight = undefined - if (end === undefined) { - // No owning turn / no clean turn/end (torn down mid-turn) → cancelled. - inflight.resolve('cancelled') - return - } - settleFromTurnEnd(inflight, end.data.reason) - } - - // On idle/disposed, reconcile any still-pending prompt from the log. A - // mid-step disposal that never appended a clean turn/end resolves `cancelled`. - // Demux via the agent→sessionId reverse map. - ctx.on('agent/status', (agent, status: AgentStatus) => { - const sessionId = bySession.get(agent) - if (sessionId === undefined) return - const rec = sessions.get(sessionId) - if (rec === undefined) return - if (status === 'idle' || status === 'disposed') settleFromLog(rec) }) // --- Approval answerer ----------------------------------------------------- @@ -894,12 +833,10 @@ export function apply(ctx: Context, config: AcpConfig): void { // Install the in-flight slot BEFORE send() (send does not synchronously // flip status to running; the session/event listener records the turn // number and settle/rejects it). Capture the log length now as the - // watermark: defensive status reconciliation can infer the owning - // turn/start if status arrives reentrantly after commit but before this - // bridge's live callback. A turn that ends in error rejects this promise (the codec - // never produces an error stop reason). + // A turn that ends in error rejects this promise (the codec never + // produces an error stop reason). const stopReason = await new Promise((resolve, reject) => { - rec.inflight = { resolve, reject, turn: undefined, logWatermark: rec.agent.session.events.length } + rec.inflight = { resolve, reject, turn: undefined } rec.agent.send([{ type: 'text', text }]) }) return { stopReason } @@ -918,8 +855,7 @@ export function apply(ctx: Context, config: AcpConfig): void { // as cancelled directly here: do NOT rely on the resulting turn/end to // settle it, because cancel() may drop the turn before any turn/end is // emitted, and removing this direct settle would move the RPC's - // resolution onto the settleFromLog/agent-status path, changing its - // timing. + // resolution onto a later observer path, changing its timing. rec.agent.cancel('session/cancel') settlePrompt(rec, 'cancelled') return Promise.resolve() diff --git a/packages/ui/acp/tests/edges.spec.ts b/packages/ui/acp/tests/edges.spec.ts index 143bd7193c..e86fb9fc94 100644 --- a/packages/ui/acp/tests/edges.spec.ts +++ b/packages/ui/acp/tests/edges.spec.ts @@ -20,7 +20,7 @@ describe('acp bridge — demux & config edges', () => { it('ignores events from an agent the bridge does not own (strict id demux)', async () => { // A second agent created directly on the registry (NOT via the bridge) runs - // a turn. Its session/event + agent/status must NOT produce ACP updates and + // a turn. Its session events must NOT produce ACP updates and // must not settle anything — the bridge demuxes strictly by its own id. harness = await makeBridgeHarness({ storageDir, script: [textResponse('foreign')] }) await harness.client.initialize({ protocolVersion: PROTOCOL_VERSION, clientCapabilities: {} }) diff --git a/packages/ui/acp/tests/turns.spec.ts b/packages/ui/acp/tests/turns.spec.ts index 3fc6eef8a6..f559ca36d9 100644 --- a/packages/ui/acp/tests/turns.spec.ts +++ b/packages/ui/acp/tests/turns.spec.ts @@ -3,7 +3,7 @@ import { mkdtemp, rm } from 'node:fs/promises' import { tmpdir } from 'node:os' import { join } from 'node:path' import { defineTool } from '@deepseek-ai/dsh-tools' -import { AgentId, agentEvents } from '@deepseek-ai/dsh-agent' +import { AgentId } from '@deepseek-ai/dsh-agent' import { PROTOCOL_VERSION } from '@agentclientprotocol/sdk' import { errorResponse, @@ -271,37 +271,6 @@ describe('acp bridge — turn outcomes', () => { expect(result.stopReason).toBe('end_turn') }) - it('status reconciliation infers the owning message turn when teardown wins after turn/start', async () => { - harness = await makeBridgeHarness({ storageDir, script: [textResponse('background completion')] }) - const sessionId = await newSession(harness) - const agent = harness.ctx.agents.get(AgentId(sessionId))! - harness.ctx.on('session/event', (session, event) => { - if (session !== agent.session || event.type !== 'turn/start') return - // Inject the signal ordering the defensive fallback handles: disposal - // status after turn/start commits but before ACP's later live observer. - // This is event-level simulation; it does not mutate the test agent. - agentEvents(harness!.ctx, agent).emit('agent/status', 'disposed') - }, { prepend: true }) - - const result = await harness.client.prompt({ sessionId, prompt: [{ type: 'text', text: 'go' }] }) - expect(result.stopReason).toBe('cancelled') - }) - - it('status reconciliation can settle from a committed turn/end before live delivery', async () => { - harness = await makeBridgeHarness({ storageDir, script: [textResponse('answer')] }) - const sessionId = await newSession(harness) - const agent = harness.ctx.agents.get(AgentId(sessionId))! - harness.ctx.on('session/event', (session, event) => { - if (session !== agent.session || event.type !== 'turn/end') return - // Inject a reentrant status signal after the boundary commits to exercise - // the defensive log path before ACP's captured callback runs. - agentEvents(harness!.ctx, agent).emit('agent/status', 'idle') - }, { prepend: true }) - - const result = await harness.client.prompt({ sessionId, prompt: [{ type: 'text', text: 'go' }] }) - expect(result.stopReason).toBe('end_turn') - }) - it('a between-turn injection does not settle the prompt early (message-trigger correlation)', async () => { // A plugin injects context (a one-shot injection-triggered turn) right after // the prompt is queued but before the prompt's own message turn runs. The diff --git a/packages/ui/user-approval/README.md b/packages/ui/user-approval/README.md index b47f112aae..a50bffad0b 100644 --- a/packages/ui/user-approval/README.md +++ b/packages/ui/user-approval/README.md @@ -2,7 +2,7 @@ User-approval seam. Owns the `ctx.approval` service ([`ApprovalService`](src/index.ts)) and the one-shot permission vocabulary the harness shares: `ApprovalRequest` (agent + tool identity + reason + abort signal), the closed `ApprovalOutcome` union (`allowed-once` / `rejected` / `cancelled` / `unavailable`), the `ApprovalRequestId` brand pairing the two log-only audit events (`approval/asked` / `approval/decided`), and the `approval/request` waterfall the answerers listen on. It lives in the UI group because its purpose is human permission, while remaining channel-neutral: it depends only on Cordis and core vocabulary packages, never on a concrete UI. -The contract in one line: `ctx.approval.request(req)` puts exactly one question — "may this specific action proceed?" — to whatever answerers the deployment composed, and its answerer phase always produces an outcome: an aborted signal yields `cancelled`, a throwing or missing answerer yields `unavailable`, and `allowed-once` is a grant for the single asked-about action, never a class of future ones. Acceptance is synchronous: the service reads the request fields and `agent.session` binding once, requires object agent/session identities, a string `toolName`, optional string `callId`/`reason`, and an AbortSignal-shaped live capability, then shallow-freezes a detached request record while preserving the exact `agent` and signal identities. A malformed request rejects before any audit append; later caller mutation cannot redirect scope, payload, cancellation, policy lookup, or either audit event. The other precondition is an open turn on the captured session — the audit pair is turn-enclosed by contract (the turn is the durable log's commit/replay boundary; a bare event between turns is crash-tail garbage on reload), so an idle ask also rejects before appending. Either audit append may reject before commit because returning an unlogged decision would violate the pair. Session contains post-commit observer failures, so an authoritative audit append cannot reject the request or suppress its matching event. +The contract in one line: `ctx.approval.request(req)` puts exactly one question — "may this specific action proceed?" — to whatever answerers the deployment composed, and its answerer phase always produces an outcome: an aborted signal yields `cancelled`, a throwing or missing answerer yields `unavailable`, and `allowed-once` is a grant for the single asked-about action, never a class of future ones. `ApprovalRequest` is a readonly same-process contract: the service borrows the exact request, agent, session, and abort signal rather than cloning or freezing them. The request requires an open turn because the audit pair is turn-enclosed by contract (the turn is the durable log's commit/replay boundary; a bare event between turns is crash-tail garbage on reload), so an idle ask rejects before appending. Either audit append may reject before commit because returning an unlogged decision would violate the pair. Session contains post-commit observer failures, so an authoritative audit append cannot reject the request or suppress its matching event. The service is the mechanism, answerers are the policy. Answerers are `approval/request` waterfall listeners occupying a single decision slot: answer for an agent you own by returning an outcome without calling `next()`, or delegate an agent you don't recognize by calling `next()` — the chain's built-in default is `unavailable`, so a deployment with no answerer (headless, CI) fails closed with zero configuration. Dispatch is keyed by `req.agent`: a listener registered through `agent.ctx` receives only that agent's questions, while a plain-context listener receives every agent's. Registration order across sibling plugins is not load-order deterministic; compose one terminal answerer per deployment and use `prepend` listeners only for decide-or-delegate gates. diff --git a/packages/ui/user-approval/src/index.ts b/packages/ui/user-approval/src/index.ts index 9b0347bb74..6b3ef962bb 100644 --- a/packages/ui/user-approval/src/index.ts +++ b/packages/ui/user-approval/src/index.ts @@ -63,10 +63,8 @@ declare module 'cordis' { * Scope-filtered dispatch (`@deepseek-ai/dsh-scope`) keys the carrier by `req.agent`: a * listener registered through `agent.ctx` receives only that agent's * questions, while a plain-context listener receives every agent's. - * `req` is the service's shallow-frozen acceptance snapshot: later caller - * mutation cannot redirect the question, while the `agent` and `signal` - * identity capabilities remain exact. - * @param req - the accepted decision (agent, tool identity, reason, signal). + * `req` is a readonly same-process value borrowed from the caller. + * @param req - the pending decision (agent, tool identity, reason, signal). * @mode waterfall */ 'approval/request'(this: Scoped, req: ApprovalRequest, next: () => Promise): Promise @@ -241,11 +239,9 @@ export function setApprovalPolicy(session: Session, policy: ApprovalPolicy): voi * for an answerer to present it and for the audit events to reconstruct what * was asked — it deliberately does NOT carry tool arguments: a UI answerer * attaches the prompt to the already-streamed tool call via `callId` instead - * of re-rendering the call. `request()` synchronously copies and shallow-freezes - * this record before crossing an asynchronous boundary. It reads each field - * and the agent's session binding once, validates the public fixed-field - * contract before audit, and detaches the scalar values; the `agent` and live - * `signal` identity capabilities are preserved rather than cloned or frozen. + * of re-rendering the call. This is a readonly same-process contract: + * `request()` borrows the request and its `agent` and `signal` capabilities + * directly rather than treating them as serialized input. */ export interface ApprovalRequest { /** @@ -253,28 +249,21 @@ export interface ApprovalRequest { * UI answerer only answers for agents it owns) and receives the audit * events on its session log. */ - agent: Agent + readonly agent: Agent /** The tool the question is about (presentation and audit). */ - toolName: string + readonly toolName: string /** * The exact tool call being decided, when the asker has one — lets a UI * attach the prompt to the tool call it already streamed. */ - callId?: CallId + readonly callId?: CallId /** The asker's human-readable explanation of WHY it is asking. */ - reason?: string + readonly reason?: string /** * Aborting withdraws the question: the request settles `'cancelled'` * immediately and a late answer from a still-pending answerer is discarded. */ - signal?: AbortSignal -} - -/** Live signal capability accepted at the synchronous request boundary. */ -interface AcceptedSignal { - signal: AbortSignal - addEventListener: AbortSignal['addEventListener'] - removeEventListener: AbortSignal['removeEventListener'] + readonly signal?: AbortSignal } /** Plugin config. All optional — `static Config` supplies the defaults. */ @@ -285,7 +274,7 @@ export interface Config { * (fail-closed with none); `'never'` auto-rejects every ask without * prompting (the deterministic CI/unattended stance). */ - policy?: ApprovalPolicy + readonly policy?: ApprovalPolicy } /** @@ -374,104 +363,26 @@ export class ApprovalService extends Service { } /** - * Ask the composed answerers to decide one request. Synchronously reads each - * request field and the agent's session binding once, validates the fixed - * agent/session, string, and live-signal contracts, and rejects before any - * audit append when malformed. The signal remains the caller's exact live - * identity capability; it is neither cloned nor frozen. Requires an open - * turn on the accepted session — the audit pair below is turn-enclosed by - * contract (the turn is the log's commit/replay boundary; an idle append - * would be dropped as crash tail on reload) — and likewise throws before - * appending anything when called idle; asking outside a turn is a deferred - * design. The answerer phase always produces an outcome: an aborted signal - * yields `'cancelled'`, a missing or throwing answerer yields `'unavailable'` - * (fail closed), and a rogue non-vocabulary return value is normalized to - * `'unavailable'`. A failure that prevents either audit append from committing - * still rejects; returning an unlogged decision would violate the audit pair. - * The caller-owned request is synchronously - * snapshotted, so later mutation cannot split routing, dispatch payload, - * cancellation, policy lookup, or the audit pair across agents/sessions. - * Appends the - * `approval/asked`/`approval/decided` audit pair (log-only) around the - * decision regardless of outcome. Session contains each post-commit observer - * failure, so an already authoritative audit event cannot make this request - * reject or suppress its matching event. + * Ask the composed answerers to decide one readonly same-process request. + * The service borrows the request, agent, session, and live signal directly. + * The request requires an open turn because the audit pair must be enclosed + * by the durable log's commit/replay boundary; an idle ask rejects before + * appending anything. The answerer phase always produces an outcome: an + * aborted signal yields `'cancelled'`, a missing or throwing answerer yields + * `'unavailable'` (fail closed), and a rogue non-vocabulary return value is + * normalized to `'unavailable'`. A failure that prevents either audit append + * from committing still rejects because returning an unlogged decision would + * violate the pair. Session contains post-commit observer failures, so an + * authoritative append cannot reject the request or suppress its matching + * audit event. * @param req - the pending decision (agent, tool identity, reason, signal). * @returns the closed outcome; `'allowed-once'` is the only grant. - * @throws when request acceptance fails, no turn is open, or either audit - * event fails before the session append commit point. + * @throws when no turn is open or either audit event fails before the session + * append commit point. */ async request(req: ApprovalRequest): Promise { - // Accept one immutable request shape before the first async boundary. The - // caller retains its record and may mutate it as soon as this async method - // returns; identity capabilities stay live, but the record is never reread. - const input: unknown = req - if (typeof input !== 'object' || input === null) { - throw new TypeError('approval.request() requires a request object') - } - const source = input as Record - const agentInput = source['agent'] - const toolName = source['toolName'] - const callId = source['callId'] - const reason = source['reason'] - const signalInput = source['signal'] - if (typeof agentInput !== 'object' || agentInput === null) { - throw new TypeError('approval request agent must be an object') - } - if (typeof toolName !== 'string') { - throw new TypeError('approval request toolName must be a string') - } - if (callId !== undefined && typeof callId !== 'string') { - throw new TypeError('approval request callId must be a string when provided') - } - if (reason !== undefined && typeof reason !== 'string') { - throw new TypeError('approval request reason must be a string when provided') - } - let acceptedSignal: AcceptedSignal | undefined - if (signalInput !== undefined) { - if (typeof signalInput !== 'object' || signalInput === null) { - throw new TypeError('approval request signal must be an AbortSignal when provided') - } - const signalRecord = signalInput as unknown as Record - const aborted = signalRecord['aborted'] - const addEventListener = signalRecord['addEventListener'] - const removeEventListener = signalRecord['removeEventListener'] - if (typeof aborted !== 'boolean' - || typeof addEventListener !== 'function' - || typeof removeEventListener !== 'function') { - throw new TypeError('approval request signal must be an AbortSignal when provided') - } - acceptedSignal = { - signal: signalInput as AbortSignal, - addEventListener: addEventListener as AbortSignal['addEventListener'], - removeEventListener: removeEventListener as AbortSignal['removeEventListener'], - } - } - const sessionInput = (agentInput as unknown as Record)['session'] - if (typeof sessionInput !== 'object' || sessionInput === null) { - throw new TypeError('approval request agent session must be an object') - } - const sessionRecord = sessionInput as unknown as Record - const events = sessionRecord['events'] - const append = sessionRecord['append'] - if (!Array.isArray(events)) { - throw new TypeError('approval request session events must be an array') - } - if (typeof append !== 'function') { - throw new TypeError('approval request session append must be a function') - } - const agent = agentInput as Agent - const session = sessionInput as Session - const acceptedCallId = callId as CallId | undefined - const signal = signalInput as AbortSignal | undefined - const accepted: Readonly = Object.freeze({ - agent, - toolName, - ...acceptedCallId !== undefined ? { callId: acceptedCallId } : {}, - ...reason !== undefined ? { reason } : {}, - ...signal !== undefined ? { signal } : {}, - }) - if (!hasOpenTurn(events)) { + const session = req.agent.session + if (!hasOpenTurn(session.events)) { throw new Error( 'approval.request() outside an open turn: the approval/asked + approval/decided audit pair ' + 'must be turn-enclosed (a bare event between turns is crash-tail garbage on reload). ' @@ -479,14 +390,14 @@ export class ApprovalService extends Service { ) } const id = ApprovalRequestId(randomUUID()) - Reflect.apply(append, session, ['approval/asked', { + session.append('approval/asked', { id, - toolName: accepted.toolName, - ...accepted.callId !== undefined ? { callId: accepted.callId } : {}, - ...accepted.reason !== undefined ? { reason: accepted.reason } : {}, - }]) - const outcome = await this.decide(accepted, session, acceptedSignal) - Reflect.apply(append, session, ['approval/decided', { id, outcome }]) + toolName: req.toolName, + ...req.callId !== undefined ? { callId: req.callId } : {}, + ...req.reason !== undefined ? { reason: req.reason } : {}, + }) + const outcome = await this.decide(req, session) + session.append('approval/decided', { id, outcome }) return outcome } @@ -502,16 +413,14 @@ export class ApprovalService extends Service { } /** - * Dispatch the waterfall, contained and raced against the accepted signal. - * @param req - the detached public request snapshot. - * @param session - the captured session used for policy lookup. - * @param acceptedSignal - the validated live signal capability, if supplied. + * Dispatch the waterfall, contained and raced against the request signal. + * @param req - the borrowed public request. + * @param session - the request agent's session used for policy lookup. * @returns the normalized closed outcome. */ - private async decide( - req: Readonly, session: Session, acceptedSignal: AcceptedSignal | undefined, - ): Promise { - if (acceptedSignal?.signal.aborted) return 'cancelled' + private async decide(req: ApprovalRequest, session: Session): Promise { + const signal = req.signal + if (signal?.aborted) return 'cancelled' // The 'never' policy is decided HERE, before any dispatch: a listener // registered with `prepend: true` after this service mounts would sit // ahead of any gate LISTENER, so a listener-shaped gate cannot keep the @@ -535,13 +444,15 @@ export class ApprovalService extends Service { // tool call open — the seam contains its callbacks. () => 'unavailable', ) - if (acceptedSignal === undefined) return answer - const { signal, addEventListener, removeEventListener } = acceptedSignal + if (signal === undefined) return answer return await new Promise((resolve) => { - const onAbort = () => { resolve('cancelled') } - addEventListener.call(signal, 'abort', onAbort, { once: true }) + const onAbort = () => { + signal.removeEventListener('abort', onAbort) + resolve('cancelled') + } + signal.addEventListener('abort', onAbort, { once: true }) void answer.then((outcome) => { - removeEventListener.call(signal, 'abort', onAbort) + signal.removeEventListener('abort', onAbort) // After an abort won the race this resolve is a settled-promise no-op: // the late answer is discarded by construction. resolve(outcome) diff --git a/packages/ui/user-approval/tests/approval.spec.ts b/packages/ui/user-approval/tests/approval.spec.ts index 4d4b7aa67f..ff0cd5c7d8 100644 --- a/packages/ui/user-approval/tests/approval.spec.ts +++ b/packages/ui/user-approval/tests/approval.spec.ts @@ -2,7 +2,8 @@ import { describe, expect, it, vi } from 'vitest' import { Context } from 'cordis' import type { Agent } from '@deepseek-ai/dsh-agent' import { CallId } from '@deepseek-ai/dsh-llm' -import { carrierKeyOf, scopeHost } from '@deepseek-ai/dsh-scope' +import { carrierKeyOf, createScope } from '@deepseek-ai/dsh-scope' +import type { Scope } from '@deepseek-ai/dsh-scope' import SessionStore, { Session, SessionId } from '@deepseek-ai/dsh-session' import type { SessionEvent } from '@deepseek-ai/dsh-session' import SystemPrompt from '@deepseek-ai/dsh-system-prompt' @@ -39,158 +40,6 @@ function requestOf(agent: Agent, overrides: Partial = {}): Appr } describe('ApprovalService.request', () => { - it('rejects malformed fixed fields and identities before appending or dispatching', async () => { - const ctx = await mounted() - const consulted = vi.fn() - ctx.on('approval/request', () => { - consulted() - return Promise.resolve('allowed-once') - }) - const { agent, appended } = fakeAgent() - const badSessionAppends: Array> = [] - const badSession = (events: unknown, append: unknown): Agent => ({ - session: { events, append }, - }) as unknown as Agent - const appendSpy = (): ReturnType => { - const append = vi.fn() - badSessionAppends.push(append) - return append - } - const validSignalShape = { - aborted: false, - addEventListener: () => {}, - removeEventListener: () => {}, - } - const cases: Array<{ request: unknown; message: string }> = [ - { request: null, message: 'requires a request object' }, - { request: 1, message: 'requires a request object' }, - { request: { agent: null, toolName: 'echo' }, message: 'agent must be an object' }, - { request: { agent: 1, toolName: 'echo' }, message: 'agent must be an object' }, - { request: { agent, toolName: 1 }, message: 'toolName must be a string' }, - { request: { agent, toolName: 'echo', callId: 1 }, message: 'callId must be a string' }, - { request: { agent, toolName: 'echo', reason: 1 }, message: 'reason must be a string' }, - { request: { agent, toolName: 'echo', signal: null }, message: 'signal must be an AbortSignal' }, - { request: { agent, toolName: 'echo', signal: 1 }, message: 'signal must be an AbortSignal' }, - { - request: { agent, toolName: 'echo', signal: { ...validSignalShape, aborted: 'no' } }, - message: 'signal must be an AbortSignal', - }, - { - request: { agent, toolName: 'echo', signal: { ...validSignalShape, addEventListener: 1 } }, - message: 'signal must be an AbortSignal', - }, - { - request: { agent, toolName: 'echo', signal: { ...validSignalShape, removeEventListener: 1 } }, - message: 'signal must be an AbortSignal', - }, - { - request: { agent: { session: null }, toolName: 'echo' }, - message: 'agent session must be an object', - }, - { - request: { agent: { session: 1 }, toolName: 'echo' }, - message: 'agent session must be an object', - }, - { - request: { agent: badSession(null, appendSpy()), toolName: 'echo' }, - message: 'session events must be an array', - }, - { - request: { agent: badSession([{ type: 'turn/start' }], 1), toolName: 'echo' }, - message: 'session append must be a function', - }, - ] - - for (const { request, message } of cases) { - await expect(ctx.approval.request(request as ApprovalRequest)).rejects.toThrow(message) - } - - expect(appended).toEqual([]) - for (const append of badSessionAppends) expect(append).not.toHaveBeenCalled() - expect(consulted).not.toHaveBeenCalled() - }) - - it('reads request fields, the agent session, and the session append method once', async () => { - const ctx = await mounted() - const { agent: acceptedSessionOwner, appended: acceptedAudit } = fakeAgent() - const { agent: replacementAgent, appended: replacementAudit } = fakeAgent() - const acceptedSession = acceptedSessionOwner.session - const acceptedAppend = acceptedSession.append.bind(acceptedSession) - const signal = new AbortController().signal - const reads = { - agent: 0, - toolName: 0, - callId: 0, - reason: 0, - signal: 0, - session: 0, - append: 0, - } - const session = { - events: acceptedSession.events, - get append(): Session['append'] { - reads.append += 1 - return reads.append === 1 ? acceptedAppend : undefined as unknown as Session['append'] - }, - } as Session - const agent = Object.defineProperty({}, 'session', { - enumerable: true, - get: () => { - reads.session += 1 - return reads.session === 1 ? session : replacementAgent.session - }, - }) as Agent - const request = Object.defineProperties({}, { - agent: { - enumerable: true, - get: () => (++reads.agent === 1 ? agent : null), - }, - toolName: { - enumerable: true, - get: () => (++reads.toolName === 1 ? 'stable-tool' : 1), - }, - callId: { - enumerable: true, - get: () => (++reads.callId === 1 ? CallId('stable-call') : {}), - }, - reason: { - enumerable: true, - get: () => (++reads.reason === 1 ? 'stable reason' : {}), - }, - signal: { - enumerable: true, - get: () => (++reads.signal === 1 ? signal : {}), - }, - }) as ApprovalRequest - let received: ApprovalRequest | undefined - ctx.on('approval/request', (accepted) => { - received = accepted - return Promise.resolve('allowed-once') - }) - - await expect(ctx.approval.request(request)).resolves.toBe('allowed-once') - - expect(reads).toEqual({ - agent: 1, - toolName: 1, - callId: 1, - reason: 1, - signal: 1, - session: 1, - append: 1, - }) - expect(received).toMatchObject({ - agent, - toolName: 'stable-tool', - callId: 'stable-call', - reason: 'stable reason', - signal, - }) - expect(Object.isFrozen(received)).toBe(true) - expect(acceptedAudit.map(event => event.type)).toEqual(['approval/asked', 'approval/decided']) - expect(replacementAudit).toEqual([]) - }) - it('throws before appending anything when no turn has ever opened (idle ask)', async () => { const ctx = await mounted() const { agent, appended } = fakeAgent([]) @@ -230,77 +79,38 @@ describe('ApprovalService.request', () => { expect(Object.keys(appended[0]?.data ?? {}).sort()).toEqual(['id', 'toolName']) }) - it('snapshots request identity, scope, payload, and audit before deferred dispatch', async () => { + it('borrows the exact readonly request for scoped dispatch and audit', async () => { const ctx = await mounted() - const { agent: acceptedAgent, appended: acceptedAudit } = fakeAgent() - const { agent: replacementAgent, appended: replacementAudit } = fakeAgent() - const host = await scopeHost(ctx, ['approval']) - const acceptedScope = host.mint(acceptedAgent) - const replacementScope = host.mint(replacementAgent) - const dispatchStarted = Promise.withResolvers<'started'>() - const answer = Promise.withResolvers() - const originalSignal = new AbortController().signal - const replacementSignal = new AbortController().signal - let heardBy: 'accepted' | 'replacement' | undefined + const { agent, appended } = fakeAgent() + let scope!: Scope + const scopeFiber = await ctx.plugin(Object.assign((inner: Context) => { + scope = createScope(inner, agent) + }, { inject: ['approval'] })) let received: ApprovalRequest | undefined let carrier: unknown - acceptedScope.ctx.on('approval/request', function (req) { - heardBy = 'accepted' + scope.ctx.on('approval/request', function (req) { received = req carrier = carrierKeyOf(this) - dispatchStarted.resolve('started') - return answer.promise + return Promise.resolve('allowed-once') }) - replacementScope.ctx.on('approval/request', function (req) { - heardBy = 'replacement' - received = req - carrier = carrierKeyOf(this) - dispatchStarted.resolve('started') - return answer.promise - }) - const request = requestOf(acceptedAgent, { - toolName: 'original-tool', - callId: CallId('original-call'), - reason: 'original reason', - signal: originalSignal, + const request = requestOf(agent, { + toolName: 'scoped-tool', + callId: CallId('scoped-call'), + reason: 'scoped reason', }) - const pending = ctx.approval.request(request) - // request() has returned, but the answerer dispatch is deliberately queued - // in a microtask. Mutating the caller-owned record must not redirect it. - request.agent = replacementAgent - request.toolName = 'mutated-before-dispatch' - request.callId = CallId('mutated-call') - request.reason = 'mutated reason' - request.signal = replacementSignal - await dispatchStarted.promise - // Mutation while the answer is pending must not redirect the final audit. - request.toolName = 'mutated-after-dispatch' - request.reason = 'mutated again' - answer.resolve('allowed-once') - - await expect(pending).resolves.toBe('allowed-once') - expect(heardBy).toBe('accepted') - expect(carrier).toBe(acceptedAgent) - expect(received).not.toBe(request) - expect(Object.isFrozen(received)).toBe(true) - expect(received).toMatchObject({ - agent: acceptedAgent, - toolName: 'original-tool', - callId: 'original-call', - reason: 'original reason', - signal: originalSignal, + await expect(ctx.approval.request(request)).resolves.toBe('allowed-once') + expect(carrier).toBe(agent) + expect(received).toBe(request) + expect(appended).toHaveLength(2) + expect(appended[0]?.data).toMatchObject({ + toolName: 'scoped-tool', + callId: 'scoped-call', + reason: 'scoped reason', }) - expect(acceptedAudit).toHaveLength(2) - expect(acceptedAudit[0]?.data).toMatchObject({ - toolName: 'original-tool', - callId: 'original-call', - reason: 'original reason', - }) - expect(acceptedAudit[1]?.data).toMatchObject({ outcome: 'allowed-once' }) - expect(acceptedAudit[1]?.data['id']).toBe(acceptedAudit[0]?.data['id']) - expect(replacementAudit).toEqual([]) - await host.dispose() + expect(appended[1]?.data).toMatchObject({ outcome: 'allowed-once' }) + expect(appended[1]?.data['id']).toBe(appended[0]?.data['id']) + await scopeFiber.dispose() }) it('contains an approval/asked observer throw after append and still completes the pair', async () => { @@ -388,9 +198,12 @@ describe('ApprovalService.request', () => { const ctx = await mounted() const { agent: agentA } = fakeAgent() const { agent: agentB } = fakeAgent() - const host = await scopeHost(ctx, ['approval']) - const scopeA = host.mint(agentA) - const scopeB = host.mint(agentB) + let scopeA!: Scope + let scopeB!: Scope + const scopesFiber = await ctx.plugin(Object.assign((inner: Context) => { + scopeA = createScope(inner, agentA) + scopeB = createScope(inner, agentB) + }, { inject: ['approval'] })) const heard: string[] = [] ctx.on('approval/request', (req, next) => { heard.push(req.agent === agentA ? 'global:A' : 'global:B') @@ -409,14 +222,16 @@ describe('ApprovalService.request', () => { await expect(ctx.approval.request(requestOf(agentB))).resolves.toBe('unavailable') expect(heard).toEqual(['global:A', 'scoped:A', 'global:B', 'scoped:B']) - await host.dispose() + await scopesFiber.dispose() }) it('keys the scoped dispatch carrier to the exact request agent', async () => { const ctx = await mounted() const { agent } = fakeAgent() - const host = await scopeHost(ctx, ['approval']) - const scope = host.mint(agent) + let scope!: Scope + const scopeFiber = await ctx.plugin(Object.assign((inner: Context) => { + scope = createScope(inner, agent) + }, { inject: ['approval'] })) let seenKey: object | undefined scope.ctx.on('approval/request', function (req, next) { seenKey = carrierKeyOf(this) @@ -427,7 +242,7 @@ describe('ApprovalService.request', () => { await expect(ctx.approval.request(requestOf(agent))).resolves.toBe('unavailable') expect(seenKey).toBe(agent) - await host.dispose() + await scopeFiber.dispose() }) it('contains a throwing answerer as unavailable', async () => { diff --git a/packages/workflow/workflow-workerthread/README.md b/packages/workflow/workflow-workerthread/README.md index 409ab57132..b9786ae1e2 100644 --- a/packages/workflow/workflow-workerthread/README.md +++ b/packages/workflow/workflow-workerthread/README.md @@ -1,59 +1,80 @@ # @deepseek-ai/dsh-workflow-workerthread -The [`WorkflowService`](../workflow/README.md) implementation, on **`node:worker_threads`**: each run gets its OWN worker thread (one run = one worker, no pooling — a run is heavyweight, so the ~tens-of-ms thread spin-up is noise), the script executes in a vm context INSIDE that worker with the workflow hooks injected, and every `agent()` call bridges back over the message port to [`ctx.subagents`](../../subagent/README.md) on the host. Child agents are I/O-bound LLM loops and stay on the host event loop; the thread isolates the SCRIPT, the only part that can spin synchronously. +This package implements `WorkflowService` with one Node worker thread per run. The worker executes the orchestration script; child agents remain on the host and are reached through `ctx.subagents` over a typed host/worker protocol. -## Trust premise: what the thread buys (and what it does not) +The split has one primary purpose: a synchronous script loop cannot block the harness event loop, and a script that ignores cancellation can be terminated with its worker. It is not a security sandbox. -Workflow scripts are **model-written** — the same trust level as the model's existing bash access — so this engine defends against **buggy** scripts, never hostile ones. A worker thread is NOT a security boundary: the vm context inside it is escapable by construction (`node:vm` shares object machinery with its surrounding realm, so a script can reach the `Function` constructor via `globalThis.constructor.constructor` and from it `process` and every Node builtin), and an escapee holds the same process privileges as the host — Node's permission model is process-wide. The absent globals are API surface that keeps honest scripts portable, not walls. What the thread concretely buys: +## Trust and isolation boundary -- **The host never blocks**: `start()` returns without running any script code on the host; a synchronous spin anywhere in the script occupies the worker's loop, not the harness's. -- **Termination is real**: a script that outlives its post-cancel grace is `worker.terminate()`d — nothing of it survives `dispose()`, where an in-process engine could only abandon the spin on its own loop. -- **No ambient credentials**: the worker spawns with an EMPTY environment (`env: {}` plus hermetic `execArgv`, the same stance as `dsh-code-runtime-worker`; the unbuilt dev shape forwards exactly one loader variable, `TSX_TSCONFIG_PATH` — path plumbing, not a secret), so an escapee reading `process.env` finds no harness secrets — ambient-channel hardening only; the process-wide privileges above (fs and the rest) remain, so a genuine sandbox is still the engine swap. -- **Serialization by construction**: everything crossing the thread is structured-clone data, and plain JSON before that — the `materializeFromRealm` walk rejects loud what JSON cannot carry, which is also what makes every postMessage hop total. +Workflow scripts are model-written and have the same trust premise as the model's existing bash access. `node:vm` inside a worker is an API-shaping mechanism, not a security boundary: an escaped script can recover Node capabilities with the host process's privileges. -What the seam guarantees regardless, because benign scripts hit these constantly: `result` never rejects, a dropped hook promise never becomes an unhandled rejection, values JSON cannot carry are rejected **loud** instead of silently mangled, and hook misuse is fatal instead of dissolving into a per-item `null`. Genuine sandboxing (containing what an escaped script may touch) remains an isolated-vm/separate-process engine swap behind the seam, still deferred. +The worker still provides useful containment: -## The script contract it executes +- Script CPU work and synchronous spins stay off the host event loop. +- `worker.terminate()` gives disposal a real final stop. +- The worker starts with an empty environment, except unbuilt loader plumbing, so ambient credentials do not cross through `process.env`. +- Host/worker messages use structured-clone data, with plain-JSON validation at the script boundary. -- **Meta as DATA** (`validateMeta`, host-side): the workflow's identity arrives on the start request as plain JSON (the tool carries it as its schema-validated `meta` parameter — never as script text) and is shape-validated loud, every violation named (`name`/`description` required; unknown fields rejected). The engine deliberately evaluates NO script text to obtain meta: an evaluated meta literal could smuggle getters that run on the host outside any vm timeout — the exact spin the worker thread exists to isolate. A body that still opens with a Claude Code-style `export const meta` statement is rejected with a pointed `SCRIPT_PARSE` message. -- **Hooks**: `agent(prompt, {label, phase, schema, model})` (schema = the [structured-output subset](../../core/tools/README.md), forwarded as `outputSchema`; result = validated object, or final text without a schema; a failed child resolves `null`), `parallel(thunks)`, `pipeline(items, ...stages)` with NO cross-stage barrier and `(prev, item, index)` stage callbacks, `phase(title)`, `log(message)`, and the `args` global. Anything else — `effort`/`isolation`/`agentType`, unknown options, malformed arguments, schemas outside the subset — throws a FATAL `WorkflowError` that `parallel`/`pipeline` re-throw rather than nulling (see the seam README's failure discipline). -- **No ambient APIs**: no timers, filesystem, or Node APIs are injected into the context (absence is API surface, not containment — see the trust premise). +A genuinely untrusted-script sandbox would require a different engine behind the same workflow seam. -## How a run executes +## Script contract -`start()` shape-validates the meta DATA host-side and parse-checks the body with the identical wrapper the worker compiles (`new vm.Script`, discarded), preserving the seam's synchronous `META_INVALID`/`SCRIPT_PARSE` throws; one redundant parse per run is the deliberate price. It then spawns the worker (unbuilt: a JavaScript data-URL bootstrap registers tsx's ESM and CommonJS transforms inside the worker before importing `src/worker.ts`, giving the whole mixed-module source graph full TypeScript and tsconfig-path transformation on every supported Node line; built: the sibling `lib/worker.js` bundle) with the meta, body, `args`, and worker-side limits as `workerData`. +The workflow's `meta` is host-provided data, not evaluated script text. The engine validates its required `name` and `description`, rejects unknown fields, and parse-checks the body before returning a run. -Inside the worker, `runWorkerSession` builds the execution core (hooks, combinators, concurrency semaphore, caps, fatal-error discipline) over a **child port**. `agent()` sends `child-start`, and the host starts the child through the holder-bound `SubagentService` handle captured synchronously by `start()`, with parent attribution, the shared per-run abort signal, and `outputSchema`/`model` pass-through. This capture is part of the seam's holder-owned lifetime: unloading the engine removes `ctx.workflows` for new calls but does not invalidate an already returned run whose worker starts another child afterward. +Inside the worker, the script receives `args` and these hooks: -The host observes `run.result` immediately but buffers its snapshotted wire projection until `run.started` fulfills. Provider `start()` is arbitrary code and can synchronously reenter workflow cancellation before its returned run reaches the host registry, so the host registers the run, attaches both promise observers, and re-checks admission after `start()` returns and again at readiness. A closed boundary never admits or announces the run to the worker: while the exact run remains registered, the host invokes explicit cancel once and disposes it; `child-start-error` is sent only while worker-message admission remains open. If the run was already retired, the identity guard sends no cleanup through the deleted call ID. An ordinary readiness rejection sends `child-start-error` while possible and disposes the provider attempt without adding an explicit cancellation. Otherwise the host replies `child-started` with the child id before forwarding settlement, so `workflow/agent-start` always names a ready child and precedes its end. The worker classifies a start error as fatal `AGENT_START` unless cancellation already owns the run. If readiness fulfills, an infrastructure result rejection crosses as `child-failed`/`AGENT_RESULT` regardless of whether that rejection settled before or after readiness. Child disposal acknowledgements complete the RPC. +- `agent(prompt, { label, phase, schema, model })` starts one host-side subagent. With a schema it returns the structured value; otherwise it returns final text. An ordinary failed child yields `null`. +- `parallel(thunks)` runs thunks under the configured concurrency limit. +- `pipeline(items, ...stages)` passes `(previous, item, index)` without a cross-stage barrier. +- `phase(title)` and `log(message)` emit observer narration. -Observer narration (`phase`/`log`/`agent-start`/`agent-end`) crosses as messages and re-emits as the seam's `workflow/*` events. A **ready→go handshake** gates the body: a cancellation racing worker boot arrives before `go`, so a run cancelled before start never executes the body at all. +Unknown options, malformed arguments, unsupported schemas, tripped caps, provider-start failures, and infrastructure result failures are fatal workflow errors. No timers, filesystem API, or Node globals are intentionally injected, though the trust caveat above still applies. -## The value boundary +## Run sequence -Values LEAVING the script (hook options/schemas, the script's return) are materialized by `materializeFromRealm`: a plain recursive walk that rejects loud everything JSON cannot carry (exotic prototypes, functions, symbols, cycles, sparse arrays, non-finite numbers, nested `undefined`), copying into plain containers via `defineProperty` so a `"__proto__"` key becomes a data property, never a prototype mutation. Getters are read ordinarily — the RESULT is what crosses; a read that throws fails loud; both run in the WORKER, never on the host. Values ENTERING the realm (`args`, `agent()` results, hook promises and their failures, combinator arrays) are handed over directly as worker-realm values — the script is trusted, so outer prototypes are not a leak; `args` is cloned once at start so a script scribbling on it cannot mutate the caller's object. One script-visible consequence: an error thrown by a hook is built OUTSIDE the script's vm context, so `e instanceof Error` inside the script is `false` — branch on `e.name`/`e.code` instead (the combinators recognize fatality by `instanceof` against their own realm's class, which a script-built object can never pass, so fatal-vs-null cannot be forged or dissolved). +`start()` validates meta and parses the body, creates the worker, and returns a holder-owned `WorkflowRun`. A ready/go handshake prevents a start-signal cancellation racing worker boot from executing the script's initial synchronous slice. -## Cancellation, death, disposal +For each `agent()` call: -Cancellation is bounded and host-driven. Per-run limits are a concurrency semaphore (`maxConcurrentAgents`), a total-`agent()` cap (`maxTotalAgents`), and a per-call item cap (`maxItemsPerCall`), all config. `cancel()` first records its reason, then posts to the worker (its hooks start throwing `CANCELLED`; the script dies at its next await) and cancels every host-side child NOW on **both seam channels**: the shared request signal aborts and each registered child's explicit `cancel()` runs host-side. The seam leaves a provider free to honor either channel, and a worker wedged in a synchronous spin could not relay its own per-child cancel RPCs. A host-side per-call gate turns the worker's later explicit-cancel relay into a no-op, because the seam does not require `SubagentRun.cancel()` to be idempotent. Each explicit child `cancel()` callback is exception-contained independently, post-cancel `phase`/`log` narration is suppressed host-side, and cancelled children still deliver paired `agent-end` events. The caller's optional start-signal callback is retained by exact identity only while the run is live and removed at the first settlement or teardown. +1. The worker sends `child-start` with a plain-data prompt and options. +2. The host calls the configured provider through async `SubagentService.start`, passing the workflow's parent and one canonical per-run abort signal. +3. If start rejects, the host sends `child-start-error`; provider startup has already reached quiescence and no child lifecycle event is emitted. +4. If start fulfills while the workflow still admits work, the host records the run, observes `result`, then sends `child-started`. Even an already-settled result is forwarded afterward, preserving start-before-result order. +5. The worker emits paired `workflow/agent-start` and `workflow/agent-end` narration and requests child disposal after collection. -Terminal arbitration is first-wins at explicit host-side claim points. A cancellation before ready→go reports `cancelled` without executing the body. For a later race, the worker queues Result before its settlement-reap `ChildCancel` messages; external `cancel()` records its reason before its fanout, while Result receipt snapshots any earlier cancellation and records the terminal outcome before settlement-cleanup fanout. Same-port FIFO and those claim points mean earlier caller/signal/dispose cancellation overrides a non-cancelled report, while an arrived report cannot be rewritten by a cleanup callback. Once Result has won, a losing reentrant `cancel()` has no state, message, child-fanout, or grace-timer effect. If no earlier terminal source settles the run, the grace callback claims `cancelled`, synthesizes missing lifecycle ends, settles the result, and terminates the worker after `disposeGraceMs`. +Provider starts are tracked separately from published children. If cancellation, worker death, or normal workflow settlement closes admission while a start is pending, the shared signal aborts it. A provider that nevertheless fulfills after closure is disposed by the host and never announced to the worker. -Worker death separates outcome ownership, message admission, and resource cleanup. An unexpected OOM, `error`, message failure, or premature exit claims `stopReason: 'error'` with diagnostics—or preserves an external cancellation already in flight—before reaping children or synthesizing observer events. Reentrant provider cancellation therefore cannot turn a death-first error into cancellation. The first death signal also closes worker-message admission because Node may deliver a queued `message` between `error` and `exit`; late protocol data cannot start a child, emit narration, or compete with the outcome. If Result or grace already claimed the outcome, death preserves it while still reaping promptly. The eventual `exit` then performs a final disposal-only sweep, joining any in-flight disposal without repeating explicit child cancellation. This separation lets grace settlement become observable before `worker.terminate()` reports exit without leaking the host-side registry. +## Value boundary -Disposal is the holder's bounded resource guarantee: cancel, begin host-driven disposal of every registered child immediately, wait for result plus child-registry quiescence up to the same grace, and unconditionally terminate the worker. `handle.dispose()` claims its public promise before that traversal invokes cancellation or disposal callbacks. Independently, every `disposeChild` path claims the call ID's promise before invoking the wrapped child disposer. Public-first reentry therefore returns the existing holder promise; worker-first reentry may begin holder disposal, whose child traversal joins the already-claimed call ID promise. Neither order can start a second provider disposal. A wedged worker can relay no dispose RPC, so host-driven teardown overlaps the grace; any later worker RPC joins the same per-child disposal. Before ordinary settlement becomes observable, the host also cancels every stray on both channels, including a fire-and-forget run still waiting on readiness. That work is settlement-only cleanup after the terminal claim, so provider reentry cannot rewrite the chosen result; `dispose()` then waits for its completion within the bound. +Values leaving the script pass through `materializeFromRealm`, which accepts plain, lossless JSON data and rejects exotic prototypes, functions, symbols, cycles, sparse arrays, non-finite numbers, and nested `undefined`. The walk runs in the worker, and defines object keys as data properties so `__proto__` cannot mutate a prototype. -Lifecycle pairing is host-guaranteed independently of outcome arbitration. Forwarded starts live in a ledger and worker-reported ends pair them on graceful paths. When death or grace is the terminal source, the host synthesizes missing ends with outcome `cancelled` before `workflow/end`. If Result settled first, later death cleanup may synthesize a survivor's end afterward; a start already crossing force-settlement may likewise surface after `workflow/end`. The same ledger still pairs every forwarded start exactly once. +Child results are projected and snapshotted before crossing from the host to the worker. This is a real process-like serialization boundary; it is deliberately different from trusted same-process workflow and subagent event payloads, which are borrowed immutable values. -**Engine-specific limitations**: worker startup is paid per run; on a termination path `agentsStarted` reports the HOST-observed count (accepted `child-start`s — calls still queued worker-side for a concurrency slot are unknowable then); and a returned promise or thenable resolves per JavaScript semantics BEFORE materialization — that is what makes an un-awaited `return agent('x')` work — with the value-boundary guard applying to the resolution. +## Cancellation and disposal + +`WorkflowRun.cancel()` records the first reason, tells the worker to cancel, aborts the one signal shared by every pending and published child, and arms the `disposeGraceMs` timer. Worker hooks then throw `CANCELLED` at their next await. If the run remains unsettled at the deadline, the host resolves it as cancelled, pairs stranded child lifecycle events, and terminates the worker. + +The subagent seam has one cancellation channel: the request signal. There is no separate child-cancel RPC. Published child teardown uses `run.dispose()`; pending provider starts remain provider-owned until their promise rejects or fulfills. + +Normal settlement also aborts pending starts and begins disposing any published fire-and-forget children before the result becomes externally settled. The host's quiescence condition includes both pending starts and published child disposals, so cleanup does not forget an async startup transaction. + +`dispose()` is idempotent. It cancels the run, starts host-driven disposal immediately, waits for result plus child quiescence up to the same grace, terminates the worker unconditionally, and performs a final survivor sweep. Per-child disposal is memoized so worker RPC, host cancellation, death cleanup, and public disposal all join one operation. + +## Outcome and event guarantees + +Terminal outcome is first-wins at host claim points. An accepted external cancellation overrides a later non-cancelled worker result; a result or worker death that claims first cannot be rewritten by reentrant cleanup callbacks. + +Worker error, message failure, or premature exit closes message admission before cleanup, then resolves `error` unless cancellation already owns the run. Late queued messages cannot create children or narrate after that logical boundary. + +The host keeps a ledger of forwarded child starts. A graceful worker supplies their ends; death or force termination synthesizes any missing end as cancelled. Every forwarded `workflow/agent-start` is therefore paired exactly once, although cleanup after an already-arrived workflow result may complete afterward. ## Config | Key | Default | Meaning | |---|---|---| -| `provider` | `spawn` | The `ctx.subagents` provider children run on (host-side). | -| `maxConcurrentAgents` | `0` (auto) | Concurrent `agent()` ceiling; `0` resolves to `min(16, max(1, cores - 2))`. | -| `maxTotalAgents` | `1000` | Total `agent()` calls one run may start (runaway-loop backstop). | -| `maxItemsPerCall` | `4096` | Items accepted by one `parallel()`/`pipeline()` call. | -| `syncTimeoutMs` | `5000` | vm timeout for the script's initial synchronous slice (in the worker). | -| `disposeGraceMs` | `5000` | How long a cancelled run may stay unsettled before force-settle + terminate; also bounds `dispose()`. | +| `provider` | `spawn` | Host-side subagent provider used by `agent()`. | +| `maxConcurrentAgents` | `0` | Concurrent `agent()` ceiling; `0` resolves from available CPU parallelism. | +| `maxTotalAgents` | `1000` | Total `agent()` calls in one run. | +| `maxItemsPerCall` | `4096` | Items accepted by one `parallel()` or `pipeline()` call. | +| `syncTimeoutMs` | `5000` | VM timeout for the script's initial synchronous slice. | +| `disposeGraceMs` | `5000` | Bound before force-settlement/termination and for public disposal. | diff --git a/packages/workflow/workflow-workerthread/src/host.ts b/packages/workflow/workflow-workerthread/src/host.ts index ef5fdcbfac..95c07f0d2d 100644 --- a/packages/workflow/workflow-workerthread/src/host.ts +++ b/packages/workflow/workflow-workerthread/src/host.ts @@ -25,33 +25,18 @@ * eventual exit performs a final disposal-only sweep without repeating child * cancellation. * - * Children live in a host-side registry (callId → run) as soon as the provider - * accepts them, so cancellation reaches even a pre-publication attempt. Both - * explicit run cancellation and the shared request signal are driven when the - * workflow is cancelled OR normally settles, so a fire-and-forget child cannot - * survive merely by honoring only one channel. A per-call gate invokes each - * explicit provider `cancel()` at most once even though host fanout and the - * worker's later relay can both request it. The host observes `result` - * immediately but acknowledges the child to the worker only after `started` - * fulfills; readiness failure is a start error and the host disposes the - * attempt because the worker never received a handle. The - * worker drives disposal by RPC on the graceful path, `dispose()` host-drives - * every registered child's disposal immediately (a wedged worker can relay no - * dispose RPC, and child teardown must overlap the grace, not start after it), - * and the registry lets the host abort and dispose every survivor when the - * worker dies or is terminated mid-flight. The three - * paths share ONE disposal per child (memoized by callId; the seam's - * dispose() is idempotent anyway, the memo keeps the bookkeeping and the - * containment warn single). Lifecycle pairing is host-guaranteed the same - * way: every forwarded `agent-start` lives in a ledger, and a start the dead - * or terminated worker never paired is closed exactly once by a synthesized - * `agent-end` (outcome `'cancelled'`). When death or grace is the terminal - * source, already-known pairs close before the run settles; cleanup after an - * earlier Result can close a survivor afterward. On a termination path - * `agentsStarted` reports the - * HOST-observed count (accepted `child-start` messages) — `agent()` calls - * still queued worker-side for a concurrency slot are unknowable then; the - * worker's own count rides the result message on every graceful path. + * Provider starts and published children are tracked separately. Every start + * receives one shared per-run abort signal; the provider owns partial setup + * until its promise fulfills. If admission closes while a start is pending, + * the signal aborts it; a late fulfillment is disposed without publication to + * the worker. Ready runs enter a callId registry whose memoized disposal is + * shared by graceful worker RPC, public disposal, normal-settlement reap, and + * worker-death cleanup. Quiescence requires both pending starts and published + * children to drain. Lifecycle pairing is host-guaranteed independently: + * every forwarded `agent-start` enters a ledger, and a dead or terminated + * worker's missing `agent-end` is synthesized exactly once as cancelled. On a + * termination path `agentsStarted` reports the host-observed child-start count; + * calls still queued worker-side for a concurrency slot are unknowable. * * @module @deepseek-ai/dsh-workflow-workerthread/host */ @@ -71,6 +56,12 @@ import { HostToWorkerType, WorkerToHostType } from './protocol.ts' import type { HostToWorkerPayloads, WorkerToHostMessage } from './protocol.ts' import type { ChildResult, ChildStartRequest, WorkerInit } from './types.ts' +/** One published child and its shared quiescent-disposal transaction. */ +interface ChildRecord { + readonly run: SubagentRun + disposal?: Promise +} + /** * Resolve the worker entry and spawn options for the current runtime shape. * Unbuilt (tsx demos, vitest — `import.meta.url` points into `src/`), the @@ -135,9 +126,9 @@ function resolveWorkerSpawn(init: WorkerInit): { entry: URL; options: WorkerOpti /** * One live worker-engine run — the seam's {@link WorkflowRun}, returned by * `start()` directly. Owns the Worker, the child registry, and the result - * settlement; `result` never rejects. `meta` is this handle's OWN clone - * (event payloads carry separate clones), so a consumer mutating it corrupts - * nothing. The holder-bound SubagentService handle is captured before the + * settlement; `result` never rejects. `meta` is trusted same-process data + * borrowed as immutable by the handle and lifecycle events. The holder-bound + * SubagentService handle is captured before the * engine returns this run, so unloading the engine removes only the ability to * start another workflow; this run can still start and clean up its children. */ @@ -157,12 +148,10 @@ export class WorkerRun implements WorkflowRun { private workerGone = false /** Accepted `child-start` messages — the terminate-path `agentsStarted` (see module doc). */ private hostStarted = 0 - /** Live children by callId; an entry leaves ONLY after its dispose settles (quiescence = empty). */ - private readonly children = new Map() - /** In-flight child disposals by callId — the memo that gives every path (worker RPC, dispose(), reap) ONE shared disposal per child. */ - private readonly childDisposals = new Map>() - /** callIds whose explicit provider cancel callback has already been invoked. */ - private readonly childCancellations = new Set() + /** Published children by callId; an entry leaves only after disposal settles. */ + private readonly children = new Map() + /** Provider starts that have not yet fulfilled or rejected. */ + private readonly pendingStarts = new Set>() /** Started-but-not-ended agents by seq — the pairing ledger the HOST guarantees (see {@link endAgent}). */ private readonly liveAgents = new Map() private readonly quiescenceWaiters: (() => void)[] = [] @@ -214,11 +203,8 @@ export class WorkerRun implements WorkflowRun { /** * Cancel the run: the worker is told (its hooks start throwing and the - * script dies at its next await), every host-side child is cancelled NOW on - * BOTH seam channels — the shared request signal aborts and each registered - * child's explicit `cancel()` is called (the seam leaves a provider free to - * honor either, and a worker wedged in a synchronous spin could not relay - * its own per-child cancel RPCs until far too late) — and the grace timer + * script dies at its next await), the required signal shared by every child + * start is aborted, and the grace timer * arms: a run still unsettled `disposeGraceMs` later force-settles * `cancelled` and its worker is TERMINATED. Idempotent; the first reason * wins. @@ -234,11 +220,7 @@ export class WorkerRun implements WorkflowRun { if (this.settled || this.terminalClaimed || this.cancelReason !== undefined) return this.cancelReason = reason ?? 'workflow cancelled' this.post(HostToWorkerType.Cancel, { reason: this.cancelReason }) - // The explicit channel is driven host-side, not left to the worker: a - // provider honoring only run.cancel() must not wait on a wedged worker's - // ChildCancel relay (the per-call cancellation gate makes those later - // RPCs no-ops without imposing idempotence on the provider). - this.cancelChildren(this.cancelReason) + this.abortChildren(this.cancelReason) this.graceTimer = setTimeout(() => { // Cancellation already owns the race through cancelReason; close the // terminal boundary explicitly before observer teardown callbacks. @@ -351,12 +333,6 @@ export class WorkerRun implements WorkflowRun { case WorkerToHostType.ChildStart: this.onChildStart(message.callId, message.request) break - case WorkerToHostType.ChildCancel: - { - const run = this.children.get(message.callId) - if (run !== undefined) this.cancelChild(message.callId, run, message.reason) - } - break case WorkerToHostType.ChildDispose: this.onChildDispose(message.callId) break @@ -369,7 +345,7 @@ export class WorkerRun implements WorkflowRun { } } - /** Why a child may no longer cross the provider readiness boundary. */ + /** Why a ready provider result may no longer be admitted to the worker. */ private childAdmissionFailure(): { reason: string; rendered: string } | undefined { if (this.cancelReason !== undefined) { return { reason: this.cancelReason, rendered: `workflow run cancelled: ${this.cancelReason}` } @@ -393,9 +369,20 @@ export class WorkerRun implements WorkflowRun { return } this.hostStarted += 1 + const task = this.startChild(callId, request) + this.pendingStarts.add(task) + void task.then( + () => { this.finishPendingStart(task) }, + /* v8 ignore next -- startChild contains provider and cleanup failures */ + () => { this.finishPendingStart(task) }, + ) + } + + /** Await one provider-owned startup transaction and publish only while admitted. */ + private async startChild(callId: number, request: ChildStartRequest): Promise { let run: SubagentRun try { - run = this.subagents.start(this.provider, { + run = await this.subagents.start(this.provider, { prompt: [{ type: 'text', text: request.prompt }], parent: this.parent, signal: this.controller.signal, @@ -403,32 +390,36 @@ export class WorkerRun implements WorkflowRun { ...request.model !== undefined ? { agentOptions: { model: request.model } } : {}, }) } catch (error: unknown) { - this.post(HostToWorkerType.ChildStartError, { callId, rendered: renderThrown(error) }) + const failure = this.childAdmissionFailure() + this.post(HostToWorkerType.ChildStartError, { + callId, + rendered: failure?.rendered ?? renderThrown(error), + }) + return + } + const failure = this.childAdmissionFailure() + if (failure !== undefined) { + this.post(HostToWorkerType.ChildStartError, { callId, rendered: failure.rendered }) + try { + await run.dispose() + } catch (error: unknown) { + this.ctx.logger.warn(`workflow-workerthread: refused child dispose failed: ${renderThrown(error)}`) + } return } - this.children.set(callId, run) - const childId = run.id - // Observe settlement IMMEDIATELY, before readiness. A provider may reject - // result and started in the same turn; delaying this handler would make the - // result transiently unhandled. Buffer a forwarding closure so the worker - // still sees ChildStarted before ChildSettled/ChildFailed. Snapshot a - // resolved result now: a provider mutating its resolved object while - // publication is pending must not change what crosses the worker boundary. + const record: ChildRecord = { run } + this.children.set(callId, record) + // Attach result forwarding before publishing the child handle. Because the + // callback itself runs in a later microtask, ChildStarted is still posted + // first even for an already-settled scripted provider. const forwardResult = run.result.then<() => void, () => void>( (result) => { try { - // Capture every provider-owned field once, then materialize the - // worker-bound value in one lossless traversal. A stateful accessor - // cannot validate one result and send another, and an exotic value is - // rejected before any prototype-erasing clone. - const output = result.output - const structured = result.structured - const stopReason = result.stopReason const snapshot = snapshotJsonValue({ - output, - ...structured !== undefined ? { structured } : {}, - stopReason, + output: result.output, + ...result.structured !== undefined ? { structured: result.structured } : {}, + stopReason: result.stopReason, }) if (snapshot === undefined) throw new TypeError('child result is not losslessly JSON-serializable') return () => { this.post(HostToWorkerType.ChildSettled, { callId, result: snapshot }) } @@ -442,67 +433,20 @@ export class WorkerRun implements WorkflowRun { return () => { this.post(HostToWorkerType.ChildFailed, { callId, rendered }) } }, ) - - // The provider owns the publication boundary. Observe both promises before - // invoking cancellation/disposal below: provider.start() itself is - // arbitrary code and may have reentered handle.cancel() before the returned - // run reached our registry. Exactly one branch answers this ChildStart. - let startReplySent = false - const refusePublication = (failure: { reason: string; rendered: string }): void => { - startReplySent = true - this.post(HostToWorkerType.ChildStartError, { callId, rendered: failure.rendered }) - // A prior dispose/death can finish and remove this run while readiness - // is still pending. In that case teardown already owned cancellation and - // disposal; touching the retired callId would repeat cancel and orphan a - // fresh gate entry after finishChild deleted it. - if (this.children.get(callId) !== run) return - this.cancelChild(callId, run, failure.reason) - void this.disposeChild(callId, run) - } - - // Only acknowledge the child after it is real, then flush any result that - // settled unusually early. Re-check admission at that exact boundary: a - // cancellation while readiness was pending is a refusal, not a late - // publication into a terminal workflow. A readiness rejection is a START - // failure, not AGENT_RESULT; the worker never receives a handle, so the - // host disposes the registered attempt. Identity guards preserve the one - // disposal memo against concurrent host teardown. - void run.started.then( - () => { - if (startReplySent) return - const failure = this.childAdmissionFailure() - if (failure !== undefined) { - refusePublication(failure) - return - } - startReplySent = true - this.post(HostToWorkerType.ChildStarted, { callId, childId }) - void forwardResult.then((forward) => { forward() }) - }, - (error: unknown) => { - if (startReplySent) return - startReplySent = true - this.post(HostToWorkerType.ChildStartError, { callId, rendered: renderThrown(error) }) - if (this.children.get(callId) === run) void this.disposeChild(callId, run) - }, - ) - - // Close the synchronous hole around provider.start(): cancel()/dispose() - // can run before the returned run is visible to their children loop. - const reentrantFailure = this.childAdmissionFailure() - if (reentrantFailure !== undefined) refusePublication(reentrantFailure) + this.post(HostToWorkerType.ChildStarted, { callId, childId: run.id }) + void forwardResult.then((forward) => { forward() }) } private onChildDispose(callId: number): void { - const run = this.children.get(callId) - if (run === undefined) { + const record = this.children.get(callId) + if (record === undefined) { // Already disposed host-side (a dispose() drive or a death reap beat // the RPC) — the ack is still owed (the worker-side wrapper awaits it). this.post(HostToWorkerType.ChildDisposed, { callId }) return } // disposeChild never rejects (containment is inside), so the ack always follows. - void this.disposeChild(callId, run).then(() => { this.post(HostToWorkerType.ChildDisposed, { callId }) }) + void this.disposeChild(callId, record).then(() => { this.post(HostToWorkerType.ChildDisposed, { callId }) }) } /** @@ -514,79 +458,55 @@ export class WorkerRun implements WorkflowRun { * not supposed to reject, but a backend that does anyway must not break * quiescence): logged, and the child still leaves the registry. * @param callId - the child's registry key. - * @param run - the registered child (the caller looked it up). + * @param record - the registered child (the caller looked it up). * @returns resolves when the disposal settled either way; never rejects. */ - private disposeChild(callId: number, run: SubagentRun): Promise { - let disposal = this.childDisposals.get(callId) - if (disposal === undefined) { - // Claim before run.dispose() invokes provider code. Reentrant holder - // disposal then joins this exact child transaction instead of entering - // the provider wrapper twice before either memo is installed. - const claimed = Promise.withResolvers() - disposal = claimed.promise - this.childDisposals.set(callId, disposal) - // The seam promises a Promise, but invoke inside an async boundary so a - // contract-violating synchronous throw is contained exactly like a - // rejected disposal and cannot break host quiescence. - void (async () => { await run.dispose() })().then( - () => { - this.finishChild(callId) - claimed.resolve(undefined) - }, - (error: unknown) => { - this.ctx.logger.warn(`workflow-workerthread: child dispose failed: ${renderThrown(error)}`) - this.finishChild(callId) - claimed.resolve(undefined) - }, - ) - } - return disposal + private disposeChild(callId: number, record: ChildRecord): Promise { + if (record.disposal !== undefined) return record.disposal + record.disposal = Promise.resolve() + .then(() => record.run.dispose()) + .catch((error: unknown) => { + this.ctx.logger.warn(`workflow-workerthread: child dispose failed: ${renderThrown(error)}`) + }) + .then(() => { this.finishChild(callId) }) + return record.disposal } - /** Drop a child from the registry (and its disposal memo), releasing quiescence waiters at zero. */ + /** Drop a child record and release quiescence waiters when all work ends. */ private finishChild(callId: number): void { this.children.delete(callId) - this.childDisposals.delete(callId) - this.childCancellations.delete(callId) - if (this.children.size === 0) { - for (const waiter of this.quiescenceWaiters.splice(0)) waiter() - } + this.notifyChildQuiescence() } - /** Resolves once the child registry is empty (every disposal settled). */ + /** Retire one provider startup transaction. */ + private finishPendingStart(task: Promise): void { + this.pendingStarts.delete(task) + this.notifyChildQuiescence() + } + + /** Release waiters only after both pending starts and published children end. */ + private notifyChildQuiescence(): void { + if (this.children.size !== 0 || this.pendingStarts.size !== 0) return + for (const waiter of this.quiescenceWaiters.splice(0)) waiter() + } + + /** Resolves once every pending start and published child has reached quiescence. */ private childQuiescence(): Promise { - if (this.children.size === 0) return Promise.resolve() + if (this.children.size === 0 && this.pendingStarts.size === 0) return Promise.resolve() return new Promise((resolve) => { this.quiescenceWaiters.push(resolve) }) } /** Abort + dispose every registered child (worker death / final teardown); disposal is contained, not awaited. */ private reapChildren(reason: string): void { - const cancellation = this.cancelReason ?? reason - this.cancelChildren(cancellation) - for (const [callId, run] of [...this.children]) { - void this.disposeChild(callId, run) + this.abortChildren(this.cancelReason ?? reason) + for (const [callId, record] of [...this.children]) { + void this.disposeChild(callId, record) } } - /** Drive both cancellation channels for every child already accepted by the host. */ - private cancelChildren(reason: string): void { - this.controller.abort(reason) - for (const [callId, run] of this.children) this.cancelChild(callId, run, reason) - } - - /** Invoke one provider-owned cancel callback at most once and contain its exception. */ - private cancelChild(callId: number, run: SubagentRun, reason?: string): void { - // Host fanout and the worker's FIFO-later ChildCancel relay are two paths - // to the same provider callback. The seam does not require cancel() to be - // idempotent, so claim the callId before invoking arbitrary provider code. - if (this.childCancellations.has(callId)) return - this.childCancellations.add(callId) - try { - run.cancel(reason) - } catch (error: unknown) { - this.ctx.logger.warn(`workflow-workerthread: child cancel failed: ${renderThrown(error)}`) - } + /** Abort the one canonical signal shared by pending and published children. */ + private abortChildren(reason: string): void { + if (!this.controller.signal.aborted) this.controller.abort(reason) } private onResult(result: WorkflowResult): void { @@ -599,17 +519,14 @@ export class WorkerRun implements WorkflowRun { // callbacks, but that internal post-result cleanup must not retroactively // rewrite the worker result that arrived first. const cancellationWasRequested = this.cancelReason !== undefined - // Claim before either settlement-cleanup cancellation channel invokes - // provider code. A provider callback can reenter cancel() synchronously or - // from a queued microtask; once Result won, that losing cancellation must - // have no state, message, child-fanout, or grace-timer side effects. + // Claim before settlement cleanup invokes provider disposal. Once Result + // won, a later cancellation cannot rewrite it. this.terminalClaimed = true - // The worker cancels handles it already received, but a fire-and-forget - // child may still be waiting on readiness and therefore have no worker - // handle. Drive BOTH provider-permitted channels from the host before the - // workflow becomes externally settled. + // Abort pending starts and begin disposing published children before the + // workflow becomes externally settled. Cleanup remains independently + // tracked by childQuiescence and the holder's dispose(). + this.reapChildren('workflow settled') if (!cancellationWasRequested) { - this.cancelChildren('workflow settled') this.settleResult(result) return } @@ -640,7 +557,7 @@ export class WorkerRun implements WorkflowRun { // accepted before death remains cancelled. If Result/grace already won, // preserve it while still performing prompt failure-time cleanup. if (!outcomeWasClaimed) this.terminalClaimed = true - if (this.children.size > 0) this.reapChildren('workflow worker gone') + if (this.children.size > 0 || this.pendingStarts.size > 0) this.reapChildren('workflow worker gone') this.endStrandedAgents() if (!outcomeWasClaimed) { if (cancellationWasRequested) { @@ -655,7 +572,7 @@ export class WorkerRun implements WorkflowRun { // precede `exit`. Admission is already closed, so this final sweep only // joins/starts disposal for registry survivors; it deliberately does not // repeat explicit provider cancellation. - for (const [callId, run] of [...this.children]) void this.disposeChild(callId, run) + for (const [callId, record] of [...this.children]) void this.disposeChild(callId, record) this.endStrandedAgents() } diff --git a/packages/workflow/workflow-workerthread/src/index.ts b/packages/workflow/workflow-workerthread/src/index.ts index 8a21847cd5..a1f5b47ccc 100644 --- a/packages/workflow/workflow-workerthread/src/index.ts +++ b/packages/workflow/workflow-workerthread/src/index.ts @@ -151,9 +151,7 @@ export class WorkerWorkflowEngine extends WorkflowService { const meta = validateMeta(request.meta) assertBodyParses(request.script, meta.name) const id = WorkflowRunId(randomUUID()) - // The event payloads and the run handle get SEPARATE meta clones: a - // listener mutating its snapshot must not corrupt the holder's view. - const info: WorkflowRunInfo = { id, meta: structuredClone(meta) } + const info: WorkflowRunInfo = { id, meta } const limits: WorkerLimits = { maxConcurrentAgents: this.config.maxConcurrentAgents === 0 ? Math.min(16, Math.max(1, availableParallelism() - 2)) @@ -180,7 +178,7 @@ export class WorkerWorkflowEngine extends WorkflowService { runCtx, subagents, id, - structuredClone(meta), + meta, request.parent, init, this.config.provider, diff --git a/packages/workflow/workflow-workerthread/src/protocol.ts b/packages/workflow/workflow-workerthread/src/protocol.ts index d70ad11613..70edc9a713 100644 --- a/packages/workflow/workflow-workerthread/src/protocol.ts +++ b/packages/workflow/workflow-workerthread/src/protocol.ts @@ -33,8 +33,6 @@ export enum WorkerToHostType { AgentEnd = 'agent-end', /** Child RPC: start a child on the host (answered by ChildStarted or ChildStartError). */ ChildStart = 'child-start', - /** Child RPC: cancel a started child (fire-and-forget). */ - ChildCancel = 'child-cancel', /** Child RPC: dispose a started child (answered by ChildDisposed). */ ChildDispose = 'child-dispose', /** The run's single terminal result. */ @@ -55,8 +53,6 @@ export interface WorkerToHostPayloads { [WorkerToHostType.AgentEnd]: { info: WorkflowAgentEndInfo } /** The RPC correlation id and the prompt plus validated options. */ [WorkerToHostType.ChildStart]: { callId: number; request: ChildStartRequest } - /** The RPC correlation id and the cancel reason (undefined = unspecified). */ - [WorkerToHostType.ChildCancel]: { callId: number; reason: string | undefined } /** The RPC correlation id of the child to dispose. */ [WorkerToHostType.ChildDispose]: { callId: number } /** The run's terminal outcome. */ @@ -69,9 +65,9 @@ export enum HostToWorkerType { Go = 'go', /** Cancel the run: hooks start throwing and the script dies at its next await. */ Cancel = 'cancel', - /** Child RPC reply: provider publication/readiness fulfilled (exactly one start reply per ChildStart). */ + /** Child RPC reply: the provider fulfilled with a ready run (exactly one start reply per ChildStart). */ ChildStarted = 'child-started', - /** Child RPC reply: synchronous start or asynchronous publication/readiness failed. */ + /** Child RPC reply: the provider's asynchronous start failed. */ ChildStartError = 'child-start-error', /** Child RPC: a started child's result RESOLVED (its JSON projection). */ ChildSettled = 'child-settled', diff --git a/packages/workflow/workflow-workerthread/src/runtime.ts b/packages/workflow/workflow-workerthread/src/runtime.ts index 9b324c1118..94282cb173 100644 --- a/packages/workflow/workflow-workerthread/src/runtime.ts +++ b/packages/workflow/workflow-workerthread/src/runtime.ts @@ -20,7 +20,7 @@ * * Failure discipline: fatal {@link WorkflowError}s (bad hook arguments, * unsupported options/schemas, tripped caps, synchronous start refusal, - * pre-publication readiness failure, ready-child result rejection, and + * provider-start failure, ready-child result rejection, and * cancellation) ALWAYS propagate through * `parallel`/`pipeline` — recognized by `instanceof` against this realm's * class, which a script inside the vm context cannot forge — and the per-item @@ -83,9 +83,8 @@ function defaultLabel(prompt: string): string { /** * One live script execution inside the worker. Constructed per run by the * session; `drive()` is called exactly once and NEVER rejects — every failure - * becomes a {@link WorkflowResult} with a non-`completed` stop reason. After - * the session publishes that result it calls {@link reapAfterResult} exactly - * once to cancel any dropped child work without racing terminal publication. + * becomes a {@link WorkflowResult} with a non-`completed` stop reason. The + * host owns cancellation and cleanup of any dropped child work. */ export class WorkflowExecution { /** 1-based count of `agent()` calls started (the `agentsStarted` result field). */ @@ -94,7 +93,6 @@ export class WorkflowExecution { private readonly slotWaiters: { resolve(): void; reject(error: unknown): void }[] = [] private cancelReason: string | undefined private cancelError: WorkflowError | undefined - private readonly controller = new AbortController() private currentPhase: string | undefined private readonly context: vm.Context private readonly compiled: vm.Script @@ -130,11 +128,8 @@ export class WorkflowExecution { pipeline: (items: unknown, ...stages: unknown[]) => this.contain(this.pipeline(items, stages)), phase: (title: unknown) => { this.phase(title) }, log: (message: unknown) => { this.log(message) }, - // Cloned once: a script scribbling on args must not mutate the - // session's init object (a benign-bug guard; args is plain JSON by the - // seam contract and already crossed one structured clone as workerData, - // so this clone is total). - args: args === undefined ? undefined : structuredClone(args), + // workerData already performed the real cross-thread structured clone. + args, } for (const [key, value] of Object.entries(globals)) { // Data properties on the contextified global; frozen shape not required — @@ -165,22 +160,18 @@ export class WorkflowExecution { } /** - * Cancel the run: in-flight children get a cancel RPC (the shared abort - * fanout), waiting `agent()` slots reject, and every future hook call + * Cancel the run: waiting `agent()` slots reject and every future hook call * throws `CANCELLED` — the script dies at its next await. A script that * never settles anyway (parked on a promise no hook owns) is the HOST's * problem: its grace timer force-settles the run and terminates the * worker. Idempotent; the first reason wins. - * @param reason - human-readable cause, carried on the CANCELLED error and - * into child cancel RPCs. Required: every caller (the session's cancel - * message and its post-result {@link reapAfterResult} call) has a concrete - * reason. + * @param reason - human-readable cause carried on the CANCELLED error. The + * host independently aborts the required signal shared by every child. */ cancel(reason: string): void { if (this.cancelReason !== undefined) return this.cancelReason = reason this.cancelError = new WorkflowError(`workflow run cancelled: ${this.cancelReason}`, 'CANCELLED') - this.controller.abort(this.cancelReason) for (const waiter of this.slotWaiters.splice(0)) waiter.reject(this.cancelledError()) } @@ -188,9 +179,8 @@ export class WorkflowExecution { * Run the script to settlement. Resolves — never rejects — with the run's * {@link WorkflowResult}: the materialized return value on `completed`, the * failure message on `error`, and `cancelled` when the script died of - * cancellation. This method only chooses the result; the session must publish - * it and then call {@link reapAfterResult}, so the terminal message precedes - * settlement-only child cancellation on the worker-to-host FIFO channel. + * cancellation. This method only chooses the result; the session publishes + * it and the host owns terminal child cancellation. * @returns the settled outcome — this promise NEVER rejects (the seam's * `result`-never-rejects contract); every failure maps to a variant. */ @@ -221,16 +211,6 @@ export class WorkflowExecution { } } - /** - * Reap strays only after the caller publishes the chosen terminal result. - * Aborting the controller synchronously sends child-cancel RPCs, so calling - * this before publication would let a provider callback reenter host - * cancellation and misclassify a result the script had already chosen. - */ - reapAfterResult(): void { - if (this.cancelReason === undefined) this.cancel('workflow settled') - } - /** * Attach a no-op rejection consumer WITHOUT changing what the caller * receives: if the script drops the promise (no await), cancellation cannot @@ -336,17 +316,11 @@ export class WorkflowExecution { // wind the fresh child down instead of leaving it live behind a dead // script. if (this.isCancelled()) { - run.cancel(this.cancelReason) await run.dispose() throw this.cancelledError() } const info: WorkflowAgentInfo = { seq, label, ...phase !== undefined ? { phase } : {}, childId: AgentId(run.id) } this.observer.agentStart(info) - // Cancellation reaches the child through an explicit cancel RPC per - // child (the host also aborts its own per-run signal, but the seam - // leaves a provider free to honor either channel, so both are driven). - const onAbort = (): void => { run.cancel(this.cancelReason) } - this.controller.signal.addEventListener('abort', onAbort, { once: true }) try { let result try { @@ -387,7 +361,6 @@ export class WorkflowExecution { this.observer.agentEnd({ ...info, outcome: 'failed' }) return null } finally { - this.controller.signal.removeEventListener('abort', onAbort) await run.dispose() } } finally { diff --git a/packages/workflow/workflow-workerthread/src/session.ts b/packages/workflow/workflow-workerthread/src/session.ts index f6b14f0fca..671a9429ae 100644 --- a/packages/workflow/workflow-workerthread/src/session.ts +++ b/packages/workflow/workflow-workerthread/src/session.ts @@ -14,11 +14,6 @@ * A `cancel` arriving instead of `go` still releases the gate: `drive()` * sees the cancelled state and settles without running the body. * - * Terminal ordering is Result first, settlement cleanup second. The session - * queues the Result message before asking the execution to reap stray children; - * MessagePort FIFO therefore lets the host atomically claim the result before a - * cleanup ChildCancel can invoke arbitrary provider code. - * * @module @deepseek-ai/dsh-workflow-workerthread/session */ @@ -64,10 +59,6 @@ class RpcChildHandle implements ChildHandle { this.result = entry.settled.promise } - cancel(reason?: string): void { - this.post(WorkerToHostType.ChildCancel, { callId: this.callId, reason }) - } - dispose(): Promise { this.post(WorkerToHostType.ChildDispose, { callId: this.callId }) return this.entry.disposed.promise @@ -76,7 +67,7 @@ class RpcChildHandle implements ChildHandle { /** * The worker-side child-RPC bridge ({@link ChildPort}): allocates callIds, - * posts the start/cancel/dispose RPCs, and owns the per-call pending + * posts the start/dispose RPCs, and owns the per-call pending * book-keeping the session's message handler settles via the `onChild*` * entry points. */ @@ -94,10 +85,10 @@ class ChildRpcBridge implements ChildPort { settled: Promise.withResolvers(), disposed: Promise.withResolvers(), } - // Containment: when synchronous start or asynchronous readiness fails (or + // Containment: when asynchronous provider start fails (or // the run is torn down), the settled promise may never gain a consumer — // it must not surface as an unhandled rejection and kill the worker. - entry.settled.promise.catch(() => { /* consumed: unconsumed child settlement after failed start/readiness */ }) + entry.settled.promise.catch(() => { /* consumed: unconsumed child settlement after failed start */ }) this.pending.set(callId, entry) this.post(WorkerToHostType.ChildStart, { callId, request }) const childId = await entry.started.promise @@ -109,7 +100,7 @@ class ChildRpcBridge implements ChildPort { this.pending.get(callId)?.started.resolve(childId) } - /** Synchronous start or asynchronous readiness failed; reject and retire the pending RPC. */ + /** Asynchronous provider start failed; reject and retire the pending RPC. */ onChildStartError(callId: number, rendered: string): void { const entry = this.pending.get(callId) this.pending.delete(callId) @@ -213,12 +204,5 @@ export async function runWorkerSession(port: MessagePort, init: WorkerInit): Pro post(WorkerToHostType.Ready, {}) await gate.promise const result = await execution.drive() - try { - // This post is the worker's terminal claim. Queue it BEFORE aborting stray - // children: MessagePort FIFO then guarantees the host claims Result before - // any settlement-only ChildCancel can invoke arbitrary provider callbacks. - post(WorkerToHostType.Result, { result }) - } finally { - execution.reapAfterResult() - } + post(WorkerToHostType.Result, { result }) } diff --git a/packages/workflow/workflow-workerthread/src/types.ts b/packages/workflow/workflow-workerthread/src/types.ts index ee5faccdda..2f29dd8137 100644 --- a/packages/workflow/workflow-workerthread/src/types.ts +++ b/packages/workflow/workflow-workerthread/src/types.ts @@ -77,8 +77,6 @@ export interface ChildHandle { * failed for its own reasons resolves with a non-`completed` stop reason. */ readonly result: Promise - /** Ask the host to cancel the child (fire-and-forget). */ - cancel(reason?: string): void /** Ask the host to dispose the child; resolves on the host's ack. */ dispose(): Promise } @@ -92,7 +90,7 @@ export interface ChildPort { * Start one child agent on the host (the `agent()` hook's start half). * @param request - the prompt and validated options. * @returns the ready child handle; rejects when synchronous start or the - * provider's asynchronous publication/readiness boundary fails. + * provider's asynchronous start fails. */ startAgent(request: ChildStartRequest): Promise } diff --git a/packages/workflow/workflow-workerthread/tests/integration.spec.ts b/packages/workflow/workflow-workerthread/tests/integration.spec.ts index 51ace9253b..0e1727f877 100644 --- a/packages/workflow/workflow-workerthread/tests/integration.spec.ts +++ b/packages/workflow/workflow-workerthread/tests/integration.spec.ts @@ -49,7 +49,7 @@ describe('dsh-workflow-workerthread over the real in-process stack', () => { ]) const childIds: string[] = [] ctx.on('workflow/agent-start', (_info, agent) => { - // The workflow bridge must honor SubagentRun.started: a start observer + // The workflow bridge must await asynchronous provider start: an observer // sees the real spawn child already published, never a reserved id. expect(ctx.agents.get(agent.childId)).toBeDefined() childIds.push(agent.childId) diff --git a/packages/workflow/workflow-workerthread/tests/session.spec.ts b/packages/workflow/workflow-workerthread/tests/session.spec.ts index e2dab8948f..5fc85ce647 100644 --- a/packages/workflow/workflow-workerthread/tests/session.spec.ts +++ b/packages/workflow/workflow-workerthread/tests/session.spec.ts @@ -212,7 +212,7 @@ describe('runWorkerSession over an in-process MessageChannel', () => { host.close() }) - it('cancel mid-run: in-flight children get cancel RPCs, hooks throw at entry, the run reports cancelled', async () => { + it('cancel mid-run: hooks throw at entry and the run reports cancelled', async () => { const host = fakeHost() void runWorkerSession(host.port, init(` phase('before') @@ -232,7 +232,6 @@ describe('runWorkerSession over an in-process MessageChannel', () => { const result = await host.result() expect(result.stopReason).toBe('cancelled') expect(result.error).toContain('stop everything') - expect(host.ofType(WorkerToHostType.ChildCancel).map(m => m.callId)).toContain(callId) expect(host.ofType(WorkerToHostType.AgentEnd)[0]!.info.outcome).toBe('cancelled') // No post-cancel narration left the runtime (the hooks threw at entry). expect(host.ofType(WorkerToHostType.Phase).map(m => m.title)).toEqual(['before']) @@ -281,36 +280,6 @@ describe('runWorkerSession over an in-process MessageChannel', () => { } }) - it('queues Result before settlement-only cancellation of a ready stray', async () => { - const host = fakeHost({ manual: true }) - const session = runWorkerSession(host.port, init(` - agent('ready stray') - return await agent('gate') - `)) - await vi.waitFor(() => { expect(host.ofType(WorkerToHostType.ChildStart)).toHaveLength(2) }) - const starts = host.ofType(WorkerToHostType.ChildStart) - const stray = starts.find(message => message.request.prompt === 'ready stray')! - const gate = starts.find(message => message.request.prompt === 'gate')! - host.send({ type: HostToWorkerType.ChildStarted, callId: stray.callId, childId: 'stray-child' }) - host.send({ type: HostToWorkerType.ChildStarted, callId: gate.callId, childId: 'gate-child' }) - host.send({ type: HostToWorkerType.ChildSettled, callId: gate.callId, result: text('gate completed') }) - - const result = await host.result() - await session - await vi.waitFor(() => { - expect(host.ofType(WorkerToHostType.ChildCancel).map(message => message.callId)).toContain(stray.callId) - }) - - expect(result).toMatchObject({ value: 'gate completed', stopReason: 'completed', agentsStarted: 2 }) - const resultIndex = host.messages.findIndex(message => message.type === WorkerToHostType.Result) - const strayCancelIndex = host.messages.findIndex(message => - message.type === WorkerToHostType.ChildCancel && message.callId === stray.callId) - expect(resultIndex).toBeGreaterThanOrEqual(0) - expect(strayCancelIndex).toBeGreaterThan(resultIndex) - host.send({ type: HostToWorkerType.ChildSettled, callId: stray.callId, result: { output: [], stopReason: 'aborted' } }) - host.close() - }) - it('an unparseable body settles an error result instead of dying without one (host pre-parse skew guard)', async () => { const host = fakeHost() await runWorkerSession(host.port, init('return (((')) @@ -463,7 +432,7 @@ describe('runWorkerSession over an in-process MessageChannel', () => { host.close() }) - it('a cancel landing DURING the start round-trip winds the fresh child down (cancel + dispose) and dies cancelled', async () => { + it('a cancel landing DURING the start round-trip disposes the fresh child and dies cancelled', async () => { const host = fakeHost({ manual: true }) void runWorkerSession(host.port, init("return await agent('p')")) await vi.waitFor(() => { expect(host.ofType(WorkerToHostType.ChildStart).length).toBe(1) }) @@ -477,7 +446,6 @@ describe('runWorkerSession over an in-process MessageChannel', () => { const result = await host.result() expect(result.stopReason).toBe('cancelled') await vi.waitFor(() => { - expect(host.ofType(WorkerToHostType.ChildCancel).map(m => m.callId)).toContain(callId) expect(host.ofType(WorkerToHostType.ChildDispose).map(m => m.callId)).toContain(callId) }) // The child never became an agent-start: it was wound down pre-lifecycle. diff --git a/packages/workflow/workflow-workerthread/tests/workflow-workerthread.spec.ts b/packages/workflow/workflow-workerthread/tests/workflow-workerthread.spec.ts index ec2f458d45..4ad00ee02f 100644 --- a/packages/workflow/workflow-workerthread/tests/workflow-workerthread.spec.ts +++ b/packages/workflow/workflow-workerthread/tests/workflow-workerthread.spec.ts @@ -48,9 +48,9 @@ const ESCAPE = "globalThis.constructor.constructor('return process')()" /** One controllable child run: the test (or auto mode) settles it. */ interface ControlledRun { request: SubagentStartRequest - /** Fulfill the provider publication/readiness boundary. */ + /** Fulfill the provider's async start with a ready child. */ publish(): void - /** Reject the provider publication/readiness boundary. */ + /** Reject the provider's async start before ownership transfer. */ rejectStart(error: unknown): void settle(result: SubagentResult): void rejectResult(error: unknown): void @@ -75,17 +75,19 @@ class StubProvider implements SubagentProvider { private readonly reply?: (request: SubagentStartRequest, index: number) => SubagentResult, private readonly disposeDelayMs = 0, private readonly deferStart = false, - private readonly onCancel?: (reason: string | undefined, index: number) => void, + private readonly onAbortString?: (reason: string | undefined, index: number) => void, private readonly onSignalAbort?: (reason: unknown, index: number) => void, ) {} - start(request: SubagentStartRequest): SubagentRun { - const readiness = Promise.withResolvers() + async start(request: SubagentStartRequest): Promise { + const startGate = Promise.withResolvers() const terminal = Promise.withResolvers() + terminal.promise.catch(() => { /* provider owns early settlement until publication */ }) + let published = false const controlled: ControlledRun = { request, - publish: () => { readiness.resolve(undefined) }, - rejectStart: (error) => { readiness.reject(error) }, + publish: () => { published = true; startGate.resolve(undefined) }, + rejectStart: (error) => { startGate.reject(error) }, settle: (result) => { terminal.resolve(result) }, rejectResult: (error) => { terminal.reject(error) }, cancelled: undefined, @@ -94,24 +96,29 @@ class StubProvider implements SubagentProvider { } this.runs.push(controlled) const index = this.runs.length - 1 - request.signal?.addEventListener('abort', () => { - this.onSignalAbort?.(request.signal?.reason, index) - terminal.resolve({ output: [], stopReason: 'aborted' }) + request.signal.addEventListener('abort', () => { + controlled.cancelled = String(request.signal.reason ?? 'cancelled') + this.onAbortString?.(String(request.signal.reason ?? 'cancelled'), index) + this.onSignalAbort?.(request.signal.reason, index) + if (published) terminal.resolve({ output: [], stopReason: 'aborted' }) + else startGate.reject(new Error('child start aborted before publication')) }, { once: true }) - if (!this.deferStart) readiness.resolve(undefined) + if (!this.deferStart) controlled.publish() if (this.reply) { const reply = this.reply queueMicrotask(() => { terminal.resolve(reply(request, index)) }) } + try { + await startGate.promise + } catch (error: unknown) { + controlled.disposeCalls += 1 + controlled.disposed = true + throw error + } + if (request.signal.aborted) throw new Error('child start aborted before publication') return { id: AgentId(`stub-child-${index}`), - started: readiness.promise, result: terminal.promise, - cancel: (reason?: string) => { - controlled.cancelled = reason ?? 'cancelled' - this.onCancel?.(reason, index) - terminal.resolve({ output: [], stopReason: 'aborted' }) - }, dispose: () => { controlled.disposeCalls += 1 if (this.disposeDelayMs === 0) { @@ -140,7 +147,7 @@ interface SetupOptions { manual?: boolean disposeDelayMs?: number deferStart?: boolean - onChildCancel?: (reason: string | undefined, index: number) => void + onChildAbortString?: (reason: string | undefined, index: number) => void onChildSignalAbort?: (reason: unknown, index: number) => void } @@ -152,7 +159,7 @@ async function setup(options?: SetupOptions) { options?.manual ? undefined : options?.reply ?? (() => text('stub reply')), options?.disposeDelayMs ?? 0, options?.deferStart ?? false, - options?.onChildCancel, + options?.onChildAbortString, options?.onChildSignalAbort, ) ctx.subagents.registerProvider(provider) @@ -243,7 +250,7 @@ describe('dsh-workflow-workerthread', () => { expect(result.error).toContain('agent() could not start a child') }) - it('waits for child readiness before announcing it and snapshots a result that settled early', async () => { + it('waits for async provider start before announcing a result that settled early', async () => { const { ctx, parent, provider } = await setup({ manual: true, deferStart: true }) const order: string[] = [] ctx.on('workflow/agent-start', (_info, agent) => { order.push(`start:${agent.seq}`) }) @@ -254,10 +261,8 @@ describe('dsh-workflow-workerthread', () => { await waitFor(() => { expect(provider.runs.length).toBe(1) }) const early = text('accepted value') provider.runs[0]!.settle(early) - // Let the host observe + snapshot result while readiness remains pending. + // The provider still owns this early result while start is pending. await new Promise(resolve => setTimeout(resolve, 0)) - const earlyText = early.output[0] as { type: 'text'; text: string } - earlyText.text = 'mutated after settlement' expect(order).toEqual([]) provider.runs[0]!.publish() @@ -268,7 +273,7 @@ describe('dsh-workflow-workerthread', () => { expect(provider.runs[0]!.disposeCalls).toBe(1) }) - it('observes an early result rejection but sends ChildStarted before ChildFailed after readiness', async () => { + it('observes an early result rejection but sends ChildStarted before ChildFailed after start fulfills', async () => { const { ctx, parent, provider } = await setup({ manual: true, deferStart: true }) const lifecycle: string[] = [] ctx.on('workflow/agent-start', () => { lifecycle.push('start') }) @@ -299,7 +304,7 @@ describe('dsh-workflow-workerthread', () => { await handle.dispose() }) - it('classifies readiness rejection as AGENT_START, drops an early result, and emits no false lifecycle pair', async () => { + it('classifies provider start rejection as AGENT_START, drops an early result, and emits no false lifecycle pair', async () => { const { ctx, parent, provider } = await setup({ manual: true, deferStart: true }) const lifecycle: string[] = [] ctx.on('workflow/agent-start', () => { lifecycle.push('start') }) @@ -311,7 +316,7 @@ describe('dsh-workflow-workerthread', () => { }) await waitFor(() => { expect(provider.runs.length).toBe(1) }) // ACP-style failure can settle result(error) before its session/publication - // boundary rejects. Readiness must dominate that buffered child outcome. + // boundary rejects. Start rejection must dominate that buffered child outcome. provider.runs[0]!.settle({ output: [], stopReason: 'error' }) await new Promise(resolve => setTimeout(resolve, 0)) provider.runs[0]!.rejectStart(new Error('publication rolled back')) @@ -328,7 +333,7 @@ describe('dsh-workflow-workerthread', () => { expect(provider.runs[0]!.disposeCalls).toBe(1) }) - it('cancels and disposes a readiness-pending child once without publishing workflow lifecycle', async () => { + it('aborts a pending provider start once without publishing workflow lifecycle', async () => { const { ctx, parent, provider } = await setup({ manual: true, deferStart: true, config: { disposeGraceMs: 500 } }) const lifecycle: string[] = [] ctx.on('workflow/agent-start', () => { lifecycle.push('start') }) @@ -342,7 +347,7 @@ describe('dsh-workflow-workerthread', () => { expect(provider.runs[0]!.disposed).toBe(true) }) // Ensure the host-driven disposal removed the registry entry before the - // late readiness rejection; its callback must not invoke dispose again. + // late start rejection; its callback must not invoke dispose again. await new Promise(resolve => setTimeout(resolve, 0)) provider.runs[0]!.rejectStart(new Error('cancelled before publication')) @@ -360,11 +365,9 @@ describe('dsh-workflow-workerthread', () => { name: 'rejecting', capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: false }, inheritsParentContext: false, - start: () => ({ + start: async () => ({ id: AgentId('reject-child'), - started: Promise.resolve(), result: Promise.reject(new Error('backend exploded')), - cancel: () => { /* nothing in flight */ }, dispose: () => Promise.resolve(), }), } @@ -385,24 +388,20 @@ describe('dsh-workflow-workerthread', () => { try { await agent('p'); return 'unreachable' } catch (e) { return { code: e.code, message: e.message } } `)) expect(result.value).toMatchObject({ code: 'AGENT_RESULT' }) - expect((result.value as { message: string }).message).toContain('subagent result must be losslessly JSON-serializable') + expect((result.value as { message: string }).message).toContain('workflow child result could not cross the worker boundary') }) it('contains a non-JSON result even if the injected subagent service violates its normalization contract', async () => { - // SubagentService normally rejects this before the workflow sees it. Stub - // the injected seam itself so the host's defensive worker-boundary guard - // remains independently covered rather than becoming dead, untested code. + // The real worker boundary must reject a non-JSON same-process result. const { ctx, parent } = await setup() const invalid = { output: [], structured: () => { /* deliberately outside lossless JSON */ }, stopReason: 'completed', } as unknown as SubagentResult - const start = vi.spyOn(ctx.subagents, 'start').mockReturnValue({ + const start = vi.spyOn(ctx.subagents, 'start').mockResolvedValue({ id: AgentId('raw-invalid-child'), - started: Promise.resolve(), result: Promise.resolve(invalid), - cancel: () => { /* already settled */ }, dispose: () => Promise.resolve(), }) @@ -416,31 +415,6 @@ describe('dsh-workflow-workerthread', () => { .toContain('workflow child result could not cross the worker boundary') }) - it('reads each resolved child-result field once before crossing the worker boundary', async () => { - let structuredReads = 0 - class DriftedStructured { readonly value = 'drifted' } - const { ctx, parent } = await setup({ - reply: () => ({ - output: [], - get structured() { - structuredReads += 1 - return structuredReads === 1 ? { value: 'accepted' } : new DriftedStructured() - }, - stopReason: 'completed', - }), - }) - - const result = await run(ctx, parent, scripted(` - const found = await agent('p', { - schema: { type: 'object', properties: { value: { type: 'string' } }, required: ['value'] } - }) - return found.value - `)) - - expect(result.value).toBe('accepted') - expect(structuredReads).toBe(1) - }) - it('a child whose dispose() throws synchronously cannot wedge the script (the host acks anyway)', async () => { const ctx = new Context() await ctx.plugin(SubagentService) @@ -448,9 +422,8 @@ describe('dsh-workflow-workerthread', () => { name: 'bad-dispose', capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: false }, inheritsParentContext: false, - start: () => ({ + start: async () => ({ id: AgentId('bad-dispose-child'), - started: Promise.resolve(), result: Promise.resolve({ output: [{ type: 'text', text: 'fine' }], stopReason: 'completed' }), cancel: () => { /* settled already */ }, dispose: () => { throw new Error('dispose exploded') }, @@ -470,9 +443,8 @@ describe('dsh-workflow-workerthread', () => { name: 'coercion-trap-dispose', capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: false }, inheritsParentContext: false, - start: () => ({ + start: async () => ({ id: AgentId('trap-child'), - started: Promise.resolve(), result: Promise.resolve({ output: [{ type: 'text', text: 'fine' }], stopReason: 'completed' }), cancel: () => { /* settled already */ }, // The rejection VALUE's own coercion throws: a warn built with bare @@ -755,7 +727,7 @@ describe('dsh-workflow-workerthread', () => { expect(provider.runs[0]!.disposed).toBe(true) }) - it('dispose() reaps a registered stray after result settlement even when the worker cannot relay disposal', async () => { + it('result settlement reaps a registered stray even when the worker cannot relay disposal', async () => { const { ctx, parent, provider } = await setup({ manual: true, config: { provider: 'stub', disposeGraceMs: 30_000 }, @@ -775,13 +747,9 @@ describe('dsh-workflow-workerthread', () => { result: { value: 'synthetic completion', stopReason: 'completed', agentsStarted: 1 }, }) await expect(handle.result).resolves.toMatchObject({ stopReason: 'completed' }) - expect(provider.runs[0]!.disposed).toBe(false) + await waitFor(() => { expect(provider.runs[0]!.disposed).toBe(true) }, 1000) const disposal = handle.dispose() - // A 30-second grace makes this assertion mutation-sensitive: without the - // settled-path host reap, no worker message can start child disposal and - // this bounded wait fails long before the grace fallback. - await waitFor(() => { expect(provider.runs[0]!.disposed).toBe(true) }, 1000) await disposal expect(provider.runs[0]!.disposeCalls).toBe(1) await ctx.fiber.dispose() @@ -795,21 +763,16 @@ describe('dsh-workflow-workerthread', () => { name: 'signal-only', capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: false }, inheritsParentContext: false, - start: (request) => { + start: async (request) => { let settle!: (result: SubagentResult) => void const result = new Promise((resolve) => { settle = resolve }) - request.signal?.addEventListener('abort', () => { - aborted.push(String(request.signal?.reason)) + request.signal.addEventListener('abort', () => { + aborted.push(String(request.signal.reason)) settle({ output: [], stopReason: 'aborted' }) }, { once: true }) return { id: AgentId('signal-only-child'), - started: Promise.resolve(), result, - // The seam leaves a provider free to honor EITHER cancel channel; - // this one deliberately ignores run.cancel() — only the request - // signal can wind it down. - cancel: () => { /* signal-only by design */ }, dispose: () => Promise.resolve(), } }, @@ -834,7 +797,7 @@ describe('dsh-workflow-workerthread', () => { await handle.dispose() }) - it('the settle-reap explicitly cancels a readiness-pending stray before workflow/end', async () => { + it('the settle-reap aborts a pending provider start before workflow/end', async () => { const { ctx, parent, provider } = await setup({ manual: true, deferStart: true }) const childLifecycle: string[] = [] let cancellationAtWorkflowEnd: string | undefined @@ -845,7 +808,7 @@ describe('dsh-workflow-workerthread', () => { }) const handle = ctx.workflows.start({ ...scripted(` - agent('readiness-pending stray') + agent('start-pending stray') return 'done' `), parent, @@ -864,129 +827,11 @@ describe('dsh-workflow-workerthread', () => { expect(provider.runs[0]!.disposeCalls).toBe(1) }) - it('post-result child cleanup cannot reentrantly rewrite a completed workflow as cancelled', async () => { - let cancelCallbacks = 0 - let signalCallbacks = 0 - const { ctx, parent, provider } = await setup({ - manual: true, - deferStart: true, - onChildCancel: () => { - cancelCallbacks += 1 - // The first callback is host cleanup for the already-arrived Result. - // Reentering cancel() here is later than that message and must not - // retroactively win the result race. Its nested child cancel is - // intentionally ignored to keep the adversarial callback finite. - if (cancelCallbacks === 1) handle.cancel('reentrant child cleanup') - }, - onChildSignalAbort: () => { - signalCallbacks += 1 - handle.cancel('reentrant signal cleanup') - }, - }) - const handle = ctx.workflows.start({ - ...scripted(` - agent('readiness-pending stray') - return 'completed first' - `), - parent, - }) - - const result = await handle.result - - expect(result).toMatchObject({ value: 'completed first', stopReason: 'completed', agentsStarted: 1 }) - expect(signalCallbacks).toBe(1) - expect(cancelCallbacks).toBe(1) - // Readiness crossing after Result is a terminal-admission refusal: no - // ChildStarted/lifecycle publication, and host-owned disposal begins. - provider.runs[0]!.publish() - await waitFor(() => { expect(provider.runs[0]!.disposed).toBe(true) }, 1000) - expect(cancelCallbacks).toBe(1) - await handle.dispose() - await ctx.fiber.dispose() - }) - - it('late readiness after completed disposal cannot cancel or dispose the retired child twice', async () => { - let explicitCancels = 0 - const lifecycle: string[] = [] - const { ctx, parent, provider } = await setup({ - manual: true, - deferStart: true, - onChildCancel: () => { explicitCancels += 1 }, - }) - ctx.on('workflow/agent-start', () => { lifecycle.push('start') }) - ctx.on('workflow/agent-end', () => { lifecycle.push('end') }) - const handle = ctx.workflows.start({ - ...scripted("agent('retired readiness')\nreturn 'done'"), - parent, - }) - - await expect(handle.result).resolves.toMatchObject({ stopReason: 'completed' }) - expect(explicitCancels).toBe(1) - await handle.dispose() - expect(provider.runs[0]!.disposed).toBe(true) - expect(provider.runs[0]!.disposeCalls).toBe(1) - - // The Promise may still fulfill after its run left every host ledger. - // Refusal replies once but must not recreate the deleted cancel gate. - provider.runs[0]!.publish() - await Promise.resolve() - await Promise.resolve() - expect(explicitCancels).toBe(1) - expect(provider.runs[0]!.disposeCalls).toBe(1) - expect(lifecycle).toEqual([]) - await ctx.fiber.dispose() - }) - - it.each([ - ['synchronous', (cancel: () => void) => { cancel() }], - ['microtask', (cancel: () => void) => { queueMicrotask(cancel) }], - ])('a ready stray %s cleanup callback cannot beat the earlier worker result claim', async (_mode, reenter) => { - let reentered = false - const explicitCancels = new Map() - const { ctx, parent, provider } = await setup({ - manual: true, - onChildCancel: (_reason, index) => { - explicitCancels.set(index, (explicitCancels.get(index) ?? 0) + 1) - if (index !== 0 || reentered) return - reentered = true - reenter(() => { handle.cancel('reentered from child cleanup') }) - }, - }) - const handle = ctx.workflows.start({ - ...scripted(` - agent('ready stray') - return await agent('gate') - `), - parent, - }) - const cancelChildSpy = vi.spyOn(handle as unknown as { - cancelChild(callId: number, run: SubagentRun, reason?: string): void - }, 'cancelChild') - await waitFor(() => { expect(provider.runs).toHaveLength(2) }) - provider.runs[1]!.settle(text('gate completed')) - - const result = await handle.result - await Promise.resolve() - - expect(result).toMatchObject({ value: 'gate completed', stopReason: 'completed', agentsStarted: 2 }) - expect(reentered).toBe(true) - // The host claim and worker's FIFO-later ChildCancel both reach the - // routing gate, but the provider callback is not an idempotent seam: - // invoke it exactly once for this callId. - await waitFor(() => { - expect(cancelChildSpy.mock.calls.filter(([callId]) => callId === 1)).toHaveLength(2) - }, 1000) - expect(explicitCancels.get(0)).toBe(1) - cancelChildSpy.mockRestore() - await handle.dispose() - await ctx.fiber.dispose() - }) - it('a duplicate Result after the terminal claim cannot repeat cleanup or rewrite the outcome', async () => { - let explicitCancels = 0 + let signalAborts = 0 const { ctx, parent, provider } = await setup({ manual: true, - onChildCancel: (_reason, index) => { if (index === 0) explicitCancels += 1 }, + onChildAbortString: (_reason, index) => { if (index === 0) signalAborts += 1 }, }) const handle = ctx.workflows.start({ ...scripted("agent('stray')\nawait new Promise(() => {})"), @@ -1005,266 +850,9 @@ describe('dsh-workflow-workerthread', () => { }) await expect(handle.result).resolves.toMatchObject({ value: 'first', stopReason: 'completed' }) - expect(explicitCancels).toBe(1) + expect(signalAborts).toBe(1) await handle.dispose() - expect(explicitCancels).toBe(1) - await ctx.fiber.dispose() - }) - - it('contains a throwing child cancel and still settles after cancelling peer strays', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - let starts = 0 - const cancelled: string[] = [] - const warnings: string[] = [] - ctx.logger.warn = ((message: unknown) => { warnings.push(String(message)) }) as typeof ctx.logger.warn - const provider: SubagentProvider = { - name: 'throwing-cancel', - capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: false }, - inheritsParentContext: false, - start: () => { - const index = starts++ - return { - id: AgentId(`throwing-cancel-${index}`), - started: new Promise(() => { /* readiness stays pending */ }), - result: new Promise(() => { /* cancellation callback owns settlement */ }), - cancel: (reason?: string) => { - if (index === 0) throw new Error('cancel callback broke') - cancelled.push(`${index}:${reason ?? 'cancelled'}`) - }, - dispose: () => Promise.resolve(), - } - }, - } - ctx.subagents.registerProvider(provider) - await ctx.plugin(WorkerWorkflowEngine, { provider: 'throwing-cancel', maxConcurrentAgents: 2 }) - const handle = ctx.workflows.start({ - ...scripted(` - agent('first stray') - agent('second stray') - return 'done' - `), - parent: fakeParent(), - }) - - const result = await handle.result - - expect(result.stopReason).toBe('completed') - expect(starts).toBe(2) - expect(cancelled).toContain('1:workflow settled') - expect(warnings.some(message => message.includes('cancel callback broke'))).toBe(true) - await handle.dispose() - }) - - it("cancel() drives each child's explicit cancel() host-side: a wedged worker cannot delay it", async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - let starts = 0 - const cancelled: string[] = [] - const provider: SubagentProvider = { - name: 'cancel-only', - capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: false }, - inheritsParentContext: false, - start: () => { - starts += 1 - return { - id: AgentId('cancel-only-child'), - started: Promise.resolve(), - result: new Promise(() => { /* only cancel() ends this child */ }), - // Deliberately ignores the request signal — the seam leaves a - // provider free to honor ONLY the explicit cancel() channel. - cancel: (reason?: string) => { cancelled.push(reason ?? 'cancelled') }, - dispose: () => Promise.resolve(), - } - }, - } - ctx.subagents.registerProvider(provider) - // A deliberately huge grace: if only the grace/terminate reap could - // reach this child, the assertion below would time out first. - await ctx.plugin(WorkerWorkflowEngine, { provider: 'cancel-only', maxConcurrentAgents: 2, disposeGraceMs: 30_000 }) - const handle = ctx.workflows.start({ - // The stray child's start RPC reaches the host, then the script wedges - // its own worker in a synchronous spin: the worker cannot process the - // Cancel message, so it can relay NO ChildCancel RPC — only the host's - // own children loop can deliver the explicit cancel in time. The - // microtask yields let the agent() continuation POST its child-start - // before the spin seizes the worker's loop (the posted message needs - // no further worker-loop turns to reach the host). - ...scripted(` - agent('wedged child') - for (let i = 0; i < 20; i++) await null - const end = Date.now() + 1500 - while (Date.now() < end) {} - return 'raced' - `), - parent: fakeParent(), - }) - await waitFor(() => { expect(starts).toBe(1) }) - handle.cancel('stop now') - await waitFor(() => { expect(cancelled).toEqual(['stop now']) }, 800) - // The wedged worker's own completion loses to the in-flight cancel. - const result = await handle.result - expect(result.stopReason).toBe('cancelled') - await handle.dispose() - }, 15_000) - - it.each(['fulfills', 'rejects'] as const)('provider.start() reentrant cancellation refuses the run when readiness later %s', async (readinessOutcome) => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const readiness = Promise.withResolvers() - let starts = 0 - let explicitCancels = 0 - let disposals = 0 - let sawAbortedSignal = false - const lifecycle: string[] = [] - const provider: SubagentProvider = { - name: 'start-reentry', - capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: false }, - inheritsParentContext: false, - start: (request) => { - starts += 1 - // This arbitrary provider callback runs before onChildStart can put - // the returned run in its registry. Cancellation must be rechecked - // after return instead of trusting the pre-start admission check. - handle.cancel('provider start reentered cancellation') - sawAbortedSignal = request.signal?.aborted === true - return { - id: AgentId('start-reentry-child'), - started: readiness.promise, - result: new Promise(() => { /* refusal owns teardown */ }), - // Deliberately honors only the explicit channel. It must still be - // reached promptly even though the first host fanout saw no run. - cancel: () => { explicitCancels += 1 }, - dispose: () => { - disposals += 1 - return Promise.resolve() - }, - } - }, - } - ctx.subagents.registerProvider(provider) - await ctx.plugin(WorkerWorkflowEngine, { - provider: 'start-reentry', - maxConcurrentAgents: 2, - disposeGraceMs: 30_000, - }) - ctx.on('workflow/agent-start', () => { lifecycle.push('start') }) - ctx.on('workflow/agent-end', () => { lifecycle.push('end') }) - const handle = ctx.workflows.start({ - ...scripted("await agent('reentrant provider')\nreturn 'unreachable'"), - parent: fakeParent(), - }) - - await waitFor(() => { expect(starts).toBe(1) }) - // Either later readiness settlement must not answer the already-refused - // start again or emit a workflow lifecycle pair. - if (readinessOutcome === 'fulfills') readiness.resolve(undefined) - else readiness.reject(new Error('late readiness rejection after refusal')) - let result: WorkflowResult | undefined - void handle.result.then((value) => { result = value }) - await waitFor(() => { - expect(explicitCancels).toBe(1) - expect(disposals).toBe(1) - expect(result?.stopReason).toBe('cancelled') - }, 1000) - expect(sawAbortedSignal).toBe(true) - expect(lifecycle).toEqual([]) - await handle.dispose() - expect(explicitCancels).toBe(1) - expect(disposals).toBe(1) - await ctx.fiber.dispose() - }) - - it('claims workflow and child disposal before a raw provider disposer reenters handle.dispose()', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const terminal = Promise.withResolvers() - const observed: { reentrant?: Promise } = {} - let starts = 0 - let rawDisposeCalls = 0 - const provider: SubagentProvider = { - name: 'dispose-reentry', - capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: false }, - inheritsParentContext: false, - start: () => { - starts += 1 - return { - id: AgentId('dispose-reentry-child'), - started: Promise.resolve(), - result: terminal.promise, - cancel: () => { terminal.resolve({ output: [], stopReason: 'aborted' }) }, - dispose: () => { - rawDisposeCalls += 1 - observed.reentrant = handle.dispose() - return Promise.resolve() - }, - } - }, - } - ctx.subagents.registerProvider(provider) - await ctx.plugin(WorkerWorkflowEngine, { provider: 'dispose-reentry', maxConcurrentAgents: 2 }) - const handle = ctx.workflows.start({ - ...scripted("await agent('live child')\nreturn 'unreachable'"), - parent: fakeParent(), - }) - await waitFor(() => { expect(starts).toBe(1) }) - - const disposal = handle.dispose() - - expect(observed.reentrant).toBe(disposal) - await disposal - expect(rawDisposeCalls).toBe(1) - await expect(handle.result).resolves.toMatchObject({ stopReason: 'cancelled' }) - await ctx.fiber.dispose() - }) - - it('claims worker-originated child disposal before its raw disposer reenters holder disposal', async () => { - const ctx = new Context() - await ctx.plugin(SubagentService) - const terminal = Promise.withResolvers() - const observed: { reentrant?: Promise } = {} - let starts = 0 - let rawDisposeCalls = 0 - const provider: SubagentProvider = { - name: 'child-dispose-reentry', - capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: false }, - inheritsParentContext: false, - start: () => { - starts += 1 - return { - id: AgentId('child-dispose-reentry-child'), - started: Promise.resolve(), - result: terminal.promise, - cancel: () => { terminal.resolve({ output: [], stopReason: 'aborted' }) }, - dispose: () => { - rawDisposeCalls += 1 - // This begins holder disposal from the worker's ChildDispose - // callback, before any public handle.dispose() call exists. - observed.reentrant = handle.dispose() - return Promise.resolve() - }, - } - }, - } - ctx.subagents.registerProvider(provider) - await ctx.plugin(WorkerWorkflowEngine, { provider: 'child-dispose-reentry', maxConcurrentAgents: 2 }) - const handle = ctx.workflows.start({ - ...scripted("return await agent('settling child')"), - parent: fakeParent(), - }) - const finishChildSpy = vi.spyOn(handle as unknown as { - finishChild(callId: number): void - }, 'finishChild') - await waitFor(() => { expect(starts).toBe(1) }) - - terminal.resolve({ output: [{ type: 'text', text: 'done' }], stopReason: 'completed' }) - - await waitFor(() => { expect(observed.reentrant).toBeDefined() }, 1000) - await observed.reentrant - expect(rawDisposeCalls).toBe(1) - expect(finishChildSpy.mock.calls.filter(([callId]) => callId === 1)).toHaveLength(1) - finishChildSpy.mockRestore() - await expect(handle.result).resolves.toMatchObject({ stopReason: 'cancelled' }) + expect(signalAborts).toBe(1) await ctx.fiber.dispose() }) @@ -1461,34 +1049,87 @@ describe('dsh-workflow-workerthread', () => { await ctx.fiber.dispose() }) + it('refuses and disposes a provider run that becomes ready after its real worker dies', async () => { + const ctx = new Context() + await ctx.plugin(SubagentService) + const requested = Promise.withResolvers() + const ready = Promise.withResolvers() + let disposeCalls = 0 + const warn = vi.spyOn(ctx.logger, 'warn').mockImplementation(() => ctx.logger) + const provider: SubagentProvider = { + name: 'late-ready', + capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: false }, + inheritsParentContext: false, + start: (request) => { + requested.resolve(request) + // Model a backend whose independent startup boundary cannot be + // interrupted promptly. The host must still reject ownership if the + // worker dies before this promise transfers the ready run. + return ready.promise + }, + } + ctx.subagents.registerProvider(provider) + await ctx.plugin(WorkerWorkflowEngine, { provider: 'late-ready', maxConcurrentAgents: 1 }) + const lifecycle: string[] = [] + ctx.on('workflow/agent-start', () => { lifecycle.push('start') }) + ctx.on('workflow/agent-end', () => { lifecycle.push('end') }) + + const handle = ctx.workflows.start({ + ...scripted("return await agent('pending startup')"), + parent: fakeParent(), + }) + const request = await requested.promise + const worker = (handle as unknown as { worker: Worker }).worker + + // Kill the actual Worker while provider startup is independently + // pending. Death closes admission and aborts the shared signal, but this + // deliberately uncooperative provider still fulfills afterward. + await worker.terminate() + const result = await handle.result + expect(result.stopReason).toBe('error') + expect(result.error).toContain('exit code') + expect(request.signal.aborted).toBe(true) + expect(request.signal.reason).toBe('workflow worker gone') + + ready.resolve({ + id: AgentId('late-ready-child'), + result: Promise.resolve({ output: [], stopReason: 'aborted' }), + dispose: () => { + disposeCalls += 1 + return Promise.reject(new Error('late ready dispose failed')) + }, + }) + await waitFor(() => { + expect(disposeCalls).toBe(1) + expect(warn).toHaveBeenCalledWith(expect.stringContaining('refused child dispose failed: Error: late ready dispose failed')) + }, 1000) + expect(lifecycle).toEqual([]) + + await handle.dispose() + expect(disposeCalls).toBe(1) + await ctx.fiber.dispose() + }) + it('a worker that exits before settling reports an error result and reaps its children', async () => { const ctx = new Context() await ctx.plugin(SubagentService) // The child's dispose() REJECTS on top of the worker death: the reap // must contain it (warn, not crash) while still emptying the registry. - const cancelled: string[] = [] const signalAborts: unknown[] = [] const provider: SubagentProvider = { name: 'doomed', capabilities: { outputSchema: true, depthLimit: true, toolFilter: true, persona: false }, inheritsParentContext: false, - start: (request) => { - request.signal?.addEventListener('abort', () => { - signalAborts.push(request.signal?.reason) + start: async (request) => { + request.signal.addEventListener('abort', () => { + signalAborts.push(request.signal.reason) // The death claim precedes the shared-signal fanout. This // synchronous callback cannot turn death into cancellation. handle.cancel('reentered from worker-death signal cleanup') }, { once: true }) return { id: AgentId('doomed-child'), - started: Promise.resolve(), result: new Promise(() => { /* never settles; the reap is the teardown */ }), - cancel: (reason?: string) => { - cancelled.push(reason ?? 'cancelled') - // Exercise the later microtask case too: terminal ownership - // remains closed after the death callback returns. - queueMicrotask(() => { handle.cancel('reentered from worker-death child cleanup') }) - }, dispose: () => Promise.reject(new Error('dispose exploded during reap')), } }, @@ -1521,7 +1162,6 @@ describe('dsh-workflow-workerthread', () => { // cold-start race; tight explicit bound (see the helper's doc comment). await waitFor(() => { expect(signalAborts).toEqual(['workflow worker gone']) - expect(cancelled).toEqual(['workflow worker gone']) }, 1000) await Promise.resolve() expect(result.stopReason).toBe('error') @@ -1647,14 +1287,14 @@ describe('dsh-workflow-workerthread', () => { }) describe('service surface', () => { - it('run ids are unique per start; the run handle and event payloads hold SEPARATE meta clones', async () => { + it('run ids are unique and lifecycle meta is the run\'s borrowed immutable value', async () => { const { ctx, parent } = await setup() let eventMeta: WorkflowRunInfo | undefined ctx.on('workflow/start', (info) => { eventMeta = info }) const first = ctx.workflows.start({ ...scripted('return 1'), parent }) const second = ctx.workflows.start({ ...scripted('return 2'), parent }) expect(first.id).not.toBe(second.id) - eventMeta!.meta.name = 'corrupted' + expect(eventMeta!.meta).toBe(second.meta) expect(second.meta.name).toBe('test-flow') await Promise.all([first.result, second.result]) await first.dispose() diff --git a/packages/workflow/workflow/README.md b/packages/workflow/workflow/README.md index 5332243a85..7caf169260 100644 --- a/packages/workflow/workflow/README.md +++ b/packages/workflow/workflow/README.md @@ -1,29 +1,43 @@ # @deepseek-ai/dsh-workflow -The **workflow seam** (`ctx.workflows`): an abstract service defining WHAT a workflow engine does — execute a model-written orchestration script that fans out subagents — without saying HOW. The bash-shaped third of the [workflow family](../README.md): implementations subclass `WorkflowService` and register as the `workflows` service (one per context); [`dsh-workflow-workerthread`](../workflow-workerthread/README.md) (one worker thread per run) is the implementation, and [`dsh-tool-workflow`](../tool-workflow/README.md) is the model-facing consumer. +The workflow seam (`ctx.workflows`) executes a model-written orchestration script that can fan out subagents. The seam defines the script, run, result, error, and event contracts; an engine decides how to isolate and execute the script. -## Service: `WorkflowService` (abstract) +`@deepseek-ai/dsh-workflow-workerthread` is the current engine and `@deepseek-ai/dsh-tool-workflow` is the model-facing consumer. A future process or sandbox engine can replace the implementation without changing the tool. -`start(request: WorkflowStartRequest): WorkflowRun` — parse and execute a script. Throws synchronously (`SCRIPT_PARSE`/`META_INVALID`) for a script that cannot begin; once a run is returned, its `result` NEVER rejects — every failure resolves with `stopReason: 'error'` (or `'cancelled'`) — and once the run is cancelled, `result` settles within the implementation's bounded grace even if the script itself never settles (a consumer awaiting `result` must never be wedged past a cancellation). `dispose()` must reach quiescence within a bounded grace (cancel → wait for the script to settle and its children to finish disposing → abandon), never hanging its caller. Runs are HOLDER-owned: the engine does not track its live runs, so disposing the engine's fiber mid-run leaves each run to its holder's teardown. +## Service and run contract -The protected `emitWorkflowEvent` helper dispatches the `workflow/*` events with PER-LISTENER containment and PER-LISTENER payload snapshots (a throwing subscriber is logged, never propagated, and cannot starve later listeners; each subscriber gets its own clone of the payload, so mutating it corrupts neither the engine nor other listeners) — the same containment guarantee as the subagent seam's lifecycle emits. +`WorkflowService.start(request): WorkflowRun` validates enough synchronously to reject a malformed meta block or unparseable script before a run exists. Once returned, `WorkflowRun.result` never rejects: execution failures resolve with `stopReason: 'error'`, and cancellation resolves with `cancelled` within the engine's bounded grace. -## Vocabulary +A run is holder-owned. Engine-plugin unload prevents new starts but does not revoke accepted runs. The holder must call `dispose()` on every path; disposal cancels remaining work and reaches or abandons quiescence within the documented bound. -- `WorkflowStartRequest` — `{ script, args?, parent: Agent, signal? }`. `parent` is REQUIRED: every child the script spawns is attributed to it. `args` must be plain host-realm JSON data. -- `WorkflowMeta` / `WorkflowPhase` — the workflow's identity block, carried as plain JSON data on the start request (Claude Code meta vocabulary: required `name`/`description`, optional `whenToUse`/`phases`) and shape-validated by the engine. -- `WorkflowRun` — `{ id, meta, result, cancel(reason?), dispose() }`; the consumer awaits `result` and MUST `dispose` on every path. -- `WorkflowResult` — `{ value, stopReason: 'completed'|'cancelled'|'error', error?, agentsStarted }`; `value` is the script's materialized return (plain JSON data; `null` for no return). -- `WorkflowError` — `HarnessError` with a `WorkflowErrorCode` and a `fatal` flag driving the combinator discipline: a fatal error (bad hook arguments, unsupported options/schemas, tripped caps, seam start failures, cancellation) always propagates through `parallel()`/`pipeline()` instead of dissolving into a per-item `null`. `isFatalWorkflowError(error)` is the catch-site predicate. +`WorkflowStartRequest` contains `{ meta, script, args?, parent, signal? }`. `parent` attributes every child agent to the invoking agent. `meta` and `args` are plain data, not script fragments. + +`WorkflowRun` exposes `{ id, meta, result, cancel(reason?), dispose() }`. `WorkflowResult` contains `{ value, stopReason, error?, agentsStarted }`; `value` is plain JSON data or `null`. ## Events -All observe-only emits carrying DATA SNAPSHOTS (`WorkflowRunInfo` = id + meta) — never the live `WorkflowRun`, so a listener cannot gain `cancel`/`dispose`; control stays with the `start()` caller: +Workflow events are observe-only. They carry `WorkflowRunInfo` (`id` plus `meta`) rather than the live run, so listeners cannot acquire cancellation or disposal authority. -- `workflow/start`(info) / `workflow/end`(info, resultInfo) — run lifecycle; `resultInfo` deliberately omits the value. -- `workflow/phase`(info, title) / `workflow/log`(info, message) — script narration. -- `workflow/agent-start`(info, agent) / `workflow/agent-end`(info, agent + outcome) — ready-child lifecycle correlated by `seq`; the [generated event contract](../../../docs/cordis-catalog/events.md#workflowagent-start--emit) defines publication and pairing. +- `workflow/start` / `workflow/end` pair the run. +- `workflow/phase` and `workflow/log` expose script narration. +- `workflow/agent-start` / `workflow/agent-end` pair each child call by `seq`; a child whose async provider start rejects emits neither. -## Non-goals (this cut) +Same-process event payloads are borrowed immutable values. Every listener is independently contained: a synchronous throw or rejected returned promise is logged without starving peers or changing execution. -Background collection, journaling/resume, saved workflows, nested `workflow()`, token budgets — see the [RFC's deferred section](../../../docs/rfc/implemented/feature/2026-07-05-dynamic-workflows.md). +## Failure discipline + +`WorkflowError` carries a code and a `fatal` flag. Fatal errors always escape `parallel()` and `pipeline()` instead of becoming an ordinary per-item `null`: + +- `SCRIPT_PARSE` / `META_INVALID` — the workflow cannot start. +- `INVALID_ARGUMENT` / `UNSUPPORTED_OPTION` / `UNSUPPORTED_SCHEMA` — a hook call violates the engine contract. +- `AGENT_CAP` / `ITEM_CAP` — configured safety limits were exceeded. +- `AGENT_START` — the provider's async start rejected. +- `AGENT_RESULT` — a ready child's result rejected with an infrastructure fault. +- `RESULT_UNSERIALIZABLE` — a script/worker value is not plain JSON data. +- `CANCELLED` — cancellation owns the run and pending/future hooks reject. + +A child that resolves normally with a non-completed stop reason is not an infrastructure exception: `agent()` returns `null`, allowing the script to handle an ordinary child failure. + +## Non-goals + +Background collection, journaling/resume, saved workflows, nested `workflow()`, and token budgets are deferred. See the [dynamic-workflows RFC](../../../docs/rfc/implemented/feature/2026-07-05-dynamic-workflows.md). diff --git a/packages/workflow/workflow/src/index.ts b/packages/workflow/workflow/src/index.ts index 19f52b5d88..0fa0289b41 100644 --- a/packages/workflow/workflow/src/index.ts +++ b/packages/workflow/workflow/src/index.ts @@ -9,14 +9,12 @@ * separate-process sandbox) swap in without touching the model-facing tool * that consumes them (`@deepseek-ai/dsh-tool-workflow`). * - * The `workflow/*` lifecycle events are OBSERVE-ONLY data snapshots: they + * The `workflow/*` lifecycle events are OBSERVE-ONLY data: they * carry {@link WorkflowRunInfo} (id + meta), never the live {@link WorkflowRun} * — a listener must not gain `cancel`/`dispose`; control stays with the - * `start()` caller holding the run. Every emit is per-listener contained (a - * throwing subscriber is logged, never propagated) and every listener gets its - * own payload clone (mutating it corrupts nothing), so one bad observer can - * neither strand a live run, starve later listeners, nor poison another - * listener's view. + * `start()` caller holding the run. Same-process payloads are borrowed + * immutable values. Every listener is independently contained, so a throw or + * rejected promise can neither strand a run nor starve peers. * * @module @deepseek-ai/dsh-workflow */ @@ -78,7 +76,7 @@ declare module 'cordis' { /** * One `agent()` call established a ready child run. Paired with * {@link Events['workflow/agent-end']} by `agent.seq`. A call that never - * crosses the provider's publication/readiness boundary emits neither + * receives a ready run from the provider emits neither * event in this pair. * @param info - the run's identity snapshot. * @param agent - the call's sequence number, label, phase, and child id. @@ -131,11 +129,10 @@ export type WorkflowEventName = * - `UNSUPPORTED_SCHEMA` — an `agent()` schema outside the structured-output * subset (see dsh-tools). * - `AGENT_CAP` / `ITEM_CAP` — the run/agent caps tripped. - * - `AGENT_START` — synchronous subagent start or the provider's asynchronous - * publication/readiness boundary failed before cancellation took precedence. - * - `AGENT_RESULT` — a run whose readiness FULFILLED had its `result` REJECT: an - * infrastructure fault at the subagent seam, even if the rejection settled - * before readiness. This is distinct from a child that failed and resolved + * - `AGENT_START` — the provider's asynchronous start rejected before + * cancellation took precedence. + * - `AGENT_RESULT` — a ready run had its `result` REJECT: an infrastructure + * fault at the subagent seam. This is distinct from a child that failed and resolved * (which is the per-item `null`, never an error). * - `RESULT_UNSERIALIZABLE` — a value crossing the script/host value boundary * is not plain JSON data. @@ -198,8 +195,8 @@ export function isFatalWorkflowError(error: unknown): boolean { * `result` SETTLES within the implementation's bounded grace even if the * script itself never settles (a consumer awaiting `result` must never be * wedged past a cancellation). - * - The `workflow/*` events fire through {@link emitWorkflowEvent} (data - * snapshots, per-listener containment); `workflow/end` fires exactly once + * - The `workflow/*` events fire through {@link emitWorkflowEvent} (borrowed + * immutable data, per-listener containment); `workflow/end` fires exactly once * per started run, after `result` is settled or as it settles. * - `dispose()` reaches quiescence within a bounded grace: it cancels, waits * for the script to settle AND its started children to finish disposing, @@ -225,12 +222,9 @@ export abstract class WorkflowService extends Service { abstract start(request: WorkflowStartRequest): WorkflowRun /** - * Emit one `workflow/*` lifecycle event with PER-LISTENER containment and - * PER-LISTENER payload snapshots: each subscriber is dispatched individually - * with its OWN structural clone of the payload (the payloads are plain JSON - * data by the seam contract), so a listener mutating what it received can - * corrupt neither the engine's live state nor any other listener's or later - * event's view; a thrown listener is logged (never propagated — the logging + * Emit one `workflow/*` lifecycle event with per-listener containment. Each + * subscriber receives the same borrowed immutable payload; a throw or + * asynchronously rejected listener is logged (never propagated — the logging * itself is total, even for a thrown value whose own string coercion * throws), so one bad subscriber can neither fail the engine mid-run, * surface as an unhandled rejection on a detached settle hook, nor starve @@ -242,9 +236,10 @@ export abstract class WorkflowService extends Service { protected emitWorkflowEvent(name: WorkflowEventName, ...args: unknown[]): void { for (const callback of this.ctx.events.dispatch('emit', [name, ...args])) { try { - // The declared workflow/* signatures are all void-returning emits; the - // dispatch callback applies the payload tuple. - ;(callback as (...payload: unknown[]) => void)(...structuredClone(args)) + const returned: unknown = (callback as (...payload: unknown[]) => unknown)(...args) + void Promise.resolve(returned).catch((error: unknown) => { + this.ctx.logger.warn(`workflow: ${name} listener rejected: ${renderListenerError(error)}`) + }) } catch (error: unknown) { this.ctx.logger.warn(`workflow: ${name} listener threw: ${renderListenerError(error)}`) } diff --git a/packages/workflow/workflow/src/types.ts b/packages/workflow/workflow/src/types.ts index 66da4f1268..981a2da172 100644 --- a/packages/workflow/workflow/src/types.ts +++ b/packages/workflow/workflow/src/types.ts @@ -128,7 +128,7 @@ export interface WorkflowRun { dispose(): Promise } -/** Identifying detail for a run, carried by every `workflow/*` event (a data snapshot, never the live run). */ +/** Identifying detail for a run, carried by every `workflow/*` event as borrowed immutable data, never the live run. */ export interface WorkflowRunInfo { /** The run's id. */ id: WorkflowRunId diff --git a/packages/workflow/workflow/tests/workflow.spec.ts b/packages/workflow/workflow/tests/workflow.spec.ts index a983e5a2b3..0df3824071 100644 --- a/packages/workflow/workflow/tests/workflow.spec.ts +++ b/packages/workflow/workflow/tests/workflow.spec.ts @@ -66,26 +66,21 @@ describe('dsh-workflow (interface)', () => { ]) }) - it('gives each listener its OWN payload snapshot: mutation corrupts neither peers nor the caller', async () => { + it('contains an asynchronously rejected listener without starving peers', async () => { const ctx = new Context() await ctx.plugin(StubEngine) + const warn = vi.spyOn(ctx.logger, 'warn').mockImplementation(() => ctx.logger) const seen: string[] = [] - ctx.on('workflow/agent-start', (info, agent) => { - agent.label = 'HACKED' - info.meta.name = 'HACKED' - seen.push('mutator') - }) - ctx.on('workflow/agent-start', (info, agent) => { - seen.push(`${info.meta.name}/${agent.label}`) - }) + // Runtime listeners may return thenables even though the declaration's observable result is void. + // eslint-disable-next-line @typescript-eslint/no-misused-promises -- exercises rejected-listener containment + ctx.on('workflow/agent-start', async () => { throw new Error('async observer failed') }) + ctx.on('workflow/agent-start', (_info, agent) => { seen.push(agent.label) }) const engine = ctx.workflows as StubEngine - const info: WorkflowRunInfo = { id: WorkflowRunId('run-2'), meta: { name: 'w', description: 'd' } } const payload = { seq: 1, label: 'original', childId: 'c' } - engine.emit('workflow/agent-start', info, payload) - expect(seen).toEqual(['mutator', 'w/original']) - // The caller's own objects are pristine too — no listener ever saw them. - expect(info.meta.name).toBe('w') - expect(payload.label).toBe('original') + engine.emit('workflow/agent-start', INFO, payload) + await Promise.resolve() + expect(seen).toEqual(['original']) + expect(String(warn.mock.calls[0]![0])).toContain('listener rejected') }) it('contains a throwing listener PER LISTENER: later listeners still run, nothing propagates', async () => { diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml index 575f2bbf9f..649a986385 100644 --- a/pnpm-lock.yaml +++ b/pnpm-lock.yaml @@ -408,9 +408,6 @@ importers: '@deepseek-ai/dsh-scope': specifier: workspace:^ version: link:../scope - '@deepseek-ai/dsh-session': - specifier: workspace:^ - version: link:../session cordis: specifier: ^4.0.0-rc.6 version: 4.0.0-rc.6(@cordisjs/plugin-include@1.0.4)(@cordisjs/plugin-loader@1.0.0-rc.4) @@ -824,9 +821,6 @@ importers: '@deepseek-ai/dsh-scope': specifier: workspace:^ version: link:../../core/scope - '@deepseek-ai/dsh-session': - specifier: workspace:^ - version: link:../../core/session '@deepseek-ai/dsh-tools': specifier: workspace:^ version: link:../../core/tools @@ -1054,12 +1048,6 @@ importers: '@deepseek-ai/dsh-session': specifier: workspace:^ version: link:../../core/session - '@deepseek-ai/dsh-system-prompt': - specifier: workspace:^ - version: link:../../core/system-prompt - '@deepseek-ai/dsh-tools': - specifier: workspace:^ - version: link:../../core/tools cordis: specifier: ^4.0.0-rc.6 version: 4.0.0-rc.6(@cordisjs/plugin-include@1.0.4)(@cordisjs/plugin-loader@1.0.0-rc.4) diff --git a/scripts/gen-cordis-catalog.ts b/scripts/gen-cordis-catalog.ts index ae1fa8d170..68139e9ff8 100644 --- a/scripts/gen-cordis-catalog.ts +++ b/scripts/gen-cordis-catalog.ts @@ -81,7 +81,6 @@ const FENCE = 'ts cordis-catalog' */ export const LINK_MAP: Record = { Agent: 'core.md', - AgentRegistrationReservation: 'core.md', ContentBlock: 'core.md', Message: 'core.md', MessageSource: 'core.md', @@ -89,7 +88,6 @@ export const LINK_MAP: Record = { LlmCallConfig: 'core.md', SessionEvent: 'core.md', SessionStartSource: 'core.md', - SessionRegistrationReservation: 'session.md', StreamChunk: 'llm-streaming.md', TurnEndReason: 'session.md', ToolDefinition: 'tools.md', diff --git a/scripts/gen-doc-graphs.ts b/scripts/gen-doc-graphs.ts index c36a50c244..c4c8186f38 100644 --- a/scripts/gen-doc-graphs.ts +++ b/scripts/gen-doc-graphs.ts @@ -588,12 +588,12 @@ function collectEventRelations(): Map { if (method === 'on') { const event = eventArg(node.arguments, method) if (event) ensure(event).listeners.add(leaf) - } else if (method === 'emit' || method === 'parallel' || method === 'serial' || method === 'strictSerial' || method === 'waterfall') { + } else if (method === 'emit' || method === 'parallel' || method === 'serial' || method === 'waterfall') { const event = eventArg(node.arguments, method) if (event) { const relation = ensure(event) const methods = relation.dispatchers.get(leaf) ?? new Set() - methods.add(method === 'strictSerial' ? 'strictSerial (serial)' : method) + methods.add(method) relation.dispatchers.set(leaf, methods) } } diff --git a/scripts/type-equiv.manifest.json b/scripts/type-equiv.manifest.json index 62111d1551..a219ea1b1d 100644 --- a/scripts/type-equiv.manifest.json +++ b/scripts/type-equiv.manifest.json @@ -11,7 +11,6 @@ { "doc": "docs/core-data-structures/core.md", "symbol": "LlmCallConfig", "source": "packages/llm/llm/src/call-config.ts" }, { "doc": "docs/core-data-structures/core.md", "symbol": "SessionEvent", "source": "packages/core/session/src/types.ts" }, { "doc": "docs/core-data-structures/core.md", "symbol": "Agent", "source": "packages/core/agent/src/types.ts" }, - { "doc": "docs/core-data-structures/core.md", "symbol": "AgentRegistrationReservation", "source": "packages/core/agent/src/index.ts" }, { "doc": "docs/core-data-structures/core.md", "symbol": "HookContext", "source": "packages/core/agent/src/types.ts" }, { "doc": "docs/core-data-structures/core.md", "symbol": "PromptDecision", "source": "packages/core/agent/src/types.ts" }, { "doc": "docs/core-data-structures/core.md", "symbol": "ContinuationDecision", "source": "packages/core/agent/src/types.ts" }, @@ -23,8 +22,8 @@ { "doc": "docs/core-data-structures/scope.md", "symbol": "Scope", "source": "packages/core/scope/src/index.ts" }, { "doc": "docs/core-data-structures/system-prompt.md", "symbol": "AssembleContext", "source": "packages/core/system-prompt/src/index.ts" }, + { "doc": "docs/core-data-structures/system-prompt.md", "symbol": "PromptSection", "source": "packages/core/system-prompt/src/index.ts" }, { "doc": "docs/core-data-structures/system-prompt.md", "symbol": "ToolProviderResult", "source": "packages/core/system-prompt/src/index.ts" }, - { "doc": "docs/core-data-structures/system-prompt.md", "symbol": "PromptProtection", "source": "packages/core/system-prompt/src/index.ts" }, { "doc": "docs/core-data-structures/llm-streaming.md", "symbol": "StreamChunk", "source": "packages/llm/llm/src/types.ts" }, { "doc": "docs/core-data-structures/llm-streaming.md", "symbol": "TokenUsage", "source": "packages/llm/llm/src/types.ts" }, @@ -41,7 +40,6 @@ { "doc": "docs/core-data-structures/session.md", "symbol": "SurfaceOp", "source": "packages/core/session/src/types.ts" }, { "doc": "docs/core-data-structures/session.md", "symbol": "SurfaceIntent", "source": "packages/core/session/src/types.ts" }, { "doc": "docs/core-data-structures/session.md", "symbol": "SurfaceNode", "source": "packages/core/session/src/surface.ts" }, - { "doc": "docs/core-data-structures/session.md", "symbol": "SessionRegistrationReservation", "source": "packages/core/session/src/index.ts" }, { "doc": "docs/core-data-structures/persistence.md", "symbol": "SessionHeader", "source": "packages/core/session/src/types.ts" }, { "doc": "docs/core-data-structures/persistence.md", "symbol": "CreateSessionOptions", "source": "packages/core/session/src/types.ts" }, @@ -54,6 +52,7 @@ { "doc": "docs/core-data-structures/tools.md", "symbol": "ToolExecutionInput", "source": "packages/core/tools/src/index.ts" }, { "doc": "docs/core-data-structures/tools.md", "symbol": "ToolExecution", "source": "packages/core/tools/src/index.ts" }, { "doc": "docs/core-data-structures/tools.md", "symbol": "ToolGuard", "source": "packages/core/tools/src/index.ts" }, + { "doc": "docs/core-data-structures/tools.md", "symbol": "ToolRestriction", "source": "packages/core/tools/src/index.ts" }, { "doc": "docs/core-data-structures/tools.md", "symbol": "ToolExecutionResult", "source": "packages/core/tools/src/index.ts" }, { "doc": "docs/core-data-structures/tools.md", "symbol": "PreToolDecision", "source": "packages/core/tools/src/index.ts" }, { "doc": "docs/core-data-structures/tools.md", "symbol": "PostToolDecision", "source": "packages/core/tools/src/index.ts" },