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

15
src/cogs/ollama.py
View file

@ -892,13 +892,11 @@ class Ollama(commands.Cog):
messages = self.history.get_history(thread) messages = self.history.get_history(thread)
embed = discord.Embed(description="*Waking Ollama up...*") embed = discord.Embed(description="*Waking Ollama up...*")
async with self.lock: async with self.lock:
msg = await message.reply(embed=embed) await ctx.respond(embed=embed, ephemeral=True)
last_edit = time.time() last_edit = time.time()
msg = None
with client.new_chat("orca-mini:3b", messages) as handler: with client.new_chat("orca-mini:3b", messages) as handler:
deleted = False
async for ln in handler: async for ln in handler:
if not deleted:
await ctx.delete()
done = ln.get("done") is True done = ln.get("done") is True
embed.description = handler.result embed.description = handler.result
if len(embed.description) >= 4096: if len(embed.description) >= 4096:
@ -909,9 +907,14 @@ class Ollama(commands.Cog):
else: else:
embed.colour = discord.Color.blurple() embed.colour = discord.Color.blurple()
embed.set_footer(text="Using server %r" % server, icon_url=CONFIG["ollama"][server].get("icon_url")) 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: if msg is None:
await ctx.edit(embed=embed) await ctx.delete(delay=0.1)
msg = await message.reply(embed=embed)
last_edit = time.time() 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: if done:
break break
embed.colour = discord.Colour.dark_theme() embed.colour = discord.Colour.dark_theme()