'commit'

dsLightRag/Start.py

@@ -2,6 +2,7 @@ import uvicorn
 import asyncio
 from fastapi import FastAPI
 from starlette.staticfiles import StaticFiles
+from fastapi.middleware.cors import CORSMiddleware  # 添加此导入
 
 from Routes.TeachingModel.tasks.BackgroundTasks import train_document_task
 from Util.PostgreSQLUtil import init_postgres_pool, close_postgres_pool
@@ -26,6 +27,7 @@ from Routes.MjRoute import router as mj_router
 from Routes.QWenImageRoute import router as qwen_image_router
 from Util.LightRagUtil import *
 from contextlib import asynccontextmanager
+import logging  # 添加此导入
 
 # 控制日志输出
 logger = logging.getLogger('lightrag')
@@ -52,6 +54,15 @@ async def lifespan(_: FastAPI):
 
 app = FastAPI(lifespan=lifespan)
 
+# 添加CORS中间件
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # 允许所有来源，生产环境中可以限制为特定域名
+    allow_credentials=True,
+    allow_methods=["*"],  # 允许所有方法
+    allow_headers=["*"],  # 允许所有头部
+)
+
 # 挂载静态文件目录
 app.mount("/static", StaticFiles(directory="Static"), name="static")