Commit 04db2475 authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

fix: disable swagger doc in prod

parent 70d2571b
import time
from fastapi import FastAPI, Request from fastapi import FastAPI, Request
from fastapi.staticfiles import StaticFiles from fastapi.staticfiles import StaticFiles
from fastapi import HTTPException from fastapi import HTTPException
...@@ -12,8 +14,7 @@ from apps.openai.main import app as openai_app ...@@ -12,8 +14,7 @@ from apps.openai.main import app as openai_app
from apps.web.main import app as webui_app from apps.web.main import app as webui_app
from apps.rag.main import app as rag_app from apps.rag.main import app as rag_app
from config import ENV
import time
class SPAStaticFiles(StaticFiles): class SPAStaticFiles(StaticFiles):
...@@ -27,7 +28,7 @@ class SPAStaticFiles(StaticFiles): ...@@ -27,7 +28,7 @@ class SPAStaticFiles(StaticFiles):
raise ex raise ex
app = FastAPI() app = FastAPI(docs_url="/docs" if ENV == "dev" else None, redoc_url=None)
origins = ["*"] origins = ["*"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment