Fix linting
This commit is contained in:
parent
a809bc7945
commit
399b2f14f6
2 changed files with 14 additions and 9 deletions
|
|
@ -833,7 +833,7 @@ class Neo4JStorage(BaseGraphStorage):
|
|||
|
||||
logger.info(
|
||||
f"Subgraph query successful | Node count: {len(result.nodes)} | Edge count: {len(result.edges)}"
|
||||
)
|
||||
)
|
||||
|
||||
except neo4jExceptions.ClientError as e:
|
||||
logger.warning(f"APOC plugin error: {str(e)}")
|
||||
|
|
|
|||
|
|
@ -445,6 +445,7 @@ async def extract_entities(
|
|||
|
||||
# Get lock manager from shared storage
|
||||
from .kg.shared_storage import get_graph_db_lock
|
||||
|
||||
graph_db_lock = get_graph_db_lock(enable_logging=False)
|
||||
|
||||
async def _user_llm_func_with_cache(
|
||||
|
|
@ -620,7 +621,11 @@ async def extract_entities(
|
|||
# Ensure edge direction consistency
|
||||
sorted_edge_key = tuple(sorted(edge_key))
|
||||
edge_data = await _merge_edges_then_upsert(
|
||||
sorted_edge_key[0], sorted_edge_key[1], edges, knowledge_graph_inst, global_config
|
||||
sorted_edge_key[0],
|
||||
sorted_edge_key[1],
|
||||
edges,
|
||||
knowledge_graph_inst,
|
||||
global_config,
|
||||
)
|
||||
chunk_relationships_data.append(edge_data)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue