diff --git a/docs/modules/models/text_embedding/examples/azureopenai.ipynb b/docs/modules/models/text_embedding/examples/azureopenai.ipynb index 435f2e06766ff4..0f09f1e969def9 100644 --- a/docs/modules/models/text_embedding/examples/azureopenai.ipynb +++ b/docs/modules/models/text_embedding/examples/azureopenai.ipynb @@ -22,7 +22,8 @@ "\n", "os.environ[\"OPENAI_API_TYPE\"] = \"azure\"\n", "os.environ[\"OPENAI_API_BASE\"] = \"https:// Dict: "OPENAI_API_TYPE", default="", ) - openai_api_version = get_from_dict_or_env( - values, - "openai_api_version", - "OPENAI_API_VERSION", - ) + if openai_api_type in ("azure", "azure_ad", "azuread"): + openai_api_version = get_from_dict_or_env( + values, + "openai_api_version", + "OPENAI_API_VERSION", + ) + else: + openai_api_version = None openai_organization = get_from_dict_or_env( values, "openai_organization", @@ -163,6 +166,7 @@ def validate_environment(cls, values: Dict) -> Dict: openai.organization = openai_organization if openai_api_base: openai.api_base = openai_api_base + if openai_api_version: openai.api_version = openai_api_version if openai_api_type: openai.api_type = openai_api_type