From 565d26634badb00e9148764ba2dda13cc3b822b2 Mon Sep 17 00:00:00 2001
From: cryscan <cryscan@umich.edu>
Date: Tue, 25 Jul 2023 12:37:47 +0800
Subject: [PATCH] Log prompt.

---
 src/chat.rs | 7 -------
 src/main.rs | 2 ++
 2 files changed, 2 insertions(+), 7 deletions(-)

diff --git a/src/chat.rs b/src/chat.rs
index 4af4c088..10479ee6 100644
--- a/src/chat.rs
+++ b/src/chat.rs
@@ -90,16 +90,9 @@ impl From<ChatRequest> for GenerateRequest {
             })
             .join("\n\n");
 
-            
-
-
         let assistant = Role::Assistant.to_string();
         let prompt = prompt + &format!("\n\n{assistant}:");
 
-
-        println!("\n\n\n\nthis is prompts \n");
-        println!("{}",prompt);
-        println!("\n\n========================================================================");
         let max_tokens = max_tokens.min(crate::MAX_TOKENS);
         let stop = stop.into();
 
diff --git a/src/main.rs b/src/main.rs
index 239a3b91..6fe5c03a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -223,6 +223,8 @@ fn model_task(
             token_counter.prompt_tokens = tokens.len();
             token_counter.total_tokens = tokens.len();
 
+            log::trace!("{prompt}");
+
             for _ in 0..max_tokens {
                 if token_sender.is_disconnected() {
                     break 'run;