From 5f78ea757cc014630eea2b938bf5ed222c92c5a1 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 22 Nov 2025 12:27:49 -0800 Subject: [PATCH] async_health_check arize fix --- litellm/integrations/arize/arize.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/litellm/integrations/arize/arize.py b/litellm/integrations/arize/arize.py index 6e0201bf5b7..a3563440ac9 100644 --- a/litellm/integrations/arize/arize.py +++ b/litellm/integrations/arize/arize.py @@ -117,10 +117,14 @@ class ArizeLogger(OpenTelemetry): try: config = self.get_arize_config() - if not config.space_id or not config.space_key: + # Prefer ARIZE_SPACE_KEY, but fall back to ARIZE_SPACE_ID for backwards compatibility + effective_space_key = config.space_key or config.space_id + + if not effective_space_key: return { "status": "unhealthy", - "error_message": "ARIZE_SPACE_ID environment variable not set", + # Tests (and users) expect the error message to reference ARIZE_SPACE_KEY + "error_message": "ARIZE_SPACE_KEY environment variable not set", } if not config.api_key: