diff options
| author | sigoden <sigoden@gmail.com> | 2024-07-01 20:19:41 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-07-01 20:19:41 +0800 |
| commit | 7c6dac061b2a33aae334c08b725e02459c5e9ade (patch) | |
| tree | 3c2067aaf80a24f988600b42aa12b647675bc383 /src/utils | |
| parent | d193950d204ed82bdb3d6a3a111511d189f084bb (diff) | |
| download | aichat-7c6dac061b2a33aae334c08b725e02459c5e9ade.tar.gz | |
feat: support `.rebuild rag` repl command(#672)
Diffstat (limited to 'src/utils')
| -rw-r--r-- | src/utils/request.rs | 4 | ||||
| -rw-r--r-- | src/utils/spinner.rs | 2 |
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)); } } |
