Skip to content

Commit 8c77c48

Browse files
committed
feat:在服务器启动时,自动从hugging face下载对应的重排序模型
1 parent f374963 commit 8c77c48

1 file changed

Lines changed: 11 additions & 4 deletions

File tree

backend/main.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,9 @@
1313

1414
from app.core.failed_response_register import register_exception_handlers
1515
from app.core.rate_limit import RateLimitMiddleware
16+
from app.core.logger_handler import logger
17+
18+
from app.rag.reorder_service import check_and_download_reranker_model
1619

1720
app = FastAPI()
1821

@@ -61,18 +64,22 @@ async def startup_event():
6164
"""应用启动时初始化会话管理器"""
6265
# 初始化数据库表结构
6366
await init_db()
64-
print("数据库表结构初始化完成")
67+
logger.info("数据库表结构初始化完成")
6568

6669
# 使用数据库版本的会话管理器
6770
await init_database_session_manager()
68-
print("数据库会话管理器初始化完成")
71+
logger.info("数据库会话管理器初始化完成")
6972

7073
# 连接Redis
7174
await connect_redis()
72-
print("Redis连接初始化完成")
75+
logger.info("Redis连接初始化完成")
76+
77+
# 检查并重排序模型
78+
check_and_download_reranker_model()
79+
logger.info("重排序模型检查完成")
7380

7481
@app.on_event("shutdown")
7582
async def shutdown_event():
7683
"""应用关闭时关闭Redis连接"""
7784
await close_redis()
78-
print("Redis连接已关闭")
85+
logger.info("Redis连接已关闭")

0 commit comments

Comments
 (0)