修复模型训练问题

backend/app/engines/text_engine.py

@@ -13,6 +13,7 @@ from typing import Any
 
 from app.config import get_settings
 from app.core.logging import logger
+from app.core.websocket import send_epoch_done, send_progress, send_completed
 from app.engines.base import BaseEngine
 
 settings = get_settings()