35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
"""
|
||
按 Agent/工作流 + 会话键持久化的用户记忆（MySQL），与 Redis 热缓存配合实现长期记忆。
|
||
"""
|
||
import uuid
|
||
|
||
from sqlalchemy import Column, DateTime, String, UniqueConstraint, func
|
||
from sqlalchemy.dialects.mysql import CHAR, JSON
|
||
|
||
from app.core.database import Base
|
||
|
||
|
||
class PersistentUserMemory(Base):
|
||
"""会话级记忆快照：同一 scope 下 session_key 唯一。"""
|
||
|
||
__tablename__ = "persistent_user_memories"
|
||
__table_args__ = (
|
||
UniqueConstraint("scope_kind", "scope_id", "session_key", name="uq_persistent_mem_scope_session"),
|
||
)
|
||
|
||
id = Column(CHAR(36), primary_key=True, default=lambda: str(uuid.uuid4()), comment="主键")
|
||
scope_kind = Column(String(16), nullable=False, comment="agent 或 workflow")
|
||
scope_id = Column(CHAR(36), nullable=False, comment="Agent ID 或 Workflow ID")
|
||
session_key = Column(String(512), nullable=False, comment="调用方传入的 user_id 等会话键")
|
||
payload = Column(JSON, nullable=False, comment="与 Redis 中 user_memory_* 结构一致的记忆 JSON")
|
||
updated_at = Column(
|
||
DateTime,
|
||
server_default=func.now(),
|
||
onupdate=func.now(),
|
||
nullable=True,
|
||
comment="更新时间",
|
||
)
|
||
|
||
def __repr__(self):
|
||
return f"<PersistentUserMemory(scope={self.scope_kind}:{self.scope_id}, session={self.session_key[:32]})>"
|