diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index ba756d05ca..e3cd257fdc 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1902,14 +1902,6 @@ async def generate_key_helper_fn( saved_token["expires"], datetime ): 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 - ttl=600, - ) if prisma_client is not None: ## CREATE USER (If necessary) verbose_proxy_logger.debug(f"prisma_client: Creating User={user_data}") diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 0d78b6d3c5..d8d63e75f4 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -1241,7 +1241,7 @@ async def test_call_with_key_never_over_budget(prisma_client): # use generated key to auth in result = await user_api_key_auth(request=request, api_key=bearer_token) - print("result from user auth with new key", result) + print("result from user auth with new key: {result}") # update spend using track_cost callback, make 2nd request, it should fail from litellm.proxy.proxy_server import (