this alembic file

alembic_db/README.md

@@ -1,4 +1,4 @@
 ## Generate new revision
 
-1. Update models in `/app/database/models.py`
+1. Update models in `/comfy/app/database/models.py`
 2. Run `alembic revision --autogenerate -m "{your message}"`

alembic_db/env.py

@@ -8,7 +8,7 @@ from alembic import context
 config = context.config
 
 
-from app.database.models import Base
+from comfy.app.database.models import Base
 target_metadata = Base.metadata
 
 # other values from the config, defined by the needs of env.py,

comfy/app/database/db.py

@@ -1,33 +1,19 @@
 import logging
 import os
 import shutil
-from app.logger import log_startup_warning
-from utils.install_util import get_missing_requirements_message
-from comfy.cli_args import args
 
-_DB_AVAILABLE = False
+from ...cli_args import args
+
 Session = None
 
+from alembic import command
+from alembic.config import Config
+from alembic.runtime.migration import MigrationContext
+from alembic.script import ScriptDirectory
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
 
-try:
+_DB_AVAILABLE = True
-    from alembic import command
-    from alembic.config import Config
-    from alembic.runtime.migration import MigrationContext
-    from alembic.script import ScriptDirectory
-    from sqlalchemy import create_engine
-    from sqlalchemy.orm import sessionmaker
-
-    _DB_AVAILABLE = True
-except ImportError as e:
-    log_startup_warning(
-        f"""
-------------------------------------------------------------------------
-Error importing dependencies: {e}
-{get_missing_requirements_message()}
-This error is happening because ComfyUI now uses a local sqlite database.
-------------------------------------------------------------------------
-""".strip()
-    )
 
 
 def dependencies_available():