AstrBot 4.9.1__py3-none-any.whl → 4.10.0__py3-none-any.whl

astrbot/dashboard/routes/config.py

@@ -6,7 +6,7 @@ from typing import Any
 
 from quart import request
 
-from astrbot.core import file_token_service, logger
+from astrbot.core import astrbot_config, file_token_service, logger
 from astrbot.core.config.astrbot_config import AstrBotConfig
 from astrbot.core.config.default import (
     CONFIG_METADATA_2,
@@ -21,6 +21,7 @@ from astrbot.core.platform.register import platform_cls_map, platform_registry
 from astrbot.core.provider import Provider
 from astrbot.core.provider.register import provider_registry
 from astrbot.core.star.star import star_registry
+from astrbot.core.utils.llm_metadata import LLM_METADATAS
 from astrbot.core.utils.webhook_utils import ensure_platform_webhook_config
 
 from .route import Response, Route, RouteContext
@@ -179,13 +180,149 @@ class ConfigRoute(Route):
             "/config/provider/new": ("POST", self.post_new_provider),
             "/config/provider/update": ("POST", self.post_update_provider),
             "/config/provider/delete": ("POST", self.post_delete_provider),
+            "/config/provider/template": ("GET", self.get_provider_template),
             "/config/provider/check_one": ("GET", self.check_one_provider_status),
             "/config/provider/list": ("GET", self.get_provider_config_list),
             "/config/provider/model_list": ("GET", self.get_provider_model_list),
             "/config/provider/get_embedding_dim": ("POST", self.get_embedding_dim),
+            "/config/provider_sources/<provider_source_id>/models": (
+                "GET",
+                self.get_provider_source_models,
+            ),
+            "/config/provider_sources/<provider_source_id>/update": (
+                "POST",
+                self.update_provider_source,
+            ),
+            "/config/provider_sources/<provider_source_id>/delete": (
+                "POST",
+                self.delete_provider_source,
+            ),
         }
         self.register_routes()
 
+    async def delete_provider_source(self, provider_source_id: str):
+        """删除 provider_source，并更新关联的 providers"""
+
+        provider_sources = self.config.get("provider_sources", [])
+        target_idx = next(
+            (
+                i
+                for i, ps in enumerate(provider_sources)
+                if ps.get("id") == provider_source_id
+            ),
+            -1,
+        )
+
+        if target_idx == -1:
+            return Response().error("未找到对应的 provider source").__dict__
+
+        # 删除 provider_source
+        del provider_sources[target_idx]
+
+        # 写回配置
+        self.config["provider_sources"] = provider_sources
+
+        # 删除引用了该 provider_source 的 providers
+        await self.core_lifecycle.provider_manager.delete_provider(
+            provider_source_id=provider_source_id
+        )
+
+        try:
+            save_config(self.config, self.config, is_core=True)
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(str(e)).__dict__
+
+        return Response().ok(message="删除 provider source 成功").__dict__
+
+    async def update_provider_source(self, provider_source_id: str):
+        """更新或新增 provider_source，并重载关联的 providers"""
+
+        post_data = await request.json
+        if not post_data:
+            return Response().error("缺少配置数据").__dict__
+
+        new_source_config = post_data.get("config") or post_data
+        original_id = provider_source_id
+
+        if not isinstance(new_source_config, dict):
+            return Response().error("缺少或错误的配置数据").__dict__
+
+        # 确保配置中有 id 字段
+        if not new_source_config.get("id"):
+            new_source_config["id"] = original_id
+
+        provider_sources = self.config.get("provider_sources", [])
+
+        for ps in provider_sources:
+            if ps.get("id") == new_source_config["id"] and ps.get("id") != original_id:
+                return (
+                    Response()
+                    .error(
+                        f"Provider source ID '{new_source_config['id']}' exists already, please try another ID.",
+                    )
+                    .__dict__
+                )
+
+        # 查找旧的 provider_source，若不存在则追加为新配置
+        target_idx = next(
+            (i for i, ps in enumerate(provider_sources) if ps.get("id") == original_id),
+            -1,
+        )
+
+        old_id = original_id
+        if target_idx == -1:
+            provider_sources.append(new_source_config)
+        else:
+            old_id = provider_sources[target_idx].get("id")
+            provider_sources[target_idx] = new_source_config
+
+        # 更新引用了该 provider_source 的 providers
+        affected_providers = []
+        for provider in self.config.get("provider", []):
+            if provider.get("provider_source_id") == old_id:
+                provider["provider_source_id"] = new_source_config["id"]
+                affected_providers.append(provider)
+
+        # 写回配置
+        self.config["provider_sources"] = provider_sources
+
+        try:
+            save_config(self.config, self.config, is_core=True)
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(str(e)).__dict__
+
+        # 重载受影响的 providers，使新的 source 配置生效
+        reload_errors = []
+        prov_mgr = self.core_lifecycle.provider_manager
+        for provider in affected_providers:
+            try:
+                await prov_mgr.reload(provider)
+            except Exception as e:
+                logger.error(traceback.format_exc())
+                reload_errors.append(f"{provider.get('id')}: {e}")
+
+        if reload_errors:
+            return (
+                Response()
+                .error("更新成功，但部分提供商重载失败: " + ", ".join(reload_errors))
+                .__dict__
+            )
+
+        return Response().ok(message="更新 provider source 成功").__dict__
+
+    async def get_provider_template(self):
+        config_schema = {
+            "provider": CONFIG_METADATA_2["provider_group"]["metadata"]["provider"]
+        }
+        data = {
+            "config_schema": config_schema,
+            "providers": astrbot_config["provider"],
+            "provider_sources": astrbot_config["provider_sources"],
+        }
+        return Response().ok(data=data).__dict__
+
     async def get_uc_table(self):
         """获取 UMOP 配置路由表"""
         return Response().ok({"routing": self.ucr.umop_to_conf_id}).__dict__
