From 62d3cab965b01dd0fd0c82d3f704cce1a0d57f53 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 14 Feb 2026 15:54:20 -0800 Subject: [PATCH] test fixes --- litellm/proxy/guardrails/guardrail_hooks/presidio.py | 3 +++ litellm/proxy/management_endpoints/common_utils.py | 4 ++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/presidio.py b/litellm/proxy/guardrails/guardrail_hooks/presidio.py index b3cc9236b1d..29d57ee4734 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/presidio.py +++ b/litellm/proxy/guardrails/guardrail_hooks/presidio.py @@ -385,6 +385,9 @@ class _OPTIONAL_PresidioPIIMasking(CustomGuardrail): ) continue return final_results + except GuardrailRaisedException: + # Re-raise GuardrailRaisedException without wrapping + raise except Exception as e: # Sanitize exception to avoid leaking the original text (which may # contain API keys or other secrets) in error responses. diff --git a/litellm/proxy/management_endpoints/common_utils.py b/litellm/proxy/management_endpoints/common_utils.py index d6f6d1337a3..34a17f6c274 100644 --- a/litellm/proxy/management_endpoints/common_utils.py +++ b/litellm/proxy/management_endpoints/common_utils.py @@ -1,4 +1,4 @@ -from typing import Any, Dict, Optional, Union, TYPE_CHECKING +from typing import TYPE_CHECKING, Any, Dict, Optional, Union from litellm._logging import verbose_proxy_logger from litellm.caching import DualCache @@ -402,7 +402,7 @@ def _update_metadata_fields(updated_kv: dict) -> None: updated_kv: The key-value dict being used for the update """ for field in LiteLLM_ManagementEndpoint_MetadataFields_Premium: - if field in updated_kv and _has_non_empty_value(updated_kv[field]): + if field in updated_kv and updated_kv[field] is not None: _update_metadata_field(updated_kv=updated_kv, field_name=field) for field in LiteLLM_ManagementEndpoint_MetadataFields: