jacob-braun-mn commited on
Commit
0b1f47b
1 Parent(s): 8cfcdac

Fix get_relevant_docs helper func

Browse files
Files changed (1) hide show
  1. helpers.py +3 -0
helpers.py CHANGED
@@ -50,6 +50,9 @@ def transform_documents(splits: List[Document]) -> List[Dict[str, Union[int, str
50
 
51
  result.append({"highlight_idx": highlight_idx, "page_content": doc.page_content.strip()})
52
 
 
 
 
53
  if combined_non_highlight_content:
54
  result.append(
55
  {"highlight_idx": None, "page_content": combined_non_highlight_content.strip()}
 
50
 
51
  result.append({"highlight_idx": highlight_idx, "page_content": doc.page_content.strip()})
52
 
53
+ else:
54
+ combined_non_highlight_content += doc.page_content.strip() + " "
55
+
56
  if combined_non_highlight_content:
57
  result.append(
58
  {"highlight_idx": None, "page_content": combined_non_highlight_content.strip()}