Align daily quota accounting across draw TTS and 429 paths

lumen_commands.py

@@ -65,6 +65,18 @@ async def cmd_start(message: Message) -> None:
 
 async def inline_draw(message: Message, prompt: str) -> None:
     import bot
+    # Дневной лимит пользователя, как у озвучки: проба без создания.
+    uid = bot._user_key_for_message(message)
+    _draw_entry = bot._user_daily_peek(uid)
+    if bot._user_daily_total_exhausted(uid, _draw_entry):
+        hours, mins = bot._user_daily_reset_in()
+        bot._record_stats_event("daily_limit_denials")
+        await bot._safe_reply(message, bot._t(
+            message.chat.id, "user_daily_total",
+            used=(_draw_entry or {}).get("total", 0), limit=bot._user_daily_limit(uid, "total", _draw_entry),
+            hours=hours, mins=mins,
+        ))
+        return
     status = await bot._tg_call(message.reply, bot._t(message.chat.id, "status_generating_image"))
     try:
         session = await bot._get_http_session()
@@ -129,6 +141,7 @@ async def inline_draw(message: Message, prompt: str) -> None:
             if sent is None:
                 raise RuntimeError("Telegram send_photo failed: connection timeout or proxy unavailable")
             # Суточный счётчик /stats: картинка ушла пользователю.
+            bot._record_user_daily(uid)
             bot._record_stats_event("answers_sent")
         else:
             if rate_limited:
@@ -395,8 +408,8 @@ async def inline_tts(message: Message, text: str) -> None:
         )
         if sent is None:
             raise RuntimeError("Telegram send_voice failed: connection timeout or proxy unavailable")
-    # Один запрос — один счёт: чанки одного сообщения не множат дневной расход.
-    bot._record_user_daily(uid, tts=True)
+        # Каждый чанк — отдельный синтез за квоту: списываем поштучно.
+        bot._record_user_daily(uid, tts=True)
     bot._record_stats_event("answers_sent")
     if shortened:
         # Обрезка по общему дедлайну — говорим прямо, что озвучено начало.

lumen_routes.py

