Update graph retrival api(abandon pydantic model)

2025-04-02 18:32:03 +08:00
parent d62a77500b
commit d7d04a0d94
3 changed files with 12 additions and 64 deletions
--- a/lightrag/base.py
+++ b/lightrag/base.py
@@ -341,7 +341,7 @@ class BaseGraphStorage(StorageNameSpace, ABC):

    @abstractmethod
    async def get_knowledge_graph(
-        self, node_label: str, max_depth: int = 3
+        self, node_label: str, max_depth: int = 3, max_nodes: int = 1000
    ) -> KnowledgeGraph:
        """Retrieve a subgraph of the knowledge graph starting from a given node."""