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

(UI) - Improvements to session handling logic (#8970)

Browse source 
* add cookieUtils

* use utils for clearing cookies

* on logout use clearTokenCookies

* ui use correct clearTokenCookies

* navbar show userEmail on UserID page

* add timestamp on token cookie

* update generate_authenticated_redirect_response

* use common getAuthToken

* fix clearTokenCookies

* fixes for get auth token

* fix invitation link sign in logic

* Revert "fix invitation link sign in logic"

This reverts commit 30e5308cb3.

* fix getAuthToken

* update setAuthToken

* fix ui session handling

* fix ui session handler
This commit is contained in:
Ishaan Jaff 2025-03-03 22:17:21 -08:00 • committed by GitHub
parent 79ed4b626e
commit c015fb34f1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 117 additions and 52 deletions
Download patch file Download diff file Expand all files Collapse all files

24
litellm/proxy/management_helpers/ui_session_handler.py Normal file
View file

@ -0,0 +1,24 @@
import time
from fastapi.responses import RedirectResponse
class UISessionHandler:
@staticmethod
def generate_authenticated_redirect_response(
redirect_url: str, jwt_token: str
) -> RedirectResponse:
redirect_response = RedirectResponse(url=redirect_url, status_code=303)
redirect_response.set_cookie(
key=UISessionHandler._generate_token_name(),
value=jwt_token,
secure=True,
samesite="strict",
)
return redirect_response
@staticmethod
def _generate_token_name() -> str:
current_timestamp = int(time.time())
cookie_name = f"token_{current_timestamp}"
return cookie_name