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

Update unit tests, files with removing model suffix to config types

Browse Source
This commit is contained in:
Saba
2021-12-09 08:50:38 -05:00
parent b3eac888fb
commit d65190c3ee
9 changed files with 55 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/utils/config.py
View File

@@ -4,7 +4,7 @@ from dataclasses import dataclass
from pathlib import Path
# Internal Packages
from src.utils.rawconfig import ProcessorConversationConfigModel
from src.utils.rawconfig import ConversationProcessorConfig
class SearchType(str, Enum):
@@ -42,8 +42,8 @@ class SearchModels():
image_search: ImageSearchModel = None
class ConversationProcessorConfigDTO():
def __init__(self, processor_config: ProcessorConversationConfigModel, verbose: bool):
class ConversationProcessorConfigModel():
def __init__(self, processor_config: ConversationProcessorConfig, verbose: bool):
self.openai_api_key = processor_config.open_api_key
self.conversation_logfile = Path(processor_config.conversation_logfile)
self.chat_log = ''
@@ -52,5 +52,5 @@ class ConversationProcessorConfigDTO():
self.verbose = verbose
@dataclass
class ProcessorConfig():
conversation: ConversationProcessorConfigDTO = None
class ProcessorConfigModel():
conversation: ConversationProcessorConfigModel = None