Fix bugs in API (#3204)

### What problem does this PR solve?

Fix bugs in API

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

---------

Co-authored-by: liuhua <10215101452@stu.ecun.edu.cn>
This commit is contained in:
liuhua
2024-11-05 14:07:31 +08:00
committed by GitHub
parent 7e89be5ed1
commit a7bf4ca8fc
8 changed files with 22 additions and 22 deletions

View File

@ -264,7 +264,7 @@ def list_chat(tenant_id):
if not chat:
return get_error_data_result(message="The chat doesn't exist")
page_number = int(request.args.get("page", 1))
items_per_page = int(request.args.get("page_size", 1024))
items_per_page = int(request.args.get("page_size", 30))
orderby = request.args.get("orderby", "create_time")
if request.args.get("desc") == "False" or request.args.get("desc") == "false":
desc = False
@ -308,4 +308,4 @@ def list_chat(tenant_id):
res["datasets"] = kb_list
res["avatar"] = res.pop("icon")
list_assts.append(res)
return get_result(data=list_assts)
return get_result(data=list_assts)

View File

@ -502,7 +502,7 @@ def list(tenant_id):
if not kbs:
return get_error_data_result(message="The dataset doesn't exist")
page_number = int(request.args.get("page", 1))
items_per_page = int(request.args.get("page_size", 1024))
items_per_page = int(request.args.get("page_size", 30))
orderby = request.args.get("orderby", "create_time")
if request.args.get("desc") == "False" or request.args.get("desc") == "false":
desc = False
@ -532,4 +532,4 @@ def list(tenant_id):
new_key = key_mapping.get(key, key)
renamed_data[new_key] = value
renamed_list.append(renamed_data)
return get_result(data=renamed_list)
return get_result(data=renamed_list)

View File

@ -386,16 +386,16 @@ def list_docs(dataset_id, tenant_id):
required: false
description: Filter by document ID.
- in: query
name: offset
name: page
type: integer
required: false
default: 1
description: Page number.
- in: query
name: limit
name: page_size
type: integer
required: false
default: 1024
default: 30
description: Number of items per page.
- in: query
name: orderby
@ -460,7 +460,7 @@ def list_docs(dataset_id, tenant_id):
return get_error_data_result(message=f"You don't own the document {name}.")
page = int(request.args.get("page", 1))
keywords = request.args.get("keywords", "")
page_size = int(request.args.get("page_size", 1024))
page_size = int(request.args.get("page_size", 30))
orderby = request.args.get("orderby", "create_time")
if request.args.get("desc") == "False":
desc = False
@ -741,13 +741,13 @@ def list_chunks(tenant_id, dataset_id, document_id):
required: true
description: ID of the document.
- in: query
name: offset
name: page
type: integer
required: false
default: 1
description: Page number.
- in: query
name: limit
name: page_size
type: integer
required: false
default: 30
@ -1323,7 +1323,7 @@ def retrieval_test(tenant_id):
if "question" not in req:
return get_error_data_result("`question` is required.")
page = int(req.get("page", 1))
size = int(req.get("page_size", 1024))
size = int(req.get("page_size", 30))
question = req["question"]
doc_ids = req.get("document_ids", [])
if not isinstance(doc_ids, list):
@ -1401,4 +1401,4 @@ def retrieval_test(tenant_id):
message="No chunk found! Check the chunk status please!",
code=RetCode.DATA_ERROR,
)
return server_error_response(e)
return server_error_response(e)

View File

@ -322,7 +322,7 @@ def list(chat_id,tenant_id):
id = request.args.get("id")
name = request.args.get("name")
page_number = int(request.args.get("page", 1))
items_per_page = int(request.args.get("page_size", 1024))
items_per_page = int(request.args.get("page_size", 30))
orderby = request.args.get("orderby", "create_time")
if request.args.get("desc") == "False" or request.args.get("desc") == "false":
desc = False
@ -391,4 +391,4 @@ def delete(tenant_id,chat_id):
if not conv:
return get_error_data_result(message="The chat doesn't own the session")
ConversationService.delete_by_id(id)
return get_result()
return get_result()