fix(ctx): prune 的查询向量改用插件 Cleaner 清洗后的有效内容(§13.8)

ContextPolicy=prune 上线时直接把**原始**工具结果传给 RelevanceContext.Prune,
而 Prune 的入参是**相关性查询向量**——它决定保留/归档哪些上下文事件。于是
ANSI 转义、base64、JSON 包装等噪声全被编进查询向量,打分失真,裁掉本该
保留的事件。

而 ToolDef.Cleaner 的契约本就写着「仅在向量化/jieba/蒸馏时调用」,裁剪正是
在向量化——所以这是**回归契约**,不是新增能力。此前只在构建事件向量
(context.go 的 toolOutputClean)时用了 Cleaner,裁剪查询这一处漏了。

回退规则(Cleaner 是计算层优化,不能因它失效而丢内容):
- 未注册 Cleaner → 原文
- RPC 失败 → 原文(proc 侧 cleanerProxy 已有此保证)
- 返回空串 → 原文(空串会让查询向量退化成零向量,所有事件相关性相同,
  等于随机裁剪)

验证:TestToolOutputForQueryAppliesCleaner(Cleaner 被调用恰好一次且用其
结果;无 Cleaner / nil stageHost 回退原文)、
TestToolOutputForQueryEmptyCleanFallsBack。

顺带把 §13.13 第 5 条(反向大结果)按核实结论结掉为「不做」:核实发现根本
不存在 llm.chat(llm.* 只映射切换 LLM 源),唯一可能返回大结果的 doc.query
没有任何外部插件使用且已被 CapDocMemory 能力门限制。留成永久 TODO 只会误导。
This commit is contained in:
JianFeeeee
2026-09-10 23:59:12 +08:00
parent 50ba4a64cb
commit b393b7072c
3 changed files with 114 additions and 5 deletions

View File

@ -0,0 +1,65 @@
package core
import (
"testing"
sdk "gitcode.com/JianFeeeee/HomeAgent/internal/sdk"
)
// ContextPolicy=prune 的查询向量必须取**清洗后**的有效内容。
//
// 裁剪的入参是相关性查询向量,它决定保留/归档哪些上下文事件。原始工具输出里
// 混着 ANSI 转义、base64、JSON 包装等噪声,直接向量化会让打分失真,裁掉本该
// 保留的事件。ToolDef.Cleaner 的契约本就写着「仅在向量化/jieba/蒸馏时调用」,
// 裁剪正是在向量化——此前只在构建事件向量时用了它,裁剪查询漏了。
func TestToolOutputForQueryAppliesCleaner(t *testing.T) {
host := NewStageHost()
called := 0
if err := host.RegisterTool("demo_tool", sdk.ToolDef{
Name: "demo_tool",
Cleaner: func(s string) string {
called++
return "cleaned:" + s
},
}, func(map[string]interface{}) (interface{}, error) { return nil, nil }); err != nil {
t.Fatalf("RegisterTool: %v", err)
}
a := &Agent{stageHost: host}
raw := "\x1b[31mresult\x1b[0m"
got := a.toolOutputForQuery("demo_tool", raw)
if called != 1 {
t.Fatalf("Cleaner 应被调用恰好一次,实际 %d", called)
}
if got != "cleaned:"+raw {
t.Fatalf("查询应使用清洗结果,实际 %q", got)
}
// 未注册 Cleaner 的工具:回退原文。
if got := a.toolOutputForQuery("no_such_tool", raw); got != raw {
t.Fatalf("无 Cleaner 应回退原文,实际 %q", got)
}
// 无 StageHost如裸 Agent不能 panic回退原文。
if got := (&Agent{}).toolOutputForQuery("demo_tool", raw); got != raw {
t.Fatalf("nil stageHost 应回退原文,实际 %q", got)
}
}
// Cleaner 返回空串时必须回退原文:空串会让查询向量退化成零向量,
// 所有事件相关性相同,裁剪就失去判据(等于随机裁)。
func TestToolOutputForQueryEmptyCleanFallsBack(t *testing.T) {
host := NewStageHost()
if err := host.RegisterTool("t", sdk.ToolDef{
Name: "t",
Cleaner: func(string) string { return "" },
}, func(map[string]interface{}) (interface{}, error) { return nil, nil }); err != nil {
t.Fatalf("RegisterTool: %v", err)
}
a := &Agent{stageHost: host}
if got := a.toolOutputForQuery("t", "raw"); got != "raw" {
t.Fatalf("Cleaner 返回空应回退原文,实际 %q", got)
}
}