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

Add Search Config for Symmetric Model. Save Model to Disk

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-01-14 16:46:56 -05:00
parent b63026d97c
commit 934ec233b0
4 changed files with 33 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/utils/rawconfig.py
View File

@@ -37,6 +37,11 @@ class ContentTypeConfig(ConfigBase):
image: Optional[ImageSearchConfig]
music: Optional[TextSearchConfig]
class SymmetricConfig(ConfigBase):
encoder: Optional[str]
cross_encoder: Optional[str]
model_directory: Optional[Path]
class AsymmetricConfig(ConfigBase):
encoder: Optional[str]
cross_encoder: Optional[str]
@@ -47,6 +52,7 @@ class ImageSearchTypeConfig(ConfigBase):
class SearchTypeConfig(ConfigBase):
asymmetric: Optional[AsymmetricConfig]
symmetric: Optional[SymmetricConfig]
image: Optional[ImageSearchTypeConfig]
class ConversationProcessorConfig(ConfigBase):