diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py index ec1dbe3..6f17d23 100644 --- a/src/cogs/ollama.py +++ b/src/cogs/ollama.py @@ -948,10 +948,9 @@ class Ollama(commands.Cog): } self.history.add_message(thread, "user", user_message["content"]) - server = self.next_server(False) - while not await self.check_server(server): + for _ in range(10): server = self.next_server() - if await self.check_server(server): + if await self.check_server(CONFIG["ollama"][server]["base_url"]): break else: return await ctx.respond("All servers are offline. Please try again later.", ephemeral=True)