fix health check

2024-07-19 15:56:35 -07:00 · 2024-07-19 15:56:35 -07:00 · 43e5890f77
commit 43e5890f77
parent 1797021d53
1 changed files with 4 additions and 3 deletions
--- a/litellm/main.py
+++ b/litellm/main.py
@ -4798,9 +4798,10 @@ async def ahealth_check(
        if isinstance(stack_trace, str):
            stack_trace = stack_trace[:1000]
        if model not in litellm.model_cost and mode is None:
-            raise Exception(
-                "Missing `mode`. Set the `mode` for the model - https://docs.litellm.ai/docs/proxy/health#embedding-models"
-            )
+            return {
+                "error": "Missing `mode`. Set the `mode` for the model - https://docs.litellm.ai/docs/proxy/health#embedding-models"
+            }
+
        error_to_return = str(e) + " stack trace: " + stack_trace
        return {"error": error_to_return}