yangdx
commited on
Commit
·
601e7a7
1
Parent(s):
a55a783
Reduce knowledge graph max_depth from 100 to 3
Browse files
lightrag/api/routers/graph_routes.py
CHANGED
|
@@ -22,6 +22,6 @@ def create_graph_routes(rag, api_key: Optional[str] = None):
|
|
| 22 |
@router.get("/graphs", dependencies=[Depends(optional_api_key)])
|
| 23 |
async def get_knowledge_graph(label: str):
|
| 24 |
"""Get knowledge graph for a specific label"""
|
| 25 |
-
return await rag.get_knowledge_graph(node_label=label, max_depth=
|
| 26 |
|
| 27 |
return router
|
|
|
|
| 22 |
@router.get("/graphs", dependencies=[Depends(optional_api_key)])
|
| 23 |
async def get_knowledge_graph(label: str):
|
| 24 |
"""Get knowledge graph for a specific label"""
|
| 25 |
+
return await rag.get_knowledge_graph(node_label=label, max_depth=3)
|
| 26 |
|
| 27 |
return router
|