Upgrade 1.4.1, change the CreatorUserRole class of models.enum

api/core/rag/extractor/pdf_extractor.py

@@ -15,7 +15,7 @@ from core.rag.extractor.extractor_base import BaseExtractor
 from core.rag.models.document import Document
 from extensions.ext_database import db
 from extensions.ext_storage import storage
-from models.enums import CreatedByRole
+from models.enums import CreatorUserRole
 from models.model import UploadFile
 
 class PdfExtractor(BaseExtractor):
@@ -121,7 +121,7 @@ class PdfExtractor(BaseExtractor):
                                     extension=image_ext,
                                     mime_type=mime_type or "",
                                     created_by=self._user_id,
-                                    created_by_role=CreatedByRole.ACCOUNT,
+                                    created_by_role=CreatorUserRole.ACCOUNT,
                                     created_at=datetime.datetime.now(datetime.timezone.utc).replace(tzinfo=None),
                                     used=True,
                                     used_by=self._user_id,
@@ -159,4 +159,4 @@ class PdfExtractor(BaseExtractor):
                         "has_tables": bool(tables)
                     }
 
-                    yield Document(page_content=content, metadata=metadata)
+                    yield Document(page_content=content, metadata=metadata)

api/pyproject.toml

@@ -83,6 +83,8 @@ dependencies = [
     "weave~=0.51.0",
     "yarl~=1.18.3",
     "webvtt-py~=0.5.1",
+    "pdfplumber~=0.11.6",
+    "pytesseract~=0.3.13"
 ]
 # Before adding new dependency, consider place it in
 # alphabet order (a-z) and suitable group.