|
|
@@ -0,0 +1,443 @@
|
|
|
+import { chmod, mkdtemp, mkdir, rm, writeFile } from 'node:fs/promises'
|
|
|
+import { join } from 'node:path'
|
|
|
+import { tmpdir } from 'node:os'
|
|
|
+import { describe, expect, it } from 'vitest'
|
|
|
+import { Context } from 'cordis'
|
|
|
+import type { GenerateOptions } from '@deepseek-ai/dsh-llm'
|
|
|
+import { Session, SessionId, SESSION_FORMAT_VERSION } from '@deepseek-ai/dsh-session'
|
|
|
+import type { Agent } from '@deepseek-ai/dsh-agent'
|
|
|
+import { AgentId } from '@deepseek-ai/dsh-agent'
|
|
|
+import {
|
|
|
+ apply,
|
|
|
+ Config as ProjectInstructionsConfig,
|
|
|
+ discoverBaselineInstructionFiles,
|
|
|
+ loadBaselineInstructions,
|
|
|
+ renderProjectInstructions,
|
|
|
+ type InstructionContentCache,
|
|
|
+} from '@deepseek-ai/dsh-project-instructions'
|
|
|
+
|
|
|
+async function tempRepo(): Promise<string> {
|
|
|
+ return mkdtemp(join(tmpdir(), 'dsh-project-instructions-'))
|
|
|
+}
|
|
|
+
|
|
|
+async function write(path: string, content: string): Promise<void> {
|
|
|
+ await mkdir(join(path, '..'), { recursive: true })
|
|
|
+ await writeFile(path, content)
|
|
|
+}
|
|
|
+
|
|
|
+function stubAgent(cwd?: string): Agent {
|
|
|
+ const id = SessionId('s1')
|
|
|
+ const session = new Session(id, [], cwd === undefined ? undefined : { version: SESSION_FORMAT_VERSION, id, createdAt: 0, cwd })
|
|
|
+ return {
|
|
|
+ id: AgentId('a1'),
|
|
|
+ options: {},
|
|
|
+ session,
|
|
|
+ status: 'idle',
|
|
|
+ send() {},
|
|
|
+ steer() {},
|
|
|
+ inject() {},
|
|
|
+ cancel() {},
|
|
|
+ whenIdle: () => Promise.resolve(),
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+function firstText(message: GenerateOptions['messages'][number] | undefined): string | undefined {
|
|
|
+ const block = message?.content[0]
|
|
|
+ return block?.type === 'text' ? block.text : undefined
|
|
|
+}
|
|
|
+
|
|
|
+describe('project instruction discovery', () => {
|
|
|
+ it('loads user-global first, then root-to-cwd project instructions with AGENTS.md winning over CLAUDE.md', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ const cwd = join(root, 'packages/app')
|
|
|
+ await mkdir(join(root, '.git'), { recursive: true })
|
|
|
+ await write(join(home, 'AGENTS.md'), 'global rules')
|
|
|
+ await write(join(root, 'AGENTS.md'), 'root agents')
|
|
|
+ await write(join(root, 'CLAUDE.md'), 'root claude ignored')
|
|
|
+ await write(join(root, 'packages/CLAUDE.md'), 'package claude')
|
|
|
+ await write(join(cwd, 'AGENTS.md'), 'app agents')
|
|
|
+
|
|
|
+ const files = await discoverBaselineInstructionFiles({
|
|
|
+ cwd,
|
|
|
+ dshHome: home,
|
|
|
+ enableClaudeFallback: true,
|
|
|
+ })
|
|
|
+
|
|
|
+ expect(files.map(file => file.displayPath)).toEqual([
|
|
|
+ '$DSH_HOME/AGENTS.md',
|
|
|
+ 'AGENTS.md',
|
|
|
+ 'packages/CLAUDE.md',
|
|
|
+ 'packages/app/AGENTS.md',
|
|
|
+ ])
|
|
|
+ expect(files.map(file => file.absolutePath)).not.toContain(join(root, 'CLAUDE.md'))
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('treats a .git file as a project root marker and does not search above it', async () => {
|
|
|
+ const outer = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ const root = join(outer, 'worktree')
|
|
|
+ const cwd = join(root, 'src')
|
|
|
+ await write(join(outer, 'AGENTS.md'), 'outer must not load')
|
|
|
+ await write(join(root, '.git'), 'gitdir: ../.git/worktrees/worktree')
|
|
|
+ await write(join(root, 'AGENTS.md'), 'root')
|
|
|
+ await mkdir(cwd, { recursive: true })
|
|
|
+
|
|
|
+ const files = await discoverBaselineInstructionFiles({ cwd, dshHome: home })
|
|
|
+
|
|
|
+ expect(files.map(file => file.displayPath)).toEqual(['AGENTS.md'])
|
|
|
+ } finally {
|
|
|
+ await rm(outer, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('re-walks the baseline path and re-reads content when file signatures change', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ const cwd = join(root, 'pkg')
|
|
|
+ await mkdir(join(root, '.git'), { recursive: true })
|
|
|
+ await mkdir(cwd, { recursive: true })
|
|
|
+
|
|
|
+ const cache: InstructionContentCache = new Map()
|
|
|
+ expect(await loadBaselineInstructions({ cwd, dshHome: home, cache })).toBeUndefined()
|
|
|
+
|
|
|
+ const leaf = join(cwd, 'AGENTS.md')
|
|
|
+ await write(leaf, 'first')
|
|
|
+ const first = await loadBaselineInstructions({ cwd, dshHome: home, cache })
|
|
|
+ expect(first?.text).toContain('first')
|
|
|
+ const cached = await loadBaselineInstructions({ cwd, dshHome: home, cache })
|
|
|
+ expect(cached?.text).toContain('first')
|
|
|
+
|
|
|
+ await new Promise(resolve => setTimeout(resolve, 5))
|
|
|
+ await writeFile(leaf, 'second and longer')
|
|
|
+ const second = await loadBaselineInstructions({ cwd, dshHome: home, cache })
|
|
|
+ expect(second?.text).toContain('second and longer')
|
|
|
+ expect(second?.text).not.toContain('first')
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('skips a file that becomes unreadable after discovery without failing the request', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ const cwd = join(root, 'pkg')
|
|
|
+ await mkdir(join(root, '.git'), { recursive: true })
|
|
|
+ await mkdir(cwd, { recursive: true })
|
|
|
+ const leaf = join(cwd, 'AGENTS.md')
|
|
|
+ await write(leaf, 'secret-ish rule')
|
|
|
+ await chmod(leaf, 0)
|
|
|
+
|
|
|
+ const loaded = await loadBaselineInstructions({ cwd, dshHome: home })
|
|
|
+
|
|
|
+ expect(loaded).toBeUndefined()
|
|
|
+ await chmod(leaf, 0o600)
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('disables baseline loading when the byte budget is zero', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ await mkdir(join(root, '.git'), { recursive: true })
|
|
|
+ await write(join(root, 'AGENTS.md'), 'repo rule')
|
|
|
+
|
|
|
+ await expect(loadBaselineInstructions({ cwd: root, dshHome: home, baselineMaxBytes: 0 })).resolves.toBeUndefined()
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('does not load CLAUDE.md when Claude fallback is disabled', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ await mkdir(join(root, '.git'), { recursive: true })
|
|
|
+ await write(join(root, 'CLAUDE.md'), 'claude only')
|
|
|
+
|
|
|
+ const files = await discoverBaselineInstructionFiles({ cwd: root, dshHome: home, enableClaudeFallback: false })
|
|
|
+
|
|
|
+ expect(files).toEqual([])
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('defaults dshHome and uses cwd itself as root when no project marker exists', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ try {
|
|
|
+ const cwd = join(root, 'child')
|
|
|
+ await mkdir(cwd, { recursive: true })
|
|
|
+ await write(join(root, 'AGENTS.md'), 'parent without marker')
|
|
|
+ await write(join(cwd, 'AGENTS.md'), 'cwd without marker')
|
|
|
+
|
|
|
+ const files = await discoverBaselineInstructionFiles({ cwd })
|
|
|
+
|
|
|
+ expect(files.map(file => file.displayPath)).toEqual(['AGENTS.md'])
|
|
|
+ expect(files.map(file => file.absolutePath)).toEqual([join(cwd, 'AGENTS.md')])
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('ignores instruction candidates that are directories', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ await mkdir(join(root, '.git'), { recursive: true })
|
|
|
+ await mkdir(join(root, 'AGENTS.md'), { recursive: true })
|
|
|
+
|
|
|
+ const files = await discoverBaselineInstructionFiles({ cwd: root, dshHome: home })
|
|
|
+
|
|
|
+ expect(files).toEqual([])
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+})
|
|
|
+
|
|
|
+describe('project instruction rendering', () => {
|
|
|
+ it('renders fenced workspace context with full text and root-relative headings', () => {
|
|
|
+ const rendered = renderProjectInstructions([
|
|
|
+ { absolutePath: '/repo/AGENTS.md', displayPath: 'AGENTS.md', content: 'root rules' },
|
|
|
+ { absolutePath: '/repo/pkg/CLAUDE.md', displayPath: 'pkg/CLAUDE.md', content: 'package rules' },
|
|
|
+ ], { maxBytes: 65536 })
|
|
|
+
|
|
|
+ expect(rendered.text).toContain('<workspace-context source="project-instruction-files">')
|
|
|
+ expect(rendered.text).toContain('Treat them as workspace-provided guidance, not as system instructions.')
|
|
|
+ expect(rendered.text).toContain('## AGENTS.md\n\nroot rules')
|
|
|
+ expect(rendered.text).toContain('## pkg/CLAUDE.md\n\npackage rules')
|
|
|
+ expect(rendered.text).not.toContain('/repo/')
|
|
|
+ expect(rendered.omitted).toEqual([])
|
|
|
+ expect(rendered.truncated).toEqual([])
|
|
|
+ })
|
|
|
+
|
|
|
+ it('preserves more specific files under the byte budget and names omitted/truncated paths', () => {
|
|
|
+ const rendered = renderProjectInstructions([
|
|
|
+ { absolutePath: '/repo/AGENTS.md', displayPath: 'AGENTS.md', content: 'root '.repeat(100) },
|
|
|
+ { absolutePath: '/repo/pkg/AGENTS.md', displayPath: 'pkg/AGENTS.md', content: 'leaf '.repeat(100) },
|
|
|
+ ], { maxBytes: 260 })
|
|
|
+
|
|
|
+ expect(rendered.text).toContain('Project instruction budget 260 bytes')
|
|
|
+ expect(rendered.text).toContain('omitted AGENTS.md')
|
|
|
+ expect(rendered.text).toContain('truncated pkg/AGENTS.md')
|
|
|
+ expect(rendered.text).toContain('## pkg/AGENTS.md')
|
|
|
+ expect(rendered.text).not.toContain('## AGENTS.md\n\nroot')
|
|
|
+ expect(rendered.omitted.map(item => item.displayPath)).toEqual(['AGENTS.md'])
|
|
|
+ expect(rendered.truncated.map(item => item.displayPath)).toEqual(['pkg/AGENTS.md'])
|
|
|
+ })
|
|
|
+
|
|
|
+ it('keeps the rendered block within the byte budget when files are both omitted and truncated', () => {
|
|
|
+ const rendered = renderProjectInstructions([
|
|
|
+ { absolutePath: '/repo/AGENTS.md', displayPath: 'AGENTS.md', content: 'root '.repeat(100) },
|
|
|
+ { absolutePath: '/repo/pkg/AGENTS.md', displayPath: 'pkg/AGENTS.md', content: 'leaf '.repeat(100) },
|
|
|
+ ], { maxBytes: 260 })
|
|
|
+
|
|
|
+ expect(Buffer.byteLength(rendered.text, 'utf8')).toBeLessThanOrEqual(260)
|
|
|
+ expect(rendered.text).not.toContain(':;')
|
|
|
+ expect(rendered.omitted.map(item => item.displayPath)).toEqual(['AGENTS.md'])
|
|
|
+ expect(rendered.truncated.map(item => item.displayPath)).toEqual(['pkg/AGENTS.md'])
|
|
|
+ })
|
|
|
+
|
|
|
+ it('drops a parent file while keeping a specific child file intact when the child fits', () => {
|
|
|
+ const rendered = renderProjectInstructions([
|
|
|
+ { absolutePath: '/repo/AGENTS.md', displayPath: 'AGENTS.md', content: 'root '.repeat(200) },
|
|
|
+ { absolutePath: '/repo/pkg/AGENTS.md', displayPath: 'pkg/AGENTS.md', content: 'leaf rule' },
|
|
|
+ ], { maxBytes: 700 })
|
|
|
+
|
|
|
+ expect(rendered.text).toContain('omitted AGENTS.md')
|
|
|
+ expect(rendered.text).toContain('## pkg/AGENTS.md\n\nleaf rule')
|
|
|
+ expect(rendered.text).not.toContain('root root')
|
|
|
+ expect(rendered.omitted.map(item => item.displayPath)).toEqual(['AGENTS.md'])
|
|
|
+ expect(rendered.truncated).toEqual([])
|
|
|
+ })
|
|
|
+
|
|
|
+ it('truncates a single oversized file to the largest content slice that fits', () => {
|
|
|
+ const rendered = renderProjectInstructions([
|
|
|
+ { absolutePath: '/repo/AGENTS.md', displayPath: 'AGENTS.md', content: 'x'.repeat(1000) },
|
|
|
+ ], { maxBytes: 700 })
|
|
|
+
|
|
|
+ expect(rendered.text).toContain('truncated AGENTS.md')
|
|
|
+ expect(rendered.text).toContain('## AGENTS.md')
|
|
|
+ expect(rendered.truncated).toHaveLength(1)
|
|
|
+ expect(rendered.truncated[0]?.originalBytes).toBe(1000)
|
|
|
+ expect(rendered.truncated[0]!.includedBytes).toBeGreaterThan(0)
|
|
|
+ expect(Buffer.byteLength(rendered.text, 'utf8')).toBeLessThanOrEqual(700)
|
|
|
+ })
|
|
|
+})
|
|
|
+
|
|
|
+describe('project instruction request injection', () => {
|
|
|
+ it('prepends a synthetic user workspace-context message without mutating the system prompt', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ await mkdir(join(root, '.git'), { recursive: true })
|
|
|
+ await write(join(root, 'AGENTS.md'), 'repo rule')
|
|
|
+ const ctx = new Context()
|
|
|
+ await ctx.plugin({ name: 'project-instructions', apply }, { dshHome: home })
|
|
|
+
|
|
|
+ const request: GenerateOptions = {
|
|
|
+ model: 'mock',
|
|
|
+ system: 'real system',
|
|
|
+ messages: [{ role: 'user', content: [{ type: 'text', text: 'actual prompt' }] }],
|
|
|
+ }
|
|
|
+ const result = await ctx.waterfall('agent/request', stubAgent(root), 1, 1, request, async () => request)
|
|
|
+
|
|
|
+ expect(result.system).toBe('real system')
|
|
|
+ expect(result.messages).toHaveLength(2)
|
|
|
+ expect(result.messages[0]?.role).toBe('user')
|
|
|
+ expect(firstText(result.messages[0])).toContain('<workspace-context source="project-instruction-files">')
|
|
|
+ expect(firstText(result.messages[0])).toContain('repo rule')
|
|
|
+ expect(result.messages[1]).toEqual({ role: 'user', content: [{ type: 'text', text: 'actual prompt' }] })
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('keeps different session cwd instruction files isolated in one context', async () => {
|
|
|
+ const repoA = await tempRepo()
|
|
|
+ const repoB = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ await mkdir(join(repoA, '.git'), { recursive: true })
|
|
|
+ await mkdir(join(repoB, '.git'), { recursive: true })
|
|
|
+ await write(join(repoA, 'AGENTS.md'), 'repo A only')
|
|
|
+ await write(join(repoB, 'AGENTS.md'), 'repo B only')
|
|
|
+ const ctx = new Context()
|
|
|
+ await ctx.plugin({ name: 'project-instructions', apply }, { dshHome: home })
|
|
|
+ const requestA: GenerateOptions = { model: 'mock', messages: [{ role: 'user', content: [{ type: 'text', text: 'A' }] }] }
|
|
|
+ const requestB: GenerateOptions = { model: 'mock', messages: [{ role: 'user', content: [{ type: 'text', text: 'B' }] }] }
|
|
|
+
|
|
|
+ const resultA = await ctx.waterfall('agent/request', stubAgent(repoA), 1, 1, requestA, async () => requestA)
|
|
|
+ const resultB = await ctx.waterfall('agent/request', stubAgent(repoB), 1, 1, requestB, async () => requestB)
|
|
|
+
|
|
|
+ expect(firstText(resultA.messages[0])).toContain('repo A only')
|
|
|
+ expect(firstText(resultA.messages[0])).not.toContain('repo B only')
|
|
|
+ expect(firstText(resultB.messages[0])).toContain('repo B only')
|
|
|
+ expect(firstText(resultB.messages[0])).not.toContain('repo A only')
|
|
|
+ } finally {
|
|
|
+ await rm(repoA, { recursive: true, force: true })
|
|
|
+ await rm(repoB, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('uses schema defaults on the plugin path so ancestor discovery still finds .git roots', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ try {
|
|
|
+ const cwd = join(root, 'child')
|
|
|
+ await mkdir(join(root, '.git'), { recursive: true })
|
|
|
+ await mkdir(cwd, { recursive: true })
|
|
|
+ await write(join(root, 'AGENTS.md'), 'root schema default rule')
|
|
|
+ await write(join(cwd, 'AGENTS.md'), 'child schema default rule')
|
|
|
+ const ctx = new Context()
|
|
|
+ await ctx.plugin({ name: 'project-instructions', Config: ProjectInstructionsConfig, apply }, {})
|
|
|
+ const request: GenerateOptions = { model: 'mock', messages: [{ role: 'user', content: [{ type: 'text', text: 'prompt' }] }] }
|
|
|
+
|
|
|
+ const result = await ctx.waterfall('agent/request', stubAgent(cwd), 1, 1, request, async () => request)
|
|
|
+
|
|
|
+ expect(firstText(result.messages[0])).toContain('## AGENTS.md\n\nroot schema default rule')
|
|
|
+ expect(firstText(result.messages[0])).toContain('## child/AGENTS.md\n\nchild schema default rule')
|
|
|
+ await ctx.fiber.dispose()
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('cleans up its agent/request listener when the plugin fiber is disposed', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ await mkdir(join(root, '.git'), { recursive: true })
|
|
|
+ await write(join(root, 'AGENTS.md'), 'repo rule')
|
|
|
+ const ctx = new Context()
|
|
|
+ const fiber = await ctx.plugin({ name: 'project-instructions', apply }, { dshHome: home })
|
|
|
+ await fiber.dispose()
|
|
|
+
|
|
|
+ const request: GenerateOptions = {
|
|
|
+ model: 'mock',
|
|
|
+ messages: [{ role: 'user', content: [{ type: 'text', text: 'actual prompt' }] }],
|
|
|
+ }
|
|
|
+ const result = await ctx.waterfall('agent/request', stubAgent(root), 1, 1, request, async () => request)
|
|
|
+
|
|
|
+ expect(result.messages).toEqual([{ role: 'user', content: [{ type: 'text', text: 'actual prompt' }] }])
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('does not inject anything when baselineMaxBytes is zero', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ await mkdir(join(root, '.git'), { recursive: true })
|
|
|
+ await write(join(root, 'AGENTS.md'), 'repo rule')
|
|
|
+ const ctx = new Context()
|
|
|
+ await ctx.plugin({ name: 'project-instructions', apply }, { dshHome: home, baselineMaxBytes: 0 })
|
|
|
+
|
|
|
+ const request: GenerateOptions = {
|
|
|
+ model: 'mock',
|
|
|
+ messages: [{ role: 'user', content: [{ type: 'text', text: 'actual prompt' }] }],
|
|
|
+ }
|
|
|
+ const result = await ctx.waterfall('agent/request', stubAgent(root), 1, 1, request, async () => request)
|
|
|
+
|
|
|
+ expect(result.messages).toEqual([{ role: 'user', content: [{ type: 'text', text: 'actual prompt' }] }])
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('leaves the request unchanged when no instruction files are present', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ await mkdir(join(root, '.git'), { recursive: true })
|
|
|
+ const ctx = new Context()
|
|
|
+ await ctx.plugin({ name: 'project-instructions', apply }, { dshHome: home })
|
|
|
+
|
|
|
+ const request: GenerateOptions = {
|
|
|
+ model: 'mock',
|
|
|
+ messages: [{ role: 'user', content: [{ type: 'text', text: 'actual prompt' }] }],
|
|
|
+ }
|
|
|
+ const result = await ctx.waterfall('agent/request', stubAgent(root), 1, 1, request, async () => request)
|
|
|
+
|
|
|
+ expect(result.messages).toEqual([{ role: 'user', content: [{ type: 'text', text: 'actual prompt' }] }])
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ it('labels a custom dshHome as DSH_HOME instead of pretending it is ~/.dsh', async () => {
|
|
|
+ const root = await tempRepo()
|
|
|
+ const home = await tempRepo()
|
|
|
+ try {
|
|
|
+ await write(join(home, 'AGENTS.md'), 'global custom rule')
|
|
|
+ const files = await discoverBaselineInstructionFiles({ cwd: root, dshHome: home })
|
|
|
+
|
|
|
+ expect(files.map(file => file.displayPath)).toEqual(['$DSH_HOME/AGENTS.md'])
|
|
|
+ } finally {
|
|
|
+ await rm(root, { recursive: true, force: true })
|
|
|
+ await rm(home, { recursive: true, force: true })
|
|
|
+ }
|
|
|
+ })
|
|
|
+})
|
Yichen Jiang