From 1f3e24cc377b5f6c554954d6a4e9c0ad0b3cac39 Mon Sep 17 00:00:00 2001 From: Wenjie Zhang Date: Wed, 21 May 2025 02:56:14 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20knowledgebase=20=E5=88=9D=E5=A7=8B?= =?UTF-8?q?=E5=8C=96=E7=9A=84=E6=97=B6=E5=80=99=E8=A1=A8=E6=A0=BC=E6=B2=A1?= =?UTF-8?q?=E6=9C=89=E5=88=9B=E5=BB=BA=20#199?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/db_manager.py | 5 ++++- server/models/__init__.py | 7 ++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/server/db_manager.py b/server/db_manager.py index 04a966f3..35af3401 100644 --- a/server/db_manager.py +++ b/server/db_manager.py @@ -7,6 +7,7 @@ from contextlib import contextmanager from src import config from server.models import Base from server.models.user_model import User +from server.models.kb_models import KnowledgeDatabase, KnowledgeFile, KnowledgeNode from src.utils import logger class DBManager: @@ -34,6 +35,8 @@ class DBManager: """创建数据库表""" # 确保所有表都会被创建 Base.metadata.create_all(self.engine) + logger.info("数据库表创建/检查完成") + def get_session(self): """获取数据库会话""" return self.Session() @@ -52,7 +55,6 @@ class DBManager: finally: session.close() - def check_first_run(self): """检查是否首次运行""" session = self.get_session() @@ -61,5 +63,6 @@ class DBManager: return session.query(User).count() == 0 finally: session.close() + # 创建全局数据库管理器实例 db_manager = DBManager() \ No newline at end of file diff --git a/server/models/__init__.py b/server/models/__init__.py index 7c2377ae..66ad4806 100644 --- a/server/models/__init__.py +++ b/server/models/__init__.py @@ -1,3 +1,8 @@ from sqlalchemy.ext.declarative import declarative_base -Base = declarative_base() \ No newline at end of file +Base = declarative_base() + +# 导入所有模型文件，确保它们被注册到Base.metadata +from . import user_model +from . import kb_models +from . import thread_model \ No newline at end of file