@@ -433,9 +570,25 @@ class ConfigRoute(Route):
             return Response().error("缺少参数 provider_type").__dict__
         provider_type_ls = provider_type.split(",")
         provider_list = []
-        astrbot_config = self.core_lifecycle.astrbot_config
-        for provider in astrbot_config["provider"]:
-            if provider.get("provider_type", None) in provider_type_ls:
+        ps = self.core_lifecycle.provider_manager.providers_config
+        p_source_pt = {
+            psrc["id"]: psrc["provider_type"]
+            for psrc in self.core_lifecycle.provider_manager.provider_sources_config
+        }
+        for provider in ps:
+            ps_id = provider.get("provider_source_id", None)
+            if (
+                ps_id
+                and ps_id in p_source_pt
+                and p_source_pt[ps_id] in provider_type_ls
+            ):
+                # chat
+                prov = self.core_lifecycle.provider_manager.get_merged_provider_config(
+                    provider
+                )
+                provider_list.append(prov)
+            elif not ps_id and provider.get("provider_type", None) in provider_type_ls:
+                # agent runner, embedding, etc
                 provider_list.append(provider)
         return Response().ok(provider_list).__dict__
 
@@ -458,9 +611,18 @@ class ConfigRoute(Route):
 
         try:
             models = await provider.get_models()
+            models = models or []
+
+            metadata_map = {}
+            for model_id in models:
+                meta = LLM_METADATAS.get(model_id)
+                if meta:
+                    metadata_map[model_id] = meta
+
             ret = {
                 "models": models,
                 "provider_id": provider_id,
+                "model_metadata": metadata_map,
             }
             return Response().ok(ret).__dict__
         except Exception as e:
@@ -522,6 +684,100 @@ class ConfigRoute(Route):
             logger.error(traceback.format_exc())
             return Response().error(f"获取嵌入维度失败: {e!s}").__dict__
 
