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()