diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 7a4c7dba5..948d913d8 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -528,10 +528,9 @@ async def user_api_key_auth( # Token passed all checks api_key = valid_token.token - hashed_token = hash_token(api_key) # Add hashed token to cache - user_api_key_cache.set_cache(key=hashed_token, value=valid_token, ttl=60) + user_api_key_cache.set_cache(key=api_key, value=valid_token, ttl=60) valid_token_dict = _get_pydantic_json_dict(valid_token) valid_token_dict.pop("token", None) """ @@ -1455,6 +1454,7 @@ async def generate_key_helper_fn( saved_token["expires"] = saved_token["expires"].isoformat() if key_data["token"] is not None and isinstance(key_data["token"], str): hashed_token = hash_token(key_data["token"]) + saved_token["token"] = hashed_token user_api_key_cache.set_cache( key=hashed_token, value=LiteLLM_VerificationToken(**saved_token), # type: ignore