AstrBot 3.5.6__py3-none-any.whl → 4.7.0__py3-none-any.whl

astrbot/dashboard/routes/session_management.py

@@ -0,0 +1,354 @@
+from quart import request
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlmodel import col, select
+
+from astrbot.core import logger, sp
+from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
+from astrbot.core.db import BaseDatabase
+from astrbot.core.db.po import ConversationV2, Preference
+from astrbot.core.provider.entities import ProviderType
+
+from .route import Response, Route, RouteContext
+
+AVAILABLE_SESSION_RULE_KEYS = [
+    "session_service_config",
+    "session_plugin_config",
+    "kb_config",
+    f"provider_perf_{ProviderType.CHAT_COMPLETION.value}",
+    f"provider_perf_{ProviderType.SPEECH_TO_TEXT.value}",
+    f"provider_perf_{ProviderType.TEXT_TO_SPEECH.value}",
+]
+
+
+class SessionManagementRoute(Route):
+    def __init__(
+        self,
+        context: RouteContext,
+        db_helper: BaseDatabase,
+        core_lifecycle: AstrBotCoreLifecycle,
+    ) -> None:
+        super().__init__(context)
+        self.db_helper = db_helper
+        self.routes = {
+            "/session/list-rule": ("GET", self.list_session_rule),
+            "/session/update-rule": ("POST", self.update_session_rule),
+            "/session/delete-rule": ("POST", self.delete_session_rule),
+            "/session/batch-delete-rule": ("POST", self.batch_delete_session_rule),
+            "/session/active-umos": ("GET", self.list_umos),
+        }
+        self.conv_mgr = core_lifecycle.conversation_manager
+        self.core_lifecycle = core_lifecycle
+        self.register_routes()
+
+    async def _get_umo_rules(
+        self, page: int = 1, page_size: int = 10, search: str = ""
+    ) -> tuple[dict, int]:
+        """获取所有带有自定义规则的 umo 及其规则内容（支持分页和搜索）。
+
+        如果某个 umo 在 preference 中有以下字段，则表示有自定义规则：
+
+        1. session_service_config (包含了 是否启用这个umo, 这个umo是否启用 llm, 这个umo是否启用tts, umo自定义名称。)
+        2. session_plugin_config (包含了 这个 umo 的 plugin set)
+        3. provider_perf_{ProviderType.value} (包含了这个 umo 所选择使用的 provider 信息)
+        4. kb_config (包含了这个 umo 的知识库相关配置)
+
+        Args:
+            page: 页码，从 1 开始
+            page_size: 每页数量
+            search: 搜索关键词，匹配 umo 或 custom_name
+
+        Returns:
+            tuple[dict, int]: (umo_rules, total) - 分页后的 umo 规则和总数
+        """
+        umo_rules = {}
+        async with self.db_helper.get_db() as session:
+            session: AsyncSession
+            result = await session.execute(
+                select(Preference).where(
+                    col(Preference.scope) == "umo",
+                    col(Preference.key).in_(AVAILABLE_SESSION_RULE_KEYS),
+                )
+            )
+            prefs = result.scalars().all()
+            for pref in prefs:
+                umo_id = pref.scope_id
+                if umo_id not in umo_rules:
+                    umo_rules[umo_id] = {}
+                umo_rules[umo_id][pref.key] = pref.value["val"]
+
+        # 搜索过滤
+        if search:
+            search_lower = search.lower()
+            filtered_rules = {}
+            for umo_id, rules in umo_rules.items():
+                # 匹配 umo
+                if search_lower in umo_id.lower():
+                    filtered_rules[umo_id] = rules
+                    continue
+                # 匹配 custom_name
+                svc_config = rules.get("session_service_config", {})
+                custom_name = svc_config.get("custom_name", "") if svc_config else ""
+                if custom_name and search_lower in custom_name.lower():
+                    filtered_rules[umo_id] = rules
+            umo_rules = filtered_rules
+
+        # 获取总数
+        total = len(umo_rules)
+
+        # 分页处理
+        all_umo_ids = list(umo_rules.keys())
+        start_idx = (page - 1) * page_size
+        end_idx = start_idx + page_size
+        paginated_umo_ids = all_umo_ids[start_idx:end_idx]
+
+        # 只返回分页后的数据
+        paginated_rules = {umo_id: umo_rules[umo_id] for umo_id in paginated_umo_ids}
+
+        return paginated_rules, total
+
+    async def list_session_rule(self):
+        """获取所有自定义的规则（支持分页和搜索）
+
+        返回已配置规则的 umo 列表及其规则内容，以及可用的 personas 和 providers
+
+        Query 参数:
+            page: 页码，默认为 1
+            page_size: 每页数量，默认为 10
+            search: 搜索关键词，匹配 umo 或 custom_name
+        """
+        try:
+            # 获取分页和搜索参数
+            page = request.args.get("page", 1, type=int)
+            page_size = request.args.get("page_size", 10, type=int)
+            search = request.args.get("search", "", type=str).strip()
+
+            # 参数校验
+            if page < 1:
+                page = 1
+            if page_size < 1:
+                page_size = 10
+            if page_size > 100:
+                page_size = 100
+
+            umo_rules, total = await self._get_umo_rules(
+                page=page, page_size=page_size, search=search
+            )
+
+            # 构建规则列表
+            rules_list = []
+            for umo, rules in umo_rules.items():
+                rule_info = {
+                    "umo": umo,
+                    "rules": rules,
+                }
+                # 解析 umo 格式: 平台:消息类型:会话ID
+                parts = umo.split(":")
+                if len(parts) >= 3:
+                    rule_info["platform"] = parts[0]
+                    rule_info["message_type"] = parts[1]
+                    rule_info["session_id"] = parts[2]
+                rules_list.append(rule_info)
+
+            # 获取可用的 providers 和 personas
+            provider_manager = self.core_lifecycle.provider_manager
+            persona_mgr = self.core_lifecycle.persona_mgr
+
+            available_personas = [
+                {"name": p["name"], "prompt": p.get("prompt", "")}
+                for p in persona_mgr.personas_v3
+            ]
+
+            available_chat_providers = [
+                {
+                    "id": p.meta().id,
+                    "name": p.meta().id,
+                    "model": p.meta().model,
+                }
+                for p in provider_manager.provider_insts
+            ]
+
+            available_stt_providers = [
+                {
+                    "id": p.meta().id,
+                    "name": p.meta().id,
+                    "model": p.meta().model,
+                }
+                for p in provider_manager.stt_provider_insts
+            ]
+
+            available_tts_providers = [
+                {
+                    "id": p.meta().id,
+                    "name": p.meta().id,
+                    "model": p.meta().model,
+                }
+                for p in provider_manager.tts_provider_insts
+            ]
+
+            return (
+                Response()
+                .ok(
+                    {
+                        "rules": rules_list,
+                        "total": total,
+                        "page": page,
+                        "page_size": page_size,
+                        "available_personas": available_personas,
+                        "available_chat_providers": available_chat_providers,
+                        "available_stt_providers": available_stt_providers,
+                        "available_tts_providers": available_tts_providers,
+                        "available_rule_keys": AVAILABLE_SESSION_RULE_KEYS,
+                    }
+                )
+                .__dict__
+            )
+        except Exception as e:
+            logger.error(f"获取规则列表失败: {e!s}")
+            return Response().error(f"获取规则列表失败: {e!s}").__dict__
+
+    async def update_session_rule(self):
+        """更新某个 umo 的自定义规则
+
+        请求体:
+        {
+            "umo": "平台:消息类型:会话ID",
+            "rule_key": "session_service_config" | "session_plugin_config" | "kb_config" | "provider_perf_xxx",
+            "rule_value": {...}  // 规则值，具体结构根据 rule_key 不同而不同
+        }
+        """
+        try:
+            data = await request.get_json()
+            umo = data.get("umo")
+            rule_key = data.get("rule_key")
+            rule_value = data.get("rule_value")
+
+            if not umo:
+                return Response().error("缺少必要参数: umo").__dict__
+            if not rule_key:
+                return Response().error("缺少必要参数: rule_key").__dict__
+            if rule_key not in AVAILABLE_SESSION_RULE_KEYS:
+                return Response().error(f"不支持的规则键: {rule_key}").__dict__
+
+            # 使用 shared preferences 更新规则
+            await sp.session_put(umo, rule_key, rule_value)
+
+            return (
+                Response()
+                .ok({"message": f"规则 {rule_key} 已更新", "umo": umo})
+                .__dict__
+            )
+        except Exception as e:
+            logger.error(f"更新会话规则失败: {e!s}")
+            return Response().error(f"更新会话规则失败: {e!s}").__dict__
+
+    async def delete_session_rule(self):
+        """删除某个 umo 的自定义规则
+
+        请求体:
+        {
+            "umo": "平台:消息类型:会话ID",
+            "rule_key": "session_service_config" | "session_plugin_config" | ... (可选，不传则删除所有规则)
+        }
+        """
+        try:
+            data = await request.get_json()
+            umo = data.get("umo")
+            rule_key = data.get("rule_key")
+
+            if not umo:
+                return Response().error("缺少必要参数: umo").__dict__
+
+            if rule_key:
+                # 删除单个规则
+                if rule_key not in AVAILABLE_SESSION_RULE_KEYS:
+                    return Response().error(f"不支持的规则键: {rule_key}").__dict__
+                await sp.session_remove(umo, rule_key)
+                return (
+                    Response()
+                    .ok({"message": f"规则 {rule_key} 已删除", "umo": umo})
+                    .__dict__
+                )
+            else:
+                # 删除该 umo 的所有规则
+                await sp.clear_async("umo", umo)
+                return Response().ok({"message": "所有规则已删除", "umo": umo}).__dict__
+        except Exception as e:
+            logger.error(f"删除会话规则失败: {e!s}")
+            return Response().error(f"删除会话规则失败: {e!s}").__dict__
+
+    async def batch_delete_session_rule(self):
+        """批量删除多个 umo 的自定义规则
+
+        请求体:
+        {
+            "umos": ["平台:消息类型:会话ID", ...]  // umo 列表
+        }
+        """
+        try:
+            data = await request.get_json()
+            umos = data.get("umos", [])
+
+            if not umos:
+                return Response().error("缺少必要参数: umos").__dict__
+
+            if not isinstance(umos, list):
+                return Response().error("参数 umos 必须是数组").__dict__
+
+            # 批量删除
+            deleted_count = 0
+            failed_umos = []
+            for umo in umos:
+                try:
+                    await sp.clear_async("umo", umo)
+                    deleted_count += 1
+                except Exception as e:
+                    logger.error(f"删除 umo {umo} 的规则失败: {e!s}")
+                    failed_umos.append(umo)
+
+            if failed_umos:
+                return (
+                    Response()
+                    .ok(
+                        {
+                            "message": f"已删除 {deleted_count} 条规则，{len(failed_umos)} 条删除失败",
+                            "deleted_count": deleted_count,
+                            "failed_umos": failed_umos,
+                        }
+                    )
+                    .__dict__
+                )
+            else:
+                return (
+                    Response()
+                    .ok(
+                        {
+                            "message": f"已删除 {deleted_count} 条规则",
+                            "deleted_count": deleted_count,
+                        }
+                    )
+                    .__dict__
+                )
+        except Exception as e:
+            logger.error(f"批量删除会话规则失败: {e!s}")
+            return Response().error(f"批量删除会话规则失败: {e!s}").__dict__
+
+    async def list_umos(self):
+        """列出所有有对话记录的 umo，从 Conversations 表中找
+
+        仅返回 umo 字符串列表，用于用户在创建规则时选择 umo
+        """
+        try:
+            # 从 Conversation 表获取所有 distinct user_id (即 umo)
+            async with self.db_helper.get_db() as session:
+                session: AsyncSession
+                result = await session.execute(
+                    select(ConversationV2.user_id)
+                    .distinct()
+                    .order_by(ConversationV2.user_id)
+                )
+                umos = [row[0] for row in result.fetchall()]
+
+            return Response().ok({"umos": umos}).__dict__
+        except Exception as e:
+            logger.error(f"获取 UMO 列表失败: {e!s}")
+            return Response().error(f"获取 UMO 列表失败: {e!s}").__dict__

