Merge pull request #51 from tianhaolin1991/patch-1

Fix NetworkXStorage get_types method
2025-03-15 13:19:23 +08:00
parent 1c59a1568e 93b6e5bd1a
commit 77aa9c911d
1 changed files with 2 additions and 2 deletions
--- a/minirag/kg/networkx_impl.py
+++ b/minirag/kg/networkx_impl.py
@@ -152,8 +152,8 @@ class NetworkXStorage(BaseGraphStorage):

        for _, data in self._graph.nodes(data=True):
            if "type" in data:
-                types.add(data["type"].lower()) 
-                types_with_case.add(data["type"])  
+                types.add(data["entity_type"].lower()) 
+                types_with_case.add(data["entity_type"])  
        return list(types), list(types_with_case)