Paillat-dev/Botator
1
0
Fork 0
mirror of https://github.com/Paillat-dev/Botator.git synced 2026-01-02 01:06:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

[PROMPTS] Added support for custom prompts

Browse Source
This commit is contained in:
Alexis LEBEL
2023-04-02 16:13:22 +02:00
parent 778330b2e2
commit d4f37afea1
1 changed files with 24 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
code/makeprompt.py
View File

@@ -167,6 +167,29 @@ async def get_data_dict(message):
+ "```", delete_after=60
)
def get_prompt(guild_data, data_dict, message, pretend_to_be):
# support for custom prompts
custom_prompt_path = f"../database/prompts/{guild_data['model']}.txt"
if(os.path.exists(custom_prompt_path)):
prompt_path = custom_prompt_path
else:
prompt_path = f"./prompts/{guild_data['model']}.txt"
# open the prompt file for the selected model with utf-8 encoding for emojis
with open(prompt_path, "r", encoding="utf-8") as f:
prompt = f.read()
# replace the variables in the prompt with the actual values
prompt = (
prompt.replace("[prompt-prefix]", data_dict['prompt_prefix'])
.replace("[server-name]", message.guild.name)
.replace("[channel-name]", message.channel.name)
.replace(
"[date-and-time]", datetime.datetime.utcnow().strftime("%d/%m/%Y %H:%M:%S")
)
.replace("[pretend-to-be]", pretend_to_be)
)
f.close()
return prompt
async def chat_process(self, message):
"""This function processes the message and sends the prompt to the API
@@ -228,7 +251,6 @@ async def chat_process(self, message):
if (await need_ignore_message(self.bot, data_dict, message, guild_data, original_message, channels)):
return
print("prompt handler")
try:
await message.channel.trigger_typing()
@@ -262,20 +284,7 @@ async def chat_process(self, message):
"pretend_enabled"] else ""
prompt_prefix = "" if data_dict["prompt_prefix"] == None else data_dict["prompt_prefix"]
# open the prompt file for the selected model with utf-8 encoding for emojis
with open(f"./prompts/{guild_data['model']}.txt", "r", encoding="utf-8") as f:
prompt = f.read()
# replace the variables in the prompt with the actual values
prompt = (
prompt.replace("[prompt-prefix]", prompt_prefix)
.replace("[server-name]", message.guild.name)
.replace("[channel-name]", message.channel.name)
.replace(
"[date-and-time]", datetime.datetime.utcnow().strftime("%d/%m/%Y %H:%M:%S")
)
.replace("[pretend-to-be]", pretend_to_be)
)
f.close()
prompt = get_prompt(guild_data, data_dict, message, pretend_to_be)
prompt_handlers = {
"gpt-3.5-turbo": gpt_prompt,