mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-22 09:58:06 +00:00
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:
@ -59,6 +59,30 @@ func isContinuationPlaceholder(m agentAPI.Message) bool {
|
||||
(m.Content == continuationPlaceholder || m.Content == replyDeliveredPlaceholder)
|
||||
}
|
||||
|
||||
// toolOutputForQuery 返回用于相关性计算的工具输出**有效内容**。
|
||||
//
|
||||
// 为什么要过 Cleaner 而不是直接用原始 result:ContextPolicy=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)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
65
internal/agent/core/prunequery_test.go
Normal file
65
internal/agent/core/prunequery_test.go
Normal 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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user