diff --git a/migrations/env.py b/migrations/env.py index 36112a3..876bd38 100644 --- a/migrations/env.py +++ b/migrations/env.py @@ -2,9 +2,12 @@ from sqlalchemy import engine_from_config from sqlalchemy import pool +from sqlmodel import SQLModel from alembic import context +from app.internal.db import ConfigTable + # this is the Alembic Config object, which provides # access to the values within the .ini file in use. config = context.config @@ -18,7 +21,7 @@ # for 'autogenerate' support # from myapp import mymodel # target_metadata = mymodel.Base.metadata -target_metadata = None +target_metadata = SQLModel.metadata # other values from the config, defined by the needs of env.py, # can be acquired: