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

ppp

Browse Source
This commit is contained in:
artie 2025-02-01 21:25:22 +01:00
parent c8f5d986a1
commit 4d941fe226
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
artemis/cogs/chat.py
View File

@ -107,10 +107,10 @@ class Chat(commands.Cog):
self.lock = asyncio.Lock() self.lock = asyncio.Lock()
def read_prompt(self): def read_prompt(self):
return Path("data/prompt").read_text() return Path("temp/prompt").read_text() if Path("temp/prompt").exists() else ""
def write_prompt(self, prompt: str): def write_prompt(self, prompt: str):
Path("data/prompt").write_text(prompt) Path("temp/prompt").write_text(prompt)
def replace_emojis(self, text: str) -> str: def replace_emojis(self, text: str) -> str:
return EMOJI_RE.sub(lambda match: emoji_map[match.group(0)], text) return EMOJI_RE.sub(lambda match: emoji_map[match.group(0)], text)
@ -124,7 +124,11 @@ 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 = self.prompt + "Following is a user chat message directed at you." + "\n\n" prompt = (
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) >= 15:
@ -152,6 +156,7 @@ class Chat(commands.Cog):
chat_response = self.replace_emojis(chat_response) chat_response = self.replace_emojis(chat_response)
chat_response = self.strip_emojis(chat_response) chat_response = self.strip_emojis(chat_response)
chat_response = re.sub(r"[ ]{2,}", " ", chat_response) chat_response = re.sub(r"[ ]{2,}", " ", chat_response)
chat_response = re.sub(r"[\n]{2,}", "\n", chat_response)
self.add_assistant_memory(chat_response) self.add_assistant_memory(chat_response)
return chat_response return chat_response
@ -182,7 +187,7 @@ class Chat(commands.Cog):
if not content: if not content:
return return
content = message.author.display_name + ": " + content content = f"[USERNAME]: {message.author.display_name}\n[MESSAGE]: {content}"
try: try:
async with message.channel.typing(): async with message.channel.typing():

1
data/prompt
View File

@ -1 +0,0 @@
You're Artemis, a bot hanging out in this Discord server, you're friendly and can answer anything.