From 34a57af998149db7e9d0f9b00ad4c6ba96646861 Mon Sep 17 00:00:00 2001 From: nex Date: Tue, 6 Feb 2024 00:54:53 +0000 Subject: [PATCH] Properly avoid text overflow --- src/cogs/ollama.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py index 7e6de18..7828bf9 100644 --- a/src/cogs/ollama.py +++ b/src/cogs/ollama.py @@ -534,8 +534,14 @@ class Ollama(commands.Cog): buffer.write(line["message"]["content"]) embed.description += line["message"]["content"] embed.timestamp = discord.utils.utcnow() - if len(embed.description) >= 4096: - embed.description = embed.description = "..." + line["message"]["content"] + if len(embed.description) >= 4000: + embed.description = "[...]" + line["message"]["content"] + if len(embed.description) >= 3250: + embed.colour = discord.Color.gold() + embed.set_footer(text="Warning: {:,}/4096 characters.".format(len(embed.description))) + else: + embed.colour = discord.Color.blurple() + embed.set_footer(text="Using server %r" % server, icon_url=server_config.get("icon_url")) if view.cancel.is_set(): break