+    async def get_provider_source_models(self, provider_source_id: str):
+        """获取指定 provider_source 支持的模型列表
+
+        本质上会临时初始化一个 Provider 实例，调用 get_models() 获取模型列表，然后销毁实例
+        """
+        try:
+            from astrbot.core.provider.register import provider_cls_map
+
+            # 从配置中查找对应的 provider_source
+            provider_sources = self.config.get("provider_sources", [])
+            provider_source = None
+            for ps in provider_sources:
+                if ps.get("id") == provider_source_id:
+                    provider_source = ps
+                    break
+
+            if not provider_source:
+                return (
+                    Response()
+                    .error(f"未找到 ID 为 {provider_source_id} 的 provider_source")
+                    .__dict__
+                )
+
+            # 获取 provider 类型
+            provider_type = provider_source.get("type", None)
+            if not provider_type:
+                return Response().error("provider_source 缺少 type 字段").__dict__
+
+            try:
+                self.core_lifecycle.provider_manager.dynamic_import_provider(
+                    provider_type
+                )
+            except ImportError as e:
+                logger.error(traceback.format_exc())
+                return Response().error(f"动态导入提供商适配器失败: {e!s}").__dict__
+
+            # 获取对应的 provider 类
+            if provider_type not in provider_cls_map:
+                return (
+                    Response()
+                    .error(f"未找到适用于 {provider_type} 的提供商适配器")
+                    .__dict__
+                )
+
+            provider_metadata = provider_cls_map[provider_type]
+            cls_type = provider_metadata.cls_type
+
+            if not cls_type:
+                return Response().error(f"无法找到 {provider_type} 的类").__dict__
+
+            # 检查是否是 Provider 类型
+            if not issubclass(cls_type, Provider):
+                return (
+                    Response()
+                    .error(f"提供商 {provider_type} 不支持获取模型列表")
+                    .__dict__
+                )
+
+            # 临时实例化 provider
+            inst = cls_type(provider_source, {})
+
+            # 如果有 initialize 方法，调用它
+            init_fn = getattr(inst, "initialize", None)
+            if inspect.iscoroutinefunction(init_fn):
+                await init_fn()
+
+            # 获取模型列表
+            models = await inst.get_models()
+            models = models or []
+
+            metadata_map = {}
+            for model_id in models:
+                meta = LLM_METADATAS.get(model_id)
+                if meta:
+                    metadata_map[model_id] = meta
+
+            # 销毁实例（如果有 terminate 方法）
+            terminate_fn = getattr(inst, "terminate", None)
+            if inspect.iscoroutinefunction(terminate_fn):
+                await terminate_fn()
+
+            logger.info(
+                f"获取到 provider_source {provider_source_id} 的模型列表: {models}",
+            )
+
+            return (
+                Response()
+                .ok({"models": models, "model_metadata": metadata_map})
+                .__dict__
+            )
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(f"获取模型列表失败: {e!s}").__dict__
+
     async def get_platform_list(self):
         """获取所有平台的列表"""
         platform_list = []
@@ -533,7 +789,15 @@ class ConfigRoute(Route):
         data = await request.json
         config = data.get("config", None)
         conf_id = data.get("conf_id", None)
+
         try:
+            # 不更新 provider_sources, provider, platform
+            # 这些配置有单独的接口进行更新
+            if conf_id == "default":
+                no_update_keys = ["provider_sources", "provider", "platform"]
+                for key in no_update_keys:
+                    config[key] = self.acm.default_conf[key]
+
             await self._save_astrbot_configs(config, conf_id)
             await self.core_lifecycle.reload_pipeline_scheduler(conf_id)
             return Response().ok(None, "保存成功~").__dict__
@@ -573,28 +837,30 @@ class ConfigRoute(Route):
 
     async def post_new_provider(self):
         new_provider_config = await request.json
-        self.config["provider"].append(new_provider_config)
+
         try:
