From 12113be6ecc16cec757c71af9ba9bf37a53643cc Mon Sep 17 00:00:00 2001
From: liunux4odoo <liunux@qq.com>
Date: Sat, 2 Dec 2023 10:52:29 +0800
Subject: [PATCH] =?UTF-8?q?=E5=9C=A8startup=E4=B8=AD=E8=87=AA=E5=8A=A8?=
 =?UTF-8?q?=E6=89=A7=E8=A1=8C=20create=5Ftables=EF=BC=8C=20=E7=A1=AE?=
 =?UTF-8?q?=E4=BF=9D=E6=95=B0=E6=8D=AE=E5=BA=93=E8=A1=A8=E8=A2=AB=E5=88=9B?=
 =?UTF-8?q?=E5=BB=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 startup.py | 3 +++
 1 file changed, 3 insertions(+)

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()