47 lines
2.3 KiB
Python
47 lines
2.3 KiB
Python
"""记忆模型。"""
|
|
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import DateTime, Float, ForeignKey, String, Text
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.core.ids import new_memory_id, new_memory_link_id
|
|
from app.core.time import utc_now
|
|
from app.db.base import Base
|
|
|
|
|
|
class Memory(Base):
|
|
__tablename__ = "memories"
|
|
|
|
id: Mapped[str] = mapped_column(String(80), primary_key=True, default=new_memory_id)
|
|
character_id: Mapped[str] = mapped_column(String(80), ForeignKey("characters.id"), index=True)
|
|
type: Mapped[str] = mapped_column(String(60), nullable=False)
|
|
title: Mapped[str] = mapped_column(String(200), nullable=False)
|
|
content: Mapped[str] = mapped_column(Text, nullable=False)
|
|
source: Mapped[str] = mapped_column(String(80), nullable=False, default="user_explicit")
|
|
confidence: Mapped[float] = mapped_column(Float, nullable=False, default=1.0)
|
|
importance: Mapped[float] = mapped_column(Float, nullable=False, default=0.5)
|
|
sensitivity: Mapped[str] = mapped_column(String(40), nullable=False, default="low")
|
|
evidence_ids_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")
|
|
status: Mapped[str] = mapped_column(String(40), nullable=False, default="active", index=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), default=utc_now, onupdate=utc_now
|
|
)
|
|
deleted_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
|
expires_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
|
temporal_context_json: Mapped[str] = mapped_column(
|
|
Text, nullable=False, default="{}", server_default="{}"
|
|
)
|
|
|
|
|
|
class MemoryLink(Base):
|
|
__tablename__ = "memory_links"
|
|
|
|
id: Mapped[str] = mapped_column(String(80), primary_key=True, default=new_memory_link_id)
|
|
character_id: Mapped[str] = mapped_column(String(80), ForeignKey("characters.id"), index=True)
|
|
source_memory_id: Mapped[str] = mapped_column(String(80), ForeignKey("memories.id"), index=True)
|
|
target_memory_id: Mapped[str] = mapped_column(String(80), ForeignKey("memories.id"), index=True)
|
|
relation: Mapped[str] = mapped_column(String(80), nullable=False)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|