rag / app /database.py
Kareman's picture
feat(ContextAI)
14faba3
raw
history blame
277 Bytes
from sqlmodel import SQLModel, create_engine, Session
DATABASE_URL = "sqlite:///./users.db"
engine = create_engine(DATABASE_URL, echo=True)
def init_db():
SQLModel.metadata.create_all(engine)
def get_session():
with Session(engine) as session:
yield session