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

v0 use token_in /key_generate

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-02 21:31:24 -07:00
parent afd81f1609
commit 15685a8f53
2 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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"] = create_key_response.token
elif custom_db_client is not None:
if table_name is None or table_name == "user":
## CREATE USER (If necessary)