mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-04-20 05:00:01 +08:00
ec560cc99d
### What problem does this PR solve? Feat: Scrolling knowledge base list and set the number of entries per page to 30 #3695 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
from common import HOST_ADDRESS, create_dataset, list_dataset, rm_dataset
|
|
import requests
|
|
|
|
|
|
def test_dataset(get_auth):
|
|
# create dataset
|
|
res = create_dataset(get_auth, "test_create_dataset")
|
|
assert res.get("code") == 0, f"{res.get('message')}"
|
|
|
|
# list dataset
|
|
page_number = 1
|
|
dataset_list = []
|
|
while True:
|
|
res = list_dataset(get_auth, page_number)
|
|
data = res.get("data")
|
|
for item in data.get("kbs"):
|
|
dataset_id = item.get("id")
|
|
dataset_list.append(dataset_id)
|
|
if len(dataset_list) < page_number * 150:
|
|
break
|
|
page_number += 1
|
|
|
|
print(f"found {len(dataset_list)} datasets")
|
|
# delete dataset
|
|
for dataset_id in dataset_list:
|
|
res = rm_dataset(get_auth, dataset_id)
|
|
assert res.get("code") == 0, f"{res.get('message')}"
|
|
print(f"{len(dataset_list)} datasets are deleted")
|
|
|
|
|
|
def test_dataset_1k_dataset(get_auth):
|
|
# create dataset
|
|
authorization = {"Authorization": get_auth}
|
|
url = f"{HOST_ADDRESS}/v1/kb/create"
|
|
for i in range(1000):
|
|
res = create_dataset(get_auth, f"test_create_dataset_{i}")
|
|
assert res.get("code") == 0, f"{res.get('message')}"
|
|
|
|
# list dataset
|
|
page_number = 1
|
|
dataset_list = []
|
|
while True:
|
|
res = list_dataset(get_auth, page_number)
|
|
data = res.get("data")
|
|
for item in data.get("kbs"):
|
|
dataset_id = item.get("id")
|
|
dataset_list.append(dataset_id)
|
|
if len(dataset_list) < page_number * 150:
|
|
break
|
|
page_number += 1
|
|
|
|
print(f"found {len(dataset_list)} datasets")
|
|
# delete dataset
|
|
for dataset_id in dataset_list:
|
|
res = rm_dataset(get_auth, dataset_id)
|
|
assert res.get("code") == 0, f"{res.get('message')}"
|
|
print(f"{len(dataset_list)} datasets are deleted")
|
|
|
|
# delete dataset
|
|
# create invalid name dataset
|
|
# update dataset with different parameters
|
|
# create duplicated name dataset
|
|
#
|