From 1c79a62fb274f53a9fc91bdc918a5fb22aaca27f Mon Sep 17 00:00:00 2001 From: Adityavardhan Agrawal Date: Mon, 21 Apr 2025 00:33:25 -0700 Subject: [PATCH] Mark ingest_text as complete when ingestion finished --- core/services/document_service.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/core/services/document_service.py b/core/services/document_service.py index fc4fdae..bae2b14 100644 --- a/core/services/document_service.py +++ b/core/services/document_service.py @@ -636,6 +636,15 @@ class DocumentService: await self._store_chunks_and_doc(chunk_objects, doc, use_colpali, chunk_objects_multivector) logger.debug(f"Successfully stored text document {doc.external_id}") + # Update the document status to completed after successful storage + # This matches the behavior in ingestion_worker.py + doc.system_metadata["status"] = "completed" + doc.system_metadata["updated_at"] = datetime.now(UTC) + await self.db.update_document( + document_id=doc.external_id, updates={"system_metadata": doc.system_metadata}, auth=auth + ) + logger.debug(f"Updated document status to 'completed' for {doc.external_id}") + return doc # TODO: check if it's unused. if so, remove it.