gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Set default max_depth to 3 for knowledge graph retrieval

Browse source
This commit is contained in:
yangdx 2025-03-07 07:34:29 +08:00
parent 6d5c04d561
commit 5942d148f0
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lightrag/base.py
View file

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

2
lightrag/lightrag.py
View file

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