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

Fix KeyError

Browse source
This commit is contained in:
Nexus 2024-01-12 15:48:58 +00:00
parent 05975de3af
commit d883f4044b
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/cogs/ollama.py
View file

@ -470,7 +470,7 @@ class Ollama(commands.Cog):
self.log.debug(f"Updating message ({last_update} -> {time.time()})") self.log.debug(f"Updating message ({last_update} -> {time.time()})")
last_update = time.time() last_update = time.time()
view.stop() view.stop()
self.history.add_message(context, "user", user_message["content"], user_message["images"]) self.history.add_message(context, "user", user_message["content"], user_message.get("images"))
self.history.add_message(context, "assistant", buffer.getvalue()) self.history.add_message(context, "assistant", buffer.getvalue())
embed.add_field(name="Context Key", value=context, inline=True) embed.add_field(name="Context Key", value=context, inline=True)
self.log.debug("Ollama finished consuming.") self.log.debug("Ollama finished consuming.")