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

fix: fix return value in TemporalRetriever

Browse source
This commit is contained in:
lxobr 2025-09-07 15:05:50 +02:00
parent 1cabd22ac8
commit 5224ec75c6
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." "No timestamps identified based on the query, performing retrieval using triplet search on events and entities."
) )
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), triplets
if ids: if ids:
relevant_events = await graph_engine.collect_events(ids=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." "No events identified based on timestamp filtering, performing retrieval using triplet search on events and entities."
) )
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), triplets
vector_engine = get_vector_engine() vector_engine = get_vector_engine()
query_vector = (await vector_engine.embedding_engine.embed_text([query]))[0] query_vector = (await vector_engine.embedding_engine.embed_text([query]))[0]