nex/college-bot-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Activity Actions

fix

Browse source
This commit is contained in:
Nexus 2024-04-18 00:49:56 +01:00
parent c5ca971134
commit 172b604a03
Signed by: nex
GPG key ID: 0FA334385D0B689F
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/cogs/ollama.py
View file

@ -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,6 +907,11 @@ 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 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()