scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix(streaming): use word boundaries in _cut_breaks_wider_context for conditional words

Browse source
This commit is contained in:
Karunasagar Mohansundar 2026-09-28 04:39:31 +00:00
parent 553db8242c
commit 40bf1eded8
1 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/content_filter.py
View file

@ -2058,7 +2058,23 @@ class ContentFilterGuardrail(CustomGuardrail):
cut_sentence: Final = (
SENTENCE_TERMINATORS.split(head.lower())[-1] + SENTENCE_TERMINATORS.split(tail_lower, maxsplit=1)[0]
)
return any(word in cut_sentence for word in plan.conditional_words)
# Use word boundary matching for single-word conditional words to match _check_conditional_categories behavior
for word in plan.conditional_words:
if " " in word:
# Multi-word phrase - use substring matching
if word in cut_sentence:
return True
else:
# Single word - use word boundary matching
if _is_word_char_pattern(word):
pattern = r"\b" + re.escape(word) + r"\b"
if re.search(pattern, cut_sentence):
return True
else:
# Punctuation-only words use substring matching
if word in cut_sentence:
return True
return False
def _trim_streamed_choice_buffer(
self, state: _StreamedChoiceState, masked_text: str, plan: _StreamedScanPlan