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

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-04-03 15:40:55 +08:00
parent 3e3338a144
commit 0826b0b80c
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lightrag/kg/postgres_impl.py
View File

@@ -1238,7 +1238,7 @@ class PGGraphStorage(BaseGraphStorage):
self, source_node_id: str, target_node_id: str self, source_node_id: str, target_node_id: str
) -> dict[str, str] | None: ) -> dict[str, str] | None:
"""Get edge properties between two nodes""" """Get edge properties between two nodes"""
src_label = source_node_id.strip('"') src_label = source_node_id.strip('"')
tgt_label = target_node_id.strip('"') tgt_label = target_node_id.strip('"')
@@ -1515,7 +1515,7 @@ class PGGraphStorage(BaseGraphStorage):
if node_id not in nodes_dict: if node_id not in nodes_dict:
nodes_dict[node_id] = KnowledgeGraphNode( nodes_dict[node_id] = KnowledgeGraphNode(
id=node_id, id=node_id,
labels=[result["n"]["properties"]["entity_id"]], labels=[result["n"]["properties"]["entity_id"]],
properties=result["n"]["properties"], properties=result["n"]["properties"],
) )
# Handle node list cases # Handle node list cases
@@ -1526,10 +1526,10 @@ class PGGraphStorage(BaseGraphStorage):
if node_id not in nodes_dict and "properties" in node: if node_id not in nodes_dict and "properties" in node:
nodes_dict[node_id] = KnowledgeGraphNode( nodes_dict[node_id] = KnowledgeGraphNode(
id=node_id, id=node_id,
labels=[node["properties"]["entity_id"]], labels=[node["properties"]["entity_id"]],
properties=node["properties"], properties=node["properties"],
) )
# Handle single edge cases # Handle single edge cases
if result.get("r") and isinstance(result["r"], dict): if result.get("r") and isinstance(result["r"], dict):
edge_id = str(result["r"]["id"]) edge_id = str(result["r"]["id"])