20da596d9a
✨ feat(base.py): add error handling when creating database and tables to provide more informative error messages
The code now checks if the `settings.database_url` is provided and raises a `RuntimeError` if it is not. This ensures that the application does not attempt to create a database connection without a valid URL. Additionally, error handling has been added when creating the database and tables. If an exception occurs during the creation process, an error message is logged and a `RuntimeError` is raised with a more informative error message. This helps in identifying and resolving any issues related to the database creation process.
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
from langflow.settings import settings
|
|
from sqlmodel import SQLModel, Session, create_engine
|
|
from langflow.utils.logger import logger
|
|
|
|
if settings.database_url and settings.database_url.startswith("sqlite"):
|
|
connect_args = {"check_same_thread": False}
|
|
else:
|
|
connect_args = {}
|
|
if not settings.database_url:
|
|
raise RuntimeError("No database_url provided")
|
|
engine = create_engine(settings.database_url, connect_args=connect_args)
|
|
|
|
|
|
def create_db_and_tables():
|
|
logger.debug("Creating database and tables")
|
|
try:
|
|
SQLModel.metadata.create_all(engine)
|
|
except Exception as exc:
|
|
logger.error(f"Error creating database and tables: {exc}")
|
|
raise RuntimeError("Error creating database and tables") from exc
|
|
# Now check if the table Flow exists, if not, something went wrong
|
|
# and we need to create the tables again.
|
|
from sqlalchemy import inspect
|
|
|
|
inspector = inspect(engine)
|
|
if "flow" not in inspector.get_table_names():
|
|
logger.error("Something went wrong creating the database and tables.")
|
|
logger.error("Please check your database settings.")
|
|
|
|
raise RuntimeError("Something went wrong creating the database and tables.")
|
|
else:
|
|
logger.debug("Database and tables created successfully")
|
|
|
|
|
|
def get_session():
|
|
with Session(engine) as session:
|
|
yield session
|