Add superuser initialization argument to server
This commit is contained in:
parent
5074bb0aff
commit
9c3b82dfe0
1 changed files with 6 additions and 1 deletions
|
|
@ -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")
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue