forked from phoenix/litellm-mirror
(fix) UI Proxy - handle base url ending with /
This commit is contained in:
parent
cbb94a77ff
commit
a6f524680c
1 changed files with 8 additions and 10 deletions
|
@ -3171,23 +3171,21 @@ async def auth_callback(request: Request):
|
||||||
param="PROXY_BASE_URL",
|
param="PROXY_BASE_URL",
|
||||||
code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||||||
)
|
)
|
||||||
if proxy_base_url.endswith("/"):
|
|
||||||
proxy_base_url += "ui"
|
|
||||||
else:
|
|
||||||
proxy_base_url += "/ui"
|
|
||||||
litellm_dashboard_ui = proxy_base_url
|
litellm_dashboard_ui = proxy_base_url
|
||||||
|
if proxy_base_url.endswith("/"):
|
||||||
|
litellm_dashboard_ui += "ui"
|
||||||
|
proxy_base_url = proxy_base_url[:-1]
|
||||||
|
else:
|
||||||
|
litellm_dashboard_ui += "/ui"
|
||||||
|
|
||||||
import jwt
|
import jwt
|
||||||
|
|
||||||
jwt_token = jwt.encode(
|
jwt_token = jwt.encode(
|
||||||
{"user_id": user_id, "key": key}, "secret", algorithm="HS256"
|
{"user_id": user_id, "key": key}, "secret", algorithm="HS256"
|
||||||
)
|
)
|
||||||
litellm_dashboard_ui += (
|
litellm_dashboard_ui += (
|
||||||
"?userID="
|
"?userID=" + user_id + "&token=" + jwt_token + "&proxyBaseUrl=" + proxy_base_url
|
||||||
+ user_id
|
|
||||||
+ "&token="
|
|
||||||
+ jwt_token
|
|
||||||
+ "&proxyBaseUrl="
|
|
||||||
+ os.getenv("PROXY_BASE_URL")
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# if a user has logged in they should be allowed to create keys - this ensures that it's set to True
|
# if a user has logged in they should be allowed to create keys - this ensures that it's set to True
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue