Log prompt.

Ai00-X · Jul 25, 2023 · 565d266 · 565d266
1 parent 2f21437
commit 565d266
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git 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
@@ -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;