From 3ff21bf30cde2cf8db3ea595b4b2b8865e53d9f2 Mon Sep 17 00:00:00 2001 From: eycjur Date: Tue, 23 Sep 2025 03:08:05 +0000 Subject: [PATCH] refactor --- litellm/llms/azure_ai/common_utils.py | 1 + litellm/llms/azure_ai/image_edit/transformation.py | 8 ++------ 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/litellm/llms/azure_ai/common_utils.py b/litellm/llms/azure_ai/common_utils.py index 64c424d23c..704c798074 100644 --- a/litellm/llms/azure_ai/common_utils.py +++ b/litellm/llms/azure_ai/common_utils.py @@ -29,6 +29,7 @@ class AzureFoundryModelInfo(BaseLLMModelInfo): api_version = ( api_version or litellm.api_version + or get_secret_str("AZURE_API_VERSION") or get_secret_str("AZURE_AI_API_VERSION") ) return api_version diff --git a/litellm/llms/azure_ai/image_edit/transformation.py b/litellm/llms/azure_ai/image_edit/transformation.py index e0be3f3d29..47f612912c 100644 --- a/litellm/llms/azure_ai/image_edit/transformation.py +++ b/litellm/llms/azure_ai/image_edit/transformation.py @@ -13,7 +13,7 @@ class AzureFoundryFluxImageEditConfig(OpenAIImageEditConfig): """ Azure AI Foundry FLUX image edit config - Supports FLUX models including FLUX-1.1-pro and FLUX-1-kontext-pro for image editing. + Supports FLUX models including FLUX-1-kontext-pro for image editing. Azure AI Foundry FLUX models handle image editing through the /images/edits endpoint, same as standard Azure OpenAI models. The request format uses multipart/form-data @@ -28,11 +28,7 @@ class AzureFoundryFluxImageEditConfig(OpenAIImageEditConfig): ) -> dict: """ Validate Azure AI Foundry environment and set up authentication - Uses Api-Key header format like Azure OpenAI - - Azure AI Foundry uses the same authentication format as Azure OpenAI: - - Header: Api-Key (not Authorization Bearer) - - Endpoint: /openai/deployments/{model}/images/edits (for image editing) + Uses Api-Key header format """ api_key = AzureFoundryModelInfo.get_api_key(api_key)