From c9ae251f8538d46ce448280e36ec6c0ddd705d5f Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 5 Aug 2023 22:16:41 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(main.py):=20refactor=20imp?= =?UTF-8?q?orts=20and=20remove=20unused=20code=20for=20better=20organizati?= =?UTF-8?q?on=20and=20readability=20=E2=9C=A8=20feat(main.py):=20add=20ini?= =?UTF-8?q?tialization=20functions=20for=20services=20and=20database=20on?= =?UTF-8?q?=20app=20startup=20to=20ensure=20proper=20setup=20and=20configu?= =?UTF-8?q?ration?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/main.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/backend/langflow/main.py b/src/backend/langflow/main.py index deef1c914..a2da92c93 100644 --- a/src/backend/langflow/main.py +++ b/src/backend/langflow/main.py @@ -6,14 +6,14 @@ from fastapi.responses import FileResponse from fastapi.staticfiles import StaticFiles from langflow.api import router -from langflow.database.base import DatabaseManager from langflow.interface.utils import setup_llm_caching +from langflow.services.database.base import initialize_database +from langflow.services.manager import initialize_services from langflow.utils.logger import configure def create_app(): """Create the FastAPI app and include the router.""" - from langflow.settings import settings configure() @@ -34,11 +34,10 @@ def create_app(): allow_methods=["*"], allow_headers=["*"], ) - database_manager = DatabaseManager(settings.DATABASE_URL) app.include_router(router) - # app.on_event("startup")(Engine.update) - app.on_event("startup")(database_manager.run_migrations) - app.on_event("startup")(database_manager.create_db_and_tables) + + app.on_event("startup")(initialize_services) + app.on_event("startup")(initialize_database) app.on_event("startup")(setup_llm_caching) return app