gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed linting

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-01-27 02:10:24 +01:00
parent a68aebb124
commit f307ed43f5
3 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lightrag/api/lightrag_server.py
View File

@@ -1791,7 +1791,7 @@ def create_app(args):
raise HTTPException(status_code=500, detail=str(e)) raise HTTPException(status_code=500, detail=str(e))
@app.get("/documents", dependencies=[Depends(optional_api_key)]) @app.get("/documents", dependencies=[Depends(optional_api_key)])
async def get_status(): async def documents():
"""Get current system status""" """Get current system status"""
return doc_manager.indexed_files return doc_manager.indexed_files

5
lightrag/kg/neo4j_impl.py
View File

@@ -340,7 +340,6 @@ class Neo4JStorage(BaseGraphStorage):
async def _node2vec_embed(self): async def _node2vec_embed(self):
print("Implemented but never called.") print("Implemented but never called.")
async def get_knowledge_graph( async def get_knowledge_graph(
self, node_label: str, max_depth: int = 5 self, node_label: str, max_depth: int = 5
) -> Dict[str, List[Dict]]: ) -> Dict[str, List[Dict]]:
@@ -448,7 +447,9 @@ class Neo4JStorage(BaseGraphStorage):
# Add node data (with complete labels) # Add node data (with complete labels)
node_data = {k: v for k, v in node.items()} node_data = {k: v for k, v in node.items()}
node_data["labels"] = [current_label] # Assume get_node method returns label information node_data["labels"] = [
current_label
] # Assume get_node method returns label information
result["nodes"].append(node_data) result["nodes"].append(node_data)
# Get all outgoing and incoming edges # Get all outgoing and incoming edges