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

Check for threads' existence properly

Browse source
This commit is contained in:
Nexus 2024-01-12 15:54:33 +00:00
parent 6600f49ddf
commit 930d86920a
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/cogs/ollama.py
View file

@ -124,7 +124,7 @@ class ChatHistory:
def get_thread(self, thread: str) -> dict[str, list[dict[str, str]] | discord.Member | int]: def get_thread(self, thread: str) -> dict[str, list[dict[str, str]] | discord.Member | int]:
"""Gets a copy of an entire thread""" """Gets a copy of an entire thread"""
return self._internal[thread].copy() return self._internal.get(thread, {}).copy()
SERVER_KEYS = list(CONFIG["ollama"].keys()) SERVER_KEYS = list(CONFIG["ollama"].keys())
@ -223,7 +223,7 @@ class Ollama(commands.Cog):
] ]
): ):
if context is not None: if context is not None:
if context not in self.contexts: if not self.history.get_thread(context):
await ctx.respond("Invalid context key.") await ctx.respond("Invalid context key.")
return return
await ctx.defer() await ctx.defer()