From bc92aaa644d349f50dbf5b1ef95fa86c9710e687 Mon Sep 17 00:00:00 2001
From: csd4ni3l <v3xq21w7@duck.com>
Date: Fri, 12 Dec 2025 20:16:15 +0100
Subject: [PATCH] Update default model in .env.example, add AI context from
 conversation history

---
 .env.example | 2 +-
 app.py       | 9 +++++----
 constants.py | 2 ++
 3 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/.env.example b/.env.example
index ccebc92..a507ef7 100644
--- a/.env.example
+++ b/.env.example
@@ -1,5 +1,5 @@
 OPENROUTER_API_KEY=API_KEY
-OPENROUTER_MODEL=qwen/qwen3-32b
+OPENROUTER_MODEL=openai/gpt-5-mini
 OPENROUTER_URL=https://ai.hackclub.com/proxy/v1
 BOT_TOKEN=BOT_TOKEN
 APP_TOKEN=APP_TOKEN
\ No newline at end of file
diff --git a/app.py b/app.py
index 534b07e..b6fc372 100644
--- a/app.py
+++ b/app.py
@@ -220,12 +220,13 @@ def message_handler(event, say, client, message):
     message_ts = message["ts"]
 
     if event.get("channel_type") == "im" and "bot_id" not in event:
+        messages = [{"role": "system", "content": AI_SYSTEM_PROMPT}]
+        messages.extend([{"role": "assistant" if "bot_id" in msg else "user", "content": msg.get("text", "")} for msg in client.conversations_history(channel=channel_id, limit=AI_CONTEXT_MSG_LIMIT).get("messages", [])])
+        messages.append({"role": "user", "content": message_text})
+
         response = openrouter_client.chat.send(
             model=os.getenv("OPENROUTER_MODEL"),
-            messages=[
-                {"role": "assistant", "content": AI_SYSTEM_PROMPT},
-                {"role": "user", "content": message_text}
-            ],
+            messages=messages,
             stream=False,
         )
 
diff --git a/constants.py b/constants.py
index 3d75f37..b5ac504 100644
--- a/constants.py
+++ b/constants.py
@@ -11,6 +11,8 @@ MEOW_PHRASES = [
     "cat"
 ]
 
+AI_CONTEXT_MSG_LIMIT = 10
+
 CAT_EMOJI = "cat"
 DUCK_EMOJI = "duck"