Added documents upload table

private_gpt/users/crud/document_crud.py

@@ -0,0 +1,16 @@
+from sqlalchemy.orm import Session
+from private_gpt.users.schemas.documents import DocumentCreate, DocumentUpdate
+from private_gpt.users.models.documents import Documents
+from private_gpt.users.crud.base import CRUDBase
+from typing import Optional
+
+
+class CRUDDocuments(CRUDBase[Documents, DocumentCreate, DocumentUpdate]):
+    def get_by_id(self, db: Session, *, id: str) -> Optional[Documents]:
+        return db.query(self.model).filter(Documents.id == id).first()
+
+    def get_by_filename(self, db: Session, *, file_name: str) -> Optional[Documents]:
+        return db.query(self.model).filter(Documents.filename == file_name).first()
+
+
+documents = CRUDDocuments(Documents)