Merge branch 'document-store' into document-entity-write

src/basic_memory/models/documents.py

@@ -0,0 +1,40 @@
+"""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
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from basic_memory.models.base import Base
+
+
+class Document(Base):
+    """
+    Tracks documents in the filesystem.
+
+    Documents are the source of truth for content, while this table
+    provides indexing and metadata storage. Like git, the filesystem
+    is the real source of truth.
+    """
+
+    __tablename__ = "documents"
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    path: Mapped[str] = mapped_column(String, unique=True, nullable=False)
+    checksum: Mapped[str] = mapped_column(String, nullable=False)
+    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}')"