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