diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py index af71532..643fb67 100644 --- a/src/cogs/ollama.py +++ b/src/cogs/ollama.py @@ -937,13 +937,19 @@ class Ollama(commands.Cog): embed.colour = discord.Colour.dark_theme() return await msg.edit(embed=embed) - @commands.command() + @commands.command(name="ollama-status", aliases=["ollama_status", "os"]) async def ollama_status(self, ctx: commands.Context): embed = discord.Embed( title="Ollama server statuses:", color=discord.Color.blurple(), timestamp=discord.utils.utcnow(), ) + if CONFIG["ollama"].get("order"): + ln = [f"Server order:"] + for n, key in enumerate(CONFIG["ollama"].get("order"), start=1): + ln.append(f"{n}. {key!r}") + embed.description = "\n".join(ln) + for server, lock in self.servers.items(): embed.add_field( name=server, @@ -959,7 +965,7 @@ class Ollama(commands.Cog): ok = await self.check_server(CONFIG["ollama"][server]["base_url"]) except Exception as e: ok = False - self.log.error("Error while checking server %r: %r", server, e) + self.log.error("Error while checking server %r: %r", server, e, exc_info=True) for field in embed.fields: if field.name == server: field.value = "\U00002705 Online" if ok else "\U0000274c Offline"