From c80dbcc7eaa8471ab268fcacc72d78f1a20dc758 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 30 Jan 2024 22:06:33 -0800 Subject: [PATCH] fix(proxy_server.py): set expires to string before saving to cache --- litellm/proxy/proxy_server.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index a2d911799..bb7ff0ae7 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1370,6 +1370,10 @@ async def generate_key_helper_fn( saved_token["config"] = json.loads(saved_token["config"]) if isinstance(saved_token["metadata"], str): 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( key=key_data["token"], value=LiteLLM_VerificationToken(**saved_token), # type: ignore