diff --git a/internal/agent/core/memorypass.go b/internal/agent/core/memorypass.go index e20179c..84a299c 100644 --- a/internal/agent/core/memorypass.go +++ b/internal/agent/core/memorypass.go @@ -65,6 +65,35 @@ func (a *Agent) sceneKeysFor(evt *agentIO.InputEvent, toolName string) []string return keys } +// mergeSceneKeys 把「声明路」与「涌现场景」两路合并成一个**无重复**的场景集合。 +// +// 为什么需要它:两路各自都去重过(sceneKeysFor 内部有 seen、resolveTurnScenes +// 内部也有),但**两路之间**没有共同的 seen。而声明路与通道派生路会产出 +// 同一个键(chan:qq 既是声明的、也是从 evt.Source 派生的)——现网日志实测到 +// `scenes=[chan:qq chan:qq]`。 +// +// 功能上 RecallByScene 内部会再去重,所以这不是 bug,但有两个实际代价: +// 日志里的 scenes=[...] 会误导排查;每次白走一遍前缀匹配。 +func mergeSceneKeys(declared, emergent []string) []string { + out := make([]string, 0, len(declared)+len(emergent)) + seen := make(map[string]bool, len(declared)+len(emergent)) + for _, k := range declared { + if k == "" || seen[k] { + continue + } + seen[k] = true + out = append(out, k) + } + for _, k := range emergent { + if k == "" || seen[k] { + continue + } + seen[k] = true + out = append(out, k) + } + return out +} + // memoryPassOut 是一次记忆操作(取进来 / 踢出去)的结果。 type memoryPassOut struct { // Archived 是被归档进文档记忆的低相关 L0 事件数(prune 的输出)。 diff --git a/internal/agent/core/scenemerge_test.go b/internal/agent/core/scenemerge_test.go new file mode 100644 index 0000000..6125b8f --- /dev/null +++ b/internal/agent/core/scenemerge_test.go @@ -0,0 +1,73 @@ +package core + +// 场景键去重的判据。 +// +// 症状:现网日志出现 `scenes=[chan:qq chan:qq]` —— 同一个键出现两次。 +// 原因:buildTaskMemoryContext / stepToolAfter 都先取 sceneKeysFor(内部 +// 有 seen 去重),再把 resolveTurnScenes 的 turn.Keys 直接 append 上去, +// **两路之间没有共同的 seen 集合**。声明路和通道派生路都会产出 chan:qq。 +// +// 功能上无害(RecallByScene 内部会去重),但它有两个实际代价: +// 1. 日志里的 scenes=[...] 具有误导性——排查时会以为场景集合有问题; +// 2. 每次多带一个重复键进召回,白走一遍前缀匹配。 +// +// 判据参照物在生产代码之外:期望值是「场景集合内不得有重复键」这条 +// 不变量,直接对合并后的切片计数,不引用被测实现。 + +import ( + "testing" + + agentIO "gitcode.com/JianFeeeee/HomeAgent/internal/agent/io" +) + +// TestSceneKeysMerged_NoDuplicates 声明路与涌现路的并集不得有重复。 +// 现状下 tooldefs.go:63 与 task.go:793 都是直接 append,缺这一步。 +func TestSceneKeysMerged_NoDuplicates(t *testing.T) { + cases := []struct { + name string + declared []string + emergent []string + }{ + { + name: "涌现键与声明键同名(现网实测 chan:qq 两路都产出)", + declared: []string{"chan:qq"}, + emergent: []string{"chan:qq"}, + }, + { + name: "涌现键已归一化后与声明键同名", + declared: []string{"chan:qq"}, + emergent: []string{"chan:QQ", "chan:qq"}, + }, + { + name: "多路重复", + declared: []string{"chan:qq", "tool:qq_get_message"}, + emergent: []string{"chan:qq", "tool:qq_get_message", "auto:chan:qq"}, + }, + } + for _, tc := range cases { + t.Run(tc.name, func(t *testing.T) { + got := mergeSceneKeys(tc.declared, tc.emergent) + seen := map[string]bool{} + for _, k := range got { + if seen[k] { + t.Errorf("场景集合含重复键 %q: %v", k, got) + } + seen[k] = true + } + }) + } +} + +// TestSceneSuppressedSkipsMerge none 声明时两路都应为空, +// 不能出现「声明路被关、涌现路还在」这种半开状态。 +func TestSceneSuppressedSkipsMerge(t *testing.T) { + a := &Agent{io: agentIO.NewIOManager()} + evt := &agentIO.InputEvent{ + Source: "system", + Payload: map[string]interface{}{"scene_policy": "none"}, + } + declared := a.sceneKeysFor(evt, "") + if len(declared) != 0 { + t.Fatalf("scene_policy=none 时声明路应为空,实际 %v", declared) + } +} diff --git a/internal/agent/core/task.go b/internal/agent/core/task.go index 70e6253..c05981a 100644 --- a/internal/agent/core/task.go +++ b/internal/agent/core/task.go @@ -790,9 +790,7 @@ func (a *Agent) stepToolAfter(f *TaskFrame) stepOutcome { // 召回用两条路的并集:声明场景(注入点/通道/工具)+ 涌现场景 scenes := a.sceneKeysFor(f.Evt, tc.Name) turn := a.resolveTurnScenes(f, tc.Name) - for _, k := range turn.Keys { - scenes = append(scenes, k) - } + scenes = mergeSceneKeys(scenes, turn.Keys) recallText = a.memoryPass(query, "tool:"+tc.Name, needPrune, needRecall, scenes).RecallText } } diff --git a/internal/agent/core/tooldefs.go b/internal/agent/core/tooldefs.go index c89e7c7..c560468 100644 --- a/internal/agent/core/tooldefs.go +++ b/internal/agent/core/tooldefs.go @@ -63,11 +63,9 @@ func (a *Agent) buildTaskMemoryContext(f *TaskFrame, input string, maxTokens int if f.Evt != nil && f.Evt.Source != "" { trigger = "input:" + f.Evt.Source } - // 场景集合 = 声明(主动)+ 涌现(被动)两条路的并集。 + // 场景集合 = 声明(主动)+ 涌现(被动)两条路的并集(去重)。 turn := a.resolveTurnScenes(f, "") - for _, k := range turn.Keys { - scenes = append(scenes, k) - } + scenes = mergeSceneKeys(scenes, turn.Keys) return a.recallText(query, trigger, maxTokens, scenes) }