tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

fix: disable swagger doc in prod

Browse source
This commit is contained in:
Timothy J. Baek 2024-01-07 02:48:21 -08:00
parent 70d2571be1
commit 04db2475f9
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
backend/main.py
View file

@ -1,3 +1,5 @@
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
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):
raise ex raise ex
app = FastAPI() app = FastAPI(docs_url="/docs" if ENV == "dev" else None, redoc_url=None)
origins = ["*"] origins = ["*"]