Luna audit round 2: review fixes (regex bounds, mixed albums, redaction guards)

lumen_limits.py

@@ -34,10 +34,16 @@ def _check_and_register_rate_limit(user_id: int | None) -> bool:
     if user_id not in user_rate_limits and len(user_rate_limits) >= MAX_RATE_LIMIT_KEYS:
         _cleanup_rate_limit_dict()
         # Чистка сносит только протухших: при флуде свежими ID словарь рос бы без потолка.
-        # Добиваем жёстко — самых давно молчавших (найдено внешним аудитом).
-        while len(user_rate_limits) >= MAX_RATE_LIMIT_KEYS:
-            oldest = min(user_rate_limits, key=lambda uid: user_rate_limits[uid][-1] if user_rate_limits[uid] else 0)
-            user_rate_limits.pop(oldest, None)
+        # Добиваем жёстко — самых давно молчавших (найдено внешним аудитом). Сносим с запасом
+        # (до 90% потолка), чтобы не сканировать весь словарь на каждый новый ID (ревью ветки).
+        if len(user_rate_limits) >= MAX_RATE_LIMIT_KEYS:
+            ordered = sorted(
+                user_rate_limits,
+                key=lambda uid: user_rate_limits[uid][-1] if user_rate_limits[uid] else 0,
+            )
+            drop = len(ordered) - int(MAX_RATE_LIMIT_KEYS * 0.9) + 1
+            for uid in ordered[:drop]:
+                user_rate_limits.pop(uid, None)
     timestamps = user_rate_limits.setdefault(user_id, [])
     while timestamps and now - timestamps[0] > RATE_LIMIT_WINDOW_SEC:
         timestamps.pop(0)

lumen_message_core.py

