From 7c6dac061b2a33aae334c08b725e02459c5e9ade Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 1 Jul 2024 20:19:41 +0800
Subject: feat: support `.rebuild rag` repl command(#672)

---
 src/config/mod.rs | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'src/config/mod.rs')

diff --git a/src/config/mod.rs b/src/config/mod.rs
index 879f511..25891bd 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -885,11 +885,23 @@ impl Config {
         Ok(())
     }
 
+    pub async fn rebuild_rag(config: &GlobalConfig, abort_signal: AbortSignal) -> Result<()> {
+        let rag_name = match config.read().rag.clone() {
+            Some(v) => v.name().to_string(),
+            None => bail!("No RAG"),
+        };
+        let rag_path = config.read().rag_file(&rag_name)?;
+        let mut rag = Rag::load(config, &rag_name, &rag_path)?;
+        rag.rebuild(config, &rag_path, abort_signal).await?;
+        config.write().rag = Some(Arc::new(rag));
+        Ok(())
+    }
+
     pub fn rag_info(&self) -> Result<String> {
         if let Some(rag) = &self.rag {
             rag.export()
         } else {
-            bail!("No rag")
+            bail!("No RAG")
         }
     }
 
-- 
cgit v1.2.3