From dfa3203af5a7b8656449054d4e2511c43b466185 Mon Sep 17 00:00:00 2001 From: lxobr <122801072+lxobr@users.noreply.github.com> Date: Sun, 7 Sep 2025 17:10:52 +0200 Subject: [PATCH] revert temporal retriever fix --- cognee/modules/retrieval/temporal_retriever.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cognee/modules/retrieval/temporal_retriever.py b/cognee/modules/retrieval/temporal_retriever.py index 0ab8e2ecf..1e56dda3a 100644 --- a/cognee/modules/retrieval/temporal_retriever.py +++ b/cognee/modules/retrieval/temporal_retriever.py @@ -114,7 +114,7 @@ class TemporalRetriever(GraphCompletionRetriever): "No timestamps identified based on the query, performing retrieval using triplet search on events and entities." ) triplets = await self.get_triplets(query) - return await self.resolve_edges_to_text(triplets), triplets + return await self.resolve_edges_to_text(triplets) if ids: relevant_events = await graph_engine.collect_events(ids=ids) @@ -123,7 +123,7 @@ class TemporalRetriever(GraphCompletionRetriever): "No events identified based on timestamp filtering, performing retrieval using triplet search on events and entities." ) triplets = await self.get_triplets(query) - return await self.resolve_edges_to_text(triplets), triplets + return await self.resolve_edges_to_text(triplets) vector_engine = get_vector_engine() query_vector = (await vector_engine.embedding_engine.embed_text([query]))[0]