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

Improve ollama UI

Browse source
This commit is contained in:
Nexus 2024-01-06 22:07:31 +00:00
parent a46e4c0797
commit 9924381b0b
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/cogs/ollama.py
View file

@ -210,9 +210,11 @@ class Ollama(commands.Cog):
embed = discord.Embed(
title="Generating response...",
description=">>> \u200b",
color=discord.Color.blurple()
description=">>> ",
color=discord.Color.blurple(),
timestamp=discord.utils.utcnow()
)
await ctx.edit(embed=embed)
async with session.post(
"/api/generate",
json={
@ -241,6 +243,7 @@ class Ollama(commands.Cog):
embed.title = "Done!"
embed.color = discord.Color.green()
embed.description += line["response"]
embed.timestamp = discord.utils.utcnow()
if len(embed.description) >= 4096:
embed.description = embed.description[:4093] + "..."
break