You can set the PYTHONPATH environment variable to control what python sees as the top level folder, eg. if you are in the root folder of your project:
PYTHONPATH=. alembic revision -m "..."
Then you can use a “normal” import in your alembic env.py, relative to your root folder, in your example:
from src.models.base import Base