fix: do not quote the divider in highlights (#138)

MartinBernstorff · Apr 9, 2024 · 2383be3 · 2383be3
1 parent a9b8b6c
commit 2383be3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/memorymarker/question_generator/main.py b/memorymarker/question_generator/main.py
@@ -81,7 +81,7 @@ def chunk_highlights(
 
     for i in range(0, len(group[1]), 5):
         subset: Sequence["Highlights"] = group[1][i : i + chunk_size]
-        combined_text = "\n---".join(
+        combined_text = "\n---\n".join(
             f"> {_.prefix}<HIGHLIGHT>{_.highlighted_text}</HIGHLIGHT>{_.suffix}"
             for _ in subset
         )

diff --git a/memorymarker/question_generator/reasoned_highlight.py b/memorymarker/question_generator/reasoned_highlight.py
@@ -13,11 +13,6 @@ class SourceDocument:
     uri: str
 
 
-def to_markdown_quote(text: str) -> str:
-    lines = [f"> {line}" for line in text.splitlines()]
-    return "\n".join(lines)
-
-
 class Highlights(BaseModel):
     source_document: SourceDocument