Update code.py

This commit is contained in:
Paillat
2022-11-29 16:07:52 +01:00
parent 3ff3546b7b
commit 868649b075

View File

@@ -96,7 +96,7 @@ async def disable(ctx):
@discord.commands.option(name="presence_penalty", description="The presence penalty", required=False)
#set the fifth argument: prompt_size, with a default value of 5
@discord.commands.option(name="prompt_size", description="The number of messages to use as a prompt", required=False)
async def advanced(ctx, max_tokens=12345, temperature=123.4, frequency_penalty=0.444555, presence_penalty=0.444555, prompt_size=23456):
async def advanced(ctx, max_tokens=None, temperature=None, frequency_penalty=None, presence_penalty=None, prompt_size=None):
#check if the guild is in the database
debug(f"The user {ctx.author} ran the advanced command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}")
c.execute("SELECT * FROM data WHERE guild_id = ?", (ctx.guild.id,))
@@ -107,15 +107,15 @@ async def advanced(ctx, max_tokens=12345, temperature=123.4, frequency_penalty=0
print(c.fetchone())
current_settings = [c.fetchone()[4], c.fetchone()[5], c.fetchone()[6], c.fetchone()[7], c.fetchone()[9]]
#get the new settings
if max_tokens == 12345:
if max_tokens == None:
max_tokens = current_settings[1]
if temperature == 123.4:
if temperature == None:
temperature = current_settings[2]
if frequency_penalty == 0.444555:
if frequency_penalty == None:
frequency_penalty = current_settings[3]
if presence_penalty == 0.444555:
if presence_penalty == None:
presence_penalty = current_settings[4]
if prompt_size == 23456:
if prompt_size == None:
prompt_size = current_settings[5]
#update the settings
c.execute("UPDATE data SET max_tokens = ?, temperature = ?, frequency_penalty = ?, presence_penalty = ?, prompt_size = ? WHERE guild_id = ?", (max_tokens, temperature, frequency_penalty, presence_penalty, prompt_size, ctx.guild.id))