diff --git a/flanabot/bots/flana_tele_bot.py b/flanabot/bots/flana_tele_bot.py index f52fffb..4c37dac 100644 --- a/flanabot/bots/flana_tele_bot.py +++ b/flanabot/bots/flana_tele_bot.py @@ -8,6 +8,7 @@ from typing import Callable, Sequence import flanautils import telethon.tl.functions +from flanautils import Media, OrderedSet from multibot import TelegramBot, find_message, user_client from flanabot.bots.flana_bot import FlanaBot @@ -58,6 +59,9 @@ class FlanaTeleBot(TelegramBot, FlanaBot): return [contact.user_id for contact in contacts_data.contacts] + async def _search_medias(self, message: Message, audio_only=False, timeout_for_media: int | float = None) -> OrderedSet[Media]: + return await super()._search_medias(message, audio_only, timeout_for_media=45) + @user_client async def _update_whitelist(self): self.whitelist_ids = [self.owner_id, self.id] + await self._get_contacts_ids()