memory_item.py 1.6 KB

1234567891011121314151617181920212223242526272829
  1. from datetime import datetime
  2. from sqlalchemy import DateTime, Integer, String, Text
  3. from sqlalchemy.dialects.sqlite import JSON
  4. from sqlalchemy.orm import Mapped, mapped_column
  5. from core_db import AuditMixin, Base, TenantMixin, VersionMixin
  6. from core_shared import JSONValue
  7. class MemoryItem(TenantMixin, AuditMixin, VersionMixin, Base):
  8. __tablename__ = "memory_item"
  9. scope_type: Mapped[str] = mapped_column(String(32), index=True)
  10. scope_id: Mapped[str] = mapped_column(String(64), index=True)
  11. memory_type: Mapped[str] = mapped_column(String(64), default="fact", index=True)
  12. content_text: Mapped[str] = mapped_column(Text)
  13. content_json: Mapped[dict[str, JSONValue] | None] = mapped_column(JSON, nullable=True)
  14. metadata_json: Mapped[dict[str, JSONValue]] = mapped_column(JSON, default=dict)
  15. embedding_model: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
  16. embedding_json: Mapped[list[float] | None] = mapped_column(JSON, nullable=True)
  17. owner_agent_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
  18. user_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
  19. session_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
  20. source_ref: Mapped[str | None] = mapped_column(String(256), nullable=True)
  21. importance_score: Mapped[int] = mapped_column(Integer, default=0)
  22. status: Mapped[str] = mapped_column(String(32), default="active", index=True)
  23. last_accessed_time: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
  24. expires_time: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)