From e9b95eaff7b5c54e40d1bafd23fd662021164887 Mon Sep 17 00:00:00 2001 From: Gnome Date: Sat, 4 Sep 2021 21:26:32 +0100 Subject: [PATCH] Fix typing on Context.reply --- discord/ext/commands/context.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/discord/ext/commands/context.py b/discord/ext/commands/context.py index 800c5d7a..8cc2afb6 100644 --- a/discord/ext/commands/context.py +++ b/discord/ext/commands/context.py @@ -473,6 +473,11 @@ class Context(discord.abc.Messageable, Generic[BotT]): return await send(content, ephemeral=ephemeral, **kwargs) # type: ignore + @overload + async def reply(self, content: Optional[str] = None, return_message: Literal[False] = False, **kwargs: Any) -> Optional[Union[Message, WebhookMessage]]: ... + @overload + async def reply(self, content: Optional[str] = None, return_message: Literal[True] = True, **kwargs: Any) -> Union[Message, WebhookMessage]: ... + @discord.utils.copy_doc(Message.reply) - async def reply(self, content: Optional[str] = None, **kwargs: Any) -> Message: - return await self.send(content, reference=self.message, **kwargs) # type: ignore + async def reply(self, content: Optional[str] = None, return_message: bool = True, **kwargs: Any) -> Optional[Union[Message, WebhookMessage]]: + return await self.send(content, return_message=return_message, reference=self.message, **kwargs) # type: ignore