知识库和智能体同步

zhaoqingang

2024-12-18 c941b948f1fa8eff615985a83110a401ae426ab3

 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)