Langchain-Chatchat/libs/chatchat-server/chatchat/server/db/base.py

18 lines
474 B
Python
Raw Normal View History

2024-12-20 16:04:03 +08:00
import json
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import DeclarativeMeta, declarative_base
from sqlalchemy.orm import sessionmaker
from chatchat.settings import Settings
engine = create_engine(
Settings.basic_settings.SQLALCHEMY_DATABASE_URI,
json_serializer=lambda obj: json.dumps(obj, ensure_ascii=False),
)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base: DeclarativeMeta = declarative_base()