feat(memory): P1 长期记忆升级 —— 异步攒批 Consolidate + 软删 + importance/last_seen #1

Merged
Blizzard merged 181 commits from feat/wails3 into main 2026-07-17 01:12:32 +00:00
2 changed files with 139 additions and 0 deletions
Showing only changes of commit 5a1a994cc2 - Show all commits
@@ -0,0 +1,75 @@
package eino
import (
"encoding/json"
"github.com/cloudwego/eino/schema"
)
// board 的字段全是未导出的,而 Eino 的 checkpoint 序列化器只认导出字段——若直接让 compose
// 持久化 board,会落成空、resume 丢全部执行态。Eino 对「同时实现 json.Marshaler+Unmarshaler」
// 的类型改走自定义 JSON(见 internal/serialization checkMarshaler),故给 *board 实现一对 JSON
// 方法、映射到导出 DTO,即可把整个黑板按需持久化,而无需把 board 字段全导出(牵连几十处调用点)。
//
// 不持久化 fatalErrerrortransient——中断点处必为 nil):审批中断时图尚未失败,
// 它是单次 Invoke 内的致命错标志,跨 resume 无意义。rejected 同理在中断点为 false
// 但仍带上以求快照完整、零歧义。
type boardSnapshot struct {
UID string `json:"uid,omitempty"`
SID string `json:"sid,omitempty"`
Query string `json:"query,omitempty"`
Profile string `json:"profile,omitempty"`
History []*schema.Message `json:"history,omitempty"`
KB string `json:"kb,omitempty"`
Refs []string `json:"refs,omitempty"`
ToolOut []string `json:"tool_out,omitempty"`
Sections []reportSection `json:"sections,omitempty"`
Answer string `json:"answer,omitempty"`
AgentOut []string `json:"agent_out,omitempty"`
Rejected bool `json:"rejected,omitempty"`
}
// MarshalJSON 把黑板序列化为快照(Eino checkpoint 经此持久化整图执行态)。
func (b *board) MarshalJSON() ([]byte, error) {
return json.Marshal(boardSnapshot{
UID: b.uid,
SID: b.sid,
Query: b.query,
Profile: b.profile,
History: b.history,
KB: b.kb,
Refs: b.refs,
ToolOut: b.toolOut,
Sections: b.sections,
Answer: b.answer,
AgentOut: b.agentOut,
Rejected: b.rejected,
})
}
// UnmarshalJSON 从快照还原黑板(resume 时 Eino 据此重建中断前的执行态)。
func (b *board) UnmarshalJSON(data []byte) error {
var s boardSnapshot
if err := json.Unmarshal(data, &s); err != nil {
return err
}
b.uid = s.UID
b.sid = s.SID
b.query = s.Query
b.profile = s.Profile
b.history = s.History
b.kb = s.KB
b.refs = s.Refs
b.toolOut = s.ToolOut
b.sections = s.Sections
b.answer = s.Answer
b.agentOut = s.AgentOut
b.rejected = s.Rejected
return nil
}
// 注册 *board 的序列化类型名:Eino checkpoint 的 State 是 any,持久化时记类型名、
// resume 时据名还原具体类型再走 UnmarshalJSON。名字一经上线不可改(旧 checkpoint 据此解码)。
func init() {
schema.RegisterName[*board]("sundynix_board")
}
@@ -0,0 +1,64 @@
package eino
import (
"encoding/json"
"reflect"
"testing"
"github.com/cloudwego/eino/schema"
)
// board 必须同时实现 json.Marshaler+UnmarshalerEino checkpoint 才会走自定义 JSON 路径
// 持久化未导出字段;任一缺失都会让序列化器回退到「只存导出字段」=空 board。
var (
_ json.Marshaler = (*board)(nil)
_ json.Unmarshaler = (*board)(nil)
)
// TestBoardSnapshotRoundtrip 钉死黑板快照往返:中断点的全部执行态都能无损还原(resume 不丢状态)。
func TestBoardSnapshotRoundtrip(t *testing.T) {
orig := &board{
uid: "u42",
sid: "s7",
query: "介绍杭州西湖",
profile: "用户偏好简洁",
history: []*schema.Message{schema.UserMessage("上一轮"), schema.AssistantMessage("上一轮回答", nil)},
kb: "u42/travel",
refs: []string{"西湖十景…", "苏堤春晓…"},
toolOut: []string{"[wiki] 杭州…"},
sections: []reportSection{
{Heading: "概况", Body: "西湖位于杭州。"},
},
answer: "西湖是…",
agentOut: []string{"草稿一", "草稿二"},
// fatalErr 故意置非空:快照应丢弃它(transient),不应影响往返。
fatalErr: errRejected,
}
data, err := json.Marshal(orig)
if err != nil {
t.Fatalf("marshal: %v", err)
}
var got board
if err := json.Unmarshal(data, &got); err != nil {
t.Fatalf("unmarshal: %v", err)
}
// 持久化字段逐一比对。
if got.uid != orig.uid || got.sid != orig.sid || got.query != orig.query ||
got.profile != orig.profile || got.kb != orig.kb || got.answer != orig.answer {
t.Fatalf("标量字段未还原: %+v", got)
}
if !reflect.DeepEqual(got.refs, orig.refs) || !reflect.DeepEqual(got.toolOut, orig.toolOut) ||
!reflect.DeepEqual(got.agentOut, orig.agentOut) || !reflect.DeepEqual(got.sections, orig.sections) {
t.Fatalf("切片字段未还原: %+v", got)
}
if len(got.history) != len(orig.history) || got.history[0].Content != "上一轮" {
t.Fatalf("history 未还原: %+v", got.history)
}
// fatalErr 是 transient,快照不持久化 → 还原后必为 nil。
if got.fatalErr != nil {
t.Fatalf("fatalErr 不应被持久化,got %v", got.fatalErr)
}
}