sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 15:38:55 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Streamline None checking in initialize_search

Browse source
This commit is contained in:
Saba 2021-11-28 11:59:45 -05:00
parent da52433d89
commit bf8ae31e6a
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/main.py
View file

@ -127,23 +127,23 @@ def initialize_search(config: FullConfig, regenerate, verbose):
search_config = SearchConfig()
# Initialize Org Notes Search
search_config.notes = TextSearchConfig(config.content_type.org, verbose) if config.content_type.org else None
if search_config.notes:
if config.content_type.org:
search_config.notes = TextSearchConfig(config.content_type.org, verbose)
model.notes_search = asymmetric.setup(search_config.notes, regenerate=regenerate)
# Initialize Org Music Search
search_config.music = TextSearchConfig(config.content_type.music, verbose) if config.content_type.music else None
if search_config.music:
if config.content_type.music:
search_config.music = TextSearchConfig(config.content_type.music, verbose)
model.music_search = asymmetric.setup(search_config.music, regenerate=regenerate)
# Initialize Ledger Search
search_config.ledger = TextSearchConfig(config.content_type.org, verbose) if config.content_type.ledger else None
if search_config.ledger:
if config.content_type.ledger:
search_config.ledger = TextSearchConfig(config.content_type.org, verbose)
model.ledger_search = symmetric_ledger.setup(search_config.ledger, regenerate=regenerate)
# Initialize Image Search
search_config.image = ImageSearchConfig(config.content_type.image, verbose) if config.content_type.image else None
if search_config.image:
if config.content_type.image:
search_config.image = ImageSearchConfig(config.content_type.image, verbose)
model.image_search = image_search.setup(search_config.image, regenerate=regenerate)
return model, search_config