99 lines
5.3 KiB
Python
99 lines
5.3 KiB
Python
from datetime import datetime
|
||
from typing import Any
|
||
|
||
from sqlalchemy import JSON, BigInteger, DateTime, Integer, Numeric, String, Text
|
||
from sqlalchemy.orm import Mapped, mapped_column
|
||
|
||
from app.model.base import Base
|
||
from app.model.profile import ProfileSnapshot
|
||
|
||
__all__ = [
|
||
"MemoryUnit", "MemoryEvidence", "MemorySyncOutbox", "MemoryConflict",
|
||
"ProfileSnapshot",
|
||
]
|
||
|
||
|
||
class MemoryUnit(Base):
|
||
"""中期记忆单元；字段与 docs/00 第 669-696 行基线一致。
|
||
|
||
生成列 `active_memory_key` 由数据库计算（仅 status='active' 时返回 memory_key），
|
||
ORM 不映射该列，唯一键 `(customer_id, active_memory_key)` 负责"每客户每键仅一条有效记忆"。
|
||
"""
|
||
|
||
__tablename__ = "memory_unit"
|
||
id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
|
||
memory_uuid: Mapped[str] = mapped_column(String(36), unique=True, nullable=False)
|
||
customer_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
||
memory_key: Mapped[str] = mapped_column(String(128), nullable=False)
|
||
content: Mapped[str] = mapped_column(Text, nullable=False)
|
||
structured_value: Mapped[dict[str, Any] | None] = mapped_column(JSON)
|
||
memory_type: Mapped[str] = mapped_column(String(24), nullable=False)
|
||
source_type: Mapped[str] = mapped_column(String(24), nullable=False)
|
||
source_confidence: Mapped[float] = mapped_column(Numeric(5, 4), nullable=False)
|
||
confidence: Mapped[float] = mapped_column(Numeric(5, 4), nullable=False)
|
||
evidence_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
conflict_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
recall_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||
status: Mapped[str] = mapped_column(String(16), nullable=False)
|
||
valid_from: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|
||
valid_until: Mapped[datetime | None] = mapped_column(DateTime)
|
||
last_evidenced_at: Mapped[datetime | None] = mapped_column(DateTime)
|
||
last_recall_at: Mapped[datetime | None] = mapped_column(DateTime)
|
||
promoted_at: Mapped[datetime | None] = mapped_column(DateTime)
|
||
version: Mapped[int] = mapped_column(BigInteger, nullable=False, default=1)
|
||
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|
||
updated_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|
||
|
||
|
||
class MemoryEvidence(Base):
|
||
"""记忆证据；`idempotency_key` 唯一键是重复消费的幂等边界。"""
|
||
|
||
__tablename__ = "memory_evidence"
|
||
id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
|
||
memory_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
||
evidence_type: Mapped[str] = mapped_column(String(24), nullable=False)
|
||
source_table: Mapped[str | None] = mapped_column(String(64))
|
||
source_record_id: Mapped[str | None] = mapped_column(String(64))
|
||
source_episode_id: Mapped[int | None] = mapped_column(BigInteger)
|
||
evidence_excerpt: Mapped[str | None] = mapped_column(Text)
|
||
evidence_snapshot: Mapped[dict[str, Any] | None] = mapped_column(JSON)
|
||
weight: Mapped[float] = mapped_column(Numeric(5, 4), nullable=False)
|
||
idempotency_key: Mapped[str] = mapped_column(String(128), unique=True, nullable=False)
|
||
occurred_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|
||
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|
||
|
||
|
||
class MemorySyncOutbox(Base):
|
||
__tablename__ = "memory_sync_outbox"
|
||
id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
|
||
event_uuid: Mapped[str] = mapped_column(String(36), nullable=False)
|
||
aggregate_type: Mapped[str] = mapped_column(String(24), nullable=False)
|
||
aggregate_uuid: Mapped[str] = mapped_column(String(36), nullable=False)
|
||
aggregate_version: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
||
target_store: Mapped[str] = mapped_column(String(16), nullable=False)
|
||
operation: Mapped[str] = mapped_column(String(16), nullable=False)
|
||
payload: Mapped[dict[str, Any]] = mapped_column(JSON, nullable=False)
|
||
status: Mapped[str] = mapped_column(String(16), nullable=False)
|
||
retry_count: Mapped[int] = mapped_column(BigInteger, nullable=False, default=0)
|
||
next_retry_at: Mapped[datetime | None] = mapped_column(DateTime)
|
||
last_error: Mapped[str | None] = mapped_column(Text)
|
||
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|
||
processed_at: Mapped[datetime | None] = mapped_column(DateTime)
|
||
|
||
|
||
class MemoryConflict(Base):
|
||
"""记忆冲突；库列名为 `status`，`resolution`/`severity`/`resolved_by` 必须映射。"""
|
||
|
||
__tablename__ = "memory_conflict"
|
||
id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
|
||
left_memory_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
||
right_memory_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
||
conflict_type: Mapped[str] = mapped_column(String(24), nullable=False)
|
||
severity: Mapped[str] = mapped_column(String(8), nullable=False)
|
||
status: Mapped[str] = mapped_column(String(16), nullable=False)
|
||
resolution: Mapped[str | None] = mapped_column(Text)
|
||
winner_memory_id: Mapped[int | None] = mapped_column(BigInteger)
|
||
resolved_by: Mapped[int | None] = mapped_column(BigInteger)
|
||
resolved_at: Mapped[datetime | None] = mapped_column(DateTime)
|
||
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|