|
|
|
@@ -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) |