update generate_authenticated_redirect_response

litellm/proxy/management_helpers/ui_session_handler.py

@@ -0,0 +1,23 @@
+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,
+        )
+        return redirect_response
+
+    @staticmethod
+    def _generate_token_name() -> str:
+        current_timestamp = int(time.time())
+        cookie_name = f"token_{current_timestamp}"
+        return cookie_name