feat: disable openapi docs in production

vlecture · May 27, 2024 · 1947a79 · 1947a79
1 parent 62cfcc1
commit 1947a79
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -31,6 +31,7 @@
 )
 from src.utils.db import Base, engine
 
+from src.utils.settings import ENV_TYPE
 
 sentry_sdk.init(
     dsn=SENTRY_DSN,
@@ -43,7 +44,9 @@
     profiles_sample_rate=1.0,
 )
 
-app = FastAPI()
+OPENAPI_URL = "/openapi.json" if ENV_TYPE == "DEV" else ""
+
+app = FastAPI(openapi_url=OPENAPI_URL)
 
 # CORS
 app.add_middleware(