astrbot/dashboard/routes/stat.py

@@ -1,14 +1,19 @@
+import threading
+import time
 import traceback
+
+import aiohttp
 import psutil
-import time
-import threading
-from .route import Route, Response, RouteContext
-from astrbot.core import logger
 from quart import request
+
+from astrbot.core import DEMO_MODE, logger
+from astrbot.core.config import VERSION
 from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
 from astrbot.core.db import BaseDatabase
-from astrbot.core.config import VERSION
-from astrbot.core import DEMO_MODE
+from astrbot.core.db.migration.helper import check_migration_needed_v4
+from astrbot.core.utils.io import get_dashboard_version
+
+from .route import Response, Route, RouteContext
 
 
 class StatRoute(Route):
@@ -24,6 +29,7 @@ class StatRoute(Route):
             "/stat/version": ("GET", self.get_version),
             "/stat/start-time": ("GET", self.get_start_time),
             "/stat/restart-core": ("POST", self.restart_core),
+            "/stat/test-ghproxy-connection": ("POST", self.test_ghproxy_connection),
         }
         self.db_helper = db_helper
         self.register_routes()
@@ -40,13 +46,36 @@ class StatRoute(Route):
         await self.core_lifecycle.restart()
         return Response().ok().__dict__
 
-    def format_sec(self, sec: int):
-        m, s = divmod(sec, 60)
-        h, m = divmod(m, 60)
-        return f"{h}小时{m}分{s}秒"
+    def _get_running_time_components(self, total_seconds: int):
+        """将总秒数转换为时分秒组件"""
+        minutes, seconds = divmod(total_seconds, 60)
+        hours, minutes = divmod(minutes, 60)
+        return {"hours": hours, "minutes": minutes, "seconds": seconds}
+
+    def is_default_cred(self):
+        username = self.config["dashboard"]["username"]
+        password = self.config["dashboard"]["password"]
+        return (
+            username == "astrbot"
+            and password == "77b90590a8945a7d36c963981a307dc9"
+            and not DEMO_MODE
+        )
 
     async def get_version(self):
-        return Response().ok({"version": VERSION}).__dict__
+        need_migration = await check_migration_needed_v4(self.core_lifecycle.db)
+
+        return (
+            Response()
+            .ok(
+                {
+                    "version": VERSION,
+                    "dashboard_version": await get_dashboard_version(),
+                    "change_pwd_hint": self.is_default_cred(),
+                    "need_migration": need_migration,
+                },
+            )
+            .__dict__
+        )
 
     async def get_start_time(self):
         return Response().ok({"start_time": self.core_lifecycle.start_time}).__dict__
@@ -61,7 +90,7 @@ class StatRoute(Route):
             message_time_based_stats = []
 
             idx = 0
-            for bucket_end in range(start_time, now, 1800):
+            for bucket_end in range(start_time, now, 3600):
                 cnt = 0
                 while (
                     idx < len(stat.platform)
@@ -87,21 +116,24 @@ class StatRoute(Route):
                 }
                 plugin_info.append(info)
 
+            # 计算运行时长组件
+            running_time = self._get_running_time_components(
+                int(time.time()) - self.core_lifecycle.start_time,
+            )
+
             stat_dict.update(
                 {
                     "platform": self.db_helper.get_grouped_base_stats(
-                        offset_sec
+                        offset_sec,
                     ).platform,
                     "message_count": self.db_helper.get_total_message_count() or 0,
                     "platform_count": len(
-                        self.core_lifecycle.platform_manager.get_insts()
+                        self.core_lifecycle.platform_manager.get_insts(),
                     ),
                     "plugin_count": len(plugins),
                     "plugins": plugin_info,
                     "message_time_series": message_time_based_stats,
-                    "running": self.format_sec(
-                        int(time.time()) - self.core_lifecycle.start_time
-                    ),
+                    "running": running_time,  # 现在返回时间组件而不是格式化的字符串
                     "memory": {
                         "process": psutil.Process().memory_info().rss >> 20,
                         "system": psutil.virtual_memory().total >> 20,
@@ -109,10 +141,45 @@ class StatRoute(Route):
                     "cpu_percent": round(cpu_percent, 1),
                     "thread_count": thread_count,
                     "start_time": self.core_lifecycle.start_time,
-                }
+                },
             )
 
             return Response().ok(stat_dict).__dict__
         except Exception as e:
             logger.error(traceback.format_exc())
             return Response().error(e.__str__()).__dict__
