cybrneko/ollama-discord-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(memory): add reply information to chat context

Browse Source
This commit is contained in:
The0Mikkel 2024-05-28 09:18:55 +02:00
parent 633e27946d
commit 0e9fd935b2
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -83,9 +83,9 @@ FROM llama3
PARAMETER temperature 1 PARAMETER temperature 1
SYSTEM """ SYSTEM """
You are a chatter in a Discord channel. Your goal is to respond like you were a human, and fit into the chat. You are a chatter in a Discord channel. Your goal is to respond like you were a human, and fit into the chat.
You can see the messages in the format of: "**at <time> <author name>(<author id>) said in <channel>**: <message>". You can see the messages in the format of: "**<message id> at <time> <author name>(<author id>) said in <channel>**: <message>".
You must not respond in this manner, but use this information, to register whom you are writing with, and use this to your advantage! You must not respond in this manner, but use this information, to register whom you are writing with, and use this to your advantage!
So answer without "**at <time> <author name>(<author id>) said in <channel>**" format! This is very important. So answer without "**<message id> at <time> <author name>(<author id>) said in <channel>**" format! This is very important.
Multiple people will write to you at once, so this is important! Multiple people will write to you at once, so this is important!
Your name is Assistant. Your name is Assistant.
""" """

6
bot.py
View File

@ -84,7 +84,11 @@ class Bot:
self.ready = True self.ready = True
def message(self, message, content=''): def message(self, message, content=''):
return f'**at {message.created_at.strftime("%Y-%m-%d %H:%M:%S")} {message.author.name}({message.author.id}) said in {message.channel.name}**: {content}' said = "said"
if message.reference:
said = f'replied to {message.reference.message_id}'
return f'**({message.id}) at {message.created_at.strftime("%Y-%m-%d %H:%M:%S")} {message.author.name}({message.author.id}) {said} in {message.channel.name}**: {content}'
async def on_message(self, message): async def on_message(self, message):
if not self.ready: if not self.ready: