diff --git a/config/base_config.yaml b/config/base_config.yaml index 5699b399..1d6c71af 100644 --- a/config/base_config.yaml +++ b/config/base_config.yaml @@ -9,7 +9,7 @@ construction: - annoy_eng - demo max_workers: 32 - mode: agent + mode: agent # set noagent to speed up overlap: 200 tree_comm: embedding_model: all-MiniLM-L6-v2 @@ -316,5 +316,5 @@ retrieval: top_k_filter: 20 triggers: constructor_trigger: true - mode: agent + mode: agent # set noagent to speed up retrieve_trigger: true diff --git a/config/config_loader.py b/config/config_loader.py index e8891aca..02252c9b 100644 --- a/config/config_loader.py +++ b/config/config_loader.py @@ -230,7 +230,7 @@ def _validate_config(self) -> None: if self.triggers.mode not in valid_modes: raise ValueError(f"Invalid mode: {self.triggers.mode}. Must be one of {valid_modes}") - if self.construction.mode not in ["agent", "basic"]: + if self.construction.mode not in valid_modes: raise ValueError(f"Invalid construction mode: {self.construction.mode}") # Validate numerical parameters