'commit'

dsLightRag/Start.py

@@ -20,6 +20,7 @@ from Routes.TeachingModel.api.DocumentController import router as document_route
 from Routes.TeachingModel.api.TeachingModelController import router as teaching_model_router
 from Routes.QA import router as qa_router
 from Routes.JiMengRoute import router as jimeng_router
+from Routes.SunoRoute import router as suno_router
 
 from Util.LightRagUtil import *
 from contextlib import asynccontextmanager
@@ -33,17 +34,19 @@ logger.addHandler(handler)
 
 
 @asynccontextmanager
-async def lifespan(_: FastAPI): 
+async def lifespan(_: FastAPI):
     pool = await init_postgres_pool()
     app.state.pool = pool
 
     asyncio.create_task(train_document_task())
-    
+
     try:
-        yield 
+        yield
     finally:
         # 应用关闭时销毁连接池
         await close_postgres_pool(pool)
+
+
 app = FastAPI(lifespan=lifespan)
 
 # 挂载静态文件目录
@@ -56,8 +59,9 @@ app.include_router(rag_router)  # LightRAG路由
 app.include_router(knowledge_router)  # 知识图谱路由
 app.include_router(oss_router)  # 阿里云OSS路由
 app.include_router(llm_router)  # 大模型路由
-app.include_router(qa_router) # 答疑路由
-app.include_router(jimeng_router) # 即梦路由
+app.include_router(qa_router)  # 答疑路由
+app.include_router(jimeng_router)  # 即梦路由
+app.include_router(suno_router)  # Suno路由
 
 # Teaching Model 相关路由
 # 登录相关（不用登录）
@@ -75,6 +79,5 @@ app.include_router(teaching_model_router, prefix="/api/teaching/model", tags=["t
 # 教学答疑
 app.include_router(teaching_model_router, prefix="/api/teaching/model", tags=["teacher_model"])
 
-
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8100)