diff --git a/src/bot.py b/src/bot.py index 763304a..c4040e6 100644 --- 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)