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

@ -59,6 +59,30 @@ func isContinuationPlaceholder(m agentAPI.Message) bool {
(m.Content == continuationPlaceholder || m.Content == replyDeliveredPlaceholder)
}
// toolOutputForQuery 返回用于相关性计算的工具输出**有效内容**。
//
// 为什么要过 Cleaner 而不是直接用原始 resultContextPolicy=prune 的入参是
// **相关性查询向量**——它决定保留/归档哪些上下文事件。原始工具输出里混着
// ANSI 转义、base64、JSON 包装等噪声,直接拿去向量化会让打分失真。
// 而 ToolDef.Cleaner 的契约本就写着“仅在向量化/jieba/蒸馏时调用”,裁剪正是
// 在向量化,所以这里必须过它(此前只在构建事件向量时用了,裁剪查询漏了)。
//
// Cleaner 未注册或 RPC 失败时回退原文(清洗是计算层优化,不能因此丢内容);
// 返回空串时也回退——空串会让查询向量退化成零向量,裁剪就失去判据。
func (a *Agent) toolOutputForQuery(toolName, raw string) string {
if a.stageHost == nil {
return raw
}
cleaner := a.stageHost.ToolDefCleaner(toolName)
if cleaner == nil {
return raw
}
if cleaned := cleaner(raw); cleaned != "" {
return cleaned
}
return raw
}
// dropContinuationPlaceholders 移除此前由本机制插入的 user 占位。
//
// 为什么必须移除而不仅仅是“不再追加”:`msgs` 在循环外创建、循环内只增不减,
@ -393,7 +417,9 @@ func (a *Agent) process(input string, stageCtx *sdk.StageContext) (response stri
if topK < 1 {
topK = 1
}
a.context.Prune(result, topK, a.docStore)
// 查询向量取**清洗后**的有效内容否则噪声ANSI/base64/JSON
// 包装)会把相关性打分带偏,裁掉本该保留的事件。
a.context.Prune(a.toolOutputForQuery(tc.Name, result), topK, a.docStore)
}
}

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)
}
}