diff --git a/libs/partners/openai/langchain_openai/embeddings/azure.py b/libs/partners/openai/langchain_openai/embeddings/azure.py index be8ddcc8b9e..70e50b2dd08 100644 --- a/libs/partners/openai/langchain_openai/embeddings/azure.py +++ b/libs/partners/openai/langchain_openai/embeddings/azure.py @@ -173,13 +173,15 @@ class AzureOpenAIEmbeddings(OpenAIEmbeddings): # type: ignore[override] # between azure_endpoint and base_url (openai_api_base). openai_api_base = self.openai_api_base if openai_api_base and self.validate_base_url: - if "/openai" not in openai_api_base: - self.openai_api_base = cast(str, self.openai_api_base) + "/openai" - raise ValueError( - "As of openai>=1.0.0, Azure endpoints should be specified via " - "the `azure_endpoint` param not `openai_api_base` " - "(or alias `base_url`). " - ) + # Only validate openai_api_base if azure_endpoint is not provided + if not self.azure_endpoint: + if "/openai" not in openai_api_base: + self.openai_api_base = cast(str, self.openai_api_base) + "/openai" + raise ValueError( + "As of openai>=1.0.0, Azure endpoints should be specified via " + "the `azure_endpoint` param not `openai_api_base` " + "(or alias `base_url`). " + ) if self.deployment: raise ValueError( "As of openai>=1.0.0, if `deployment` (or alias "