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

Merge branch 'notebook_tutorial_improvements' into notebook_tutorial

Browse source 
# Conflicts:
#	notebooks/tutorial.ipynb
This commit is contained in:
lxobr 2025-09-07 15:20:27 +02:00
commit 2233a646ba
2 changed files with 177 additions and 468 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]

641
notebooks/tutorial.ipynb vendored
View file

File diff suppressed because one or more lines are too long