28 lines
565 B
Python
28 lines
565 B
Python
|
from sqlalchemy.ext.declarative import declarative_base
|
||
|
|
|
||
|
|
Base = declarative_base()
|
||
|
|
|
||
|
|
# 导入所有模型，确保它们被注册到 Base.metadata
|
||
|
|
from src.storage.db.models import ( # noqa: E402, F401
|
||
|
|
User,
|
||
|
|
OperationLog,
|
||
|
|
Thread,
|
||
|
|
KnowledgeDatabase,
|
||
|
|
KnowledgeFile,
|
||
|
|
KnowledgeNode,
|
||
|
|
) # noqa: E402
|
||
|
|
|
||
|
|
# 导入数据库管理器
|
||
|
|
from src.storage.db.manager import db_manager # noqa: E402
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"Base",
|
||
|
|
"User",
|
||
|
|
"OperationLog",
|
||
|
|
"Thread",
|
||
|
|
"KnowledgeDatabase",
|
||
|
|
"KnowledgeFile",
|
||
|
|
"KnowledgeNode",
|
||
|
|
"db_manager",
|
||
|
|
]
|