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

ruff fix

Browse source
This commit is contained in:
hajdul88 2025-09-01 15:31:30 +02:00
parent 9df440c020
commit d336511c57
1 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
cognee/tests/unit/modules/retrieval/temporal_retriever_test.py
View file

@ -167,11 +167,15 @@ class _FakeRetriever(TemporalRetriever):
return ["e1", "e2"]
async def _fake_graph_collect_events(self, ids):
return [{"events": [
{"id": "e1", "description": "E1"},
{"id": "e2", "description": "E2"},
{"id": "e3", "description": "E3"},
]}]
return [
{
"events": [
{"id": "e1", "description": "E1"},
{"id": "e2", "description": "E2"},
{"id": "e3", "description": "E3"},
]
}
]
async def _fake_vector_embed(self, texts):
assert isinstance(texts, list) and texts
@ -190,9 +194,7 @@ class _FakeRetriever(TemporalRetriever):
triplets = await self.get_triplets(query)
return await self.resolve_edges_to_text(triplets)
ids = await self._fake_graph_collect_ids(
time_from=time_from, time_to=time_to
)
ids = await self._fake_graph_collect_ids(time_from=time_from, time_to=time_to)
relevant_events = await self._fake_graph_collect_events(ids)
_ = await self._fake_vector_embed([query])