phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix azure.py health check

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-22 17:33:39 -08:00
parent 49464df216
commit 04dec96f79
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/AzureOpenAI/azure.py
View file

@ -1528,7 +1528,8 @@ class AzureChatCompletion(BaseLLM):
prompt: Optional[str] = None, prompt: Optional[str] = None,
) -> dict: ) -> dict:
client_session = ( client_session = (
litellm.aclient_session or httpx.AsyncClient() litellm.aclient_session
or get_async_httpx_client(llm_provider=litellm.LlmProviders.AZURE).client
) # handle dall-e-2 calls ) # handle dall-e-2 calls
if "gateway.ai.cloudflare.com" in api_base: if "gateway.ai.cloudflare.com" in api_base: