From 0e9fd935b265357dbfa3bd09f97b1f9c61d9418e Mon Sep 17 00:00:00 2001
From: The0Mikkel <mikkel.alb@gmail.com>
Date: Tue, 28 May 2024 09:18:55 +0200
Subject: [PATCH] fix(memory): add reply information to chat context

---
 README.md | 4 ++--
 bot.py    | 6 +++++-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 574247b..94ada2d 100644
--- a/README.md
+++ b/README.md
@@ -83,9 +83,9 @@ FROM llama3
 PARAMETER temperature 1
 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 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! 
-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!
 Your name is Assistant.
 """
diff --git a/bot.py b/bot.py
index 476a901..1eda2d6 100644
--- a/bot.py
+++ b/bot.py
@@ -84,7 +84,11 @@ class Bot:
     self.ready = True
 
   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):
     if not self.ready: