From ac9c1509d5e6469607ceedb6426ebba52481734f Mon Sep 17 00:00:00 2001
From: "Rodrigo Rodriguez (Pragmatismo)" <me@rodrigorodriguez.com>
Date: Wed, 5 Nov 2025 10:35:54 -0300
Subject: [PATCH] feat(llm): improve logging and content processing

- Add 'keyword' to LLM processing log message for better context
- Replace simple string replace with regex for removing <think> tags in DeepseekR3 model
- The changes provide more precise logging and more robust content processing
---
 src/basic/keywords/llm_keyword.rs | 2 +-
 src/llm_models/deepseek_r3.rs     | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/basic/keywords/llm_keyword.rs b/src/basic/keywords/llm_keyword.rs
index b3b8657b..1c8e0c45 100644
--- a/src/basic/keywords/llm_keyword.rs
+++ b/src/basic/keywords/llm_keyword.rs
@@ -13,7 +13,7 @@ pub fn llm_keyword(state: Arc<AppState>, _user: UserSession, engine: &mut Engine
         .register_custom_syntax(&["LLM", "$expr$"], false, move |context, inputs| {
             let text = context.eval_expression_tree(&inputs[0])?.to_string();
 
-            info!("LLM processing text: {}", text);
+            info!("LLM keyword processing text: {}", text);
 
             let state_for_thread = Arc::clone(&state_clone);
             let prompt = build_llm_prompt(&text);
diff --git a/src/llm_models/deepseek_r3.rs b/src/llm_models/deepseek_r3.rs
index 686a32a6..cce8a624 100644
--- a/src/llm_models/deepseek_r3.rs
+++ b/src/llm_models/deepseek_r3.rs
@@ -1,4 +1,5 @@
 use super::ModelHandler;
+use regex;
 
 pub struct DeepseekR3Handler;
 
@@ -8,7 +9,8 @@ impl ModelHandler for DeepseekR3Handler {
     }
 
     fn process_content(&self, content: &str) -> String {
-        content.replace("<think>", "").replace("</think>", "")
+        let re = regex::Regex::new(r"<think>.*?</think>").unwrap();
+        re.replace_all(content, "").to_string()
     }
 
     fn has_analysis_markers(&self, buffer: &str) -> bool {