split lightrag_servery.py to smaller files

lightrag/api/routers/graph_routes.py

@@ -0,0 +1,26 @@
+"""
+This module contains all graph-related routes for the LightRAG API.
+"""
+
+from typing import Optional
+
+from fastapi import APIRouter, Depends, HTTPException
+
+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"""
+        return await rag.get_graph_labels()
+
+    @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(nodel_label=label, max_depth=100)
+
+    return router