Bladeren bron

Fix: remove dup ids for APIs. (#6263)

### What problem does this PR solve?

#6234

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.18.0
Kevin Hu 7 maanden geleden
bovenliggende
commit
c2302abaf1
No account linked to committer's email address

+ 1
- 1
api/apps/sdk/dataset.py Bestand weergeven

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)

+ 2
- 2
api/apps/sdk/doc.py Bestand weergeven

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)

+ 1
- 1
sdk/python/test/test_http_api/test_dataset_mangement/test_delete_dataset.py Bestand weergeven

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

Laden…
Annuleren
Opslaan