phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add redirect_ui_middleware

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-06 18:01:23 -07:00
parent 070d8f60f8
commit e207ac577e
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/proxy/proxy_server.py
View file

@ -285,8 +285,6 @@ except Exception as e:
server_root_path = os.getenv("SERVER_ROOT_PATH", "") server_root_path = os.getenv("SERVER_ROOT_PATH", "")
print("server root path: ", server_root_path) # noqa print("server root path: ", server_root_path) # noqa
if server_root_path != "":
setup_admin_ui_on_server_root_path(server_root_path)
_license_check = LicenseCheck() _license_check = LicenseCheck()
premium_user: bool = _license_check.is_premium() premium_user: bool = _license_check.is_premium()
ui_link = f"{server_root_path}/ui/" ui_link = f"{server_root_path}/ui/"
@ -388,6 +386,16 @@ try:
src = os.path.join(ui_path, filename) src = os.path.join(ui_path, filename)
dst = os.path.join(folder_path, "index.html") dst = os.path.join(folder_path, "index.html")
os.rename(src, dst) os.rename(src, dst)
if server_root_path != "":
@app.middleware("http")
async def redirect_ui_middleware(request: Request, call_next):
if request.url.path.startswith("/ui"):
new_path = request.url.path.replace("/ui", f"{server_root_path}/ui", 1)
return RedirectResponse(new_path)
return await call_next(request)
except: except:
pass pass
app.add_middleware( app.add_middleware(