优化model

2025-07-17 16:17:57 +08:00
parent 9b30115444
commit 6ed606f7a4
11 changed files with 316 additions and 193 deletions
--- a/chat/db/session.py
+++ b/chat/db/session.py
@@ -1,5 +1,5 @@
 from sqlalchemy import create_engine
-from sqlalchemy.orm import sessionmaker
+from sqlalchemy.orm import sessionmaker, declarative_base
 from config import SQLALCHEMY_DATABASE_URL

 engine = create_engine(SQLALCHEMY_DATABASE_URL, pool_pre_ping=True)
@@ -10,4 +10,6 @@ def get_db():
    try:
        yield db
    finally:
-        db.close() 
+        db.close()
+
+Base = declarative_base()