@@ -305,6 +305,13 @@ async def _handle_message_core(message: Message, extra_media: list[tuple[bytes,
          pass
 
     media_mime = media_tuple[1] if media_tuple else None
+    # Смешанный альбом (фото + видео-слайды): достаточно одного видео — весь набор едет
+    # в Gemini-ветку, иначе видео-слайды молча терялись (ревью ветки: смотрели только media[0]).
+    if media_mime and media_mime.startswith("image/") and extra_media:
+        for _, extra_mime in extra_media:
+            if not (extra_mime or "").startswith("image/"):
+                media_mime = extra_mime
+                break
     is_heavy = _looks_like_heavy_query(clean_prompt)
     needs_freshness = _looks_like_freshness_query(clean_prompt)
     route = _build_route(

lumen_message_parse.py

@@ -50,7 +50,7 @@ DRAW_TRIGGER_PREFIXES = [
     "сгенери картинку", "сгенери изображение", "можешь нарисовать", "можешь нарисовать мне",
     # EN: /start обещает "draw a cat" — чиним (найдено внешним аудитом: был только русский).
     "generate an image", "generate me an image", "create an image", "draw a picture",
-    "draw me", "draw an", "draw a", "draw some", "paint me", "paint", "sketch",
+    "draw me", "draw an", "draw a", "paint me", "paint", "sketch",
 ]
 # "хочу картинку" не триггер: путается с показом/редактурой. Длинные фразы выше коротких, иначе вернётся первое совпадение.
 TTS_TRIGGER_PREFIXES = [

lumen_router_config.py

@@ -403,11 +403,12 @@ _HEAVY_QUERY_RE = re.compile(
     r"|многошагов\w*|пошагов\w*\s+(инструкц\w*|план\w*)"
     r"|архитектур\w*|алгоритм\w*"
     # EN-набор (найдено внешним аудитом: heavy-детект был почти весь русский).
-    r"|write\s+(a\s+|an\s+|the\s+)?(\w+\s+)?(code|function|script|program|class|sql|regex|test|parser|bot|website|app)\b"
+    # Границы слов обязательны: голый prove ловил improve, solve — resolve (ревью ветки).
+    r"|write\s+(a\s+|an\s+|the\s+)?(\w+\s+)?(code|function\w*|script\w*|program\w*|class\w*|sql|regex|test\w*|parser\w*|bot|website\w*|app\w*)\b"
     r"|generate\s+code|fix\s+(this\s+|that\s+)?(code|bug|error|issue)\b|refactor\w*|optimiz\w*|debug"
     r"|explain\s+(code|error)|code\s+review|algorithm|architecture"
     r"|write\s+(an?\s+)?(essay|article|report|paper|thesis|cv|resume)\b"
-    r"|compar(e|ison)|prove|proof|solve|equation",
+    r"|compar(e|ison)|\bprove\b|\bproof\b|\bsolve\b|equation",
     re.IGNORECASE,
 )
 
@@ -441,9 +442,10 @@ _FRESHNESS_QUERY_RE = re.compile(
     r"|в\s+эт(ом|ой)\s+(году|месяце|неделе)"
     r"|\b202[6-9]\b"
     # EN-набор (найдено внешним аудитом: детект был только русским при DEFAULT_LANG=en
-    # и 25 языках; ложные срабатывания так же дёшевы).
-    r"|now|today|current\w*|latest|recent\w*|breaking"
-    r"|news|weather|forecast|price\w*|cost\w*|how\s+much|exchange|score\w*|schedule"
+    # и 25 языках; ложные срабатывания так же дёшевы). Границы слов обязательны:
+    # голый now ловил know/snow (ревью ветки).
+    r"|\bnow\b|\btoday\b|current\w*|latest|recent\w*|\bbreaking\b"
+    r"|\bnews\b|weather|forecast|price\w*|\bcost\w*|how\s+much|exchange|\bscore\w*|schedule"
     r"|who\s+is\s+(now|currently|the\s+(president|ceo|prime\s+minister|mayor))"
     r"|best\s+(to\s+buy|buy)|should\s+i\s+buy|worth\s+buying",
     re.IGNORECASE,

lumen_routes.py

@@ -84,7 +84,7 @@ async def _groq_request(path: str, method: str = "GET", *, json_body: dict | Non
         raise
     except Exception as exc:
         exc_str = str(exc) or repr(exc) or exc.__class__.__name__
-        if bot.GROQ_API_KEY:
+        if bot.GROQ_API_KEY and len(bot.GROQ_API_KEY) > 8:
             exc_str = exc_str.replace(bot.GROQ_API_KEY, "<KEY>")
         raise bot.GroqAPIError(f"Groq network error: {exc_str}") from exc
 
@@ -118,7 +118,7 @@ async def _or_request(path: str, method: str = "GET", *, json_body: dict | None
         # str(exc) у таймаутов часто пуст — берём repr/имя класса, иначе в логах пустая строка.
         exc_str = str(exc) or repr(exc) or exc.__class__.__name__
         # Вычищаем OPENROUTER_API_KEY из текста ошибок (defense-in-depth: обычно ключ только в заголовке, но прокси может процитировать заголовки).
-        if bot.OPENROUTER_API_KEY:
+        if bot.OPENROUTER_API_KEY and len(bot.OPENROUTER_API_KEY) > 8:
             exc_str = exc_str.replace(bot.OPENROUTER_API_KEY, "<KEY>")
         raise bot.OpenRouterAPIError(f"OpenRouter network error: {exc_str}") from exc
 
@@ -346,7 +346,7 @@ async def ask_openrouter_multimodal(
     for raw_bytes, mime in media_items[:10]:
         # Видео-слайды альбомов сюда не ходят (OpenRouter принимает только картинки) —
         # их забирает Gemini-ветка; раньше они просто молча отваливались вместе со 2-9 фото.
-        if not mime.startswith("image/"):
+        if not (mime or "").startswith("image/"):
             continue
         b64 = base64.b64encode(raw_bytes).decode("utf-8")
         parts.append({"type": "image_url", "image_url": {"url": f"data:{mime};base64,{b64}"}})
@@ -776,7 +776,7 @@ async def _run_route(
             tried_stream_model, tried_stream_provider = head_model, "groq"
             reusable_placeholder = placeholder
 
-    is_video_or_audio = bool(media) and not media[0][1].startswith("image/")
+    is_video_or_audio = bool(media) and any(not (m[1] or "").startswith("image/") for m in media)
     last_exc: Exception | None = None
     for provider in provider_order:
         ids = list(groups.get(provider) or [])

lumen_tts.py

@@ -86,7 +86,7 @@ async def _fish_audio_tts_bytes(
     except Exception as exc:
         # Ключ из текста ошибок вычищаем (тот же defense-in-depth, что у _or_request/_groq_request).
         exc_str = str(exc) or repr(exc) or exc.__class__.__name__
-        if api_key:
+        if api_key and len(api_key) > 8:
             exc_str = exc_str.replace(api_key, "<KEY>")
         log.warning("[tts] Fish Audio request failed, falling back to Gemini TTS: %s", exc_str)
         return None

tests/test_bot_commands.py

@@ -219,6 +219,11 @@ def test_match_trigger_prefix_english_triggers():
     assert bot._match_trigger_prefix("how are you", bot.TTS_TRIGGER_PREFIXES) is None
 
 
+def test_match_trigger_prefix_rejects_draw_idiom():
+    # Ревью ветки: "draw some" ловил идиому "draw conclusions".
+    assert bot._match_trigger_prefix("draw some conclusions here", bot.DRAW_TRIGGER_PREFIXES) is None
+
+
 def test_tts_trigger_this_with_reply_voices_replied_message(rate_guard_setup):
     # Регрессия (сентябрь 2026): "озвучь это" в ответ на сообщение озвучивало
     # само слово "это" — остаток после триггера считался содержанием.

tests/test_bot_routes.py

@@ -832,6 +832,21 @@ def test_ask_openrouter_multimodal_skips_video_slides_for_gemini():
         bot.chat_state.pop(chat_id, None)
 
 
+def test_ask_openrouter_multimodal_rejects_video_only_album():
+    # Только видео без картинок — нечего слать в OpenRouter: ошибка уводит маршрут в Gemini.
+    chat_id = 999413
+    original_key = bot.OPENROUTER_API_KEY
+    bot.OPENROUTER_API_KEY = "fake-key"
+    try:
+        with pytest.raises(RuntimeError):
+            asyncio.run(bot.ask_openrouter_multimodal(
+                chat_id, "что это?", [(b"vid", "video/mp4")], "a.mp4", model_chain=["m"],
+            ))
+    finally:
+        bot.OPENROUTER_API_KEY = original_key
+        bot.chat_state.pop(chat_id, None)
+
+
 def test_route_error_reply_text_youtube_takes_priority_over_exception_type():
     exc = bot.OpenRouterAPIError("boom", status_code=500)
     text = bot._route_error_reply_text(exc, "gemini-3.6-flash", youtube_url_to_analyze="https://youtu.be/x")

tests/test_bot_state.py

@@ -737,6 +737,31 @@ def test_voice_message_falls_back_to_gemini_audio_on_transcribe_failure(rate_gua
     bot.chat_state.pop(123, None)
 
 
+def test_mixed_album_with_video_slide_routes_to_gemini(rate_guard_setup, monkeypatch):
+    # Ревью ветки: смотрели только первое вложение — фото+видео уходило image-маршрутом и видео терялось.
+    message = rate_guard_setup()
+    message.text = ""
+    captured = {}
+
+    async def fake_resolve(message, state, clean_prompt, *, is_private):
+        return None, "", "photo.jpg", (b"pic", "image/jpeg")
+
+    async def fake_run_route(chat_id, ai_prompt, route, message, **kwargs):
+        captured["route"] = route
+        captured["media"] = kwargs.get("media")
+        return "ok", False
+
+    monkeypatch.setattr(bot, "_resolve_incoming_media", fake_resolve)
+    monkeypatch.setattr(bot, "_run_route", fake_run_route)
+    monkeypatch.setattr(bot, "_safe_reply", AsyncMock())
+    asyncio.run(bot._handle_message_core(
+        message, extra_media=[(b"vid", "video/mp4")],
+    ))
+    assert captured["route"][0][0] == "gemini"
+    assert len(captured["media"]) == 2
+    bot.chat_state.pop(123, None)
+
+
 def test_unfetchable_attachment_gets_honest_error_not_silence(rate_guard_setup, monkeypatch):
     # Вложение есть, а скачать не вышло — честная ошибка вместо "Слушаю" в пустоту (прод 23.09.2026).
     message = rate_guard_setup()

tests/test_lumen_router_config.py

@@ -101,6 +101,13 @@ def test_looks_like_freshness_query_detects_english():
     assert lumen_router_config._looks_like_freshness_query("just chatting about life") is False
 
 
+def test_looks_like_freshness_query_ignores_substrings():
+    # Ревью ветки: голый now ловил know/snow — высокочастотные ложняки на обычной болтовне.
+    assert lumen_router_config._looks_like_freshness_query("I know kung fu") is False
+    assert lumen_router_config._looks_like_freshness_query("snow here") is False
+    assert lumen_router_config._looks_like_freshness_query("groundbreaking discovery") is False
+
+
 def test_looks_like_heavy_query_detects_english():
     # Внешний аудит: "write a Python parser" пролетал в light-ветку.
     assert lumen_router_config._looks_like_heavy_query("write a Python parser for csv") is True
@@ -109,6 +116,13 @@ def test_looks_like_heavy_query_detects_english():
     assert lumen_router_config._looks_like_heavy_query("just saying hello") is False
 
 
+def test_looks_like_heavy_query_ignores_substrings_and_handles_plurals():
+    # Ревью ветки: prove ловил improve, solve — resolve; "write two scripts" пролетал мимо.
+    assert lumen_router_config._looks_like_heavy_query("please improve my text") is False
+    assert lumen_router_config._looks_like_heavy_query("lets resolve this conflict") is False
+    assert lumen_router_config._looks_like_heavy_query("write two scripts for backup") is True
+
+
 def test_build_route_youtube_link_forces_gemini_only():
     route = lumen_router_config._build_route(needs_youtube=True, needs_website=False, media_mime=None, is_heavy=False, needs_freshness=False)
     assert all(p == "gemini" for p, _ in route)