+
+    async def test_ghproxy_connection(self):
+        """测试 GitHub 代理连接是否可用。"""
+        try:
+            data = await request.get_json()
+            proxy_url: str = data.get("proxy_url")
+
+            if not proxy_url:
+                return Response().error("proxy_url is required").__dict__
+
+            proxy_url = proxy_url.rstrip("/")
+
+            test_url = f"{proxy_url}/https://github.com/AstrBotDevs/AstrBot/raw/refs/heads/master/.python-version"
+            start_time = time.time()
+
+            async with (
+                aiohttp.ClientSession() as session,
+                session.get(
+                    test_url,
+                    timeout=aiohttp.ClientTimeout(total=10),
+                ) as response,
+            ):
+                if response.status == 200:
+                    end_time = time.time()
+                    _ = await response.text()
+                    ret = {
+                        "latency": round((end_time - start_time) * 1000, 2),
+                    }
+                    return Response().ok(data=ret).__dict__
+                return (
+                    Response().error(f"Failed. Status code: {response.status}").__dict__
+                )
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(f"Error: {e!s}").__dict__

astrbot/dashboard/routes/static_file.py

@@ -12,7 +12,10 @@ class StaticFileRoute(Route):
             "/logs",
             "/extension",
             "/dashboard/default",
-            "/project-atri",
+            "/alkaid",
+            "/alkaid/knowledge-base",
+            "/alkaid/long-term-memory",
+            "/alkaid/other",
             "/console",
             "/chat",
             "/settings",
@@ -28,7 +31,7 @@ class StaticFileRoute(Route):
 
         @self.app.errorhandler(404)
         async def page_not_found(e):
-            return "404 Not found。如果你初次使用打开面板发现 404, 请参考文档: https://astrbot.app/faq.html。"
+            return "404 Not found。如果你初次使用打开面板发现 404, 请参考文档: https://astrbot.app/faq.html。如果你正在测试回调地址可达性，显示这段文字说明测试成功了。"
 
     async def index(self):
         return await self.app.send_static_file("index.html")