add graph depth and layout iteration settings

2025-02-25 18:28:31 +08:00
parent 7f96ab654a
commit 40a1a94a31
11 changed files with 233 additions and 86 deletions
--- a/lightrag/api/routers/graph_routes.py
+++ b/lightrag/api/routers/graph_routes.py
@@ -20,8 +20,8 @@ def create_graph_routes(rag, api_key: Optional[str] = None):
        return await rag.get_graph_labels()

    @router.get("/graphs", dependencies=[Depends(optional_api_key)])
-    async def get_knowledge_graph(label: str):
+    async def get_knowledge_graph(label: str, max_depth: int = 3):
        """Get knowledge graph for a specific label"""
-        return await rag.get_knowledge_graph(node_label=label, max_depth=3)
+        return await rag.get_knowledge_graph(node_label=label, max_depth=max_depth)

    return router