diff --git a/src/backend/langflow/__main__.py b/src/backend/langflow/__main__.py index b5ec034de..a08ae9fb0 100644 --- a/src/backend/langflow/__main__.py +++ b/src/backend/langflow/__main__.py @@ -356,7 +356,7 @@ def superuser( with session_getter(db_manager) as session: from langflow.services.auth.utils import create_super_user - if create_super_user(session, username, password): + if create_super_user(db=session, username=username, password=password): # Verify that the superuser was created from langflow.services.database.models.user.user import User diff --git a/tests/test_user.py b/tests/test_user.py index 35b724cc6..bc617e127 100644 --- a/tests/test_user.py +++ b/tests/test_user.py @@ -12,7 +12,7 @@ def super_user(client, session): settings_manager = get_settings_manager() auth_settings = settings_manager.auth_settings return create_super_user( - session, + db=session, username=auth_settings.FIRST_SUPERUSER, password=auth_settings.FIRST_SUPERUSER_PASSWORD, )