浏览代码

fix: disable swagger doc in prod

Timothy J. Baek 1 年之前
父节点
当前提交
04db2475f9
共有 1 个文件被更改,包括 4 次插入3 次删除
  1. 4 3
      backend/main.py

+ 4 - 3
backend/main.py

@@ -1,3 +1,5 @@
+import time
+
 from fastapi import FastAPI, Request
 from fastapi.staticfiles import StaticFiles
 from fastapi import HTTPException
@@ -12,8 +14,7 @@ from apps.openai.main import app as openai_app
 from apps.web.main import app as webui_app
 from apps.rag.main import app as rag_app
 
-
-import time
+from config import ENV
 
 
 class SPAStaticFiles(StaticFiles):
@@ -27,7 +28,7 @@ class SPAStaticFiles(StaticFiles):
                 raise ex
 
 
-app = FastAPI()
+app = FastAPI(docs_url="/docs" if ENV == "dev" else None, redoc_url=None)
 
 origins = ["*"]