AstrBot 4.5.1__py3-none-any.whl → 4.5.3__py3-none-any.whl

astrbot/dashboard/routes/t2i.py

@@ -1,11 +1,13 @@
 # astrbot/dashboard/routes/t2i.py
 
 from dataclasses import asdict
+
 from quart import jsonify, request
 
 from astrbot.core import logger
 from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
 from astrbot.core.utils.t2i.template_manager import TemplateManager
+
 from .route import Response, Route, RouteContext
 
 
@@ -49,7 +51,7 @@ class T2iRoute(Route):
         try:
             active_template = self.config.get("t2i_active_template", "base")
             return jsonify(
-                asdict(Response().ok(data={"active_template": active_template}))
+                asdict(Response().ok(data={"active_template": active_template})),
             )
         except Exception as e:
             logger.error("Error in get_active_template", exc_info=True)
@@ -62,7 +64,7 @@ class T2iRoute(Route):
         try:
             content = self.manager.get_template(name)
             return jsonify(
-                asdict(Response().ok(data={"name": name, "content": content}))
+                asdict(Response().ok(data={"name": name, "content": content})),
             )
         except FileNotFoundError:
             response = jsonify(asdict(Response().error("Template not found")))
@@ -81,7 +83,7 @@ class T2iRoute(Route):
             content = data.get("content")
             if not name or not content:
                 response = jsonify(
-                    asdict(Response().error("Name and content are required."))
+                    asdict(Response().error("Name and content are required.")),
                 )
                 response.status_code = 400
                 return response
@@ -91,15 +93,16 @@ class T2iRoute(Route):
             response = jsonify(
                 asdict(
                     Response().ok(
-                        data={"name": name}, message="Template created successfully."
-                    )
-                )
+                        data={"name": name},
+                        message="Template created successfully.",
+                    ),
+                ),
             )
             response.status_code = 201
             return response
         except FileExistsError:
             response = jsonify(
-                asdict(Response().error("Template with this name already exists."))
+                asdict(Response().error("Template with this name already exists.")),
             )
             response.status_code = 409
             return response
@@ -149,7 +152,7 @@ class T2iRoute(Route):
             name = name.strip()
             self.manager.delete_template(name)
             return jsonify(
-                asdict(Response().ok(message="Template deleted successfully."))
+                asdict(Response().ok(message="Template deleted successfully.")),
             )
         except FileNotFoundError:
             response = jsonify(asdict(Response().error("Template not found.")))
@@ -189,7 +192,7 @@ class T2iRoute(Route):
 
         except FileNotFoundError:
             response = jsonify(
-                asdict(Response().error(f"模板 '{name}' 不存在，无法应用。"))
+                asdict(Response().error(f"模板 '{name}' 不存在，无法应用。")),
             )
             response.status_code = 404
             return response
@@ -215,9 +218,9 @@ class T2iRoute(Route):
             return jsonify(
                 asdict(
                     Response().ok(
-                        message="Default template has been reset and activated."
-                    )
-                )
+                        message="Default template has been reset and activated.",
+                    ),
+                ),
             )
         except FileNotFoundError as e:
             response = jsonify(asdict(Response().error(str(e))))

astrbot/dashboard/routes/tools.py

@@ -13,7 +13,9 @@ DEFAULT_MCP_CONFIG = {"mcpServers": {}}
 
 class ToolsRoute(Route):
     def __init__(
-        self, context: RouteContext, core_lifecycle: AstrBotCoreLifecycle
+        self,
+        context: RouteContext,
+        core_lifecycle: AstrBotCoreLifecycle,
     ) -> None:
         super().__init__(context)
         self.core_lifecycle = core_lifecycle
@@ -64,7 +66,7 @@ class ToolsRoute(Route):
             return Response().ok(servers).__dict__
         except Exception as e:
             logger.error(traceback.format_exc())
-            return Response().error(f"获取 MCP 服务器列表失败: {str(e)}").__dict__
+            return Response().error(f"获取 MCP 服务器列表失败: {e!s}").__dict__
 
     async def add_mcp_server(self):
         try:
@@ -105,23 +107,22 @@ class ToolsRoute(Route):
             if self.tool_mgr.save_mcp_config(config):
                 try:
                     await self.tool_mgr.enable_mcp_server(
-                        name, server_config, timeout=30
+                        name,
+                        server_config,
+                        timeout=30,
                     )
                 except TimeoutError:
                     return Response().error(f"启用 MCP 服务器 {name} 超时。").__dict__
                 except Exception as e:
                     logger.error(traceback.format_exc())
                     return (
-                        Response()
-                        .error(f"启用 MCP 服务器 {name} 失败: {str(e)}")
-                        .__dict__
+                        Response().error(f"启用 MCP 服务器 {name} 失败: {e!s}").__dict__
                     )
                 return Response().ok(None, f"成功添加 MCP 服务器 {name}").__dict__
-            else:
-                return Response().error("保存配置失败").__dict__
+            return Response().error("保存配置失败").__dict__
         except Exception as e:
             logger.error(traceback.format_exc())
-            return Response().error(f"添加 MCP 服务器失败: {str(e)}").__dict__
+            return Response().error(f"添加 MCP 服务器失败: {e!s}").__dict__
 
     async def update_mcp_server(self):
         try:
@@ -139,7 +140,8 @@ class ToolsRoute(Route):
 
             # 获取活动状态
             active = server_data.get(
-                "active", config["mcpServers"][name].get("active", True)
+                "active",
+                config["mcpServers"][name].get("active", True),
             )
 
             # 创建新的配置对象
@@ -177,19 +179,21 @@ class ToolsRoute(Route):
                         except TimeoutError as e:
                             return (
                                 Response()
-                                .error(f"启用前停用 MCP 服务器时 {name} 超时: {str(e)}")
+                                .error(f"启用前停用 MCP 服务器时 {name} 超时: {e!s}")
                                 .__dict__
                             )
                         except Exception as e:
                             logger.error(traceback.format_exc())
                             return (
                                 Response()
-                                .error(f"启用前停用 MCP 服务器时 {name} 失败: {str(e)}")
+                                .error(f"启用前停用 MCP 服务器时 {name} 失败: {e!s}")
                                 .__dict__
                             )
                     try:
                         await self.tool_mgr.enable_mcp_server(
-                            name, config["mcpServers"][name], timeout=30
+                            name,
+                            config["mcpServers"][name],
+                            timeout=30,
                         )
                     except TimeoutError:
                         return (
@@ -199,34 +203,30 @@ class ToolsRoute(Route):
                         logger.error(traceback.format_exc())
                         return (
                             Response()
-                            .error(f"启用 MCP 服务器 {name} 失败: {str(e)}")
+                            .error(f"启用 MCP 服务器 {name} 失败: {e!s}")
+                            .__dict__
+                        )
+                # 如果要停用服务器
+                elif name in self.tool_mgr.mcp_client_dict:
+                    try:
+                        await self.tool_mgr.disable_mcp_server(name, timeout=10)
+                    except TimeoutError:
+                        return (
+                            Response().error(f"停用 MCP 服务器 {name} 超时。").__dict__
+                        )
+                    except Exception as e:
+                        logger.error(traceback.format_exc())
+                        return (
+                            Response()
+                            .error(f"停用 MCP 服务器 {name} 失败: {e!s}")
                             .__dict__
                         )
-                else:
-                    # 如果要停用服务器
-                    if name in self.tool_mgr.mcp_client_dict:
-                        try:
-                            await self.tool_mgr.disable_mcp_server(name, timeout=10)
-                        except TimeoutError:
-                            return (
-                                Response()
-                                .error(f"停用 MCP 服务器 {name} 超时。")
-                                .__dict__
-                            )
-                        except Exception as e:
-                            logger.error(traceback.format_exc())
-                            return (
-                                Response()
-                                .error(f"停用 MCP 服务器 {name} 失败: {str(e)}")
-                                .__dict__
-                            )
 
                 return Response().ok(None, f"成功更新 MCP 服务器 {name}").__dict__
-            else:
-                return Response().error("保存配置失败").__dict__
+            return Response().error("保存配置失败").__dict__
         except Exception as e:
             logger.error(traceback.format_exc())
-            return Response().error(f"更新 MCP 服务器失败: {str(e)}").__dict__
+            return Response().error(f"更新 MCP 服务器失败: {e!s}").__dict__
 
     async def delete_mcp_server(self):
         try:
@@ -255,20 +255,17 @@ class ToolsRoute(Route):
                         logger.error(traceback.format_exc())
                         return (
                             Response()
-                            .error(f"停用 MCP 服务器 {name} 失败: {str(e)}")
+                            .error(f"停用 MCP 服务器 {name} 失败: {e!s}")
                             .__dict__
                         )
                 return Response().ok(None, f"成功删除 MCP 服务器 {name}").__dict__
-            else:
-                return Response().error("保存配置失败").__dict__
+            return Response().error("保存配置失败").__dict__
         except Exception as e:
             logger.error(traceback.format_exc())
-            return Response().error(f"删除 MCP 服务器失败: {str(e)}").__dict__
+            return Response().error(f"删除 MCP 服务器失败: {e!s}").__dict__
 
     async def test_mcp_connection(self):
-        """
-        测试 MCP 服务器连接
-        """
+        """测试 MCP 服务器连接"""
         try:
             server_data = await request.json
             config = server_data.get("mcp_server_config", None)
@@ -283,9 +280,8 @@ class ToolsRoute(Route):
                 if len(keys) > 1:
                     return Response().error("一次只能配置一个 MCP 服务器配置").__dict__
                 config = config["mcpServers"][keys[0]]
-            else:
-                if not config:
-                    return Response().error("MCP 服务器配置不能为空").__dict__
+            elif not config:
+                return Response().error("MCP 服务器配置不能为空").__dict__
 
             tools_name = await self.tool_mgr.test_mcp_server_connection(config)
             return (
@@ -294,17 +290,25 @@ class ToolsRoute(Route):
 
         except Exception as e:
             logger.error(traceback.format_exc())
-            return Response().error(f"测试 MCP 连接失败: {str(e)}").__dict__
+            return Response().error(f"测试 MCP 连接失败: {e!s}").__dict__
 
     async def get_tool_list(self):
         """获取所有注册的工具列表"""
         try:
             tools = self.tool_mgr.func_list
-            tools_dict = [tool.__dict__() for tool in tools]
+            tools_dict = [
+                {
+                    "name": tool.name,
+                    "description": tool.description,
+                    "parameters": tool.parameters,
+                    "active": tool.active,
+                }
+                for tool in tools
+            ]
             return Response().ok(data=tools_dict).__dict__
         except Exception as e:
             logger.error(traceback.format_exc())
-            return Response().error(f"获取工具列表失败: {str(e)}").__dict__
+            return Response().error(f"获取工具列表失败: {e!s}").__dict__
 
     async def toggle_tool(self):
         """启用或停用指定的工具"""
@@ -320,18 +324,17 @@ class ToolsRoute(Route):
                 try:
                     ok = self.tool_mgr.activate_llm_tool(tool_name, star_map=star_map)
                 except ValueError as e:
-                    return Response().error(f"启用工具失败: {str(e)}").__dict__
+                    return Response().error(f"启用工具失败: {e!s}").__dict__
             else:
                 ok = self.tool_mgr.deactivate_llm_tool(tool_name)
 
             if ok:
                 return Response().ok(None, "操作成功。").__dict__
-            else:
-                return Response().error(f"工具 {tool_name} 不存在或操作失败。").__dict__
+            return Response().error(f"工具 {tool_name} 不存在或操作失败。").__dict__
 
         except Exception as e:
             logger.error(traceback.format_exc())
-            return Response().error(f"操作工具失败: {str(e)}").__dict__
+            return Response().error(f"操作工具失败: {e!s}").__dict__
 
     async def sync_provider(self):
         """同步 MCP 提供者配置"""
@@ -348,4 +351,4 @@ class ToolsRoute(Route):
             return Response().ok(message="同步成功").__dict__
         except Exception as e:
             logger.error(traceback.format_exc())
-            return Response().error(f"同步失败: {str(e)}").__dict__
+            return Response().error(f"同步失败: {e!s}").__dict__

astrbot/dashboard/routes/update.py

@@ -1,13 +1,15 @@
 import traceback
-from .route import Route, Response, RouteContext
+
 from quart import request
+
+from astrbot.core import DEMO_MODE, logger, pip_installer
+from astrbot.core.config.default import VERSION
 from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
+from astrbot.core.db.migration.helper import check_migration_needed_v4, do_migration_v4
 from astrbot.core.updator import AstrBotUpdator
-from astrbot.core import logger, pip_installer
 from astrbot.core.utils.io import download_dashboard, get_dashboard_version
-from astrbot.core.config.default import VERSION
-from astrbot.core import DEMO_MODE
-from astrbot.core.db.migration.helper import do_migration_v4, check_migration_needed_v4
+
+from .route import Response, Route, RouteContext
 
 CLEAR_SITE_DATA_HEADERS = {"Clear-Site-Data": '"cache"'}
 
@@ -40,12 +42,14 @@ class UpdateRoute(Route):
             data = await request.json
             pim = data.get("platform_id_map", {})
             await do_migration_v4(
-                self.core_lifecycle.db, pim, self.core_lifecycle.astrbot_config
+                self.core_lifecycle.db,
+                pim,
+                self.core_lifecycle.astrbot_config,
             )
             return Response().ok(None, "迁移成功。").__dict__
         except Exception as e:
             logger.error(f"迁移失败: {traceback.format_exc()}")
-            return Response().error(f"迁移失败: {str(e)}").__dict__
+            return Response().error(f"迁移失败: {e!s}").__dict__
 
     async def check_update(self):
         type_ = request.args.get("type", None)
@@ -58,20 +62,19 @@ class UpdateRoute(Route):
                     .ok({"has_new_version": dv != f"v{VERSION}", "current_version": dv})
                     .__dict__
                 )
-            else:
-                ret = await self.astrbot_updator.check_update(None, None, False)
-                return Response(
-                    status="success",
-                    message=str(ret) if ret is not None else "已经是最新版本了。",
-                    data={
-                        "version": f"v{VERSION}",
-                        "has_new_version": ret is not None,
-                        "dashboard_version": dv,
-                        "dashboard_has_new_version": bool(dv and dv != f"v{VERSION}"),
-                    },
-                ).__dict__
+            ret = await self.astrbot_updator.check_update(None, None, False)
+            return Response(
+                status="success",
+                message=str(ret) if ret is not None else "已经是最新版本了。",
+                data={
+                    "version": f"v{VERSION}",
+                    "has_new_version": ret is not None,
+                    "dashboard_version": dv,
+                    "dashboard_has_new_version": bool(dv and dv != f"v{VERSION}"),
+                },
+            ).__dict__
         except Exception as e:
-            logger.warning(f"检查更新失败: {str(e)} (不影响除项目更新外的正常使用)")
+            logger.warning(f"检查更新失败: {e!s} (不影响除项目更新外的正常使用)")
             return Response().error(e.__str__()).__dict__
 
     async def get_releases(self):
@@ -98,7 +101,9 @@ class UpdateRoute(Route):
 
         try:
             await self.astrbot_updator.update(
-                latest=latest, version=version, proxy=proxy
+                latest=latest,
+                version=version,
+                proxy=proxy,
             )
 
             try:
@@ -121,13 +126,12 @@ class UpdateRoute(Route):
                     .__dict__
                 )
                 return ret, 200, CLEAR_SITE_DATA_HEADERS
-            else:
-                ret = (
-                    Response()
-                    .ok(None, "更新成功，AstrBot 将在下次启动时应用新的代码。")
-                    .__dict__
-                )
-                return ret, 200, CLEAR_SITE_DATA_HEADERS
+            ret = (
+                Response()
+                .ok(None, "更新成功，AstrBot 将在下次启动时应用新的代码。")
+                .__dict__
+            )
+            return ret, 200, CLEAR_SITE_DATA_HEADERS
         except Exception as e:
             logger.error(f"/api/update_project: {traceback.format_exc()}")
             return Response().error(e.__str__()).__dict__

astrbot/dashboard/server.py

@@ -39,7 +39,7 @@ class AstrBotDashboard:
             self.data_path = os.path.abspath(webui_dir)
         else:
             self.data_path = os.path.abspath(
-                os.path.join(get_astrbot_data_path(), "dist")
+                os.path.join(get_astrbot_data_path(), "dist"),
             )
 
         self.app = Quart("dashboard", static_folder=self.data_path, static_url_path="/")
@@ -53,11 +53,15 @@ class AstrBotDashboard:
         logging.getLogger(self.app.name).removeHandler(default_handler)
         self.context = RouteContext(self.config, self.app)
         self.ur = UpdateRoute(
-            self.context, core_lifecycle.astrbot_updator, core_lifecycle
+            self.context,
+            core_lifecycle.astrbot_updator,
+            core_lifecycle,
         )
         self.sr = StatRoute(self.context, db, core_lifecycle)
         self.pr = PluginRoute(
-            self.context, core_lifecycle, core_lifecycle.plugin_manager
+            self.context,
+            core_lifecycle,
+            core_lifecycle.plugin_manager,
         )
         self.cr = ConfigRoute(self.context, core_lifecycle)
         self.lr = LogRoute(self.context, core_lifecycle.log_broker)
@@ -68,7 +72,9 @@ class AstrBotDashboard:
         self.conversation_route = ConversationRoute(self.context, db, core_lifecycle)
         self.file_route = FileRoute(self.context)
         self.session_management_route = SessionManagementRoute(
-            self.context, db, core_lifecycle
+            self.context,
+            db,
+            core_lifecycle,
         )
         self.persona_route = PersonaRoute(self.context, db, core_lifecycle)
         self.t2i_route = T2iRoute(self.context, core_lifecycle)
@@ -85,9 +91,7 @@ class AstrBotDashboard:
         self._init_jwt_secret()
 
     async def srv_plug_route(self, subpath, *args, **kwargs):
-        """
-        插件路由
-        """
+        """插件路由"""
         registered_web_apis = self.core_lifecycle.star_context.registered_web_apis
         for api in registered_web_apis:
             route, view_handler, methods, _ = api
@@ -97,18 +101,17 @@ class AstrBotDashboard:
 
     async def auth_middleware(self):
         if not request.path.startswith("/api"):
-            return
+            return None
         allowed_endpoints = ["/api/auth/login", "/api/file"]
         if any(request.path.startswith(prefix) for prefix in allowed_endpoints):
-            return
-        # claim jwt
+            return None
+        # 声明 JWT
         token = request.headers.get("Authorization")
         if not token:
             r = jsonify(Response().error("未授权").__dict__)
             r.status_code = 401
             return r
-        if token.startswith("Bearer "):
-            token = token[7:]
+        token = token.removeprefix("Bearer ")
         try:
             payload = jwt.decode(token, self._jwt_secret, algorithms=["HS256"])
             g.username = payload["username"]
@@ -122,9 +125,7 @@ class AstrBotDashboard:
             return r
 
     def check_port_in_use(self, port: int) -> bool:
-        """
-        跨平台检测端口是否被占用
-        """
+        """跨平台检测端口是否被占用"""
         try:
             # 创建 IPv4 TCP Socket
             sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@@ -136,7 +137,7 @@ class AstrBotDashboard:
             # result 为 0 表示端口被占用
             return result == 0
         except Exception as e:
-            logger.warning(f"检查端口 {port} 时发生错误: {str(e)}")
+            logger.warning(f"检查端口 {port} 时发生错误: {e!s}")
             # 如果出现异常，保守起见认为端口可能被占用
             return True
 
@@ -157,10 +158,10 @@ class AstrBotDashboard:
                         ]
                         return "\n           ".join(proc_info)
                     except (psutil.NoSuchProcess, psutil.AccessDenied) as e:
-                        return f"无法获取进程详细信息(可能需要管理员权限): {str(e)}"
+                        return f"无法获取进程详细信息(可能需要管理员权限): {e!s}"
             return "未找到占用进程"
         except Exception as e:
-            return f"获取进程信息失败: {str(e)}"
+            return f"获取进程信息失败: {e!s}"
 
     def _init_jwt_secret(self):
         if not self.config.get("dashboard", {}).get("jwt_secret", None):
@@ -182,13 +183,13 @@ class AstrBotDashboard:
 
         if not enable:
             logger.info("WebUI 已被禁用")
-            return
+            return None
 
         logger.info(f"正在启动 WebUI, 监听地址: http://{host}:{port}")
 
         if host == "0.0.0.0":
             logger.info(
-                "提示: WebUI 将监听所有网络接口，请注意安全。（可在 data/cmd_config.json 中配置 dashboard.host 以修改 host）"
+                "提示: WebUI 将监听所有网络接口，请注意安全。（可在 data/cmd_config.json 中配置 dashboard.host 以修改 host）",
             )
 
         if host not in ["localhost", "127.0.0.1"]:
@@ -207,16 +208,17 @@ class AstrBotDashboard:
                 f"请确保：\n"
                 f"1. 没有其他 AstrBot 实例正在运行\n"
                 f"2. 端口 {port} 没有被其他程序占用\n"
-                f"3. 如需使用其他端口，请修改配置文件"
+                f"3. 如需使用其他端口，请修改配置文件",
             )
 
             raise Exception(f"端口 {port} 已被占用")
 
-        display = f"\n ✨✨✨\n  AstrBot v{VERSION} WebUI 已启动，可访问\n\n"
-        display += f"   ➜  本地: http://localhost:{port}\n"
+        parts = [f"\n ✨✨✨\n  AstrBot v{VERSION} WebUI 已启动，可访问\n\n"]
+        parts.append(f"   ➜  本地: http://localhost:{port}\n")
         for ip in ip_addr:
-            display += f"   ➜  网络: http://{ip}:{port}\n"
-        display += "   ➜  默认用户名和密码: astrbot\n ✨✨✨\n"
+            parts.append(f"   ➜  网络: http://{ip}:{port}\n")
+        parts.append("   ➜  默认用户名和密码: astrbot\n ✨✨✨\n")
+        display = "".join(parts)
 
         if not ip_addr:
             display += (
@@ -226,7 +228,9 @@ class AstrBotDashboard:
         logger.info(display)
 
         return self.app.run_task(
-            host=host, port=port, shutdown_trigger=self.shutdown_trigger
+            host=host,
+            port=port,
+            shutdown_trigger=self.shutdown_trigger,
         )
 
     async def shutdown_trigger(self):

astrbot/dashboard/utils.py

@@ -2,14 +2,17 @@ import base64
 import os
 import traceback
 from io import BytesIO
+
 from astrbot.api import logger
+from astrbot.core.db.vec_db.faiss_impl import FaissVecDB
 from astrbot.core.knowledge_base.kb_helper import KBHelper
 from astrbot.core.knowledge_base.kb_mgr import KnowledgeBaseManager
-from astrbot.core.db.vec_db.faiss_impl import FaissVecDB
 
 
 async def generate_tsne_visualization(
-    query: str, kb_names: list[str], kb_manager: KnowledgeBaseManager
+    query: str,
+    kb_names: list[str],
+    kb_manager: KnowledgeBaseManager,
 ) -> str | None:
     """生成 t-SNE 可视化图片
 
@@ -20,18 +23,19 @@ async def generate_tsne_visualization(
 
     Returns:
         图片路径或 None
+
     """
     try:
         import faiss
-        import numpy as np
         import matplotlib
+        import numpy as np
 
         matplotlib.use("Agg")  # 使用非交互式后端
         import matplotlib.pyplot as plt
         from sklearn.manifold import TSNE
     except ImportError as e:
         raise Exception(
-            "缺少必要的库以生成 t-SNE 可视化。请安装 matplotlib 和 scikit-learn: {e}"
+            "缺少必要的库以生成 t-SNE 可视化。请安装 matplotlib 和 scikit-learn: {e}",
         ) from e
 
     try:

{astrbot-4.5.1.dist-info → astrbot-4.5.3.dist-info}/METADATA

@@ -1,8 +1,9 @@
 Metadata-Version: 2.4
 Name: AstrBot
-Version: 4.5.1
+Version: 4.5.3
 Summary: 易上手的多平台 LLM 聊天机器人及开发框架
 License-File: LICENSE
+Keywords: Astrbot,Astrbot Module,Astrbot Plugin
 Requires-Python: >=3.10
 Requires-Dist: aiocqhttp>=1.4.4
 Requires-Dist: aiodocker>=0.24.0