From dc78d16d055b2a6895dace5c9c1ce70933312bfb Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 27 Jan 2024 15:25:00 -0800
Subject: [PATCH] (fix) auth google

---
 litellm/proxy/proxy_server.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 6b5502b3c..3da2e40e3 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -2875,8 +2875,10 @@ async def google_callback(code: str, request: Request):
     import httpx
 
     scheme = request.url.scheme
-    host = request.url.hostname
+    host = request.url.hostname or "localhost"
     port = request.url.port or 4000
+    if "localhost" not in host:
+        scheme = "https"
 
     GOOGLE_REDIRECT_URI = f"{scheme}://{host}:{port}/google-callback"
     GOOGLE_CLIENT_ID = (