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

(fix) print_verbose health check

Browse source
This commit is contained in:
ishaan-jaff 2023-12-06 10:58:00 -08:00
parent f6546076b0
commit 4f02b3c161
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/health_check.py
View file

@ -50,7 +50,7 @@ async def _perform_health_check(model_list: list):
try: try:
await litellm.aembedding(**model_params) await litellm.aembedding(**model_params)
except Exception as e: except Exception as e:
print_verbose("\n\n Got Exception, ", e) print_verbose(f"\n\n Got Exception, {e}")
print_verbose(f"Health check failed for model {model_params['model']}. Error: {e}") print_verbose(f"Health check failed for model {model_params['model']}. Error: {e}")
return False return False
return True return True
@ -60,7 +60,7 @@ async def _perform_health_check(model_list: list):
try: try:
await litellm.acompletion(**model_params) await litellm.acompletion(**model_params)
except Exception as e: except Exception as e:
print_verbose("\n\n Got Exception, ", e) print_verbose(f"\n\n Got Exception, {e}")
error_str = (str(e)) error_str = (str(e))
if "This is not a chat model" in error_str or "The chatCompletion operation does not work with the specified model" in error_str: if "This is not a chat model" in error_str or "The chatCompletion operation does not work with the specified model" in error_str:
return await _check_embedding_model(model_params) return await _check_embedding_model(model_params)