fix

IlyaGusev · Jul 8, 2024 · 6d0c724 · 6d0c724
1 parent 942ea34
commit 6d0c724
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/bot.py b/src/bot.py
@@ -429,6 +429,9 @@ async def get_count(self, message: Message) -> None:
         user_id = message.from_user.id
         chat_id = message.chat.id
         model = self.db.get_current_model(chat_id)
+        if model not in self.providers:
+            await message.reply(self.localization.MODEL_NOT_SUPPORTED)
+            return
         remaining_count = self._count_remaining_messages(user_id=user_id, model=model)
         text = self.localization.REMAINING_MESSAGES.format(model=model, remaining_count=remaining_count)
         await message.reply(text)