difyq 接入

zhaoqingang

2024-11-25 dc478b065693dd24e4cae719186d6aafb2d24f6d

 app/models/base_model.py

@@ -1,10 +1,12 @@
import os

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker, Session

from app.config.config import settings

DATABASE_URL = settings.database_url
DATABASE_URL = os.getenv('DATABASE_URL') or settings.database_url

engine = create_engine(DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)