feat(AI功能): 添加AI功能开关配置及处理逻辑

在config.py和r.py中添加AI功能开关配置，并在bot_client.py中根据开关状态调整消息处理逻辑。当AI功能关闭时，使用简单回复代替AI聊天功能。

Author: bestzyq

bot_client.py

@@ -20,7 +20,7 @@
 from handlers.bus import query_bus
 from handlers.classroom import query_empty_classroom
 
-from config import APPID, SECRET
+from config import APPID, SECRET, AI_GROUP_ENABLED, AI_DIRECT_ENABLED
 
 _log = botpy.logging.get_logger()
 
@@ -73,11 +73,14 @@ async def on_ready(self):
 
     async def on_c2c_message_create(self, message: DirectMessage):
         """私聊消息处理"""
-        try:
-            if await direct_chat_with_clawdbot(api=self.api, message=message):
-                return
-        except Exception as e:
-            # 出错时回退到简单回复
+        if AI_DIRECT_ENABLED:
+            try:
+                if await direct_chat_with_clawdbot(api=self.api, message=message):
+                    return
+            except Exception as e:
+                content = message.content
+                await message.reply(content=content)
+        else:
             content = message.content
             await message.reply(content=content)
 
@@ -87,21 +90,30 @@ async def on_group_at_message_create(self, message: GroupMessage):
             if await handler(api=self.api, message=message):
                 return
 
-        # 如果没有处理器处理，尝试使用 group_chat_with_clawdbot
-        try:
-            if await group_chat_with_clawdbot(api=self.api, message=message):
-                return
-        except Exception as e:
-            # 出错时回退到群组查找
+        if AI_GROUP_ENABLED:
+            try:
+                if await group_chat_with_clawdbot(api=self.api, message=message):
+                    return
+            except Exception as e:
+                user_input = message.content.strip().replace("群", "")
+                if user_input:
+                    try:
+                        await internal_find_group(api=self.api, message=message, search_key=user_input)
+                        return
+                    except Exception as find_error:
+                        await message.reply(content=f"调用出错: {str(find_error)}")
+                else:
+                    await message.reply(content=f"调用出错: {str(e)}")
+        else:
             user_input = message.content.strip().replace("群", "")
             if user_input:
                 try:
                     await internal_find_group(api=self.api, message=message, search_key=user_input)
                     return
-                except Exception as find_error:
-                    await message.reply(content=f"调用出错: {str(find_error)}")
+                except Exception as e:
+                    await message.reply(content=f"调用出错: {str(e)}")
             else:
-                await message.reply(content=f"调用出错: {str(e)}")
+                await message.reply(content=f"不明白你在说什么哦(๑• . •๑)")
 
     async def on_group_add_robot(self, message: GroupManageEvent):
         """机器人被添加到群组事件"""

config.py

@@ -53,4 +53,8 @@
 
 # 三角洲行动API配置
 DELTAFORCE_API_TOKEN = r.deltaforce_api_token
-CLASS_API_KEY = r.class_api_token
+CLASS_API_KEY = r.class_api_token
+
+# AI功能开关
+AI_GROUP_ENABLED = r.ai_group_enabled
+AI_DIRECT_ENABLED = r.ai_direct_enabled

r.py

@@ -124,4 +124,8 @@ def update_env_variable(key, value):
 # 查询空教室API配置
 class_api_token = os.getenv("CLASS_API_KEY")
 if class_api_token is None:
-    raise Exception('Missing "CLASS_API_KEY" environment variable for CLASS_API_KEY')
+    raise Exception('Missing "CLASS_API_KEY" environment variable for CLASS_API_KEY')
+
+# AI功能开关配置
+ai_group_enabled = os.getenv("AI_GROUP_ENABLED", "false").lower() == "true"
+ai_direct_enabled = os.getenv("AI_DIRECT_ENABLED", "false").lower() == "true"