Corrected OpenAI API usage

.dockerignore

@@ -15,6 +15,7 @@ usage/*
 feedback/*
 test_code/*
 unsloth_compiled_cache/*
+.vscode/*
 input/
 output/
 logs/

.gitignore

@@ -16,4 +16,5 @@ feedback/*
 test_code/*
 config/*
 tmp/*
-unsloth_compiled_cache/*
+unsloth_compiled_cache/*
+.vscode/*

app.py

@@ -314,8 +314,9 @@ with app:
             google_api_key_textbox = gr.Textbox(value = GEMINI_API_KEY, label="Enter Gemini API key (only if using Google API models)", lines=1, type="password")
 
         with gr.Accordion("Azure/OpenAI Inference", open = False):
-            azure_api_key_textbox = gr.Textbox(value = AZURE_OPENAI_API_KEY, label="Enter Azure/OpenAI Inference API key (only if using Azure/OpenAI models)", lines=1, type="password")
-            azure_endpoint_textbox = gr.Textbox(value = AZURE_OPENAI_INFERENCE_ENDPOINT, label="Enter Azure/OpenAI Inference endpoint URL (only if using Azure/OpenAI models)", lines=1)
+            with gr.Row():
+                azure_api_key_textbox = gr.Textbox(value = AZURE_OPENAI_API_KEY, label="Enter Azure/OpenAI Inference API key (only if using Azure/OpenAI models)", lines=1, type="password")
+                azure_endpoint_textbox = gr.Textbox(value = AZURE_OPENAI_INFERENCE_ENDPOINT, label="Enter Azure Inference endpoint URL (only if using Azure models)", lines=1)
 
         with gr.Accordion("Hugging Face token for downloading gated models", open = False):
             hf_api_key_textbox = gr.Textbox(value = HF_TOKEN, label="Enter Hugging Face API key (only for gated models that need a token to download)", lines=1, type="password")

tools/llm_funcs.py

@@ -685,13 +685,18 @@ def construct_azure_client(in_api_key: str, endpoint: str) -> Tuple[object, dict
         if not endpoint:
             endpoint = os.environ.get("AZURE_OPENAI_INFERENCE_ENDPOINT", "")
             if not endpoint:
-                raise Warning("No Azure/OpenAI inference endpoint found.")
+                # Assume using OpenAI API
+                client = OpenAI(
+                api_key=key,
+                )
+            else:
+                # Use the provided endpoint
+                client = OpenAI(
+                api_key=key,
+                base_url=f"{endpoint}",
+                )
 
-        # Use the provided endpoint instead of hardcoded value
-        client = OpenAI(
-        api_key=key,
-        base_url=f"{endpoint}",
-        )
+        
 
         return client, dict()
     except Exception as e: