Undefined-bot 2.1.0__py3-none-any.whl

Undefined/skills/tools/render_latex/handler.py

@@ -0,0 +1,78 @@
+from typing import Any, Dict
+import logging
+from pathlib import Path
+import uuid
+import matplotlib.pyplot as plt
+import matplotlib
+
+logger = logging.getLogger(__name__)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    content = args.get("content", "")
+    target_id = args.get("target_id")
+    message_type = args.get("message_type")
+
+    if not content:
+        return "内容不能为空"
+    if not target_id:
+        return "目标 ID 不能为空"
+    if not message_type:
+        return "消息类型不能为空"
+    if message_type not in ["group", "private"]:
+        return "消息类型必须是 group 或 private"
+
+    try:
+        filename = f"render_{uuid.uuid4().hex[:16]}.png"
+        filepath = Path.cwd() / "img" / filename
+        filepath.parent.mkdir(exist_ok=True)
+
+        matplotlib.use("Agg")
+
+        fig, ax = plt.subplots(figsize=(10, 6))
+        ax.axis("off")
+
+        ax.text(
+            0.5,
+            0.5,
+            content,
+            transform=ax.transAxes,
+            fontsize=12,
+            verticalalignment="center",
+            horizontalalignment="center",
+            usetex=True,
+            wrap=True,
+        )
+
+        plt.tight_layout()
+        plt.savefig(filepath, dpi=150, bbox_inches="tight", pad_inches=0.1)
+        plt.close(fig)
+
+        send_image_callback = context.get("send_image_callback")
+        sender = context.get("sender")
+
+        if sender:
+            import os
+
+            abs_path = os.path.abspath(str(filepath))
+            message = f"[CQ:image,file={abs_path}]"
+
+            if message_type == "group":
+                await sender.send_group_message(int(target_id), message)
+            elif message_type == "private":
+                await sender.send_private_message(int(target_id), message)
+
+            return f"LaTeX 图片已渲染并发送到 {message_type} {target_id}"
+
+        elif send_image_callback:
+            await send_image_callback(target_id, message_type, str(filepath))
+            return f"LaTeX 图片已渲染并发送到 {message_type} {target_id}"
+        else:
+            return "发送图片回调未设置"
+
+    except ImportError as e:
+        missing_pkg = str(e).split("'")[1] if "'" in str(e) else "未知包"
+        return f"渲染失败：缺少依赖包 {missing_pkg}，请运行: uv add {missing_pkg}"
+    except Exception as e:
+        logger.exception(f"渲染并发送 LaTeX 图片失败: {e}")
+        return f"渲染失败: {e}"

Undefined/skills/tools/render_markdown/config.json

@@ -0,0 +1,26 @@
+{
+    "type": "function",
+    "function": {
+        "name": "render_markdown",
+        "description": "将 Markdown 文本渲染为图片并发送到指定目标（群聊或私聊）。支持标准 Markdown 格式，包括标题、列表、代码块、表格等。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "content": {
+                    "type": "string",
+                    "description": "要渲染的 Markdown 内容。支持标准 Markdown 格式。"
+                },
+                "target_id": {
+                    "type": "integer",
+                    "description": "目标 ID（群号或用户 QQ 号）"
+                },
+                "message_type": {
+                    "type": "string",
+                    "description": "消息类型",
+                    "enum": ["group", "private"]
+                }
+            },
+            "required": ["content", "target_id", "message_type"]
+        }
+    }
+}

Undefined/skills/tools/render_markdown/handler.py

@@ -0,0 +1,63 @@
+from typing import Any, Dict
+import logging
+from pathlib import Path
+import uuid
+from Undefined.render import render_markdown_to_html, render_html_to_image
+
+logger = logging.getLogger(__name__)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    content = args.get("content", "")
+    target_id = args.get("target_id")
+    message_type = args.get("message_type")
+
+    if not content:
+        return "内容不能为空"
+    if not target_id:
+        return "目标 ID 不能为空"
+    if not message_type:
+        return "消息类型不能为空"
+    if message_type not in ["group", "private"]:
+        return "消息类型必须是 group 或 private"
+
+    try:
+        filename = f"render_{uuid.uuid4().hex[:16]}.png"
+        filepath = Path.cwd() / "img" / filename
+        filepath.parent.mkdir(exist_ok=True)
+
+        try:
+            html_content = await render_markdown_to_html(content)
+            await render_html_to_image(html_content, str(filepath))
+        except Exception as e:
+            logger.exception(f"Markdown 渲染失败: {e}")
+            return f"Markdown 渲染失败: {e}"
+
+        send_image_callback = context.get("send_image_callback")
+        sender = context.get("sender")
+
+        if sender:
+            import os
+
+            abs_path = os.path.abspath(str(filepath))
+            message = f"[CQ:image,file={abs_path}]"
+
+            if message_type == "group":
+                await sender.send_group_message(int(target_id), message)
+            elif message_type == "private":
+                await sender.send_private_message(int(target_id), message)
+
+            return f"Markdown 图片已渲染并发送到 {message_type} {target_id}"
+
+        elif send_image_callback:
+            await send_image_callback(target_id, message_type, str(filepath))
+            return f"Markdown 图片已渲染并发送到 {message_type} {target_id}"
+        else:
+            return "发送图片回调未设置"
+
+    except ImportError as e:
+        missing_pkg = str(e).split("'")[1] if "'" in str(e) else "未知包"
+        return f"渲染失败：缺少依赖包 {missing_pkg}，请运行: uv add {missing_pkg}"
+    except Exception as e:
+        logger.exception(f"渲染并发送 Markdown 图片失败: {e}")
+        return f"渲染失败: {e}"

Undefined/skills/tools/save_memory/config.json

@@ -0,0 +1,17 @@
+{
+    "type": "function",
+    "function": {
+        "name": "save_memory",
+        "description": "保存一条记忆。当你觉得某些信息值得记住（比如用户的偏好、重要的事情、约定等），可以调用此工具保存。就事论事，就人论人，不做会话隔离。上限100条，超出上限会自动移除最旧的记忆。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "fact": {
+                    "type": "string",
+                    "description": "要记住的内容（简短、具体、有价值的信息）"
+                }
+            },
+            "required": ["fact"]
+        }
+    }
+}

Undefined/skills/tools/save_memory/handler.py

@@ -0,0 +1,17 @@
+from typing import Any, Dict
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    fact = args.get("fact", "")
+    if not fact:
+        return "记忆内容不能为空"
+    
+    memory_storage = context.get("memory_storage")
+    
+    if memory_storage:
+        success = memory_storage.add(fact)
+        if success:
+            return f"已保存记忆: {fact[:50]}..."
+        else:
+            return "保存失败（可能已存在相同记忆）"
+    else:
+        return "记忆存储未初始化"

Undefined/skills/tools/send_message/config.json

@@ -0,0 +1,21 @@
+{
+    "type": "function",
+    "function": {
+        "name": "send_message",
+        "description": "发送群聊消息。可以在回答过程中多次调用，用于发送中间结果、进展信息或最终答案。最后必须调用 end 工具结束对话。注意：群号已自动设置，无需传递 group_id 参数。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "message": {
+                    "type": "string",
+                    "description": "要发送的消息内容"
+                },
+                "at_user": {
+                    "type": "integer",
+                    "description": "要 @ 的用户 QQ 号（可选）"
+                }
+            },
+            "required": ["message"]
+        }
+    }
+}

Undefined/skills/tools/send_message/handler.py

@@ -0,0 +1,60 @@
+from typing import Any, Dict
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    message = args.get("message", "")
+    if not message:
+        logger.warning("[发送消息] 收到空消息请求")
+        return "消息内容不能为空"
+
+    message = message.replace("\\", "")
+
+    # 如果可用，使用 context.recent_replies 检查重复
+    recent_replies = context.get("recent_replies")
+    if recent_replies is not None and message in recent_replies:
+        logger.info(f"[发送消息] 检测到重复消息内容: {message[:50]}...")
+
+    at_user = args.get("at_user")
+    send_message_callback = context.get("send_message_callback")
+    sender = context.get("sender")
+
+    # 优先使用 sender 接口
+    if sender:
+        ai_client = context.get("ai_client")
+        group_id = ai_client.current_group_id if ai_client else None
+
+        if group_id:
+            logger.info(f"[发送消息] 准备发送到群 {group_id}: {message[:100]}")
+            if at_user:
+                logger.debug(f"[发送消息] 同时 @ 用户: {at_user}")
+                message = f"[CQ:at,qq={at_user}] {message}"
+            try:
+                await sender.send_group_message(group_id, message)
+                if recent_replies is not None:
+                    recent_replies.append(message)
+                return "消息已发送"
+            except Exception as e:
+                logger.exception(f"[发送消息] 发送到群 {group_id} 失败: {e}")
+                return f"发送失败: {e}"
+        elif send_message_callback:
+            logger.info(f"[发送消息] 无法确定群ID，尝试使用回调发送: {message[:100]}")
+            await send_message_callback(message, at_user)
+            if recent_replies is not None:
+                recent_replies.append(message)
+            return "消息已发送"
+        else:
+            logger.error("[发送消息] 发送失败：无法确定群组 ID 且无回调可用")
+            return "发送失败：无法确定群组 ID"
+
+    elif send_message_callback:
+        logger.info(f"[发送消息] 使用回调发送私聊或默认消息: {message[:100]}")
+        await send_message_callback(message, at_user)
+        if recent_replies is not None:
+            recent_replies.append(message)
+        return "消息已发送"
+    else:
+        logger.error("[发送消息] 发送消息回调和 sender 均未设置")
+        return "发送消息回调未设置"

Undefined/skills/tools/send_private_message/config.json

@@ -0,0 +1,21 @@
+{
+    "type": "function",
+    "function": {
+        "name": "send_private_message",
+        "description": "发送私聊消息给指定用户。只能在群聊中使用，用于给特定用户发送私聊消息。最后必须调用 end 工具结束对话。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "user_id": {
+                    "type": "integer",
+                    "description": "目标用户的 QQ 号"
+                },
+                "message": {
+                    "type": "string",
+                    "description": "要发送的私聊消息内容"
+                }
+            },
+            "required": ["user_id", "message"]
+        }
+    }
+}

Undefined/skills/tools/send_private_message/handler.py

@@ -0,0 +1,35 @@
+from typing import Any, Dict
+import logging
+
+logger = logging.getLogger(__name__)
+
+async def execute(args: Dict[str, Any], context: Dict[str, Any]) -> str:
+    user_id = args.get("user_id")
+    message = args.get("message", "")
+    
+    if not user_id:
+        return "目标用户 QQ 号不能为空"
+    if not message:
+        return "消息内容不能为空"
+    
+    message = message.replace("\\", "")
+    
+    recent_replies = context.get("recent_replies")
+    if recent_replies is not None and message in recent_replies:
+        logger.info(f"发送了重复私聊消息（已移除屏蔽）: {message[:50]}...")
+    
+    send_private_message_callback = context.get("send_private_message_callback")
+    sender = context.get("sender")
+    
+    if sender:
+        await sender.send_private_message(user_id, message)
+        if recent_replies is not None:
+            recent_replies.append(message)
+        return f"私聊消息已发送给用户 {user_id}"
+    elif send_private_message_callback:
+        await send_private_message_callback(user_id, message)
+        if recent_replies is not None:
+            recent_replies.append(message)
+        return f"私聊消息已发送给用户 {user_id}"
+    else:
+        return "私聊发送回调未设置"

Undefined/utils/common.py

@@ -0,0 +1,186 @@
+"""通用工具函数"""
+
+import re
+import logging
+from typing import Any, Awaitable, Callable
+
+logger = logging.getLogger(__name__)
+
+
+def extract_text(message_content: list[dict[str, Any]], bot_qq: int = 0) -> str:
+    """提取消息中的文本内容
+
+    参数:
+        message_content: 消息内容列表
+        bot_qq: 机器人 QQ 号（用于过滤 @ 机器人的内容），默认为 0（不过滤）
+
+    返回:
+        提取的文本
+    """
+    texts: list[str] = []
+    for segment in message_content:
+        type_ = segment.get("type", "")
+        data = segment.get("data", {})
+
+        if type_ == "text":
+            texts.append(data.get("text", ""))
+        elif type_ == "at":
+            qq = data.get("qq", "")
+            # 如果指定了 bot_qq 且 @ 的是 bot，则不显示 @
+            if bot_qq and str(qq) == str(bot_qq):
+                continue
+            texts.append(f"[@ {qq}]")
+        elif type_ == "image":
+            file = data.get("file", "") or data.get("url", "")
+            texts.append(f"[图片: {file}]")
+
+    return "".join(texts).strip()
+
+
+async def parse_message_content_for_history(
+    message_content: list[dict[str, Any]],
+    bot_qq: int,
+    get_msg_func: Callable[[int], Awaitable[dict[str, Any] | None]] | None = None,
+) -> str:
+    """解析消息内容用于历史记录（支持合并转发和 @ 格式化）
+
+    参数:
+        message_content: 消息内容列表
+        bot_qq: 机器人 QQ 号
+        get_msg_func: 获取消息详情的异步函数（可选，用于处理回复引用）
+
+    返回:
+        解析后的文本
+    """
+    texts: list[str] = []
+    for segment in message_content:
+        type_ = segment.get("type")
+        data = segment.get("data", {})
+
+        if type_ == "text":
+            texts.append(data.get("text", ""))
+
+        elif type_ == "at":
+            qq = data.get("qq", "")
+            # 仅当 @ 的不是机器人时才显示，且使用指定格式
+            if str(qq) != str(bot_qq):
+                texts.append(f"[@ {qq}]")
+
+        elif type_ == "image":
+            file = data.get("file", "") or data.get("url", "")
+            texts.append(f"[图片: {file}]")
+
+        elif type_ == "forward":
+            msg_id = data.get("id")
+            if msg_id:
+                texts.append(f"[合并转发: {msg_id}]")
+
+        elif type_ == "reply":
+            msg_id = data.get("id")
+            if msg_id and get_msg_func:
+                try:
+                    # 尝试获取引用的消息内容
+                    reply_msg = await get_msg_func(int(msg_id))
+                    if reply_msg:
+                        sender = reply_msg.get("sender", {}).get("nickname", "未知")
+                        content = reply_msg.get("message", [])
+                        # 使用 extract_text 解析引用内容
+                        quote_text = extract_text(content, bot_qq)
+                        texts.append(f'<quote sender="{sender}">{quote_text}</quote>\n')
+                except Exception as e:
+                    logger.warning(f"获取回复消息失败: {e}")
+
+        elif type_ == "face":
+            texts.append("[表情]")
+
+    return "".join(texts).strip()
+
+
+def message_to_segments(message: str) -> list[dict[str, Any]]:
+    """将包含 CQ 码的字符串转换为 OneBot 消息段数组
+
+    参数:
+        message: 包含 CQ 码的字符串
+
+    返回:
+        消息段列表
+    """
+    segments = []
+    # 匹配 CQ 码的正则
+    # [CQ:type,arg1=val1,arg2=val2]
+    cq_pattern = re.compile(r"\[CQ:([a-zA-Z0-9_-]+),?([^\]]*)\]")
+
+    last_pos = 0
+    for match in cq_pattern.finditer(message):
+        # 处理 CQ 码之前的文本
+        text_part = message[last_pos : match.start()]
+        if text_part:
+            segments.append({"type": "text", "data": {"text": text_part}})
+
+        # 处理 CQ 码
+        cq_type = match.group(1)
+        cq_args_str = match.group(2)
+
+        # 解析参数
+        data = {}
+        if cq_args_str:
+            for arg_pair in cq_args_str.split(","):
+                if "=" in arg_pair:
+                    k, v = arg_pair.split("=", 1)
+                    # 注意：这里假设输入的 CQ 码已经是经过 OneBot 转义的格式
+                    data[k.strip()] = v.strip()
+
+        segments.append({"type": cq_type, "data": data})
+        last_pos = match.end()
+
+    # 处理剩余的文本
+    remaining_text = message[last_pos:]
+    if remaining_text:
+        segments.append({"type": "text", "data": {"text": remaining_text}})
+
+    return segments
+
+
+def matches_xinliweiyuan(text: str) -> bool:
+    """判断文本是否匹配心理委员触发规则
+
+    规则:
+    1. 单独“心理委员” (可选加标点/空格)
+    2. 前后（同时仅1）添加 5 个字以内（标点/空格不计入字数）
+    """
+    keyword = "心理委员"
+    if keyword not in text:
+        return False
+
+    # 分割文本，找到关键字的位置
+    parts = text.split(keyword)
+    # 如果出现多次关键词，只要其中一个位置满足条件即可触发
+    # 但为了简单和符合直觉，我们检查是否【任何】一种分割方式满足条件
+    # 通常文本里只会有一个“心理委员”用于触发
+
+    # 标点符号和空白字符正则
+    # \s 匹配空白，[^\w\s] 在大多数情况下匹配标点（但在 Python 3 中 \w 包含中文）
+    # 我们直接定义非“字”的模式：空白、常见标点
+    punc_pattern = r'[ \t\n\r\f\v\s!"#$%&\'()*+,\-./:;<=>?@\[\\\]^_`{|}~，。！？、；：""\'\'（）【】「」《》—…·]'
+
+    def count_real_chars(s: str) -> int:
+        """移除标点和空格后的长度"""
+        return len(re.sub(punc_pattern, "", s))
+
+    # 遍历所有可能的分割（以防文本中有多个“心理委员”）
+    for i in range(len(parts) - 1):
+        prefix = keyword.join(parts[: i + 1])
+        suffix = keyword.join(parts[i + 1 :])
+
+        prefix_count = count_real_chars(prefix)
+        suffix_count = count_real_chars(suffix)
+
+        # 同时仅1：不能前后都有字（标点不计）
+        if prefix_count > 0 and suffix_count > 0:
+            continue
+
+        # 字数限制：添加的部分总字数 <= 5
+        if (prefix_count + suffix_count) <= 5:
+            return True
+
+    return False