Spaces:
Sleeping
Sleeping
jacob-braun-mn
commited on
Commit
•
0b1f47b
1
Parent(s):
8cfcdac
Fix get_relevant_docs helper func
Browse files- 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()}
|