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

astrbot/core/utils/migra_helper.py

@@ -32,6 +32,92 @@ def _migra_agent_runner_configs(conf: AstrBotConfig, ids_map: dict) -> None:
         logger.error(traceback.format_exc())
 
 
+def _migra_provider_to_source_structure(conf: AstrBotConfig) -> None:
+    """
+    Migrate old provider structure to new provider-source separation.
+    Provider only keeps: id, provider_source_id, model, modalities, custom_extra_body
+    All other fields move to provider_sources.
+    """
+    providers = conf.get("provider", [])
+    provider_sources = conf.get("provider_sources", [])
+
+    # Track if any migration happened
+    migrated = False
+
+    # Provider-only fields that should stay in provider
+    provider_only_fields = {
+        "id",
+        "provider_source_id",
+        "model",
+        "modalities",
+        "custom_extra_body",
+        "enable",
+    }
+
+    # Fields that should not go to source
+    source_exclude_fields = provider_only_fields | {"model_config"}
+
+    for provider in providers:
+        # Skip if already has provider_source_id
+        if provider.get("provider_source_id"):
+            continue
+
+        # Skip non-chat-completion types (they don't need source separation)
+        provider_type = provider.get("provider_type", "")
+        if provider_type != "chat_completion":
+            # For old types without provider_type, check type field
+            old_type = provider.get("type", "")
+            if "chat_completion" not in old_type:
+                continue
+
+        migrated = True
+        logger.info(f"Migrating provider {provider.get('id')} to new structure")
+
+        # Extract source fields from provider
+        source_fields = {}
+        for key, value in list(provider.items()):
+            if key not in source_exclude_fields:
+                source_fields[key] = value
+
+        # Create new provider_source
+        source_id = provider.get("id", "") + "_source"
+        new_source = {"id": source_id, **source_fields}
+
+        # Update provider to only keep necessary fields
+        provider["provider_source_id"] = source_id
+
+        # Extract model from model_config if exists
+        if "model_config" in provider and isinstance(provider["model_config"], dict):
+            model_config = provider["model_config"]
+            provider["model"] = model_config.get("model", "")
+
+            # Put other model_config fields into custom_extra_body
+            extra_body_fields = {k: v for k, v in model_config.items() if k != "model"}
+            if extra_body_fields:
+                if "custom_extra_body" not in provider:
+                    provider["custom_extra_body"] = {}
+                provider["custom_extra_body"].update(extra_body_fields)
+
+        # Initialize new fields if not present
+        if "modalities" not in provider:
+            provider["modalities"] = []
+        if "custom_extra_body" not in provider:
+            provider["custom_extra_body"] = {}
+
+        # Remove fields that should be in source
+        keys_to_remove = [k for k in provider.keys() if k not in provider_only_fields]
+        for key in keys_to_remove:
+            del provider[key]
+
+        # Add source to provider_sources
+        provider_sources.append(new_source)
+
+    if migrated:
+        conf["provider_sources"] = provider_sources
+        conf.save_config()
+        logger.info("Provider-source structure migration completed")
+
+
 async def migra(
     db, astrbot_config_mgr, umop_config_router, acm: AstrBotConfigManager
 ) -> None:
@@ -71,3 +157,10 @@ async def migra(
 
     for conf in acm.confs.values():
         _migra_agent_runner_configs(conf, ids_map)
+
+    # Migrate providers to new structure: extract source fields to provider_sources
+    try:
+        _migra_provider_to_source_structure(astrbot_config)
+    except Exception as e:
+        logger.error(f"Migration for provider-source structure failed: {e!s}")
+        logger.error(traceback.format_exc())

astrbot/core/utils/plugin_kv_store.py

@@ -0,0 +1,28 @@
+from typing import TypeVar
+
+from astrbot.core import sp
+
+SUPPORTED_VALUE_TYPES = int | float | str | bytes | bool | dict | list | None
+_VT = TypeVar("_VT")
+
+
+class PluginKVStoreMixin:
+    """为插件提供键值存储功能的 Mixin 类"""
+
+    plugin_id: str
+
+    async def put_kv_data(
+        self,
+        key: str,
+        value: SUPPORTED_VALUE_TYPES,
+    ) -> None:
+        """为指定插件存储一个键值对"""
+        await sp.put_async("plugin", self.plugin_id, key, value)
+
+    async def get_kv_data(self, key: str, default: _VT) -> _VT | None:
+        """获取指定插件存储的键值对"""
+        return await sp.get_async("plugin", self.plugin_id, key, default)
+
+    async def delete_kv_data(self, key: str) -> None:
+        """删除指定插件存储的键值对"""
+        await sp.remove_async("plugin", self.plugin_id, key)

astrbot/dashboard/routes/__init__.py

@@ -1,5 +1,6 @@
 from .auth import AuthRoute
 from .chat import ChatRoute
+from .command import CommandRoute
 from .config import ConfigRoute
 from .conversation import ConversationRoute
 from .file import FileRoute
@@ -17,6 +18,7 @@ from .update import UpdateRoute
 __all__ = [
     "AuthRoute",
     "ChatRoute",
+    "CommandRoute",
     "ConfigRoute",
     "ConversationRoute",
     "FileRoute",

astrbot/dashboard/routes/chat.py

@@ -227,16 +227,19 @@ class ChatRoute(Route):
         text: str,
         media_parts: list,
         reasoning: str,
+        agent_stats: dict,
     ):
         """保存 bot 消息到历史记录，返回保存的记录"""
         bot_message_parts = []
+        bot_message_parts.extend(media_parts)
         if text:
             bot_message_parts.append({"type": "plain", "text": text})
-        bot_message_parts.extend(media_parts)
 
         new_his = {"type": "bot", "message": bot_message_parts}
         if reasoning:
             new_his["reasoning"] = reasoning
+        if agent_stats:
+            new_his["agent_stats"] = agent_stats
 
         record = await self.platform_history_mgr.insert(
             platform_id="webchat",
@@ -294,7 +297,8 @@ class ChatRoute(Route):
             accumulated_parts = []
             accumulated_text = ""
             accumulated_reasoning = ""
-
+            tool_calls = {}
+            agent_stats = {}
             try:
                 async with track_conversation(self.running_convs, webchat_conv_id):
                     while True:
@@ -314,6 +318,16 @@ class ChatRoute(Route):
                         result_text = result["data"]
                         msg_type = result.get("type")
                         streaming = result.get("streaming", False)
+                        chain_type = result.get("chain_type")
+
+                        if chain_type == "agent_stats":
+                            stats_info = {
+                                "type": "agent_stats",
+                                "data": json.loads(result_text),
+                            }
+                            yield f"data: {json.dumps(stats_info, ensure_ascii=False)}\n\n"
+                            agent_stats = stats_info["data"]
+                            continue
 
                         # 发送 SSE 数据
                         try:
@@ -335,11 +349,35 @@ class ChatRoute(Route):
 
                         # 累积消息部分
                         if msg_type == "plain":
-                            chain_type = result.get("chain_type", "normal")
-                            if chain_type == "reasoning":
+                            chain_type = result.get("chain_type")
+                            if chain_type == "tool_call":
+                                tool_call = json.loads(result_text)
+                                tool_calls[tool_call.get("id")] = tool_call
+                                if accumulated_text:
+                                    # 如果累积了文本，则先保存文本
+                                    accumulated_parts.append(
+                                        {"type": "plain", "text": accumulated_text}
+                                    )
+                                    accumulated_text = ""
+                            elif chain_type == "tool_call_result":
+                                tcr = json.loads(result_text)
+                                tc_id = tcr.get("id")
+                                if tc_id in tool_calls:
+                                    tool_calls[tc_id]["result"] = tcr.get("result")
+                                    tool_calls[tc_id]["finished_ts"] = tcr.get("ts")
+                                accumulated_parts.append(
+                                    {
+                                        "type": "tool_call",
+                                        "tool_calls": [tool_calls[tc_id]],
+                                    }
+                                )
+                                tool_calls.pop(tc_id, None)
+                            elif chain_type == "reasoning":
                                 accumulated_reasoning += result_text
-                            else:
+                            elif streaming:
                                 accumulated_text += result_text
+                            else:
+                                accumulated_text = result_text
                         elif msg_type == "image":
                             filename = result_text.replace("[IMAGE]", "")
                             part = await self._create_attachment_from_file(
@@ -367,15 +405,20 @@ class ChatRoute(Route):
                         if msg_type == "end":
                             break
                         elif (
-                            (streaming and msg_type == "complete")
-                            or not streaming
-                            or msg_type == "break"
+                            (streaming and msg_type == "complete") or not streaming
+                            # or msg_type == "break"
                         ):
+                            if (
+                                chain_type == "tool_call"
+                                or chain_type == "tool_call_result"
+                            ):
+                                continue
                             saved_record = await self._save_bot_message(
                                 webchat_conv_id,
                                 accumulated_text,
                                 accumulated_parts,
                                 accumulated_reasoning,
+                                agent_stats,
                             )
                             # 发送保存的消息信息给前端
                             if saved_record and not client_disconnected:
@@ -390,11 +433,11 @@ class ChatRoute(Route):
                                     yield f"data: {json.dumps(saved_info, ensure_ascii=False)}\n\n"
                                 except Exception:
                                     pass
-                            # 重置累积变量 (对于 break 后的下一段消息)
-                            if msg_type == "break":
-                                accumulated_parts = []
-                                accumulated_text = ""
-                                accumulated_reasoning = ""
+                            accumulated_parts = []
+                            accumulated_text = ""
+                            accumulated_reasoning = ""
+                            # tool_calls = {}
+                            agent_stats = {}
             except BaseException as e:
                 logger.exception(f"WebChat stream unexpected error: {e}", exc_info=True)
 

astrbot/dashboard/routes/command.py

@@ -0,0 +1,82 @@
+from quart import request
+
+from astrbot.core.star.command_management import (
+    list_command_conflicts,
+    list_commands,
+)
+from astrbot.core.star.command_management import (
+    rename_command as rename_command_service,
+)
+from astrbot.core.star.command_management import (
+    toggle_command as toggle_command_service,
+)
+
+from .route import Response, Route, RouteContext
+
+
+class CommandRoute(Route):
+    def __init__(self, context: RouteContext) -> None:
+        super().__init__(context)
+        self.routes = {
+            "/commands": ("GET", self.get_commands),
+            "/commands/conflicts": ("GET", self.get_conflicts),
+            "/commands/toggle": ("POST", self.toggle_command),
+            "/commands/rename": ("POST", self.rename_command),
+        }
+        self.register_routes()
+
+    async def get_commands(self):
+        commands = await list_commands()
+        summary = {
+            "total": len(commands),
+            "disabled": len([cmd for cmd in commands if not cmd["enabled"]]),
+            "conflicts": len([cmd for cmd in commands if cmd.get("has_conflict")]),
+        }
+        return Response().ok({"items": commands, "summary": summary}).__dict__
+
+    async def get_conflicts(self):
+        conflicts = await list_command_conflicts()
+        return Response().ok(conflicts).__dict__
+
+    async def toggle_command(self):
+        data = await request.get_json()
+        handler_full_name = data.get("handler_full_name")
+        enabled = data.get("enabled")
+
+        if handler_full_name is None or enabled is None:
+            return Response().error("handler_full_name 与 enabled 均为必填。").__dict__
+
+        if isinstance(enabled, str):
+            enabled = enabled.lower() in ("1", "true", "yes", "on")
+
+        try:
+            await toggle_command_service(handler_full_name, bool(enabled))
+        except ValueError as exc:
+            return Response().error(str(exc)).__dict__
+
+        payload = await _get_command_payload(handler_full_name)
+        return Response().ok(payload).__dict__
+
+    async def rename_command(self):
+        data = await request.get_json()
+        handler_full_name = data.get("handler_full_name")
+        new_name = data.get("new_name")
+
+        if not handler_full_name or not new_name:
+            return Response().error("handler_full_name 与 new_name 均为必填。").__dict__
+
+        try:
+            await rename_command_service(handler_full_name, new_name)
+        except ValueError as exc:
+            return Response().error(str(exc)).__dict__
+
+        payload = await _get_command_payload(handler_full_name)
+        return Response().ok(payload).__dict__
+
+
+async def _get_command_payload(handler_full_name: str):
+    commands = await list_commands()
+    for cmd in commands:
+        if cmd["handler_full_name"] == handler_full_name:
+            return cmd
+    return {}