48 lines
1.0 KiB
Python
48 lines
1.0 KiB
Python
from sqlmodel import SQLModel
|
|
|
|
from .engine import engine
|
|
from .model_aii import AiiModel, AiiModelPublic, AiiProvider, AiiProviderPublic, z_aii_model, z_aii_provider
|
|
from .model_story import (
|
|
Chatroom,
|
|
ChatroomChat,
|
|
ChatroomChatAccept,
|
|
ChatroomChatDelete,
|
|
ChatroomChatEdit,
|
|
ChatroomPublic,
|
|
ChatScript,
|
|
ScriptTemplate,
|
|
)
|
|
from .model_user import ModelUploadFile, ModelUser
|
|
from .session import async_get_session, get_session
|
|
|
|
|
|
# 创建数据库连接和数据库文件
|
|
def create_db() -> None: # noqa: RUF067
|
|
try:
|
|
SQLModel.metadata.create_all(engine)
|
|
except Exception as e:
|
|
raise RuntimeError(f"连接或创建数据库失败：{e}") from e
|
|
|
|
|
|
__all__ = [
|
|
AiiModel,
|
|
AiiModelPublic,
|
|
AiiProvider,
|
|
AiiProviderPublic,
|
|
ChatScript,
|
|
Chatroom,
|
|
ChatroomChat,
|
|
ChatroomChatAccept,
|
|
ChatroomChatDelete,
|
|
ChatroomChatEdit,
|
|
ChatroomPublic,
|
|
ModelUploadFile,
|
|
ModelUser,
|
|
ScriptTemplate,
|
|
async_get_session,
|
|
create_db,
|
|
get_session,
|
|
z_aii_model,
|
|
z_aii_provider,
|
|
]
|