From 8a43fed20cabb1c038cf2b5389efd89dfa0b7a24 Mon Sep 17 00:00:00 2001 From: yassin Date: Wed, 16 Sep 2026 16:48:15 +0000 Subject: [PATCH] fix(guardrails): use explicit returns in _configured_event_hooks Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../guardrails/guardrail_initializers.py | 26 +++++++------------ 1 file changed, 10 insertions(+), 16 deletions(-) diff --git a/litellm/proxy/guardrails/guardrail_initializers.py b/litellm/proxy/guardrails/guardrail_initializers.py index 98bc6b5451d..b7ab215a2cd 100644 --- a/litellm/proxy/guardrails/guardrail_initializers.py +++ b/litellm/proxy/guardrails/guardrail_initializers.py @@ -1,8 +1,6 @@ # litellm/proxy/guardrails/guardrail_initializers.py from typing import Any, Final -from typing_extensions import assert_never - import litellm from litellm.integrations.custom_guardrail import CustomGuardrail from litellm.proxy._types import CommonProxyErrors @@ -99,20 +97,16 @@ _MCP_EVENT_HOOKS: Final = frozenset( def _configured_event_hooks(mode: str | list[str] | Mode) -> tuple[str, ...]: - match mode: - case str(): - return (mode,) - case list(): - return tuple(mode) - case Mode(): - return tuple( - hook - for value in (*mode.tags.values(), mode.default) - if value is not None - for hook in ((value,) if isinstance(value, str) else value) - ) - case _: - assert_never(mode) + if isinstance(mode, str): + return (mode,) + if isinstance(mode, list): + return tuple(mode) + return tuple( + hook + for value in (*mode.tags.values(), mode.default) + if value is not None + for hook in ((value,) if isinstance(value, str) else value) + ) def _is_mcp_only_mode(mode: str | list[str] | Mode) -> bool: