Struct ConfigManager 

pub struct ConfigManager { /* private fields */ }

Thread-safe mutable configuration manager.

Wraps Settings in Arc<RwLock<>> to allow runtime mutation from setter methods. Tracks a monotonically increasing version counter so that crate::ComponentManager can detect stale cached components and reinitialize them.

Example

use cognee_lib::config::{ConfigManager, Settings};

let cfg = ConfigManager::new(Settings::default());
assert_eq!(cfg.version(), 0);

cfg.set_llm_model("gpt-4o");
assert_eq!(cfg.version(), 1);
assert_eq!(cfg.read().llm_model, "gpt-4o");

Implementations

impl ConfigManager

pub fn new(settings: Settings) -> Self

Create a new ConfigManager wrapping the given settings.

pub fn from_env() -> Self

Convenience constructor: Settings::load_from_env() + wrap.

pub fn read(&self) -> RwLockReadGuard<'_, Settings>

Obtain a read-lock on the current settings.

pub fn version(&self) -> u64

Current config version (monotonically increasing on each mutation).

impl ConfigManager

pub fn set_llm_provider(&self, provider: &str)

pub fn set_llm_model(&self, model: &str)

pub fn set_llm_api_key(&self, key: &str)

pub fn set_llm_endpoint(&self, endpoint: &str)

pub fn set_llm_fallback_model(&self, model: &str)

pub fn set_llm_fallback_provider(&self, provider: &str)

pub fn set_llm_fallback_endpoint(&self, endpoint: &str)

pub fn set_llm_fallback_api_key(&self, key: &str)

pub fn set_embedding_provider(&self, provider: &str)

pub fn set_embedding_model(&self, model: &str)

pub fn set_embedding_dimensions(&self, dims: u32)

pub fn set_embedding_endpoint(&self, endpoint: &str)

pub fn set_embedding_api_key(&self, key: &str)

pub fn set_embedding_api_version(&self, version: &str)

pub fn set_transcription_model(&self, model: &str)

pub fn set_vector_db_provider(&self, provider: &str)

pub fn set_vector_db_url(&self, url: &str)

pub fn set_relational_db_url(&self, url: &str)

Override the relational database URL (e.g. "sqlite:///path/to/db?mode=rwc").

Primarily used by language-binding tests to redirect each test’s DB to an isolated tmp directory so tests do not share the default on-disk DB.

pub fn set_relational_db_config( &self, url: Option<&str>, provider: Option<&str>, host: Option<&str>, port: Option<u16>, name: Option<&str>, username: Option<&str>, password: Option<&str>, )

Set all relational DB connection fields at once.

pub fn set_migration_db_config(&self, url: &str)

Set the migration DB URL.

pub fn set_vector_db_key(&self, key: &str)

pub fn set_graph_database_provider(&self, provider: &str)

pub fn set_graph_model(&self, model: &str)

pub fn set_chunk_strategy(&self, strategy: &str)

pub fn set_chunk_engine(&self, engine: &str)

pub fn set_chunk_size(&self, size: u32)

pub fn set_chunk_overlap(&self, overlap: u32)

pub fn set_data_root_directory(&self, path: &str)

pub fn set_system_root_directory(&self, path: &str)

Set system root directory and cascade derived path updates.

Matches Python config.system_root_directory() (config.py lines 41-67):

• graph_file_path updated if it was under the old system root
• vector_db_url updated if it was under the old system root

pub fn set_monitoring_tool(&self, tool: &str)

pub fn set_classification_model(&self, model: &str)

pub fn set_summarization_model(&self, model: &str)

pub fn set_summarization_schema(&self, schema: Value) -> Result<(), ConfigError>

Set a custom JSON schema for the summarization output stage.

Mirrors Python’s cognee.config.set_summarization_model(CustomSchema): accepts a JSON Schema Value describing the expected LLM output. The schema must contain a summary string field. The stored value is intended to be read by callers when constructing a CognifyConfig via CognifyConfig::with_summary_schema.

Returns Err if the schema fails validation (missing summary field).

pub fn set_llm_api_version(&self, version: &str)

pub fn set_llm_temperature(&self, temperature: f64)

pub fn set_llm_streaming(&self, streaming: bool)

pub fn set_llm_max_completion_tokens(&self, tokens: u32)

pub fn set_llm_max_retries(&self, retries: u32)

pub fn set_llm_max_parallel_requests(&self, parallel: u32)

pub fn set_llm_mock(&self, mock: bool)

Select the record/replay mock LLM (MOCK_LLM parity).

pub fn set_llm_cassette(&self, cassette: &str)

Set the cassette path used by the replay mock (MOCK_LLM_CASSETTE).

pub fn set_llm_record_path(&self, path: &str)

Set the recording cassette output path (COGNEE_RECORD_LLM); empty = unset.

pub fn set_embedding_model_path(&self, path: &str)

pub fn set_embedding_tokenizer_path(&self, path: &str)

pub fn set_vector_db_host(&self, host: &str)

pub fn set_vector_db_port(&self, port: u16)

pub fn set_vector_db_name(&self, name: &str)

pub fn set_graph_file_path(&self, path: &str)

Set graph_file_path directly.

Unlike set_system_root_directory, this is a plain field write and does not cascade to other paths.

pub fn set_cache_root_directory(&self, path: &str)

pub fn set_logs_root_directory(&self, path: &str)

pub fn set_ontology_file_path(&self, path: &str)

pub fn set_ontology_resolver(&self, resolver: &str)

pub fn set_ontology_matching_strategy(&self, strategy: &str)

pub fn get_settings(&self) -> HashMap<String, Value>

Return a snapshot of the current settings with secrets masked.

All secret-bearing fields (*_api_key, *_password, *_key) are replaced with "<redacted>" when non-empty, matching Python’s config.get_settings() behaviour so callers can safely log or expose the output without leaking credentials.

The returned map can be serialized to JSON for logging or debugging.

impl ConfigManager

pub fn set_llm_config( &self, values: &HashMap<String, Value>, ) -> Result<(), ConfigError>

Bulk-update LLM config from a map. Matches Python config.set_llm_config().

pub fn set_embedding_config( &self, values: &HashMap<String, Value>, ) -> Result<(), ConfigError>

Bulk-update embedding config from a map. Matches Python config.set_embedding_config().

pub fn set_vector_db_config( &self, values: &HashMap<String, Value>, ) -> Result<(), ConfigError>

Bulk-update vector DB config from a map. Matches Python config.set_vector_db_config().

pub fn set_graph_db_config( &self, values: &HashMap<String, Value>, ) -> Result<(), ConfigError>

Bulk-update graph DB config from a map. Matches Python config.set_graph_db_config().

pub fn set(&self, key: &str, value: Value) -> Result<(), ConfigError>

Generic setter matching Python’s config.set(key, value).

Dispatches to the appropriate typed setter based on the key name. Returns ConfigError::UnknownKey for unrecognized keys and ConfigError::TypeMismatch when the JSON value type doesn’t match.

Trait Implementations

impl Clone for ConfigManager

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.