Reduce knowledge graph max_depth from 100 to 3
This commit is contained in:
parent
6cf555ebd5
commit
facf7c11fe
1 changed files with 1 additions and 1 deletions
|
|
@ -22,6 +22,6 @@ def create_graph_routes(rag, api_key: Optional[str] = None):
|
||||||
@router.get("/graphs", dependencies=[Depends(optional_api_key)])
|
@router.get("/graphs", dependencies=[Depends(optional_api_key)])
|
||||||
async def get_knowledge_graph(label: str):
|
async def get_knowledge_graph(label: str):
|
||||||
"""Get knowledge graph for a specific label"""
|
"""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
|
return router
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue