LightRAG/get_all_edges_nx.py

import networkx as nx

G = nx.read_graphml("./dickensTestEmbedcall/graph_chunk_entity_relation.graphml")


def get_all_edges_and_nodes(G):
    # Get all edges and their properties
    edges_with_properties = []
    for u, v, data in G.edges(data=True):
        edges_with_properties.append(
            {
                "start": u,
                "end": v,
                "label": data.get(
                    "label", ""
                ),  # Assuming 'label' is used for edge type
                "properties": data,
                "start_node_properties": G.nodes[u],
                "end_node_properties": G.nodes[v],
            }
        )

    return edges_with_properties


# Example usage
if __name__ == "__main__":
    # Assume G is your NetworkX graph loaded from Neo4j

    all_edges = get_all_edges_and_nodes(G)

    # Print all edges and node properties
    for edge in all_edges:
        print(f"Edge Label: {edge['label']}")
        print(f"Edge Properties: {edge['properties']}")
        print(f"Start Node: {edge['start']}")
        print(f"Start Node Properties: {edge['start_node_properties']}")
        print(f"End Node: {edge['end']}")
        print(f"End Node Properties: {edge['end_node_properties']}")
        print("---")