From 92fdf79d545121e18bb5636a285339a5429ef9fc Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Wed, 1 Aug 2018 21:19:11 +0200
Subject: [PATCH] Expose the new InputMedia types

---
 pyrogram/__init__.py              | 9 ++++-----
 pyrogram/client/types/__init__.py | 4 ++++
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/pyrogram/__init__.py b/pyrogram/__init__.py
index f1442596..24bdc1a5 100644
--- a/pyrogram/__init__.py
+++ b/pyrogram/__init__.py
@@ -28,11 +28,10 @@ __version__ = "0.7.6dev1"
 from .api.errors import Error
 from .client.types import (
     Audio, Chat, ChatMember, ChatMembers, ChatPhoto, Contact, Document, InputMediaPhoto,
-    InputMediaVideo, InputPhoneContact, Location, Message, MessageEntity,
-    Dialog, Dialogs, Photo, PhotoSize, Sticker, Update, User, UserProfilePhotos,
-    Venue, GIF, Video, VideoNote, Voice, CallbackQuery, Messages, ForceReply,
-    InlineKeyboardButton, InlineKeyboardMarkup, KeyboardButton, ReplyKeyboardMarkup,
-    ReplyKeyboardRemove
+    InputMediaVideo, InputMediaDocument, InputMediaAudio, InputMediaAnimation, InputPhoneContact,
+    Location, Message, MessageEntity, Dialog, Dialogs, Photo, PhotoSize, Sticker, Update, User,
+    UserProfilePhotos, Venue, GIF, Video, VideoNote, Voice, CallbackQuery, Messages, ForceReply,
+    InlineKeyboardButton, InlineKeyboardMarkup, KeyboardButton, ReplyKeyboardMarkup, ReplyKeyboardRemove
 )
 from .client import (
     Client, ChatAction, ParseMode, Emoji,
diff --git a/pyrogram/client/types/__init__.py b/pyrogram/client/types/__init__.py
index e6fcc8aa..23c15818 100644
--- a/pyrogram/client/types/__init__.py
+++ b/pyrogram/client/types/__init__.py
@@ -23,6 +23,10 @@ from .chat_members import ChatMembers
 from .chat_photo import ChatPhoto
 from .dialog import Dialog
 from .dialogs import Dialogs
+from .input_media import InputMedia
+from .input_media_animation import InputMediaAnimation
+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 .input_phone_contact import InputPhoneContact