From 1335f6fdd07420391958d460910d73e28049ea06 Mon Sep 17 00:00:00 2001 From: lxobr <122801072+lxobr@users.noreply.github.com> Date: Sun, 7 Sep 2025 17:12:39 +0200 Subject: [PATCH] fix: fix return values --- 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 1e56dda3a..0ab8e2ecf 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) + return await self.resolve_edges_to_text(triplets), 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) + return await self.resolve_edges_to_text(triplets), triplets vector_engine = get_vector_engine() query_vector = (await vector_engine.embedding_engine.embed_text([query]))[0]