Skip to main content

entrenar/citl/pattern_store/
config.rs

1//! Configuration for the pattern store.
2
3use serde::{Deserialize, Serialize};
4
5/// Default embedding dimension for the pattern store
6const DEFAULT_EMBEDDING_DIM: usize = 384;
7
8/// Configuration for the pattern store
9#[derive(Debug, Clone, Serialize, Deserialize)]
10pub struct PatternStoreConfig {
11    /// Chunk size for the chunker (default: 256)
12    pub chunk_size: usize,
13    /// Embedding dimension (default: 384)
14    pub embedding_dim: usize,
15    /// RRF k constant (default: 60.0)
16    pub rrf_k: f32,
17}
18
19impl Default for PatternStoreConfig {
20    fn default() -> Self {
21        Self { chunk_size: 256, embedding_dim: DEFAULT_EMBEDDING_DIM, rrf_k: 60.0 }
22    }
23}