fix linting
This commit is contained in:
@@ -739,57 +739,59 @@ class OracleGraphStorage(BaseGraphStorage):
|
||||
|
||||
async def remove_nodes(self, nodes: list[str]) -> None:
|
||||
"""Delete multiple nodes from the graph
|
||||
|
||||
|
||||
Args:
|
||||
nodes: List of node IDs to be deleted
|
||||
"""
|
||||
if not nodes:
|
||||
return
|
||||
|
||||
|
||||
try:
|
||||
for node in nodes:
|
||||
# For each node, first delete all its relationships
|
||||
delete_relations_sql = SQL_TEMPLATES["delete_entity_relations"]
|
||||
params_relations = {"workspace": self.db.workspace, "entity_name": node}
|
||||
await self.db.execute(delete_relations_sql, params_relations)
|
||||
|
||||
|
||||
# Then delete the node itself
|
||||
delete_node_sql = SQL_TEMPLATES["delete_entity"]
|
||||
params_node = {"workspace": self.db.workspace, "entity_name": node}
|
||||
await self.db.execute(delete_node_sql, params_node)
|
||||
|
||||
logger.info(f"Successfully deleted {len(nodes)} nodes and their relationships")
|
||||
|
||||
logger.info(
|
||||
f"Successfully deleted {len(nodes)} nodes and their relationships"
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"Error during batch node deletion: {e}")
|
||||
raise
|
||||
|
||||
async def remove_edges(self, edges: list[tuple[str, str]]) -> None:
|
||||
"""Delete multiple edges from the graph
|
||||
|
||||
|
||||
Args:
|
||||
edges: List of edges to be deleted, each edge is a (source, target) tuple
|
||||
"""
|
||||
if not edges:
|
||||
return
|
||||
|
||||
|
||||
try:
|
||||
for source, target in edges:
|
||||
# Check if the edge exists before attempting to delete
|
||||
if await self.has_edge(source, target):
|
||||
# Delete the edge using a SQL query that matches both source and target
|
||||
delete_edge_sql = """
|
||||
DELETE FROM LIGHTRAG_GRAPH_EDGES
|
||||
WHERE workspace = :workspace
|
||||
AND source_name = :source_name
|
||||
DELETE FROM LIGHTRAG_GRAPH_EDGES
|
||||
WHERE workspace = :workspace
|
||||
AND source_name = :source_name
|
||||
AND target_name = :target_name
|
||||
"""
|
||||
params = {
|
||||
"workspace": self.db.workspace,
|
||||
"source_name": source,
|
||||
"target_name": target
|
||||
"target_name": target,
|
||||
}
|
||||
await self.db.execute(delete_edge_sql, params)
|
||||
|
||||
|
||||
logger.info(f"Successfully deleted {len(edges)} edges from the graph")
|
||||
except Exception as e:
|
||||
logger.error(f"Error during batch edge deletion: {e}")
|
||||
|
Reference in New Issue
Block a user