Reduce knowledge graph max_depth from 100 to 3

2025-02-24 03:34:44 +08:00
parent 6cf555ebd5
commit facf7c11fe
1 changed files with 1 additions and 1 deletions
--- a/lightrag/api/routers/graph_routes.py
+++ b/lightrag/api/routers/graph_routes.py
@@ -22,6 +22,6 @@ def create_graph_routes(rag, api_key: Optional[str] = None):
    @router.get("/graphs", dependencies=[Depends(optional_api_key)])
    async def get_knowledge_graph(label: str):
        """Get knowledge graph for a specific label"""
-        return await rag.get_knowledge_graph(node_label=label, max_depth=100)
+        return await rag.get_knowledge_graph(node_label=label, max_depth=3)

    return router