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