From 172b604a032e9f44d655a823bf32cca2a5b27d8d Mon Sep 17 00:00:00 2001 From: nexy7574 Date: Thu, 18 Apr 2024 00:49:56 +0100 Subject: [PATCH] fix --- src/cogs/ollama.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py index 0617846..49feb39 100644 --- a/src/cogs/ollama.py +++ b/src/cogs/ollama.py @@ -892,13 +892,11 @@ class Ollama(commands.Cog): messages = self.history.get_history(thread) embed = discord.Embed(description="*Waking Ollama up...*") async with self.lock: - msg = await message.reply(embed=embed) + await ctx.respond(embed=embed, ephemeral=True) last_edit = time.time() + msg = None with client.new_chat("orca-mini:3b", messages) as handler: - deleted = False async for ln in handler: - if not deleted: - await ctx.delete() done = ln.get("done") is True embed.description = handler.result if len(embed.description) >= 4096: @@ -909,9 +907,14 @@ class Ollama(commands.Cog): else: embed.colour = discord.Color.blurple() embed.set_footer(text="Using server %r" % server, icon_url=CONFIG["ollama"][server].get("icon_url")) - if time.time() >= (last_edit + 5.1) or done is True: - await ctx.edit(embed=embed) + if msg is None: + await ctx.delete(delay=0.1) + msg = await message.reply(embed=embed) last_edit = time.time() + else: + if time.time() >= (last_edit + 5.1) or done is True: + await ctx.edit(embed=embed) + last_edit = time.time() if done: break embed.colour = discord.Colour.dark_theme()