refactor(subagent): retire standalone mock package
This commit is contained in:
104
packages/subagent/tool-subagent/tests/scripted-provider.ts
Normal file
104
packages/subagent/tool-subagent/tests/scripted-provider.ts
Normal file
@@ -0,0 +1,104 @@
|
||||
/** Package-local scripted child boundary for deterministic tool-subagent tests. */
|
||||
|
||||
import type { Context } from 'cordis'
|
||||
import type { ContentBlock } from '@deepseek-ai/dsh-llm'
|
||||
import { SessionId } from '@deepseek-ai/dsh-session'
|
||||
import type {
|
||||
SubagentCapabilities,
|
||||
SubagentProvider,
|
||||
SubagentResult,
|
||||
SubagentRun,
|
||||
SubagentStartRequest,
|
||||
SubagentStopReason,
|
||||
} from '@deepseek-ai/dsh-subagent'
|
||||
|
||||
const DEFAULT_CAPABILITIES: SubagentCapabilities = {
|
||||
outputSchema: true,
|
||||
depthLimit: true,
|
||||
toolFilter: true,
|
||||
persona: true,
|
||||
}
|
||||
|
||||
/** Options for one scripted provider fixture. */
|
||||
export interface Config {
|
||||
/** Registry name to register under. */
|
||||
name: string
|
||||
/** Final text returned by the scripted child. */
|
||||
reply?: string
|
||||
/** Terminal result reason. */
|
||||
stopReason?: SubagentStopReason
|
||||
/** Start-time features advertised by the provider. */
|
||||
capabilities?: Partial<SubagentCapabilities>
|
||||
/** Whether tool descriptions say the child inherits completed turns. */
|
||||
inheritsParentContext?: boolean
|
||||
/** Structured value returned when the request asks for one. */
|
||||
structured?: unknown
|
||||
}
|
||||
|
||||
/** Scripted provider whose result aborts if its signal or disposer wins first. */
|
||||
class ScriptedSubagentProvider implements SubagentProvider {
|
||||
readonly capabilities: SubagentCapabilities
|
||||
readonly inheritsParentContext: boolean
|
||||
|
||||
constructor(
|
||||
readonly name: string,
|
||||
private readonly config: Config,
|
||||
) {
|
||||
this.capabilities = { ...DEFAULT_CAPABILITIES, ...config.capabilities }
|
||||
this.inheritsParentContext = config.inheritsParentContext ?? false
|
||||
}
|
||||
|
||||
async start(request: SubagentStartRequest): Promise<SubagentRun> {
|
||||
if (request.signal.aborted) throw new Error('scripted subagent start aborted before publication')
|
||||
const reply = this.config.reply ?? 'scripted subagent reply'
|
||||
const output: ContentBlock[] = [{ type: 'text', text: reply }]
|
||||
const wantsStructured = request.outputSchema !== undefined && this.capabilities.outputSchema
|
||||
const stopReason = this.config.stopReason ?? 'completed'
|
||||
const state = { cancelled: false }
|
||||
const onAbort = (): void => { state.cancelled = true }
|
||||
request.signal.addEventListener('abort', onAbort, { once: true })
|
||||
await Promise.resolve()
|
||||
if (state.cancelled) {
|
||||
request.signal.removeEventListener('abort', onAbort)
|
||||
throw new Error('scripted subagent start aborted before publication')
|
||||
}
|
||||
|
||||
const resultFor = (): SubagentResult => ({
|
||||
output,
|
||||
...wantsStructured ? { structured: this.config.structured ?? { reply } } : {},
|
||||
stopReason: state.cancelled ? 'aborted' : stopReason,
|
||||
})
|
||||
const result = new Promise<SubagentResult>((resolve) => {
|
||||
setTimeout(() => { resolve(resultFor()) }, 0)
|
||||
}).finally(() => {
|
||||
request.signal.removeEventListener('abort', onAbort)
|
||||
})
|
||||
|
||||
return {
|
||||
id: SessionId(`scripted-subagent:${this.name}:${request.parent.id}`),
|
||||
localAgent: undefined,
|
||||
result,
|
||||
dispose(): Promise<void> {
|
||||
state.cancelled = true
|
||||
request.signal.removeEventListener('abort', onAbort)
|
||||
return Promise.resolve()
|
||||
},
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Mount one scripted provider through an effect-scoped local plugin.
|
||||
* @param ctx - context carrying the real subagent registry.
|
||||
* @param config - scripted provider identity and outcome.
|
||||
* @returns the fixture plugin's disposable fiber.
|
||||
*/
|
||||
export function mountScriptedProvider(ctx: Context, config: Config) {
|
||||
return ctx.plugin({
|
||||
name: 'scripted-subagent-provider',
|
||||
inject: ['subagents'],
|
||||
apply(pluginCtx: Context): void {
|
||||
pluginCtx.subagents.registerProvider(new ScriptedSubagentProvider(config.name, config))
|
||||
},
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user