AstrBot 4.10.1__py3-none-any.whl → 4.10.3__py3-none-any.whl

astrbot/builtin_stars/builtin_commands/commands/alter_cmd.py

@@ -0,0 +1,173 @@
+from astrbot.api import star
+from astrbot.api.event import AstrMessageEvent, MessageChain
+from astrbot.core.star.filter.command import CommandFilter
+from astrbot.core.star.filter.command_group import CommandGroupFilter
+from astrbot.core.star.filter.permission import PermissionTypeFilter
+from astrbot.core.star.star import star_map
+from astrbot.core.star.star_handler import StarHandlerMetadata, star_handlers_registry
+from astrbot.core.utils.command_parser import CommandParserMixin
+
+from .utils.rst_scene import RstScene
+
+
+class AlterCmdCommands(CommandParserMixin):
+    def __init__(self, context: star.Context):
+        self.context = context
+
+    async def update_reset_permission(self, scene_key: str, perm_type: str):
+        """更新reset命令在特定场景下的权限设置"""
+        from astrbot.api import sp
+
+        alter_cmd_cfg = await sp.global_get("alter_cmd", {})
+        plugin_cfg = alter_cmd_cfg.get("astrbot", {})
+        reset_cfg = plugin_cfg.get("reset", {})
+        reset_cfg[scene_key] = perm_type
+        plugin_cfg["reset"] = reset_cfg
+        alter_cmd_cfg["astrbot"] = plugin_cfg
+        await sp.global_put("alter_cmd", alter_cmd_cfg)
+
+    async def alter_cmd(self, event: AstrMessageEvent):
+        token = self.parse_commands(event.message_str)
+        if token.len < 3:
+            await event.send(
+                MessageChain().message(
+                    "该指令用于设置指令或指令组的权限。\n"
+                    "格式: /alter_cmd <cmd_name> <admin/member>\n"
+                    "例1: /alter_cmd c1 admin 将 c1 设为管理员指令\n"
+                    "例2: /alter_cmd g1 c1 admin 将 g1 指令组的 c1 子指令设为管理员指令\n"
+                    "/alter_cmd reset config 打开 reset 权限配置",
+                ),
+            )
+            return
+
+        # 兼容 reset scene 的专门配置
+        cmd_name = token.get(1)
+        cmd_type = token.get(2)
+
+        if cmd_name == "reset" and cmd_type == "config":
+            from astrbot.api import sp
+
+            alter_cmd_cfg = await sp.global_get("alter_cmd", {})
+            plugin_ = alter_cmd_cfg.get("astrbot", {})
+            reset_cfg = plugin_.get("reset", {})
+
+            group_unique_on = reset_cfg.get("group_unique_on", "admin")
+            group_unique_off = reset_cfg.get("group_unique_off", "admin")
+            private = reset_cfg.get("private", "member")
+
+            config_menu = f"""reset命令权限细粒度配置
+                当前配置：
+                1. 群聊+会话隔离开: {group_unique_on}
+                2. 群聊+会话隔离关: {group_unique_off}
+                3. 私聊: {private}
+                修改指令格式：
+                /alter_cmd reset scene <场景编号> <admin/member>
+                例如: /alter_cmd reset scene 2 member"""
+            await event.send(MessageChain().message(config_menu))
+            return
+
+        if cmd_name == "reset" and cmd_type == "scene" and token.len >= 4:
+            scene_num = token.get(3)
+            perm_type = token.get(4)
+
+            if scene_num is None or perm_type is None:
+                await event.send(MessageChain().message("场景编号和权限类型不能为空"))
+                return
+
+            if not scene_num.isdigit() or int(scene_num) < 1 or int(scene_num) > 3:
+                await event.send(
+                    MessageChain().message("场景编号必须是 1-3 之间的数字"),
+                )
+                return
+
+            if perm_type not in ["admin", "member"]:
+                await event.send(
+                    MessageChain().message("权限类型错误，只能是 admin 或 member"),
+                )
+                return
+
+            scene_num = int(scene_num)
+            scene = RstScene.from_index(scene_num)
+            scene_key = scene.key
+
+            await self.update_reset_permission(scene_key, perm_type)
+
+            await event.send(
+                MessageChain().message(
+                    f"已将 reset 命令在{scene.name}场景下的权限设为{perm_type}",
+                ),
+            )
+            return
+
+        if cmd_type not in ["admin", "member"]:
+            await event.send(
+                MessageChain().message("指令类型错误，可选类型有 admin, member"),
+            )
+            return
+
+        # 查找指令
+        cmd_name = " ".join(token.tokens[1:-1])
+        cmd_type = token.get(-1)
+        found_command = None
+        cmd_group = False
+        for handler in star_handlers_registry:
+            assert isinstance(handler, StarHandlerMetadata)
+            for filter_ in handler.event_filters:
+                if isinstance(filter_, CommandFilter):
+                    if filter_.equals(cmd_name):
+                        found_command = handler
+                        break
+                elif isinstance(filter_, CommandGroupFilter):
+                    if filter_.equals(cmd_name):
+                        found_command = handler
+                        cmd_group = True
+                        break
+
+        if not found_command:
+            await event.send(MessageChain().message("未找到该指令"))
+            return
+
+        found_plugin = star_map[found_command.handler_module_path]
+
+        from astrbot.api import sp
+
+        alter_cmd_cfg = await sp.global_get("alter_cmd", {})
+        plugin_ = alter_cmd_cfg.get(found_plugin.name, {})
+        cfg = plugin_.get(found_command.handler_name, {})
+        cfg["permission"] = cmd_type
+        plugin_[found_command.handler_name] = cfg
+        alter_cmd_cfg[found_plugin.name] = plugin_
+
+        await sp.global_put("alter_cmd", alter_cmd_cfg)
+
+        # 注入权限过滤器
+        found_permission_filter = False
+        for filter_ in found_command.event_filters:
+            if isinstance(filter_, PermissionTypeFilter):
+                if cmd_type == "admin":
+                    from astrbot.api.event import filter
+
+                    filter_.permission_type = filter.PermissionType.ADMIN
+                else:
+                    from astrbot.api.event import filter
+
+                    filter_.permission_type = filter.PermissionType.MEMBER
+                found_permission_filter = True
+                break
+        if not found_permission_filter:
+            from astrbot.api.event import filter
+
+            found_command.event_filters.insert(
+                0,
+                PermissionTypeFilter(
+                    filter.PermissionType.ADMIN
+                    if cmd_type == "admin"
+                    else filter.PermissionType.MEMBER,
+                ),
+            )
+        cmd_group_str = "指令组" if cmd_group else "指令"
+        await event.send(
+            MessageChain().message(
+                f"已将「{cmd_name}」{cmd_group_str} 的权限级别调整为 {cmd_type}。",
+            ),
+        )

astrbot/builtin_stars/builtin_commands/commands/conversation.py

@@ -0,0 +1,366 @@
+import datetime
+
+from astrbot.api import sp, star
+from astrbot.api.event import AstrMessageEvent, MessageEventResult
+from astrbot.core.platform.astr_message_event import MessageSession
+from astrbot.core.platform.message_type import MessageType
+
+from .utils.rst_scene import RstScene
+
+THIRD_PARTY_AGENT_RUNNER_KEY = {
+    "dify": "dify_conversation_id",
+    "coze": "coze_conversation_id",
+    "dashscope": "dashscope_conversation_id",
+}
+THIRD_PARTY_AGENT_RUNNER_STR = ", ".join(THIRD_PARTY_AGENT_RUNNER_KEY.keys())
+
+
+class ConversationCommands:
+    def __init__(self, context: star.Context):
+        self.context = context
+
+    async def _get_current_persona_id(self, session_id):
+        curr = await self.context.conversation_manager.get_curr_conversation_id(
+            session_id,
+        )
+        if not curr:
+            return None
+        conv = await self.context.conversation_manager.get_conversation(
+            session_id,
+            curr,
+        )
+        if not conv:
+            return None
+        return conv.persona_id
+
+    async def reset(self, message: AstrMessageEvent):
+        """重置 LLM 会话"""
+        umo = message.unified_msg_origin
+        cfg = self.context.get_config(umo=message.unified_msg_origin)
+        is_unique_session = cfg["platform_settings"]["unique_session"]
+        is_group = bool(message.get_group_id())
+
+        scene = RstScene.get_scene(is_group, is_unique_session)
+
+        alter_cmd_cfg = await sp.get_async("global", "global", "alter_cmd", {})
+        plugin_config = alter_cmd_cfg.get("astrbot", {})
+        reset_cfg = plugin_config.get("reset", {})
+
+        required_perm = reset_cfg.get(
+            scene.key,
+            "admin" if is_group and not is_unique_session else "member",
+        )
+
+        if required_perm == "admin" and message.role != "admin":
+            message.set_result(
+                MessageEventResult().message(
+                    f"在{scene.name}场景下，reset命令需要管理员权限，"
+                    f"您 (ID {message.get_sender_id()}) 不是管理员，无法执行此操作。",
+                ),
+            )
+            return
+
+        agent_runner_type = cfg["provider_settings"]["agent_runner_type"]
+        if agent_runner_type in THIRD_PARTY_AGENT_RUNNER_KEY:
+            await sp.remove_async(
+                scope="umo",
+                scope_id=umo,
+                key=THIRD_PARTY_AGENT_RUNNER_KEY[agent_runner_type],
+            )
+            message.set_result(MessageEventResult().message("重置对话成功。"))
+            return
+
+        if not self.context.get_using_provider(umo):
+            message.set_result(
+                MessageEventResult().message("未找到任何 LLM 提供商。请先配置。"),
+            )
+            return
+
+        cid = await self.context.conversation_manager.get_curr_conversation_id(umo)
+
+        if not cid:
+            message.set_result(
+                MessageEventResult().message(
+                    "当前未处于对话状态，请 /switch 切换或者 /new 创建。",
+                ),
+            )
+            return
+
+        await self.context.conversation_manager.update_conversation(
+            umo,
+            cid,
+            [],
+        )
+
+        ret = "清除聊天历史成功！"
+
+        message.set_extra("_clean_ltm_session", True)
+
+        message.set_result(MessageEventResult().message(ret))
+
+    async def his(self, message: AstrMessageEvent, page: int = 1):
+        """查看对话记录"""
+        if not self.context.get_using_provider(message.unified_msg_origin):
+            message.set_result(
+                MessageEventResult().message("未找到任何 LLM 提供商。请先配置。"),
+            )
+            return
+
+        size_per_page = 6
+
+        conv_mgr = self.context.conversation_manager
+        umo = message.unified_msg_origin
+        session_curr_cid = await conv_mgr.get_curr_conversation_id(umo)
+
+        if not session_curr_cid:
+            session_curr_cid = await conv_mgr.new_conversation(
+                umo,
+                message.get_platform_id(),
+            )
+
+        contexts, total_pages = await conv_mgr.get_human_readable_context(
+            umo,
+            session_curr_cid,
+            page,
+            size_per_page,
+        )
+
+        parts = []
+        for context in contexts:
+            if len(context) > 150:
+                context = context[:150] + "..."
+            parts.append(f"{context}\n")
+
+        history = "".join(parts)
+        ret = (
+            f"当前对话历史记录："
+            f"{history or '无历史记录'}\n\n"
+            f"第 {page} 页 | 共 {total_pages} 页\n"
+            f"*输入 /history 2 跳转到第 2 页"
+        )
+
+        message.set_result(MessageEventResult().message(ret).use_t2i(False))
+
+    async def convs(self, message: AstrMessageEvent, page: int = 1):
+        """查看对话列表"""
+        cfg = self.context.get_config(umo=message.unified_msg_origin)
+        agent_runner_type = cfg["provider_settings"]["agent_runner_type"]
+        if agent_runner_type in THIRD_PARTY_AGENT_RUNNER_KEY:
+            message.set_result(
+                MessageEventResult().message(
+                    f"{THIRD_PARTY_AGENT_RUNNER_STR} 对话列表功能暂不支持。",
+                ),
+            )
+            return
+
+        size_per_page = 6
+        """获取所有对话列表"""
+        conversations_all = await self.context.conversation_manager.get_conversations(
+            message.unified_msg_origin,
+        )
+        """计算总页数"""
+        total_pages = (len(conversations_all) + size_per_page - 1) // size_per_page
+        """确保页码有效"""
+        page = max(1, min(page, total_pages))
+        """分页处理"""
+        start_idx = (page - 1) * size_per_page
+        end_idx = start_idx + size_per_page
+        conversations_paged = conversations_all[start_idx:end_idx]
+
+        parts = ["对话列表：\n---\n"]
+        """全局序号从当前页的第一个开始"""
+        global_index = start_idx + 1
+
+        """生成所有对话的标题字典"""
+        _titles = {}
+        for conv in conversations_all:
+            title = conv.title if conv.title else "新对话"
+            _titles[conv.cid] = title
+
+        """遍历分页后的对话生成列表显示"""
+        for conv in conversations_paged:
+            persona_id = conv.persona_id
+            if not persona_id or persona_id == "[%None]":
+                persona = await self.context.persona_manager.get_default_persona_v3(
+                    umo=message.unified_msg_origin,
+                )
+                persona_id = persona["name"]
+            title = _titles.get(conv.cid, "新对话")
+            parts.append(
+                f"{global_index}. {title}({conv.cid[:4]})\n  人格情景: {persona_id}\n  上次更新: {datetime.datetime.fromtimestamp(conv.updated_at).strftime('%m-%d %H:%M')}\n"
+            )
+            global_index += 1
+
+        parts.append("---\n")
+        ret = "".join(parts)
+        curr_cid = await self.context.conversation_manager.get_curr_conversation_id(
+            message.unified_msg_origin,
+        )
+        if curr_cid:
+            """从所有对话的标题字典中获取标题"""
+            title = _titles.get(curr_cid, "新对话")
+            ret += f"\n当前对话: {title}({curr_cid[:4]})"
+        else:
+            ret += "\n当前对话: 无"
+
+        cfg = self.context.get_config(umo=message.unified_msg_origin)
+        unique_session = cfg["platform_settings"]["unique_session"]
+        if unique_session:
+            ret += "\n会话隔离粒度: 个人"
+        else:
+            ret += "\n会话隔离粒度: 群聊"
+
+        ret += f"\n第 {page} 页 | 共 {total_pages} 页"
+        ret += "\n*输入 /ls 2 跳转到第 2 页"
+
+        message.set_result(MessageEventResult().message(ret).use_t2i(False))
+        return
+
+    async def new_conv(self, message: AstrMessageEvent):
+        """创建新对话"""
+        cfg = self.context.get_config(umo=message.unified_msg_origin)
+        agent_runner_type = cfg["provider_settings"]["agent_runner_type"]
+        if agent_runner_type in THIRD_PARTY_AGENT_RUNNER_KEY:
+            await sp.remove_async(
+                scope="umo",
+                scope_id=message.unified_msg_origin,
+                key=THIRD_PARTY_AGENT_RUNNER_KEY[agent_runner_type],
+            )
+            message.set_result(MessageEventResult().message("已创建新对话。"))
+            return
+
+        cpersona = await self._get_current_persona_id(message.unified_msg_origin)
+        cid = await self.context.conversation_manager.new_conversation(
+            message.unified_msg_origin,
+            message.get_platform_id(),
+            persona_id=cpersona,
+        )
+
+        message.set_extra("_clean_ltm_session", True)
+
+        message.set_result(
+            MessageEventResult().message(f"切换到新对话: 新对话({cid[:4]})。"),
+        )
+
+    async def groupnew_conv(self, message: AstrMessageEvent, sid: str = ""):
+        """创建新群聊对话"""
+        if sid:
+            session = str(
+                MessageSession(
+                    platform_name=message.platform_meta.id,
+                    message_type=MessageType("GroupMessage"),
+                    session_id=sid,
+                ),
+            )
+
+            cpersona = await self._get_current_persona_id(session)
+            cid = await self.context.conversation_manager.new_conversation(
+                session,
+                message.get_platform_id(),
+                persona_id=cpersona,
+            )
+            message.set_result(
+                MessageEventResult().message(
+                    f"群聊 {session} 已切换到新对话: 新对话({cid[:4]})。",
+                ),
+            )
+        else:
+            message.set_result(
+                MessageEventResult().message("请输入群聊 ID。/groupnew 群聊ID。"),
+            )
+
+    async def switch_conv(
+        self,
+        message: AstrMessageEvent,
+        index: int | None = None,
+    ):
+        """通过 /ls 前面的序号切换对话"""
+        if not isinstance(index, int):
+            message.set_result(
+                MessageEventResult().message("类型错误，请输入数字对话序号。"),
+            )
+            return
+
+        if index is None:
+            message.set_result(
+                MessageEventResult().message(
+                    "请输入对话序号。/switch 对话序号。/ls 查看对话 /new 新建对话",
+                ),
+            )
+            return
+        conversations = await self.context.conversation_manager.get_conversations(
+            message.unified_msg_origin,
+        )
+        if index > len(conversations) or index < 1:
+            message.set_result(
+                MessageEventResult().message("对话序号错误，请使用 /ls 查看"),
+            )
+        else:
+            conversation = conversations[index - 1]
+            title = conversation.title if conversation.title else "新对话"
+            await self.context.conversation_manager.switch_conversation(
+                message.unified_msg_origin,
+                conversation.cid,
+            )
+            message.set_result(
+                MessageEventResult().message(
+                    f"切换到对话: {title}({conversation.cid[:4]})。",
+                ),
+            )
+
+    async def rename_conv(self, message: AstrMessageEvent, new_name: str = ""):
+        """重命名对话"""
+        if not new_name:
+            message.set_result(MessageEventResult().message("请输入新的对话名称。"))
+            return
+        await self.context.conversation_manager.update_conversation_title(
+            message.unified_msg_origin,
+            new_name,
+        )
+        message.set_result(MessageEventResult().message("重命名对话成功。"))
+
+    async def del_conv(self, message: AstrMessageEvent):
+        """删除当前对话"""
+        cfg = self.context.get_config(umo=message.unified_msg_origin)
+        is_unique_session = cfg["platform_settings"]["unique_session"]
+        if message.get_group_id() and not is_unique_session and message.role != "admin":
+            # 群聊，没开独立会话，发送人不是管理员
+            message.set_result(
+                MessageEventResult().message(
+                    f"会话处于群聊，并且未开启独立会话，并且您 (ID {message.get_sender_id()}) 不是管理员，因此没有权限删除当前对话。",
+                ),
+            )
+            return
+
+        agent_runner_type = cfg["provider_settings"]["agent_runner_type"]
+        if agent_runner_type in THIRD_PARTY_AGENT_RUNNER_KEY:
+            await sp.remove_async(
+                scope="umo",
+                scope_id=message.unified_msg_origin,
+                key=THIRD_PARTY_AGENT_RUNNER_KEY[agent_runner_type],
+            )
+            message.set_result(MessageEventResult().message("重置对话成功。"))
+            return
+
+        session_curr_cid = (
+            await self.context.conversation_manager.get_curr_conversation_id(
+                message.unified_msg_origin,
+            )
+        )
+
+        if not session_curr_cid:
+            message.set_result(
+                MessageEventResult().message(
+                    "当前未处于对话状态，请 /switch 序号 切换或 /new 创建。",
+                ),
+            )
+            return
+
+        await self.context.conversation_manager.delete_conversation(
+            message.unified_msg_origin,
+            session_curr_cid,
+        )
+
+        ret = "删除当前对话成功。不再处于对话状态，使用 /switch 序号 切换到其他对话或 /new 创建。"
+        message.set_extra("_clean_ltm_session", True)
+        message.set_result(MessageEventResult().message(ret))

astrbot/builtin_stars/builtin_commands/commands/help.py

@@ -0,0 +1,88 @@
+import aiohttp
+
+from astrbot.api import star
+from astrbot.api.event import AstrMessageEvent, MessageEventResult
+from astrbot.core.config.default import VERSION
+from astrbot.core.star import command_management
+from astrbot.core.utils.io import get_dashboard_version
+
+
+class HelpCommand:
+    def __init__(self, context: star.Context):
+        self.context = context
+
+    async def _query_astrbot_notice(self):
+        try:
+            async with aiohttp.ClientSession(trust_env=True) as session:
+                async with session.get(
+                    "https://astrbot.app/notice.json",
+                    timeout=2,
+                ) as resp:
+                    return (await resp.json())["notice"]
+        except BaseException:
+            return ""
+
+    async def _build_reserved_command_lines(self) -> list[str]:
+        """
+        使用实时指令配置生成内置指令清单，确保重命名/禁用后与实际生效状态保持一致。
+        """
+        try:
+            commands = await command_management.list_commands()
+        except BaseException:
+            return []
+
+        lines: list[str] = []
+        hidden_commands = {"set", "unset", "websearch"}
+
+        def walk(items: list[dict], indent: int = 0):
+            for item in items:
+                if not item.get("reserved") or not item.get("enabled"):
+                    continue
+                # 仅展示顶级指令或指令组
+                if item.get("type") == "sub_command":
+                    continue
+                if item.get("parent_signature"):
+                    continue
+
+                effective = (
+                    item.get("effective_command")
+                    or item.get("original_command")
+                    or item.get("handler_name")
+                )
+                if not effective:
+                    continue
+                if effective in hidden_commands:
+                    continue
+
+                description = item.get("description") or ""
+                desc_text = f" - {description}" if description else ""
+                indent_prefix = "  " * indent
+                lines.append(f"{indent_prefix}/{effective}{desc_text}")
+
+        walk(commands)
+        return lines
+
+    async def help(self, event: AstrMessageEvent):
+        """查看帮助"""
+        notice = ""
+        try:
+            notice = await self._query_astrbot_notice()
+        except BaseException:
+            pass
+
+        dashboard_version = await get_dashboard_version()
+        command_lines = await self._build_reserved_command_lines()
+        commands_section = (
+            "\n".join(command_lines) if command_lines else "暂无启用的内置指令"
+        )
+
+        msg_parts = [
+            f"AstrBot v{VERSION}(WebUI: {dashboard_version})",
+            "内置指令:",
+            commands_section,
+        ]
+        if notice:
+            msg_parts.append(notice)
+        msg = "\n".join(msg_parts)
+
+        event.set_result(MessageEventResult().message(msg).use_t2i(False))

astrbot/builtin_stars/builtin_commands/commands/llm.py

@@ -0,0 +1,20 @@
+from astrbot.api import star
+from astrbot.api.event import AstrMessageEvent, MessageChain
+
+
+class LLMCommands:
+    def __init__(self, context: star.Context):
+        self.context = context
+
+    async def llm(self, event: AstrMessageEvent):
+        """开启/关闭 LLM"""
+        cfg = self.context.get_config(umo=event.unified_msg_origin)
+        enable = cfg["provider_settings"].get("enable", True)
+        if enable:
+            cfg["provider_settings"]["enable"] = False
+            status = "关闭"
+        else:
+            cfg["provider_settings"]["enable"] = True
+            status = "开启"
+        cfg.save_config()
+        await event.send(MessageChain().message(f"{status} LLM 聊天功能。"))