Disable docs endpoints in prod env

osuAkatsuki · Jun 22, 2024 · 6b8b477 · 6b8b477
1 parent 225eeb7
commit 6b8b477
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/app/init_api.py b/app/init_api.py
@@ -5,6 +5,7 @@
 from fastapi import Response
 from starlette.middleware.base import RequestResponseEndpoint
 
+from app import settings
 from app.api import api_router
 
 
@@ -29,7 +30,10 @@ async def http_middleware(
 
 
 def init_api() -> FastAPI:
-    app = FastAPI()
+    app = FastAPI(
+        docs_url="/docs" if settings.APP_ENV != "production" else None,
+        redoc_url="/redoc" if settings.APP_ENV != "production" else None,
+    )
     app = init_routes(app)
     app = init_middleware(app)
     return app