From 0ffa9903c48cabada0781fa8263d6a1e38e0f0f6 Mon Sep 17 00:00:00 2001 From: YoursFunny Date: Fri, 26 Apr 2024 15:19:04 +0800 Subject: [PATCH] add hint --- tweet.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tweet.py b/tweet.py index f719cb0..71d35f5 100644 --- a/tweet.py +++ b/tweet.py @@ -189,7 +189,8 @@ class TGTweet(Tweet): ) @property - def inline_query_generator(self) -> Generator[InlineQueryResultPhoto | InlineQueryResultVideo, None, None]: + def inline_query_generator(self) -> Generator[ + InlineQueryResultPhoto | InlineQueryResultVideo | InlineQueryResultMpeg4Gif, None, None]: for i, tweet_media in enumerate(self.media): logger.info(str(tweet_media)) if tweet_media.type == "image": @@ -217,7 +218,7 @@ class TGTweet(Tweet): ) @property - def pm_media_generator(self) -> Generator[InputMediaPhoto | InputMediaVideo, None, None]: + def pm_media_generator(self) -> Generator[InputMediaPhoto | InputMediaVideo | InputMediaAnimation, None, None]: for tweet_media in self.media: logger.info(str(tweet_media)) if tweet_media.type == "image":