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

fix: fixes kuzu and neo4j tests

Browse source
This commit is contained in:
hajdul88 2025-08-18 14:14:12 +02:00
parent b6be61776a
commit 0529d4b87f
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/tests/test_kuzu.py
View file

@ -94,12 +94,12 @@ async def main():
await cognee.cognify([dataset_name]) await cognee.cognify([dataset_name])
context_nonempty = await GraphCompletionRetriever( context_nonempty, _ = await GraphCompletionRetriever(
node_type=NodeSet, node_type=NodeSet,
node_name=["first"], node_name=["first"],
).get_context("What is in the context?") ).get_context("What is in the context?")
context_empty = await GraphCompletionRetriever( context_empty, _ = await GraphCompletionRetriever(
node_type=NodeSet, node_type=NodeSet,
node_name=["nonexistent"], node_name=["nonexistent"],
).get_context("What is in the context?") ).get_context("What is in the context?")

4
cognee/tests/test_neo4j.py
View file

@ -98,12 +98,12 @@ async def main():
await cognee.cognify([dataset_name]) await cognee.cognify([dataset_name])
context_nonempty = await GraphCompletionRetriever( context_nonempty, _ = await GraphCompletionRetriever(
node_type=NodeSet, node_type=NodeSet,
node_name=["first"], node_name=["first"],
).get_context("What is in the context?") ).get_context("What is in the context?")
context_empty = await GraphCompletionRetriever( context_empty, _ = await GraphCompletionRetriever(
node_type=NodeSet, node_type=NodeSet,
node_name=["nonexistent"], node_name=["nonexistent"],
).get_context("What is in the context?") ).get_context("What is in the context?")