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

fix(proxy_server.py): add custom json helper func to work across multiple pydantic versions

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-09 11:12:52 -08:00
parent 7ed68da149
commit c1c81af9d4
2 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View file

@ -1128,9 +1128,7 @@ async def generate_key_fn(request: Request, data: GenerateKeyRequest, Authorizat
- user_id: (str) Unique user id - used for tracking spend across multiple keys for same user id.
"""
# data = await request.json()
print(f"type of object: {type(data)}")
data_json = json.loads(data.model_dump_json()) # type: ignore
print(f"data_json: {data_json}")
data_json = json.loads(data.json()) # type: ignore
response = await generate_key_helper_fn(**data_json)
return GenerateKeyResponse(key=response["token"], expires=response["expires"], user_id=response["user_id"])