From 7a76621e80582f0ad7b439fc61150a3ee486fd64 Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Thu, 30 Nov 2023 12:07:35 -0500 Subject: [PATCH] Disable user-specified OpenAI Base URL in prod --- backend/main.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/backend/main.py b/backend/main.py index 91914e5..65fc15e 100644 --- a/backend/main.py +++ b/backend/main.py @@ -113,13 +113,15 @@ async def stream_code(websocket: WebSocket): # Get the OpenAI Base URL from the request. Fall back to environment variable if not provided. openai_base_url = None - if "openAiBaseURL" in params and params["openAiBaseURL"]: - openai_base_url = params["openAiBaseURL"] - print("Using OpenAI Base URL from client-side settings dialog") - else: - openai_base_url = os.environ.get("OPENAI_BASE_URL") - if openai_base_url: - print("Using OpenAI Base URL from environment variable") + # Disable user-specified OpenAI Base URL in prod + if not os.environ.get("IS_PROD"): + if "openAiBaseURL" in params and params["openAiBaseURL"]: + openai_base_url = params["openAiBaseURL"] + print("Using OpenAI Base URL from client-side settings dialog") + else: + openai_base_url = os.environ.get("OPENAI_BASE_URL") + if openai_base_url: + print("Using OpenAI Base URL from environment variable") if not openai_base_url: print("Using official OpenAI URL")