feat(memory): P1 长期记忆升级 —— 异步攒批 Consolidate + 软删 + importance/last_seen #1
@@ -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 字段全导出(牵连几十处调用点)。
|
||||
//
|
||||
// 不持久化 fatalErr(error,transient——中断点处必为 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+Unmarshaler,Eino 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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user