sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-30 10:53:02 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove "Path" prefix from org ancestor heading in compiled entry

Browse source
This commit is contained in:
Debanjum Singh Solanky 2024-06-06 16:51:26 +05:30
parent dd2225b1aa
commit 18f7e6e7ed
2 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/khoj/processor/content/org_mode/org_to_entries.py
View file

@ -182,7 +182,7 @@ class OrgToEntries(TextToEntries):
# Children nodes do not need ancestors trail as root parent node will have it # Children nodes do not need ancestors trail as root parent node will have it
if not entry_heading: if not entry_heading:
ancestors_trail = " / ".join(parsed_entry.ancestors) or Path(entry_to_file_map[parsed_entry]) ancestors_trail = " / ".join(parsed_entry.ancestors) or Path(entry_to_file_map[parsed_entry])
heading = f"* Path: {ancestors_trail}\n{heading}" if heading else f"* Path: {ancestors_trail}." heading = f"* {ancestors_trail}\n{heading}" if heading else f"* {ancestors_trail}."
compiled = heading compiled = heading

16
tests/test_org_to_entries.py
View file

@ -50,14 +50,14 @@ def test_entry_split_when_exceeds_max_tokens():
data = { data = {
f"{tmp_path}": entry, f"{tmp_path}": entry,
} }
expected_heading = f"* Path: {tmp_path}\n** Heading" expected_heading = f"* {tmp_path}\n** Heading"
# Act # Act
# Extract Entries from specified Org files # Extract Entries from specified Org files
entries = OrgToEntries.extract_org_entries(org_files=data) entries = OrgToEntries.extract_org_entries(org_files=data)
# Split each entry from specified Org files by max tokens # Split each entry from specified Org files by max tokens
entries = TextToEntries.split_entries_by_max_tokens(entries, max_tokens=6) entries = TextToEntries.split_entries_by_max_tokens(entries, max_tokens=5)
# Assert # Assert
assert len(entries) == 2 assert len(entries) == 2
@ -139,7 +139,7 @@ longer body line 2.1
f"{tmp_path}": entry, f"{tmp_path}": entry,
} }
first_expected_entry = f""" first_expected_entry = f"""
* Path: {tmp_path} * {tmp_path}
** Heading 1. ** Heading 1.
body line 1 body line 1
@ -148,13 +148,13 @@ longer body line 2.1
""".lstrip() """.lstrip()
second_expected_entry = f""" second_expected_entry = f"""
* Path: {tmp_path} * {tmp_path}
** Heading 2. ** Heading 2.
body line 2 body line 2
""".lstrip() """.lstrip()
third_expected_entry = f""" third_expected_entry = f"""
* Path: {tmp_path} / Heading 2 * {tmp_path} / Heading 2
** Subheading 2.1. ** Subheading 2.1.
longer body line 2.1 longer body line 2.1
@ -192,7 +192,7 @@ body line 3.1
f"{tmp_path}": entry, f"{tmp_path}": entry,
} }
first_expected_entry = f""" first_expected_entry = f"""
* Path: {tmp_path} * {tmp_path}
** Heading 1. ** Heading 1.
body line 1 body line 1
@ -201,7 +201,7 @@ body line 3.1
""".lstrip() """.lstrip()
second_expected_entry = f""" second_expected_entry = f"""
* Path: {tmp_path} * {tmp_path}
** Heading 2. ** Heading 2.
body line 2 body line 2
@ -210,7 +210,7 @@ body line 3.1
""".lstrip() """.lstrip()
third_expected_entry = f""" third_expected_entry = f"""
* Path: {tmp_path} * {tmp_path}
** Heading 3. ** Heading 3.
body line 3 body line 3