🔧 chore(main.py): refactor imports and remove unused code for better organization and readability

✨ feat(main.py): add initialization functions for services and database on app startup to ensure proper setup and configuration
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-05 22:16:41 -03:00
commit c9ae251f85

View file

@ -6,14 +6,14 @@ from fastapi.responses import FileResponse
from fastapi.staticfiles import StaticFiles from fastapi.staticfiles import StaticFiles
from langflow.api import router from langflow.api import router
from langflow.database.base import DatabaseManager
from langflow.interface.utils import setup_llm_caching 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 from langflow.utils.logger import configure
def create_app(): def create_app():
"""Create the FastAPI app and include the router.""" """Create the FastAPI app and include the router."""
from langflow.settings import settings
configure() configure()
@ -34,11 +34,10 @@ def create_app():
allow_methods=["*"], allow_methods=["*"],
allow_headers=["*"], allow_headers=["*"],
) )
database_manager = DatabaseManager(settings.DATABASE_URL)
app.include_router(router) app.include_router(router)
# app.on_event("startup")(Engine.update)
app.on_event("startup")(database_manager.run_migrations) app.on_event("startup")(initialize_services)
app.on_event("startup")(database_manager.create_db_and_tables) app.on_event("startup")(initialize_database)
app.on_event("startup")(setup_llm_caching) app.on_event("startup")(setup_llm_caching)
return app return app