refactor(guardrails): tighten Wingback initialize_guardrail typing

This commit is contained in:
bsk-wb 2026-10-09 22:57:20 +05:30
parent c16af16c73
commit 35359bdfc2

View file

@ -1,3 +1,5 @@
from __future__ import annotations
from typing import TYPE_CHECKING, Final
from litellm.types.guardrails import SupportedGuardrailIntegrations
@ -8,7 +10,10 @@ if TYPE_CHECKING:
from litellm.types.guardrails import Guardrail, LitellmParams
def initialize_guardrail(litellm_params: "LitellmParams", guardrail: "Guardrail") -> WingbackGuardrail:
def initialize_guardrail(
litellm_params: LitellmParams,
guardrail: Guardrail,
) -> WingbackGuardrail:
import litellm
_wingback_callback: Final = WingbackGuardrail(