Function add_node_recursive

visualize_mcts.py:353–405

(node_id: str)

    
    # 添加节点和边的递归函数
    def add_node_recursive(node_id: str):
        if node_id not in nodes:
            return
        
        node = nodes[node_id]
        
        # 构建节点标签
        status_sym = "✓" if node.run_status == "OK" else "✗" if node.run_status == "FAIL" else "?"
        stage_name = {"root": "R", "draft": "D", "improve": "I", "debug": "DB"}.get(node.stage, node.stage[:1].upper())
        
        label_lines = [
            f"{status_sym} {stage_name} {node.id[:8]}",
        ]
        
        if node.metric is not None:
            label_lines.append(f"metric: {node.metric:.4f}")
        else:
            label_lines.append("metric: N/A")
        
        if node.is_terminal:
            label_lines.append("[TERMINAL]")
        
        label = "\n".join(label_lines)
        
        # 根据状态和阶段设置颜色
        if node.run_status == "OK":
            if node.metric is not None and node.metric > 0.9:
                fillcolor = "#90EE90"  # 浅绿色 - 高metric
            else:
                fillcolor = "#E8F5E9"  # 浅绿色 - 成功但metric一般
        elif node.run_status == "FAIL":
            fillcolor = "#FFCDD2"  # 浅红色 - 失败
        else:
            fillcolor = "#F5F5F5"  # 灰色 - 未知
        
        # 根据阶段调整颜色
        if node.stage == "root":
            fillcolor = "#BBDEFB"  # 浅蓝色 - 根节点
        elif node.stage == "draft":
            fillcolor = "#FFF3E0"  # 浅橙色 - draft节点
        
        # 添加节点
        g.node(
            node_id,
            label=label,
            fillcolor=fillcolor
        )
        
        # 递归添加子节点和边
        for child_id in node.children_ids:
            if child_id in nodes:
                add_node_recursive(child_id)
                g.edge(node_id, child_id)
    
    # 从根节点开始构建
    if root_id not in nodes: