private-gpt/private_gpt/users/db/session.py
2024-03-03 17:12:09 +05:45

11 lines
523 B
Python

from private_gpt.users.core.db_config import SQLALCHEMY_DATABASE_URI
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
import logging
logging.basicConfig()
logging.getLogger("sqlalchemy.engine").setLevel(logging.INFO)
logging.getLogger("sqlalchemy.pool").setLevel(logging.DEBUG)
engine = create_engine(SQLALCHEMY_DATABASE_URI, echo=True,
future=True, pool_pre_ping=True, logging_name="myengine")
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)