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

(fix) use GOOGLE_REDIRECT_URI.

Browse source
This commit is contained in:
ishaan-jaff 2024-01-27 16:50:53 -08:00
parent 308351458b
commit 9d5bfa45c1
1 changed files with 23 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

23
litellm/proxy/proxy_server.py
View file

@ -2860,6 +2860,17 @@ async def user_auth(request: Request):
@app.get("/google-login/key/generate", tags=["experimental"]) @app.get("/google-login/key/generate", tags=["experimental"])
async def google_login(request: Request): async def google_login(request: Request):
GOOGLE_REDIRECT_URI = os.getenv("GOOGLE_REDIRECT_URI") GOOGLE_REDIRECT_URI = os.getenv("GOOGLE_REDIRECT_URI")
if GOOGLE_REDIRECT_URI is None:
raise ProxyException(
message="GOOGLE_REDIRECT_URI not set. Set it in .env file",
type="auth_error",
param="GOOGLE_REDIRECT_URI",
code=status.HTTP_500_INTERNAL_SERVER_ERROR,
)
if GOOGLE_REDIRECT_URI.endswith("/"):
GOOGLE_REDIRECT_URI += "google-callback"
else:
GOOGLE_REDIRECT_URI += "/google-callback"
GOOGLE_CLIENT_ID = ( GOOGLE_CLIENT_ID = (
"246483686424-clje5sggkjma26ilktj6qssakqhoon0m.apps.googleusercontent.com" "246483686424-clje5sggkjma26ilktj6qssakqhoon0m.apps.googleusercontent.com"
) )
@ -2872,6 +2883,18 @@ async def google_callback(code: str, request: Request):
import httpx import httpx
GOOGLE_REDIRECT_URI = os.getenv("GOOGLE_REDIRECT_URI") GOOGLE_REDIRECT_URI = os.getenv("GOOGLE_REDIRECT_URI")
if GOOGLE_REDIRECT_URI is None:
raise ProxyException(
message="GOOGLE_REDIRECT_URI not set. Set it in .env file",
type="auth_error",
param="GOOGLE_REDIRECT_URI",
code=status.HTTP_500_INTERNAL_SERVER_ERROR,
)
# Add "/google-callback"" to your callback URL
if GOOGLE_REDIRECT_URI.endswith("/"):
GOOGLE_REDIRECT_URI += "google-callback"
else:
GOOGLE_REDIRECT_URI += "/google-callback"
GOOGLE_CLIENT_ID = ( GOOGLE_CLIENT_ID = (
"246483686424-clje5sggkjma26ilktj6qssakqhoon0m.apps.googleusercontent.com" "246483686424-clje5sggkjma26ilktj6qssakqhoon0m.apps.googleusercontent.com"
) )