From 5e5f51b3a0fcfe1c28d8f8688301edf62cfa37e4 Mon Sep 17 00:00:00 2001 From: CREDO23 Date: Tue, 2 Jun 2026 16:10:44 +0200 Subject: [PATCH] feat(db): register DocumentFile and document.files relationship --- surfsense_backend/app/db.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/surfsense_backend/app/db.py b/surfsense_backend/app/db.py index f632c18af..5be10427f 100644 --- a/surfsense_backend/app/db.py +++ b/surfsense_backend/app/db.py @@ -1478,6 +1478,11 @@ class Document(BaseModel, TimestampMixin): chunks = relationship( "Chunk", back_populates="document", cascade="all, delete-orphan" ) + # Original upload + future derived artifacts (redacted, filled-form). + # Model lives in app.file_storage.persistence to keep that feature cohesive. + files = relationship( + "DocumentFile", back_populates="document", cascade="all, delete-orphan" + ) class DocumentVersion(BaseModel, TimestampMixin): @@ -2931,6 +2936,7 @@ from app.automations.persistence import ( # noqa: E402, F401 AutomationRun, AutomationTrigger, ) +from app.file_storage.persistence import DocumentFile # noqa: E402, F401 engine = create_async_engine( DATABASE_URL,