From 96667696d2e6803e6dc0ded2ce21889c59d30ea0 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Tue, 17 Dec 2024 13:58:26 +0800 Subject: [PATCH] Compatible with former API keys. (#4055) ### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- api/apps/system_app.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/api/apps/system_app.py b/api/apps/system_app.py index 28b74d658..0d979b233 100644 --- a/api/apps/system_app.py +++ b/api/apps/system_app.py @@ -256,8 +256,14 @@ def token_list(): if not tenants: return get_data_error_result(message="Tenant not found!") - objs = APITokenService.query(tenant_id=tenants[0].tenant_id) - return get_json_result(data=[o.to_dict() for o in objs]) + tenant_id = tenants[0].tenant_id + objs = APITokenService.query(tenant_id=tenant_id) + objs = [o.to_dict() for o in objs] + for o in objs: + if not o["beta"]: + o["beta"] = generate_confirmation_token(generate_confirmation_token(tenants[0].tenant_id)).replace("ragflow-", "")[:32] + APITokenService.filter_update([APIToken.tenant_id == tenant_id, APIToken.token == o["token"]], o) + return get_json_result(data=objs) except Exception as e: return server_error_response(e)