rename to _get_azure_openai_client

This commit is contained in:
Ishaan Jaff 2025-03-18 09:25:26 -07:00
parent 01ed8e8a23
commit 9e00fa8221

View file

@ -141,7 +141,7 @@ class AzureChatCompletion(BaseAzureLLM, BaseLLM):
return headers return headers
def _get_sync_azure_client( def _get_azure_openai_client(
self, self,
api_version: Optional[str], api_version: Optional[str],
api_base: Optional[str], api_base: Optional[str],
@ -1240,7 +1240,7 @@ class AzureChatCompletion(BaseAzureLLM, BaseLLM):
litellm_params=litellm_params, litellm_params=litellm_params,
) # type: ignore ) # type: ignore
azure_client: AzureOpenAI = self._get_sync_azure_client( azure_client: AzureOpenAI = self._get_azure_openai_client(
api_base=api_base, api_base=api_base,
api_version=api_version, api_version=api_version,
api_key=api_key, api_key=api_key,
@ -1279,7 +1279,7 @@ class AzureChatCompletion(BaseAzureLLM, BaseLLM):
litellm_params: Optional[dict] = None, litellm_params: Optional[dict] = None,
) -> HttpxBinaryResponseContent: ) -> HttpxBinaryResponseContent:
azure_client: AsyncAzureOpenAI = self._get_sync_azure_client( azure_client: AsyncAzureOpenAI = self._get_azure_openai_client(
api_base=api_base, api_base=api_base,
api_version=api_version, api_version=api_version,
api_key=api_key, api_key=api_key,