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

Update unit tests to use the new BaseModel types

Browse Source
This commit is contained in:
Saba
2021-12-05 09:31:39 -05:00
parent c4cd4b57f1
commit 76e9e9da2f
5 changed files with 19 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/utils/cli.py
View File

@@ -38,12 +38,14 @@ def cli(args=None):
config_from_file = yaml.safe_load(config_file)
args.config = merge_dicts(priority_dict=config_from_file, default_dict=args.config)
args.config = FullConfigModel.parse_obj(args.config)
else:
args.config = FullConfigModel.parse_obj(args.config)
if args.org_files:
args.config['content-type']['org']['input-files'] = args.org_files
args.config.content_type.org.input_files = args.org_files
if args.org_filter:
args.config['content-type']['org']['input-filter'] = args.org_filter
args.config.content_type.org.input_filter = args.org_filter
return args

2
src/utils/rawconfig.py
View File

@@ -27,7 +27,7 @@ class TextSearchConfigModel(ConfigBaseModel):
class ImageSearchConfigModel(ConfigBaseModel):
use_xmp_metadata: Optional[str]
batch_size: Optional[int]
input_directory: Optional[List[str]]
input_directory: Optional[Path]
input_filter: Optional[str]
embeddings_file: Optional[Path]