### What problem does this PR solve? #6234 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)tags/v0.18.0
| if not req: | if not req: | ||||
| ids = None | ids = None | ||||
| else: | else: | ||||
| ids = req.get("ids") | |||||
| ids = set(req.get("ids")) | |||||
| if not ids: | if not ids: | ||||
| id_list = [] | id_list = [] | ||||
| kbs = KnowledgebaseService.query(tenant_id=tenant_id) | kbs = KnowledgebaseService.query(tenant_id=tenant_id) |
| if not req: | if not req: | ||||
| doc_ids = None | doc_ids = None | ||||
| else: | else: | ||||
| doc_ids = req.get("ids") | |||||
| doc_ids = set(req.get("ids")) | |||||
| if not doc_ids: | if not doc_ids: | ||||
| doc_list = [] | doc_list = [] | ||||
| docs = DocumentService.query(kb_id=dataset_id) | docs = DocumentService.query(kb_id=dataset_id) | ||||
| if not req.get("document_ids"): | if not req.get("document_ids"): | ||||
| return get_error_data_result("`document_ids` is required") | return get_error_data_result("`document_ids` is required") | ||||
| not_found = [] | not_found = [] | ||||
| for id in req["document_ids"]: | |||||
| for id in set(req["document_ids"]): | |||||
| doc = DocumentService.query(id=id, kb_id=dataset_id) | doc = DocumentService.query(id=id, kb_id=dataset_id) | ||||
| if not doc: | if not doc: | ||||
| not_found.append(id) | not_found.append(id) |
| ids = create_datasets(get_http_api_auth, 1) | ids = create_datasets(get_http_api_auth, 1) | ||||
| res = delete_dataset(get_http_api_auth, {"ids": ids + ids}) | res = delete_dataset(get_http_api_auth, {"ids": ids + ids}) | ||||
| assert res["code"] == 0 | assert res["code"] == 0 | ||||
| assert res["data"]["success_count"] == 1 | |||||
| #assert res["data"]["success_count"] == 1 | |||||
| res = list_dataset(get_http_api_auth) | res = list_dataset(get_http_api_auth) | ||||
| assert len(res["data"]) == 0 | assert len(res["data"]) == 0 |