jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update DatabaseService to use settings_service for database URL

Browse source
This commit is contained in:
ogabrielluiz 2024-06-03 13:08:12 -03:00
commit 5ea2699b29
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/base/langflow/services/database/service.py
View file

@ -20,7 +20,8 @@ from langflow.services.utils import teardown_superuser
if TYPE_CHECKING: if TYPE_CHECKING:
from sqlalchemy.engine import Engine from sqlalchemy.engine import Engine
from langflow.services.settings.manager import SettingsService
from langflow.services.settings.service import SettingsService
class DatabaseService(Service): class DatabaseService(Service):
@ -28,7 +29,9 @@ class DatabaseService(Service):
def __init__(self, settings_service: "SettingsService"): def __init__(self, settings_service: "SettingsService"):
self.settings_service = settings_service self.settings_service = settings_service
self.database_url = settings_service.settings.database_url if settings_service.settings.database_url is None:
raise ValueError("No database URL provided")
self.database_url: str = settings_service.settings.database_url
# This file is in langflow.services.database.manager.py # This file is in langflow.services.database.manager.py
# the ini is in langflow # the ini is in langflow
langflow_dir = Path(__file__).parent.parent.parent langflow_dir = Path(__file__).parent.parent.parent
@ -274,3 +277,4 @@ class DatabaseService(Service):
logger.error(f"Error tearing down database: {exc}") logger.error(f"Error tearing down database: {exc}")
self.engine.dispose() self.engine.dispose()
self.engine.dispose()