From dd7a51dd3468e423c625f07c79ba90bacfae4183 Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Wed, 31 Jul 2024 10:16:03 -0400 Subject: [PATCH] refactor --- backend/routes/generate_code.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py index 76b1fbe..d436e83 100644 --- a/backend/routes/generate_code.py +++ b/backend/routes/generate_code.py @@ -170,8 +170,8 @@ async def stream_code(websocket: WebSocket): openai_base_url: Union[str, None] = None # 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"] + openai_base_url = params.get("openAiBaseURL") + if openai_base_url: print("Using OpenAI Base URL from client-side settings dialog") else: openai_base_url = os.environ.get("OPENAI_BASE_URL")