klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 05:40:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Consolidate jsonl helper methods in a single file under utils module

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-07-21 03:28:37 +04:00
parent de726c4b6c
commit 0917f1574d
6 changed files with 57 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/search_type/asymmetric.py
View File

@@ -10,10 +10,11 @@ import torch
from sentence_transformers import SentenceTransformer, CrossEncoder, util
# Internal Packages
from src.utils.helpers import get_absolute_path, resolve_absolute_path, load_model, load_jsonl
from src.utils.helpers import get_absolute_path, resolve_absolute_path, load_model
from src.processor.org_mode.org_to_jsonl import org_to_jsonl
from src.utils.config import TextSearchModel
from src.utils.rawconfig import AsymmetricSearchConfig, TextContentConfig
from src.utils.jsonl import load_jsonl
def initialize_model(search_config: AsymmetricSearchConfig):

3
src/search_type/symmetric_ledger.py
View File

@@ -8,10 +8,11 @@ import torch
from sentence_transformers import SentenceTransformer, CrossEncoder, util
# Internal Packages
from src.utils.helpers import get_absolute_path, resolve_absolute_path, load_model, load_jsonl
from src.utils.helpers import get_absolute_path, resolve_absolute_path, load_model
from src.processor.ledger.beancount_to_jsonl import beancount_to_jsonl
from src.utils.config import TextSearchModel
from src.utils.rawconfig import SymmetricSearchConfig, TextContentConfig
from src.utils.jsonl import load_jsonl
def initialize_model(search_config: SymmetricSearchConfig):