fix(guardrails): log Conduct apply_guardrail decisions via log_guardrail_information

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-11 21:17:16 +00:00
parent 747dda483e
commit 0f729bd767

View file

@ -10,7 +10,7 @@ from collections.abc import Mapping
from types import MappingProxyType
from typing import TYPE_CHECKING, Final, Literal
from litellm.integrations.custom_guardrail import CustomGuardrail
from litellm.integrations.custom_guardrail import CustomGuardrail, log_guardrail_information
from litellm.types.llms.openai import ChatCompletionUserMessage
if TYPE_CHECKING:
@ -51,6 +51,7 @@ except ImportError as import_error:
else:
class ConductGuardrail(ConductGuard): # pyright: ignore[reportUntypedBaseClass] # optional dep, absent at type-check
@log_guardrail_information
async def apply_guardrail(
self,
inputs: GenericGuardrailAPIInputs,