persona

robot.py

@@ -10,19 +10,18 @@ import os
 import random
 import shutil
 import copy
-from image import AliyunImage, GeminiImage
 from image.img_manager import ImageGenerationManager
 
 from wcferry import Wcf, WxMsg
 
 from ai_providers.ai_chatgpt import ChatGPT
 from ai_providers.ai_deepseek import DeepSeek
-from ai_providers.ai_gemini import Gemini
 from ai_providers.ai_perplexity import Perplexity
 from function.func_weather import Weather
 from function.func_news import News
 from function.func_summary import MessageSummary  # 导入新的MessageSummary类
 from function.func_reminder import ReminderManager  # 导入ReminderManager类
+from function.func_persona import PersonaManager  # 导入PersonaManager用于人设存储
 from configuration import Config
 from constants import ChatType
 from job_mgmt import Job
@@ -155,31 +154,6 @@ class Robot(Job):
             except Exception as e:
                 self.LOG.error(f"初始化 DeepSeek 模型时出错: {str(e)}")
         
-        # 初始化Gemini
-        if Gemini.value_check(self.config.GEMINI):
-            try:
-                gemini_flash_conf = copy.deepcopy(self.config.GEMINI)
-                flash_model_name = gemini_flash_conf.get("model_flash", Gemini.DEFAULT_MODEL)
-                gemini_flash_conf["model_name"] = flash_model_name
-                self.chat_models[ChatType.GEMINI.value] = Gemini(
-                    gemini_flash_conf,
-                    message_summary_instance=self.message_summary,
-                    bot_wxid=self.wxid
-                )
-                self.LOG.info(f"已加载 Gemini 模型: {flash_model_name}")
-
-                reasoning_model_name = self.config.GEMINI.get("model_reasoning")
-                if reasoning_model_name and reasoning_model_name != flash_model_name:
-                    gemini_reason_conf = copy.deepcopy(self.config.GEMINI)
-                    gemini_reason_conf["model_name"] = reasoning_model_name
-                    self.reasoning_chat_models[ChatType.GEMINI.value] = Gemini(
-                        gemini_reason_conf,
-                        message_summary_instance=self.message_summary,
-                        bot_wxid=self.wxid
-                    )
-                    self.LOG.info(f"已加载 Gemini 推理模型: {reasoning_model_name}")
-            except Exception as e:
-                self.LOG.error(f"初始化 Gemini 模型时出错: {str(e)}")
             
         # 初始化Perplexity
         if Perplexity.value_check(self.config.PERPLEXITY):
@@ -266,6 +240,15 @@ class Robot(Job):
         except Exception as e:
             self.LOG.error(f"初始化提醒管理器失败: {e}", exc_info=True)
         
+        # 初始化人设管理器
+        persona_db_path = getattr(self.message_summary, 'db_path', "data/message_history.db") if getattr(self, 'message_summary', None) else "data/message_history.db"
+        try:
+            self.persona_manager = PersonaManager(persona_db_path)
+            self.LOG.info("人设管理器已初始化。")
+        except Exception as e:
+            self.LOG.error(f"初始化人设管理器失败: {e}", exc_info=True)
+            self.persona_manager = None
+        
     @staticmethod
     def value_check(args: dict) -> bool:
         if args:
@@ -293,12 +276,23 @@ class Robot(Job):
             # 确保context能访问到当前选定的chat模型及特定历史限制
             setattr(ctx, 'chat', self.chat)
             setattr(ctx, 'specific_max_history', specific_limit)
+            persona_text = None
+            if getattr(self, 'persona_manager', None):
+                try:
+                    persona_text = self.persona_manager.get_persona(ctx.get_receiver())
+                except Exception as persona_error:
+                    self.LOG.error(f"获取会话人设失败: {persona_error}", exc_info=True)
+                    persona_text = None
+            setattr(ctx, 'persona', persona_text)
             ctx.reasoning_requested = bool(
                 ctx.text
                 and "想想" in ctx.text
                 and (not ctx.is_group or ctx.is_at_bot)
             )
 
+            if self._handle_persona_command(ctx):
+                return
+
             if ctx.reasoning_requested:
                 self.LOG.info("检测到推理模式触发词，跳过AI路由，直接进入闲聊推理模式。")
                 self._handle_chitchat(ctx, None)
@@ -581,6 +575,12 @@ class Robot(Job):
         if hasattr(self, 'message_summary') and self.message_summary:
             self.LOG.info("正在关闭消息历史数据库...")
             self.message_summary.close_db()
+        if hasattr(self, 'persona_manager') and self.persona_manager:
+            self.LOG.info("正在关闭人设数据库连接...")
+            try:
+                self.persona_manager.close()
+            except Exception as e:
+                self.LOG.error(f"关闭人设数据库时出错: {e}")
         
         self.LOG.info("机器人资源清理完成")
                 
@@ -609,6 +609,126 @@ class Robot(Job):
             
         return None
     
