111 lines
4.5 KiB
TypeScript
111 lines
4.5 KiB
TypeScript
import { mkdtemp, rm, writeFile } from 'node:fs/promises'
|
|
import { tmpdir } from 'node:os'
|
|
import { join } from 'node:path'
|
|
import { pathToFileURL } from 'node:url'
|
|
import { afterEach, describe, expect, it } from 'vitest'
|
|
import { Context } from 'cordis'
|
|
import Loader from '@cordisjs/plugin-loader'
|
|
import Include from '@cordisjs/plugin-include'
|
|
import AgentRegistry, { Inbox } from '@deepseek-ai/dsh-agent'
|
|
import type { Agent, AgentStatus } from '@deepseek-ai/dsh-agent'
|
|
import CommandService from '@deepseek-ai/dsh-commands'
|
|
import SessionStore, { SessionId } from '@deepseek-ai/dsh-session'
|
|
import * as CommandFeedback from '@deepseek-ai/dsh-command-feedback'
|
|
|
|
let root: string | undefined
|
|
let context: Context | undefined
|
|
|
|
afterEach(async () => {
|
|
await context?.fiber.dispose()
|
|
context = undefined
|
|
if (root !== undefined) await rm(root, { recursive: true, force: true })
|
|
root = undefined
|
|
})
|
|
|
|
/** Register one idle agent over a store-owned session, as an app's spine does. */
|
|
function agent(ctx: Context): Agent {
|
|
const scope = ctx.plugin(() => {})
|
|
const id = SessionId('feedback-loader-agent')
|
|
const session = ctx.sessions.create(id)
|
|
const inbox = new Inbox(session, { inserted: () => {}, discarded: () => {}, claimed: () => {} })
|
|
let status: AgentStatus = 'idle'
|
|
const value: Agent = {
|
|
id,
|
|
options: {},
|
|
session,
|
|
inbox,
|
|
ctx: scope.ctx,
|
|
get status() { return status },
|
|
send: () => {},
|
|
followup: () => {},
|
|
steer: () => {},
|
|
inject: () => {},
|
|
cancel() { status = 'idle' },
|
|
runMaintenance: task => task(new AbortController().signal),
|
|
whenIdle: () => Promise.resolve(),
|
|
}
|
|
ctx.agents.register(value)
|
|
return value
|
|
}
|
|
|
|
describe('/feedback real Loader composition through cordis.yml', () => {
|
|
it('boots cordis.yml and records feedback without model-visible output', async () => {
|
|
root = await mkdtemp(join(tmpdir(), 'dsh-command-feedback-loader-'))
|
|
const configPath = join(root, 'cordis.yml')
|
|
await writeFile(configPath, [
|
|
"- name: '@deepseek-ai/dsh-agent'",
|
|
"- name: '@deepseek-ai/dsh-session'",
|
|
"- name: '@deepseek-ai/dsh-commands'",
|
|
"- name: '@deepseek-ai/dsh-command-feedback'",
|
|
'',
|
|
].join('\n'))
|
|
|
|
context = new Context()
|
|
context.baseUrl = pathToFileURL(root).href + '/'
|
|
await context.plugin(Loader)
|
|
context.loader.builtins.include = Include
|
|
const modules = new Map<string, unknown>([
|
|
['@deepseek-ai/dsh-agent', AgentRegistry],
|
|
['@deepseek-ai/dsh-session', SessionStore],
|
|
['@deepseek-ai/dsh-commands', CommandService],
|
|
['@deepseek-ai/dsh-command-feedback', CommandFeedback],
|
|
])
|
|
context.loader.internal = {
|
|
version: 'v2',
|
|
async import(specifier: string) {
|
|
if (!modules.has(specifier)) throw new Error(`unexpected Loader import: ${specifier}`)
|
|
return modules.get(specifier)
|
|
},
|
|
} as unknown as NonNullable<typeof context.loader.internal>
|
|
await context.loader.create({ name: 'cordis:include', config: { path: pathToFileURL(configPath).href } })
|
|
await context.loader.await()
|
|
|
|
const owner = agent(context)
|
|
const signal = new AbortController().signal
|
|
|
|
// Discoverable through the composed registry, as a UI adapter finds it.
|
|
expect(context.commands.list(owner).map(command => command.name)).toContain('feedback')
|
|
|
|
const accepted = await context.commands.execute(owner, '/feedback the diff view is unreadable', signal)
|
|
expect(accepted?.result).toEqual({ kind: 'success', text: 'Feedback recorded.' })
|
|
const rejected = await context.commands.execute(owner, '/feedback', signal)
|
|
expect(rejected?.result).toEqual({
|
|
kind: 'error',
|
|
text: 'Feedback text is required. Usage: /feedback <text>',
|
|
})
|
|
|
|
// The domain event owns the payload; generic command bookkeeping omits it.
|
|
expect(owner.session.events.map(event => event.type))
|
|
.toEqual(['command/run', 'feedback/record', 'command/done', 'command/run', 'command/done'])
|
|
const run = owner.session.events.find(event => event.type === 'command/run')
|
|
expect(run?.type === 'command/run' && Object.hasOwn(run.data, 'args')).toBe(false)
|
|
const feedback = owner.session.events.find(event => event.type === 'feedback/record')
|
|
expect(feedback?.type === 'feedback/record' && feedback.data.text).toBe('the diff view is unreadable')
|
|
expect(JSON.stringify(owner.session.events).match(/the diff view is unreadable/gu)).toHaveLength(1)
|
|
|
|
// Nothing reached the model.
|
|
expect(owner.session.deriveMessages()).toEqual([])
|
|
expect(owner.session.surface.nodes).toEqual([])
|
|
})
|
|
})
|