nex/college-bot-v1
1
0
Fork 0
mirror of https://github.com/nexy7574/LCC-bot.git synced 2024-09-19 18:16:34 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add killswitch & client discovery

Browse source
This commit is contained in:
Nexus 2023-11-11 18:49:54 +00:00
parent 474fe98b38
commit a44dfb725c
Signed by: nex
GPG key ID: 0FA334385D0B689F
1 changed files with 20 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
cogs/other.py
View file

@ -1857,10 +1857,28 @@ class OtherCog(commands.Cog):
except ValueError:
host += ":11434"
else:
host = "192.168.0.90:11434"
try_hosts = [
"127.0.0.1:11434", # Localhost
"100.106.34.86:11434", # Laptop
"100.66.187.46:11434", # optiplex
"100.116.242.161:11434" # Raspberry Pi
]
async with httpx.AsyncClient(follow_redirects=True) as client:
for host in try_hosts:
try:
response = await client.get(
f"http://{host}/api/tags",
)
response.raise_for_status()
except (httpx.TransportError, httpx.NetworkError, httpx.HTTPStatusError):
continue
else:
break
else:
return await ctx.reply(":x: No servers available.")
msg = await ctx.reply(f"Preparing [{model!r}](http://{host}) <a:loading:1101463077586735174>")
async with httpx.AsyncClient(base_url=f"http://{host}/api") as client:
async with httpx.AsyncClient(base_url=f"http://{host}/api", follow_redirects=True) as client:
# get models
try:
response = await client.post("/show", json={"name": model})