mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-01 23:07:30 +00:00
4e70b8d5d8
批 1，不含 SDK 仓与目录改名。 - go.mod 第 1 行 module 路径 - 366 个 .go 文件的完整前缀 github.com/JianFeeeee/HomeAgent 机械替换 （857 处，仅完整前缀；展示名/注释里的 HomeAgent 字样留给批 5） - 未改：third_party 目录名、SDK 自身 go.mod module、go.mod replace 目标、 homed/waiter 可执行名、.hmap 后缀
66 lines
2.3 KiB
Go
66 lines
2.3 KiB
Go
package core
|
||
|
||
import (
|
||
"testing"
|
||
|
||
sdk "github.com/JianFeeeee/HologramAgent/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)
|
||
}
|
||
}
|