diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py index 8bbfeaa..95114f2 100644 --- a/src/cogs/ollama.py +++ b/src/cogs/ollama.py @@ -121,7 +121,7 @@ class ChatHistory: instance = cog.history return list( filter( - lambda v: ctx.value in v, map( + lambda v: (ctx.value or v) in v, map( lambda d: list(d.keys()), instance.threads_for(ctx.interaction.user) )