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

Logging

Browse source
This commit is contained in:
Nexus 2024-04-18 00:52:15 +01:00
parent 709ca94e29
commit 33319feb86
Signed by: nex
GPG key ID: 0FA334385D0B689F
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/cogs/ollama.py
View file

@ -891,11 +891,13 @@ class Ollama(commands.Cog):
await ctx.delete(delay=0.1) await ctx.delete(delay=0.1)
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...*")
self.log.debug("Acquiring lock")
async with self.lock: async with self.lock:
await ctx.respond(embed=embed, ephemeral=True) await ctx.respond(embed=embed, ephemeral=True)
last_edit = time.time() last_edit = time.time()
msg = None msg = None
with client.new_chat("orca-mini:3b", messages) as handler: with client.new_chat("orca-mini:3b", messages) as handler:
self.log.info("New chat connection established.")
async for ln in handler: async for ln in handler:
done = ln.get("done") is True done = ln.get("done") is True
embed.description = handler.result embed.description = handler.result