From f4d3dbf3bf8dfc8411e46900b147c11eba54a09e Mon Sep 17 00:00:00 2001
From: Danipulok <45077699+Danipulok@users.noreply.github.com>
Date: Fri, 21 Oct 2022 22:23:58 +0300
Subject: [PATCH] Fix typehints for `Message` media group methods (#1029)

* Fix typehints for `Message` media group methods

* Create 1029.doc.rst

* Fix importes
---
 CHANGES/1029.doc.rst     | 1 +
 aiogram/types/message.py | 6 ++++--
 2 files changed, 5 insertions(+), 2 deletions(-)
 create mode 100644 CHANGES/1029.doc.rst

diff --git a/CHANGES/1029.doc.rst b/CHANGES/1029.doc.rst
new file mode 100644
index 00000000..0f13c395
--- /dev/null
+++ b/CHANGES/1029.doc.rst
@@ -0,0 +1 @@
+Fix typehints for `Message` `reply_media_group` & `answer_media_group` methods
diff --git a/aiogram/types/message.py b/aiogram/types/message.py
index f4b26087..43ade2b4 100644
--- a/aiogram/types/message.py
+++ b/aiogram/types/message.py
@@ -45,6 +45,8 @@ if TYPE_CHECKING:
     from .game import Game
     from .inline_keyboard_markup import InlineKeyboardMarkup
     from .input_file import InputFile
+    from .input_media_audio import InputMediaAudio
+    from .input_media_document import InputMediaDocument
     from .input_media_photo import InputMediaPhoto
     from .input_media_video import InputMediaVideo
     from .invoice import Invoice
@@ -897,7 +899,7 @@ class Message(TelegramObject):
 
     def reply_media_group(
         self,
-        media: List[Union[InputMediaPhoto, InputMediaVideo]],
+        media: List[Union[InputMediaAudio, InputMediaDocument, InputMediaPhoto, InputMediaVideo]],
         disable_notification: Optional[bool] = None,
         allow_sending_without_reply: Optional[bool] = None,
     ) -> SendMediaGroup:
@@ -921,7 +923,7 @@ class Message(TelegramObject):
 
     def answer_media_group(
         self,
-        media: List[Union[InputMediaPhoto, InputMediaVideo]],
+        media: List[Union[InputMediaAudio, InputMediaDocument, InputMediaPhoto, InputMediaVideo]],
         disable_notification: Optional[bool] = None,
     ) -> SendMediaGroup:
         """