diff --git a/src/backend/langflow/__main__.py b/src/backend/langflow/__main__.py index b362d2fa9..c2b7b0fd6 100644 --- a/src/backend/langflow/__main__.py +++ b/src/backend/langflow/__main__.py @@ -95,7 +95,7 @@ def serve_on_jcloud(): @app.command() -def serve( +def run( host: str = typer.Option( "127.0.0.1", help="Host to bind the server to.", envvar="LANGFLOW_HOST" ), @@ -326,7 +326,16 @@ def superuser( from langflow.services.auth.utils import create_super_user if create_super_user(session, username, password): + # Verify that the superuser was created + from langflow.services.database.models.user.user import User + + user = session.query(User).filter(User.username == username).first() + if user is None: + typer.echo("Superuser creation failed.") + return + typer.echo("Superuser created successfully.") + else: typer.echo("Superuser creation failed.") diff --git a/tests/test_cli.py b/tests/test_cli.py index c5fba1da6..4ed00893e 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -23,8 +23,14 @@ def test_components_path(runner, client, default_settings): result = runner.invoke( app, - ["serve", "--components-path", str(temp_dir), *default_settings], + ["run", "--components-path", str(temp_dir), *default_settings], ) assert result.exit_code == 0, result.stdout settings_manager = utils.get_settings_manager() assert str(temp_dir) in settings_manager.settings.COMPONENTS_PATH + + +def test_superuser(runner, client, session): + result = runner.invoke(app, ["superuser"], input="admin\nadmin\n") + assert result.exit_code == 0, result.stdout + assert "Superuser created successfully." in result.stdout