summaryrefslogtreecommitdiffstats
path: root/src/utils
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-07-01 20:19:41 +0800
committerGitHub <noreply@github.com>2024-07-01 20:19:41 +0800
commit7c6dac061b2a33aae334c08b725e02459c5e9ade (patch)
tree3c2067aaf80a24f988600b42aa12b647675bc383 /src/utils
parentd193950d204ed82bdb3d6a3a111511d189f084bb (diff)
downloadaichat-7c6dac061b2a33aae334c08b725e02459c5e9ade.tar.gz
feat: support `.rebuild rag` repl command(#672)
Diffstat (limited to 'src/utils')
-rw-r--r--src/utils/request.rs4
-rw-r--r--src/utils/spinner.rs2
2 files changed, 5 insertions, 1 deletions
diff --git a/src/utils/request.rs b/src/utils/request.rs
index fadd3b8..678e2bb 100644
--- a/src/utils/request.rs
+++ b/src/utils/request.rs
@@ -48,7 +48,9 @@ pub async fn fetch(loaders: &HashMap<String, String>, path: &str) -> Result<(Str
}
let result = match loaders.get(&extension) {
Some(loader_command) => {
- let save_path = temp_file("-download-", "").display().to_string();
+ let save_path = temp_file("-download-", &format!(".{extension}"))
+ .display()
+ .to_string();
let mut save_file = tokio::fs::File::create(&save_path).await?;
let mut size = 0;
while let Some(chunk) = res.chunk().await? {
diff --git a/src/utils/spinner.rs b/src/utils/spinner.rs
index a04a18f..8f386db 100644
--- a/src/utils/spinner.rs
+++ b/src/utils/spinner.rs
@@ -78,11 +78,13 @@ impl Drop for Spinner {
impl Spinner {
pub fn set_message(&self, message: String) -> Result<()> {
self.0.send(SpinnerEvent::SetMessage(message))?;
+ std::thread::sleep(Duration::from_millis(10));
Ok(())
}
pub fn stop(&self) {
let _ = self.0.send(SpinnerEvent::Stop);
+ std::thread::sleep(Duration::from_millis(10));
}
}