mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-27 12:53:35 +00:00
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:
@ -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 的输出)。
|
||||||
|
|||||||
73
internal/agent/core/scenemerge_test.go
Normal file
73
internal/agent/core/scenemerge_test.go
Normal 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)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -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
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -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)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user