fix(memory): 场景键两路合并去重(现网日志实测 scenes=[chan:qq chan:qq])

现网 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。
This commit is contained in:
JianFeeeee
2026-09-26 21:55:12 +08:00
parent 13a070fbb2
commit 2567a22be5
4 changed files with 105 additions and 7 deletions

View File

@ -65,6 +65,35 @@ func (a *Agent) sceneKeysFor(evt *agentIO.InputEvent, toolName string) []string
return keys 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 是一次记忆操作(取进来 / 踢出去)的结果。 // memoryPassOut 是一次记忆操作(取进来 / 踢出去)的结果。
type memoryPassOut struct { type memoryPassOut struct {
// Archived 是被归档进文档记忆的低相关 L0 事件数(prune 的输出)。 // Archived 是被归档进文档记忆的低相关 L0 事件数(prune 的输出)。

View File

@ -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)
}
}

View File

@ -790,9 +790,7 @@ func (a *Agent) stepToolAfter(f *TaskFrame) stepOutcome {
// 召回用两条路的并集:声明场景(注入点/通道/工具)+ 涌现场景 // 召回用两条路的并集:声明场景(注入点/通道/工具)+ 涌现场景
scenes := a.sceneKeysFor(f.Evt, tc.Name) scenes := a.sceneKeysFor(f.Evt, tc.Name)
turn := a.resolveTurnScenes(f, tc.Name) turn := a.resolveTurnScenes(f, tc.Name)
for _, k := range turn.Keys { scenes = mergeSceneKeys(scenes, turn.Keys)
scenes = append(scenes, k)
}
recallText = a.memoryPass(query, "tool:"+tc.Name, needPrune, needRecall, scenes).RecallText recallText = a.memoryPass(query, "tool:"+tc.Name, needPrune, needRecall, scenes).RecallText
} }
} }

View File

@ -63,11 +63,9 @@ func (a *Agent) buildTaskMemoryContext(f *TaskFrame, input string, maxTokens int
if f.Evt != nil && f.Evt.Source != "" { if f.Evt != nil && f.Evt.Source != "" {
trigger = "input:" + f.Evt.Source trigger = "input:" + f.Evt.Source
} }
// 场景集合 = 声明(主动)+ 涌现(被动)两条路的并集。 // 场景集合 = 声明(主动)+ 涌现(被动)两条路的并集(去重)。
turn := a.resolveTurnScenes(f, "") turn := a.resolveTurnScenes(f, "")
for _, k := range turn.Keys { scenes = mergeSceneKeys(scenes, turn.Keys)
scenes = append(scenes, k)
}
return a.recallText(query, trigger, maxTokens, scenes) return a.recallText(query, trigger, maxTokens, scenes)
} }