zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-28 03:38:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: add document delete permission check (#14472)

Browse Source 
### What problem does this PR solve?

add document delete permission check

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
buua436
2026-04-30 11:01:09 +08:00
committed by GitHub
parent 47129fdd08
commit 06c6da5d94
3 changed files with 27 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/apps/restful_apis/document_api.py
View File

@@ -1027,6 +1027,13 @@ async def delete_documents(tenant_id, dataset_id):
if delete_all:
doc_ids = [doc.id for doc in DocumentService.query(kb_id=dataset_id)]
dataset_doc_ids = {doc.id for doc in DocumentService.query(kb_id=dataset_id)}
invalid_ids = [doc_id for doc_id in doc_ids if doc_id not in dataset_doc_ids]
if invalid_ids:
return get_error_data_result(
message=f"These documents do not belong to dataset {dataset_id} or Document not found: {', '.join(invalid_ids)}"
)
# make sure each id is unique
unique_doc_ids, duplicate_messages = check_duplicate_ids(doc_ids, "document")
if duplicate_messages: