mirror of
https://github.com/Zippland/Bubbles.git
synced 2026-01-19 09:41:17 +08:00
618 lines
27 KiB
Python
618 lines
27 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
import logging
|
||
import re
|
||
import time
|
||
import xml.etree.ElementTree as ET
|
||
from queue import Empty
|
||
from threading import Thread
|
||
import os
|
||
import random
|
||
import shutil
|
||
from ai_providers.ai_zhipu import ZhiPu
|
||
from image import CogView, AliyunImage, GeminiImage
|
||
from image.img_manager import ImageGenerationManager
|
||
|
||
from wcferry import Wcf, WxMsg
|
||
|
||
from ai_providers.ai_bard import BardAssistant
|
||
from ai_providers.ai_chatglm import ChatGLM
|
||
from ai_providers.ai_ollama import Ollama
|
||
from ai_providers.ai_chatgpt import ChatGPT
|
||
from ai_providers.ai_deepseek import DeepSeek
|
||
from ai_providers.ai_perplexity import Perplexity
|
||
from function.func_weather import Weather
|
||
from function.func_news import News
|
||
from ai_providers.ai_tigerbot import TigerBot
|
||
from ai_providers.ai_xinghuo_web import XinghuoWeb
|
||
from function.func_duel import start_duel, get_rank_list, get_player_stats, change_player_name, DuelManager, attempt_sneak_attack
|
||
from function.func_summary import MessageSummary # 导入新的MessageSummary类
|
||
from function.func_reminder import ReminderManager # 导入ReminderManager类
|
||
from configuration import Config
|
||
from constants import ChatType
|
||
from job_mgmt import Job
|
||
from function.func_xml_process import XmlProcessor
|
||
from function.func_goblin_gift import GoblinGiftManager
|
||
|
||
# 导入命令路由系统
|
||
from commands.context import MessageContext
|
||
from commands.router import CommandRouter
|
||
from commands.registry import COMMANDS, get_commands_info
|
||
from commands.handlers import handle_chitchat # 导入闲聊处理函数
|
||
|
||
__version__ = "39.2.4.0"
|
||
|
||
|
||
class Robot(Job):
|
||
"""个性化自己的机器人
|
||
"""
|
||
|
||
def __init__(self, config: Config, wcf: Wcf, chat_type: int) -> None:
|
||
# 调用父类构造函数
|
||
super().__init__()
|
||
|
||
self.wcf = wcf
|
||
self.config = config
|
||
self.LOG = logging.getLogger("Robot")
|
||
self.wxid = self.wcf.get_self_wxid()
|
||
self.allContacts = self.getAllContacts()
|
||
self._msg_timestamps = []
|
||
# 创建决斗管理器
|
||
self.duel_manager = DuelManager(self.sendDuelMsg)
|
||
|
||
# 初始化消息总结功能
|
||
self.message_summary = MessageSummary(max_history=200)
|
||
|
||
# 初始化XML处理器
|
||
self.xml_processor = XmlProcessor(self.LOG)
|
||
|
||
# 初始化所有可能需要的AI模型实例
|
||
self.chat_models = {}
|
||
self.LOG.info("开始初始化各种AI模型...")
|
||
|
||
# 初始化TigerBot
|
||
if TigerBot.value_check(self.config.TIGERBOT):
|
||
self.chat_models[ChatType.TIGER_BOT.value] = TigerBot(self.config.TIGERBOT)
|
||
self.LOG.info(f"已加载 TigerBot 模型")
|
||
|
||
# 初始化ChatGPT
|
||
if ChatGPT.value_check(self.config.CHATGPT):
|
||
self.chat_models[ChatType.CHATGPT.value] = ChatGPT(self.config.CHATGPT)
|
||
self.LOG.info(f"已加载 ChatGPT 模型")
|
||
|
||
# 初始化讯飞星火
|
||
if XinghuoWeb.value_check(self.config.XINGHUO_WEB):
|
||
self.chat_models[ChatType.XINGHUO_WEB.value] = XinghuoWeb(self.config.XINGHUO_WEB)
|
||
self.LOG.info(f"已加载 讯飞星火 模型")
|
||
|
||
# 初始化ChatGLM
|
||
if ChatGLM.value_check(self.config.CHATGLM):
|
||
try:
|
||
# 检查key是否有实际内容而不只是存在
|
||
if self.config.CHATGLM.get('key') and self.config.CHATGLM.get('key').strip():
|
||
self.chat_models[ChatType.CHATGLM.value] = ChatGLM(self.config.CHATGLM)
|
||
self.LOG.info(f"已加载 ChatGLM 模型")
|
||
else:
|
||
self.LOG.warning("ChatGLM 配置中缺少有效的API密钥，跳过初始化")
|
||
except Exception as e:
|
||
self.LOG.error(f"初始化 ChatGLM 模型时出错: {str(e)}")
|
||
|
||
# 初始化BardAssistant
|
||
if BardAssistant.value_check(self.config.BardAssistant):
|
||
self.chat_models[ChatType.BardAssistant.value] = BardAssistant(self.config.BardAssistant)
|
||
self.LOG.info(f"已加载 BardAssistant 模型")
|
||
|
||
# 初始化ZhiPu
|
||
if ZhiPu.value_check(self.config.ZhiPu):
|
||
self.chat_models[ChatType.ZhiPu.value] = ZhiPu(self.config.ZhiPu)
|
||
self.LOG.info(f"已加载 智谱 模型")
|
||
|
||
# 初始化Ollama
|
||
if Ollama.value_check(self.config.OLLAMA):
|
||
self.chat_models[ChatType.OLLAMA.value] = Ollama(self.config.OLLAMA)
|
||
self.LOG.info(f"已加载 Ollama 模型")
|
||
|
||
# 初始化DeepSeek
|
||
if DeepSeek.value_check(self.config.DEEPSEEK):
|
||
self.chat_models[ChatType.DEEPSEEK.value] = DeepSeek(self.config.DEEPSEEK)
|
||
self.LOG.info(f"已加载 DeepSeek 模型")
|
||
|
||
# 初始化Perplexity
|
||
if Perplexity.value_check(self.config.PERPLEXITY):
|
||
self.chat_models[ChatType.PERPLEXITY.value] = Perplexity(self.config.PERPLEXITY)
|
||
self.perplexity = self.chat_models[ChatType.PERPLEXITY.value] # 单独保存一个引用用于特殊处理
|
||
self.LOG.info(f"已加载 Perplexity 模型")
|
||
|
||
# 根据chat_type参数选择默认模型
|
||
if chat_type > 0 and chat_type in self.chat_models:
|
||
self.chat = self.chat_models[chat_type]
|
||
self.default_model_id = chat_type
|
||
else:
|
||
# 如果没有指定chat_type或指定的模型不可用，尝试使用配置文件中指定的默认模型
|
||
self.default_model_id = self.config.GROUP_MODELS.get('default', 0)
|
||
if self.default_model_id in self.chat_models:
|
||
self.chat = self.chat_models[self.default_model_id]
|
||
elif self.chat_models: # 如果有任何可用模型，使用第一个
|
||
self.default_model_id = list(self.chat_models.keys())[0]
|
||
self.chat = self.chat_models[self.default_model_id]
|
||
else:
|
||
self.LOG.warning("未配置任何可用的模型")
|
||
self.chat = None
|
||
self.default_model_id = 0
|
||
|
||
self.LOG.info(f"默认模型: {self.chat}，模型ID: {self.default_model_id}")
|
||
|
||
# 显示群组-模型映射信息
|
||
if hasattr(self.config, 'GROUP_MODELS'):
|
||
# 显示群聊映射信息
|
||
if self.config.GROUP_MODELS.get('mapping'):
|
||
self.LOG.info("群聊-模型映射配置:")
|
||
for mapping in self.config.GROUP_MODELS.get('mapping', []):
|
||
room_id = mapping.get('room_id', '')
|
||
model_id = mapping.get('model', 0)
|
||
if room_id and model_id in self.chat_models:
|
||
model_name = self.chat_models[model_id].__class__.__name__
|
||
self.LOG.info(f" 群聊 {room_id} -> 模型 {model_name}(ID:{model_id})")
|
||
elif room_id:
|
||
self.LOG.warning(f" 群聊 {room_id} 配置的模型ID {model_id} 不可用")
|
||
|
||
# 显示私聊映射信息
|
||
if self.config.GROUP_MODELS.get('private_mapping'):
|
||
self.LOG.info("私聊-模型映射配置:")
|
||
for mapping in self.config.GROUP_MODELS.get('private_mapping', []):
|
||
wxid = mapping.get('wxid', '')
|
||
model_id = mapping.get('model', 0)
|
||
if wxid and model_id in self.chat_models:
|
||
model_name = self.chat_models[model_id].__class__.__name__
|
||
contact_name = self.allContacts.get(wxid, wxid)
|
||
self.LOG.info(f" 私聊用户 {contact_name}({wxid}) -> 模型 {model_name}(ID:{model_id})")
|
||
elif wxid:
|
||
self.LOG.warning(f" 私聊用户 {wxid} 配置的模型ID {model_id} 不可用")
|
||
|
||
# 初始化图像生成管理器
|
||
self.image_manager = ImageGenerationManager(self.config, self.wcf, self.LOG, self.sendTextMsg)
|
||
|
||
# 初始化古灵阁妖精馈赠管理器
|
||
self.goblin_gift_manager = GoblinGiftManager(self.config, self.wcf, self.LOG, self.sendTextMsg)
|
||
|
||
# 初始化命令路由器
|
||
self.command_router = CommandRouter(COMMANDS, robot_instance=self)
|
||
self.LOG.info(f"命令路由系统初始化完成，共加载 {len(COMMANDS)} 条命令")
|
||
|
||
# 初始化提醒管理器
|
||
try:
|
||
# 使用与MessageSummary相同的数据库路径
|
||
db_path = getattr(self.message_summary, 'db_path', "data/message_history.db")
|
||
self.reminder_manager = ReminderManager(self, db_path)
|
||
self.LOG.info("提醒管理器已初始化，与消息历史使用相同数据库。")
|
||
except Exception as e:
|
||
self.LOG.error(f"初始化提醒管理器失败: {e}", exc_info=True)
|
||
|
||
# 输出命令列表信息，便于调试
|
||
# self.LOG.debug(get_commands_info()) # 如果需要在日志中输出所有命令信息，取消本行注释
|
||
|
||
@staticmethod
|
||
def value_check(args: dict) -> bool:
|
||
if args:
|
||
return all(value is not None for key, value in args.items() if key != 'proxy')
|
||
return False
|
||
|
||
def processMsg(self, msg: WxMsg) -> None:
|
||
"""
|
||
处理收到的微信消息
|
||
:param msg: 微信消息对象
|
||
"""
|
||
try:
|
||
# 1. 使用MessageSummary记录消息(保持不变)
|
||
self.message_summary.process_message_from_wxmsg(msg, self.wcf, self.allContacts, self.wxid)
|
||
|
||
# 2. 根据消息来源选择使用的AI模型
|
||
self._select_model_for_message(msg)
|
||
|
||
# 3. 预处理消息，生成MessageContext
|
||
ctx = self.preprocess(msg)
|
||
# 确保context能访问到当前选定的chat模型
|
||
setattr(ctx, 'chat', self.chat)
|
||
|
||
# 4. 使用命令路由器分发处理消息
|
||
handled = self.command_router.dispatch(ctx)
|
||
|
||
# 5. 如果没有命令处理器处理，则进行特殊逻辑处理
|
||
if not handled:
|
||
# 5.1 好友请求自动处理
|
||
if msg.type == 37: # 好友请求
|
||
self.autoAcceptFriendRequest(msg)
|
||
return
|
||
|
||
# 5.2 系统消息处理
|
||
elif msg.type == 10000:
|
||
# 5.2.1 处理新成员入群
|
||
if "加入了群聊" in msg.content and msg.from_group():
|
||
new_member_match = re.search(r'"(.+?)"邀请"(.+?)"加入了群聊', msg.content)
|
||
if new_member_match:
|
||
inviter = new_member_match.group(1) # 邀请人
|
||
new_member = new_member_match.group(2) # 新成员
|
||
# 使用配置文件中的欢迎语，支持变量替换
|
||
welcome_msg = self.config.WELCOME_MSG.format(new_member=new_member, inviter=inviter)
|
||
self.sendTextMsg(welcome_msg, msg.roomid)
|
||
self.LOG.info(f"已发送欢迎消息给新成员 {new_member} 在群 {msg.roomid}")
|
||
return
|
||
# 5.2.2 处理新好友添加
|
||
elif "你已添加了" in msg.content:
|
||
self.sayHiToNewFriend(msg)
|
||
return
|
||
|
||
# 5.3 群聊消息，且配置了响应该群
|
||
if msg.from_group() and msg.roomid in self.config.GROUPS:
|
||
# 如果在群里被@了，但命令路由器没有处理，则进行闲聊
|
||
if msg.is_at(self.wxid):
|
||
# 调用handle_chitchat函数处理闲聊
|
||
handle_chitchat(ctx, None)
|
||
else:
|
||
pass
|
||
|
||
# 5.4 私聊消息，未被命令处理，进行闲聊
|
||
elif not msg.from_group() and not msg.from_self():
|
||
# 检查是否是文本消息(type 1)或者是包含用户输入的类型49消息
|
||
if msg.type == 1 or (msg.type == 49 and ctx.text):
|
||
self.LOG.info(f"准备回复私聊消息: 类型={msg.type}, 文本内容='{ctx.text}'")
|
||
# 调用handle_chitchat函数处理闲聊
|
||
handle_chitchat(ctx, None)
|
||
|
||
except Exception as e:
|
||
self.LOG.error(f"处理消息时发生错误: {str(e)}", exc_info=True)
|
||
|
||
def enableRecvMsg(self) -> None:
|
||
self.wcf.enable_recv_msg(self.onMsg)
|
||
|
||
def enableReceivingMsg(self) -> None:
|
||
def innerProcessMsg(wcf: Wcf):
|
||
while wcf.is_receiving_msg():
|
||
try:
|
||
msg = wcf.get_msg()
|
||
self.LOG.info(msg)
|
||
self.processMsg(msg)
|
||
except Empty:
|
||
continue # Empty message
|
||
except Exception as e:
|
||
self.LOG.error(f"Receiving message error: {e}")
|
||
|
||
self.wcf.enable_receiving_msg()
|
||
Thread(target=innerProcessMsg, name="GetMessage", args=(self.wcf,), daemon=True).start()
|
||
|
||
def sendTextMsg(self, msg: str, receiver: str, at_list: str = "") -> None:
|
||
""" 发送消息
|
||
:param msg: 消息字符串
|
||
:param receiver: 接收人wxid或者群id
|
||
:param at_list: 要@的wxid, @所有人的wxid为：notify@all
|
||
"""
|
||
# 随机延迟0.3-1.3秒，并且一分钟内发送限制
|
||
time.sleep(float(str(time.time()).split('.')[-1][-2:]) / 100.0 + 0.3)
|
||
now = time.time()
|
||
if self.config.SEND_RATE_LIMIT > 0:
|
||
# 清除超过1分钟的记录
|
||
self._msg_timestamps = [t for t in self._msg_timestamps if now - t < 60]
|
||
if len(self._msg_timestamps) >= self.config.SEND_RATE_LIMIT:
|
||
self.LOG.warning(f"发送消息过快，已达到每分钟{self.config.SEND_RATE_LIMIT}条上限。")
|
||
return
|
||
self._msg_timestamps.append(now)
|
||
|
||
# msg 中需要有 @ 名单中一样数量的 @
|
||
ats = ""
|
||
if at_list:
|
||
if at_list == "notify@all": # @所有人
|
||
ats = " @所有人"
|
||
else:
|
||
wxids = at_list.split(",")
|
||
for wxid in wxids:
|
||
# 根据 wxid 查找群昵称
|
||
ats += f" @{self.wcf.get_alias_in_chatroom(wxid, receiver)}"
|
||
|
||
# {msg}{ats} 表示要发送的消息内容后面紧跟@，例如 北京天气情况为：xxx @张三
|
||
if ats == "":
|
||
self.LOG.info(f"To {receiver}: {msg}")
|
||
self.wcf.send_text(f"{msg}", receiver, at_list)
|
||
else:
|
||
self.LOG.info(f"To {receiver}:\n{ats}\n{msg}")
|
||
self.wcf.send_text(f"{ats}\n\n{msg}", receiver, at_list)
|
||
|
||
def getAllContacts(self) -> dict:
|
||
"""
|
||
获取联系人（包括好友、公众号、服务号、群成员……）
|
||
格式: {"wxid": "NickName"}
|
||
"""
|
||
contacts = self.wcf.query_sql("MicroMsg.db", "SELECT UserName, NickName FROM Contact;")
|
||
return {contact["UserName"]: contact["NickName"] for contact in contacts}
|
||
|
||
def keepRunningAndBlockProcess(self) -> None:
|
||
"""
|
||
保持机器人运行，不让进程退出
|
||
"""
|
||
while True:
|
||
self.runPendingJobs()
|
||
time.sleep(1)
|
||
|
||
def autoAcceptFriendRequest(self, msg: WxMsg) -> None:
|
||
try:
|
||
xml = ET.fromstring(msg.content)
|
||
v3 = xml.attrib["encryptusername"]
|
||
v4 = xml.attrib["ticket"]
|
||
scene = int(xml.attrib["scene"])
|
||
self.wcf.accept_new_friend(v3, v4, scene)
|
||
|
||
except Exception as e:
|
||
self.LOG.error(f"同意好友出错：{e}")
|
||
|
||
def sayHiToNewFriend(self, msg: WxMsg) -> None:
|
||
nickName = re.findall(r"你已添加了(.*)，现在可以开始聊天了。", msg.content)
|
||
if nickName:
|
||
# 添加了好友，更新好友列表
|
||
self.allContacts[msg.sender] = nickName[0]
|
||
self.sendTextMsg(f"Hi {nickName[0]}，我是泡泡，我自动通过了你的好友请求。", msg.sender)
|
||
|
||
def newsReport(self) -> None:
|
||
receivers = self.config.NEWS
|
||
if not receivers:
|
||
self.LOG.info("未配置定时新闻接收人，跳过。")
|
||
return
|
||
|
||
self.LOG.info("开始执行定时新闻推送任务...")
|
||
# 获取新闻，解包返回的元组
|
||
is_today, news_content = News().get_important_news()
|
||
|
||
# 必须是当天的新闻 (is_today=True) 并且有有效内容 (news_content非空) 才发送
|
||
if is_today and news_content:
|
||
self.LOG.info(f"成功获取当天新闻，准备推送给 {len(receivers)} 个接收人...")
|
||
for r in receivers:
|
||
self.sendTextMsg(news_content, r)
|
||
self.LOG.info("定时新闻推送完成。")
|
||
else:
|
||
# 记录没有发送的原因
|
||
if not is_today and news_content:
|
||
self.LOG.warning("获取到的是旧闻，定时推送已跳过。")
|
||
elif not news_content:
|
||
self.LOG.warning("获取新闻内容失败或为空，定时推送已跳过。")
|
||
else: # 理论上不会执行到这里
|
||
self.LOG.warning("获取新闻失败（未知原因），定时推送已跳过。")
|
||
|
||
def weatherReport(self, receivers: list = None) -> None:
|
||
if receivers is None:
|
||
receivers = self.config.WEATHER
|
||
if not receivers or not self.config.CITY_CODE:
|
||
self.LOG.warning("未配置天气城市代码或接收人")
|
||
return
|
||
|
||
report = Weather(self.config.CITY_CODE).get_weather()
|
||
for r in receivers:
|
||
self.sendTextMsg(report, r)
|
||
|
||
def sendDuelMsg(self, msg: str, receiver: str) -> None:
|
||
"""发送决斗消息，不受消息频率限制，不记入历史记录
|
||
:param msg: 消息字符串
|
||
:param receiver: 接收人wxid或者群id
|
||
"""
|
||
try:
|
||
self.wcf.send_text(f"{msg}", receiver, "")
|
||
except Exception as e:
|
||
self.LOG.error(f"发送决斗消息失败: {e}")
|
||
|
||
def cleanup_perplexity_threads(self):
|
||
"""清理所有Perplexity线程"""
|
||
# 如果已初始化Perplexity实例，调用其清理方法
|
||
perplexity_instance = self.get_perplexity_instance()
|
||
if perplexity_instance:
|
||
perplexity_instance.cleanup()
|
||
|
||
# 检查并等待决斗线程结束
|
||
if hasattr(self, 'duel_manager') and self.duel_manager.is_duel_running():
|
||
self.LOG.info("等待决斗线程结束...")
|
||
# 最多等待5秒
|
||
for i in range(5):
|
||
if not self.duel_manager.is_duel_running():
|
||
break
|
||
time.sleep(1)
|
||
|
||
if self.duel_manager.is_duel_running():
|
||
self.LOG.warning("决斗线程在退出时仍在运行")
|
||
else:
|
||
self.LOG.info("决斗线程已结束")
|
||
|
||
def cleanup(self):
|
||
"""清理所有资源，在程序退出前调用"""
|
||
self.LOG.info("开始清理机器人资源...")
|
||
|
||
# 清理Perplexity线程
|
||
self.cleanup_perplexity_threads()
|
||
|
||
# 关闭消息历史数据库连接
|
||
if hasattr(self, 'message_summary') and self.message_summary:
|
||
self.LOG.info("正在关闭消息历史数据库...")
|
||
self.message_summary.close_db()
|
||
|
||
self.LOG.info("机器人资源清理完成")
|
||
|
||
def get_perplexity_instance(self):
|
||
"""获取Perplexity实例
|
||
|
||
Returns:
|
||
Perplexity: Perplexity实例，如果未配置则返回None
|
||
"""
|
||
# 检查是否已有Perplexity实例
|
||
if hasattr(self, 'perplexity'):
|
||
return self.perplexity
|
||
|
||
# 检查config中是否有Perplexity配置
|
||
if hasattr(self.config, 'PERPLEXITY') and Perplexity.value_check(self.config.PERPLEXITY):
|
||
self.perplexity = Perplexity(self.config.PERPLEXITY)
|
||
return self.perplexity
|
||
|
||
# 检查chat是否是Perplexity类型
|
||
if isinstance(self.chat, Perplexity):
|
||
return self.chat
|
||
|
||
# 如果存在chat_models字典，尝试从中获取
|
||
if hasattr(self, 'chat_models') and ChatType.PERPLEXITY.value in self.chat_models:
|
||
return self.chat_models[ChatType.PERPLEXITY.value]
|
||
|
||
return None
|
||
|
||
def try_trigger_goblin_gift(self, msg: WxMsg) -> None:
|
||
"""尝试触发古灵阁妖精的馈赠事件
|
||
|
||
用户与机器人互动时，有概率获得随机积分
|
||
根据配置决定是否启用及在哪些群聊启用
|
||
|
||
Args:
|
||
msg: 微信消息对象
|
||
"""
|
||
# 调用管理器的触发方法
|
||
self.goblin_gift_manager.try_trigger(msg)
|
||
|
||
def _select_model_for_message(self, msg: WxMsg) -> None:
|
||
"""根据消息来源选择对应的AI模型
|
||
:param msg: 接收到的消息
|
||
"""
|
||
if not hasattr(self, 'chat_models') or not self.chat_models:
|
||
return # 没有可用模型，无需切换
|
||
|
||
# 获取消息来源ID
|
||
source_id = msg.roomid if msg.from_group() else msg.sender
|
||
|
||
# 检查配置
|
||
if not hasattr(self.config, 'GROUP_MODELS'):
|
||
# 没有配置，使用默认模型
|
||
if self.default_model_id in self.chat_models:
|
||
self.chat = self.chat_models[self.default_model_id]
|
||
return
|
||
|
||
# 群聊消息处理
|
||
if msg.from_group():
|
||
model_mappings = self.config.GROUP_MODELS.get('mapping', [])
|
||
for mapping in model_mappings:
|
||
if mapping.get('room_id') == source_id:
|
||
model_id = mapping.get('model')
|
||
if model_id in self.chat_models:
|
||
# 切换到指定模型
|
||
if self.chat != self.chat_models[model_id]:
|
||
self.chat = self.chat_models[model_id]
|
||
self.LOG.info(f"已为群 {source_id} 切换到模型: {self.chat.__class__.__name__}")
|
||
else:
|
||
self.LOG.warning(f"群 {source_id} 配置的模型ID {model_id} 不可用，使用默认模型")
|
||
if self.default_model_id in self.chat_models:
|
||
self.chat = self.chat_models[self.default_model_id]
|
||
return
|
||
# 私聊消息处理
|
||
else:
|
||
private_mappings = self.config.GROUP_MODELS.get('private_mapping', [])
|
||
for mapping in private_mappings:
|
||
if mapping.get('wxid') == source_id:
|
||
model_id = mapping.get('model')
|
||
if model_id in self.chat_models:
|
||
# 切换到指定模型
|
||
if self.chat != self.chat_models[model_id]:
|
||
self.chat = self.chat_models[model_id]
|
||
self.LOG.info(f"已为私聊用户 {source_id} 切换到模型: {self.chat.__class__.__name__}")
|
||
else:
|
||
self.LOG.warning(f"私聊用户 {source_id} 配置的模型ID {model_id} 不可用，使用默认模型")
|
||
if self.default_model_id in self.chat_models:
|
||
self.chat = self.chat_models[self.default_model_id]
|
||
return
|
||
|
||
# 如果没有找到对应配置，使用默认模型
|
||
if self.default_model_id in self.chat_models:
|
||
self.chat = self.chat_models[self.default_model_id]
|
||
|
||
def onMsg(self, msg: WxMsg) -> int:
|
||
try:
|
||
self.LOG.info(msg)
|
||
self.processMsg(msg)
|
||
except Exception as e:
|
||
self.LOG.error(e)
|
||
|
||
return 0
|
||
|
||
def preprocess(self, msg: WxMsg) -> MessageContext:
|
||
"""
|
||
预处理消息，生成MessageContext对象
|
||
:param msg: 微信消息对象
|
||
:return: MessageContext对象
|
||
"""
|
||
is_group = msg.from_group()
|
||
is_at_bot = False
|
||
pure_text = msg.content # 默认使用原始内容
|
||
|
||
# 初始化引用图片相关属性
|
||
is_quoted_image = False
|
||
quoted_msg_id = None
|
||
quoted_image_extra = None
|
||
|
||
# 处理引用消息等特殊情况
|
||
if msg.type == 49 and ("<title>" in msg.content or "<appmsg" in msg.content):
|
||
# 尝试提取引用消息中的文本
|
||
if is_group:
|
||
msg_data = self.xml_processor.extract_quoted_message(msg)
|
||
else:
|
||
msg_data = self.xml_processor.extract_private_quoted_message(msg)
|
||
|
||
if msg_data and msg_data.get("new_content"):
|
||
pure_text = msg_data["new_content"]
|
||
# 检查是否包含@机器人
|
||
if is_group and pure_text.startswith(f"@{self.allContacts.get(self.wxid, '')}"):
|
||
is_at_bot = True
|
||
pure_text = re.sub(r"^@.*?[\u2005|\s]", "", pure_text).strip()
|
||
elif "<title>" in msg.content:
|
||
# 备选：直接从title标签提取
|
||
title_match = re.search(r'<title>(.*?)</title>', msg.content)
|
||
if title_match:
|
||
pure_text = title_match.group(1).strip()
|
||
# 检查是否@机器人
|
||
if is_group and pure_text.startswith(f"@{self.allContacts.get(self.wxid, '')}"):
|
||
is_at_bot = True
|
||
pure_text = re.sub(r"^@.*?[\u2005|\s]", "", pure_text).strip()
|
||
|
||
# 检查并提取图片引用信息
|
||
if msg_data and msg_data.get("media_type") == "引用图片" and \
|
||
msg_data.get("quoted_msg_id") and \
|
||
msg_data.get("quoted_image_extra"):
|
||
is_quoted_image = True
|
||
quoted_msg_id = msg_data["quoted_msg_id"]
|
||
quoted_image_extra = msg_data["quoted_image_extra"]
|
||
self.LOG.info(f"预处理已提取引用图片信息: msg_id={quoted_msg_id}")
|
||
|
||
# 处理文本消息
|
||
elif msg.type == 1: # 文本消息
|
||
# 检查是否@机器人
|
||
if is_group and msg.is_at(self.wxid):
|
||
is_at_bot = True
|
||
# 移除@前缀
|
||
pure_text = re.sub(r"^@.*?[\u2005|\s]", "", msg.content).strip()
|
||
else:
|
||
pure_text = msg.content.strip()
|
||
|
||
# 构造上下文对象
|
||
ctx = MessageContext(
|
||
msg=msg,
|
||
wcf=self.wcf,
|
||
config=self.config,
|
||
all_contacts=self.allContacts,
|
||
robot_wxid=self.wxid,
|
||
robot=self, # 传入Robot实例本身，便于handlers访问其方法
|
||
logger=self.LOG,
|
||
text=pure_text,
|
||
is_group=is_group,
|
||
is_at_bot=is_at_bot or (is_group and msg.is_at(self.wxid)), # 确保is_at_bot正确
|
||
)
|
||
|
||
# 将图片引用信息添加到 ctx
|
||
setattr(ctx, 'is_quoted_image', is_quoted_image)
|
||
if is_quoted_image:
|
||
setattr(ctx, 'quoted_msg_id', quoted_msg_id)
|
||
setattr(ctx, 'quoted_image_extra', quoted_image_extra)
|
||
|
||
# 获取发送者昵称
|
||
ctx.sender_name = ctx.get_sender_alias_or_name()
|
||
|
||
self.LOG.debug(f"预处理消息: text='{ctx.text}', is_group={ctx.is_group}, is_at_bot={ctx.is_at_bot}, sender='{ctx.sender_name}', is_quoted_image={is_quoted_image}")
|
||
return ctx
|
||
|