loader-composition.spec.ts 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. import { mkdtemp, rm, writeFile } from 'node:fs/promises'
  2. import { tmpdir } from 'node:os'
  3. import { join } from 'node:path'
  4. import { pathToFileURL } from 'node:url'
  5. import { afterEach, describe, expect, it } from 'vitest'
  6. import { Context } from 'cordis'
  7. import Loader from '@cordisjs/plugin-loader'
  8. import Include from '@cordisjs/plugin-include'
  9. import AgentRegistry, { Inbox } from '@deepseek-ai/dsh-agent'
  10. import type { Agent, AgentStatus } from '@deepseek-ai/dsh-agent'
  11. import CommandService from '@deepseek-ai/dsh-commands'
  12. import SessionStore, { SessionId } from '@deepseek-ai/dsh-session'
  13. import * as CommandFeedback from '@deepseek-ai/dsh-command-feedback'
  14. let root: string | undefined
  15. let context: Context | undefined
  16. afterEach(async () => {
  17. await context?.fiber.dispose()
  18. context = undefined
  19. if (root !== undefined) await rm(root, { recursive: true, force: true })
  20. root = undefined
  21. })
  22. /** Register one idle agent over a store-owned session, as an app's spine does. */
  23. function agent(ctx: Context): Agent {
  24. const scope = ctx.plugin(() => {})
  25. const id = SessionId('feedback-loader-agent')
  26. const session = ctx.sessions.create(id)
  27. const inbox = new Inbox(session, { inserted: () => {}, discarded: () => {}, claimed: () => {} })
  28. let status: AgentStatus = 'idle'
  29. const value: Agent = {
  30. id,
  31. options: {},
  32. session,
  33. inbox,
  34. ctx: scope.ctx,
  35. get status() { return status },
  36. send: () => {},
  37. followup: () => {},
  38. steer: () => {},
  39. inject: () => {},
  40. cancel() { status = 'idle' },
  41. runMaintenance: task => task(new AbortController().signal),
  42. whenIdle: () => Promise.resolve(),
  43. }
  44. ctx.agents.register(value)
  45. return value
  46. }
  47. describe('/feedback real Loader composition through cordis.yml', () => {
  48. it('boots cordis.yml and records feedback without model-visible output', async () => {
  49. root = await mkdtemp(join(tmpdir(), 'dsh-command-feedback-loader-'))
  50. const configPath = join(root, 'cordis.yml')
  51. await writeFile(configPath, [
  52. "- name: '@deepseek-ai/dsh-agent'",
  53. "- name: '@deepseek-ai/dsh-session'",
  54. "- name: '@deepseek-ai/dsh-commands'",
  55. "- name: '@deepseek-ai/dsh-command-feedback'",
  56. '',
  57. ].join('\n'))
  58. context = new Context()
  59. context.baseUrl = pathToFileURL(root).href + '/'
  60. await context.plugin(Loader)
  61. context.loader.builtins.include = Include
  62. const modules = new Map<string, unknown>([
  63. ['@deepseek-ai/dsh-agent', AgentRegistry],
  64. ['@deepseek-ai/dsh-session', SessionStore],
  65. ['@deepseek-ai/dsh-commands', CommandService],
  66. ['@deepseek-ai/dsh-command-feedback', CommandFeedback],
  67. ])
  68. context.loader.internal = {
  69. version: 'v2',
  70. async import(specifier: string) {
  71. if (!modules.has(specifier)) throw new Error(`unexpected Loader import: ${specifier}`)
  72. return modules.get(specifier)
  73. },
  74. } as unknown as NonNullable<typeof context.loader.internal>
  75. await context.loader.create({ name: 'cordis:include', config: { path: pathToFileURL(configPath).href } })
  76. await context.loader.await()
  77. const owner = agent(context)
  78. const signal = new AbortController().signal
  79. // Discoverable through the composed registry, as a UI adapter finds it.
  80. expect(context.commands.list(owner).map(command => command.name)).toContain('feedback')
  81. const accepted = await context.commands.execute(owner, '/feedback the diff view is unreadable', signal)
  82. expect(accepted?.result).toEqual({ kind: 'success', text: 'Feedback recorded.' })
  83. const rejected = await context.commands.execute(owner, '/feedback', signal)
  84. expect(rejected?.result).toEqual({
  85. kind: 'error',
  86. text: 'Feedback text is required. Usage: /feedback <text>',
  87. })
  88. // The domain event owns the payload; generic command bookkeeping omits it.
  89. expect(owner.session.events.map(event => event.type))
  90. .toEqual(['command/run', 'feedback/record', 'command/done', 'command/run', 'command/done'])
  91. const run = owner.session.events.find(event => event.type === 'command/run')
  92. expect(run?.type === 'command/run' && Object.hasOwn(run.data, 'args')).toBe(false)
  93. const feedback = owner.session.events.find(event => event.type === 'feedback/record')
  94. expect(feedback?.type === 'feedback/record' && feedback.data.text).toBe('the diff view is unreadable')
  95. expect(JSON.stringify(owner.session.events).match(/the diff view is unreadable/gu)).toHaveLength(1)
  96. // Nothing reached the model.
  97. expect(owner.session.deriveMessages()).toEqual([])
  98. expect(owner.session.surface.nodes).toEqual([])
  99. })
  100. })