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-02 16:36:28 +08:00
parent 3ed7f11b4c
commit c2f007199f
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
lightrag/kg/neo4j_impl.py
View File

@@ -164,16 +164,18 @@ class Neo4JStorage(BaseGraphStorage):
check_result = await session.run(check_query) check_result = await session.run(check_query)
record = await check_result.single() record = await check_result.single()
await check_result.consume() await check_result.consume()
index_exists = record and record.get("exists", False) index_exists = record and record.get("exists", False)
if not index_exists: if not index_exists:
# Create index only if it doesn't exist # Create index only if it doesn't exist
result = await session.run( result = await session.run(
"CREATE INDEX FOR (n:base) ON (n.entity_id)" "CREATE INDEX FOR (n:base) ON (n.entity_id)"
) )
await result.consume() await result.consume()
logger.info(f"Created index for base nodes on entity_id in {database}") logger.info(
f"Created index for base nodes on entity_id in {database}"
)
except Exception: except Exception:
# Fallback if db.indexes() is not supported in this Neo4j version # Fallback if db.indexes() is not supported in this Neo4j version
result = await session.run( result = await session.run(
@@ -709,7 +711,7 @@ class Neo4JStorage(BaseGraphStorage):
WITH node, COALESCE(count(r), 0) AS degree, start, nodes, relationships WITH node, COALESCE(count(r), 0) AS degree, start, nodes, relationships
ORDER BY ORDER BY
CASE CASE
WHEN node = start THEN 0 WHEN node = start THEN 0
ELSE length(shortestPath((start)--(node))) ELSE length(shortestPath((start)--(node)))
END ASC, END ASC,
degree DESC degree DESC
@@ -778,13 +780,13 @@ class Neo4JStorage(BaseGraphStorage):
logger.warning( logger.warning(
"Neo4j: falling back to basic Cypher recursive search..." "Neo4j: falling back to basic Cypher recursive search..."
) )
return await self._robust_fallback( return await self._robust_fallback(node_label, max_depth, max_nodes)
node_label, max_depth, max_nodes
)
return result return result
async def _robust_fallback(self, node_label: str, max_depth: int, max_nodes: int) -> KnowledgeGraph: async def _robust_fallback(
self, node_label: str, max_depth: int, max_nodes: int
) -> KnowledgeGraph:
""" """
Fallback implementation when APOC plugin is not available or incompatible. Fallback implementation when APOC plugin is not available or incompatible.
This method implements the same functionality as get_knowledge_graph but uses This method implements the same functionality as get_knowledge_graph but uses