From 9924381b0b5ee484875176a1e23768f7f9a42107 Mon Sep 17 00:00:00 2001
From: nexy7574 <me@nexy7574.co.uk>
Date: Sat, 6 Jan 2024 22:07:31 +0000
Subject: [PATCH] Improve ollama UI

---
 src/cogs/ollama.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py
index 46867cd..f50f07d 100644
--- a/src/cogs/ollama.py
+++ b/src/cogs/ollama.py
@@ -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