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)])
|
||||
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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue