diff --git a/services/ml/app/db.py b/services/ml/app/db.py index b051727..13811b7 100644 --- a/services/ml/app/db.py +++ b/services/ml/app/db.py @@ -87,18 +87,3 @@ def get_db() -> Generator[Session, None, None]: yield db finally: db.close() - - -def get_db_session() -> Session: - """ - Get a database session (for dependency injection). - - Usage with FastAPI: - @app.get("/") - def endpoint(db: Session = Depends(get_db_session)): - # Use db here - - Returns: - Database session (caller must close it) - """ - return SessionLocal()