This commit is contained in:
ArnoChen
2025-02-09 22:22:44 +08:00
parent 7f407ce949
commit c1d7fbe02b
2 changed files with 2 additions and 2 deletions

View File

@@ -1413,7 +1413,7 @@ def create_app(args):
# query all graph # query all graph
@app.get("/graphs") @app.get("/graphs")
async def get_graphs(label: str): async def get_graphs(label: str):
return await rag.get_graps(nodel_label=label, max_depth=100) return await rag.get_graphs(nodel_label=label, max_depth=100)
# Add Ollama API routes # Add Ollama API routes
ollama_api = OllamaAPI(rag) ollama_api = OllamaAPI(rag)

View File

@@ -320,7 +320,7 @@ class LightRAG:
text = await self.chunk_entity_relation_graph.get_all_labels() text = await self.chunk_entity_relation_graph.get_all_labels()
return text return text
async def get_graps(self, nodel_label: str, max_depth: int): async def get_graphs(self, nodel_label: str, max_depth: int):
return await self.chunk_entity_relation_graph.get_knowledge_graph( return await self.chunk_entity_relation_graph.get_knowledge_graph(
node_label=nodel_label, max_depth=max_depth node_label=nodel_label, max_depth=max_depth
) )