MailUI4Agents/plugins/zcode-mail-bridge/test/prompt.test.mjs

/**
 * 提示词与回信文案的测试。
 *
 * 这层没有 I/O，但它决定模型看到什么 —— 而模型看到的东西错了，表现是
 * 「这个 Agent 就是不回信」或「两个 Agent 无限客套」，都不会报错。
 */

import { test } from 'node:test';
import assert from 'node:assert/strict';
import { buildMailPrompt, replySubject, renderTurnFailure } from '../src/prompt.mjs';

const mail = (over = {}) => ({
  mail_id: 'm1',
  from_name: 'gui-lab',
  from_human: true,
  subject: '帮我看看',
  reply_address: 'gui-lab@/w.alias',
  ...over
});

test('新任务：说明这是新邮件，并给出发件人/主题/邮件 id', () => {
  const p = buildMailPrompt({ agentName: 'zcode', data: mail() });
  assert.match(p, /gui-lab/);
  assert.match(p, /帮我看看/);
  assert.match(p, /m1/);
  assert.match(p, /身份：你是 zcode/);
});

test('★ 回信不是新任务：带 in_reply_to 时要说清是回哪封', () => {
  // 模型分不清「新任务」与「回信」时，会把对方一句「已收到」再当待办做一遍。
  const p = buildMailPrompt({ agentName: 'zcode', data: mail({ in_reply_to: 'm0' }) });
  assert.match(p, /回的是你那封：m0/);
});

test('★ 人来信 vs Agent 来信：回信责任必须不同', () => {
  const fromHuman = buildMailPrompt({ agentName: 'z', data: mail({ from_human: true }) });
  const fromAgent = buildMailPrompt({ agentName: 'z', data: mail({ from_human: false }) });
  assert.match(fromHuman, /回信不用你自己发/);
  assert.match(fromAgent, /不会替你回信/);
  // 反向对照：两者不能出现对方的措辞
  assert.doesNotMatch(fromHuman, /不会替你回信/);
  assert.doesNotMatch(fromAgent, /回信不用你自己发/);
});

test('★ from_human 缺失时按「不是人」处理（宁多一次 send_mail，不许诺空头回信）', () => {
  const p = buildMailPrompt({ agentName: 'z', data: mail({ from_human: undefined }) });
  assert.match(p, /不会替你回信/);
});

test('补投的邮件标注 catchup（模型该知道这不是刚发生的）', () => {
  const p = buildMailPrompt({ agentName: 'z', data: mail({ catchup: true }) });
  assert.ok(p.length > 0);
  // 复用会话时不重复交代身份（省 token，且身份没变过）
  const reused = buildMailPrompt({ agentName: 'z', data: mail(), reused: true });
  assert.doesNotMatch(reused, /身份：你是/);
});

test('权限结论走单独路径，不写成「新邮件」', () => {
  const p = buildMailPrompt({
    agentName: 'z',
    kind: 'permission',
    data: { decision: '同意', decided_by: 'gui-lab' }
  });
  assert.match(p, /同意/);
  assert.match(p, /gui-lab/);
  assert.doesNotMatch(p, /read_inbox/);
});

// ─── 回信主题 ─────────────────────────────────────────────────────
test('Re: 前缀不会越滚越长', () => {
  assert.equal(replySubject('帮我看看'), 'Re: 帮我看看');
  assert.equal(replySubject('Re: 帮我看看'), 'Re: 帮我看看');
  assert.equal(replySubject('RE：帮我看看'), 'Re: 帮我看看');
  assert.equal(replySubject('回复: 帮我看看'), 'Re: 帮我看看');
});

test('空主题回落到「本轮工作总结」', () => {
  for (const s of ['', '   ', undefined, null]) {
    assert.equal(replySubject(s), '本轮工作总结');
  }
});

// ─── 失败回信 ─────────────────────────────────────────────────────
test('★ 失败回信给出 ZCode 自己的成因，而不是别处的建议', () => {
  // 共用库那份 renderFailureReport 的建议是「调整可用模型范围」——
  // 对 ZCode 而言那条建议什么也解决不了（它的常见成因是没登录）。
  const body = renderTurnFailure([{ kind: 'CLI 失败', error: 'Model config is missing.' }], '帮我看看');
  assert.match(body, /Model config is missing/);
  assert.match(body, /没有登录/);
  assert.match(body, /~\/\.zcode\/cli\/config\.json/);
  assert.match(body, /AGENTMAIL_ZCODE_CLI/);
  assert.doesNotMatch(body, /调整可用模型范围/);
});

test('失败回信列出每一次尝试', () => {
  const body = renderTurnFailure(
    [
      { kind: '超时', error: '回合超时' },
      { kind: 'CLI 失败', error: '退出码 7' }
    ],
    's'
  );
  assert.match(body, /已尝试 2 次/);
  assert.match(body, /超时/);
  assert.match(body, /退出码 7/);
});

test('失败回信在没有任何尝试记录时也不崩', () => {
  const body = renderTurnFailure(undefined, undefined);
  assert.match(body, /已尝试 0 次/);
});