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

Set author to indicate which model is in use

Browse source
This commit is contained in:
Nexus 2024-01-10 15:35:18 +00:00
parent f3aadd2ce0
commit 8ccdeb8a0e
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/cogs/ollama.py
View file

@ -200,6 +200,11 @@ class Ollama(commands.Cog):
color=discord.Color.blurple(), color=discord.Color.blurple(),
timestamp=discord.utils.utcnow() 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( embed.add_field(
name="Prompt", name="Prompt",
value=">>> " + textwrap.shorten(query, width=1020, placeholder="..."), value=">>> " + textwrap.shorten(query, width=1020, placeholder="..."),