phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(key_management_endpoints.py): return 'created_at' and 'updated_at' on /key/generate

Browse source 
Show 'created_at' on UI when key created
This commit is contained in:
Krrish Dholakia 2025-04-21 16:47:31 -07:00
parent 57b887478a
commit f13b56d09e
2 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/_types.py
View file

@ -687,6 +687,8 @@ class GenerateKeyResponse(KeyRequestBase):
token: Optional[str] = None token: Optional[str] = None
created_by: Optional[str] = None created_by: Optional[str] = None
updated_by: Optional[str] = None updated_by: Optional[str] = None
created_at: Optional[datetime] = None
updated_at: Optional[datetime] = None
@model_validator(mode="before") @model_validator(mode="before")
@classmethod @classmethod

3
litellm/proxy/management_endpoints/key_management_endpoints.py
View file

@ -1347,10 +1347,13 @@ async def generate_key_helper_fn( # noqa: PLR0915
create_key_response = await prisma_client.insert_data( create_key_response = await prisma_client.insert_data(
data=key_data, table_name="key" data=key_data, table_name="key"
) )
key_data["token_id"] = getattr(create_key_response, "token", None) key_data["token_id"] = getattr(create_key_response, "token", None)
key_data["litellm_budget_table"] = getattr( key_data["litellm_budget_table"] = getattr(
create_key_response, "litellm_budget_table", None 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: except Exception as e:
verbose_proxy_logger.error( verbose_proxy_logger.error(
"litellm.proxy.proxy_server.generate_key_helper_fn(): Exception occured - {}".format( "litellm.proxy.proxy_server.generate_key_helper_fn(): Exception occured - {}".format(