diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index e13f162656..56814cbf6e 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -2854,8 +2854,11 @@ async def user_auth(request: Request): @app.get("/google-login", tags=["experimental"]) -async def google_login(): - GOOGLE_REDIRECT_URI = "http://localhost:4000/google-callback" +async def google_login(request: Request): + scheme = request.url.scheme + host = request.url.hostname + port = request.url.port or 4000 + GOOGLE_REDIRECT_URI = f"{scheme}://{host}:{port}/google-callback" GOOGLE_CLIENT_ID = ( "246483686424-clje5sggkjma26ilktj6qssakqhoon0m.apps.googleusercontent.com" ) @@ -2864,10 +2867,14 @@ async def google_login(): @app.get("/google-callback", tags=["experimental"], response_model=GenerateKeyResponse) -async def google_callback(code: str): +async def google_callback(code: str, request: Request): import httpx - GOOGLE_REDIRECT_URI = "http://localhost:4000/google-callback" + scheme = request.url.scheme + host = request.url.hostname + port = request.url.port or 4000 + + GOOGLE_REDIRECT_URI = f"{scheme}://{host}:{port}/google-callback" GOOGLE_CLIENT_ID = ( "246483686424-clje5sggkjma26ilktj6qssakqhoon0m.apps.googleusercontent.com" )