phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

build(user.py): fix page param read issue

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-01 17:25:52 +05:30
parent ff3f3844c7
commit e1e3721917
2 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/proxy_server.py
View file

@ -1130,7 +1130,10 @@ async def completion(
else: # router is not set else: # router is not set
response = await litellm.atext_completion(**data) 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}") print_verbose(f"final response: {response}")
if ( if (

7
ui/pages/user.py
View file

@ -117,19 +117,20 @@ def user_page(page_param: str, user_id: str, token: str):
def router(): def router():
query_params = st.experimental_get_query_params() query_params = st.experimental_get_query_params()
page_param = query_params.get("page", None)[0] page_param = query_params.get("page", None)
if ( if (
query_params.get("token", None) is not None query_params.get("token", None) is not None
and query_params.get("user_id", None) is not None and query_params.get("user_id", None) is not None
and page_param is not None
): ):
# render user page # render user page
user_page( user_page(
page_param=page_param, page_param=page_param[0],
user_id=query_params.get("user_id")[0], user_id=query_params.get("user_id")[0],
token=query_params.get("token")[0], token=query_params.get("token")[0],
) )
elif page_param is not None: elif page_param is not None:
auth_page(page_param=page_param) auth_page(page_param=page_param[0])
else: else:
st.write("Please setup proxy") st.write("Please setup proxy")