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

Fix lint

Browse source
This commit is contained in:
Cyrill Bannwart 2026-05-05 21:06:53 +00:00
parent 1d70f3621b
commit e9499ed4c6
1 changed files with 7 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/proxy/guardrails/guardrail_hooks/unified_guardrail/unified_guardrail.py
View file

@ -174,7 +174,8 @@ def _accumulate_tool_calls(chunks: List[Any]) -> List[dict]:
if idx is None:
continue
slot = by_index.setdefault(
idx, {"id": None, "type": None, "function": {"name": None, "arguments": ""}}
idx,
{"id": None, "type": None, "function": {"name": None, "arguments": ""}},
)
def _g(obj: Any, key: str) -> Any:
@ -855,9 +856,8 @@ class UnifiedLLMGuardrails(CustomLogger):
if action == GENERIC_GUARDRAIL_ACTION_BLOCKED:
error_message = (
(decision.blocked_reason if decision is not None else None)
or "Content violates policy"
)
decision.blocked_reason if decision is not None else None
) or "Content violates policy"
verbose_proxy_logger.warning(
"UnifiedLLMGuardrails action mode: BLOCKED mid-stream by %s: %s",
guardrail_name,
@ -910,9 +910,8 @@ class UnifiedLLMGuardrails(CustomLogger):
if action == GENERIC_GUARDRAIL_ACTION_BLOCKED:
error_message = (
(decision.blocked_reason if decision is not None else None)
or "Content violates policy"
)
decision.blocked_reason if decision is not None else None
) or "Content violates policy"
verbose_proxy_logger.warning(
"UnifiedLLMGuardrails action mode: BLOCKED at EOS by %s: %s",
guardrail_name,
@ -985,10 +984,7 @@ class UnifiedLLMGuardrails(CustomLogger):
if decision is None:
return accumulated_text, GENERIC_GUARDRAIL_ACTION_NONE
action = decision.action or GENERIC_GUARDRAIL_ACTION_NONE
if (
action == GENERIC_GUARDRAIL_ACTION_GUARDRAIL_INTERVENED
and decision.texts
):
if action == GENERIC_GUARDRAIL_ACTION_GUARDRAIL_INTERVENED and decision.texts:
return decision.texts[0], action
if action not in (
GENERIC_GUARDRAIL_ACTION_NONE,