Tianyi Cui
3b63bcbeec
CI's full-tree coverage flagged three untaken paths this PR introduced: - shapeDispatchLog's catch (a throwing tools/code-dispatch-log listener must be contained — the settle event logs the unshaped content); - the spill listener's flatten-decline arm (non-text sub-result content passes through unchanged); - the generated scope-key extractor row for tools/code-dispatch-log (registered in the scope invariant matrix like the other tools events).
98 lines
5.2 KiB
TypeScript
98 lines
5.2 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
import { Context } from 'cordis'
|
|
import type { Events } from 'cordis'
|
|
import { AgentMessageId, type Agent } from '@deepseek-ai/dsh-agent'
|
|
import { scopeTarget } from '@deepseek-ai/dsh-scope'
|
|
import * as ScopeInvariant from '@deepseek-ai/dsh-scope/invariant'
|
|
import InvariantService from '@deepseek-ai/dsh-invariants'
|
|
|
|
async function setup(): Promise<Context> {
|
|
const ctx = new Context()
|
|
await ctx.plugin(InvariantService)
|
|
await ctx.plugin(ScopeInvariant)
|
|
return ctx
|
|
}
|
|
|
|
function emit(ctx: Context, receiver: object | undefined, event: string, args: unknown[]): void {
|
|
const dispatch = ctx.emit.bind(ctx) as (...values: unknown[]) => void
|
|
if (receiver === undefined) dispatch(event, ...args)
|
|
else dispatch(receiver, event, ...args)
|
|
}
|
|
|
|
describe('scoped-dispatch invariants', () => {
|
|
type AgentEventName = Extract<keyof Events, `agent/${string}`>
|
|
type EventArgs<K extends keyof Events> = Events[K] extends (...args: infer Args) => unknown ? Args : never
|
|
|
|
it('ignores ordinary events and rejects a scoped dispatch without a carrier', async () => {
|
|
const ctx = await setup()
|
|
expect(() => { emit(ctx, undefined, 'ordinary/event', []) }).not.toThrow()
|
|
const agent = { id: 'a1' }
|
|
expect(() => { emit(ctx, undefined, 'agent/error', [agent, 1, 0, new Error('x')]) })
|
|
.toThrow(/dispatched without a scope carrier/)
|
|
})
|
|
|
|
it('checks every generated subject resolver against the carrier key', async () => {
|
|
const ctx = await setup()
|
|
const agent = { id: 'a1' } as unknown as Agent
|
|
const other = { id: 'a2' } as unknown as Agent
|
|
const signal = new AbortController().signal
|
|
const config = { provider: 'p', model: 'm' }
|
|
const message = { role: 'assistant' as const, content: [] }
|
|
const agentRows = {
|
|
'agent/created': [agent],
|
|
'agent/disposed': [agent],
|
|
'agent/status': [agent, 'idle'],
|
|
'agent/inbox/enqueue': [agent, { id: AgentMessageId('m'), content: [], source: { kind: 'user' }, contexts: [], steering: false, wakeup: true }],
|
|
'agent/inbox/dequeue': [agent, { id: AgentMessageId('m'), content: [], source: { kind: 'user' }, contexts: [], steering: false, wakeup: true }],
|
|
'agent/inbox/discard': [agent, []],
|
|
'agent/cancel-requested': [agent, { kind: 'user' }],
|
|
'agent/session-start': [agent, 'startup'],
|
|
'agent/pre-step': [agent, 1, 1, signal],
|
|
'agent/post-step': [agent, 1, 1, signal],
|
|
'agent/prompt-submit': [agent, [], { kind: 'user' }, signal, () => Promise.resolve({ kind: 'allow' })],
|
|
'agent/request': [agent, 1, 1, config, signal, () => Promise.resolve(config)],
|
|
'agent/request-error': [agent, 1, 1, new Error('request failed'), { message: 'request failed', code: 'UNKNOWN' }, [], signal, () => Promise.resolve({ action: 'fail' })],
|
|
'agent/session-prefix': [agent, [], signal, () => Promise.resolve([])],
|
|
'agent/step-result': [agent, 1, 1, message, signal, () => Promise.resolve(message)],
|
|
'agent/turn-continuation': [agent, 1, { action: 'stop' }, signal, () => Promise.resolve({ action: 'stop' })],
|
|
'agent/turn-stop': [agent, 1, signal],
|
|
'agent/error': [agent, 1, 0, new Error('x')],
|
|
} satisfies { [K in AgentEventName]: EventArgs<K> }
|
|
const rows: Array<[string, unknown[]]> = [
|
|
...Object.entries(agentRows),
|
|
['approval/request', [{ agent, toolName: 'echo' }, () => Promise.resolve('unavailable')]],
|
|
['goal/changed', [agent, { operation: 'create', ref: { id: 'goal-a', revision: 1 } }]],
|
|
['system-prompt/assemble', [[], { scope: agent }]],
|
|
['tools/code-dispatch-log', [{ exec: { callId: 'c', name: 't', arguments: {} }, agent, subCallId: 'c:code:1', name: 't', isError: false, content: [] }, () => Promise.resolve([])]],
|
|
['tools/execute', [{ callId: 'c', name: 't', arguments: {}, agent }, () => Promise.resolve({ content: [], isError: false })]],
|
|
['tools/post-execute', [{ callId: 'c', name: 't', arguments: {}, agent }, { content: [], isError: false }, () => Promise.resolve({ kind: 'accept' })]],
|
|
['tools/pre-execute', [{ callId: 'c', name: 't', arguments: {}, agent }, () => Promise.resolve({ kind: 'allow' })]],
|
|
['tools/result', [{ callId: 'c', name: 't', arguments: {}, agent }, { content: [], isError: false }]],
|
|
]
|
|
|
|
for (const [event, args] of rows) {
|
|
expect(() => { emit(ctx, scopeTarget(agent, agent), event, args) }, `${event} matching`).not.toThrow()
|
|
expect(() => { emit(ctx, scopeTarget(agent, other), event, args) }, `${event} mismatched`)
|
|
.toThrow(/DIFFERENT subject/)
|
|
}
|
|
})
|
|
|
|
it('requires carriers for generated presence-only scoped events without comparing a payload subject', async () => {
|
|
const ctx = await setup()
|
|
const agent = { id: 'a1' }
|
|
const rows: Array<[string, unknown[]]> = [
|
|
['session/created', [{}]],
|
|
['session/disposed', [{}]],
|
|
['session/event', [{}, {}]],
|
|
['session/flush', [{}]],
|
|
['subagent/end', [{}]],
|
|
['subagent/start', [{}]],
|
|
]
|
|
for (const [event, args] of rows) {
|
|
expect(() => { emit(ctx, scopeTarget(agent, agent), event, args) }, `${event} carrier`).not.toThrow()
|
|
expect(() => { emit(ctx, undefined, event, args) }, `${event} no carrier`)
|
|
.toThrow(/dispatched without a scope carrier/)
|
|
}
|
|
})
|
|
})
|