Преглед на файлове

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 месеца
родител
ревизия
c2302abaf1
No account linked to committer's email address
променени са 3 файла, в които са добавени 4 реда и са изтрити 4 реда
  1. 1
    1
      api/apps/sdk/dataset.py
  2. 2
    2
      api/apps/sdk/doc.py
  3. 1
    1
      sdk/python/test/test_http_api/test_dataset_mangement/test_delete_dataset.py

+ 1
- 1
api/apps/sdk/dataset.py Целия файл

@@ -237,7 +237,7 @@ def delete(tenant_id):
if not req:
ids = None
else:
ids = req.get("ids")
ids = set(req.get("ids"))
if not ids:
id_list = []
kbs = KnowledgebaseService.query(tenant_id=tenant_id)

+ 2
- 2
api/apps/sdk/doc.py Целия файл

@@ -584,7 +584,7 @@ def delete(tenant_id, dataset_id):
if not req:
doc_ids = None
else:
doc_ids = req.get("ids")
doc_ids = set(req.get("ids"))
if not doc_ids:
doc_list = []
docs = DocumentService.query(kb_id=dataset_id)
@@ -681,7 +681,7 @@ def parse(tenant_id, dataset_id):
if not req.get("document_ids"):
return get_error_data_result("`document_ids` is required")
not_found = []
for id in req["document_ids"]:
for id in set(req["document_ids"]):
doc = DocumentService.query(id=id, kb_id=dataset_id)
if not doc:
not_found.append(id)

+ 1
- 1
sdk/python/test/test_http_api/test_dataset_mangement/test_delete_dataset.py Целия файл

@@ -120,7 +120,7 @@ class TestDatasetDeletion:
ids = create_datasets(get_http_api_auth, 1)
res = delete_dataset(get_http_api_auth, {"ids": ids + ids})
assert res["code"] == 0
assert res["data"]["success_count"] == 1
#assert res["data"]["success_count"] == 1

res = list_dataset(get_http_api_auth)
assert len(res["data"]) == 0

Loading…
Отказ
Запис