fix issue with openAiBaseURL not being set

yjc930214 · Nov 30, 2023 · fa6f579 · fa6f579
1 parent 0c532e5
commit fa6f579
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/main.py b/backend/main.py
@@ -113,7 +113,7 @@ 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 params["openAiBaseURL"]:
+    if "openAiBaseURL" in params and params["openAiBaseURL"]:
         openai_base_url = params["openAiBaseURL"]
         print("Using OpenAI Base URL from client-side settings dialog")
     else:
@@ -122,7 +122,7 @@ async def stream_code(websocket: WebSocket):
             print("Using OpenAI Base URL from environment variable")
 
     if not openai_base_url:
-        print("Using Offical OpenAI Base URL")
+        print("Using offical OpenAI URL")
 
     # Get the image generation flag from the request. Fall back to True if not provided.
     should_generate_images = (