mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-04-21 05:29:57 +08:00
b6f3f15f0b
### What problem does this PR solve? 1. Read KB list path parameter, page_number and page_size, which type isn't int 2. Add cases on create / list / delete datasets. ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) - [x] Test cases Signed-off-by: jinhai <haijin.chn@gmail.com>
27 lines
799 B
Python
27 lines
799 B
Python
import os
|
|
import requests
|
|
|
|
HOST_ADDRESS = os.getenv('HOST_ADDRESS', 'http://127.0.0.1:9380')
|
|
|
|
def create_dataset(auth, dataset_name):
|
|
authorization = {"Authorization": auth}
|
|
url = f"{HOST_ADDRESS}/v1/kb/create"
|
|
json = {"name": dataset_name}
|
|
res = requests.post(url=url, headers=authorization, json=json)
|
|
return res.json()
|
|
|
|
|
|
def list_dataset(auth, page_number):
|
|
authorization = {"Authorization": auth}
|
|
url = f"{HOST_ADDRESS}/v1/kb/list?page={page_number}"
|
|
res = requests.get(url=url, headers=authorization)
|
|
return res.json()
|
|
|
|
|
|
def rm_dataset(auth, dataset_id):
|
|
authorization = {"Authorization": auth}
|
|
url = f"{HOST_ADDRESS}/v1/kb/rm"
|
|
json = {"kb_id": dataset_id}
|
|
res = requests.post(url=url, headers=authorization, json=json)
|
|
return res.json()
|