diff --git a/cogs/text_service_cog.py b/cogs/text_service_cog.py index 2f29ea4..a16926d 100644 --- a/cogs/text_service_cog.py +++ b/cogs/text_service_cog.py @@ -425,9 +425,10 @@ class GPT3ComCon(discord.Cog, name="GPT3ComCon"): first = True else: if from_context: - await ctx.send_followup(chunk) + response_message = await ctx.send_followup(chunk) else: - await ctx.channel.send(chunk) + response_message = await ctx.channel.send(chunk) + return response_message async def paginate_embed(self, response_text, codex, prompt=None, instruction=None): """Given a response text make embed pages and return a list of the pages. Codex makes it a codeblock in the embed""" diff --git a/services/text_service.py b/services/text_service.py index f3cf460..8efa191 100644 --- a/services/text_service.py +++ b/services/text_service.py @@ -432,7 +432,7 @@ class TextService: if len(response_text) > converser_cog.TEXT_CUTOFF: if not from_context: paginator = None - await converser_cog.paginate_and_send(response_text, ctx) + response_message = await converser_cog.paginate_and_send(response_text, ctx) else: embed_pages = await converser_cog.paginate_embed( response_text, codex, prompt, instruction