-            save_config(self.config, self.config, is_core=True)
-            await self.core_lifecycle.provider_manager.load_provider(
-                new_provider_config,
+            await self.core_lifecycle.provider_manager.create_provider(
+                new_provider_config
             )
         except Exception as e:
             return Response().error(str(e)).__dict__
-        return Response().ok(None, "新增服务提供商配置成功~").__dict__
+        return Response().ok(None, "新增服务提供商配置成功").__dict__
 
     async def post_update_platform(self):
         update_platform_config = await request.json
-        platform_id = update_platform_config.get("id", None)
+        origin_platform_id = update_platform_config.get("id", None)
         new_config = update_platform_config.get("config", None)
-        if not platform_id or not new_config:
+        if not origin_platform_id or not new_config:
             return Response().error("参数错误").__dict__
 
+        if origin_platform_id != new_config.get("id", None):
+            return Response().error("机器人名称不允许修改").__dict__
+
         # 如果是支持统一 webhook 模式的平台，且启用了统一 webhook 模式，确保有 webhook_uuid
         ensure_platform_webhook_config(new_config)
 
         for i, platform in enumerate(self.config["platform"]):
-            if platform["id"] == platform_id:
+            if platform["id"] == origin_platform_id:
                 self.config["platform"][i] = new_config
                 break
         else:
@@ -609,21 +875,15 @@ class ConfigRoute(Route):
 
     async def post_update_provider(self):
         update_provider_config = await request.json
-        provider_id = update_provider_config.get("id", None)
+        origin_provider_id = update_provider_config.get("id", None)
         new_config = update_provider_config.get("config", None)
-        if not provider_id or not new_config:
+        if not origin_provider_id or not new_config:
             return Response().error("参数错误").__dict__
 
-        for i, provider in enumerate(self.config["provider"]):
-            if provider["id"] == provider_id:
-                self.config["provider"][i] = new_config
-                break
-        else:
-            return Response().error("未找到对应服务提供商").__dict__
-
         try:
-            save_config(self.config, self.config, is_core=True)
-            await self.core_lifecycle.provider_manager.reload(new_config)
+            await self.core_lifecycle.provider_manager.update_provider(
+                origin_provider_id, new_config
+            )
         except Exception as e:
             return Response().error(str(e)).__dict__
         return Response().ok(None, "更新成功，已经实时生效~").__dict__
@@ -646,19 +906,17 @@ class ConfigRoute(Route):
 
     async def post_delete_provider(self):
         provider_id = await request.json
-        provider_id = provider_id.get("id")
-        for i, provider in enumerate(self.config["provider"]):
-            if provider["id"] == provider_id:
-                del self.config["provider"][i]
-                break
-        else:
-            return Response().error("未找到对应服务提供商").__dict__
+        provider_id = provider_id.get("id", "")
+        if not provider_id:
+            return Response().error("缺少参数 id").__dict__
+
         try:
-            save_config(self.config, self.config, is_core=True)
-            await self.core_lifecycle.provider_manager.terminate_provider(provider_id)
+            await self.core_lifecycle.provider_manager.delete_provider(
+                provider_id=provider_id
+            )
         except Exception as e:
             return Response().error(str(e)).__dict__
-        return Response().ok(None, "删除成功，已经实时生效~").__dict__
+        return Response().ok(None, "删除成功，已经实时生效。").__dict__
 
     async def get_llm_tools(self):
         """获取函数调用工具。包含了本地加载的以及 MCP 服务的工具"""

astrbot/dashboard/routes/stat.py

@@ -1,6 +1,9 @@
+import os
+import re
 import threading
 import time
 import traceback
+from functools import cmp_to_key
 
 import aiohttp
 import psutil
@@ -11,7 +14,9 @@ from astrbot.core.config import VERSION
 from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
 from astrbot.core.db import BaseDatabase
 from astrbot.core.db.migration.helper import check_migration_needed_v4
+from astrbot.core.utils.astrbot_path import get_astrbot_path
 from astrbot.core.utils.io import get_dashboard_version
+from astrbot.core.utils.version_comparator import VersionComparator
 
 from .route import Response, Route, RouteContext
 
@@ -30,6 +35,8 @@ class StatRoute(Route):
             "/stat/start-time": ("GET", self.get_start_time),
             "/stat/restart-core": ("POST", self.restart_core),
             "/stat/test-ghproxy-connection": ("POST", self.test_ghproxy_connection),
+            "/stat/changelog": ("GET", self.get_changelog),
+            "/stat/changelog/list": ("GET", self.list_changelog_versions),
         }
         self.db_helper = db_helper
         self.register_routes()
@@ -183,3 +190,92 @@ class StatRoute(Route):
         except Exception as e:
             logger.error(traceback.format_exc())
             return Response().error(f"Error: {e!s}").__dict__
