From d883f4044b40c052b97e58e43ed32150672277eb Mon Sep 17 00:00:00 2001 From: nexy7574 Date: Fri, 12 Jan 2024 15:48:58 +0000 Subject: [PATCH] Fix KeyError --- src/cogs/ollama.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py index 3239617..773e408 100644 --- a/src/cogs/ollama.py +++ b/src/cogs/ollama.py @@ -470,7 +470,7 @@ class Ollama(commands.Cog): self.log.debug(f"Updating message ({last_update} -> {time.time()})") last_update = time.time() 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()) embed.add_field(name="Context Key", value=context, inline=True) self.log.debug("Ollama finished consuming.")