From 4e32c76fa8ace5d3e32ccdf71c22f6374e01e7af Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 27 Jan 2024 14:17:03 -0800 Subject: [PATCH] (fix) return dict response --- litellm/proxy/proxy_server.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 263e858ea..300780086 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -2863,7 +2863,7 @@ async def google_login(): 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): import httpx @@ -2902,10 +2902,16 @@ async def google_callback(code: str): # 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. - return JSONResponse( - content={"user_email": user_email, "user_name": user_name}, - status_code=200, + response = await generate_key_helper_fn( + **{"duration": "24hr", "models": [], "aliases": {}, "config": {}, "spend": 0, "user_id": user_email, "team_id": "litellm-dashboard"} # type: ignore ) + + key = response["token"] # type: ignore + user_id = response["user_id"] # type: ignore + { + "key": key, + "user_id": user_id, + } else: # Handle user info retrieval error raise HTTPException(