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

chore: fix search db tests

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

8
cognee/tests/test_search_db.py
View file

@ -44,16 +44,16 @@ async def main():
await cognee.cognify([dataset_name])
context_gk = await GraphCompletionRetriever().get_context(
context_gk, _ = await GraphCompletionRetriever().get_context(
query="Next to which country is Germany located?"
)
context_gk_cot = await GraphCompletionCotRetriever().get_context(
context_gk_cot, _ = await GraphCompletionCotRetriever().get_context(
query="Next to which country is Germany located?"
)
context_gk_ext = await GraphCompletionContextExtensionRetriever().get_context(
context_gk_ext, _ = await GraphCompletionContextExtensionRetriever().get_context(
query="Next to which country is Germany located?"
)
context_gk_sum = await GraphSummaryCompletionRetriever().get_context(
context_gk_sum, _ = await GraphSummaryCompletionRetriever().get_context(
query="Next to which country is Germany located?"
)