fix: _extract_entities crashes on a non-string query (#1724)

This commit is contained in:
Afonso Coutinho
2026-06-03 05:30:28 +01:00
committed by GitHub
parent 0c37943267
commit c5bc39de88
2 changed files with 17 additions and 0 deletions

View File

@@ -27,6 +27,8 @@ def _detect_question_type(query: str) -> Optional[str]:
def _extract_entities(query: str) -> Dict[str, List[str]]:
"""Lightweight entity extraction: capitalized words and date patterns."""
if not isinstance(query, str):
return {"names": [], "dates": []}
entities: Dict[str, List[str]] = {"names": [], "dates": []}
qtype = _detect_question_type(query)
cleaned = query