use KnowledgeGraph typed dict for graph API response

2025-02-13 17:32:51 +08:00
parent e4562c761c
commit c674905a98
5 changed files with 54 additions and 30 deletions
--- a/lightrag/api/lightrag_server.py
+++ b/lightrag/api/lightrag_server.py
@@ -1424,8 +1424,8 @@ def create_app(args):

    # query all graph
    @app.get("/graphs")
-    async def get_graphs(label: str):
-        return await rag.get_graphs(nodel_label=label, max_depth=100)
+    async def get_knowledge_graph(label: str):
+        return await rag.get_knowledge_graph(nodel_label=label, max_depth=100)

    # Add Ollama API routes
    ollama_api = OllamaAPI(rag)