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

feat: checks neo4j test for bruteforce retriever

Browse source
This commit is contained in:
hajdul88 2024-11-26 16:33:33 +01:00
parent db07179856
commit c9f66145f5
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/tests/test_neo4j.py
View file

@ -4,6 +4,7 @@ import logging
import pathlib import pathlib
import cognee import cognee
from cognee.api.v1.search import SearchType from cognee.api.v1.search import SearchType
from cognee.modules.retrieval.brute_force_triplet_search import brute_force_triplet_search
logging.basicConfig(level=logging.DEBUG) logging.basicConfig(level=logging.DEBUG)
@ -61,6 +62,9 @@ async def main():
assert len(history) == 6, "Search history is not correct." assert len(history) == 6, "Search history is not correct."
results = await brute_force_triplet_search('Who has the most experience with graphic design?')
assert len(results)>0
if __name__ == "__main__": if __name__ == "__main__":
import asyncio import asyncio
asyncio.run(main()) asyncio.run(main())