diff --git a/startup.py b/startup.py index e610cda..3bb508a 100644 --- a/startup.py +++ b/startup.py @@ -36,6 +36,7 @@ from server.utils import (fschat_controller_address, fschat_model_worker_address fschat_openai_api_address, set_httpx_config, get_httpx_client, get_model_worker_config, get_all_model_worker_configs, MakeFastAPIOffline, FastAPI, llm_device, embedding_device) +from server.knowledge_base.migrate import create_tables import argparse from typing import Tuple, List, Dict from configs import VERSION @@ -866,6 +867,8 @@ async def start_main_server(): logger.info("Process status: %s", p) if __name__ == "__main__": + # 确保数据库表被创建 + create_tables() if sys.version_info < (3, 10): loop = asyncio.get_event_loop()