From f13b56d09e9d88bef26948c9c89a9968d4dd2a38 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 21 Apr 2025 16:47:31 -0700 Subject: [PATCH] fix(key_management_endpoints.py): return 'created_at' and 'updated_at' on `/key/generate` Show 'created_at' on UI when key created --- litellm/proxy/_types.py | 2 ++ litellm/proxy/management_endpoints/key_management_endpoints.py | 3 +++ 2 files changed, 5 insertions(+) diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 150528fd8f..354f6bb54c 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -687,6 +687,8 @@ class GenerateKeyResponse(KeyRequestBase): token: Optional[str] = None created_by: Optional[str] = None updated_by: Optional[str] = None + created_at: Optional[datetime] = None + updated_at: Optional[datetime] = None @model_validator(mode="before") @classmethod diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py index 8fd3b555d4..15edab8909 100644 --- a/litellm/proxy/management_endpoints/key_management_endpoints.py +++ b/litellm/proxy/management_endpoints/key_management_endpoints.py @@ -1347,10 +1347,13 @@ async def generate_key_helper_fn( # noqa: PLR0915 create_key_response = await prisma_client.insert_data( data=key_data, table_name="key" ) + key_data["token_id"] = getattr(create_key_response, "token", None) key_data["litellm_budget_table"] = getattr( create_key_response, "litellm_budget_table", None ) + key_data["created_at"] = getattr(create_key_response, "created_at", None) + key_data["updated_at"] = getattr(create_key_response, "updated_at", None) except Exception as e: verbose_proxy_logger.error( "litellm.proxy.proxy_server.generate_key_helper_fn(): Exception occured - {}".format(