diff --git a/litellm/proxy/health_endpoints/_health_endpoints.py b/litellm/proxy/health_endpoints/_health_endpoints.py index 35c9edb937d..9760e7e4a52 100644 --- a/litellm/proxy/health_endpoints/_health_endpoints.py +++ b/litellm/proxy/health_endpoints/_health_endpoints.py @@ -126,6 +126,7 @@ services = Union[ "datadog_llm_observability", "generic_api", "arize", + "s3", "sqs", ], str, @@ -200,6 +201,7 @@ async def health_services_endpoint( # noqa: PLR0915 "datadog_llm_observability", "generic_api", "arize", + "s3", "sqs", ]: raise HTTPException( diff --git a/tests/test_litellm/proxy/health_endpoints/test_health_services_callbacks.py b/tests/test_litellm/proxy/health_endpoints/test_health_services_callbacks.py new file mode 100644 index 00000000000..c82b02c5d65 --- /dev/null +++ b/tests/test_litellm/proxy/health_endpoints/test_health_services_callbacks.py @@ -0,0 +1,17 @@ +from unittest.mock import AsyncMock, patch + +import pytest + +from litellm.proxy.health_endpoints._health_endpoints import health_services_endpoint + + +@pytest.mark.asyncio +async def test_health_services_endpoint_accepts_s3_callback(): + with ( + patch("litellm.success_callback", ["s3"]), + patch("litellm.acompletion", new=AsyncMock(return_value={})), + ): + result = await health_services_endpoint(service="s3") + + assert result["status"] == "success" + assert "s3" in result["message"]