From 69d262d65b13add7ebbcfae56716aa5aff790dae Mon Sep 17 00:00:00 2001
From: nexy7574 <git@nexy7574.co.uk>
Date: Sun, 9 Jun 2024 19:27:53 +0100
Subject: [PATCH] Indicate GPU servers in os

---
 src/cogs/ollama.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py
index 1dcb66f..08074f7 100644
--- a/src/cogs/ollama.py
+++ b/src/cogs/ollama.py
@@ -921,7 +921,8 @@ class Ollama(commands.Cog):
         if CONFIG["ollama"].get("order"):
             ln = ["Server order:"]
             for n, key in enumerate(CONFIG["ollama"].get("order"), start=1):
-                ln.append(f"{n}. {key!r}")
+                zap = '\N{high voltage sign}'
+                ln.append(f"{n}. {key!r} {f'({zap})' if CONFIG['ollama'][key].get('is_gpu') else ''}")
             embed.description = "\n".join(ln)
 
         for server, lock in self.servers.items():