🔧 fix(__main__.py): import correct function for initializing settings manager

🔧 fix(utils.py): fix condition for checking if user is superuser

src/backend/langflow/__main__.py

@@ -2,8 +2,9 @@ import sys
 import time
 import httpx
 from langflow.services.database.utils import session_getter
-from langflow.services.manager import initialize_services, initialize_settings_manager
+from langflow.services.utils import initialize_services
 from langflow.services.getters import get_db_manager, get_settings_manager
+from langflow.services.utils import initialize_settings_manager
 
 from multiprocess import Process, cpu_count  # type: ignore
 import platform

src/backend/langflow/services/utils.py

@@ -1,4 +1,6 @@
 from langflow.services.auth.utils import create_super_user
+from langflow.services.manager import service_manager
+from langflow.services.schema import ServiceType
 from langflow.services.settings.constants import (
     DEFAULT_SUPERUSER,
     DEFAULT_SUPERUSER_PASSWORD,
@@ -15,6 +17,10 @@ def setup_superuser():
     # vars on settings_manager.auth_settings to create the superuser
     # if it does not exist.
     settings_manager = get_settings_manager()
+    if settings_manager.auth_settings.AUTO_LOGIN:
+        logger.debug("AUTO_LOGIN is set to True. Not creating superuser automatically.")
+        return
+
     session = next(get_session())
     username = settings_manager.auth_settings.SUPERUSER
     password = settings_manager.auth_settings.SUPERUSER_PASSWORD
@@ -28,7 +34,7 @@ def setup_superuser():
         from langflow.services.database.models.user.user import User
 
         user = session.query(User).filter(User.username == username).first()
-        if user and user.is_superuser:
+        if user and user.is_superuser is True:
             return
     except Exception as exc:
         logger.exception(exc)
@@ -46,3 +52,90 @@ def setup_superuser():
     # reset superuser credentials
     settings_manager.auth_settings.reset_credentials()
     logger.debug("Superuser created successfully.")
+
+
+def teardown_superuser():
+    """
+    Teardown the superuser.
+    """
+    # If AUTO_LOGIN is True, we will remove the default superuser
+    # from the database.
+    settings_manager = get_settings_manager()
+    if settings_manager.auth_settings.AUTO_LOGIN:
+        logger.debug("AUTO_LOGIN is set to True. Removing default superuser.")
+        session = next(get_session())
+        username = settings_manager.auth_settings.SUPERUSER
+        from langflow.services.database.models.user.user import User
+
+        user = session.query(User).filter(User.username == username).first()
+        if user and user.is_superuser:
+            session.delete(user)
+            session.commit()
+            logger.debug("Default superuser removed successfully.")
+        else:
+            logger.debug("Default superuser not found.")
+
+
+def teardown_services():
+    """
+    Teardown all the services.
+    """
+    teardown_superuser()
+    service_manager.teardown()
+
+
+def initialize_settings_manager():
+    """
+    Initialize the settings manager.
+    """
+    from langflow.services.settings import factory as settings_factory
+
+    service_manager.register_factory(settings_factory.SettingsManagerFactory())
+
+
+def initialize_session_manager():
+    """
+    Initialize the session manager.
+    """
+    from langflow.services.session import factory as session_manager_factory  # type: ignore
+    from langflow.services.cache import factory as cache_factory
+
+    initialize_settings_manager()
+
+    service_manager.register_factory(
+        cache_factory.CacheManagerFactory(), dependencies=[ServiceType.SETTINGS_MANAGER]
+    )
+
+    service_manager.register_factory(
+        session_manager_factory.SessionManagerFactory(),
+        dependencies=[ServiceType.CACHE_MANAGER],
+    )
+
+    setup_superuser()
+
+
+def initialize_services():
+    """
+    Initialize all the services needed.
+    """
+    from langflow.services.database import factory as database_factory
+    from langflow.services.cache import factory as cache_factory
+    from langflow.services.chat import factory as chat_factory
+    from langflow.services.settings import factory as settings_factory
+    from langflow.services.auth import factory as auth_factory
+
+    service_manager.register_factory(settings_factory.SettingsManagerFactory())
+    service_manager.register_factory(
+        auth_factory.AuthManagerFactory(), dependencies=[ServiceType.SETTINGS_MANAGER]
+    )
+    service_manager.register_factory(
+        database_factory.DatabaseManagerFactory(),
+        dependencies=[ServiceType.SETTINGS_MANAGER],
+    )
+    service_manager.register_factory(cache_factory.CacheManagerFactory())
+    service_manager.register_factory(chat_factory.ChatManagerFactory())
+
+    # Test cache connection
+    service_manager.get(ServiceType.CACHE_MANAGER)
+    # Test database connection
+    service_manager.get(ServiceType.DATABASE_MANAGER)