From 2567a22be534b8cf3429908116879be575e46ed3 Mon Sep 17 00:00:00 2001 From: JianFeeeee Date: Sat, 26 Sep 2026 21:55:12 +0800 Subject: [PATCH] =?UTF-8?q?fix(memory):=20=E5=9C=BA=E6=99=AF=E9=94=AE?= =?UTF-8?q?=E4=B8=A4=E8=B7=AF=E5=90=88=E5=B9=B6=E5=8E=BB=E9=87=8D=EF=BC=88?= =?UTF-8?q?=E7=8E=B0=E7=BD=91=E6=97=A5=E5=BF=97=E5=AE=9E=E6=B5=8B=20scenes?= =?UTF-8?q?=3D[chan:qq=20chan:qq]=EF=BC=89?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 现网 21:49 的 QQ 轮次日志打出 `scenes=[chan:qq chan:qq]` —— 同一键 出现两次。查因：tooldefs.go:63 与 task.go:793 是同一段拼接写法， scenes := a.sceneKeysFor(...) // 内部有 seen 去重 turn := a.resolveTurnScenes(...) for _, k := range turn.Keys { scenes = append(scenes, k) // ← 两路之间没有共同的 seen } 而声明路与通道派生路都会产出 chan:qq（既是插件声明的、也是从 evt.Source 派生的），于是重复。 功能上无害（RecallByScene 内部会再去重），但有两个实际代价：日志里 的 scenes=[...] 误导排查——会让人以为场景集合本身有问题；以及每次 白走一遍前缀匹配。 修法：抽 mergeSceneKeys 共用函数（顺带消掉两处重复代码），两处调用点 都走它。判据 scenemerge_test.go 5 例：3 个合并场景（同名 / 归一化后 同名 / 多路重复）+ scene_policy=none 时两路皆空（防「声明路关了但 涌现路还开着」的半开状态）。 记忆 8 包 + agent/core 全绿，8 包齐全、无 FAIL/panic/race。 --- internal/agent/core/memorypass.go | 29 ++++++++++ internal/agent/core/scenemerge_test.go | 73 ++++++++++++++++++++++++++ internal/agent/core/task.go | 4 +- internal/agent/core/tooldefs.go | 6 +-- 4 files changed, 105 insertions(+), 7 deletions(-) create mode 100644 internal/agent/core/scenemerge_test.go 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) }