vrr/open-notebook
2
0
Fork 0
mirror of https://github.com/lfnovo/open-notebook.git synced 2026-04-30 12:30:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix insight context to improve citations

Browse source
This commit is contained in:
LUIS NOVO 2024-11-13 17:02:18 -03:00
parent 066c7a06e2
commit 06c6842f11
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
open_notebook/domain/notebook.py
View file

@ -141,15 +141,16 @@ class Source(ObjectModel):
def get_context(
self, context_size: Literal["short", "long"] = "short"
) -> Dict[str, Any]:
insights = [insight.model_dump() for insight in self.insights]
if context_size == "long":
return dict(
id=self.id,
title=self.title,
insights=[insight.model_dump() for insight in self.insights],
insights=insights,
full_text=self.full_text,
)
else:
return dict(id=self.id, title=self.title, insights=self.insights)
return dict(id=self.id, title=self.title, insights=insights)
@property
def embedded_chunks(self) -> int: