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

fix(mypy): add type annotation for words_in_match in airline competitor check

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-21 10:00:33 -08:00
parent 4f9bcf9a73
commit 08fedb8fd0
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/competitor_intent/airline.py
View file

@ -14,8 +14,11 @@ from pathlib import Path
from typing import Any, Dict, List, Tuple
from litellm.proxy.guardrails.guardrail_hooks.litellm_content_filter.competitor_intent.base import (
BaseCompetitorIntentChecker, _compile_marker, _count_signals,
_word_boundary_match)
BaseCompetitorIntentChecker,
_compile_marker,
_count_signals,
_word_boundary_match,
)
# Location/travel context: prepositions, travel verbs, booking nouns, entry/geo nouns.
# No place-name list; these patterns detect "destination context" generically.
@ -132,7 +135,7 @@ def _load_competitors_excluding_brand(brand_self: List[str]) -> List[str]:
continue
match_str = entry.get("match") or ""
variants = [v.strip().lower() for v in match_str.split("|") if v.strip()]
words_in_match = set()
words_in_match: set[str] = set()
for v in variants:
words_in_match.update(v.split())
if brand_set & words_in_match or any(v in brand_set for v in variants):