mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-03 07:47:45 +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 缺失。
109 lines
3.0 KiB
Go
109 lines
3.0 KiB
Go
package sdk
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
"sync"
|
||
|
||
"github.com/JianFeeeee/HologramAgent/internal/knowledge"
|
||
"github.com/JianFeeeee/HologramAgent/internal/memory"
|
||
coldcontext "github.com/JianFeeeee/HologramAgent/internal/memory/coldcontext"
|
||
"github.com/JianFeeeee/HologramAgent/internal/memory/text"
|
||
)
|
||
|
||
// VirtualInstance 是完全隔离的虚拟存储集合，供内置插件（如 healthcheck）
|
||
// 做不污染生产存储的"写→查→删"往返自检。所有写入发生在独立临时目录，
|
||
// 由 Cleanup 统一销毁。
|
||
type VirtualInstance struct {
|
||
Memory MemoryAPI
|
||
Knowledge KnowledgeAPI
|
||
DocMemory DocMemoryAPI
|
||
TextMemory TextMemoryAPI
|
||
|
||
dir string
|
||
mu sync.Mutex
|
||
created bool
|
||
}
|
||
|
||
// newBaseDir 返回一个隔离的临时根目录（hot 前缀，避免与生产数据混淆）。
|
||
func newBaseDir(scope string) (string, error) {
|
||
return os.MkdirTemp("", "homeagent_selftest_"+scope+"_")
|
||
}
|
||
|
||
// NewVirtualInstance 创建完全隔离的测试实例，使用独立临时目录。
|
||
func NewVirtualInstance(scope string) (*VirtualInstance, error) {
|
||
if scope == "" {
|
||
scope = "hc"
|
||
}
|
||
base, err := newBaseDir(scope)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
v := &VirtualInstance{dir: base, created: true}
|
||
if err := v.initLocked(); err != nil {
|
||
os.RemoveAll(base)
|
||
return nil, err
|
||
}
|
||
return v, nil
|
||
}
|
||
|
||
// initLocked 初始化各隔离存储。调用方须持 v.mu。
|
||
func (v *VirtualInstance) initLocked() error {
|
||
memDB, err := memory.NewGraphDB(filepath.Join(v.dir, "graph.db"))
|
||
if err != nil {
|
||
return fmt.Errorf("virtual graph db: %w", err)
|
||
}
|
||
v.Memory = NewGraphMemory(memDB)
|
||
|
||
ks := knowledge.NewStore(filepath.Join(v.dir, "knowledge"))
|
||
if err := ks.Start(); err != nil {
|
||
return fmt.Errorf("virtual knowledge store: %w", err)
|
||
}
|
||
v.Knowledge = NewKnowledge(ks)
|
||
|
||
ds := coldcontext.NewStore(filepath.Join(v.dir, "documents"), memory.TokenizeWords)
|
||
if err := ds.Start(); err != nil {
|
||
return fmt.Errorf("virtual doc store: %w", err)
|
||
}
|
||
v.DocMemory = NewDocMemory(ds)
|
||
|
||
tm := text.New(filepath.Join(v.dir, "text"))
|
||
if err := tm.Start(); err != nil {
|
||
return fmt.Errorf("virtual text memory: %w", err)
|
||
}
|
||
v.TextMemory = NewTextMemory(tm)
|
||
return nil
|
||
}
|
||
|
||
// Cleanup 关闭并销毁该虚拟实例的全部临时存储。之后实例不可再用。
|
||
func (v *VirtualInstance) Cleanup() {
|
||
v.mu.Lock()
|
||
defer v.mu.Unlock()
|
||
if !v.created {
|
||
return
|
||
}
|
||
if tm, ok := v.TextMemory.(*textMemoryImpl); ok && tm.tm != nil {
|
||
tm.tm.Stop()
|
||
}
|
||
if ds, ok := v.DocMemory.(*docMemoryImpl); ok && ds.ds != nil {
|
||
ds.ds.Stop()
|
||
}
|
||
if ks, ok := v.Knowledge.(*knowledgeImpl); ok && ks.ks != nil {
|
||
ks.ks.Stop()
|
||
}
|
||
if gm, ok := v.Memory.(*graphMemory); ok && gm.db != nil {
|
||
gm.db.Close()
|
||
}
|
||
os.RemoveAll(v.dir)
|
||
v.created = false
|
||
}
|
||
|
||
// Reset 清理当前实例并重建一个全新的隔离实例，用于每轮自检前重置状态。
|
||
// 返回新实例；失败时旧实例已被清理、返回 err，调用方需重新创建。
|
||
func (v *VirtualInstance) Reset(scope string) (*VirtualInstance, error) {
|
||
v.Cleanup()
|
||
return NewVirtualInstance(scope)
|
||
}
|