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

Remove Test suffix from Config models

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

22
src/utils/rawconfig.py
View File

@@ -13,18 +13,18 @@ class ConfigBaseModel(BaseModel):
alias_generator = to_snake_case_from_dash alias_generator = to_snake_case_from_dash
allow_population_by_field_name = True allow_population_by_field_name = True
class SearchConfigTest(ConfigBaseModel): class SearchConfig(ConfigBaseModel):
input_files: Optional[List[str]] input_files: Optional[List[str]]
input_filter: Optional[str] input_filter: Optional[str]
embeddings_file: Optional[Path] embeddings_file: Optional[Path]
class TextSearchConfigTest(ConfigBaseModel): class TextSearchConfig(ConfigBaseModel):
compressed_jsonl: Optional[Path] compressed_jsonl: Optional[Path]
input_files: Optional[List[str]] input_files: Optional[List[str]]
input_filter: Optional[str] input_filter: Optional[str]
embeddings_file: Optional[Path] embeddings_file: Optional[Path]
class ImageSearchConfigTest(ConfigBaseModel): class ImageSearchConfig(ConfigBaseModel):
use_xmp_metadata: Optional[str] use_xmp_metadata: Optional[str]
batch_size: Optional[int] batch_size: Optional[int]
input_directory: Optional[List[str]] input_directory: Optional[List[str]]
@@ -32,10 +32,10 @@ class ImageSearchConfigTest(ConfigBaseModel):
embeddings_file: Optional[Path] embeddings_file: Optional[Path]
class ContentType(ConfigBaseModel): class ContentType(ConfigBaseModel):
org: Optional[TextSearchConfigTest] org: Optional[TextSearchConfig]
ledger: Optional[TextSearchConfigTest] ledger: Optional[TextSearchConfig]
image: Optional[ImageSearchConfigTest] image: Optional[ImageSearchConfig]
music: Optional[TextSearchConfigTest] music: Optional[TextSearchConfig]
class AsymmetricConfig(ConfigBaseModel): class AsymmetricConfig(ConfigBaseModel):
encoder: Optional[str] encoder: Optional[str]
@@ -44,7 +44,7 @@ class AsymmetricConfig(ConfigBaseModel):
class ImageSearchTypeConfig(ConfigBaseModel): class ImageSearchTypeConfig(ConfigBaseModel):
encoder: Optional[str] encoder: Optional[str]
class SearchTypeConfigTest(ConfigBaseModel): class SearchTypeConfig(ConfigBaseModel):
asymmetric: Optional[AsymmetricConfig] asymmetric: Optional[AsymmetricConfig]
image: Optional[ImageSearchTypeConfig] image: Optional[ImageSearchTypeConfig]
@@ -53,10 +53,10 @@ class ProcessorConversationConfig(ConfigBaseModel):
conversation_logfile: Optional[str] conversation_logfile: Optional[str]
conversation_history: Optional[str] conversation_history: Optional[str]
class ProcessorConfigTest(ConfigBaseModel): class ProcessorConfig(ConfigBaseModel):
conversation: Optional[ProcessorConversationConfig] conversation: Optional[ProcessorConversationConfig]
class FullConfig(ConfigBaseModel): class FullConfig(ConfigBaseModel):
content_type: Optional[ContentType] content_type: Optional[ContentType]
search_type: Optional[SearchTypeConfigTest] search_type: Optional[SearchTypeConfig]
processor: Optional[ProcessorConfigTest] processor: Optional[ProcessorConfig]