mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-01 06:47:29 +00:00
b850feaef2
记忆层： - MemoryBlock 承载 FrameTS/PageSeq/PageKind/BlockRole/CallID - coldcontext 归档与回填按完整页传递，工具调用参数和配对键不丢失 - 召回先估算总 token，超预算时按全局页相似度标记 inactive - inactive 仅是 context 渲染期状态，不落盘、不进入 coldcontext，Prune 后重置 - Peek/确认/Remove 两阶段召回，失败保留冷上下文副本 - stablemem 记录 last_recalled_at，删块连带清理边 命名： - internal/memory/document → internal/memory/coldcontext - Doc/DocEvent/DocHit → ColdDoc/ColdEvent/ColdHit - ContextToColdContext/RecentColdContexts/FindColdContexts - 保留既有 JSON 标签、doc_ 文件前缀与 doc_count 状态键，兼容存量数据 安全治理： - 工具策略门禁、安全配置、事件总线发布者身份与入站脱敏 - 声明式 SecurityAPI 与 QQ owner/guest profile - 日志保留策略与相关 e2e/回归判据 验证：go build ./...、go vet ./...、go test ./...（62 包）； 生产副本 e2e：7013 块迁移后 ID/向量/指纹 0 变化、0 缺失。
108 lines
4.0 KiB
Go
108 lines
4.0 KiB
Go
package core
|
||
|
||
import (
|
||
"path/filepath"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/JianFeeeee/HologramAgent/internal/memory"
|
||
"github.com/JianFeeeee/HologramAgent/internal/memory/coldcontext"
|
||
)
|
||
|
||
// memoryPass 是「裁剪」与「召回」的唯一入口：两根正交轴，但共用同一份 query。
|
||
//
|
||
// 这一组测试锁死三件事：
|
||
// 1. 两个策略都不声明时是 no-op（不裁剪、不召回）；
|
||
// 2. 同时声明时一次调用同时产出「归档数」与「召回文本」；
|
||
// 3. 单一策略只产出对应的那一个输出（正交，不互相触发）。
|
||
func TestMemoryPass_NoPolicyIsNoOp(t *testing.T) {
|
||
a := &Agent{
|
||
context: newPruneableContext(15),
|
||
maxContextSize: 4,
|
||
indexer: newTestIndexer(t, "咖啡", "张三"),
|
||
}
|
||
out := a.memoryPass("咖啡", "test", false, false, nil)
|
||
if out.Archived != 0 || out.RecallText != "" {
|
||
t.Fatalf("未声明任何策略时不应有任何输出，实际 %+v", out)
|
||
}
|
||
}
|
||
|
||
func TestMemoryPass_PruneAndRecallTogether(t *testing.T) {
|
||
a := newMemoryPassAgent(t)
|
||
before := a.context.Len()
|
||
out := a.memoryPass("咖啡", "tool:test", true, true, nil)
|
||
if out.Archived == 0 {
|
||
t.Fatal("声明 prune 应归档低相关事件")
|
||
}
|
||
if a.context.Len() >= before {
|
||
t.Fatalf("裁剪后上下文应变短：%d → %d", before, a.context.Len())
|
||
}
|
||
if !strings.Contains(out.RecallText, "【记忆索引】") {
|
||
t.Fatalf("声明 recall 应产出记忆索引文本，实际 %q", out.RecallText)
|
||
}
|
||
}
|
||
|
||
func TestMemoryPass_PoliciesAreOrthogonal(t *testing.T) {
|
||
// 只裁不召回：输出只有归档数。
|
||
onlyPrune := &Agent{
|
||
context: newPruneableContext(15),
|
||
maxContextSize: 4,
|
||
indexer: newTestIndexer(t, "咖啡", "张三"),
|
||
}
|
||
if out := onlyPrune.memoryPass("咖啡", "test", true, false, nil); out.RecallText != "" {
|
||
t.Fatalf("只声明 prune 不应召回，实际 %q", out.RecallText)
|
||
}
|
||
// 只召回不裁剪：输出只有召回文本，上下文条数不变。
|
||
onlyRecall := &Agent{
|
||
context: newPruneableContext(15),
|
||
maxContextSize: 4,
|
||
indexer: newTestIndexer(t, "咖啡", "张三"),
|
||
}
|
||
before := onlyRecall.context.Len()
|
||
out := onlyRecall.memoryPass("咖啡", "test", false, true, nil)
|
||
if out.Archived != 0 {
|
||
t.Fatalf("只声明 recall 不应裁剪，实际归档 %d", out.Archived)
|
||
}
|
||
if onlyRecall.context.Len() != before {
|
||
t.Fatalf("只声明 recall 不应改变上下文条数：%d → %d", before, onlyRecall.context.Len())
|
||
}
|
||
}
|
||
|
||
// 输入侧召回必须用**清洗后**的 query（通道 Cleaner 的输出），与裁剪侧一致。
|
||
// 用无关原文 + 命中清洗文本做区分，锁死「用的是 CleanInput 而不是 Input」。
|
||
func TestBuildTaskMemoryContext_UsesCleanInput(t *testing.T) {
|
||
a := &Agent{indexer: newTestIndexer(t, "咖啡", "张三")}
|
||
|
||
// CleanInput 命中实体、原文完全不相关 → 应召回（证明用了清洗文本）。
|
||
f := &TaskFrame{Evt: nil, CleanInput: "咖啡"}
|
||
if got := a.buildTaskMemoryContext(f, "zzz", 0); !strings.Contains(got, "【记忆索引】") {
|
||
t.Fatalf("应据清洗后的 query 召回，实际 %q", got)
|
||
}
|
||
// 清洗为空 → 回退原文；原文无关则不召回。
|
||
f2 := &TaskFrame{CleanInput: ""}
|
||
if got := a.buildTaskMemoryContext(f2, "zzz", 0); got != "" {
|
||
t.Fatalf("清洗为空且原文无关时不应召回，实际 %q", got)
|
||
}
|
||
}
|
||
|
||
// newMemoryPassAgent 造一个同时能做裁剪与召回的 agent（含 doc 记忆落点）。
|
||
func newMemoryPassAgent(t *testing.T) *Agent {
|
||
t.Helper()
|
||
return &Agent{
|
||
context: newPruneableContext(15),
|
||
maxContextSize: 4,
|
||
indexer: newTestIndexer(t, "咖啡", "张三"),
|
||
coldcontextStore: coldcontext.NewStore(filepath.Join(t.TempDir(), "docs"), memory.TokenizeWords),
|
||
}
|
||
}
|
||
|
||
// newPruneableContext 造 n 条可被裁剪的上下文（最近 10 条受保护）。
|
||
func newPruneableContext(n int) *RelevanceContext {
|
||
ctx := NewRelevanceContext("", memory.NewStaticEmbedder(""))
|
||
for i := 0; i < n; i++ {
|
||
ctx.Append(ContextEvent{Timestamp: time.Now(), Source: "user", Input: "事件内容"})
|
||
}
|
||
return ctx
|
||
}
|