From 53b66499ee8aa6590bf8505bd987a18733a67662 Mon Sep 17 00:00:00 2001
From: F4ria <f4riacool@gmail.com>
Date: Mon, 15 Apr 2024 17:58:50 +0800
Subject: [PATCH] feat: if the message starts with "new ", clear the player's
 history and start a new chat

---
 handlers/claude.py | 6 ++++++
 handlers/gemini.py | 6 ++++++
 handlers/yi.py     | 3 +++
 3 files changed, 15 insertions(+)

diff --git a/handlers/claude.py b/handlers/claude.py
index 8ff60f5..461fc9e 100644
--- a/handlers/claude.py
+++ b/handlers/claude.py
@@ -45,6 +45,9 @@ def claude_handler(message: Message, bot: TeleBot) -> None:
         )
         player_message.clear()
         return
+    if m[:4].lower() == "new ":
+        m = m[4:].strip()
+        player_message.clear()
 
     who = "Claude"
     # show something, make it more responsible
@@ -103,6 +106,9 @@ def claude_pro_handler(message: Message, bot: TeleBot) -> None:
         )
         player_message.clear()
         return
+    if m[:4].lower() == "new ":
+        m = m[4:].strip()
+        player_message.clear()
 
     who = "Claude Pro"
     # show something, make it more responsible
diff --git a/handlers/gemini.py b/handlers/gemini.py
index de6882c..0d395ec 100644
--- a/handlers/gemini.py
+++ b/handlers/gemini.py
@@ -68,6 +68,9 @@ def gemini_handler(message: Message, bot: TeleBot) -> None:
         )
         player.history.clear()
         return
+    if m[:4].lower() == "new ":
+        m = m[4:].strip()
+        player.history.clear()
 
     who = "Gemini"
     # show something, make it more responsible
@@ -114,6 +117,9 @@ def gemini_pro_handler(message: Message, bot: TeleBot) -> None:
         )
         player.history.clear()
         return
+    if m[:4].lower() == "new ":
+        m = m[4:].strip()
+        player.history.clear()
 
     who = "Gemini Pro"
     # show something, make it more responsible
diff --git a/handlers/yi.py b/handlers/yi.py
index 6442b21..3bd234b 100644
--- a/handlers/yi.py
+++ b/handlers/yi.py
@@ -42,6 +42,9 @@ def yi_handler(message: Message, bot: TeleBot) -> None:
         )
         player_message.clear()
         return
+    if m[:4].lower() == "new ":
+        m = m[4:].strip()
+        player_message.clear()
 
     who = "Yi"
     # show something, make it more responsible