Set default max_depth to 3 for knowledge graph retrieval

lightrag/base.py

@@ -204,7 +204,7 @@ class BaseGraphStorage(StorageNameSpace, ABC):
 
     @abstractmethod
     async def get_knowledge_graph(
-        self, node_label: str, max_depth: int = 5
+        self, node_label: str, max_depth: int = 3
     ) -> KnowledgeGraph:
         """Retrieve a subgraph of the knowledge graph starting from a given node."""
 

lightrag/lightrag.py

@@ -506,7 +506,7 @@ class LightRAG:
     async def get_knowledge_graph(
         self,
         node_label: str,
-        max_depth: int,
+        max_depth: int = 3,
         min_degree: int = 0,
         inclusive: bool = False,
     ) -> KnowledgeGraph: