Revert "removed get_knowledge_graph"

This commit is contained in:
Yannick Stephan
2025-02-20 14:29:36 +01:00
committed by GitHub
parent 0d4c580859
commit 439685e69c
11 changed files with 324 additions and 0 deletions

View File

@@ -5,6 +5,7 @@ from typing import Any, final
import numpy as np
from lightrag.types import KnowledgeGraph
from lightrag.utils import (
logger,
)
@@ -166,3 +167,8 @@ class NetworkXStorage(BaseGraphStorage):
for source, target in edges:
if self._graph.has_edge(source, target):
self._graph.remove_edge(source, target)
async def get_knowledge_graph(
self, node_label: str, max_depth: int = 5
) -> KnowledgeGraph:
raise NotImplementedError