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

test fix

Browse source
This commit is contained in:
prestonrasmussen 2025-05-08 00:27:05 -04:00
parent 501fec8512
commit f7b2c3845b
1 changed files with 1 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_graphiti_int.py
View file

@ -26,7 +26,6 @@ from graphiti_core.edges import EntityEdge, EpisodicEdge
from graphiti_core.graphiti import Graphiti from graphiti_core.graphiti import Graphiti
from graphiti_core.helpers import semaphore_gather from graphiti_core.helpers import semaphore_gather
from graphiti_core.nodes import EntityNode, EpisodicNode from graphiti_core.nodes import EntityNode, EpisodicNode
from graphiti_core.search.search_config_recipes import COMBINED_HYBRID_SEARCH_MMR
from graphiti_core.search.search_helpers import search_results_to_context_string from graphiti_core.search.search_helpers import search_results_to_context_string
pytestmark = pytest.mark.integration pytestmark = pytest.mark.integration
@ -67,9 +66,7 @@ async def test_graphiti_init():
graphiti = Graphiti(NEO4J_URI, NEO4j_USER, NEO4j_PASSWORD) graphiti = Graphiti(NEO4J_URI, NEO4j_USER, NEO4j_PASSWORD)
results = await graphiti.search_( results = await graphiti.search_(
query='Who is the Tania?', query='Who is the user?',
group_ids=['98602bec-2c05-4f34-876e-741b39fe6ae8'],
config=COMBINED_HYBRID_SEARCH_MMR,
) )
pretty_results = search_results_to_context_string(results) pretty_results = search_results_to_context_string(results)