gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

entity typo (#274)

Browse source
This commit is contained in:
Preston Rasmussen 2025-02-24 12:44:17 -05:00 • committed by GitHub
parent e1022fa21d
commit 9efa6762d7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
graphiti_core/search/search_utils.py
View file

@ -158,7 +158,7 @@ async def edge_fulltext_search(
""" """
CALL db.index.fulltext.queryRelationships("edge_name_and_fact", $query, {limit: $limit}) CALL db.index.fulltext.queryRelationships("edge_name_and_fact", $query, {limit: $limit})
YIELD relationship AS rel, score YIELD relationship AS rel, score
MATCH (:ENTITY)-[r:RELATES_TO]->(:ENTITY) MATCH (:Entity)-[r:RELATES_TO]->(:Entity)
WHERE r.group_id IN $group_ids""" WHERE r.group_id IN $group_ids"""
+ filter_query + filter_query
+ """\nWITH r, score, startNode(r) AS n, endNode(r) AS m + """\nWITH r, score, startNode(r) AS n, endNode(r) AS m
@ -229,8 +229,8 @@ async def edge_similarity_search(
query: LiteralString = ( query: LiteralString = (
""" """
MATCH (n:Entity)-[r:RELATES_TO]->(m:Entity) MATCH (n:Entity)-[r:RELATES_TO]->(m:Entity)
""" """
+ group_filter_query + group_filter_query
+ filter_query + filter_query
+ """\nWITH DISTINCT r, vector.similarity.cosine(r.fact_embedding, $search_vector) AS score + """\nWITH DISTINCT r, vector.similarity.cosine(r.fact_embedding, $search_vector) AS score

10
tests/utils/search/search_utils_test.py
View file

@ -14,7 +14,7 @@ async def test_hybrid_node_search_deduplication():
# Mock the node_fulltext_search and entity_similarity_search functions # Mock the node_fulltext_search and entity_similarity_search functions
with patch( with patch(
'graphiti_core.search.search_utils.node_fulltext_search' 'graphiti_core.search.search_utils.node_fulltext_search'
) as mock_fulltext_search, patch( ) as mock_fulltext_search, patch(
'graphiti_core.search.search_utils.node_similarity_search' 'graphiti_core.search.search_utils.node_similarity_search'
) as mock_similarity_search: ) as mock_similarity_search:
@ -48,7 +48,7 @@ async def test_hybrid_node_search_empty_results():
mock_driver = AsyncMock() mock_driver = AsyncMock()
with patch( with patch(
'graphiti_core.search.search_utils.node_fulltext_search' 'graphiti_core.search.search_utils.node_fulltext_search'
) as mock_fulltext_search, patch( ) as mock_fulltext_search, patch(
'graphiti_core.search.search_utils.node_similarity_search' 'graphiti_core.search.search_utils.node_similarity_search'
) as mock_similarity_search: ) as mock_similarity_search:
@ -67,7 +67,7 @@ async def test_hybrid_node_search_only_fulltext():
mock_driver = AsyncMock() mock_driver = AsyncMock()
with patch( with patch(
'graphiti_core.search.search_utils.node_fulltext_search' 'graphiti_core.search.search_utils.node_fulltext_search'
) as mock_fulltext_search, patch( ) as mock_fulltext_search, patch(
'graphiti_core.search.search_utils.node_similarity_search' 'graphiti_core.search.search_utils.node_similarity_search'
) as mock_similarity_search: ) as mock_similarity_search:
@ -91,7 +91,7 @@ async def test_hybrid_node_search_with_limit():
mock_driver = AsyncMock() mock_driver = AsyncMock()
with patch( with patch(
'graphiti_core.search.search_utils.node_fulltext_search' 'graphiti_core.search.search_utils.node_fulltext_search'
) as mock_fulltext_search, patch( ) as mock_fulltext_search, patch(
'graphiti_core.search.search_utils.node_similarity_search' 'graphiti_core.search.search_utils.node_similarity_search'
) as mock_similarity_search: ) as mock_similarity_search:
@ -134,7 +134,7 @@ async def test_hybrid_node_search_with_limit_and_duplicates():
mock_driver = AsyncMock() mock_driver = AsyncMock()
with patch( with patch(
'graphiti_core.search.search_utils.node_fulltext_search' 'graphiti_core.search.search_utils.node_fulltext_search'
) as mock_fulltext_search, patch( ) as mock_fulltext_search, patch(
'graphiti_core.search.search_utils.node_similarity_search' 'graphiti_core.search.search_utils.node_similarity_search'
) as mock_similarity_search: ) as mock_similarity_search: