Update app.py

app.py

@@ -20,9 +20,7 @@ def make_llm():
     """
     provider = os.getenv("LLM_PROVIDER", "gemini").lower()
     
-    # LangChain helper; will raise if GOOGLE_API_KEY missing
-    # return ChatGoogleGenerativeAI(model="gemini-2.0-flash-exp")
-    
+    # --- Google Gemini stays exactly as before ------------------- #
     if provider == "gemini":
         return ChatGoogleGenerativeAI(
             model=os.getenv("GEMINI_MODEL", "gemini-2.0-flash"),
@@ -31,23 +29,21 @@ def make_llm():
             timeout=60,
         )
 
+    # --- Azure OpenAI ------------------------------------------- #
     elif provider in {"azure", "azure-openai"}:
-        return ChatOpenAI(
-            openai_api_base=os.getenv("AZURE_OPENAI_ENDPOINT"),         # ← NEW NAME
-            openai_api_version=os.getenv("AZURE_OPENAI_API_VERSION", "2024-02-15-preview"),
-            azure_deployment_name=os.getenv("AZURE_OPENAI_DEPLOYMENT", "gpt-35-turbo"),
-            openai_api_key=os.getenv("AZURE_OPENAI_API_KEY"),
-            openai_api_type="azure",                                    # required flag
+        return AzureChatOpenAI(                         # ✅ correct wrapper
+            azure_deployment=os.getenv(                 # ✅ correct kwarg name
+                "AZURE_OPENAI_DEPLOYMENT", "gpt-35-turbo"
+            ),
+            api_version=os.getenv(                      # ✅ correct kwarg name
+                "AZURE_OPENAI_API_VERSION", "2024-05-01-preview"
+            ),
             temperature=float(os.getenv("AZURE_OPENAI_TEMPERATURE", "0.3")),
             max_retries=3,
             timeout=60,
         )
 
-    else:
-        raise ValueError(
-            f"Unsupported LLM_PROVIDER '{provider}'. "
-            "Use 'gemini' or 'azure-openai'."
-        )
+    raise ValueError(f"Unsupported LLM_PROVIDER '{provider}'")
 
 @app.post("/run")
 async def run_task(t: Task):