gooberinc/goober
3
0
Fork 1
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

add ability for server admins to retrain the bot periodically

Browse source
This commit is contained in:
WhatDidYouExpect 2025-04-04 00:05:42 +02:00
parent ddfd67df5c
commit 7877db4785
1 changed files with 17 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

15
bot.py
View file

@ -401,18 +401,23 @@ async def send_message(ctx, message=None, embed=None, file=None, edit=False, mes
app_commands.Choice(name="Select servers to retrain", value="select") app_commands.Choice(name="Select servers to retrain", value="select")
]) ])
async def retrain_models(ctx, option: app_commands.Choice[str]): async def retrain_models(ctx, option: app_commands.Choice[str]):
if ctx.author.id != ownerid: if ctx.author.id == ownerid:
return await ctx.send("You don't have permission to use this command.", ephemeral=True)
if option.value == "current": if option.value == "current":
server_id = ctx.guild.id if ctx.guild else "DM" server_id = ctx.guild.id if ctx.guild else "DM"
await retrain_single_server(ctx, server_id) await retrain_single_server(ctx, server_id)
elif option.value == "all": elif option.value == "all":
await retrain_all_servers(ctx) await retrain_all_servers(ctx)
elif option.value == "select": elif option.value == "select":
await show_server_selection(ctx) await show_server_selection(ctx)
else:
if ctx.guild and (ctx.author.guild_permissions.administrator or any(role.permissions.administrator for role in ctx.author.roles)):
if option.value == "current":
server_id = ctx.guild.id
await retrain_single_server(ctx, server_id)
else:
await ctx.send("You can only retrain the current server.", ephemeral=True)
else:
await ctx.send("You don't have permission to use this command.", ephemeral=True)
async def retrain_single_server(ctx, server_id): async def retrain_single_server(ctx, server_id):
memory_file = f"memories/memory_{server_id}.json" memory_file = f"memories/memory_{server_id}.json"