Method load_retrieval_config

AskAgent/python/core/config.py:318–353

(self, path: str = "config_retrieval.yaml")

            )

    def load_retrieval_config(self, path: str = "config_retrieval.yaml"):
        # Build the full path to the config file using the config directory
        full_path = os.path.join(self.config_directory, path)

        try:
            with open(full_path) as f:
                data = yaml.safe_load(f)
        except FileNotFoundError:
            # If config file doesn't exist, use defaults
            print(f"Warning: {path} not found. Using default retrieval configuration.")
            data = {
                "preferred_endpoint": "default",
                "endpoints": {}
            }

        # No longer using preferred_endpoint - now using enabled field on each endpoint
        self.retrieval_endpoints: dict[str, RetrievalProviderConfig] = {}

        # Get the write endpoint for database modifications
        self.write_endpoint: str = data.get("write_endpoint")

        # Changed from providers to endpoints
        for name, cfg in data.get("endpoints", {}).items():
            # Use the new method for all configuration values
            self.retrieval_endpoints[name] = RetrievalProviderConfig(
                api_key=self._get_config_value(cfg.get("api_key_env")),
                api_key_env=cfg.get("api_key_env"),  # Store the env var name
                api_endpoint=self._get_config_value(cfg.get("api_endpoint_env")),
                api_endpoint_env=cfg.get("api_endpoint_env"),  # Store the env var name
                database_path=self._get_config_value(cfg.get("database_path")),
                index_name=self._get_config_value(cfg.get("index_name")),
                db_type=self._get_config_value(cfg.get("db_type")),  # Add db_type
                enabled=cfg.get("enabled", False),  # Add enabled field
                use_knn=cfg.get("use_knn"),
                vector_type=cfg.get("vector_type")
            )

    def load_webserver_config(self, path: str = "config_webserver.yaml"):
        # Build the full path to the config file using the config directory