phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(proxy_server.py): fix key info to handle pydantic v1

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-19 18:52:09 -08:00
parent b726b04309
commit 27d2f2246e
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/proxy/proxy_server.py
View file

@ -2227,7 +2227,11 @@ async def info_key_fn(
) )
key_info = await prisma_client.get_data(token=key) key_info = await prisma_client.get_data(token=key)
## REMOVE HASHED TOKEN INFO BEFORE RETURNING ## ## REMOVE HASHED TOKEN INFO BEFORE RETURNING ##
key_info = key_info.model_dump() try:
key_info = key_info.model_dump() # noqa
except:
# if using pydantic v1
key_info = key_info.dict()
key_info.pop("token") key_info.pop("token")
return {"key": key, "info": key_info} return {"key": key, "info": key_info}
except Exception as e: except Exception as e:
@ -2366,7 +2370,11 @@ async def user_info(
) )
## REMOVE HASHED TOKEN INFO before returning ## ## REMOVE HASHED TOKEN INFO before returning ##
for key in keys: for key in keys:
key = key.model_dump() try:
key = key.model_dump() # noqa
except:
# if using pydantic v1
key = key.dict()
key.pop("token", None) key.pop("token", None)
return {"user_id": user_id, "user_info": user_info, "keys": keys} return {"user_id": user_id, "user_info": user_info, "keys": keys}
except Exception as e: except Exception as e: