mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-21 17:38:10 +00:00
b393b7072c
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 只会误导。
66 lines
2.3 KiB
Go
66 lines
2.3 KiB
Go
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)
|
||
}
|
||
}
|