fix document could be None (#15818)

2025-03-14 16:40:01 +08:00 · 2025-03-14 16:40:01 +08:00 · 84a866028a
commit 84a866028a
parent 10bd03611c
2 changed files with 26 additions and 21 deletions
--- a/api/core/rag/datasource/retrieval_service.py
+++ b/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