Update app.py

app.py

@@ -24,25 +24,23 @@ def make_llm():
     # return ChatGoogleGenerativeAI(model="gemini-2.0-flash-exp")
     
     if provider == "gemini":
-        # Google Gemini via Generative AI API
-        model = os.getenv("GEMINI_MODEL", "gemini-2.0-flash")  # or 1.5-pro, etc.
         return ChatGoogleGenerativeAI(
-            model=model,
+            model=os.getenv("GEMINI_MODEL", "gemini-2.0-flash"),
+            temperature=0.3,
             max_retries=3,
             timeout=60,
-            temperature=0.3,
         )
 
     elif provider in {"azure", "azure-openai"}:
-        # Azure OpenAI  (ChatGPT family)
         return ChatOpenAI(
-            azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
+            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"),
-            azure_api_version=os.getenv("AZURE_OPENAI_API_VERSION", "2024-02-15-preview"),
-            api_key=os.getenv("AZURE_OPENAI_API_KEY"),        # picked up by ChatOpenAI
+            openai_api_key=os.getenv("AZURE_OPENAI_API_KEY"),
+            openai_api_type="azure",                                    # required flag
+            temperature=float(os.getenv("AZURE_OPENAI_TEMPERATURE", "0.3")),
             max_retries=3,
             timeout=60,
-            temperature=float(os.getenv("AZURE_OPENAI_TEMPERATURE", "0.3")),
         )
 
     else:
@@ -58,7 +56,7 @@ async def run_task(t: Task):
         agent = Agent(task=t.task, llm=llm)   # ← constructor has no max_steps
 
         # pick limits from env (or fall back to sane defaults)
-        max_steps = int(os.getenv("AGENT_MAX_STEPS", 10))
+        max_steps = int(os.getenv("AGENT_MAX_STEPS", 8))
         result = await agent.run_async(max_steps=max_steps)      # ← here!
         return result                        # Browser-Use returns dict
     except Exception as e: