Преглед на файлове

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 = ["*"]