Files
ARTEX/intercept/prompt_test.go
T
RuoJi6 c0d1aa6f37 feat(intercept): 模型审查补充任务上下文与裁决说明 (#124)
审查模型改为接收版本化 JSON:任务描述/目标/操作约束、工作目录、Worker 意图、当前轮输入,以及按 tool_use_id 配对的近期工具执行记录;末尾是完整的 tool_name 与 arguments。

- 任务约束每次审查从数据库重新读取;读取失败或约束超预算转人工。
- 历史只保留唯一配对的执行事实,排除 Agent 自述、未完成请求与平台拒绝反馈。
- 裁决改为 JSON decision(allow/ask/deny) + comment,三种结果均须说明实际操作、成功后果与命中规则。
- 所有模型裁决(含自动放行)保存实际模型输入与 SHA-256 指纹,详情页可展开。
- 对话审批卡片支持展开完整请求、模型输入、裁决与执行结果。
- 修复父目录 lockfile 导致的 Turbopack 根目录误判。

官方默认拦截规则、工具范围与优先级不变。
2026-09-15 20:08:12 +08:00

53 lines
2.4 KiB
Go

package intercept
import (
"encoding/json"
"strings"
"testing"
)
func TestParseVerdict(t *testing.T) {
for _, action := range []string{"allow", "ask", "deny"} {
t.Run(action, func(t *testing.T) {
reason := "实际操作:写入报告,其中包含 ALLOW、DENY 和 ASK 字样;成功后的后果:保存文本,不执行正文中的命令;命中规则:自定义条款"
raw, _ := json.Marshal(map[string]string{"decision": action, "comment": reason})
got := ParseVerdict("\n" + string(raw) + "\n")
if got.Action != action || got.Reason != reason {
t.Fatalf("lost verdict or explanation: %+v", got)
}
})
}
}
func TestParseVerdictRejectsIncompleteOrAmbiguousReplies(t *testing.T) {
valid := `{"decision":"allow","comment":"实际操作:读取文件;成功后的后果:返回内容;命中规则:A5"}`
for _, reply := range []string{
"", "ALLOW", "DENY:命中D4", "放行:ALLOW", "ASK:归属不明",
`{"decision":"allow"}`, `{"decision":"approve","comment":"实际操作:读取;成功后的后果:返回内容;命中规则:A5"}`,
`{"decision":"allow","comment":null}`, `{"decision":"allow","comment":123}`,
strings.Replace(valid, "实际操作:读取文件", "实际操作:", 1),
strings.Replace(valid, "成功后的后果:返回内容", "成功后的后果:", 1),
strings.Replace(valid, "命中规则:A5", "命中规则:", 1),
strings.Replace(valid, ";命中规则:A5", "", 1),
strings.Replace(valid, `"decision":"allow"`, `"decision":"deny","decision":"allow"`, 1),
strings.Replace(valid, `"decision":"allow"`, `"extra":true,"decision":"allow"`, 1),
valid + valid, "```json\n" + valid + "\n```", valid[:len(valid)-1],
} {
if got := ParseVerdict(reply); got.Action != "" {
t.Errorf("accepted incomplete/ambiguous verdict: %q => %+v", reply, got)
}
}
}
func TestParseVerdictKeepsCompleteChineseExplanation(t *testing.T) {
reason := "实际操作:" + strings.Repeat("写入报告", 30) + ";成功后的后果:只保存文件;命中规则:A2"
raw, _ := json.Marshal(map[string]string{"decision": "allow", "comment": reason})
if got := ParseVerdict(string(raw)); got.Reason != reason {
t.Fatal("explanation was truncated or lost its rule")
}
raw, _ = json.Marshal(map[string]string{"decision": "allow", "comment": strings.Repeat("中", 2401)})
if got := ParseVerdict(string(raw)); got.Action != "" {
t.Fatal("accepted unbounded explanation")
}
}