from typing import Generator

# Use the improved get_db function from session.py
from app.db.session import get_db