From 35359bdfc28444e88023cdfddd0403b0156297e9 Mon Sep 17 00:00:00 2001
From: bsk-wb <bharath@wingback.ai>
Date: Fri, 9 Oct 2026 22:57:20 +0530
Subject: [PATCH] refactor(guardrails): tighten Wingback initialize_guardrail
 typing

---
 .../proxy/guardrails/guardrail_hooks/wingback/__init__.py  | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/litellm/proxy/guardrails/guardrail_hooks/wingback/__init__.py b/litellm/proxy/guardrails/guardrail_hooks/wingback/__init__.py
index fc16cad9d16..9e172cea68a 100644
--- a/litellm/proxy/guardrails/guardrail_hooks/wingback/__init__.py
+++ b/litellm/proxy/guardrails/guardrail_hooks/wingback/__init__.py
@@ -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(