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

Fix ollama server selector [pt 1]

Browse source
This commit is contained in:
Nexus 2023-11-22 15:50:15 +00:00
parent 487def9693
commit 0ec616ea3f
Signed by: nex
GPG key ID: 0FA334385D0B689F
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cogs/other.py
View file

@ -1904,7 +1904,7 @@ class OtherCog(commands.Cog):
except ValueError:
model = model + ":latest"
servers = {
servers: dict[str, dict[str, str, list[str] | int]] = {
"100.106.34.86:11434": {
"name": "NexTop",
"allow": [
@ -1926,7 +1926,7 @@ class OtherCog(commands.Cog):
},
"100.66.187.46:11434": {
"name": "NexBox",
"restrict-to": [
"allow": [
"orca-mini:latest",
"orca-mini:3b",
"orca-mini:7b",
@ -1935,6 +1935,12 @@ class OtherCog(commands.Cog):
},
}
def model_is_allowed(model_name: str, srv: dict[str, str | list[str] | int]) -> bool:
for pat in srv.get("allow", ['*']):
if not fnmatch.fnmatch(model_name.casefold(), pat.casefold()):
return False
return True
class ServerSelector(discord.ui.View):
def __init__(self):
super().__init__(
@ -1951,7 +1957,7 @@ class OtherCog(commands.Cog):
value=x
)
for x, y in servers.items()
if fnmatch.fnmatch(model, y.get("restrict-to", ['*']))
if model_is_allowed(model, y)
] + [
discord.SelectOption(
label="Custom",