fix document could be None (#15818)

api/core/rag/datasource/retrieval_service.py

@@ -276,6 +276,8 @@ class RetrievalService:
                     continue
 
                 dataset_document = dataset_documents[document_id]
+                if not dataset_document:
+                    continue
 
                 if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX:
                     # Handle parent-child documents

api/core/rag/retrieval/dataset_retrieval.py

@@ -433,6 +433,7 @@ class DatasetRetrieval:
                 dataset_document = DatasetDocument.query.filter(
                     DatasetDocument.id == document.metadata["document_id"]
                 ).first()
+                if dataset_document:
                     if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX:
                         child_chunk = ChildChunk.query.filter(
                             ChildChunk.index_node_id == document.metadata["doc_id"],
@@ -454,7 +455,9 @@ class DatasetRetrieval:
                             query = query.filter(DocumentSegment.dataset_id == document.metadata["dataset_id"])
 
                         # add hit count to document segment
-                    query.update({DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False)
+                        query.update(
+                            {DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False
+                        )
 
                     db.session.commit()