gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add empty graph creation logging in NetworkXStorage

Browse Source
This commit is contained in:
yangdx
2025-02-26 17:42:30 +08:00
parent 41f5d208a9
commit 145bacc773
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
lightrag/kg/networkx_impl.py
View File

@@ -80,16 +80,22 @@ class NetworkXStorage(BaseGraphStorage):
if self._graph.value is None: if self._graph.value is None:
preloaded_graph = NetworkXStorage.load_nx_graph(self._graphml_xml_file) preloaded_graph = NetworkXStorage.load_nx_graph(self._graphml_xml_file)
self._graph.value = preloaded_graph or nx.Graph() self._graph.value = preloaded_graph or nx.Graph()
logger.info( if preloaded_graph:
f"Loaded graph from {self._graphml_xml_file} with {preloaded_graph.number_of_nodes()} nodes, {preloaded_graph.number_of_edges()} edges" logger.info(
) f"Loaded graph from {self._graphml_xml_file} with {preloaded_graph.number_of_nodes()} nodes, {preloaded_graph.number_of_edges()} edges"
)
else:
logger.info("Created new empty graph")
else: else:
if self._graph is None: if self._graph is None:
preloaded_graph = NetworkXStorage.load_nx_graph(self._graphml_xml_file) preloaded_graph = NetworkXStorage.load_nx_graph(self._graphml_xml_file)
self._graph = preloaded_graph or nx.Graph() self._graph = preloaded_graph or nx.Graph()
logger.info( if preloaded_graph:
f"Loaded graph from {self._graphml_xml_file} with {preloaded_graph.number_of_nodes()} nodes, {preloaded_graph.number_of_edges()} edges" logger.info(
) f"Loaded graph from {self._graphml_xml_file} with {preloaded_graph.number_of_nodes()} nodes, {preloaded_graph.number_of_edges()} edges"
)
else:
logger.info("Created new empty graph")
self._node_embed_algorithms = { self._node_embed_algorithms = {
"node2vec": self._node2vec_embed, "node2vec": self._node2vec_embed,