[changes] Hide the user knowledge base and unify the display of memory capacity

api/app/tasks.py

@@ -62,7 +62,7 @@ def process_item(item: dict):
 
 
 @celery_app.task(name="app.core.rag.tasks.parse_document")
-def parse_document(file_path: str, document_id: uuid.UUID):
+def parse_document(file_path: str, document_id: str):
     """
     Document parsing, vectorization, and storage
     """
@@ -74,6 +74,9 @@ def parse_document(file_path: str, document_id: uuid.UUID):
     db = next(get_db())  # Manually call the generator
     db_document = None
     db_knowledge = None
+    # 确保 document_id 是 UUID 对象
+    if not isinstance(document_id, uuid.UUID):
+        document_id = uuid.UUID(str(document_id))
     progress_msg = f"{datetime.now().strftime('%H:%M:%S')} Task has been received.\n"
     try:
         db_document = db.query(Document).filter(Document.id == document_id).first()
@@ -286,11 +289,13 @@ def parse_document(file_path: str, document_id: uuid.UUID):
         result = f"parse document '{db_document.file_name}' processed successfully."
         return result
     except Exception as e:
-        if 'db_document' in locals():
-            db_document.progress_msg += f"Failed to vectorize and import the parsed document:{str(e)}\n"
+        if db_document is not None:
+            db_document.progress_msg = (db_document.progress_msg or "") + f"Failed to vectorize and import the parsed document: {str(e)}\n"
             db_document.run = 0
             db.commit()
-        result = f"parse document '{db_document.file_name}' failed."
+            result = f"parse document '{db_document.file_name}' failed."
+        else:
+            result = f"parse document '{document_id}' failed: document not found in DB. error={str(e)}"
         return result
     finally:
         db.close()