From 8e8597262d94c8858ff85f392ab0bd35bff8fcd6 Mon Sep 17 00:00:00 2001
From: cryscan <cryscan@umich.edu>
Date: Wed, 26 Jul 2023 18:32:40 +0800
Subject: [PATCH] Log remaining prompt.

---
 src/main.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 43b6cb95..f0ea3840 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -222,7 +222,7 @@ fn model_task(
             token_counter.prompt_tokens = tokens.len();
             token_counter.total_tokens = tokens.len();
 
-            log::trace!("{prompt}");
+            log::trace!("{}", String::from_utf8(remain).unwrap_or_default());
 
             for _ in 0..max_tokens {
                 if token_sender.is_disconnected() {
@@ -267,7 +267,7 @@ fn model_task(
             let _ = token_sender.send(Token::Stop(FinishReason::Length, token_counter));
         }
 
-        print!("\n\n");
+        println!("[DONE]");
 
         if let Ok(back) = state.back() {
             if embedding {