From 8a8233e428588dfe40dd28b4c9c4101fdc491e58 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 2 Apr 2024 21:40:35 -0700 Subject: [PATCH] fix safe use token id --- litellm/proxy/proxy_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 4f4519035e..060f64b94d 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -2566,7 +2566,7 @@ async def generate_key_helper_fn( create_key_response = await prisma_client.insert_data( data=key_data, table_name="key" ) - key_data["token_id"] = create_key_response.token + 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)