artie/artemis
1
0
Fork 0
mirror of https://github.com/artiemis/artemis.git synced 2026-02-14 08:31:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

dum

Browse Source
This commit is contained in:
artie 2025-02-02 15:59:15 +01:00
parent 4d941fe226
commit af202e8014
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
artemis/cogs/chat.py
View File

@ -124,18 +124,17 @@ class Chat(commands.Cog):
return content return content
def add_memory(self, role: str, message: str): def add_memory(self, role: str, message: str):
prompt = ( prompt = self.prompt + "\n\n"
self.prompt
+ "The following is a user chat message directed at you, the format will be the same for subsequent messages, respond with only the message content, without specyfing actions."
+ "\n\n"
)
if len(self.memory) == 0: if len(self.memory) == 0:
message = prompt + message message = prompt + message
if len(self.memory) >= 15: if len(self.memory) >= 20:
del self.memory[0] del self.memory[0]
del self.memory[0] del self.memory[0]
self.memory[0] = {"role": "user", "content": prompt + self.memory[0]["content"]} self.memory[0] = {
self.memory.append({"role": role, "content": message}) "role": "user",
"content": (prompt + self.memory[0]["content"]).strip(),
}
self.memory.append({"role": role, "content": message.strip()})
def add_user_memory(self, message: str): def add_user_memory(self, message: str):
self.add_memory("user", message) self.add_memory("user", message)
@ -187,8 +186,6 @@ class Chat(commands.Cog):
if not content: if not content:
return return
content = f"[USERNAME]: {message.author.display_name}\n[MESSAGE]: {content}"
try: try:
async with message.channel.typing(): async with message.channel.typing():
async with self.lock: async with self.lock: