phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix linting error

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-18 11:38:31 -07:00
parent e3752cd32a
commit 40736c3263
2 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/llms/azure/common_utils.py
View file

@ -1,6 +1,6 @@
import json
import os
from typing import Callable, Optional, Union
from typing import Any, Callable, Dict, Optional, Union
import httpx
from openai import AsyncAzureOpenAI, AzureOpenAI
@ -385,7 +385,7 @@ class BaseAzureLLM(BaseOpenAILLM):
api_base += "/"
api_base += f"{model}"
azure_client_params = {
azure_client_params: Dict[str, Any] = {
"api_version": api_version,
"base_url": f"{api_base}",
"http_client": litellm.client_session,
@ -399,11 +399,11 @@ class BaseAzureLLM(BaseOpenAILLM):
azure_ad_token = get_azure_ad_token_from_oidc(azure_ad_token)
azure_client_params["azure_ad_token"] = azure_ad_token
elif azure_ad_token_provider is not None:
if azure_ad_token_provider is not None:
azure_client_params["azure_ad_token_provider"] = azure_ad_token_provider
if acompletion is True:
client = AsyncAzureOpenAI(**azure_client_params)
client = AsyncAzureOpenAI(**azure_client_params) # type: ignore
else:
client = AzureOpenAI(**azure_client_params)
client = AzureOpenAI(**azure_client_params) # type: ignore
return client