mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-14 04:09:09 +00:00
refactor(guardrails): tighten Wingback initialize_guardrail typing
This commit is contained in:
parent
c16af16c73
commit
35359bdfc2
1 changed files with 6 additions and 1 deletions
|
|
@ -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(
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue