Reorder setup and cleanup tasks in database initialization process

langflow-ai · Nov 19, 2024 · 092d899 · 092d899
1 parent 3be43f5
commit 092d899
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/backend/base/langflow/services/utils.py b/src/backend/base/langflow/services/utils.py
@@ -244,12 +244,12 @@ async def initialize_services(*, fix_migration: bool = False) -> None:
     await asyncio.to_thread(initialize_database, fix_migration=fix_migration)
     async with get_db_service().with_async_session() as session:
         settings_service = get_service(ServiceType.SETTINGS_SERVICE)
+        await setup_superuser(settings_service, session)
     try:
         await get_db_service().migrate_flows_if_auto_login()
     except Exception as exc:
         msg = "Error migrating flows"
         logger.exception(msg)
         raise RuntimeError(msg) from exc
-        await setup_superuser(settings_service, session)
-        await clean_transactions(settings_service, session)
-        await clean_vertex_builds(settings_service, session)
+    await clean_transactions(settings_service, session)
+    await clean_vertex_builds(settings_service, session)