forked from phoenix/litellm-mirror
fix safe use token id
This commit is contained in:
parent
20b479725d
commit
8a8233e428
1 changed files with 1 additions and 1 deletions
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue