mirror of
https://github.com/Paillat-dev/Botator.git
synced 2026-01-02 01:06:19 +00:00
Paillat
8771247612
🐛 fix(ChatProcess.py): fix logic error in the return criteria for determining if the bot should respond to a message 🐛 fix(ChatProcess.py): fix typo in the 'functions' variable name 🐛 fix(ChatProcess.py): fix typo in the 'functions' parameter name in the request function call 🐛 fix(ChatProcess.py): fix typo in the 'functions' parameter name in the processFunctioncallResponse function call 🐛 fix(ChatProcess.py): remove unnecessary print statement in the processMessage function 🐛 fix(prompts.py): remove unnecessary print statement in the createPrompt function 🐛 fix(channelSetup.py): fix logic error in the is_owner function call 🐛 fix(moderation.py): remove unnecessary code for disabling moderation 🐛 fix(config.py): remove unnecessary code for creating tables in the database 🐛 fix(functionscalls.py): fix type hint for the return value of the call_function function 🐛 fix(guild.py): fix handling of serialized data in the load function 🐛 fix(SqlConnector.py): create setup_data table if it does not exist
269 lines
10 KiB
Python
269 lines
10 KiB
Python
import discord
|
|
import datetime
|
|
|
|
from discord import SlashCommandGroup
|
|
from discord import default_permissions
|
|
from discord.ext.commands import guild_only
|
|
from discord.ext import commands
|
|
from src.utils.variousclasses import models, characters, apis
|
|
from src.guild import Guild
|
|
|
|
sampleDataFormatExample = {
|
|
"guild_id": 1234567890,
|
|
"premium": False,
|
|
"premium_expiration": 0,
|
|
}
|
|
|
|
|
|
class ChannelSetup(commands.Cog):
|
|
def __init__(self, bot: discord.Bot):
|
|
super().__init__()
|
|
self.bot = bot
|
|
|
|
setup = SlashCommandGroup(
|
|
"setup",
|
|
description="Setup commands for the bot, inlcuding channels, models, and more.",
|
|
)
|
|
|
|
setup_channel = setup.create_subgroup(
|
|
name="channel", description="Setup, add, or remove channels for the bot to use."
|
|
)
|
|
|
|
setup_guild = setup.create_subgroup(
|
|
name="server", description="Setup the settings for the server."
|
|
)
|
|
|
|
@setup_channel.command(
|
|
name="add",
|
|
description="Add a channel for the bot to use. Can also specify server-wide settings.",
|
|
)
|
|
@discord.option(
|
|
name="channel",
|
|
description="The channel to setup. If not specified, will use the current channel.",
|
|
type=discord.TextChannel,
|
|
required=False,
|
|
)
|
|
@discord.option(
|
|
name="model",
|
|
description="The model to use for this channel.",
|
|
type=str,
|
|
required=False,
|
|
autocomplete=models.autocomplete,
|
|
)
|
|
@discord.option(
|
|
name="character",
|
|
description="The character to use for this channel.",
|
|
type=str,
|
|
required=False,
|
|
autocomplete=characters.autocomplete,
|
|
)
|
|
@guild_only()
|
|
async def channel(
|
|
self,
|
|
ctx: discord.ApplicationContext,
|
|
channel: discord.TextChannel = None,
|
|
model: str = models.default,
|
|
character: str = characters.default,
|
|
):
|
|
if channel is None:
|
|
channel = ctx.channel
|
|
guild = Guild(ctx.guild.id)
|
|
guild.load()
|
|
if not guild.premium:
|
|
if (
|
|
len(guild.channels) >= 1
|
|
and guild.channels.get(str(channel.id), None) is None
|
|
):
|
|
await ctx.respond(
|
|
"`Warning: You are not a premium user, and can only have one channel setup. The settings will still be saved, but will not be used.`",
|
|
ephemeral=True,
|
|
)
|
|
if model != models.default:
|
|
await ctx.respond(
|
|
"`Warning: You are not a premium user, and can only use the default model. The settings will still be saved, but will not be used.`",
|
|
ephemeral=True,
|
|
)
|
|
if character != characters.default:
|
|
await ctx.respond(
|
|
"`Warning: You are not a premium user, and can only use the default character. The settings will still be saved, but will not be used.`",
|
|
ephemeral=True,
|
|
)
|
|
if guild.api_keys.get("openai", None) is None:
|
|
await ctx.respond(
|
|
"`Error: No openai api key is set. The api key is needed for the openai models, as well as for the content moderation. The openai models will cost you tokens in your openai account. However, if you use one of the llama models, you will not be charged, but the api key is still needed for content moderation, wich is free but requires an api key.`",
|
|
ephemeral=True,
|
|
)
|
|
guild.addChannel(
|
|
channel, models.matchingDict[model], characters.matchingDict[character]
|
|
)
|
|
await ctx.respond(
|
|
f"Set channel {channel.mention} with model `{model}` and character `{character}`.",
|
|
ephemeral=True,
|
|
)
|
|
|
|
@setup_channel.command(
|
|
name="remove", description="Remove a channel from the bot's usage."
|
|
)
|
|
@discord.option(
|
|
name="channel",
|
|
description="The channel to remove. If not specified, will use the current channel.",
|
|
type=discord.TextChannel,
|
|
required=False,
|
|
)
|
|
@guild_only()
|
|
async def remove(
|
|
self, ctx: discord.ApplicationContext, channel: discord.TextChannel = None
|
|
):
|
|
if channel is None:
|
|
channel = ctx.channel
|
|
guild = Guild(ctx.guild.id)
|
|
guild.load()
|
|
if guild.getChannelInfo(str(channel.id)) is None:
|
|
await ctx.respond("That channel is not setup.")
|
|
return
|
|
guild.delChannel(channel)
|
|
await ctx.respond(f"Removed channel {channel.mention}.", ephemeral=True)
|
|
|
|
@setup_guild.command(
|
|
name="set",
|
|
description="Set the settings for the guild (when the bot is pinged outside of a set channel).",
|
|
)
|
|
@discord.option(
|
|
name="model",
|
|
description="The model to use for this channel.",
|
|
type=str,
|
|
required=False,
|
|
autocomplete=models.autocomplete,
|
|
)
|
|
@discord.option(
|
|
name="character",
|
|
description="The character to use for this channel.",
|
|
type=str,
|
|
required=False,
|
|
autocomplete=characters.autocomplete,
|
|
)
|
|
@guild_only()
|
|
async def setSettings(
|
|
self,
|
|
ctx: discord.ApplicationContext,
|
|
model: str = models.default,
|
|
character: str = characters.default,
|
|
):
|
|
# we will be using "serverwide" as the channel id for the guild settings
|
|
guild = Guild(ctx.guild.id)
|
|
guild.load()
|
|
if not guild.premium:
|
|
if model != models.default:
|
|
await ctx.respond(
|
|
"`Warning: You are not a premium user, and can only use the default model. The settings will still be saved, but will not be used.`",
|
|
ephemeral=True,
|
|
)
|
|
if character != characters.default:
|
|
await ctx.respond(
|
|
"`Warning: You are not a premium user, and can only use the default character. The settings will still be saved, but will not be used.`",
|
|
ephemeral=True,
|
|
)
|
|
if guild.api_keys.get("openai", None) is None:
|
|
await ctx.respond(
|
|
"`Error: No openai api key is set. The api key is needed for the openai models, as well as for the content moderation. The openai models will cost you tokens in your openai account. However, if you use one of the llama models, you will not be charged, but the api key is still needed for content moderation, wich is free but requires an api key.`",
|
|
ephemeral=True,
|
|
)
|
|
guild.addChannel(
|
|
"serverwide", models.matchingDict[model], characters.matchingDict[character]
|
|
)
|
|
await ctx.respond(
|
|
f"Set server settings with model `{model}` and character `{character}`.",
|
|
ephemeral=True,
|
|
)
|
|
|
|
@setup_guild.command(name="remove", description="Remove the guild settings.")
|
|
@guild_only()
|
|
async def removeSettings(self, ctx: discord.ApplicationContext):
|
|
guild = Guild(ctx.guild.id)
|
|
guild.load()
|
|
if "serverwide" not in guild.channels:
|
|
await ctx.respond("No guild settings are setup.")
|
|
return
|
|
guild.delChannel("serverwide")
|
|
await ctx.respond(f"Removed serverwide settings.", ephemeral=True)
|
|
|
|
@setup.command(name="list", description="List all channels that are setup.")
|
|
@guild_only()
|
|
async def list(self, ctx: discord.ApplicationContext):
|
|
guild = Guild(ctx.guild.id)
|
|
guild.load()
|
|
if len(guild.channels) == 0:
|
|
await ctx.respond("No channels are setup.")
|
|
return
|
|
embed = discord.Embed(
|
|
title="Channels",
|
|
description="All channels that are setup.",
|
|
color=discord.Color.nitro_pink(),
|
|
)
|
|
channels = guild.sanitizedChannels
|
|
for channel in channels:
|
|
if channel == "serverwide":
|
|
mention = "Serverwide"
|
|
else:
|
|
mention = f"<#{channel}>"
|
|
model = models.reverseMatchingDict[channels[channel]["model"]]
|
|
character = characters.reverseMatchingDict[channels[channel]["character"]]
|
|
embed.add_field(
|
|
name=f"{mention}",
|
|
value=f"Model: `{model}`\nCharacter: `{character}`",
|
|
inline=False,
|
|
)
|
|
await ctx.respond(embed=embed)
|
|
|
|
@setup.command(name="api", description="Set an API key for the bot to use.")
|
|
@discord.option(
|
|
name="api",
|
|
description="The API to set. Currently only OpenAI is supported.",
|
|
type=str,
|
|
required=True,
|
|
autocomplete=apis.autocomplete,
|
|
)
|
|
@discord.option(name="key", description="The key to set.", type=str, required=True)
|
|
@guild_only()
|
|
async def api(self, ctx: discord.ApplicationContext, api: str, key: str):
|
|
guild = Guild(ctx.guild.id)
|
|
guild.load()
|
|
guild.api_keys[apis.matchingDict[api]] = key
|
|
guild.updateDbData()
|
|
await ctx.respond(f"Set API key for {api} to `secret`.", ephemeral=True)
|
|
|
|
@setup.command(name="premium", description="Set the guild to premium.")
|
|
async def premium(self, ctx: discord.ApplicationContext):
|
|
guild = Guild(ctx.guild.id)
|
|
guild.load()
|
|
if await self.bot.is_owner(ctx.author):
|
|
guild.premium = True
|
|
# also set expiry date in 6 months isofromat
|
|
guild.premium_expiration = datetime.datetime.now() + datetime.timedelta(
|
|
days=180
|
|
)
|
|
guild.updateDbData()
|
|
return await ctx.respond("Set guild to premium.", ephemeral=True)
|
|
if not guild.premium:
|
|
await ctx.respond(
|
|
"You can get your premium subscription at https://www.botator.dev/premium.",
|
|
ephemeral=True,
|
|
)
|
|
else:
|
|
await ctx.respond("This guild is already premium.", ephemeral=True)
|
|
|
|
@setup.command(name="help", description="Show the help page for setup.")
|
|
async def help(self, ctx: discord.ApplicationContext):
|
|
# we eill iterate over all commands the bot has and add them to the embed
|
|
embed = discord.Embed(
|
|
title="Setup Help",
|
|
description="Here is the help page for setup.",
|
|
color=discord.Color.dark_teal(),
|
|
)
|
|
for command in self.setup.walk_commands():
|
|
fieldname = command.name
|
|
fielddescription = command.description
|
|
embed.add_field(name=fieldname, value=fielddescription, inline=False)
|
|
embed.set_footer(text="Made with ❤️ by paillat : https://paillat.dev")
|
|
await ctx.respond(embed=embed, ephemeral=True)
|