From cca1ea1592e440f9585dd2a50a36da4e5067bf85 Mon Sep 17 00:00:00 2001 From: Yuta Saito Date: Sun, 24 Aug 2025 08:37:17 +0900 Subject: [PATCH] Fix missing and unused imports in custom_guardrail docs example --- docs/my-website/docs/proxy/guardrails/custom_guardrail.md | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/docs/my-website/docs/proxy/guardrails/custom_guardrail.md b/docs/my-website/docs/proxy/guardrails/custom_guardrail.md index 657ccab68e..b8ba64d333 100644 --- a/docs/my-website/docs/proxy/guardrails/custom_guardrail.md +++ b/docs/my-website/docs/proxy/guardrails/custom_guardrail.md @@ -23,15 +23,14 @@ A CustomGuardrail has 4 methods to enforce guardrails Create a new file called `custom_guardrail.py` and add this code to it ```python -from typing import Any, Dict, List, Literal, Optional, Union +from typing import Any, AsyncGenerator, Literal, Optional, Union import litellm from litellm._logging import verbose_proxy_logger from litellm.caching.caching import DualCache from litellm.integrations.custom_guardrail import CustomGuardrail from litellm.proxy._types import UserAPIKeyAuth -from litellm.proxy.guardrails.guardrail_helpers import should_proceed_based_on_metadata -from litellm.types.guardrails import GuardrailEventHooks +from litellm.types.utils import ModelResponseStream class myCustomGuardrail(CustomGuardrail):