100 lines
3.8 KiB
Go
100 lines
3.8 KiB
Go
// Package models 定义织忆 MemoryWeave 的核心数据模型。
|
||
package models
|
||
|
||
import "time"
|
||
|
||
// Time 是 time.Time 的安全版本，支持空字符串解析
|
||
type Time struct{ time.Time }
|
||
|
||
// UnmarshalJSON 实现 json.Unmarshaler：空字符串解析为当前时间（而非报错）
|
||
func (t *Time) UnmarshalJSON(data []byte) error {
|
||
s := string(data)
|
||
if s == `""` || s == `"0001-01-01T00:00:00Z"` {
|
||
t.Time = time.Time{}
|
||
return nil
|
||
}
|
||
// 去掉前后引号
|
||
if len(s) >= 2 && s[0] == '"' && s[len(s)-1] == '"' {
|
||
s = s[1 : len(s)-1]
|
||
}
|
||
if s == "" || s == "null" {
|
||
t.Time = time.Time{}
|
||
return nil
|
||
}
|
||
return t.Time.UnmarshalJSON(data)
|
||
}
|
||
|
||
// VersionRecord 记忆版本记录（溯源链中的单次修改）
|
||
type VersionRecord struct {
|
||
Version int `json:"version"`
|
||
Content string `json:"content"`
|
||
UpdatedBy string `json:"updated_by"`
|
||
Source string `json:"source"`
|
||
Trigger string `json:"trigger"`
|
||
Reason string `json:"reason"`
|
||
Timestamp time.Time `json:"timestamp"`
|
||
}
|
||
|
||
// MemoryRecord 织忆中的单条记忆（存储在 LanceDB memories 表中）。
|
||
// v3.8 完整 Schema — 23 字段，与设计文档一致。
|
||
type MemoryRecord struct {
|
||
ID string `json:"id"`
|
||
AgentID string `json:"agent_id"`
|
||
Namespace string `json:"namespace"`
|
||
Content string `json:"content"`
|
||
ContentMD5 string `json:"content_md5"` // content 的 MD5 前 12 字符，用于精确去重
|
||
Category string `json:"category"` // system_fact, user_pref, proj_context, tool_usage, code_snippet
|
||
Vector []float32 `json:"vector"` // bge-m3 1024维，L2归一化
|
||
Tier string `json:"tier"` // normal, core（core永不衰减）
|
||
Importance float64 `json:"importance"` // 重要性得分 = recency_factor × (1+log(1+recall_count))
|
||
QualityScore float64 `json:"quality_score"`
|
||
UsefulCount int `json:"useful_count"` // 被标记 useful 的次数
|
||
NotUsefulCount int `json:"not_useful_count"` // 被标记 not-useful 的次数
|
||
Version int `json:"version"` // 当前版本号
|
||
VersionHistory []VersionRecord `json:"version_history,omitempty"` // 完整修改链
|
||
Source string `json:"source"` // 来源：牧尘口头 / 配置解析 / Agent推断 / LLM蒸馏
|
||
VolatileFlag bool `json:"volatile_flag"` // 频繁变更标记（≥3次修正自动设置）
|
||
RecallCount int `json:"recall_count"`
|
||
Freshness string `json:"freshness"` // fresh, stale, verified
|
||
LastRecalledAt time.Time `json:"last_recalled_at"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
UpdatedAt time.Time `json:"updated_at"`
|
||
IsDeleted bool `json:"is_deleted"`
|
||
DependsOn []string `json:"depends_on,omitempty"` // 依赖的记忆 ID 列表
|
||
DerivedFrom string `json:"derived_from,omitempty"` // 蒸馏来源 episode ID
|
||
}
|
||
|
||
// EpisodeRecord 原始对话/任务日志（存储在 LanceDB episodes 表中）。
|
||
type EpisodeRecord struct {
|
||
ID string `json:"id"`
|
||
AgentID string `json:"agent_id"`
|
||
Namespace string `json:"namespace"`
|
||
Content string `json:"content"`
|
||
Category string `json:"category"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
}
|
||
|
||
// TombstoneRecord 已删除/废弃记忆的墓碑记录。
|
||
type TombstoneRecord struct {
|
||
ID string `json:"id"`
|
||
OriginalID string `json:"original_id"`
|
||
Reason string `json:"reason"` // deprecated, merged, corrected, evicted
|
||
CreatedAt time.Time `json:"created_at"`
|
||
}
|
||
|
||
// RecallResult recall 接口返回的单条结果。
|
||
type RecallResult struct {
|
||
ID string `json:"id"`
|
||
Content string `json:"content"`
|
||
Category string `json:"category"`
|
||
Score float64 `json:"score"`
|
||
Timestamp string `json:"timestamp"`
|
||
}
|
||
|
||
// RerankResult rerank 重排后的单条结果。
|
||
type RerankResult struct {
|
||
Index int `json:"index"`
|
||
Score float64 `json:"score"`
|
||
Text string `json:"text"`
|
||
}
|