scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-19 00:01:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 88

fix(guardrails): import ModelResponse lazily to avoid cyclic import alert

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
joshua-berri 2026-09-13 04:50:29 +00:00 • committed by yucheng
parent c65ace9164
commit c1d0d29d01
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/integrations/custom_guardrail.py
View file

@ -31,7 +31,6 @@ from litellm.types.utils import (
GuardrailStatus,
GuardrailTracingDetail,
LLMResponseTypes,
ModelResponse,
StandardLoggingGuardrailInformation,
)
@ -907,6 +906,8 @@ class CustomGuardrail(CustomLogger):
response: Final = (
kwargs.get("async_complete_streaming_response") or kwargs.get("complete_streaming_response") or result
)
from litellm.types.utils import ModelResponse
output_translation: Final = (
get_guardrail_translation_mapping(CallTypes.acompletion)()
if isinstance(response, ModelResponse)