vent_ai_platform/app/db/init_db.py

from __future__ import annotations

from app.db.base import Base
from app.db.session import SessionLocal, engine

# 导入模型后 Base.metadata 才能完整建表
from app.db import models  # noqa: F401


def init_db() -> None:
    Base.metadata.create_all(bind=engine)
    from app.services.document_business_templates import ensure_default_templates

    with SessionLocal() as db:
        ensure_default_templates(db)


def drop_db() -> None:
    Base.metadata.drop_all(bind=engine)