Function parse_nodes_from_json

src/WorkFlow.py:32–45

Parses node data from a subgraph's JSON structure. Args: graph_data: A dictionary representing a subgraph. Returns: A dictionary of NodeData objects keyed by their unique IDs.

(graph_data: Dict[str, Any])

    return func

def parse_nodes_from_json(graph_data: Dict[str, Any]) -> Dict[str, NodeData]:
    """
    Parses node data from a subgraph's JSON structure.

    Args:
        graph_data: A dictionary representing a subgraph.
    Returns:
        A dictionary of NodeData objects keyed by their unique IDs.
    """
    node_map = {}
    for node_data in graph_data.get("nodes", []):
        node = NodeData.from_dict(node_data)
        node_map[node.uniq_id] = node
    return node_map

def find_nodes_by_type(node_map: Dict[str, NodeData], node_type: str) -> List[NodeData]:
    return [node for node in node_map.values() if node.type == node_type]