mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
"""Document model for tracking files in the knowledge base."""
|
|
|
|
from datetime import datetime
|
|
from typing import Optional, List
|
|
|
|
from sqlalchemy import String, DateTime, text, JSON, Index
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from basic_memory.models.base import Base
|
|
|
|
|
|
class Document(Base):
|
|
"""
|
|
Tracks documents in the filesystem.
|
|
|
|
Documents files are the source of truth for document content, while this table
|
|
provides indexing and metadata storage. Like git, the filesystem
|
|
is the real source of truth.
|
|
"""
|
|
|
|
__tablename__ = "document"
|
|
__table_args__ = (
|
|
Index("ix_document_created_at", "created_at"),
|
|
Index("ix_document_updated_at", "updated_at")
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
path: Mapped[str] = mapped_column(String, unique=True, nullable=False, index=True)
|
|
checksum: Mapped[str] = mapped_column(String, nullable=True)
|
|
doc_metadata: Mapped[Optional[dict]] = mapped_column(
|
|
JSON, nullable=True
|
|
) # renamed from metadata
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, server_default=text("CURRENT_TIMESTAMP"))
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime, server_default=text("CURRENT_TIMESTAMP"), onupdate=text("CURRENT_TIMESTAMP")
|
|
)
|
|
|
|
# Relationships
|
|
entities: Mapped[List["Entity"]] = relationship( # pyright: ignore [reportUndefinedVariable] # noqa: F821
|
|
"Entity", back_populates="document", cascade="all, delete-orphan"
|
|
)
|
|
|
|
def __repr__(self) -> str:
|
|
return f"Document(id={self.id}, path='{self.path}', checksum='{self.checksum}', created_at='{self.created_at}', updated_at='{self.updated_at}')"
|