openclaw-agent-dashboard 1.0.4

package/src/backend/data/chain_reader.py

@@ -0,0 +1,299 @@
+"""
+任务链路读取器 - 解析 Agent 间的任务派发关系
+"""
+import json
+import os
+from pathlib import Path
+from typing import List, Dict, Any, Optional
+from dataclasses import dataclass, asdict
+from enum import Enum
+from datetime import datetime
+
+
+class ChainNodeStatus(str, Enum):
+    PENDING = "pending"
+    RUNNING = "running"
+    COMPLETED = "completed"
+    ERROR = "error"
+
+
+class ChainStatus(str, Enum):
+    RUNNING = "running"
+    COMPLETED = "completed"
+    ERROR = "error"
+
+
+from data.config_reader import get_openclaw_root
+
+
+def _get_agents_config() -> Dict[str, Any]:
+    """获取 agents 配置"""
+    config_file = get_openclaw_root() / "openclaw.json"
+    if not config_file.exists():
+        return {}
+
+    try:
+        with open(config_file, 'r', encoding='utf-8') as f:
+            return json.load(f)
+    except:
+        return {}
+
+
+def _get_agent_info(agent_id: str) -> Dict[str, Any]:
+    """获取单个 agent 的信息"""
+    config = _get_agents_config()
+    agents = config.get('agents', {}).get('list', [])
+    for a in agents:
+        if a.get('id') == agent_id:
+            return a
+    return {}
+
+
+def _parse_session_key(session_key: str) -> Dict[str, str]:
+    """解析 session key，提取 agent_id 等信息"""
+    # 格式: agent:agent-id:subagent:uuid 或 agent:agent-id:main
+    parts = session_key.split(':')
+    result = {'type': parts[0] if len(parts) > 0 else 'unknown'}
+
+    if len(parts) >= 2:
+        result['agent_id'] = parts[1]
+    if len(parts) >= 3:
+        result['session_type'] = parts[2]  # main 或 subagent
+    if len(parts) >= 4:
+        result['uuid'] = parts[3]
+
+    return result
+
+
+def _load_runs() -> Dict[str, Any]:
+    """加载 runs.json"""
+    runs_file = get_openclaw_root() / "subagents" / "runs.json"
+    if not runs_file.exists():
+        return {"version": 2, "runs": {}}
+
+    try:
+        with open(runs_file, 'r', encoding='utf-8') as f:
+            return json.load(f)
+    except:
+        return {"version": 2, "runs": {}}
+
+
+def _get_workflow_state(project_id: str) -> Dict[str, Any]:
+    """获取项目的 workflow 状态"""
+    # 尝试多个可能的项目路径
+    possible_paths = [
+        get_openclaw_root() / f"workspace-{project_id}" / ".staging" / "workflow_state.json",
+        Path.home() / "vrt-projects" / "projects" / project_id / ".staging" / "workflow_state.json",
+    ]
+
+    for path in possible_paths:
+        if path.exists():
+            try:
+                with open(path, 'r', encoding='utf-8') as f:
+                    return json.load(f)
+            except:
+                pass
+
+    return {}
+
+
+def build_task_chains(limit: int = 20) -> List[Dict[str, Any]]:
+    """
+    构建任务链路列表
+
+    通过解析 runs.json 中的派发关系，构建完整的任务执行链路
+    """
+    runs_data = _load_runs()
+    runs = runs_data.get('runs', {})
+
+    if not runs:
+        return []
+
+    # 按 requesterSessionKey 分组，构建链路
+    chains_map: Dict[str, Dict[str, Any]] = {}
+
+    for run_id, run_info in runs.items():
+        if not isinstance(run_info, dict):
+            continue
+
+        requester_key = run_info.get('requesterSessionKey', '')
+        child_key = run_info.get('childSessionKey', '')
+
+        if not requester_key or not child_key:
+            continue
+
+        # 解析 requester 和 child
+        requester = _parse_session_key(requester_key)
+        child = _parse_session_key(child_key)
+
+        requester_id = requester.get('agent_id', 'main')
+        child_id = child.get('agent_id', 'unknown')
+
+        # 使用 requester 的 session 作为链路 ID（简化处理）
+        chain_id = requester_key.split(':subagent:')[0] if ':subagent:' in requester_key else requester_key
+
+        if chain_id not in chains_map:
+            chains_map[chain_id] = {
+                'chainId': chain_id,
+                'rootTask': '',
+                'startedAt': None,
+                'status': ChainStatus.RUNNING.value,
+                'nodes': {},
+                'edges': [],
+                'projectId': None
+            }
+
+        chain = chains_map[chain_id]
+
+        # 添加 requester 节点（如果不存在）
+        if requester_id not in chain['nodes']:
+            agent_info = _get_agent_info(requester_id)
+            chain['nodes'][requester_id] = {
+                'agentId': requester_id,
+                'agentName': agent_info.get('name', requester_id),
+                'role': requester_id.split('-')[0] if '-' in requester_id else requester_id,
+                'status': ChainNodeStatus.COMPLETED.value,
+                'startedAt': None,
+                'endedAt': None,
+                'duration': None,
+                'task': None,
+                'runId': None,
+                'input': None,
+                'output': None,
+                'artifacts': [],
+                'toolCallCount': 0,
+                'tokenUsage': {'input': 0, 'output': 0}
+            }
+
+        # 添加子节点
+        agent_info = _get_agent_info(child_id)
+        started_at = run_info.get('startedAt')
+        ended_at = run_info.get('endedAt')
+
+        node_status = ChainNodeStatus.PENDING.value
+        if started_at and ended_at:
+            node_status = ChainNodeStatus.COMPLETED.value if run_info.get('outcome') == 'ok' else ChainNodeStatus.ERROR.value
+        elif started_at:
+            node_status = ChainNodeStatus.RUNNING.value
+
+        chain['nodes'][child_id] = {
+            'agentId': child_id,
+            'agentName': agent_info.get('name', child_id),
+            'role': child_id.split('-')[0] if '-' in child_id else child_id,
+            'status': node_status,
+            'startedAt': started_at,
+            'endedAt': ended_at,
+            'duration': (ended_at - started_at) if started_at and ended_at else None,
+            'task': run_info.get('task', ''),
+            'runId': run_id,
+            'input': None,
+            'output': None,
+            'artifacts': [],
+            'toolCallCount': 0,
+            'tokenUsage': {'input': 0, 'output': 0}
+        }
+
+        # 添加边
+        edge = {'from': requester_id, 'to': child_id}
+        if edge not in chain['edges']:
+            chain['edges'].append(edge)
+
+        # 更新链路开始时间
+        if started_at:
+            if chain['startedAt'] is None or started_at < chain['startedAt']:
+                chain['startedAt'] = started_at
+
+        # 设置根任务（使用第一个子节点的任务）
+        if not chain['rootTask'] and run_info.get('task'):
+            chain['rootTask'] = run_info.get('task')
+
+        # 保存 archiveAtMs（用于超时倒计时）
+        if run_info.get('archiveAtMs') and not chain.get('archiveAtMs'):
+            chain['archiveAtMs'] = run_info.get('archiveAtMs')
+
+    # 转换节点为列表并计算统计信息
+    chains = []
+    for chain_id, chain in chains_map.items():
+        nodes_list = list(chain['nodes'].values())
+
+        # 计算进度
+        completed = sum(1 for n in nodes_list if n['status'] == ChainNodeStatus.COMPLETED.value)
+        running = sum(1 for n in nodes_list if n['status'] == ChainNodeStatus.RUNNING.value)
+        total = len(nodes_list)
+
+        progress = completed / total if total > 0 else 0
+
+        # 计算总耗时
+        total_duration = sum(n['duration'] or 0 for n in nodes_list)
+
+        # 确定链路状态
+        if any(n['status'] == ChainNodeStatus.ERROR.value for n in nodes_list):
+            chain_status = ChainStatus.ERROR.value
+        elif running > 0:
+            chain_status = ChainStatus.RUNNING.value
+        else:
+            chain_status = ChainStatus.COMPLETED.value
+
+        chains.append({
+            'chainId': chain_id,
+            'projectId': chain.get('projectId'),
+            'rootTask': chain.get('rootTask', '未知任务'),
+            'startedAt': chain.get('startedAt'),
+            'archiveAtMs': chain.get('archiveAtMs'),
+            'status': chain_status,
+            'nodes': nodes_list,
+            'edges': chain['edges'],
+            'progress': progress,
+            'completedNodes': completed,
+            'totalNodes': total,
+            'totalDuration': total_duration
+        })
+
+    # 按开始时间排序
+    chains.sort(key=lambda x: x.get('startedAt') or 0, reverse=True)
+
+    return chains[:limit]
+
+
+def get_task_chain(chain_id: str) -> Optional[Dict[str, Any]]:
+    """获取单个任务链的详情"""
+    chains = build_task_chains(limit=100)
+    for chain in chains:
+        if chain['chainId'] == chain_id:
+            return chain
+    return None
+
+
+def get_active_chain() -> Optional[Dict[str, Any]]:
+    """获取当前活跃的任务链（正在执行的）"""
+    chains = build_task_chains(limit=50)
+    for chain in chains:
+        if chain['status'] == ChainStatus.RUNNING.value:
+            return chain
+    return None
+
+
+def get_chains_summary() -> Dict[str, Any]:
+    """获取任务链摘要统计"""
+    chains = build_task_chains(limit=100)
+
+    running = sum(1 for c in chains if c['status'] == ChainStatus.RUNNING.value)
+    completed = sum(1 for c in chains if c['status'] == ChainStatus.COMPLETED.value)
+    error = sum(1 for c in chains if c['status'] == ChainStatus.ERROR.value)
+
+    return {
+        'total': len(chains),
+        'running': running,
+        'completed': completed,
+        'error': error,
+        'chains': [
+            {
+                'chainId': c['chainId'],
+                'rootTask': c['rootTask'][:50] + '...' if len(c['rootTask']) > 50 else c['rootTask'],
+                'status': c['status'],
+                'progress': c['progress'],
+                'startedAt': c['startedAt']
+            }
+            for c in chains[:10]
+        ]
+    }

package/src/backend/data/config_reader.py

@@ -0,0 +1,153 @@
+"""
+配置读取器 - 读取 openclaw.json
+支持 OPENCLAW_STATE_DIR、OPENCLAW_HOME 环境变量（跨平台，含 Windows）
+"""
+import json
+import os
+from pathlib import Path
+from typing import List, Dict, Any
+
+
+def get_openclaw_root() -> Path:
+    """OpenClaw 根目录，统一解析优先级（兼容 Windows）：
+    1. OPENCLAW_STATE_DIR（最高优先级）
+    2. OPENCLAW_HOME（兼容两种写法：直接指向 .openclaw，或指向用户 Home 需拼接 .openclaw）
+    3. ~/.openclaw（兜底）
+    """
+    # 1. OPENCLAW_STATE_DIR
+    env_state = os.environ.get("OPENCLAW_STATE_DIR")
+    if env_state:
+        return Path(env_state).expanduser().resolve()
+
+    # 2. OPENCLAW_HOME
+    env_home = os.environ.get("OPENCLAW_HOME")
+    if env_home:
+        p = Path(env_home).expanduser().resolve()
+        # 兼容两种写法：直接指向 .openclaw 目录，或指向用户 Home
+        if p.name in (".openclaw", "openclaw"):
+            return p
+        return p / ".openclaw"
+
+    # 3. 兜底
+    return Path.home() / ".openclaw"
+
+
+def load_config() -> Dict[str, Any]:
+    """加载 openclaw.json"""
+    config_path = get_openclaw_root() / "openclaw.json"
+    if not config_path.exists():
+        return {}
+    
+    with open(config_path, 'r', encoding='utf-8') as f:
+        return json.load(f)
+
+
+def get_agents_list() -> List[Dict[str, Any]]:
+    """获取 Agent 列表"""
+    config = load_config()
+    agents = config.get('agents')
+    if agents is None or not isinstance(agents, dict):
+        return []
+    return agents.get('list', [])
+
+
+def get_main_agent_id() -> str:
+    """获取主 Agent ID（配置中 id 为 main 的，或列表第一个）"""
+    agents = get_agents_list()
+    for a in agents:
+        if a.get('id') == 'main':
+            return 'main'
+    return agents[0].get('id', 'main') if agents else 'main'
+
+
+def get_workspace_paths() -> List[Path]:
+    """获取所有 Agent 的 workspace 路径（用于 model-failures.log 等）"""
+    agents = get_agents_list()
+    paths = []
+    seen = set()
+    for a in agents:
+        ws = a.get('workspace')
+        if ws and ws not in seen:
+            p = Path(ws).expanduser() if isinstance(ws, str) else Path(ws)
+            if p.exists():
+                paths.append(p)
+                seen.add(ws)
+    if not paths:
+        paths.append(get_openclaw_root() / "workspace-main")
+    return paths
+
+
+def get_agent_config(agent_id: str) -> Dict[str, Any]:
+    """获取单个 Agent 配置"""
+    agents = get_agents_list()
+    for agent in agents:
+        if agent.get('id') == agent_id:
+            return agent
+    return {}
+
+
+def get_default_config() -> Dict[str, Any]:
+    """获取默认配置"""
+    config = load_config()
+    agents = config.get('agents')
+    if agents is None or not isinstance(agents, dict):
+        return {}
+    return agents.get('defaults', {})
+
+
+def get_agent_models(agent_id: str) -> Dict[str, Any]:
+    """获取 Agent 的模型配置（primary + fallbacks）"""
+    agent = get_agent_config(agent_id)
+    model_cfg = agent.get('model') or {}
+    defaults = get_default_config()
+    default_model = defaults.get('model', {})
+    primary = model_cfg.get('primary') or default_model.get('primary') or ''
+    fallbacks = model_cfg.get('fallbacks') or default_model.get('fallbacks') or []
+    return {'primary': primary, 'fallbacks': fallbacks}
+
+
+def get_models_configured_by_agents() -> List[str]:
+    """
+    从配置中收集「各 Agent 实际配置使用」的模型 ID（仅 primary + fallbacks）。
+    用于协作流程右侧模型面板：只显示有 Agent 配置的模型，不含白名单中未使用的。
+    """
+    agents = get_agents_list()
+    model_ids = set()
+    defaults = get_default_config()
+    default_model = defaults.get('model', {})
+    if default_model.get('primary'):
+        model_ids.add(default_model['primary'])
+    for fb in default_model.get('fallbacks') or []:
+        model_ids.add(fb)
+    for agent in agents:
+        cfg = get_agent_models(agent.get('id', ''))
+        if cfg.get('primary'):
+            model_ids.add(cfg['primary'])
+        for fb in cfg.get('fallbacks', []):
+            model_ids.add(fb)
+    return sorted(model_ids)
+
+
+def get_all_models_from_agents() -> List[str]:
+    """
+    从配置中收集模型 ID（provider/model 格式），用于无 models.providers 时的下拉选项。
+    来源（合并去重）：
+    1. 各 Agent 实际配置（primary + fallbacks）
+    2. agents.defaults.models（白名单 key，确保配置过的能选）
+    """
+    model_ids = set(get_models_configured_by_agents())
+    defaults = get_default_config()
+    models_cfg = defaults.get('models', {}) or {}
+    if isinstance(models_cfg, dict):
+        for mid in models_cfg.keys():
+            if mid and isinstance(mid, str):
+                model_ids.add(mid)
+    return sorted(model_ids)
+
+
+def get_model_display_name(model_id: str) -> str:
+    """获取模型显示名。展示策略：使用 id 不用别名（与 OpenClaw 白名单逻辑一致）"""
+    if not model_id:
+        return ''
+    parts = model_id.split('/')
+    return parts[-1] if len(parts) > 1 else model_id