feat: checks neo4j test for bruteforce retriever

This commit is contained in:
hajdul88 2024-11-26 16:33:33 +01:00
parent db07179856
commit c9f66145f5

View file

@ -4,6 +4,7 @@ import logging
import pathlib
import cognee
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)
@ -61,6 +62,9 @@ async def main():
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__":
import asyncio
asyncio.run(main())