Litellm merge pr (#7161)

* build: merge branch

* test: fix openai naming

* fix(main.py): fix openai renaming

* style: ignore function length for config factory

* fix(sagemaker/): fix routing logic

* fix: fix imports

* fix: fix override

litellm/llms/azure/azure.py

@@ -35,38 +35,11 @@ from ...types.llms.openai import (
     RetrieveBatchRequest,
 )
 from ..base import BaseLLM
-from .common_utils import process_azure_headers
+from .common_utils import AzureOpenAIError, process_azure_headers
 
 azure_ad_cache = DualCache()
 
 
-class AzureOpenAIError(Exception):
-    def __init__(
-        self,
-        status_code,
-        message,
-        request: Optional[httpx.Request] = None,
-        response: Optional[httpx.Response] = None,
-        headers: Optional[httpx.Headers] = None,
-    ):
-        self.status_code = status_code
-        self.message = message
-        self.headers = headers
-        if request:
-            self.request = request
-        else:
-            self.request = httpx.Request(method="POST", url="https://api.openai.com/v1")
-        if response:
-            self.response = response
-        else:
-            self.response = httpx.Response(
-                status_code=status_code, request=self.request
-            )
-        super().__init__(
-            self.message
-        )  # Call the base class constructor with the parameters it needs
-
-
 class AzureOpenAIAssistantsAPIConfig:
     """
     Reference: https://learn.microsoft.com/en-us/azure/ai-services/openai/assistants-reference-messages?tabs=python#create-message
@@ -412,8 +385,12 @@ class AzureChatCompletion(BaseLLM):
 
                 data = {"model": None, "messages": messages, **optional_params}
             else:
-                data = litellm.AzureOpenAIConfig.transform_request(
-                    model=model, messages=messages, optional_params=optional_params
+                data = litellm.AzureOpenAIConfig().transform_request(
+                    model=model,
+                    messages=messages,
+                    optional_params=optional_params,
+                    litellm_params=litellm_params,
+                    headers=headers or {},
                 )
 
             if acompletion is True: