mirror of
https://github.com/HKUDS/LightRAG.git
synced 2025-07-28 19:29:44 +00:00
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
"""
|
|
This module contains all graph-related routes for the LightRAG API.
|
|
"""
|
|
|
|
from typing import Optional
|
|
|
|
from fastapi import APIRouter, Depends
|
|
|
|
from ..utils_api import get_api_key_dependency
|
|
|
|
router = APIRouter(tags=["graph"])
|
|
|
|
|
|
def create_graph_routes(rag, api_key: Optional[str] = None):
|
|
optional_api_key = get_api_key_dependency(api_key)
|
|
|
|
@router.get("/graph/label/list", dependencies=[Depends(optional_api_key)])
|
|
async def get_graph_labels():
|
|
"""
|
|
Get all graph labels
|
|
|
|
Returns:
|
|
List[str]: List of graph labels
|
|
"""
|
|
return await rag.get_graph_labels()
|
|
|
|
@router.get("/graphs", dependencies=[Depends(optional_api_key)])
|
|
async def get_knowledge_graph(label: str, max_depth: int = 3):
|
|
"""
|
|
Get knowledge graph for a specific label.
|
|
Maximum number of nodes is limited to env MAX_GRAPH_NODES(default: 1000)
|
|
|
|
Args:
|
|
label (str): Label to get knowledge graph for
|
|
max_depth (int, optional): Maximum depth of graph. Defaults to 3.
|
|
|
|
Returns:
|
|
Dict[str, List[str]]: Knowledge graph for label
|
|
"""
|
|
return await rag.get_knowledge_graph(node_label=label, max_depth=max_depth)
|
|
|
|
return router
|