yaoyao/tests/server/workspacePrompt.test.ts

// @vitest-environment node
import { describe, expect, it } from 'vitest'
import { buildWorkspacePrompt, workspaceEnvironmentPrompt, type WorkspacePromptEnvironment, type WorkspacePromptInput } from '../../src/server/workspacePrompt'

function environment(): WorkspacePromptEnvironment {
  return {
    open: { computer: true, server: true, vm: true, cloud: true },
    tools: { desktopView: true, desktopFile: true, vm: true, cloud: true, plugins: false },
    deviceContext: '本轮用户消息来自电脑「工作 Mac」。「本机」指这台电脑。',
    onlineHosts: '工作 Mac（host=mac-id）', cwd: '/srv/hermes/project', fileTransferMaxMiB: 50,
  }
}
function input(): WorkspacePromptInput {
  return {
    agent: { id: 'bot', name: '助手', avatar: '', instructions: '规则原文\n保留格式', description: '处理开发任务', job: '开发与验证', voice: 'concise', actBias: 'act_now', nodeId: 'local', profile: 'dev', archived: false, revision: 7, createdAt: 1, updatedAt: 1 },
    conversation: { id: 'chat', kind: 'direct', name: '对话', avatar: '', memberIds: ['bot'], administratorId: 'bot', mode: 'free', instructions: '', autoReplyIds: [], maxReplyRounds: 1, archived: false, pinned: false, readSeq: 0, lastSeq: 0, preview: '', createdAt: 1, updatedAt: 1 },
    members: [{ id: 'bot', name: '助手' }],
    work: { depth: 0, requiredReply: true, replyMode: 'mentioned', messageId: 'message' },
    run: { mentionIds: [] }, environment: environment(),
    memory: '已确认的项目事实', noReply: '[[YAOYAO_NO_REPLY_V1]]', marker: '[yaoyao-run:run:result]',
    content: '读取本机文件\n然后核对结果', contentKind: 'user', attachmentRefs: ['[附件](ref:file)'],
  }
}

describe('Bot environment routing', () => {
  it('describes all mounted environments conditionally without moving the model or the message origin', () => {
    const text = workspaceEnvironmentPrompt(environment())
    expect(text).toContain('模型会话运行在服务端 Hermes')
    expect(text).toContain('「本机」只指本轮消息来源电脑')
    expect(text).toContain('不得切换其他机器代做')
    expect(text).toContain('host="server"')
    expect(text).toContain('desktop_shell')
    expect(text).toContain('computer_shell')
    expect(text).toContain('cloud_computer_*')
    expect(text).toContain('最大 50 MiB')
    expect(text).not.toContain('25 MiB')
    expect(text).not.toContain('当前环境是')
    expect(text).not.toContain('一律用 desktop_*')
    expect(text.match(/工作 Mac/g)).toHaveLength(2) // source and inventory, not one list per environment
    expect(text).toContain('该路径仅属于 Hermes 运行环境')
  })

  it('distinguishes global policy from mounted capabilities and does not promise native tools', () => {
    const env = environment()
    env.tools = { desktopView: false, desktopFile: false, vm: false, cloud: false, plugins: false }
    const text = workspaceEnvironmentPrompt(env)
    expect(text).toContain('开放不等于在线或已授权')
    expect(text).toContain('以本轮实际工具目录为准')
    expect(text).not.toContain('工具均可使用')
    for (const tool of ['desktop_file_copy', 'desktop_environment_view', 'computer_shell', 'cloud_computer_*']) expect(text).not.toContain(tool)
  })

  it('does not require a screenshot for file-only computers', () => {
    const env = environment()
    env.tools.desktopView = false
    const text = workspaceEnvironmentPrompt(env)
    expect(text).toContain('文件与命令操作无需先截图')
    expect(text).not.toContain('desktop_environment_view')
    expect(text).toContain('desktop_file_copy')
    env.tools.desktopFile = false
    env.tools.desktopView = true
    const view = workspaceEnvironmentPrompt(env)
    expect(view).toContain('desktop_environment_view')
    expect(view).not.toContain('电脑文件与命令工具已挂载')
    expect(view).not.toContain('desktop_file_copy')
  })

  it('cannot turn Grok model discussion, old VM text or a follow-up into environment selection', () => {
    const fixture = input()
    fixture.environment.tools.cloud = false
    fixture.environment.tools.vm = false
    const prompts = ['介绍 Grok 模型', '以前使用过虚拟机', '继续'].map(content => buildWorkspacePrompt({ ...fixture, content }))
    const environments = prompts.map(text => text.split('【本轮环境与设备】')[1]!.split('【Bot 记忆与相关事实】')[0])
    expect(new Set(environments).size).toBe(1)
    expect(environments[0]).not.toContain('cloud_computer_*')
    expect(environments[0]).not.toContain('computer_shell')
  })
})

describe('Bot prompt composition', () => {
  it('keeps persona, provenance, user content and attachments in their own sections', () => {
    const fixture = input(), text = buildWorkspacePrompt(fixture)
    expect(text).toContain(`版本 7）：\n${fixture.agent.instructions}`)
    expect(text).toContain('你是 助手。')
    expect(text).toContain('直接行动：合理默认即可开工')
    expect(text).toContain(`【本轮用户消息】\n${fixture.content}`)
    expect(text).toContain('【本轮附件】\n[附件](ref:file)')
    expect(text).toContain(fixture.marker)
    expect(text.indexOf('【Bot 身份与长期规则】')).toBeLessThan(text.indexOf('【本轮任务与协作】'))
    expect(text.indexOf('【本轮环境与设备】')).toBeLessThan(text.indexOf('【Bot 记忆与相关事实】'))
    expect(text.indexOf('【Bot 记忆与相关事实】')).toBeLessThan(text.indexOf('【本轮用户消息】'))
  })

  it('preserves discussion participation, member roles and history boundaries', () => {
    const fixture = input()
    fixture.conversation = { ...fixture.conversation, kind: 'group', memberRoles: { bot: { name: '审核', description: '核实事实' } } }
    fixture.run = { mentionIds: ['bot'], discussion: { memberIds: ['bot'], rounds: 2 }, projectId: 'project' }
    fixture.work.depth = 1
    fixture.contentKind = 'history'
    const text = buildWorkspacePrompt(fixture)
    expect(text).toContain('平等讨论，第 2 轮')
    expect(text).toContain('@助手：审核；核实事实')
    expect(text).toContain('本轮用户指定成员：@助手')
    expect(text).toContain('不要用私信或 @ 再次派发本轮工作')
    expect(text).toContain('【本轮对话上下文】')
    expect(text).toContain('当前项目 ID：project')
  })

  it('preserves automatic silence and does not offer unmounted export tools to a helper', () => {
    const fixture = input()
    fixture.work.requiredReply = false
    fixture.work.replyMode = 'automatic'
    fixture.agent.temporaryGoalId = 'goal'
    fixture.environment.tools.vm = false
    const text = buildWorkspacePrompt(fixture)
    expect(text).toContain('完整答复只能是 [[YAOYAO_NO_REPLY_V1]]')
    expect(text).toContain('不要创建团队或改变自身权限')
    expect(text).not.toContain('computer_export')
  })
})