From b522edcdf1f097b1668fed3d7e4e4f0392165d2b Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 27 Jan 2024 15:36:52 -0800
Subject: [PATCH] Update proxy_server.py

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

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 3da2e40e3f..10748042ca 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -2862,6 +2862,8 @@ async def google_login(request: Request):
     scheme = request.url.scheme
     host = request.url.hostname
     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 = (
         "246483686424-clje5sggkjma26ilktj6qssakqhoon0m.apps.googleusercontent.com"