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

fix include request_typ in key_generate requests

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-31 18:46:36 -07:00
parent 8f87da4b20
commit ffa16a2f6a
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/proxy_server.py
View file

@ -12611,6 +12611,7 @@ async def login(request: Request):
if secrets.compare_digest(password, _password):
if os.getenv("DATABASE_URL") is not None:
response = await generate_key_helper_fn(
request_type="key",
**{ # type: ignore
"user_role": user_role,
"duration": "2hr",
@ -12621,7 +12622,7 @@ async def login(request: Request):
"spend": 0,
"user_id": user_id,
"team_id": "litellm-dashboard",
}
},
)
else:
raise ProxyException(
@ -12727,6 +12728,7 @@ async def onboarding(invite_link: str):
user_email = user_obj.user_email
response = await generate_key_helper_fn(
request_type="key",
**{
"user_role": user_obj.user_role,
"duration": "2hr",
@ -12737,7 +12739,7 @@ async def onboarding(invite_link: str):
"spend": 0,
"user_id": user_obj.user_id,
"team_id": "litellm-dashboard",
} # type: ignore
}, # type: ignore
)
key = response["token"] # type: ignore