diff --git a/api/ragflow_server.py b/api/ragflow_server.py index fb49f3d8b..bbfe6a032 100644 --- a/api/ragflow_server.py +++ b/api/ragflow_server.py @@ -39,7 +39,7 @@ from api.db.services.document_service import DocumentService from api import utils from api.db.db_models import init_database_tables as init_web_db -from api.db.init_data import init_web_data +from api.db.init_data import init_web_data, init_superuser from api.versions import get_ragflow_version from api.utils.configs import show_configs from rag.settings import print_rag_settings @@ -112,11 +112,16 @@ if __name__ == '__main__': parser.add_argument( "--debug", default=False, help="debug mode", action="store_true" ) + parser.add_argument( + "--init-superuser", default=False, help="init superuser", action="store_true" + ) args = parser.parse_args() if args.version: print(get_ragflow_version()) sys.exit(0) + if args.init_superuser: + init_superuser() RuntimeConfig.DEBUG = args.debug if RuntimeConfig.DEBUG: logging.info("run on debug mode")