diff --git a/src/chore/GenerationContext.py b/src/chore/GenerationContext.py index ec8cf2f..d2e2e28 100644 --- a/src/chore/GenerationContext.py +++ b/src/chore/GenerationContext.py @@ -8,6 +8,7 @@ from ..models import Video, SessionLocal class GenerationContext: def store_in_db(self): + return with SessionLocal() as db: db.add( Video(