fix
This commit is contained in:
parent
c5ca971134
commit
172b604a03
1 changed files with 9 additions and 6 deletions
|
@ -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()
|
||||||
|
|
Loading…
Reference in a new issue