From fa51d5a4d7186766d01f62e7b8e8676573906a2e Mon Sep 17 00:00:00 2001
From: Martim Santos <72747170+martimfasantos@users.noreply.github.com>
Date: Thu, 6 Mar 2025 19:30:04 +0000
Subject: [PATCH] AzureOpenAI: api_base and azure_endpoint are mutually
 exclusive (#18037)

---
 .../llama_index/llms/azure_openai/base.py                  | 7 ++++---
 .../llms/llama-index-llms-azure-openai/pyproject.toml      | 2 +-
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/llama-index-integrations/llms/llama-index-llms-azure-openai/llama_index/llms/azure_openai/base.py b/llama-index-integrations/llms/llama-index-llms-azure-openai/llama_index/llms/azure_openai/base.py
index 332a13ebad091..16b4a54cf4657 100644
--- a/llama-index-integrations/llms/llama-index-llms-azure-openai/llama_index/llms/azure_openai/base.py
+++ b/llama-index-integrations/llms/llama-index-llms-azure-openai/llama_index/llms/azure_openai/base.py
@@ -146,9 +146,10 @@ def __init__(
         if engine is None:
             raise ValueError("You must specify an `engine` parameter.")
 
-        azure_endpoint = get_from_param_or_env(
-            "azure_endpoint", azure_endpoint, "AZURE_OPENAI_ENDPOINT", ""
-        )
+        if api_base is None:
+            azure_endpoint = get_from_param_or_env(
+                "azure_endpoint", azure_endpoint, "AZURE_OPENAI_ENDPOINT", ""
+            )
 
         super().__init__(
             engine=engine,
diff --git a/llama-index-integrations/llms/llama-index-llms-azure-openai/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-azure-openai/pyproject.toml
index 931c554be79b1..8c4443425a26c 100644
--- a/llama-index-integrations/llms/llama-index-llms-azure-openai/pyproject.toml
+++ b/llama-index-integrations/llms/llama-index-llms-azure-openai/pyproject.toml
@@ -29,7 +29,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-llms-azure-openai"
 readme = "README.md"
-version = "0.3.1"
+version = "0.3.2"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"