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

Update imports for raw config models in config.py

Browse Source
This commit is contained in:
Saba
2021-12-04 10:44:55 -05:00
parent 4d6284b0af
commit e068968b35
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/utils/config.py
View File

@@ -6,7 +6,7 @@ from pathlib import Path
# Internal Packages # Internal Packages
from src.utils.helpers import get_from_dict from src.utils.helpers import get_from_dict
from src.utils.rawconfig import TextSearchConfigTest, ImageSearchConfigTest, ProcessorConversationConfig from src.utils.rawconfig import TextSearchConfig, ImageSearchConfig, ProcessorConversationConfig
class SearchType(str, Enum): class SearchType(str, Enum):
@@ -45,7 +45,7 @@ class SearchModels():
class TextSearchConfig(): class TextSearchConfig():
def __init__(self, text_search_config: TextSearchConfigTest, verbose: bool): def __init__(self, text_search_config: TextSearchConfig, verbose: bool):
self.input_files = text_search_config.input_files self.input_files = text_search_config.input_files
self.input_filter = text_search_config.input_filter self.input_filter = text_search_config.input_filter
self.compressed_jsonl = Path(text_search_config.compressed_jsonl) self.compressed_jsonl = Path(text_search_config.compressed_jsonl)
@@ -54,7 +54,7 @@ class TextSearchConfig():
class ImageSearchConfig(): class ImageSearchConfig():
def __init__(self, image_search_config: ImageSearchConfigTest, verbose): def __init__(self, image_search_config: ImageSearchConfig, verbose):
self.input_directory = Path(image_search_config.input_directory) self.input_directory = Path(image_search_config.input_directory)
self.embeddings_file = Path(image_search_config.embeddings_file) self.embeddings_file = Path(image_search_config.embeddings_file)
self.batch_size = image_search_config.batch_size self.batch_size = image_search_config.batch_size