From 5bdb1c99bb6102dbf10886b80e95b7f3da3332a6 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Tue, 17 Dec 2024 13:52:57 -0800 Subject: [PATCH] refac --- backend/open_webui/main.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py index 48cb8c5b3..e7f602311 100644 --- a/backend/open_webui/main.py +++ b/backend/open_webui/main.py @@ -18,6 +18,8 @@ from typing import Optional from aiocache import cached import aiohttp import requests + + from fastapi import ( Depends, FastAPI, @@ -29,8 +31,9 @@ from fastapi import ( status, applications, ) -from fastapi.middleware.cors import CORSMiddleware from fastapi.openapi.docs import get_swagger_ui_html + +from fastapi.middleware.cors import CORSMiddleware from fastapi.responses import JSONResponse, RedirectResponse from fastapi.staticfiles import StaticFiles @@ -1109,7 +1112,7 @@ app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static") app.mount("/cache", StaticFiles(directory=CACHE_DIR), name="cache") -def swagger_monkey_patch(*args, **kwargs): +def swagger_ui_html(*args, **kwargs): return get_swagger_ui_html( *args, **kwargs, @@ -1119,7 +1122,7 @@ def swagger_monkey_patch(*args, **kwargs): ) -applications.get_swagger_ui_html = swagger_monkey_patch +applications.get_swagger_ui_html = swagger_ui_html if os.path.exists(FRONTEND_BUILD_DIR): mimetypes.add_type("text/javascript", ".js")