From 44631284368ef49b4de5fe707fe5a7559538620b Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Tue, 15 Oct 2024 16:48:38 +0800 Subject: [PATCH] add rm token (#2850) ### What problem does this PR solve? #2846 ### Type of change - [x] New Feature (non-breaking change which adds functionality) --- api/apps/system_app.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/api/apps/system_app.py b/api/apps/system_app.py index d59ab71a3..030c6a7e3 100644 --- a/api/apps/system_app.py +++ b/api/apps/system_app.py @@ -17,13 +17,15 @@ import json from datetime import datetime from flask_login import login_required, current_user + +from api.db.db_models import APIToken from api.db.services.api_service import APITokenService from api.db.services.knowledgebase_service import KnowledgebaseService from api.db.services.user_service import UserTenantService from api.settings import DATABASE_TYPE from api.utils import current_timestamp, datetime_format from api.utils.api_utils import get_json_result, get_data_error_result, server_error_response, \ - generate_confirmation_token + generate_confirmation_token, request, validate_request from api.versions import get_rag_version from rag.utils.es_conn import ELASTICSEARCH from rag.utils.storage_factory import STORAGE_IMPL, STORAGE_IMPL_TYPE @@ -130,3 +132,16 @@ def token_list(): except Exception as e: return server_error_response(e) + +@manager.route('/rm', methods=['POST']) +@validate_request("tokens", "tenant_id") +@login_required +def rm(): + req = request.json + try: + for token in req["tokens"]: + APITokenService.filter_delete( + [APIToken.tenant_id == req["tenant_id"], APIToken.token == token]) + return get_json_result(data=True) + except Exception as e: + return server_error_response(e)