From 47818159a7ca3910cdc6ed00109a331d2ac2f906 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 27 Oct 2023 09:33:34 +0800
Subject: refactor: streaming input text in dry-run mode (#154)

---
 src/client/mod.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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

diff --git a/src/client/mod.rs b/src/client/mod.rs
index bf83c60..1f9e66c 100644
--- a/src/client/mod.rs
+++ b/src/client/mod.rs
@@ -17,6 +17,7 @@ use crate::{
     client::localai::LocalAIClient,
     config::{Config, SharedConfig},
     repl::{ReplyStreamHandler, SharedAbortSignal},
+    utils::split_text,
 };
 
 #[allow(clippy::struct_excessive_bools)]
@@ -92,7 +93,11 @@ pub trait Client {
             tokio::select! {
                 ret = async {
                     if self.get_config().read().dry_run {
-                        handler.text(&self.get_config().read().echo_messages(content))?;
+                        let words = split_text(content)?;
+                        for word in words {
+                            tokio::time::sleep(Duration::from_millis(25)).await;
+                            handler.text(&self.get_config().read().echo_messages(&word))?;
+                        }
                         return Ok(());
                     }
                     self.send_message_streaming_inner(content, handler).await
-- 
cgit v1.2.3