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

(fix) return dict response

Browse source
This commit is contained in:
ishaan-jaff 2024-01-27 14:17:03 -08:00
parent dabc4591e7
commit 4e32c76fa8
1 changed files with 10 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/proxy/proxy_server.py
View file

@ -2863,7 +2863,7 @@ async def google_login():
return RedirectResponse(url=google_auth_url) return RedirectResponse(url=google_auth_url)
@app.get("/google-callback", tags=["experimental"]) @app.get("/google-callback", tags=["experimental"], response_model=GenerateKeyResponse)
async def google_callback(code: str): async def google_callback(code: str):
import httpx import httpx
@ -2902,10 +2902,16 @@ async def google_callback(code: str):
# we can use user_email on litellm proxy now # we can use user_email on litellm proxy now
# TODO: Handle user info as needed, for example, store it in a database, authenticate the user, etc. # TODO: Handle user info as needed, for example, store it in a database, authenticate the user, etc.
return JSONResponse( response = await generate_key_helper_fn(
content={"user_email": user_email, "user_name": user_name}, **{"duration": "24hr", "models": [], "aliases": {}, "config": {}, "spend": 0, "user_id": user_email, "team_id": "litellm-dashboard"} # type: ignore
status_code=200,
) )
key = response["token"] # type: ignore
user_id = response["user_id"] # type: ignore
{
"key": key,
"user_id": user_id,
}
else: else:
# Handle user info retrieval error # Handle user info retrieval error
raise HTTPException( raise HTTPException(