+
+    async def get_changelog(self):
+        """获取指定版本的更新日志"""
+        try:
+            version = request.args.get("version")
+            if not version:
+                return Response().error("version parameter is required").__dict__
+
+            version = version.lstrip("v")
+
+            # 防止路径遍历攻击
+            if not re.match(r"^[a-zA-Z0-9._-]+$", version):
+                return Response().error("Invalid version format").__dict__
+            if ".." in version or "/" in version or "\\" in version:
+                return Response().error("Invalid version format").__dict__
+
+            filename = f"v{version}.md"
+            project_path = get_astrbot_path()
+            changelogs_dir = os.path.join(project_path, "changelogs")
+            changelog_path = os.path.join(changelogs_dir, filename)
+
+            # 规范化路径，防止符号链接攻击
+            changelog_path = os.path.realpath(changelog_path)
+            changelogs_dir = os.path.realpath(changelogs_dir)
+
+            # 验证最终路径在预期的 changelogs 目录内（防止路径遍历）
+            # 确保规范化后的路径以 changelogs_dir 开头，且是目录内的文件
+            changelog_path_normalized = os.path.normpath(changelog_path)
+            changelogs_dir_normalized = os.path.normpath(changelogs_dir)
+
+            # 检查路径是否在预期目录内（必须是目录的子文件，不能是目录本身）
+            expected_prefix = changelogs_dir_normalized + os.sep
+            if not changelog_path_normalized.startswith(expected_prefix):
+                logger.warning(
+                    f"Path traversal attempt detected: {version} -> {changelog_path}",
+                )
+                return Response().error("Invalid version format").__dict__
+
+            if not os.path.exists(changelog_path):
+                return (
+                    Response()
+                    .error(f"Changelog for version {version} not found")
+                    .__dict__
+                )
+            if not os.path.isfile(changelog_path):
+                return (
+                    Response()
+                    .error(f"Changelog for version {version} not found")
+                    .__dict__
+                )
+
+            with open(changelog_path, encoding="utf-8") as f:
+                content = f.read()
+
+            return Response().ok({"content": content, "version": version}).__dict__
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(f"Error: {e!s}").__dict__
+
+    async def list_changelog_versions(self):
+        """获取所有可用的更新日志版本列表"""
+        try:
+            project_path = get_astrbot_path()
+            changelogs_dir = os.path.join(project_path, "changelogs")
+
+            if not os.path.exists(changelogs_dir):
+                return Response().ok({"versions": []}).__dict__
+
+            versions = []
+            for filename in os.listdir(changelogs_dir):
+                if filename.endswith(".md") and filename.startswith("v"):
+                    # 提取版本号（去除 v 前缀和 .md 后缀）
+                    version = filename[1:-3]  # 去掉 "v" 和 ".md"
+                    # 验证版本号格式
+                    if re.match(r"^[a-zA-Z0-9._-]+$", version):
+                        versions.append(version)
+
+            # 按版本号排序（降序，最新的在前）
+            # 使用项目中的 VersionComparator 进行语义化版本号排序
+            versions.sort(
+                key=cmp_to_key(
+                    lambda v1, v2: VersionComparator.compare_version(v2, v1),
+                ),
+            )
+
+            return Response().ok({"versions": versions}).__dict__
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(f"Error: {e!s}").__dict__

astrbot/dashboard/routes/tools.py

@@ -3,6 +3,7 @@ import traceback
 from quart import request
 
 from astrbot.core import logger
+from astrbot.core.agent.mcp_client import MCPTool
 from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
 from astrbot.core.star import star_map
 
@@ -296,15 +297,30 @@ class ToolsRoute(Route):
         """获取所有注册的工具列表"""
         try:
             tools = self.tool_mgr.func_list
-            tools_dict = [
-                {
+            tools_dict = []
+            for tool in tools:
+                if isinstance(tool, MCPTool):
+                    origin = "mcp"
+                    origin_name = tool.mcp_server_name
+                elif tool.handler_module_path and star_map.get(
+                    tool.handler_module_path
+                ):
+                    star = star_map[tool.handler_module_path]
+                    origin = "plugin"
+                    origin_name = star.name
+                else:
+                    origin = "unknown"
+                    origin_name = "unknown"
+
+                tool_info = {
                     "name": tool.name,
                     "description": tool.description,
                     "parameters": tool.parameters,
                     "active": tool.active,
+                    "origin": origin,
+                    "origin_name": origin_name,
                 }
-                for tool in tools
-            ]
+                tools_dict.append(tool_info)
             return Response().ok(data=tools_dict).__dict__
         except Exception as e:
             logger.error(traceback.format_exc())

astrbot/dashboard/server.py

@@ -67,6 +67,7 @@ class AstrBotDashboard:
             core_lifecycle,
             core_lifecycle.plugin_manager,
         )
+        self.command_route = CommandRoute(self.context)
         self.cr = ConfigRoute(self.context, core_lifecycle)
         self.lr = LogRoute(self.context, core_lifecycle.log_broker)
         self.sfr = StaticFileRoute(self.context)

{astrbot-4.9.1.dist-info → astrbot-4.10.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: AstrBot
-Version: 4.9.1
+Version: 4.10.0
 Summary: Easy-to-use multi-platform LLM chatbot and development framework
 License-File: LICENSE
 Keywords: Astrbot,Astrbot Module,Astrbot Plugin
@@ -26,7 +26,7 @@ Requires-Dist: dingtalk-stream>=0.22.1
 Requires-Dist: docstring-parser>=0.16
 Requires-Dist: faiss-cpu==1.10.0
 Requires-Dist: filelock>=3.18.0
-Requires-Dist: google-genai<1.51.0,>=1.14.0
+Requires-Dist: google-genai>=1.56.0
 Requires-Dist: jieba>=0.42.1
 Requires-Dist: lark-oapi>=1.4.15
 Requires-Dist: lxml-html-clean>=0.4.2