From 2484e26cb59c16f624f9f0d5e997350792c851dc Mon Sep 17 00:00:00 2001 From: Dada Hsueh Date: Wed, 18 Sep 2024 14:30:45 +0800 Subject: [PATCH] fix `superuser` password not base64 encoded (#2475) ### What problem does this PR solve? Fixes the _superuser_ `admin@ragflow.io` not being accessible due to how entered passwords are used. Unless this is expected behavior? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- api/db/init_data.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/api/db/init_data.py b/api/db/init_data.py index e4f61093b..d8837ddf0 100644 --- a/api/db/init_data.py +++ b/api/db/init_data.py @@ -13,6 +13,7 @@ # See the License for the specific language governing permissions and # limitations under the License. # +import base64 import json import os import time @@ -30,11 +31,14 @@ from api.db.services.user_service import TenantService, UserTenantService from api.settings import CHAT_MDL, EMBEDDING_MDL, ASR_MDL, IMAGE2TEXT_MDL, PARSERS, LLM_FACTORY, API_KEY, LLM_BASE_URL from api.utils.file_utils import get_project_base_directory +def encode_to_base64(input_string): + base64_encoded = base64.b64encode(input_string.encode('utf-8')) + return base64_encoded.decode('utf-8') def init_superuser(): user_info = { "id": uuid.uuid1().hex, - "password": "admin", + "password": encode_to_base64("admin"), "nickname": "admin", "is_superuser": True, "email": "admin@ragflow.io",