fix: disable swagger doc in prod

04db2475 · Timothy J. Baek · 70d2571b · 04db2475
Commit 04db2475 authored Jan 07, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

backend/main.py backend/main.py +4 -3

No files found.
--- a/backend/main.py
+++ b/backend/main.py
+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 = ["*"]