Files
my-pi/pi-permission-system/test/handlers/gates/skill-read.test.ts
T
云服务部-叶林立andDevin <158243242+devin-ai-integration[bot]@users.noreply.github.com> 7e6a8faae1 feat(pi-permission-system): AutoReview 改为跟随主模型的对话式工具调用审查
AutoReview:

- 审查器回到 Codex Guardian 风格的对话式审查,默认跟随会话当前主模型
  (每次审查实时读取 ctx.model);provider/model 必须同时设置才固定审查
  模型,只设其一视为配置错误并转人工。
- 结论只通过 approve / reject 两个工具给出,参数仅 reason;不强制
  toolChoice,正文文本(即使是合法 JSON)一律忽略。
- approve 自动放行;reject 转人工并在弹窗显示理由(标注模型生成)。
  超时、Provider 错误、截断、未调用工具、多次调用、参数无效等异常同样
  转人工,并附固定的系统说明;用户按 Esc 中止时直接拒绝。
- 模型调用改为 modelRegistry.streamSimple(...).result():Pi 1.3.0 面向
  扩展的 ModelRegistry 没有 completeSimple,原写法在真实环境中每次都会
  转人工。
- 对话记录保留可信度标注;工具结果在有更短的 pi-compaction 摘要时替换
  为摘要;token 估算改为非 ASCII 字符按 1 token 计;预算可配置。
- 同一条用户消息内完全相同的请求只缓存 approve 结论。
- 默认 reasoning low、总超时 30 秒、最多 2 次尝试、maxOutputTokens 4000。

删除:

- 未发布过的 TypeSafe Jev 分类器实现、DenialCircuitBreaker。
- 整个 annotator 机制(model-explanation、注册表、registerAnnotator、
  provider-timeout);annotators 配置键保留为已弃用并忽略,避免旧配置
  被拒绝。

测试:

- 新增 reviewer / transcript / 弹窗说明 / 默认配置测试;清理重复、恒真
  或只测实现细节的旧测试。
- 使用 deepseek-v4.1-flash 实测 5 个场景各 5 次,25 次结论全部符合预期。

Generated with [Devin](https://devin.ai)

Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
2026-10-19 01:54:17 -07:00

155 lines
5.2 KiB
TypeScript

import { describe, expect, it, vi } from "vitest";
import { describeSkillReadGate } from "#src/handlers/gates/skill-read";
import type { ToolCallContext } from "#src/handlers/gates/types";
import { posixPathFlavor } from "#src/path/path-flavor";
import { PathNormalizer } from "#src/path-normalizer";
import type { SkillPromptEntry } from "#src/skill-prompt-sanitizer";
// All test tccs use cwd "/test/project"; one normalizer serves every call.
const normalizer = new PathNormalizer(posixPathFlavor, "/test/project");
// ── SDK stubs ──────────────────────────────────────────────────────────────
vi.mock("@earendil-works/pi-coding-agent", async (importOriginal) => {
const original =
await importOriginal<typeof import("@earendil-works/pi-coding-agent")>();
return { ...original };
});
// ── helpers ────────────────────────────────────────────────────────────────
function makeSkillEntry(
overrides: Partial<SkillPromptEntry> = {},
): SkillPromptEntry {
return {
name: "librarian",
description: "Research skills",
location: "/skills/librarian/SKILL.md",
state: "ask",
normalizedLocation: "/skills/librarian/SKILL.md",
normalizedBaseDir: "/skills/librarian",
...overrides,
};
}
function makeTcc(overrides: Partial<ToolCallContext> = {}): ToolCallContext {
return {
toolName: "read",
agentName: null,
input: { path: "/skills/librarian/SKILL.md" },
toolCallId: "tc-1",
cwd: "/test/project",
...overrides,
};
}
// ── tests ──────────────────────────────────────────────────────────────────
describe("describeSkillReadGate", () => {
it("returns null when tool is not read", () => {
const result = describeSkillReadGate(
makeTcc({ toolName: "write" }),
normalizer,
() => [makeSkillEntry()],
);
expect(result).toBeNull();
});
it("returns null when no active skill entries", () => {
const result = describeSkillReadGate(makeTcc(), normalizer, () => []);
expect(result).toBeNull();
});
it("returns null when read path does not match any skill", () => {
const result = describeSkillReadGate(
makeTcc({ input: { path: "/test/project/src/index.ts" } }),
normalizer,
() => [makeSkillEntry()],
);
expect(result).toBeNull();
});
it("returns null when input has no path", () => {
const result = describeSkillReadGate(
makeTcc({ input: {} }),
normalizer,
() => [makeSkillEntry()],
);
expect(result).toBeNull();
});
it.each(["ask", "allow", "deny"] as const)(
"returns GateDescriptor with preResolved.state matching skill entry state (%s)",
(state) => {
const result = describeSkillReadGate(makeTcc(), normalizer, () => [
makeSkillEntry({ state }),
]);
expect(result).not.toBeNull();
const desc = result!;
expect(desc.preResolved).toEqual({ state });
},
);
it("decision surface is 'skill' and decision value is the skill name", () => {
const result = describeSkillReadGate(makeTcc(), normalizer, () => [
makeSkillEntry({ name: "my-skill" }),
])!;
expect(result.decision.surface).toBe("skill");
expect(result.decision.value).toBe("my-skill");
});
it("carries the skill name as single-value access facts on promptDetails", () => {
const result = describeSkillReadGate(makeTcc(), normalizer, () => [
makeSkillEntry({ name: "my-skill" }),
])!;
expect(result.promptDetails.accessIntent).toEqual({
surface: "skill",
matchValues: ["my-skill"],
boundaryValue: null,
});
});
it("payload contains the skill name and the path it was reached through", () => {
const result = describeSkillReadGate(makeTcc(), normalizer, () => [
makeSkillEntry({ name: "librarian" }),
])!;
expect(result.payload.kind).toBe("skill_read");
expect(result.payload.request.value).toBe("librarian");
expect(result.payload.request.requester.agentName).toBeNull();
expect(result.payload.evidence).toEqual([
{
label: "read path",
text: "/skills/librarian/SKILL.md",
detail: null,
},
]);
});
it("promptDetails includes skill_read source and skillName", () => {
const result = describeSkillReadGate(
makeTcc({ agentName: "test-agent", toolCallId: "tc-42" }),
normalizer,
() => [makeSkillEntry({ name: "my-skill" })],
)!;
expect(result.promptDetails).toMatchObject({
source: "skill_read",
agentName: "test-agent",
toolCallId: "tc-42",
toolName: "read",
skillName: "my-skill",
});
});
it("logContext includes skill_read source and skillName", () => {
const result = describeSkillReadGate(
makeTcc({ agentName: "agent-1" }),
normalizer,
() => [makeSkillEntry({ name: "librarian" })],
)!;
expect(result.logContext).toMatchObject({
source: "skill_read",
skillName: "librarian",
agentName: "agent-1",
});
});
});