From 5a4e64e741a0325fd518c90789bcfb674b4f4e13 Mon Sep 17 00:00:00 2001 From: wwwlll Date: Mon, 19 Aug 2024 10:02:31 +0800 Subject: [PATCH] Add API to get doc info by doc ids (#1986) Supports use API to get doc info by doc ids ### What problem does this PR solve? feat: Supports use API to get doc info by doc ids ### Type of change - [x] New Feature (non-breaking change which adds functionality) --------- Co-authored-by: Kevin Hu --- api/apps/api_app.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/api/apps/api_app.py b/api/apps/api_app.py index 47ca514d..a5243aea 100644 --- a/api/apps/api_app.py +++ b/api/apps/api_app.py @@ -571,6 +571,19 @@ def list_kb_docs(): except Exception as e: return server_error_response(e) +@manager.route('/document/infos', methods=['POST']) +@validate_request("doc_ids") +def docinfos(): + token = request.headers.get('Authorization').split()[1] + objs = APIToken.query(token=token) + if not objs: + return get_json_result( + data=False, retmsg='Token is not valid!"', retcode=RetCode.AUTHENTICATION_ERROR) + req = request.json + doc_ids = req["doc_ids"] + docs = DocumentService.get_by_ids(doc_ids) + return get_json_result(data=list(docs.dicts())) + @manager.route('/document', methods=['DELETE']) # @login_required