Node (tree_search.py)

MCTS search node; parses LLM state JSON, tracks value/visits/parent/children, enables selection, expansion, rollout, backup in tree search.

import json
import os
import pickle
import shutil

import numpy as np
import pandas as pd

from metagpt.ext.sela.data.custom_task import (
    get_mle_bench_requirements,
    get_mle_task_id,
)
from metagpt.ext.sela.data.dataset import (
    generate_task_requirement,
    get_split_dataset_path,
)
from metagpt.ext.sela.evaluation.evaluation import evaluate_score
from metagpt.ext.sela.experimenter import Experimenter, TimeoutException
from metagpt.ext.sela.insights.instruction_generator import InstructionGenerator
from metagpt.ext.sela.utils import get_exp_pool_path, load_execute_notebook, mcts_logger
from metagpt.tools.tool_recommend import ToolRecommender
from metagpt.utils.common import read_json_file


def initialize_di_root_node(state: dict, reflection: bool = True):
    """
    Initia

... (truncated -- full source via MCP)