Kaynağa Gözat

fix: disable swagger doc in prod

Timothy J. Baek 1 yıl önce
ebeveyn
işleme
04db2475f9
1 değiştirilmiş dosya ile 4 ekleme ve 3 silme
  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 = ["*"]