diff --git a/litellm/proxy/health_check.py b/litellm/proxy/health_check.py
index 62721799662..f4cffcd1875 100644
--- a/litellm/proxy/health_check.py
+++ b/litellm/proxy/health_check.py
@@ -368,11 +368,15 @@ def _update_litellm_params_for_health_check(
     """
     _NON_CHAT_MODES = {
         "image_generation",
+        "image_edit",
         "video_generation",
         "embedding",
         "rerank",
-        "transcription",
+        "audio_transcription",
         "audio_speech",
+        "ocr",
+        "search",
+        "moderation",
     }
     _mode = model_info.get("mode", None)
     litellm_params["messages"] = _get_random_llm_message()
diff --git a/tests/test_litellm/proxy/test_health_check_max_tokens.py b/tests/test_litellm/proxy/test_health_check_max_tokens.py
index 2d2b4eac9fe..ad1264c7ca8 100644
--- a/tests/test_litellm/proxy/test_health_check_max_tokens.py
+++ b/tests/test_litellm/proxy/test_health_check_max_tokens.py
@@ -90,11 +90,15 @@ async def test_ahealth_check_wildcard_models_respects_max_tokens():
     "mode",
     [
         "image_generation",
+        "image_edit",
         "video_generation",
         "embedding",
         "rerank",
-        "transcription",
+        "audio_transcription",
         "audio_speech",
+        "ocr",
+        "search",
+        "moderation",
     ],
 )
 @pytest.mark.asyncio