diff --git a/src/core/kb/kb_indexer.rs b/src/core/kb/kb_indexer.rs index a6476b8e..fde6c3c4 100644 --- a/src/core/kb/kb_indexer.rs +++ b/src/core/kb/kb_indexer.rs @@ -747,7 +747,16 @@ pub struct KbFolderMonitor { impl KbFolderMonitor { pub fn new(work_root: PathBuf, embedding_config: EmbeddingConfig) -> Self { - let qdrant_config = QdrantConfig::default(); + let qdrant_config = if let Some(sm) = crate::core::shared::utils::get_secrets_manager_sync() { + let (url, api_key) = sm.get_vectordb_config_sync(); + QdrantConfig { + url, + api_key, + timeout_secs: 30, + } + } else { + QdrantConfig::default() + }; let indexer = KbIndexer::new(embedding_config, qdrant_config); Self { indexer, work_root } diff --git a/src/core/kb/website_crawler_service.rs b/src/core/kb/website_crawler_service.rs index ce06795d..e7ebda0f 100644 --- a/src/core/kb/website_crawler_service.rs +++ b/src/core/kb/website_crawler_service.rs @@ -241,9 +241,9 @@ impl WebsiteCrawlerService { let embedding_config = embedding_config_with_vault(&db_pool, &website.bot_id, &bot_name).await; info!("Using embedding URL: {} for bot {}", embedding_config.embedding_url, bot_name); - // Create bot-specific KB indexer with correct embedding config - let qdrant_config = QdrantConfig::default(); - let bot_indexer = KbIndexer::new(embedding_config, qdrant_config); + // Create bot-specific KB indexer with correct embedding config + let qdrant_config = QdrantConfig::from_config(db_pool.clone(), &website.bot_id); + let bot_indexer = KbIndexer::new(embedding_config, qdrant_config); // Process pages in small batches to prevent memory exhaustion const BATCH_SIZE: usize = 5; diff --git a/src/drive/local_file_monitor.rs b/src/drive/local_file_monitor.rs index 4dc21e34..3bc15b22 100644 --- a/src/drive/local_file_monitor.rs +++ b/src/drive/local_file_monitor.rs @@ -291,11 +291,11 @@ impl LocalFileMonitor { continue; } - info!("Indexing KB '{}' for bot '{}'", kb_name, bot_name); + info!("Indexing KB '{}' for bot '{}'", kb_name, bot_name); - // Create a temporary KbIndexer with the bot-specific config - let qdrant_config = crate::core::kb::QdrantConfig::default(); - let indexer = crate::core::kb::KbIndexer::new(embedding_config.clone(), qdrant_config); + // Create a temporary KbIndexer with the bot-specific config + let qdrant_config = crate::core::kb::QdrantConfig::from_config(self.state.conn.clone(), &bot_id); + let indexer = crate::core::kb::KbIndexer::new(embedding_config.clone(), qdrant_config); if let Err(e) = indexer.index_kb_folder( bot_id,