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

fix: temporal retriever return values (#1338)

Browse source 
<!-- .github/pull_request_template.md -->

## Description
<!-- Provide a clear description of the changes in this PR -->

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
This commit is contained in:
Vasilije 2025-09-07 08:44:23 -07:00 • committed by GitHub
commit e0b4e4ee5e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/modules/retrieval/temporal_retriever.py
View file

@ -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]