feat: checks neo4j test for bruteforce retriever
This commit is contained in:
parent
db07179856
commit
c9f66145f5
1 changed files with 4 additions and 0 deletions
|
|
@ -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())
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue