137 lines
4.8 KiB
TypeScript
137 lines
4.8 KiB
TypeScript
/** Covers fail-closed per-call classification and model-schema isolation. */
|
|
|
|
import { describe, expect, expectTypeOf, it } from 'vitest'
|
|
import { Context } from 'cordis'
|
|
import { CallId } from '@deepseek-ai/dsh-llm'
|
|
import SystemPrompt from '@deepseek-ai/dsh-system-prompt'
|
|
import ToolRegistry, {
|
|
defineTool,
|
|
type ToolDefinition,
|
|
type ToolExecutionInput,
|
|
type ToolExecutionMode,
|
|
} from '@deepseek-ai/dsh-tools'
|
|
|
|
async function setup() {
|
|
const ctx = new Context()
|
|
await ctx.plugin(SystemPrompt)
|
|
await ctx.plugin(ToolRegistry)
|
|
return ctx
|
|
}
|
|
|
|
function exec(name: string, args: unknown): ToolExecutionInput {
|
|
return { callId: CallId('c1'), name, arguments: args }
|
|
}
|
|
|
|
describe('ToolRegistry.executionMode', () => {
|
|
it('returns parallel only for an explicit true classifier', async () => {
|
|
const ctx = await setup()
|
|
ctx.tools.register(defineTool({
|
|
name: 'safe',
|
|
description: 'parallel-safe',
|
|
parameters: {},
|
|
isConcurrencySafe: () => true,
|
|
async execute() { return [] },
|
|
}))
|
|
expect(ctx.tools.executionMode(exec('safe', {}))).toEqual({ kind: 'parallel' })
|
|
})
|
|
|
|
it('defaults to exclusive for a tool with no isConcurrencySafe declaration', async () => {
|
|
const ctx = await setup()
|
|
ctx.tools.register(defineTool({
|
|
name: 'plain',
|
|
description: 'no declaration',
|
|
parameters: {},
|
|
async execute() { return [] },
|
|
}))
|
|
expect(ctx.tools.executionMode(exec('plain', {}))).toEqual({ kind: 'exclusive' })
|
|
})
|
|
|
|
it('returns exclusive for an unknown tool', async () => {
|
|
const ctx = await setup()
|
|
expect(ctx.tools.executionMode(exec('nonexistent', {}))).toEqual({ kind: 'exclusive' })
|
|
})
|
|
|
|
it('returns exclusive when the classifier returns false for these args', async () => {
|
|
const ctx = await setup()
|
|
ctx.tools.register(defineTool({
|
|
name: 'rw',
|
|
description: 'read or write',
|
|
parameters: { mode: { type: 'string', required: true } },
|
|
isConcurrencySafe: args => args.mode === 'read',
|
|
async execute() { return [] },
|
|
}))
|
|
expect(ctx.tools.executionMode(exec('rw', { mode: 'read' }))).toEqual({ kind: 'parallel' })
|
|
expect(ctx.tools.executionMode(exec('rw', { mode: 'write' }))).toEqual({ kind: 'exclusive' })
|
|
})
|
|
|
|
it('classifies invalid defineTool arguments as exclusive without throwing', async () => {
|
|
const ctx = await setup()
|
|
ctx.tools.register(defineTool({
|
|
name: 'needs-mode',
|
|
description: 'requires mode',
|
|
parameters: { mode: { type: 'string', required: true } },
|
|
isConcurrencySafe: () => true,
|
|
async execute() { return [] },
|
|
}))
|
|
expect(ctx.tools.executionMode(exec('needs-mode', {}))).toEqual({ kind: 'exclusive' })
|
|
})
|
|
|
|
it('treats a throwing raw classifier as exclusive', async () => {
|
|
const ctx = await setup()
|
|
const raw: ToolDefinition = {
|
|
name: 'thrower',
|
|
description: 'classifier throws',
|
|
parameters: { type: 'object', properties: {} },
|
|
isConcurrencySafe() { throw new Error('boom') },
|
|
async execute() { return [] },
|
|
}
|
|
ctx.tools.register(raw)
|
|
expect(ctx.tools.executionMode(exec('thrower', {}))).toEqual({ kind: 'exclusive' })
|
|
})
|
|
|
|
it('treats a truthy non-boolean raw result as exclusive', async () => {
|
|
const ctx = await setup()
|
|
const raw = {
|
|
name: 'truthy',
|
|
description: 'classifier returns a truthy string',
|
|
parameters: { type: 'object', properties: {} },
|
|
isConcurrencySafe() { return 'yes' },
|
|
async execute() { return [] },
|
|
} as unknown as ToolDefinition
|
|
ctx.tools.register(raw)
|
|
expect(ctx.tools.executionMode(exec('truthy', {}))).toEqual({ kind: 'exclusive' })
|
|
})
|
|
|
|
it('passes parsed arguments directly to a raw definition', async () => {
|
|
const ctx = await setup()
|
|
let seen: unknown
|
|
ctx.tools.register({
|
|
name: 'raw-safe',
|
|
description: 'raw',
|
|
parameters: { type: 'object', properties: {} },
|
|
isConcurrencySafe(args) { seen = args; return true },
|
|
async execute() { return [] },
|
|
})
|
|
expect(ctx.tools.executionMode(exec('raw-safe', { anything: 1 }))).toEqual({ kind: 'parallel' })
|
|
expect(seen).toEqual({ anything: 1 })
|
|
})
|
|
|
|
it('isConcurrencySafe never reaches the model-facing schemas() projection', async () => {
|
|
const ctx = await setup()
|
|
ctx.tools.register(defineTool({
|
|
name: 'safe',
|
|
description: 'parallel-safe',
|
|
parameters: { x: { type: 'string', required: true } },
|
|
isConcurrencySafe: () => true,
|
|
async execute() { return [] },
|
|
}))
|
|
const schema = ctx.tools.schemas()[0] as unknown as Record<string, unknown>
|
|
expect(Object.keys(schema).sort()).toEqual(['description', 'name', 'parameters'])
|
|
expect(schema.isConcurrencySafe).toBeUndefined()
|
|
})
|
|
|
|
it('ToolExecutionMode is the object-tagged union', () => {
|
|
expectTypeOf<ToolExecutionMode>().toEqualTypeOf<{ kind: 'parallel' } | { kind: 'exclusive' }>()
|
|
})
|
|
})
|