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

Update the search_config instantiated for tests in conftest

Browse Source
This commit is contained in:
Saba
2021-12-11 14:14:31 -05:00
parent 9ebf00e29b
commit ba8dc9ed5f
2 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/conftest.py
View File

@@ -14,7 +14,7 @@ def model_dir(tmp_path_factory):
# Generate Image Embeddings from Test Images
search_config = ContentTypeConfig()
search_config.image = ImageSearchConfig(
input_directory = Path('tests/data'),
input_directory = 'tests/data',
embeddings_file = model_dir.joinpath('.image_embeddings.pt'),
batch_size = 10,
use_xmp_metadata = False)
@@ -23,12 +23,12 @@ def model_dir(tmp_path_factory):
# Generate Notes Embeddings from Test Notes
search_config.org = TextSearchConfig(
input_files = [Path('tests/data/main_readme.org'), Path('tests/data/interface_emacs_readme.org')],
input_files = ['tests/data/main_readme.org', 'tests/data/interface_emacs_readme.org'],
input_filter = None,
compressed_jsonl = model_dir.joinpath('.notes.jsonl.gz'),
embeddings_file = model_dir.joinpath('.note_embeddings.pt'))
asymmetric.setup(search_config.notes, regenerate=False, verbose=True)
asymmetric.setup(search_config.org, regenerate=False, verbose=True)
return model_dir
@@ -37,14 +37,14 @@ def model_dir(tmp_path_factory):
def search_config(model_dir):
search_config = ContentTypeConfig()
search_config.org = TextSearchConfig(
input_files = [Path('tests/data/main_readme.org'), Path('tests/data/interface_emacs_readme.org')],
input_files = ['tests/data/main_readme.org', 'tests/data/interface_emacs_readme.org'],
input_filter = None,
compressed_jsonl = model_dir.joinpath('.notes.jsonl.gz'),
embeddings_file = model_dir.joinpath('.note_embeddings.pt'))
search_config.image = ImageSearchConfig(
input_directory = Path('tests/data'),
embeddings_file = Path('tests/data/.image_embeddings.pt'),
input_directory = 'tests/data',
embeddings_file = 'tests/data/.image_embeddings.pt',
batch_size = 10,
use_xmp_metadata = False)