mirror of
https://github.com/zylon-ai/private-gpt.git
synced 2025-12-22 20:12:55 +01:00
11 lines
244 B
Python
11 lines
244 B
Python
from private_gpt.users.db.session import SessionLocal
|
|
|
|
from sqlalchemy.orm import Session
|
|
|
|
def init_db(db: Session) -> None:
|
|
"""Database session generator"""
|
|
db = SessionLocal()
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|