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

fix(proxy_server.py): set expires to string before saving to cache

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-30 22:06:33 -08:00
parent 5429698bf3
commit c80dbcc7ea
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View file

@ -1370,6 +1370,10 @@ async def generate_key_helper_fn(
saved_token["config"] = json.loads(saved_token["config"]) saved_token["config"] = json.loads(saved_token["config"])
if isinstance(saved_token["metadata"], str): if isinstance(saved_token["metadata"], str):
saved_token["metadata"] = json.loads(saved_token["metadata"]) saved_token["metadata"] = json.loads(saved_token["metadata"])
if saved_token.get("expires", None) is not None and isinstance(
saved_token["expires"], datetime
):
saved_token["expires"] = saved_token["expires"].isoformat()
user_api_key_cache.set_cache( user_api_key_cache.set_cache(
key=key_data["token"], key=key_data["token"],
value=LiteLLM_VerificationToken(**saved_token), # type: ignore value=LiteLLM_VerificationToken(**saved_token), # type: ignore