完成训练模块的转移

main.py

@@ -17,7 +17,7 @@ from starlette.middleware.cors import CORSMiddleware
 from application import settings
 from application import urls
 from starlette.staticfiles import StaticFiles  # 依赖安装：pip install aiofiles
-from core.docs import custom_api_docs
+from core.websocket_app import websocket_config
 from core.exception import register_exception
 import typer
 from scripts.initialize.initialize import InitializeData, Environment
@@ -64,7 +64,9 @@ def create_app():
     for url in urls.urlpatterns:
         app.include_router(url["ApiRouter"], prefix=url["prefix"], tags=url["tags"])
     # 配置接口文档静态资源
-    custom_api_docs(app)
+    # custom_api_docs(app)
+    # 开启websocket
+    websocket_config(app)
     return app