@@ -835,8 +835,13 @@ async def ask_gemini(
                     log.warning("[gemini] Model %s quota exhausted (429). Switching to %s", curr_model_id, next_model)
                     curr_model_id = next_model
                     continue
-                log.warning("[gemini] All Gemini models in route exhausted their quota (429): %s", ", ".join(quota_exhausted_models))
-                raise bot.GeminiAllModelsExhaustedError(quota_exhausted_models) from exc
+                if quota_exhausted_models:
+                    log.warning("[gemini] All Gemini models in route exhausted their quota (429): %s", ", ".join(quota_exhausted_models))
+                    raise bot.GeminiAllModelsExhaustedError(quota_exhausted_models) from exc
+                # Одни минутные всплески: суточного исчерпания не было — отдаём
+                # последний 429 как есть, а не пустой "исчерпано всё".
+                log.warning("[gemini] All Gemini models in route rate-limited (429), no daily exhaustion.")
+                raise
 
             # Остальные исходы — одна попытка и сразу следующая модель, без ретраев.
             next_model = bot._next_fallback_model(tried_models, chain)

lumen_streaming.py

@@ -393,7 +393,10 @@ async def _run_streaming_reply(
             try:
                 _txt = bot._error_text(exc).strip() or exc.__class__.__name__
                 if bot._classify_model_error(bot._error_status(exc, _txt), _txt) == "rate_limit":
-                    if bot._is_account_wide_or_rate_limit(_txt.lower()):
+                    # Дневную квоту опознаём предикатом провайдера, как текстовый
+                    # путь: общий account-wide ловит только OpenRouter-тексты.
+                    _daily = bot._is_gemini_daily_quota(_txt) if provider == "gemini" else bot._is_account_wide_or_rate_limit(_txt.lower())
+                    if _daily:
                         bot._mark_quota_exhausted(provider, model_id)
                     else:
                         bot._mark_rate_limited(provider, model_id)

tests/test_bot_commands.py

@@ -584,6 +584,61 @@ def test_inline_draw_picks_model_from_prompt_without_touching_chat_state():
         bot.chat_state.pop(chat_id, None)
 
 
+def test_inline_draw_refuses_when_user_total_limit_exhausted(monkeypatch):
+    # /draw без дневного лимита обходился: исчерпанный total — отказ до генерации.
+    from types import SimpleNamespace
+    incoming = _FakeIncomingMessage(999491)
+    incoming.message_id = 1
+    incoming.from_user = SimpleNamespace(id=777021)
+    monkeypatch.setattr(bot, "OWNER_ID", 1)
+    monkeypatch.setattr(bot, "DAILY_USER_MESSAGE_LIMIT", 2)
+    bot._record_user_daily(777021)
+    bot._record_user_daily(777021)
+
+    async def fail_generate(session, model_id, prompt, timeout_sec=None):
+        raise AssertionError("generation must not run for an exhausted user")
+
+    monkeypatch.setattr(bot, "_pollinations_text_to_image", fail_generate)
+    replies = []
+
+    async def fake_safe_reply(msg, text, **kwargs):
+        replies.append(text)
+
+    monkeypatch.setattr(bot, "_safe_reply", fake_safe_reply)
+    try:
+        asyncio.run(bot.inline_draw(incoming, "нарисуй кота"))
+        assert len(replies) == 1
+        assert bot._user_daily_entry(777021)["total"] == 2
+    finally:
+        bot.GLOBAL_QUOTA.get("user_daily", {}).pop("777021", None)
+        bot.chat_state.pop(999491, None)
+
+
+def test_inline_draw_records_user_daily_on_success(monkeypatch):
+    # Успешная картинка растит total, как озвучка.
+    from types import SimpleNamespace
+    incoming = _FakeIncomingMessage(999492)
+    incoming.message_id = 1
+    incoming.from_user = SimpleNamespace(id=777022)
+    monkeypatch.setattr(bot, "OWNER_ID", 1)
+
+    async def fake_generate(session, model_id, prompt, timeout_sec=None):
+        return b"\x89PNG fake bytes"
+
+    class _FakePhotoBot:
+        async def send_photo(self, **kwargs):
+            return SimpleNamespace()
+
+    monkeypatch.setattr(bot, "_pollinations_text_to_image", fake_generate)
+    monkeypatch.setattr(bot, "bot", _FakePhotoBot())
+    try:
+        asyncio.run(bot.inline_draw(incoming, "нарисуй кота"))
+        assert bot._user_daily_entry(777022)["total"] == 1
+    finally:
+        bot.GLOBAL_QUOTA.get("user_daily", {}).pop("777022", None)
+        bot.chat_state.pop(999492, None)
+
+
 def test_inline_draw_stops_fallback_chain_when_time_budget_exceeded():
     # Регрессия на находку код-ревью (28 августа 2026): раньше у /draw не было
     # общего бюджета времени на всю фолбэк-цепочку — при недоступности сервиса

tests/test_bot_routes.py

@@ -927,6 +927,31 @@ def test_ask_gemini_raises_all_models_exhausted_when_entire_chain_429s():
         bot.GLOBAL_QUOTA["gemini"].pop("gemini-2.5-flash", None)
 
 
+def test_ask_gemini_minute_429s_do_not_raise_empty_exhausted():
+    # Одни минутные всплески: суточного исчерпания не было — последний 429
+    # идёт как есть, а не пустой "исчерпано всё".
+    chat_id = 999017
+
+    class _Minute429(Exception):
+        status_code = 429
+
+    def fake_generate_content(*, model, contents, config=None):
+        raise _Minute429("RESOURCE_EXHAUSTED: quota exceeded for quota metric Generate requests per minute")
+
+    fake_client = MagicMock()
+    fake_client.aio.models.generate_content = AsyncMock(side_effect=fake_generate_content)
+    original_client = bot.client
+    bot.client = fake_client
+    try:
+        with pytest.raises(_Minute429):
+            asyncio.run(bot.ask_gemini(chat_id, "Привет", model_chain=["gemini-3.6-flash", "gemini-2.5-flash"]))
+    finally:
+        bot.client = original_client
+        bot.chat_state.pop(chat_id, None)
+        bot.GLOBAL_QUOTA["gemini"].pop("gemini-3.6-flash", None)
+        bot.GLOBAL_QUOTA["gemini"].pop("gemini-2.5-flash", None)
+
+
 def test_ask_gemini_falls_back_to_next_model_on_timeout():
     # Прежняя версия использовала блокирующий time.sleep в async-фейке и один и тот
     # же текст ответа для обеих моделей: таймаут никогда не срабатывал, фолбэка не

tests/test_bot_streaming.py

@@ -920,6 +920,30 @@ def test_send_text_pauses_between_chunks(monkeypatch):
         bot.chat_state.pop(999114, None)
 
 
+def test_gemini_streaming_daily_429_marks_exhausted_like_text_path():
+    # Стриминг опознавал суточную квоту чужим предикатом и ставил минутную
+    # остывку вместо метки до утра.
+    chat_id = 999116
+    model = bot.DEFAULT_GEMINI_MODEL
+    fake_stream = _fake_gemini_stream(raises=RuntimeError("429 Quota exceeded for the day"))
+    fake_client = MagicMock()
+    fake_client.aio.models.generate_content_stream = fake_stream
+    incoming = _FakeIncomingMessage(chat_id)
+    original_client = bot.client
+    bot.client = fake_client
+    bot.GLOBAL_QUOTA.setdefault("gemini", {}).pop(model, None)
+    try:
+        answer, _ = asyncio.run(bot._try_gemini_streaming(chat_id, "Привет!", incoming, model))
+        assert answer is None
+        entry = bot.GLOBAL_QUOTA["gemini"][model]
+        assert entry["exhausted_at"] is not None
+        assert lumen_router_config._is_quota_exhausted("gemini", model) is True
+    finally:
+        bot.client = original_client
+        bot.chat_state.pop(chat_id, None)
+        bot.GLOBAL_QUOTA["gemini"].pop(model, None)
+
+
 def test_rich_send_used_for_final_answer_with_table():
     chat_id = 999401
     incoming = _FakeIncomingMessage(chat_id)

tests/test_bot_tts.py

@@ -533,3 +533,32 @@ def test_inline_tts_records_user_daily_on_success(monkeypatch):
         bot.chat_state.pop(999989, None)
         bot.GLOBAL_QUOTA["gemini"].pop("gemini-3.1-flash-tts-preview", None)
 
+
+def test_inline_tts_charges_each_synthesized_chunk():
+    # Каждый чанк — отдельный синтез за квоту: списываем поштучно, а не раз за сообщение.
+    limit = bot.TTS_MAX_CHARS
+    part = "Предложение номер раз про интересные вещи. "
+    text = part * ((limit // len(part)) + 2)
+    assert len(text) > limit
+
+    incoming = _FakeIncomingMessage(999810)
+    incoming.message_id = 12360
+    incoming.from_user = SimpleNamespace(id=777015)
+
+    original_client = bot.client
+    original_bot = bot.bot
+    bot.client = _wav_client()
+    counting = _CountingVoiceBot()
+    bot.bot = counting
+    try:
+        asyncio.run(bot.inline_tts(incoming, text))
+        assert len(counting.voices) == 2
+        entry = bot.GLOBAL_QUOTA.get("user_daily", {}).get("777015")
+        assert entry is not None
+        assert entry["tts"] == 2 and entry["total"] == 2
+    finally:
+        bot.client = original_client
+        bot.bot = original_bot
+        bot.GLOBAL_QUOTA.get("user_daily", {}).pop("777015", None)
+        bot.chat_state.pop(999810, None)
+