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

astrbot/builtin_stars/builtin_commands/commands/tool.py

@@ -0,0 +1,31 @@
+from astrbot.api import star
+from astrbot.api.event import AstrMessageEvent, MessageEventResult
+
+
+class ToolCommands:
+    def __init__(self, context: star.Context):
+        self.context = context
+
+    async def tool_ls(self, event: AstrMessageEvent):
+        """查看函数工具列表"""
+        event.set_result(
+            MessageEventResult().message("tool 指令在 AstrBot v4.0.0 已经被移除。"),
+        )
+
+    async def tool_on(self, event: AstrMessageEvent, tool_name: str = ""):
+        """启用一个函数工具"""
+        event.set_result(
+            MessageEventResult().message("tool 指令在 AstrBot v4.0.0 已经被移除。"),
+        )
+
+    async def tool_off(self, event: AstrMessageEvent, tool_name: str = ""):
+        """停用一个函数工具"""
+        event.set_result(
+            MessageEventResult().message("tool 指令在 AstrBot v4.0.0 已经被移除。"),
+        )
+
+    async def tool_all_off(self, event: AstrMessageEvent):
+        """停用所有函数工具"""
+        event.set_result(
+            MessageEventResult().message("tool 指令在 AstrBot v4.0.0 已经被移除。"),
+        )

astrbot/builtin_stars/builtin_commands/commands/tts.py

@@ -0,0 +1,36 @@
+"""文本转语音命令"""
+
+from astrbot.api import star
+from astrbot.api.event import AstrMessageEvent, MessageEventResult
+from astrbot.core.star.session_llm_manager import SessionServiceManager
+
+
+class TTSCommand:
+    """文本转语音命令类"""
+
+    def __init__(self, context: star.Context):
+        self.context = context
+
+    async def tts(self, event: AstrMessageEvent):
+        """开关文本转语音（会话级别）"""
+        umo = event.unified_msg_origin
+        ses_tts = SessionServiceManager.is_tts_enabled_for_session(umo)
+        cfg = self.context.get_config(umo=umo)
+        tts_enable = cfg["provider_tts_settings"]["enable"]
+
+        # 切换状态
+        new_status = not ses_tts
+        SessionServiceManager.set_tts_status_for_session(umo, new_status)
+
+        status_text = "已开启" if new_status else "已关闭"
+
+        if new_status and not tts_enable:
+            event.set_result(
+                MessageEventResult().message(
+                    f"{status_text}当前会话的文本转语音。但 TTS 功能在配置中未启用，请前往 WebUI 开启。",
+                ),
+            )
+        else:
+            event.set_result(
+                MessageEventResult().message(f"{status_text}当前会话的文本转语音。"),
+            )

astrbot/builtin_stars/builtin_commands/commands/utils/rst_scene.py

@@ -0,0 +1,26 @@
+from enum import Enum
+
+
+class RstScene(Enum):
+    GROUP_UNIQUE_ON = ("group_unique_on", "群聊+会话隔离开启")
+    GROUP_UNIQUE_OFF = ("group_unique_off", "群聊+会话隔离关闭")
+    PRIVATE = ("private", "私聊")
+
+    @property
+    def key(self) -> str:
+        return self.value[0]
+
+    @property
+    def name(self) -> str:
+        return self.value[1]
+
+    @classmethod
+    def from_index(cls, index: int) -> "RstScene":
+        mapping = {1: cls.GROUP_UNIQUE_ON, 2: cls.GROUP_UNIQUE_OFF, 3: cls.PRIVATE}
+        return mapping[index]
+
+    @classmethod
+    def get_scene(cls, is_group: bool, is_unique_session: bool) -> "RstScene":
+        if is_group:
+            return cls.GROUP_UNIQUE_ON if is_unique_session else cls.GROUP_UNIQUE_OFF
+        return cls.PRIVATE

astrbot/builtin_stars/builtin_commands/main.py

@@ -0,0 +1,237 @@
+from astrbot.api import star
+from astrbot.api.event import AstrMessageEvent, filter
+
+from .commands import (
+    AdminCommands,
+    AlterCmdCommands,
+    ConversationCommands,
+    HelpCommand,
+    LLMCommands,
+    PersonaCommands,
+    PluginCommands,
+    ProviderCommands,
+    SetUnsetCommands,
+    SIDCommand,
+    T2ICommand,
+    ToolCommands,
+    TTSCommand,
+)
+
+
+class Main(star.Star):
+    def __init__(self, context: star.Context) -> None:
+        self.context = context
+
+        self.help_c = HelpCommand(self.context)
+        self.llm_c = LLMCommands(self.context)
+        self.tool_c = ToolCommands(self.context)
+        self.plugin_c = PluginCommands(self.context)
+        self.admin_c = AdminCommands(self.context)
+        self.conversation_c = ConversationCommands(self.context)
+        self.provider_c = ProviderCommands(self.context)
+        self.persona_c = PersonaCommands(self.context)
+        self.alter_cmd_c = AlterCmdCommands(self.context)
+        self.setunset_c = SetUnsetCommands(self.context)
+        self.t2i_c = T2ICommand(self.context)
+        self.tts_c = TTSCommand(self.context)
+        self.sid_c = SIDCommand(self.context)
+
+    @filter.command("help")
+    async def help(self, event: AstrMessageEvent):
+        """查看帮助"""
+        await self.help_c.help(event)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("llm")
+    async def llm(self, event: AstrMessageEvent):
+        """开启/关闭 LLM"""
+        await self.llm_c.llm(event)
+
+    @filter.command_group("tool")
+    def tool(self):
+        """函数工具管理"""
+
+    @tool.command("ls")
+    async def tool_ls(self, event: AstrMessageEvent):
+        """查看函数工具列表"""
+        await self.tool_c.tool_ls(event)
+
+    @tool.command("on")
+    async def tool_on(self, event: AstrMessageEvent, tool_name: str):
+        """启用一个函数工具"""
+        await self.tool_c.tool_on(event, tool_name)
+
+    @tool.command("off")
+    async def tool_off(self, event: AstrMessageEvent, tool_name: str):
+        """停用一个函数工具"""
+        await self.tool_c.tool_off(event, tool_name)
+
+    @tool.command("off_all")
+    async def tool_all_off(self, event: AstrMessageEvent):
+        """停用所有函数工具"""
+        await self.tool_c.tool_all_off(event)
+
+    @filter.command_group("plugin")
+    def plugin(self):
+        """插件管理"""
+
+    @plugin.command("ls")
+    async def plugin_ls(self, event: AstrMessageEvent):
+        """获取已经安装的插件列表。"""
+        await self.plugin_c.plugin_ls(event)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @plugin.command("off")
+    async def plugin_off(self, event: AstrMessageEvent, plugin_name: str = ""):
+        """禁用插件"""
+        await self.plugin_c.plugin_off(event, plugin_name)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @plugin.command("on")
+    async def plugin_on(self, event: AstrMessageEvent, plugin_name: str = ""):
+        """启用插件"""
+        await self.plugin_c.plugin_on(event, plugin_name)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @plugin.command("get")
+    async def plugin_get(self, event: AstrMessageEvent, plugin_repo: str = ""):
+        """安装插件"""
+        await self.plugin_c.plugin_get(event, plugin_repo)
+
+    @plugin.command("help")
+    async def plugin_help(self, event: AstrMessageEvent, plugin_name: str = ""):
+        """获取插件帮助"""
+        await self.plugin_c.plugin_help(event, plugin_name)
+
+    @filter.command("t2i")
+    async def t2i(self, event: AstrMessageEvent):
+        """开关文本转图片"""
+        await self.t2i_c.t2i(event)
+
+    @filter.command("tts")
+    async def tts(self, event: AstrMessageEvent):
+        """开关文本转语音（会话级别）"""
+        await self.tts_c.tts(event)
+
+    @filter.command("sid")
+    async def sid(self, event: AstrMessageEvent):
+        """获取会话 ID 和 管理员 ID"""
+        await self.sid_c.sid(event)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("op")
+    async def op(self, event: AstrMessageEvent, admin_id: str = ""):
+        """授权管理员。op <admin_id>"""
+        await self.admin_c.op(event, admin_id)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("deop")
+    async def deop(self, event: AstrMessageEvent, admin_id: str):
+        """取消授权管理员。deop <admin_id>"""
+        await self.admin_c.deop(event, admin_id)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("wl")
+    async def wl(self, event: AstrMessageEvent, sid: str = ""):
+        """添加白名单。wl <sid>"""
+        await self.admin_c.wl(event, sid)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("dwl")
+    async def dwl(self, event: AstrMessageEvent, sid: str):
+        """删除白名单。dwl <sid>"""
+        await self.admin_c.dwl(event, sid)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("provider")
+    async def provider(
+        self,
+        event: AstrMessageEvent,
+        idx: str | int | None = None,
+        idx2: int | None = None,
+    ):
+        """查看或者切换 LLM Provider"""
+        await self.provider_c.provider(event, idx, idx2)
+
+    @filter.command("reset")
+    async def reset(self, message: AstrMessageEvent):
+        """重置 LLM 会话"""
+        await self.conversation_c.reset(message)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("model")
+    async def model_ls(
+        self,
+        message: AstrMessageEvent,
+        idx_or_name: int | str | None = None,
+    ):
+        """查看或者切换模型"""
+        await self.provider_c.model_ls(message, idx_or_name)
+
+    @filter.command("history")
+    async def his(self, message: AstrMessageEvent, page: int = 1):
+        """查看对话记录"""
+        await self.conversation_c.his(message, page)
+
+    @filter.command("ls")
+    async def convs(self, message: AstrMessageEvent, page: int = 1):
+        """查看对话列表"""
+        await self.conversation_c.convs(message, page)
+
+    @filter.command("new")
+    async def new_conv(self, message: AstrMessageEvent):
+        """创建新对话"""
+        await self.conversation_c.new_conv(message)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("groupnew")
+    async def groupnew_conv(self, message: AstrMessageEvent, sid: str):
+        """创建新群聊对话"""
+        await self.conversation_c.groupnew_conv(message, sid)
+
+    @filter.command("switch")
+    async def switch_conv(self, message: AstrMessageEvent, index: int | None = None):
+        """通过 /ls 前面的序号切换对话"""
+        await self.conversation_c.switch_conv(message, index)
+
+    @filter.command("rename")
+    async def rename_conv(self, message: AstrMessageEvent, new_name: str):
+        """重命名对话"""
+        await self.conversation_c.rename_conv(message, new_name)
+
+    @filter.command("del")
+    async def del_conv(self, message: AstrMessageEvent):
+        """删除当前对话"""
+        await self.conversation_c.del_conv(message)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("key")
+    async def key(self, message: AstrMessageEvent, index: int | None = None):
+        """查看或者切换 Key"""
+        await self.provider_c.key(message, index)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("persona")
+    async def persona(self, message: AstrMessageEvent):
+        """查看或者切换 Persona"""
+        await self.persona_c.persona(message)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("dashboard_update")
+    async def update_dashboard(self, event: AstrMessageEvent):
+        """更新管理面板"""
+        await self.admin_c.update_dashboard(event)
+
+    @filter.command("set")
+    async def set_variable(self, event: AstrMessageEvent, key: str, value: str):
+        await self.setunset_c.set_variable(event, key, value)
+
+    @filter.command("unset")
+    async def unset_variable(self, event: AstrMessageEvent, key: str):
+        await self.setunset_c.unset_variable(event, key)
+
+    @filter.permission_type(filter.PermissionType.ADMIN)
+    @filter.command("alter_cmd", alias={"alter"})
+    async def alter_cmd(self, event: AstrMessageEvent):
+        """修改命令权限"""
+        await self.alter_cmd_c.alter_cmd(event)

astrbot/builtin_stars/builtin_commands/metadata.yaml

@@ -0,0 +1,4 @@
+name: builtin_commands
+desc: AstrBot 自带指令，提供常用的对话管理、工具使用、插件管理等功能。
+author: Soulter
+version: 0.0.1