feat(azure.py): support health checking azure deployments

Fixes https://github.com/BerriAI/litellm/issues/5279
This commit is contained in:
Krrish Dholakia 2024-08-22 16:11:14 -07:00
parent 63cd94c32a
commit d7d3eee349
4 changed files with 42 additions and 3 deletions

View file

@ -4825,7 +4825,7 @@ def speech(
async def ahealth_check(
model_params: dict,
mode: Optional[
Literal["completion", "embedding", "image_generation", "chat"]
Literal["completion", "embedding", "image_generation", "chat", "batch"]
] = None,
prompt: Optional[str] = None,
input: Optional[List] = None,