diff --git a/api/apps/api_app.py b/api/apps/api_app.py index 47ca514d9..a5243aea1 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