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

Split save_settings method into smaller methods for modularization

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-08-11 20:00:52 +03:00
parent 56ba91fec8
commit dad9133598
1 changed files with 16 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/interface/desktop/configure_screen.py
View File

@@ -141,9 +141,8 @@ class ConfigureScreen(QtWidgets.QDialog):
error_message.setStyleSheet("color: red") error_message.setStyleSheet("color: red")
parent_layout.addWidget(error_message) parent_layout.addWidget(error_message)
def save_settings(self, _): def update_search_settings(self):
"Save the settings to khoj.yml" "Update config with search settings from UI"
# Update config with search settings from UI
for settings_panel in self.search_settings_panels: for settings_panel in self.search_settings_panels:
for child in settings_panel.children(): for child in settings_panel.children():
if not isinstance(child, (SearchCheckBox, FileBrowser)): if not isinstance(child, (SearchCheckBox, FileBrowser)):
@@ -160,7 +159,8 @@ class ConfigureScreen(QtWidgets.QDialog):
elif isinstance(child, FileBrowser) and child.search_type in self.new_config['content-type']: elif isinstance(child, FileBrowser) and child.search_type in self.new_config['content-type']:
self.new_config['content-type'][child.search_type.value]['input-files'] = child.getPaths() self.new_config['content-type'][child.search_type.value]['input-files'] = child.getPaths()
# Update config with conversation settings from UI def update_processor_settings(self):
"Update config with conversation settings from UI"
for settings_panel in self.processor_settings_panels: for settings_panel in self.processor_settings_panels:
for child in settings_panel.children(): for child in settings_panel.children():
if isinstance(child, QtWidgets.QWidget) and child.findChild(ProcessorLineEdit): if isinstance(child, QtWidgets.QWidget) and child.findChild(ProcessorLineEdit):
@@ -180,13 +180,14 @@ class ConfigureScreen(QtWidgets.QDialog):
if child.processor_type == ProcessorType.Conversation: if child.processor_type == ProcessorType.Conversation:
self.new_config['processor'][child.processor_type.value]['openai-api-key'] = child.text() if child.text() != '' else None self.new_config['processor'][child.processor_type.value]['openai-api-key'] = child.text() if child.text() != '' else None
def save_settings_to_file(self) -> bool:
# Validate config before writing to file # Validate config before writing to file
try: try:
yaml_utils.parse_config_from_string(self.new_config) yaml_utils.parse_config_from_string(self.new_config)
except Exception as e: except Exception as e:
print(f"Error validating config: {e}") print(f"Error validating config: {e}")
self.add_error_message(f"Error validating config: {e}", self.layout()) self.add_error_message(f"Error validating config: {e}", self.layout())
return return False
else: else:
# Remove error message if present # Remove error message if present
for i in range(self.layout().count()): for i in range(self.layout().count()):
@@ -197,7 +198,10 @@ class ConfigureScreen(QtWidgets.QDialog):
# Save the config to app config file # Save the config to app config file
yaml_utils.save_config_to_file(self.new_config, self.config_file) yaml_utils.save_config_to_file(self.new_config, self.config_file)
return True
def load_updated_settings(self):
"Hot swap to using the updated config from file"
# Load parsed, validated config from app config file # Load parsed, validated config from app config file
args = cli(state.cli_args) args = cli(state.cli_args)
self.current_config = self.new_config self.current_config = self.new_config
@@ -205,7 +209,13 @@ class ConfigureScreen(QtWidgets.QDialog):
# Configure server with loaded config # Configure server with loaded config
configure_server(args, required=True) configure_server(args, required=True)
self.hide() def save_settings(self):
"Save the settings to khoj.yml"
self.update_search_settings()
self.update_processor_settings()
if self.save_settings_to_file():
self.load_updated_settings()
self.hide()
class SearchCheckBox(QtWidgets.QCheckBox): class SearchCheckBox(QtWidgets.QCheckBox):