From 5ea2699b29629ea6b4298e1fa5c9702808027854 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 3 Jun 2024 13:08:12 -0300 Subject: [PATCH] refactor: Update DatabaseService to use settings_service for database URL --- src/backend/base/langflow/services/database/service.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/services/database/service.py b/src/backend/base/langflow/services/database/service.py index df1bbe224..cf3795610 100644 --- a/src/backend/base/langflow/services/database/service.py +++ b/src/backend/base/langflow/services/database/service.py @@ -20,7 +20,8 @@ from langflow.services.utils import teardown_superuser if TYPE_CHECKING: from sqlalchemy.engine import Engine - from langflow.services.settings.manager import SettingsService + + from langflow.services.settings.service import SettingsService class DatabaseService(Service): @@ -28,7 +29,9 @@ class DatabaseService(Service): def __init__(self, settings_service: "SettingsService"): 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 # the ini is in langflow langflow_dir = Path(__file__).parent.parent.parent @@ -274,3 +277,4 @@ class DatabaseService(Service): logger.error(f"Error tearing down database: {exc}") self.engine.dispose() + self.engine.dispose()