ping doesn't require a GPU server

This commit is contained in:
Nexus 2024-07-29 00:21:06 +01:00
parent 8ae75afb64
commit 3e110097bf

View file

@ -61,7 +61,7 @@ class AIModule(niobot.Module):
async def ping_command(self, ctx: niobot.Context):
"""Checks the bot is running."""
reply = await ctx.respond("Pong!")
server = await self.find_server()
server = await self.find_server(gpu_only=False)
if not server:
await reply.edit("Pong :(\nNo servers available.")
return
@ -117,7 +117,7 @@ class AIModule(niobot.Module):
await ctx.respond("You need to set a model first. See: `h!help ollama.set-model`")
return
model = users[ctx.message.sender]
server = await self.find_server()
server = await self.find_server(gpu_only=False)
if not server:
await ctx.respond("No servers available.")
return