🔀 refactor(alembic/env.py): update import statement for SQLModel in env.py to reflect new location in manager module

🔀 refactor(interface/custom/custom_component.py): update import statement for session_getter in custom_component.py to reflect new location in utils module

🔀 refactor(main.py): update import statement for initialize_database in main.py to reflect new location in utils module

🔀 refactor(database/factory.py): update import statement for DatabaseManager in factory.py to reflect new location in manager module

src/backend/langflow/alembic/env.py

@@ -5,7 +5,7 @@ from sqlalchemy import pool
 
 from alembic import context
 
-from langflow.services.database.base import SQLModel
+from langflow.services.database.manager import SQLModel
 
 # this is the Alembic Config object, which provides
 # access to the values within the .ini file in use.

src/backend/langflow/interface/custom/custom_component.py

@@ -7,7 +7,7 @@ from langflow.services.utils import get_db_manager
 
 from langflow.utils import validate
 
-from langflow.services.database.base import session_getter
+from langflow.services.database.utils import session_getter
 from langflow.services.database.models.flow import Flow
 from pydantic import Extra
 import yaml

src/backend/langflow/main.py

@@ -7,7 +7,7 @@ from fastapi.staticfiles import StaticFiles
 
 from langflow.api import router
 from langflow.interface.utils import setup_llm_caching
-from langflow.services.database.base import initialize_database
+from langflow.services.database.utils import initialize_database
 from langflow.services.manager import initialize_services
 from langflow.utils.logger import configure
 

src/backend/langflow/services/database/factory.py

@@ -1,5 +1,5 @@
 from typing import TYPE_CHECKING
-from langflow.services.database.base import DatabaseManager
+from langflow.services.database.manager import DatabaseManager
 from langflow.services.factory import ServiceFactory
 
 if TYPE_CHECKING: