feat(zcode): 工具加 MCP 注解 + headless 档位映射改为 plan(否则一个工具都用不了)
## 逆出 ZCode 的 MCP 权限判定,并据此让工具真的可用
逐字逆自 CLI 产物:
Ari(): annotations.readOnlyHint === true → riskLevel "low"
annotations.destructiveHint === true → riskLevel "high"
needsApproval = true ← **硬编码为真,与注解无关**
checkBuildMode(): needsApproval || destructive || sideEffectScope !== "none" → ask
checkPlanMode(): permissionName === "mcp" && !destructive → allow
两条合起来的结论不直观但很关键:
- **build 档下每一个 MCP 工具都要审批**(needsApproval 恒真),而 headless
模式没有交互式审批客户端 ⇒ 全被拒。实测:模型连 read_inbox 都调不动,
只能从提示词里猜;更糟的是它**绕道**用 Bash 去读网关的 sqlite WAL 文件
(它自己在回信里如实交代了这件事)。
- **plan 档下只要不声明 destructive,MCP 工具直接放行**。
于是两处改动:
1. `lib/tools.mjs` 给每个工具加真实注解(读类 readOnlyHint,写类
destructiveHint:false——它们确实不破坏任何东西);`lib/mcp-rpc.mjs` 透传
annotations。**漏传不是"少个提示",而是工具在该档下全被拒**。
2. `src/turn-mode.mjs` 的 workspace 档映射从 build 改为 **plan**。
build 在本环境等于「什么都不能做」,那不是保守而是不可用;plan 才是真的
fail-closed:危险的自带工具被平台直接拒,能用的只有我们声明为非破坏性的工具。
日志会明确写出为什么退档。可用 `AGENTMAIL_ZCODE_MODE_MAP` 覆盖
(平台修好钩子后只改配置就能恢复 build,不必等发版)。
## 真模型验证
场景 A 的判据同时加强:**正文本标记只出现在邮件正文里**(驱动的提示词只带主题
与 mail_id),所以模型必须真的读信才可能答对。通过 —— 约 20-30 秒一轮。
反过来说,早先那版「通过」是假的:标记在主题里,模型从提示词抄一遍就行。
## 仍然做不到的(见 README 已知缺口)
授权桥(PermissionRequest 钩子)在本版本(3.10.2 / CLI 0.16.5)**不可用**:
有时根本不触发,触发时在 ~5ms 内失败且**命令从未被 spawn**
(用「钩子写 marker 文件」的副作用验证,process 与 command 两种类型都一样)。
所以 workspace 档「危险操作问人」目前在 headless 下无法实现。
单元 329/329。
This commit is contained in:
@ -218,7 +218,8 @@ test('Agent 来信的提示词必须说清「插件不会替你回信」', async
|
||||
test('★ 档位随邮件传下去,并作为 --mode 与钩子环境变量注入', async () => {
|
||||
for (const [tier, mode] of [
|
||||
['plan', 'plan'],
|
||||
['workspace', 'build'],
|
||||
// workspace 默认映射到 plan:build 在 headless 下连 MCP 工具都要审批而无人可批
|
||||
['workspace', 'plan'],
|
||||
['full', 'yolo']
|
||||
]) {
|
||||
const h = await harness();
|
||||
|
||||
@ -200,17 +200,27 @@ async function main() {
|
||||
|
||||
// ── A. 不需要授权:模型读信 → 产出文本 → 驱动回信 ──────────
|
||||
if (SCENARIO === 'a' || SCENARIO === 'all') {
|
||||
const marker = `ZC-A-${Date.now()}`;
|
||||
console.log(`── A: 纯文本问答(标记 ${marker})`);
|
||||
await humanSend(cookie, `真模型验证A ${marker}`, `请把下面这一行原样回给我,不要改动:${marker}`);
|
||||
const reply = await findReply(cookie, marker, 300000);
|
||||
// 两个标记分工不同:
|
||||
// subjectTag —— 只用来**定位回信**(收件箱是跨轮次共享的持久状态)
|
||||
// bodyTag —— 只出现在**邮件正文**里,用来**证明模型真读了信**
|
||||
// 驱动的提示词只带主题与 mail_id、不带正文,所以主题里放 bodyTag 就等于
|
||||
// 把答案送给模型(早先的版本正是如此,那条判据什么也没证明)。
|
||||
const subjectTag = `SUBJ-${Date.now()}`;
|
||||
const bodyTag = `ZC-A-${Date.now()}`;
|
||||
console.log(`── A: 模型必须自己读信(正文标记 ${bodyTag})`);
|
||||
await humanSend(
|
||||
cookie,
|
||||
`真模型验证A ${subjectTag}`,
|
||||
`请把下面这一行原样回给我,不要改动:${bodyTag}`
|
||||
);
|
||||
const reply = await findReply(cookie, subjectTag, 300000);
|
||||
if (!reply) {
|
||||
const fail = await findFailure(cookie, marker, 3000);
|
||||
record('A · 模型读信并回信', '失败', fail ? `收到失败信:${String(fail.body).slice(0, 80)}` : '5 分钟内没有回信');
|
||||
} else if (String(reply.body || '').includes(marker)) {
|
||||
record('A · 模型读信并回信', '通过', `回信含标记,正文 ${String(reply.body).length} 字`);
|
||||
} else if (String(reply.body || '').includes(bodyTag)) {
|
||||
record('A · 模型读信并回信', '通过', `回信含正文标记(必须读信才可能答对),正文 ${String(reply.body).length} 字`);
|
||||
} else {
|
||||
record('A · 模型读信并回信', '失败', `回信正文没有标记:${String(reply.body).slice(0, 80)}`);
|
||||
record('A · 模型读信并回信', '失败', `回信正文没有正文本标记:${String(reply.body).slice(0, 80)}`);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@ -50,7 +50,7 @@ test('任何无 id 的消息都不回响应', async () => {
|
||||
assert.equal(out, null);
|
||||
});
|
||||
|
||||
test('tools/list 只暴露 name/description/inputSchema(多带的字段会被客户端拒绝)', async () => {
|
||||
test('tools/list 只暴露 name/description/inputSchema/annotations(多带的字段会被客户端拒绝)', async () => {
|
||||
const out = await handleMessage({ jsonrpc: '2.0', id: 2, method: 'tools/list' }, makeCtx());
|
||||
assert.equal(out.result.tools.length, 2);
|
||||
for (const t of out.result.tools) {
|
||||
@ -58,6 +58,23 @@ test('tools/list 只暴露 name/description/inputSchema(多带的字段会被
|
||||
}
|
||||
});
|
||||
|
||||
test('★ annotations 必须透传(ZCode 靠它算风险等级,plan 档据此放行)', async () => {
|
||||
// 漏传的后果不是「少个提示」而是「工具在该档下全被拒」:
|
||||
// ZCode 的 MCP 工具 needsApproval 恒为真,只有 plan 档的
|
||||
// 「!destructive → allow」能放行,而 destructive 正是从 annotations 读的。
|
||||
const ctx = {
|
||||
tools: [{ name: 'read_inbox', description: 'd', inputSchema: {}, annotations: { readOnlyHint: true, destructiveHint: false } }],
|
||||
call: async () => 'x'
|
||||
};
|
||||
const out = await handleMessage({ jsonrpc: '2.0', id: 3, method: 'tools/list' }, ctx);
|
||||
assert.deepEqual(out.result.tools[0].annotations, { readOnlyHint: true, destructiveHint: false });
|
||||
});
|
||||
|
||||
test('★ 反向对照:没有注解的工具不该凭空多出 annotations 字段', async () => {
|
||||
const out = await handleMessage({ jsonrpc: '2.0', id: 4, method: 'tools/list' }, makeCtx());
|
||||
assert.equal('annotations' in out.result.tools[0], false);
|
||||
});
|
||||
|
||||
test('tools/call 成功时回 content 文本数组', async () => {
|
||||
const out = await handleMessage(
|
||||
{ jsonrpc: '2.0', id: 3, method: 'tools/call', params: { name: 'read_inbox', arguments: {} } },
|
||||
|
||||
@ -10,12 +10,23 @@ import { test } from 'node:test';
|
||||
import assert from 'node:assert/strict';
|
||||
import { zcodeModeForTier, modeReachesPermissionHook, describeTier, ZCODE_MODES } from '../src/turn-mode.mjs';
|
||||
|
||||
test('三个档位映射到三个不同的 mode', () => {
|
||||
test('★ workspace 映射到 plan(不是 build),full 映射到 yolo', () => {
|
||||
// build 在 headless 下等于「什么都不行」:MCP 工具的 needsApproval 硬编码为真,
|
||||
// 而 headless 没有审批客户端 → 连 read_inbox 都被拒(实测)。
|
||||
// plan 是真的 fail-closed:危险的自带工具被平台拒,我们的非破坏性工具放行。
|
||||
assert.equal(zcodeModeForTier('plan'), 'plan');
|
||||
assert.equal(zcodeModeForTier('workspace'), 'build');
|
||||
assert.equal(zcodeModeForTier('workspace'), 'plan');
|
||||
assert.equal(zcodeModeForTier('full'), 'yolo');
|
||||
});
|
||||
|
||||
test('★ 映射可被 AGENTMAIL_ZCODE_MODE_MAP 覆盖(平台修好后不必等发版)', () => {
|
||||
assert.equal(zcodeModeForTier('workspace', { AGENTMAIL_ZCODE_MODE_MAP: 'workspace:build' }), 'build');
|
||||
assert.equal(zcodeModeForTier('workspace', { AGENTMAIL_ZCODE_MODE_MAP: 'workspace:yolo,full:plan' }), 'yolo');
|
||||
// 非法值被忽略,不改变默认
|
||||
assert.equal(zcodeModeForTier('workspace', { AGENTMAIL_ZCODE_MODE_MAP: 'workspace:nonsense' }), 'plan');
|
||||
assert.equal(zcodeModeForTier('workspace', { AGENTMAIL_ZCODE_MODE_MAP: '' }), 'plan');
|
||||
});
|
||||
|
||||
test('★ 只有 full 档会得到 yolo', () => {
|
||||
// 反向对照:如果任何其它档位(含拼错的、空的、未知的、大小写不对的)
|
||||
// 也能得到 yolo,那就意味着一个打字错误会关掉整个授权系统。
|
||||
@ -33,13 +44,13 @@ test('★ 大写 FULL 不认,落在安全侧', () => {
|
||||
// 这是**刻意保留**的好性质:认不出来时不会掉进「免授权」那一档,
|
||||
// 而是退回 default。这条断言把它钉住 —— 哪天有人「顺手」改成大小写不敏感,
|
||||
// 就会有一个打字错误变成全权授权的风险面。
|
||||
assert.equal(zcodeModeForTier('FULL'), 'build');
|
||||
assert.equal(zcodeModeForTier('PLAN'), 'build');
|
||||
assert.equal(zcodeModeForTier('FULL'), 'plan');
|
||||
assert.equal(zcodeModeForTier('PLAN'), 'plan');
|
||||
});
|
||||
|
||||
test('未知档位退回 build(安全侧),不是 yolo', () => {
|
||||
assert.equal(zcodeModeForTier('nonsense'), 'build');
|
||||
assert.equal(zcodeModeForTier(undefined), 'build');
|
||||
test('未知档位退回 plan(安全侧),不是 yolo', () => {
|
||||
assert.equal(zcodeModeForTier('nonsense'), 'plan');
|
||||
assert.equal(zcodeModeForTier(undefined), 'plan');
|
||||
});
|
||||
|
||||
test('产出的 mode 必须是 ZCode 认识的值', () => {
|
||||
@ -48,6 +59,12 @@ test('产出的 mode 必须是 ZCode 认识的值', () => {
|
||||
}
|
||||
});
|
||||
|
||||
test('★ 只有 full 档会得到 yolo(覆盖后仍成立)', () => {
|
||||
assert.equal(zcodeModeForTier('full', {}), 'yolo');
|
||||
assert.equal(zcodeModeForTier('workspace', {}), 'plan');
|
||||
assert.equal(zcodeModeForTier('plan', {}), 'plan');
|
||||
});
|
||||
|
||||
test('只有 build / edit 会让危险操作走到授权钩子', () => {
|
||||
assert.equal(modeReachesPermissionHook('build'), true);
|
||||
assert.equal(modeReachesPermissionHook('edit'), true);
|
||||
@ -64,5 +81,9 @@ test('★ 反向对照:plan 与 full 都不产生询问,但原因不同', ()
|
||||
assert.notEqual(plan, full);
|
||||
assert.match(plan, /只读/);
|
||||
assert.match(full, /全权/);
|
||||
assert.match(describeTier('workspace'), /授权钩子/);
|
||||
// workspace 在本平台退到 plan,日志里必须说清**为什么**退
|
||||
// (否则人只会看到「为什么它什么都不做」而无从判断)
|
||||
assert.match(describeTier('workspace'), /headless 做不到|只读/);
|
||||
// 显式覆盖回 build 时,说明恢复成「会走到授权钩子」
|
||||
assert.match(describeTier('workspace', 'build'), /授权钩子/);
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user