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

🐛 fix(base.py): handle case where database and tables already exist to prevent error and log a debug message

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-08 17:41:17 -03:00
commit d6bb6bf3d9
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -3,6 +3,7 @@ import os
from sqlmodel import SQLModel, Session, create_engine from sqlmodel import SQLModel, Session, create_engine
from langflow.utils.logger import logger from langflow.utils.logger import logger
import sqlalchemy as sa
class Engine: class Engine:
@ -43,6 +44,13 @@ def create_db_and_tables():
logger.debug("Creating database and tables") logger.debug("Creating database and tables")
try: try:
SQLModel.metadata.create_all(Engine.get()) SQLModel.metadata.create_all(Engine.get())
except sa.exc.OperationalError as exc:
# Check if the error is because the table already exists
if "already exists" in str(exc):
logger.debug("Database and tables already exist")
else:
logger.error(f"Error creating database and tables: {exc}")
raise RuntimeError("Error creating database and tables") from exc
except Exception as exc: except Exception as exc:
logger.error(f"Error creating database and tables: {exc}") logger.error(f"Error creating database and tables: {exc}")
raise RuntimeError("Error creating database and tables") from exc raise RuntimeError("Error creating database and tables") from exc