klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 21:19:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Scheduled, Deadlne date to Model Embeddings for Date Aware Search

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-06-17 02:55:27 +03:00
parent cfbd5c4ecc
commit a2a08d1354
2 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/processor/org_mode/org_to_jsonl.py
View File

@@ -120,6 +120,16 @@ def convert_org_entries_to_jsonl(entries, verbose=0):
if verbose > 2:
print(f"Body: {entry.Body()}")
if entry.Scheduled():
entry_dict["Scheduled"] = entry.Scheduled().strftime("%Y-%m-%d")
if verbose > 2:
print(f"Scheduled: {entry.Scheduled().strftime('%Y-%m-%d')}")
if entry.Deadline():
entry_dict["Deadline"] = entry.Deadline().strftime("%Y-%m-%d")
if verbose > 2:
print(f"Deadline: {entry.Deadline().strftime('%Y-%m-%d')}")
if entry_dict:
entry_dict["Raw"] = f'{entry}'

4
src/search_type/asymmetric.py
View File

@@ -60,7 +60,9 @@ def extract_entries(notesfile, verbose=0):
if not "Body" in note or note["Body"].strip(empty_escape_sequences) == "":
continue
note_string = f'{note["Title"]}\t{note["Tags"] if "Tags" in note else ""}\n{note["Body"] if "Body" in note else ""}'
note_string = f'''{note["Title"]}\t{note["Tags"] if "Tags" in note else ""}
{"Scheduled on " + note["Scheduled"] if "Scheduled" in note else ""}
{note["Body"] if "Body" in note else ""}'''
entries.append([note_string, note["Raw"]])
# Close File