config.rs

// Copyright (c) 2026 vectorless developers
// SPDX-License-Identifier: Apache-2.0

//! Config Python wrapper.

use pyo3::prelude::*;

/// Advanced configuration for Engine internals.
///
/// Create a Config to customize storage, retrieval, concurrency,
/// and other engine parameters beyond the basic builder API.
///
/// Example:
///
/// ```python
/// from vectorless import Config, Engine
///
/// config = Config()
/// config.set_workspace_dir("/data/vectorless")
/// config.set_top_k(10)
/// config.set_max_concurrent_requests(20)
///
/// engine = Engine(api_key="sk-...", model="gpt-4o", config=config)
/// ```
#[pyclass(name = "Config")]
pub struct PyConfig {
    pub(crate) inner: vectorless::Config,
}

#[pymethods]
impl PyConfig {
    /// Create a new Config with defaults.
    #[new]
    fn new() -> Self {
        Self {
            inner: vectorless::Config::default(),
        }
    }

    /// Set the workspace directory for persisted documents.
    ///
    /// Default: ~/.vectorless
    fn set_workspace_dir(&mut self, dir: &str) {
        self.inner.storage.workspace_dir = std::path::PathBuf::from(dir);
    }

    /// Set the number of top-k results to return from queries.
    ///
    /// Default: 3
    fn set_top_k(&mut self, k: usize) {
        self.inner.retrieval.top_k = k;
    }

    /// Set the maximum concurrent LLM API calls.
    ///
    /// Default: 10
    fn set_max_concurrent_requests(&mut self, max: usize) {
        self.inner.llm.throttle.max_concurrent_requests = max;
    }

    /// Set the rate limit (requests per minute).
    ///
    /// Default: 500
    fn set_requests_per_minute(&mut self, rpm: usize) {
        self.inner.llm.throttle.requests_per_minute = rpm;
    }

    /// Set the maximum iterations for retrieval search.
    fn set_max_iterations(&mut self, max: usize) {
        self.inner.retrieval.search.max_iterations = max;
    }

    /// Enable or disable metrics collection.
    ///
    /// Default: True
    fn set_metrics_enabled(&mut self, enabled: bool) {
        self.inner.metrics.enabled = enabled;
    }
}