adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #742 from SuanmoSuanyangTechnology/fix/cypher-query

Browse Source 
[fix] Fix the alias query statement
This commit is contained in:
Mark
2026-03-30 19:59:48 +08:00
committed by GitHub
parent a00684c67d 3419bb137a
commit 5c87a2beeb
1 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
api/app/repositories/neo4j/cypher_queries.py
View File

@@ -340,17 +340,22 @@ SEARCH_ENTITIES_BY_NAME_OR_ALIAS = """
CALL db.index.fulltext.queryNodes("entitiesFulltext", $q) YIELD node AS e, score CALL db.index.fulltext.queryNodes("entitiesFulltext", $q) YIELD node AS e, score
WHERE ($end_user_id IS NULL OR e.end_user_id = $end_user_id) WHERE ($end_user_id IS NULL OR e.end_user_id = $end_user_id)
WITH e, score WITH e, score
UNION WITH collect({entity: e, score: score}) AS fulltextResults
MATCH (e:ExtractedEntity)
WHERE ($end_user_id IS NULL OR e.end_user_id = $end_user_id) OPTIONAL MATCH (ae:ExtractedEntity)
AND e.aliases IS NOT NULL WHERE ($end_user_id IS NULL OR ae.end_user_id = $end_user_id)
AND ANY(alias IN e.aliases WHERE toLower(alias) CONTAINS toLower($q)) AND ae.aliases IS NOT NULL
WITH e, AND ANY(alias IN ae.aliases WHERE toLower(alias) CONTAINS toLower($q))
WITH fulltextResults, collect(ae) AS aliasEntities
UNWIND (fulltextResults + [x IN aliasEntities | {entity: x, score:
CASE CASE
WHEN ANY(alias IN e.aliases WHERE toLower(alias) = toLower($q)) THEN 1.0 WHEN ANY(alias IN x.aliases WHERE toLower(alias) = toLower($q)) THEN 1.0
WHEN ANY(alias IN e.aliases WHERE toLower(alias) STARTS WITH toLower($q)) THEN 0.9 WHEN ANY(alias IN x.aliases WHERE toLower(alias) STARTS WITH toLower($q)) THEN 0.9
ELSE 0.8 ELSE 0.8
END AS score END
}]) AS row
WITH row.entity AS e, row.score AS score
WITH DISTINCT e, MAX(score) AS score WITH DISTINCT e, MAX(score) AS score
OPTIONAL MATCH (s:Statement)-[:REFERENCES_ENTITY]->(e) OPTIONAL MATCH (s:Statement)-[:REFERENCES_ENTITY]->(e)
OPTIONAL MATCH (c:Chunk)-[:CONTAINS]->(s) OPTIONAL MATCH (c:Chunk)-[:CONTAINS]->(s)