From e1e37219174d796b4e2fef6d532944b9df15f2a2 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 1 Jan 2024 17:25:52 +0530 Subject: [PATCH] build(user.py): fix page param read issue --- litellm/proxy/proxy_server.py | 5 ++++- ui/pages/user.py | 7 ++++--- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index d7a9b1024..31e13ac19 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1130,7 +1130,10 @@ async def completion( else: # router is not set response = await litellm.atext_completion(**data) - model_id = response._hidden_params.get("model_id", None) or "" + if hasattr(response, "_hidden_params"): + model_id = response._hidden_params.get("model_id", None) or "" + else: + model_id = "" print_verbose(f"final response: {response}") if ( diff --git a/ui/pages/user.py b/ui/pages/user.py index 8bc65ca02..666458301 100644 --- a/ui/pages/user.py +++ b/ui/pages/user.py @@ -117,19 +117,20 @@ def user_page(page_param: str, user_id: str, token: str): def router(): query_params = st.experimental_get_query_params() - page_param = query_params.get("page", None)[0] + page_param = query_params.get("page", None) if ( query_params.get("token", None) is not None and query_params.get("user_id", None) is not None + and page_param is not None ): # render user page user_page( - page_param=page_param, + page_param=page_param[0], user_id=query_params.get("user_id")[0], token=query_params.get("token")[0], ) elif page_param is not None: - auth_page(page_param=page_param) + auth_page(page_param=page_param[0]) else: st.write("Please setup proxy")