dev 分支暂存

2026-05-16 08:26:56 +08:00
parent 58c8caa570
commit eb4129176c
71 changed files with 8474 additions and 214 deletions
@@ -0,0 +1,68 @@
+package model
+
+import "time"
+
+// MemoryPriority 记忆优先级
+type MemoryPriority int
+
+const (
+	MemoryTemp     MemoryPriority = 0 // 临时记忆 (会话内)
+	MemoryNormal   MemoryPriority = 1 // 普通记忆
+	MemoryImportant MemoryPriority = 2 // 重要记忆
+	MemoryCore     MemoryPriority = 3 // 核心记忆 (永远保留)
+)
+
+// String 返回优先级的中文描述
+func (p MemoryPriority) String() string {
+	switch p {
+	case MemoryCore:
+		return "核心"
+	case MemoryImportant:
+		return "重要"
+	case MemoryNormal:
+		return "普通"
+	case MemoryTemp:
+		return "临时"
+	default:
+		return "未知"
+	}
+}
+
+// MemoryCategory 记忆分类
+type MemoryCategory string
+
+const (
+	CategoryPreference  MemoryCategory = "preference"   // 喜好/偏好
+	CategoryFact        MemoryCategory = "fact"         // 事实信息
+	CategoryEvent       MemoryCategory = "event"        // 事件/经历
+	CategoryRelationship MemoryCategory = "relationship" // 关系/情感
+	CategoryHabit       MemoryCategory = "habit"        // 习惯
+	CategoryOther       MemoryCategory = "other"        // 其他
+)
+
+// MemoryEntry 记忆条目
+type MemoryEntry struct {
+	ID        string          `json:"id" db:"id"`
+	UserID    string          `json:"user_id" db:"user_id"`
+	Content   string          `json:"content" db:"content"`
+	Summary   string          `json:"summary" db:"summary"`   // 简短摘要
+	Category  MemoryCategory  `json:"category" db:"category"`
+	Priority  MemoryPriority  `json:"priority" db:"priority"`
+	SessionID string          `json:"session_id" db:"session_id"` // 来源会话
+	Source    string          `json:"source" db:"source"`         // 来源文本片断
+	Embedding []float32       `json:"-" db:"embedding"`           // 向量 (pgvector)
+	AccessCount int           `json:"access_count" db:"access_count"`
+	LastAccess  time.Time     `json:"last_access" db:"last_access"`
+	CreatedAt   time.Time     `json:"created_at" db:"created_at"`
+	ExpiresAt   *time.Time    `json:"expires_at,omitempty" db:"expires_at"` // 临时记忆过期时间
+}
+
+// MemoryQuery 记忆查询参数
+type MemoryQuery struct {
+	UserID   string
+	Query    string   // 查询文本
+	Category MemoryCategory
+	Priority MemoryPriority
+	Limit    int
+	Offset   int
+}