diff --git a/api/apps/sdk/doc.py b/api/apps/sdk/doc.py index c551bbfe1..bef03d38e 100644 --- a/api/apps/sdk/doc.py +++ b/api/apps/sdk/doc.py @@ -1286,6 +1286,9 @@ async def rm_chunk(tenant_id, dataset_id, document_id): if "chunk_ids" in req: unique_chunk_ids, duplicate_messages = check_duplicate_ids(req["chunk_ids"], "chunk") condition["id"] = unique_chunk_ids + else: + unique_chunk_ids = [] + duplicate_messages = [] chunk_number = settings.docStoreConn.delete(condition, search.index_name(tenant_id), dataset_id) if chunk_number != 0: DocumentService.decrement_chunk_num(document_id, dataset_id, 1, chunk_number, 0)