phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #2809 from BerriAI/ui_use_token_id_in_key_gen

Browse source 
[UI] QA Fix Edit Key flow - return `token_id` in /key/generate respose
This commit is contained in:
Ishaan Jaff 2024-04-02 21:48:40 -07:00 • committed by GitHub
commit 326f95244a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/proxy/_types.py
View file

@ -306,6 +306,7 @@ class GenerateKeyResponse(GenerateKeyRequest):
key_name: Optional[str] = None
expires: Optional[datetime]
user_id: Optional[str] = None
token_id: Optional[str] = None
@root_validator(pre=True)
def set_model_info(cls, values):

5
litellm/proxy/proxy_server.py
View file

@ -2563,7 +2563,10 @@ async def generate_key_helper_fn(
## CREATE KEY
verbose_proxy_logger.debug("prisma_client: Creating Key= %s", key_data)
await prisma_client.insert_data(data=key_data, table_name="key")
create_key_response = await prisma_client.insert_data(
data=key_data, table_name="key"
)
key_data["token_id"] = getattr(create_key_response, "token", None)
elif custom_db_client is not None:
if table_name is None or table_name == "user":
## CREATE USER (If necessary)

5
ui/litellm-dashboard/src/components/view_key_table.tsx
View file

@ -213,6 +213,11 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
const handleEditClick = (token: any) => {
console.log("handleEditClick:", token);
// set token.token to token.token_id if token_id is not null
if (token.token_id !== null) {
token.token = token.token_id;
}
setSelectedToken(token);
setEditModalVisible(true);
};