Method dfs

toolbench/tooleval/evaluation/dataclass.py:75–88

(node:ExecutionNode)

        all_visited_nodes = set()
        for node in all_start_nodes:
            def dfs(node:ExecutionNode)->Dict[Any,Any]:
                if node.node_id in all_visited_nodes:
                    return None
                all_visited_nodes.add(node.node_id)
                node_json={
                    'role':node.role,
                    'message':node.message if node.role != 'system' and node.role !='user' else '',
                    'next':[]
                }
                for next_node in self.get_adjacent_node(node):
                    next_node_dict = dfs(self.nodes[next_node])
                    if next_node_dict is not None:
                        node_json['next'].append(next_node_dict)
                return node_json
            
            data.append(dfs(self.nodes[node]))