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

fix(azure.py): support health checks to text completion endpoints

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-12 00:13:01 +05:30
parent 0a76269541
commit a7f182b8ec
3 changed files with 20 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/main.py
View file

@ -3123,7 +3123,10 @@ async def ahealth_check(
prompt=prompt,
input=input,
)
elif custom_llm_provider == "openai":
elif (
custom_llm_provider == "openai"
or custom_llm_provider == "text-completion-openai"
):
api_key = model_params.get("api_key") or get_secret("OPENAI_API_KEY")
timeout = (