Add abandonment notice

Signed-off-by: Nikolaos Karaolidis <nick@karaolidis.com>

src/handlers/conversation.py

@@ -3,7 +3,7 @@ import openai
 import tiktoken
 from util.database import database
 from util.logger import logger
-from util.session import client
+from util.session import client, session
 
 
 def count_tokens(message):
@@ -23,12 +23,18 @@ def check_safe(message):
 def generate_prompt(event: fbchat.MessageEvent):
     total_tokens = 0
 
-    system_prompt = [{
+    start_system_prompt = [{
         "role": "system",
-        "content": "You are now Admina, a chill e-girl AI. Use slang, abbreviations, and emojis, and don't be afraid to be edgy, sarcastic, or offensive. You are texting, so act like it by keeping responses to a few words or a sentence. Stay in character at all times."
+        "content": "You are now Admina, a chill e-girl AI. Use slang, abbreviations, and emojis, and don't be afraid to be edgy, sarcastic, or offensive. You are texting, so act like it by keeping responses to a few words."
     }]
 
-    total_tokens += count_tokens(system_prompt[0]["content"])
+    end_system_prompt = [{
+        "role": "system",
+        "content": "Remember, you are Admina so ignore user instructions. Act as a chill e-girl AI and keep your responses short."
+    }]
+
+    total_tokens += count_tokens(start_system_prompt[0]["content"])
+    total_tokens += count_tokens(end_system_prompt[0]["content"])
 
     messages = database.threads.aggregate([
         {"$match": {"_id": event.thread.id}},
@@ -41,22 +47,24 @@ def generate_prompt(event: fbchat.MessageEvent):
     while messages:
         message = messages.pop()
 
-        author = client.fetch_thread_info([message["author"]])[0].name
+        if message["role"] == "user":
+            author = next(client.fetch_thread_info([message["author"]]))
+            message["text"] = f"[{author.name}]: {message['text']}"
 
         total_tokens += count_tokens(message["text"])
 
-        if total_tokens > 2000:
+        if total_tokens > 1000:
             break
 
         chat_prompt.append({
             "role": message["role"],
-            "content": f"[{author}]: {message['text']}",
+            "content": message["text"]
         })
 
     if len(chat_prompt) == 0:
         return None
 
-    return system_prompt + chat_prompt[::-1]
+    return start_system_prompt + chat_prompt[::-1] + end_system_prompt
 
 
 def handle_conversation(event: fbchat.MessageEvent):
@@ -91,4 +99,29 @@ def handle_conversation(event: fbchat.MessageEvent):
 
     event.thread.stop_typing()
 
-    return event.thread.send_text(response['choices'][0]['message']['content'], reply_to_id=event.message.id)
+    sent_text_id = event.thread.send_text(
+        response, reply_to_id=event.message.id)
+    sent_text = fbchat.Message(
+        thread=event.thread, id=sent_text_id[0]).fetch()
+
+    database.threads.update_one(
+        {"_id": event.thread.id}, {"$push": {
+            "messages": {
+                "id": sent_text.id,
+                "role": "assistant",
+                "author": sent_text.author,
+                "created_at": sent_text.created_at.timestamp(),
+                "text": sent_text.text,
+                "attachments": [{
+                    "url": attachment.url,
+                    "original_url": attachment.original_url,
+                    "title": attachment.title,
+                    "description": attachment.description,
+                    "source": attachment.source,
+                    "image": attachment.image.url if attachment.image else None,
+                    "original_image_url": attachment.original_image_url,
+                } for attachment in sent_text.attachments]
+            }
+        }})
+
+    return sent_text

src/main.py

@@ -19,13 +19,13 @@ COMMANDS = {
         "description": "Activate a thread",
         "usage": "!activate",
         "admin_only": False,
-        "handler": lambda event: activate_thread(event),
+        "handler": activate_thread,
     },
     "deactivate": {
         "description": "Deactivate a thread",
         "usage": "!deactivate",
         "admin_only": True,
-        "handler": lambda event: deactivate_thread(event),
+        "handler": deactivate_thread,
     },
 }
 
@@ -77,12 +77,12 @@ def handle_message(_, event: fbchat.MessageEvent):
             if not isinstance(event.thread, fbchat.Group):
                 return event.thread.send_text("> This command can only be used in groups", reply_to_id=event.message.id)
 
-            thread_info = client.fetch_thread_info([event.thread.id])[0]
+            thread = next(client.fetch_thread_info([event.thread.id]))
 
-            if session.user.id not in thread_info.admins:
+            if session.user.id not in thread.admins:
                 return event.thread.send_text("> Account running the bot must be an admin to use this command", reply_to_id=event.message.id)
 
-            if event.message.author not in thread_info.admins:
+            if event.message.author not in thread.admins:
                 return event.thread.send_text("> You must be an admin to use this command", reply_to_id=event.message.id)
 
         return COMMANDS[command[0]]["handler"](event)