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

test_model_connection

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-14 18:33:49 -07:00
parent 5aec90c513
commit 6787d0dabe
2 changed files with 31 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/main.py
View file

@ -5487,7 +5487,9 @@ async def ahealth_check(
model_params["litellm_logging_obj"] = litellm_logging_obj
mode_handlers = {
"chat": lambda: litellm.acompletion(**model_params),
"chat": lambda: litellm.acompletion(
**model_params,
),
"completion": lambda: litellm.atext_completion(
**_filter_model_params(model_params),
prompt=prompt or "test",
@ -5544,12 +5546,7 @@ async def ahealth_check(
"error": f"error:{str(e)}. Missing `mode`. Set the `mode` for the model - https://docs.litellm.ai/docs/proxy/health#embedding-models \nstacktrace: {stack_trace}"
}
error_to_return = (
str(e)
+ "\nHave you set 'mode' - https://docs.litellm.ai/docs/proxy/health#embedding-models"
+ "\nstack trace: "
+ stack_trace
)
error_to_return = str(e) + "\nstack trace: " + stack_trace
raw_request_typed_dict = litellm_logging_obj.model_call_details.get(
"raw_request_typed_dict"