From 8ccdeb8a0e7ff6366343747336e144f423150d9b Mon Sep 17 00:00:00 2001 From: nexy7574 Date: Wed, 10 Jan 2024 15:35:18 +0000 Subject: [PATCH] Set author to indicate which model is in use --- src/cogs/ollama.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py index 51f7e95..c9669a8 100644 --- a/src/cogs/ollama.py +++ b/src/cogs/ollama.py @@ -200,6 +200,11 @@ class Ollama(commands.Cog): color=discord.Color.blurple(), timestamp=discord.utils.utcnow() ) + embed.set_author( + name=model, + url="https://ollama.ai/library/" + model.split(":")[0], + icon_url="https://ollama.ai/public/ollama.png" + ) embed.add_field( name="Prompt", value=">>> " + textwrap.shorten(query, width=1020, placeholder="..."),