diff --git a/api/apps/document_app.py b/api/apps/document_app.py index e96791a2f..ec7db5f70 100644 --- a/api/apps/document_app.py +++ b/api/apps/document_app.py @@ -347,7 +347,7 @@ def rm(): @manager.route('/run', methods=['POST']) # noqa: F821 @login_required @validate_request("doc_ids", "run") -def run(): +def run(): req = request.json for doc_id in req["doc_ids"]: if not DocumentService.accessible(doc_id, current_user.id): diff --git a/api/apps/sdk/dataset.py b/api/apps/sdk/dataset.py index 80daf2109..17d1f939f 100644 --- a/api/apps/sdk/dataset.py +++ b/api/apps/sdk/dataset.py @@ -276,7 +276,7 @@ def delete(tenant_id): return get_result(code=settings.RetCode.SUCCESS) -@manager.route("/datasets/", methods=["PUT"]) # noqa: F821 +@manager.route("/datasets/", methods=["PUT"]) # noqa: F821 @token_required def update(tenant_id, dataset_id): """ @@ -330,7 +330,7 @@ def update(tenant_id, dataset_id): return get_error_data_result(message="You don't own the dataset") req = request.json e, t = TenantService.get_by_id(tenant_id) - invalid_keys = {"id", "embd_id", "chunk_num", "doc_num", "parser_id"} + invalid_keys = {"id", "embd_id", "chunk_num", "doc_num", "parser_id", "create_date", "create_time", "created_by", "status","token_num","update_date","update_time"} if any(key in req for key in invalid_keys): return get_error_data_result(message="The input parameters are invalid.") permission = req.get("permission") diff --git a/sdk/python/test/test_frontend_api/common.py b/sdk/python/test/test_frontend_api/common.py index 19aa6df45..f94e1e6f5 100644 --- a/sdk/python/test/test_frontend_api/common.py +++ b/sdk/python/test/test_frontend_api/common.py @@ -68,7 +68,7 @@ def upload_file(auth, dataset_id, path): def list_document(auth, dataset_id): authorization = {"Authorization": auth} - url = f"{HOST_ADDRESS}/v1/document/list?kb_id={dataset_id}" + url = f"{HOST_ADDRESS}/v1/document/list?kb_id={dataset_id}" res = requests.get(url=url, headers=authorization) return res.json() @@ -85,7 +85,7 @@ def parse_docs(auth, doc_ids): authorization = {"Authorization": auth} json_req = { "doc_ids": doc_ids, - "run": 1 + "run": 1 } url = f"{HOST_ADDRESS}/v1/document/run" res = requests.post(url=url, headers=authorization, json=json_req) diff --git a/sdk/python/test/test_http_api/test_dataset_mangement/common.py b/sdk/python/test/test_http_api/test_dataset_mangement/common.py index 049cb537e..9afbf029e 100644 --- a/sdk/python/test/test_http_api/test_dataset_mangement/common.py +++ b/sdk/python/test/test_http_api/test_dataset_mangement/common.py @@ -39,7 +39,7 @@ def list_dataset(auth, params=None): def update_dataset(auth, dataset_id, payload): res = requests.put( - url=f"{API_URL}/{dataset_id}", headers=HEADERS, auth=auth, json=payload + url=f"{API_URL}/{dataset_id}", headers=HEADERS, auth=auth, json=payload ) return res.json()