fix(web): make plan transitions admission-safe
This commit is contained in:
@@ -4,7 +4,9 @@ import ToolRegistry from '@deepseek-ai/dsh-tools'
|
||||
import SystemPrompt from '@deepseek-ai/dsh-system-prompt'
|
||||
import UserInteractionService from '@deepseek-ai/dsh-user-interaction'
|
||||
import { EXIT_PLAN_MODE } from '@deepseek-ai/dsh-plan-mode'
|
||||
import { WEB_PLAN_SECTION, apply, inject } from '../src/index.ts'
|
||||
import type { Agent } from '@deepseek-ai/dsh-agent'
|
||||
import type { SessionEvent } from '@deepseek-ai/dsh-session'
|
||||
import { WEB_DEFAULT_SECTION, WEB_PLAN_SECTION, apply, inject } from '../src/index.ts'
|
||||
|
||||
let ctx: Context | undefined
|
||||
|
||||
@@ -27,13 +29,45 @@ describe('ui-plan node plugin', () => {
|
||||
expect(WEB_PLAN_SECTION).toContain('Stay in plan mode until exit_plan_mode succeeds')
|
||||
expect(WEB_PLAN_SECTION).toContain('Do not edit or write files')
|
||||
expect(WEB_PLAN_SECTION).toContain('Make exit_plan_mode the only and final tool call')
|
||||
expect(WEB_DEFAULT_SECTION).toContain('default mode, not plan mode')
|
||||
expect(WEB_DEFAULT_SECTION).toContain('Do not call exit_plan_mode in default mode')
|
||||
expect((await ctx.systemPrompt.assemble()).sections)
|
||||
.toEqual(expect.arrayContaining([expect.objectContaining({ name: 'plan:policy', text: '' })]))
|
||||
.toEqual(expect.arrayContaining([
|
||||
expect.objectContaining({ name: 'plan:default-policy', text: '' }),
|
||||
expect.objectContaining({ name: 'plan:policy', text: '' }),
|
||||
]))
|
||||
|
||||
await feature.dispose()
|
||||
expect(ctx.get('planMode')).toBeUndefined()
|
||||
expect(ctx.tools.get(EXIT_PLAN_MODE)).toBeUndefined()
|
||||
expect((await ctx.systemPrompt.assemble()).sections.map(section => section.name))
|
||||
.not.toContain('plan:policy')
|
||||
.toEqual(expect.not.arrayContaining(['plan:default-policy', 'plan:policy']))
|
||||
})
|
||||
|
||||
it('states the exact Web collaboration mode at every agent assembly', async () => {
|
||||
ctx = new Context()
|
||||
await ctx.plugin(SystemPrompt)
|
||||
await ctx.plugin(ToolRegistry)
|
||||
await ctx.plugin(UserInteractionService)
|
||||
const feature = ctx.plugin({ inject: [...inject], apply })
|
||||
await feature.await()
|
||||
|
||||
const events: SessionEvent[] = []
|
||||
const agent = { session: { events } } as unknown as Agent
|
||||
let sections = (await ctx.systemPrompt.assemble({ agent })).sections
|
||||
expect(sections.find(section => section.name === 'plan:default-policy')?.text)
|
||||
.toBe(WEB_DEFAULT_SECTION)
|
||||
expect(sections.find(section => section.name === 'plan:policy')?.text).toBe('')
|
||||
|
||||
events.push({
|
||||
type: 'plan/mode',
|
||||
seq: 0,
|
||||
time: 1,
|
||||
data: { active: true },
|
||||
})
|
||||
sections = (await ctx.systemPrompt.assemble({ agent })).sections
|
||||
expect(sections.find(section => section.name === 'plan:default-policy')?.text).toBe('')
|
||||
expect(sections.find(section => section.name === 'plan:policy')?.text)
|
||||
.toBe(WEB_PLAN_SECTION)
|
||||
})
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user