fix: coordinate overlapping configured reloads
This commit is contained in:
@@ -131,7 +131,7 @@ export interface Config {
|
|||||||
|
|
||||||
Depends on: [`AgentOptions`](../packages/core/agent/src/index.ts) · [`SessionId`](../packages/core/session/src/index.ts)
|
Depends on: [`AgentOptions`](../packages/core/agent/src/index.ts) · [`SessionId`](../packages/core/session/src/index.ts)
|
||||||
|
|
||||||
Source: [`packages/core/agent-loop/src/index.ts:354`](../packages/core/agent-loop/src/index.ts)
|
Source: [`packages/core/agent-loop/src/index.ts:361`](../packages/core/agent-loop/src/index.ts)
|
||||||
|
|
||||||
## `@deepseek-ai/dsh-bash-local`
|
## `@deepseek-ai/dsh-bash-local`
|
||||||
|
|
||||||
|
|||||||
@@ -185,7 +185,7 @@ A declarative agent entry failed before it could publish a live agent. Consumers
|
|||||||
'agent-loop/config-start-failed'(sessionId: SessionId, error: unknown): void
|
'agent-loop/config-start-failed'(sessionId: SessionId, error: unknown): void
|
||||||
```
|
```
|
||||||
|
|
||||||
Source: [`packages/core/agent-loop/src/index.ts:349`](../../packages/core/agent-loop/src/index.ts)
|
Source: [`packages/core/agent-loop/src/index.ts:356`](../../packages/core/agent-loop/src/index.ts)
|
||||||
|
|
||||||
## `approval/*`
|
## `approval/*`
|
||||||
|
|
||||||
|
|||||||
@@ -19,7 +19,7 @@ async createAgent(ownerCtx: Context, options: CreateAgentOptions): Promise<Agent
|
|||||||
async resume(ownerCtx: Context, options: ResumeAgentOptions): Promise<AgentHandle>
|
async resume(ownerCtx: Context, options: ResumeAgentOptions): Promise<AgentHandle>
|
||||||
```
|
```
|
||||||
|
|
||||||
Source: [`packages/core/agent-loop/src/index.ts:369`](../../packages/core/agent-loop/src/index.ts)
|
Source: [`packages/core/agent-loop/src/index.ts:376`](../../packages/core/agent-loop/src/index.ts)
|
||||||
|
|
||||||
## `ctx.agents` — `AgentRegistry`
|
## `ctx.agents` — `AgentRegistry`
|
||||||
|
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ This matrix shows which packages dispatch each harness-owned event and which pac
|
|||||||
|
|
||||||
| Event | Mode | Declared in | Dispatchers | Listeners |
|
| Event | Mode | Declared in | Dispatchers | Listeners |
|
||||||
| --- | --- | --- | --- | --- |
|
| --- | --- | --- | --- | --- |
|
||||||
| `agent-loop/config-start-failed` | `emit` | [`packages/core/agent-loop/src/index.ts:349`](../packages/core/agent-loop/src/index.ts) | [`agent-loop`](../packages/core/agent-loop) (`events.dispatch`) | [`stdio-agent`](../packages/ui/stdio-agent) |
|
| `agent-loop/config-start-failed` | `emit` | [`packages/core/agent-loop/src/index.ts:356`](../packages/core/agent-loop/src/index.ts) | [`agent-loop`](../packages/core/agent-loop) (`events.dispatch`) | [`stdio-agent`](../packages/ui/stdio-agent) |
|
||||||
| `agent/created` | `emit` | [`packages/core/agent/src/types.ts:304`](../packages/core/agent/src/types.ts) | [`agent`](../packages/core/agent) (`events.dispatch`) | [`jsonrpc`](../packages/ui/jsonrpc), [`stdio-agent`](../packages/ui/stdio-agent) |
|
| `agent/created` | `emit` | [`packages/core/agent/src/types.ts:304`](../packages/core/agent/src/types.ts) | [`agent`](../packages/core/agent) (`events.dispatch`) | [`jsonrpc`](../packages/ui/jsonrpc), [`stdio-agent`](../packages/ui/stdio-agent) |
|
||||||
| `agent/disposed` | `emit` | [`packages/core/agent/src/types.ts:319`](../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:319`](../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:593`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`emit`) | - |
|
| `agent/error` | `emit` | [`packages/core/agent/src/types.ts:593`](../packages/core/agent/src/types.ts) | [`agent-loop`](../packages/core/agent-loop) (`emit`) | - |
|
||||||
|
|||||||
@@ -41,7 +41,7 @@ interface Config {
|
|||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
Agents listed in config are auto-created at startup. `cwd` seeds a fresh config-created session; a materialized exact `sessionId` remount and an explicit `resumeSessionId` keep the persisted session header. While the factory is active, a declarative lookup, resume, setup, or publication failure is contained, logged, and emitted as `agent-loop/config-start-failed(sessionId, error)` because no live `Agent` exists for an `agent/*` signal; cancellation caused by factory teardown is silent. 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 `assembleContextFor(agent)` — the helper couples the typed agent with its matching scope selector. These are 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.
|
Agents listed in config are auto-created at startup. `cwd` seeds a fresh config-created session; a materialized exact `sessionId` remount and an explicit `resumeSessionId` keep the persisted session header. An overlapping remount waits for an already-disposed same-id agent to finish detaching both registries before it inspects persistence, so asynchronous teardown cannot strand the configured identity. While the factory is active, a declarative lookup, resume, setup, or publication failure is contained, logged, and emitted as `agent-loop/config-start-failed(sessionId, error)` because no live `Agent` exists for an `agent/*` signal; cancellation caused by factory teardown is silent. 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 `assembleContextFor(agent)` — the helper couples the typed agent with its matching scope selector. These are 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
|
### Exported concrete class
|
||||||
|
|
||||||
|
|||||||
@@ -53,6 +53,7 @@ function renderThrown(value: unknown): string {
|
|||||||
/** Factory-level ownership of every preparing or live transaction. */
|
/** Factory-level ownership of every preparing or live transaction. */
|
||||||
class FactoryOwnership {
|
class FactoryOwnership {
|
||||||
private accepting = true
|
private accepting = true
|
||||||
|
private readonly inactive = Promise.withResolvers<void>()
|
||||||
private transactions = new Set<AgentCreationTransaction>()
|
private transactions = new Set<AgentCreationTransaction>()
|
||||||
private startupTasks = new Set<Promise<void>>()
|
private startupTasks = new Set<Promise<void>>()
|
||||||
|
|
||||||
@@ -74,8 +75,14 @@ class FactoryOwnership {
|
|||||||
void task.then(forget, forget)
|
void task.then(forget, forget)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/** Resolve `task`, or stop waiting when factory teardown begins. */
|
||||||
|
async waitWhileActive(task: Promise<void>): Promise<void> {
|
||||||
|
await Promise.race([task, this.inactive.promise])
|
||||||
|
}
|
||||||
|
|
||||||
async dispose(): Promise<void> {
|
async dispose(): Promise<void> {
|
||||||
this.accepting = false
|
this.accepting = false
|
||||||
|
this.inactive.resolve()
|
||||||
const reason = new Error('agent loop is not active')
|
const reason = new Error('agent loop is not active')
|
||||||
await Promise.all([
|
await Promise.all([
|
||||||
...[...this.transactions].map(transaction => transaction.disposeForFactory(reason)),
|
...[...this.transactions].map(transaction => transaction.disposeForFactory(reason)),
|
||||||
@@ -455,6 +462,8 @@ export class AgentLoop extends Service implements AgentFactory {
|
|||||||
agentOptions: AgentOptions,
|
agentOptions: AgentOptions,
|
||||||
meta: Pick<SessionHeader, 'cwd'>,
|
meta: Pick<SessionHeader, 'cwd'>,
|
||||||
): Promise<void> {
|
): Promise<void> {
|
||||||
|
await this.waitForDrainingConfiguredIdentity(ownerCtx, sessionId)
|
||||||
|
if (!this.ownership.isActive()) return
|
||||||
const exists = (await persistence.list()).some(header => header.id === sessionId)
|
const exists = (await persistence.list()).some(header => header.id === sessionId)
|
||||||
if (!this.ownership.isActive()) return
|
if (!this.ownership.isActive()) return
|
||||||
if (exists) {
|
if (exists) {
|
||||||
@@ -464,6 +473,32 @@ export class AgentLoop extends Service implements AgentFactory {
|
|||||||
this.create(sessionId, agentOptions, meta)
|
this.create(sessionId, agentOptions, meta)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/** Wait for an already-disposed same-id lifecycle to finish registry teardown. */
|
||||||
|
private async waitForDrainingConfiguredIdentity(ownerCtx: Context, sessionId: SessionId): Promise<void> {
|
||||||
|
const current = ownerCtx.agents.get(sessionId)
|
||||||
|
if (current?.status !== 'disposed') return
|
||||||
|
|
||||||
|
const released = Promise.withResolvers<void>()
|
||||||
|
const checkReleased = (): void => {
|
||||||
|
if (ownerCtx.agents.get(sessionId) === undefined && ownerCtx.sessions.get(sessionId) === undefined) {
|
||||||
|
released.resolve()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const disposeAgentListener = ownerCtx.on('agent/disposed', (agent) => {
|
||||||
|
if (agent.id === sessionId) checkReleased()
|
||||||
|
})
|
||||||
|
const disposeSessionListener = ownerCtx.on('session/disposed', (session) => {
|
||||||
|
if (session.id === sessionId) checkReleased()
|
||||||
|
})
|
||||||
|
try {
|
||||||
|
checkReleased()
|
||||||
|
await this.ownership.waitWhileActive(released.promise)
|
||||||
|
} finally {
|
||||||
|
disposeAgentListener()
|
||||||
|
disposeSessionListener()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Create an agent and session under one caller-supplied identity, owned by
|
* Create an agent and session under one caller-supplied identity, owned by
|
||||||
* the accessing fiber. Constructor-driven config calls mint a fresh combined
|
* the accessing fiber. Constructor-driven config calls mint a fresh combined
|
||||||
|
|||||||
@@ -92,6 +92,50 @@ describe('config-driven session id', () => {
|
|||||||
await ctx.fiber.dispose()
|
await ctx.fiber.dispose()
|
||||||
})
|
})
|
||||||
|
|
||||||
|
it('waits for a draining exact-id lifecycle during an overlapping reload', async () => {
|
||||||
|
const root = await mkdtemp(join(tmpdir(), 'dsh-cfg-exact-overlap-'))
|
||||||
|
dirs.push(root)
|
||||||
|
const ctx = await makeCoreContext()
|
||||||
|
await ctx.plugin(SessionPersistenceJsonl, { root })
|
||||||
|
const sessionId = SessionId('stdio-exact-overlap')
|
||||||
|
const config = { agents: [{ id: 'main', sessionId, model: 'mock' }] }
|
||||||
|
const firstLoop = await ctx.plugin(AgentLoop, config)
|
||||||
|
await expect.poll(() => ctx.agents.get(sessionId)).toBeDefined()
|
||||||
|
const first = ctx.agents.get(sessionId) as ReactLoopAgent
|
||||||
|
|
||||||
|
const flushGate = Promise.withResolvers<undefined>()
|
||||||
|
let flushStarted = false
|
||||||
|
ctx.on('session/flush', (session) => {
|
||||||
|
if (session !== first.session) return
|
||||||
|
flushStarted = true
|
||||||
|
return flushGate.promise
|
||||||
|
})
|
||||||
|
first.inject([{ type: 'text', text: 'persist before replacement' }], {
|
||||||
|
source: { kind: 'plugin', plugin: 'test' },
|
||||||
|
})
|
||||||
|
expect(flushStarted).toBe(true)
|
||||||
|
|
||||||
|
const firstDisposal = firstLoop.dispose()
|
||||||
|
await expect.poll(() => first.status).toBe('disposed')
|
||||||
|
const failures: unknown[] = []
|
||||||
|
ctx.on('agent-loop/config-start-failed', (_id, error) => { failures.push(error) })
|
||||||
|
const secondLoop = await ctx.plugin(AgentLoop, config)
|
||||||
|
await new Promise(resolve => setTimeout(resolve, 0))
|
||||||
|
expect(ctx.agents.get(sessionId)).toBe(first)
|
||||||
|
expect(failures).toEqual([])
|
||||||
|
|
||||||
|
flushGate.resolve(undefined)
|
||||||
|
await firstDisposal
|
||||||
|
await expect.poll(() => ctx.agents.get(sessionId)).toBeDefined()
|
||||||
|
const second = ctx.agents.get(sessionId) as ReactLoopAgent
|
||||||
|
expect(second).not.toBe(first)
|
||||||
|
expect(JSON.stringify(second.session.deriveMessages())).toContain('persist before replacement')
|
||||||
|
expect(failures).toEqual([])
|
||||||
|
|
||||||
|
await secondLoop.dispose()
|
||||||
|
await ctx.fiber.dispose()
|
||||||
|
})
|
||||||
|
|
||||||
it('contains an exact-id persistence lookup failure', async () => {
|
it('contains an exact-id persistence lookup failure', async () => {
|
||||||
const root = await mkdtemp(join(tmpdir(), 'dsh-cfg-exact-failure-'))
|
const root = await mkdtemp(join(tmpdir(), 'dsh-cfg-exact-failure-'))
|
||||||
dirs.push(root)
|
dirs.push(root)
|
||||||
|
|||||||
@@ -67,6 +67,15 @@ function isTTYPair(input: Readable, output: Writable): boolean {
|
|||||||
return Boolean((input as { isTTY?: boolean }).isTTY && (output as { isTTY?: boolean }).isTTY)
|
return Boolean((input as { isTTY?: boolean }).isTTY && (output as { isTTY?: boolean }).isTTY)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/** Render an arbitrary failure without allowing hostile coercion to escape the UI boundary. */
|
||||||
|
function renderThrown(value: unknown): string {
|
||||||
|
try {
|
||||||
|
return String(value)
|
||||||
|
} catch {
|
||||||
|
return '<unrenderable thrown value>'
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
interface PendingQuestion {
|
interface PendingQuestion {
|
||||||
request: AskUserQuestionRequest
|
request: AskUserQuestionRequest
|
||||||
questionIndex: number
|
questionIndex: number
|
||||||
@@ -230,7 +239,7 @@ export function createStdioChat(ctx: Context, config: Config, runtime: StdioRunt
|
|||||||
queuedInput.length = 0
|
queuedInput.length = 0
|
||||||
submittedWork = sawRunning
|
submittedWork = sawRunning
|
||||||
if (dropped > 0) {
|
if (dropped > 0) {
|
||||||
ctx.logger.error(`ui-stdio: main agent failed to start; dropped queued stdin (${dropped} line(s)): ${String(error)}`)
|
ctx.logger.error(`ui-stdio: main agent failed to start; dropped queued stdin (${dropped} line(s)): ${renderThrown(error)}`)
|
||||||
}
|
}
|
||||||
maybeExit()
|
maybeExit()
|
||||||
})
|
})
|
||||||
@@ -390,7 +399,7 @@ export function createStdioChat(ctx: Context, config: Config, runtime: StdioRunt
|
|||||||
const text = line.trim()
|
const text = line.trim()
|
||||||
if (!text) return
|
if (!text) return
|
||||||
if (failedStartup !== undefined) {
|
if (failedStartup !== undefined) {
|
||||||
ctx.logger.error(`ui-stdio: main agent failed to start; dropped queued stdin (1 line(s)): ${String(failedStartup.error)}`)
|
ctx.logger.error(`ui-stdio: main agent failed to start; dropped queued stdin (1 line(s)): ${renderThrown(failedStartup.error)}`)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
const agent = target
|
const agent = target
|
||||||
|
|||||||
@@ -83,6 +83,10 @@ function chunkEvent(chunk: StreamChunk): SessionEvent {
|
|||||||
|
|
||||||
const CONFIG: Config = { welcome: 'hi there', sessionId: 'main' }
|
const CONFIG: Config = { welcome: 'hi there', sessionId: 'main' }
|
||||||
|
|
||||||
|
function unrenderableFailure(): unknown {
|
||||||
|
return { [Symbol.toPrimitive](): never { throw new Error('coercion escaped') } }
|
||||||
|
}
|
||||||
|
|
||||||
async function setup(config: Config = CONFIG, runtimeOver: Partial<StdioRuntime> = {}) {
|
async function setup(config: Config = CONFIG, runtimeOver: Partial<StdioRuntime> = {}) {
|
||||||
const ctx = new Context()
|
const ctx = new Context()
|
||||||
await ctx.plugin(AgentRegistry)
|
await ctx.plugin(AgentRegistry)
|
||||||
@@ -753,14 +757,14 @@ describe('createStdioChat input', () => {
|
|||||||
it('drops later input after the configured startup fails', async () => {
|
it('drops later input after the configured startup fails', async () => {
|
||||||
const { ctx, input } = await setup()
|
const { ctx, input } = await setup()
|
||||||
const error = vi.spyOn(ctx.logger, 'error').mockImplementation(() => {})
|
const error = vi.spyOn(ctx.logger, 'error').mockImplementation(() => {})
|
||||||
const failure = new Error('persisted session is corrupt')
|
const failure = unrenderableFailure()
|
||||||
ctx.emit('agent-loop/config-start-failed', SessionId('main'), failure)
|
ctx.emit('agent-loop/config-start-failed', SessionId('main'), failure)
|
||||||
|
|
||||||
input.feed('cannot run')
|
input.feed('cannot run')
|
||||||
await new Promise(r => setImmediate(r))
|
await new Promise(r => setImmediate(r))
|
||||||
|
|
||||||
expect(error).toHaveBeenCalledWith(
|
expect(error).toHaveBeenCalledWith(
|
||||||
'ui-stdio: main agent failed to start; dropped queued stdin (1 line(s)): Error: persisted session is corrupt',
|
'ui-stdio: main agent failed to start; dropped queued stdin (1 line(s)): <unrenderable thrown value>',
|
||||||
)
|
)
|
||||||
})
|
})
|
||||||
|
|
||||||
@@ -844,11 +848,11 @@ describe('createStdioChat EOF exit', () => {
|
|||||||
await flushExit()
|
await flushExit()
|
||||||
expect(exit).not.toHaveBeenCalled()
|
expect(exit).not.toHaveBeenCalled()
|
||||||
|
|
||||||
ctx.emit('agent-loop/config-start-failed', SessionId('main'), new Error('missing persisted session'))
|
ctx.emit('agent-loop/config-start-failed', SessionId('main'), unrenderableFailure())
|
||||||
await flushExit()
|
await flushExit()
|
||||||
|
|
||||||
expect(error).toHaveBeenCalledWith(
|
expect(error).toHaveBeenCalledWith(
|
||||||
'ui-stdio: main agent failed to start; dropped queued stdin (1 line(s)): Error: missing persisted session',
|
'ui-stdio: main agent failed to start; dropped queued stdin (1 line(s)): <unrenderable thrown value>',
|
||||||
)
|
)
|
||||||
expect(exit).toHaveBeenCalledWith(0)
|
expect(exit).toHaveBeenCalledWith(0)
|
||||||
})
|
})
|
||||||
|
|||||||
Reference in New Issue
Block a user