Merge branch 'notebook_tutorial_improvements' into notebook_tutorial

# 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

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]

File diff suppressed because one or more lines are too long