mirror of
https://github.com/Zippland/Bubbles.git
synced 2026-02-12 02:05:13 +08:00
refactor(handlers): 清理未使用的导入并重构工具处理逻辑
This commit is contained in:
zihanjian
@@ -1,14 +1,9 @@
|
||||
import re
|
||||
from typing import Optional, Match, Dict, Any
|
||||
import json
|
||||
from datetime import datetime
|
||||
import os
|
||||
import time as time_mod
|
||||
from typing import Optional, Match, TYPE_CHECKING
|
||||
|
||||
from function.func_persona import build_persona_system_prompt
|
||||
|
||||
# 前向引用避免循环导入
|
||||
from typing import TYPE_CHECKING
|
||||
if TYPE_CHECKING:
|
||||
from .context import MessageContext
|
||||
|
||||
@@ -91,22 +86,18 @@ def handle_chitchat(ctx: 'MessageContext', match: Optional[Match]) -> bool:
|
||||
|
||||
# 插嘴模式不使用工具，减少 token 消耗
|
||||
if not is_auto_random_reply:
|
||||
from tools import tool_registry
|
||||
# 导入工具模块以触发注册（仅首次生效）
|
||||
import tools.history
|
||||
import tools.reminder
|
||||
import tools.web_search
|
||||
import skills
|
||||
|
||||
openai_tools = tool_registry.get_openai_tools()
|
||||
openai_tools = skills.get_openai_tools()
|
||||
if openai_tools:
|
||||
tools = openai_tools
|
||||
tool_handler = tool_registry.create_handler(ctx)
|
||||
tool_handler = skills.create_handler(ctx)
|
||||
|
||||
# ── 构建系统提示 ──────────────────────────────────────
|
||||
persona_text = getattr(ctx, 'persona', None)
|
||||
system_prompt_override = None
|
||||
|
||||
# 工具使用指引：告诉 LLM 何时该用工具
|
||||
# 工具使用指引
|
||||
tool_guidance = ""
|
||||
if tools:
|
||||
tool_guidance = (
|
||||
@@ -116,7 +107,7 @@ def handle_chitchat(ctx: 'MessageContext', match: Optional[Match]) -> bool:
|
||||
"- 用户想设置/查看/删除提醒 → 调用 reminder_create / reminder_list / reminder_delete\n"
|
||||
"- 用户提到之前聊过的内容、或你需要回顾更早的对话 → 调用 lookup_chat_history\n"
|
||||
"- 日常闲聊、观点讨论、情感交流 → 直接回复，不需要调用任何工具\n"
|
||||
"你可以在一次对话中多次调用工具（例如先搜索再设提醒），每次调用的结果会反馈给你继续推理。"
|
||||
"你可以在一次对话中多次调用工具，每次调用的结果会反馈给你继续推理。"
|
||||
)
|
||||
|
||||
if persona_text:
|
||||
@@ -143,7 +134,7 @@ def handle_chitchat(ctx: 'MessageContext', match: Optional[Match]) -> bool:
|
||||
specific_max_history=specific_max_history,
|
||||
tools=tools,
|
||||
tool_handler=tool_handler,
|
||||
tool_max_iterations=10,
|
||||
tool_max_iterations=20,
|
||||
)
|
||||
|
||||
if rsp:
|
||||
@@ -207,671 +198,3 @@ def _handle_quoted_image(ctx, chat_model) -> bool:
|
||||
ctx.logger.error(f"处理引用图片出错: {e}", exc_info=True)
|
||||
ctx.send_text(f"处理图片时发生错误: {str(e)}")
|
||||
return True
|
||||
|
||||
def handle_perplexity_ask(ctx: 'MessageContext', match: Optional[Match]) -> bool:
|
||||
"""
|
||||
处理 "ask" 命令，调用 Perplexity AI
|
||||
|
||||
匹配: ask [问题内容]
|
||||
"""
|
||||
if not match: # 理论上正则匹配成功才会被调用，但加个检查更安全
|
||||
return False
|
||||
|
||||
# 1. 尝试从 Robot 实例获取 Perplexity 实例
|
||||
perplexity_instance = getattr(ctx.robot, 'perplexity', None)
|
||||
|
||||
# 2. 检查 Perplexity 实例是否存在
|
||||
if not perplexity_instance:
|
||||
if ctx.logger:
|
||||
ctx.logger.warning("尝试调用 Perplexity，但实例未初始化或未配置。")
|
||||
ctx.send_text("❌ Perplexity 功能当前不可用或未正确配置。")
|
||||
return True # 命令已被处理（错误处理也是处理）
|
||||
|
||||
# 3. 从匹配结果中提取问题内容
|
||||
prompt = match.group(1).strip()
|
||||
if not prompt: # 如果 'ask' 后面没有内容
|
||||
ctx.send_text("请在 'ask' 后面加上您想问的问题。", ctx.msg.sender if ctx.is_group else None)
|
||||
return True # 命令已被处理
|
||||
|
||||
# 4. 准备调用 Perplexity 实例的 process_message 方法
|
||||
if ctx.logger:
|
||||
ctx.logger.info(f"检测到 Perplexity 请求，发送者: {ctx.sender_name}, 问题: {prompt[:50]}...")
|
||||
|
||||
# 准备参数并调用 process_message
|
||||
# 确保无论用户输入有没有空格，都以标准格式"ask 问题"传给process_message
|
||||
content_for_perplexity = f"ask {prompt}" # 重构包含触发词的内容
|
||||
chat_id = ctx.get_receiver()
|
||||
sender_wxid = ctx.msg.sender
|
||||
room_id = ctx.msg.roomid if ctx.is_group else None
|
||||
is_group = ctx.is_group
|
||||
|
||||
# 5. 调用 process_message 并返回其结果
|
||||
was_handled, fallback_prompt = perplexity_instance.process_message(
|
||||
content=content_for_perplexity,
|
||||
chat_id=chat_id,
|
||||
sender=sender_wxid,
|
||||
roomid=room_id,
|
||||
from_group=is_group,
|
||||
send_text_func=ctx.send_text
|
||||
)
|
||||
|
||||
# 6. 如果没有被处理且有备选prompt，使用默认AI处理
|
||||
if not was_handled and fallback_prompt:
|
||||
if ctx.logger:
|
||||
ctx.logger.info(f"使用备选prompt '{fallback_prompt[:20]}...' 调用默认AI处理")
|
||||
|
||||
# 获取当前选定的AI模型
|
||||
chat_model = None
|
||||
if hasattr(ctx, 'chat'):
|
||||
chat_model = ctx.chat
|
||||
elif ctx.robot and hasattr(ctx.robot, 'chat'):
|
||||
chat_model = ctx.robot.chat
|
||||
|
||||
if chat_model:
|
||||
# 使用与 handle_chitchat 类似的逻辑，但使用备选prompt
|
||||
try:
|
||||
# 格式化消息，与 handle_chitchat 保持一致
|
||||
if ctx.robot and hasattr(ctx.robot, "xml_processor"):
|
||||
if ctx.is_group:
|
||||
msg_data = ctx.robot.xml_processor.extract_quoted_message(ctx.msg)
|
||||
q_with_info = ctx.robot.xml_processor.format_message_for_ai(msg_data, ctx.sender_name)
|
||||
else:
|
||||
msg_data = ctx.robot.xml_processor.extract_private_quoted_message(ctx.msg)
|
||||
q_with_info = ctx.robot.xml_processor.format_message_for_ai(msg_data, ctx.sender_name)
|
||||
|
||||
if not q_with_info:
|
||||
import time
|
||||
current_time = time.strftime("%H:%M", time.localtime())
|
||||
q_with_info = f"[{current_time}] {ctx.sender_name}: {prompt or '[空内容]'}"
|
||||
else:
|
||||
import time
|
||||
current_time = time.strftime("%H:%M", time.localtime())
|
||||
q_with_info = f"[{current_time}] {ctx.sender_name}: {prompt or '[空内容]'}"
|
||||
|
||||
if ctx.is_group and not ctx.is_at_bot and getattr(ctx, 'auto_random_reply', False):
|
||||
latest_message_prompt = (
|
||||
"你目前是在群聊里主动接话，没有人点名让你发言。\n"
|
||||
"请根据下面这句最新消息插入一条自然、不突兀的中文回复，语气放松随和即可：\n"
|
||||
f"“{q_with_info}”\n"
|
||||
"不要重复对方的话，也不要显得过于正式。"
|
||||
)
|
||||
else:
|
||||
latest_message_prompt = (
|
||||
"请你基于下面这条最新收到的消息，直接面向发送者进行自然的中文回复：\n"
|
||||
f"{q_with_info}\n"
|
||||
"请只围绕这条消息的内容作答，不要泛泛而谈。"
|
||||
)
|
||||
|
||||
if ctx.logger:
|
||||
ctx.logger.info(f"发送给默认AI的消息内容: {latest_message_prompt}")
|
||||
|
||||
# 调用 AI 模型时传入备选 prompt
|
||||
# 需要调整 get_answer 方法以支持 system_prompt_override 参数
|
||||
# 这里我们假设已对各AI模型实现了这个参数
|
||||
specific_max_history = getattr(ctx, 'specific_max_history', None)
|
||||
override_prompt = fallback_prompt
|
||||
persona_text = getattr(ctx, 'persona', None)
|
||||
if persona_text:
|
||||
try:
|
||||
override_prompt = build_persona_system_prompt(
|
||||
chat_model,
|
||||
persona_text,
|
||||
override_prompt=fallback_prompt
|
||||
)
|
||||
except Exception as persona_exc:
|
||||
if ctx.logger:
|
||||
ctx.logger.error(f"构建人设系统提示失败: {persona_exc}", exc_info=True)
|
||||
override_prompt = fallback_prompt
|
||||
|
||||
rsp = chat_model.get_answer(
|
||||
question=latest_message_prompt,
|
||||
wxid=ctx.get_receiver(),
|
||||
system_prompt_override=override_prompt,
|
||||
specific_max_history=specific_max_history
|
||||
)
|
||||
|
||||
if rsp:
|
||||
ctx.send_text(rsp, "")
|
||||
return True
|
||||
else:
|
||||
if ctx.logger:
|
||||
ctx.logger.error("无法从默认AI获得答案")
|
||||
except Exception as e:
|
||||
if ctx.logger:
|
||||
ctx.logger.error(f"使用备选prompt调用默认AI时出错: {e}")
|
||||
|
||||
return was_handled
|
||||
|
||||
def handle_reminder(ctx: 'MessageContext', match: Optional[Match]) -> bool:
|
||||
"""处理来自私聊或群聊的 '提醒' 命令，支持批量添加多个提醒"""
|
||||
# 2. 获取用户输入的提醒内容 (现在从完整消息获取)
|
||||
raw_text = ctx.msg.content.strip() # 修改：从 ctx.msg.content 获取
|
||||
if not raw_text: # 修改：仅检查是否为空
|
||||
# 在群聊中@用户回复
|
||||
at_list = ctx.msg.sender if ctx.is_group else ""
|
||||
ctx.send_text("请告诉我需要提醒什么内容和时间呀~ (例如：提醒我明天下午3点开会)", at_list)
|
||||
return True
|
||||
|
||||
# 3. 构造给 AI 的 Prompt，更新为支持批量提醒
|
||||
sys_prompt = """
|
||||
你是提醒解析助手。请仔细分析用户输入的提醒信息，**识别其中可能包含的所有独立提醒请求**。将所有成功解析的提醒严格按照以下 JSON **数组** 格式输出结果，数组中的每个元素代表一个独立的提醒:
|
||||
[
|
||||
{{
|
||||
"type": "once" | "daily" | "weekly", // 提醒类型: "once" (一次性) 或 "daily" (每日重复) 或 "weekly" (每周重复)
|
||||
"time": "YYYY-MM-DD HH:MM" | "HH:MM", // "once"类型必须是 'YYYY-MM-DD HH:MM' 格式, "daily"与"weekly"类型必须是 'HH:MM' 格式。时间必须是未来的。
|
||||
"content": "提醒的具体内容文本",
|
||||
"weekday": 0-6, // 仅当 type="weekly" 时需要，周一=0, 周二=1, ..., 周日=6
|
||||
"extra": {{}} // 保留字段，目前为空对象即可
|
||||
}},
|
||||
// ... 可能有更多提醒对象 ...
|
||||
]
|
||||
|
||||
**重要:** 你的回复必须仅包含有效的JSON数组，不要包含任何其他说明文字。所有JSON中的布尔值、数字应该没有引号，字符串需要有引号。
|
||||
|
||||
- **仔细分析用户输入，识别所有独立的提醒请求。**
|
||||
- 对每一个识别出的提醒，判断其类型 (`once`, `daily`, `weekly`) 并计算准确时间。
|
||||
- "once"类型时间必须是 'YYYY-MM-DD HH:MM' 格式, "daily"/"weekly"类型必须是 'HH:MM' 格式。时间必须是未来的。
|
||||
- "weekly"类型必须提供 weekday (周一=0...周日=6)。
|
||||
- **将所有解析成功的提醒对象放入一个 JSON 数组中返回。**
|
||||
- 如果只识别出一个提醒，返回包含单个元素的数组。
|
||||
- **如果无法识别出任何有效提醒，返回空数组 `[]`。**
|
||||
- 如果用户输入的某个提醒部分信息不完整或格式错误，请尝试解析其他部分，并在最终数组中仅包含解析成功的提醒。
|
||||
- 输出结果必须是纯 JSON 数组，不包含任何其他说明文字。
|
||||
|
||||
当前准确时间是：{current_datetime}
|
||||
"""
|
||||
current_dt_str = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||
formatted_prompt = sys_prompt.format(current_datetime=current_dt_str)
|
||||
|
||||
# 4. 调用AI模型并解析
|
||||
q_for_ai = f"请解析以下用户提醒，识别所有独立的提醒请求:\n{raw_text}"
|
||||
try:
|
||||
# 检查AI模型
|
||||
if not hasattr(ctx, 'chat') or not ctx.chat:
|
||||
raise ValueError("当前上下文中没有可用的AI模型")
|
||||
|
||||
# 获取AI回答
|
||||
at_list = ctx.msg.sender if ctx.is_group else ""
|
||||
|
||||
# 实现最多尝试3次解析AI回复的逻辑
|
||||
max_retries = 3
|
||||
retry_count = 0
|
||||
parsed_reminders = [] # 初始化为空列表
|
||||
ai_parsing_success = False
|
||||
|
||||
while retry_count < max_retries and not ai_parsing_success:
|
||||
# 如果是重试，更新提示信息
|
||||
if retry_count > 0:
|
||||
enhanced_prompt = sys_prompt + f"\n\n**重要提示:** 这是第{retry_count+1}次尝试。你之前的回复格式有误，无法被解析为有效的JSON。请确保你的回复仅包含有效的JSON数组，没有其他任何文字。"
|
||||
formatted_prompt = enhanced_prompt.format(current_datetime=current_dt_str)
|
||||
# 在重试时提供更明确的信息
|
||||
retry_q = f"请再次解析以下提醒，并返回严格的JSON数组格式(第{retry_count+1}次尝试):\n{raw_text}"
|
||||
q_for_ai = retry_q
|
||||
|
||||
ai_response = ctx.chat.get_answer(q_for_ai, ctx.get_receiver(), system_prompt_override=formatted_prompt)
|
||||
|
||||
# 尝试匹配 [...] 或 {...} (兼容单个提醒的情况，但优先列表)
|
||||
json_match_list = re.search(r'\[.*\]', ai_response, re.DOTALL)
|
||||
json_match_obj = re.search(r'\{.*\}', ai_response, re.DOTALL)
|
||||
|
||||
json_str = None
|
||||
if json_match_list:
|
||||
json_str = json_match_list.group(0)
|
||||
elif json_match_obj: # 如果没找到列表，尝试找单个对象 (增加兼容性)
|
||||
json_str = f"[{json_match_obj.group(0)}]" # 将单个对象包装成数组
|
||||
else:
|
||||
json_str = ai_response # 如果都找不到，直接尝试解析原始回复
|
||||
|
||||
try:
|
||||
# 尝试解析JSON
|
||||
parsed_data = json.loads(json_str)
|
||||
# 确保解析结果是一个列表
|
||||
if isinstance(parsed_data, dict):
|
||||
parsed_reminders = [parsed_data] # 包装成单元素列表
|
||||
elif isinstance(parsed_data, list):
|
||||
parsed_reminders = parsed_data # 本身就是列表
|
||||
else:
|
||||
# 解析结果不是列表也不是字典，无法处理
|
||||
raise ValueError("AI 返回的不是有效的 JSON 列表或对象")
|
||||
|
||||
# 如果能到这里，说明解析成功
|
||||
ai_parsing_success = True
|
||||
|
||||
except (json.JSONDecodeError, ValueError) as e:
|
||||
# JSON解析失败
|
||||
retry_count += 1
|
||||
if ctx.logger:
|
||||
ctx.logger.warning(f"AI 返回 JSON 解析失败(第{retry_count}次尝试): {ai_response}, 错误: {str(e)}")
|
||||
|
||||
if retry_count >= max_retries:
|
||||
# 达到最大重试次数，返回错误
|
||||
ctx.send_text(f"❌ 抱歉，无法理解您的提醒请求。请尝试换一种方式表达，或分开设置多个提醒。", at_list)
|
||||
if ctx.logger: ctx.logger.error(f"解析AI回复失败，已达到最大重试次数({max_retries}): {ai_response}")
|
||||
return True
|
||||
# 否则继续下一次循环重试
|
||||
|
||||
# 检查 ReminderManager 是否存在
|
||||
if not hasattr(ctx.robot, 'reminder_manager'):
|
||||
ctx.send_text("❌ 内部错误：提醒管理器未初始化。", at_list)
|
||||
if ctx.logger: ctx.logger.error("handle_reminder 无法访问 ctx.robot.reminder_manager")
|
||||
return True
|
||||
|
||||
# 如果AI返回空列表，告知用户
|
||||
if not parsed_reminders:
|
||||
ctx.send_text("🤔 嗯... 我好像没太明白您想设置什么提醒，可以换种方式再说一次吗？", at_list)
|
||||
return True
|
||||
|
||||
# 批量处理提醒
|
||||
results = [] # 用于存储每个提醒的处理结果
|
||||
roomid = ctx.msg.roomid if ctx.is_group else None
|
||||
|
||||
for index, data in enumerate(parsed_reminders):
|
||||
reminder_label = f"提醒{index+1}" # 给每个提醒一个标签，方便反馈
|
||||
validation_error = None # 存储验证错误信息
|
||||
|
||||
# **验证单个提醒数据**
|
||||
if not isinstance(data, dict):
|
||||
validation_error = "格式错误 (不是有效的提醒对象)"
|
||||
elif not data.get("type") or not data.get("time") or not data.get("content"):
|
||||
validation_error = "缺少必要字段(类型/时间/内容)"
|
||||
elif len(data.get("content", "").strip()) < 2:
|
||||
validation_error = "提醒内容太短"
|
||||
else:
|
||||
# 验证时间格式
|
||||
time_value = data.get("time", "")
|
||||
if data["type"] == "once":
|
||||
parsed_dt = None
|
||||
for fmt in ("%Y-%m-%d %H:%M", "%Y-%m-%d %H:%M:%S"):
|
||||
try:
|
||||
parsed_dt = datetime.strptime(time_value, fmt)
|
||||
break
|
||||
except ValueError:
|
||||
continue
|
||||
if not parsed_dt:
|
||||
validation_error = f"时间格式错误 ({time_value})"
|
||||
elif parsed_dt < datetime.now():
|
||||
validation_error = f"时间 ({time_value}) 必须是未来的时间"
|
||||
else:
|
||||
# 统一存储格式为分钟精度，避免比较时出错
|
||||
data["time"] = parsed_dt.strftime("%Y-%m-%d %H:%M")
|
||||
elif data["type"] in ["daily", "weekly"]:
|
||||
parsed_time = None
|
||||
for fmt in ("%H:%M", "%H:%M:%S"):
|
||||
try:
|
||||
parsed_time = datetime.strptime(time_value, fmt)
|
||||
break
|
||||
except ValueError:
|
||||
continue
|
||||
if not parsed_time:
|
||||
validation_error = f"时间格式错误 ({time_value})"
|
||||
else:
|
||||
data["time"] = parsed_time.strftime("%H:%M")
|
||||
else:
|
||||
validation_error = f"不支持的提醒类型: {data.get('type')}"
|
||||
|
||||
# 验证周提醒 (如果类型是 weekly 且无验证错误)
|
||||
if not validation_error and data["type"] == "weekly":
|
||||
if not (isinstance(data.get("weekday"), int) and 0 <= data.get("weekday") <= 6):
|
||||
validation_error = "每周提醒需要指定周几(0-6)"
|
||||
|
||||
# 如果验证通过，尝试添加到数据库
|
||||
if not validation_error:
|
||||
try:
|
||||
success, result_or_id = ctx.robot.reminder_manager.add_reminder(ctx.msg.sender, data, roomid=roomid)
|
||||
if success:
|
||||
results.append({"label": reminder_label, "success": True, "id": result_or_id, "data": data})
|
||||
if ctx.logger: ctx.logger.info(f"成功添加提醒 {result_or_id} for {ctx.msg.sender} (来自批量处理)")
|
||||
else:
|
||||
# add_reminder 返回错误信息
|
||||
results.append({"label": reminder_label, "success": False, "error": result_or_id, "data": data})
|
||||
if ctx.logger: ctx.logger.warning(f"添加提醒失败 (来自批量处理): {result_or_id}")
|
||||
except Exception as db_e:
|
||||
# 捕获 add_reminder 可能抛出的其他异常
|
||||
error_msg = f"数据库错误: {db_e}"
|
||||
results.append({"label": reminder_label, "success": False, "error": error_msg, "data": data})
|
||||
if ctx.logger: ctx.logger.error(f"添加提醒时数据库出错 (来自批量处理): {db_e}", exc_info=True)
|
||||
else:
|
||||
# 验证失败
|
||||
results.append({"label": reminder_label, "success": False, "error": validation_error, "data": data})
|
||||
if ctx.logger: ctx.logger.warning(f"提醒数据验证失败 ({reminder_label}): {validation_error} - Data: {data}")
|
||||
|
||||
# 构建汇总反馈消息
|
||||
reply_parts = []
|
||||
successful_count = sum(1 for res in results if res["success"])
|
||||
failed_count = len(results) - successful_count
|
||||
|
||||
# 添加总览信息
|
||||
if len(results) > 1: # 只有多个提醒时才需要总览
|
||||
if successful_count > 0 and failed_count > 0:
|
||||
reply_parts.append(f"✅ 已设置 {successful_count} 个提醒，{failed_count} 个设置失败：\n")
|
||||
elif successful_count > 0:
|
||||
reply_parts.append(f"✅ 已设置 {successful_count} 个提醒：\n")
|
||||
else:
|
||||
reply_parts.append(f"❌ 抱歉，所有 {len(results)} 个提醒设置均失败：\n")
|
||||
|
||||
# 添加每个提醒的详细信息
|
||||
for res in results:
|
||||
content_preview = res['data'].get('content', '未知内容')
|
||||
# 如果内容太长，截取前20个字符加省略号
|
||||
if len(content_preview) > 20:
|
||||
content_preview = content_preview[:20] + "..."
|
||||
|
||||
if res["success"]:
|
||||
reminder_id = res['id']
|
||||
type_str = {"once": "一次性", "daily": "每日", "weekly": "每周"}.get(res['data'].get('type'), "未知")
|
||||
time_display = res['data'].get("time", "?")
|
||||
|
||||
# 为周提醒格式化显示
|
||||
if res['data'].get("type") == "weekly" and "weekday" in res['data']:
|
||||
weekdays = ["周一", "周二", "周三", "周四", "周五", "周六", "周日"]
|
||||
if 0 <= res['data']["weekday"] <= 6:
|
||||
time_display = f"{weekdays[res['data']['weekday']]} {time_display}"
|
||||
|
||||
# 单个提醒或多个提醒的第一个，不需要标签
|
||||
if len(results) == 1:
|
||||
reply_parts.append(f"✅ 已为您设置{type_str}提醒:\n"
|
||||
f"时间: {time_display}\n"
|
||||
f"内容: {res['data'].get('content', '无')}")
|
||||
else:
|
||||
reply_parts.append(f"✅ {res['label']}: {type_str}\n {time_display} - \"{content_preview}\"")
|
||||
else:
|
||||
# 失败的提醒
|
||||
if len(results) == 1:
|
||||
reply_parts.append(f"❌ 设置提醒失败: {res['error']}")
|
||||
else:
|
||||
reply_parts.append(f"❌ {res['label']}: \"{content_preview}\" - {res['error']}")
|
||||
|
||||
# 发送汇总消息
|
||||
ctx.send_text("\n".join(reply_parts), at_list)
|
||||
|
||||
return True # 命令处理流程结束
|
||||
|
||||
except Exception as e: # 捕获代码块顶层的其他潜在错误
|
||||
at_list = ctx.msg.sender if ctx.is_group else ""
|
||||
error_message = f"处理提醒时发生意外错误: {str(e)}"
|
||||
ctx.send_text(f"❌ {error_message}", at_list)
|
||||
if ctx.logger:
|
||||
ctx.logger.error(f"handle_reminder 顶层错误: {e}", exc_info=True)
|
||||
return True
|
||||
|
||||
def handle_list_reminders(ctx: 'MessageContext', match: Optional[Match]) -> bool:
|
||||
"""处理查看提醒命令（支持群聊和私聊）"""
|
||||
if not hasattr(ctx.robot, 'reminder_manager'):
|
||||
ctx.send_text("❌ 内部错误：提醒管理器未初始化。", ctx.msg.sender if ctx.is_group else "")
|
||||
return True
|
||||
|
||||
reminders = ctx.robot.reminder_manager.list_reminders(ctx.msg.sender)
|
||||
# 在群聊中@用户
|
||||
at_list = ctx.msg.sender if ctx.is_group else ""
|
||||
|
||||
if not reminders:
|
||||
ctx.send_text("您还没有设置任何提醒。", at_list)
|
||||
return True
|
||||
|
||||
reply_parts = ["📝 您设置的提醒列表（包括私聊和群聊）：\n"]
|
||||
for i, r in enumerate(reminders):
|
||||
# 格式化星期几（如果存在）
|
||||
weekday_str = ""
|
||||
if r.get("weekday") is not None:
|
||||
weekdays = ["周一", "周二", "周三", "周四", "周五", "周六", "周日"]
|
||||
weekday_str = f" (每周{weekdays[r['weekday']]})" if 0 <= r['weekday'] <= 6 else ""
|
||||
|
||||
# 格式化时间
|
||||
time_display = r['time_str']
|
||||
# 添加设置位置标记（群聊/私聊）
|
||||
scope_tag = ""
|
||||
if r.get('roomid'):
|
||||
# 尝试获取群聊名称，如果获取不到就用 roomid
|
||||
room_name = ctx.all_contacts.get(r['roomid']) or r['roomid'][:8]
|
||||
scope_tag = f"[群:{room_name}]"
|
||||
else:
|
||||
scope_tag = "[私聊]"
|
||||
|
||||
if r['type'] == 'once':
|
||||
# 一次性提醒显示完整日期时间
|
||||
time_display = f"{scope_tag}{r['time_str']} (一次性)"
|
||||
elif r['type'] == 'daily':
|
||||
time_display = f"{scope_tag}每天 {r['time_str']}"
|
||||
elif r['type'] == 'weekly':
|
||||
if 0 <= r.get('weekday', -1) <= 6:
|
||||
time_display = f"{scope_tag}每周{weekdays[r['weekday']]} {r['time_str']}"
|
||||
else:
|
||||
time_display = f"{scope_tag}每周 {r['time_str']}"
|
||||
|
||||
reply_parts.append(
|
||||
f"{i+1}. [ID: {r['id'][:6]}] {time_display}: {r['content']}"
|
||||
)
|
||||
ctx.send_text("\n".join(reply_parts), at_list)
|
||||
|
||||
return True
|
||||
|
||||
def handle_delete_reminder(ctx: 'MessageContext', match: Optional[Match]) -> bool:
|
||||
"""处理删除提醒命令（支持群聊和私聊），通过 AI 理解用户意图并执行操作。"""
|
||||
# 1. 获取用户输入的完整内容
|
||||
raw_text = ctx.msg.content.strip()
|
||||
|
||||
# 2. 检查 ReminderManager 是否存在
|
||||
if not hasattr(ctx.robot, 'reminder_manager'):
|
||||
# 这个检查需要保留，是内部依赖
|
||||
ctx.send_text("❌ 内部错误：提醒管理器未初始化。", ctx.msg.sender if ctx.is_group else "")
|
||||
return True # 确实是想处理，但内部错误，返回 True
|
||||
|
||||
# 在群聊中@用户
|
||||
at_list = ctx.msg.sender if ctx.is_group else ""
|
||||
|
||||
# --- 核心流程：直接使用 AI 分析 ---
|
||||
|
||||
# 3. 获取用户的所有提醒作为 AI 的上下文
|
||||
reminders = ctx.robot.reminder_manager.list_reminders(ctx.msg.sender)
|
||||
if not reminders:
|
||||
# 如果用户没有任何提醒，直接告知
|
||||
ctx.send_text("您当前没有任何提醒可供删除。", at_list)
|
||||
return True
|
||||
|
||||
# 将提醒列表转换为 JSON 字符串给 AI 参考
|
||||
try:
|
||||
reminders_json_str = json.dumps(reminders, ensure_ascii=False, indent=2)
|
||||
except Exception as e:
|
||||
ctx.send_text("❌ 内部错误：准备数据给 AI 时出错。", at_list)
|
||||
if ctx.logger: ctx.logger.error(f"序列化提醒列表失败: {e}", exc_info=True)
|
||||
return True
|
||||
|
||||
# 4. 构造 AI Prompt (与之前相同，AI 需要能处理所有情况)
|
||||
# 注意：确保 prompt 中的 {{ 和 }} 转义正确
|
||||
sys_prompt = """
|
||||
你是提醒删除助手。用户会提出删除提醒的请求。我会提供用户的**完整请求原文**，以及一个包含该用户所有当前提醒的 JSON 列表。
|
||||
|
||||
你的任务是：根据用户请求和提醒列表，判断用户的意图，并确定要删除哪些提醒。用户可能要求删除特定提醒（通过描述内容、时间、ID等），也可能要求删除所有提醒。
|
||||
|
||||
**必须严格**按照以下几种 JSON 格式之一返回结果：
|
||||
|
||||
1. **删除特定提醒:** 如果你能明确匹配到一个或多个特定提醒，返回：
|
||||
```json
|
||||
{{
|
||||
"action": "delete_specific",
|
||||
"ids": ["<full_reminder_id_1>", "<full_reminder_id_2>", ...]
|
||||
}}
|
||||
```
|
||||
(`ids` 列表中包含所有匹配到的提醒的 **完整 ID**)
|
||||
|
||||
2. **删除所有提醒:** 如果用户明确表达了删除所有/全部提醒的意图，返回：
|
||||
```json
|
||||
{{
|
||||
"action": "delete_all"
|
||||
}}
|
||||
```
|
||||
|
||||
3. **需要澄清:** 如果用户描述模糊，匹配到多个可能的提醒，无法确定具体是哪个，返回：
|
||||
```json
|
||||
{{
|
||||
"action": "clarify",
|
||||
"message": "抱歉，您的描述可能匹配多个提醒，请问您想删除哪一个？（建议使用 ID 精确删除）",
|
||||
"options": [ {{ "id": "id_prefix_1...", "description": "提醒1的简短描述(如: 周一 09:00 开会)" }}, ... ]
|
||||
}}
|
||||
```
|
||||
(`message` 是给用户的提示，`options` 包含可能的选项及其简短描述和 ID 前缀)
|
||||
|
||||
4. **未找到:** 如果在列表中找不到任何与用户描述匹配的提醒，返回：
|
||||
```json
|
||||
{{
|
||||
"action": "not_found",
|
||||
"message": "抱歉，在您的提醒列表中没有找到与您描述匹配的提醒。"
|
||||
}}
|
||||
```
|
||||
|
||||
5. **错误:** 如果处理中遇到问题或无法理解请求，返回：
|
||||
```json
|
||||
{{
|
||||
"action": "error",
|
||||
"message": "抱歉，处理您的删除请求时遇到问题。"
|
||||
}}
|
||||
```
|
||||
|
||||
**重要:**
|
||||
- 仔细分析用户的**完整请求原文**和提供的提醒列表 JSON 进行匹配。
|
||||
- 用户请求中可能直接包含 ID，也需要你能识别并匹配。
|
||||
- 匹配时要综合考虑内容、时间、类型（一次性/每日/每周）等信息。
|
||||
- 如果返回 `delete_specific`，必须提供 **完整** 的 reminder ID。
|
||||
- **只输出 JSON 结构，不要包含任何额外的解释性文字。**
|
||||
|
||||
用户的提醒列表如下 (JSON 格式):
|
||||
{reminders_list_json}
|
||||
|
||||
当前时间（供参考）: {current_datetime}
|
||||
"""
|
||||
current_dt_str = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||
try:
|
||||
# 将用户的自然语言请求和提醒列表JSON传入Prompt
|
||||
formatted_prompt = sys_prompt.format(
|
||||
reminders_list_json=reminders_json_str,
|
||||
current_datetime=current_dt_str
|
||||
)
|
||||
except KeyError as e:
|
||||
ctx.send_text("❌ 内部错误：构建 AI 请求时出错。", at_list)
|
||||
if ctx.logger: ctx.logger.error(f"格式化删除提醒 prompt 失败: {e}，可能是 sys_prompt 中的 {{}} 未正确转义", exc_info=True)
|
||||
return True
|
||||
|
||||
|
||||
# 5. 调用 AI (使用完整的用户原始输入)
|
||||
q_for_ai = f"请根据以下用户完整请求，分析需要删除哪个提醒：\n{raw_text}" # 使用 raw_text
|
||||
try:
|
||||
if not hasattr(ctx, 'chat') or not ctx.chat:
|
||||
raise ValueError("当前上下文中没有可用的AI模型")
|
||||
|
||||
# 实现最多尝试3次解析AI回复的逻辑
|
||||
max_retries = 3
|
||||
retry_count = 0
|
||||
parsed_ai_response = None
|
||||
ai_parsing_success = False
|
||||
|
||||
while retry_count < max_retries and not ai_parsing_success:
|
||||
# 如果是重试，更新提示信息
|
||||
if retry_count > 0:
|
||||
enhanced_prompt = sys_prompt + f"\n\n**重要提示:** 这是第{retry_count+1}次尝试。你之前的回复格式有误，无法被解析为有效的JSON。请确保你的回复仅包含有效的JSON对象，没有其他任何文字。"
|
||||
try:
|
||||
formatted_prompt = enhanced_prompt.format(
|
||||
reminders_list_json=reminders_json_str,
|
||||
current_datetime=current_dt_str
|
||||
)
|
||||
except Exception as e:
|
||||
ctx.send_text("❌ 内部错误：构建重试请求时出错。", at_list)
|
||||
if ctx.logger: ctx.logger.error(f"格式化重试 prompt 失败: {e}", exc_info=True)
|
||||
return True
|
||||
|
||||
# 在重试时提供更明确的信息
|
||||
retry_q = f"请再次分析以下删除提醒请求，并返回严格的JSON格式(第{retry_count+1}次尝试):\n{raw_text}"
|
||||
q_for_ai = retry_q
|
||||
|
||||
# 获取AI回答
|
||||
ai_response = ctx.chat.get_answer(q_for_ai, ctx.get_receiver(), system_prompt_override=formatted_prompt)
|
||||
|
||||
# 6. 解析 AI 的 JSON 回复
|
||||
json_str = None
|
||||
json_match_obj = re.search(r'\{.*\}', ai_response, re.DOTALL)
|
||||
if json_match_obj:
|
||||
json_str = json_match_obj.group(0)
|
||||
else:
|
||||
json_str = ai_response
|
||||
|
||||
try:
|
||||
parsed_ai_response = json.loads(json_str)
|
||||
if not isinstance(parsed_ai_response, dict) or "action" not in parsed_ai_response:
|
||||
raise ValueError("AI 返回的 JSON 格式不符合预期（缺少 action 字段）")
|
||||
|
||||
# 如果能到这里，说明解析成功
|
||||
ai_parsing_success = True
|
||||
|
||||
except (json.JSONDecodeError, ValueError) as e:
|
||||
# JSON解析失败
|
||||
retry_count += 1
|
||||
if ctx.logger:
|
||||
ctx.logger.warning(f"AI 删除提醒 JSON 解析失败(第{retry_count}次尝试): {ai_response}, 错误: {str(e)}")
|
||||
|
||||
if retry_count >= max_retries:
|
||||
# 达到最大重试次数，返回错误
|
||||
ctx.send_text(f"❌ 抱歉，无法理解您的删除提醒请求。请尝试换一种方式表达，或使用提醒ID进行精确删除。", at_list)
|
||||
if ctx.logger: ctx.logger.error(f"解析AI删除提醒回复失败，已达到最大重试次数({max_retries}): {ai_response}")
|
||||
return True
|
||||
# 否则继续下一次循环重试
|
||||
|
||||
# 7. 根据 AI 指令执行操作 (与之前相同)
|
||||
action = parsed_ai_response.get("action")
|
||||
|
||||
if action == "delete_specific":
|
||||
reminder_ids_to_delete = parsed_ai_response.get("ids", [])
|
||||
if not reminder_ids_to_delete or not isinstance(reminder_ids_to_delete, list):
|
||||
ctx.send_text("❌ AI 指示删除特定提醒，但未提供有效的 ID 列表。", at_list)
|
||||
return True
|
||||
|
||||
delete_results = []
|
||||
successful_deletes = 0
|
||||
deleted_descriptions = []
|
||||
|
||||
for r_id in reminder_ids_to_delete:
|
||||
original_reminder = next((r for r in reminders if r['id'] == r_id), None)
|
||||
desc = f"ID:{r_id[:6]}..."
|
||||
if original_reminder:
|
||||
desc = f"ID:{r_id[:6]}... 内容: \"{original_reminder['content'][:20]}...\""
|
||||
|
||||
success, message = ctx.robot.reminder_manager.delete_reminder(ctx.msg.sender, r_id)
|
||||
delete_results.append({"id": r_id, "success": success, "message": message, "description": desc})
|
||||
if success:
|
||||
successful_deletes += 1
|
||||
deleted_descriptions.append(desc)
|
||||
|
||||
if successful_deletes == len(reminder_ids_to_delete):
|
||||
reply_msg = f"✅ 已删除 {successful_deletes} 个提醒:\n" + "\n".join([f"- {d}" for d in deleted_descriptions])
|
||||
elif successful_deletes > 0:
|
||||
reply_msg = f"⚠️ 部分提醒删除完成 ({successful_deletes}/{len(reminder_ids_to_delete)}):\n"
|
||||
for res in delete_results:
|
||||
status = "✅ 成功" if res["success"] else f"❌ 失败: {res['message']}"
|
||||
reply_msg += f"- {res['description']}: {status}\n"
|
||||
else:
|
||||
reply_msg = f"❌ 未能删除 AI 指定的提醒。\n"
|
||||
for res in delete_results:
|
||||
reply_msg += f"- {res['description']}: 失败原因: {res['message']}\n"
|
||||
|
||||
ctx.send_text(reply_msg.strip(), at_list)
|
||||
|
||||
elif action == "delete_all":
|
||||
success, message, count = ctx.robot.reminder_manager.delete_all_reminders(ctx.msg.sender)
|
||||
ctx.send_text(message, at_list)
|
||||
|
||||
elif action in ["clarify", "not_found", "error"]:
|
||||
message_to_user = parsed_ai_response.get("message", "抱歉，我没能处理您的请求。")
|
||||
if action == "clarify" and "options" in parsed_ai_response:
|
||||
options_text = "\n可能的选项：\n" + "\n".join([f"- ID: {opt.get('id', 'N/A')} ({opt.get('description', '无描述')})" for opt in parsed_ai_response["options"]])
|
||||
message_to_user += options_text
|
||||
ctx.send_text(message_to_user, at_list)
|
||||
|
||||
else:
|
||||
ctx.send_text("❌ AI 返回了无法理解的指令。", at_list)
|
||||
if ctx.logger: ctx.logger.error(f"AI 删除提醒返回未知 action: {action} - Response: {ai_response}")
|
||||
|
||||
return True # AI 处理流程结束
|
||||
|
||||
except Exception as e: # 捕获 AI 调用和处理过程中的其他顶层错误
|
||||
ctx.send_text(f"❌ 处理删除提醒时发生意外错误。", at_list)
|
||||
if ctx.logger:
|
||||
ctx.logger.error(f"handle_delete_reminder AI 部分顶层错误: {e}", exc_info=True)
|
||||
return True
|
||||
|
||||
Reference in New Issue
Block a user