+    def _handle_persona_command(self, ctx: MessageContext) -> bool:
+        """处理 /set 人设命令"""
+        text = (ctx.text or "").strip()
+        if not text or not text.startswith("/"):
+            return False
+
+        parts = text.split(None, 1)
+        command = parts[0].lower()
+        payload = parts[1] if len(parts) > 1 else ""
+
+        at_list = ctx.msg.sender if ctx.is_group else ""
+        scope_label = "本群" if ctx.is_group else "当前会话"
+
+        if command == "/persona":
+            if not getattr(self, "persona_manager", None):
+                ctx.send_text("❌ 人设功能暂不可用。", at_list)
+                return True
+
+            persona_text = getattr(ctx, "persona", None)
+            if persona_text is None:
+                try:
+                    persona_text = self.persona_manager.get_persona(ctx.get_receiver())
+                    setattr(ctx, "persona", persona_text)
+                except Exception as exc:
+                    self.LOG.error(f"查询人设失败: {exc}", exc_info=True)
+                    persona_text = None
+
+            if persona_text:
+                ctx.send_text(f"{scope_label}当前的人设是：\n## 角色\n{persona_text}", at_list)
+            else:
+                ctx.send_text(f"{scope_label}当前没有设置人设，可发送“/set 你的人设描述”来设定。", at_list)
+            return True
+
+        if command != "/set":
+            return False
+
+        if not getattr(self, "persona_manager", None):
+            ctx.send_text("❌ 人设功能暂不可用。", at_list)
+            return True
+
+        persona_body = payload.strip()
+        chat_id = ctx.get_receiver()
+
+        if not persona_body:
+            current = getattr(ctx, "persona", None)
+            if current:
+                ctx.send_text(
+                    f"{scope_label}当前的人设是：\n{current}\n发送“/set clear”可以清空，或重新发送“/set + 新人设”进行更新。\n也可以使用“/persona”随时查看当前人设。",
+                    at_list
+                )
+            else:
+                ctx.send_text("请在 /set 后输入人设描述，例如：/set 你是一个幽默的机器人助手。", at_list)
+            return True
+
+        if persona_body.lower() in {"clear", "reset"}:
+            cleared = self.persona_manager.clear_persona(chat_id)
+            setattr(ctx, "persona", None)
+            if cleared:
+                ctx.send_text(f"✅ 已清空{scope_label}的人设。", at_list)
+            else:
+                ctx.send_text(f"{scope_label}当前没有设置人设。", at_list)
+            return True
+
+        try:
+            self.persona_manager.set_persona(chat_id, persona_body, setter_wxid=ctx.msg.sender)
+            setattr(ctx, "persona", persona_body)
+            preview = persona_body if len(persona_body) <= 120 else persona_body[:120] + "..."
+            ctx.send_text(
+                f"✅ {scope_label}人设设定成功：\n## 角色\n{preview}"
+                f"{'' if len(preview) == len(persona_body) else '...'}\n如需查看完整内容，可发送“/persona”。",
+                at_list
+            )
+        except Exception as exc:
+            self.LOG.error(f"设置人设失败: {exc}", exc_info=True)
+            ctx.send_text("❌ 设置人设时遇到问题，请稍后再试。", at_list)
+        return True
+
+    @staticmethod
+    def _get_model_base_prompt(chat_model):
+        """获取模型默认的系统提示"""
+        if not chat_model:
+            return None
+
+        system_msg = getattr(chat_model, "system_content_msg", None)
+        if isinstance(system_msg, dict):
+            prompt = system_msg.get("content")
+            if prompt:
+                return prompt
+
+        if hasattr(chat_model, "_base_prompt"):
+            prompt = getattr(chat_model, "_base_prompt")
+            if prompt:
+                return prompt
+
+        if hasattr(chat_model, "prompt"):
+            prompt = getattr(chat_model, "prompt")
+            if prompt:
+                return prompt
+
+        return None
+
+    @staticmethod
+    def _merge_prompt_with_persona(prompt, persona):
+        """将人设信息附加到系统提示后"""
+        persona = (persona or "").strip()
+        prompt = (prompt or "").strip() if prompt else ""
+
+        if persona:
+            persona_section = f"## 角色\n{persona}"
+            if prompt:
+                return f"{persona_section}\n\n{prompt}"
+            return persona_section
+
+        return prompt or None
+
+    def _build_system_prompt(self, chat_model, persona=None, override_prompt=None):
+        """生成包含人设的系统提示"""
+        base_prompt = override_prompt if override_prompt is not None else self._get_model_base_prompt(chat_model)
+        return self._merge_prompt_with_persona(base_prompt, persona)
+
     def _get_reasoning_chat_model(self):
         """获取当前聊天模型对应的推理模型实例"""
         model_id = getattr(self, 'current_model_id', None)
@@ -668,7 +788,6 @@ class Robot(Job):
         mapping = {
             ChatType.CHATGPT.value: getattr(self.config, 'CHATGPT', None),
             ChatType.DEEPSEEK.value: getattr(self.config, 'DEEPSEEK', None),
-            ChatType.GEMINI.value: getattr(self.config, 'GEMINI', None),
             ChatType.PERPLEXITY.value: getattr(self.config, 'PERPLEXITY', None),
         }
         return mapping.get(model_id)