Update app.py

app.py

@@ -165,7 +165,15 @@ class WebContentFetcher:
             results.append(result)
             time.sleep(1)  # Be respectful to servers
         return results
-
+        
+def remove_thinking_tags(text):
+    import re
+    # Remove <think>...</think> blocks
+    cleaned = re.sub(r'<think>.*?</think>', '', text, flags=re.DOTALL)
+    # Remove thinking markers
+    cleaned = re.sub(r'<thinking>.*?</thinking>', '', cleaned, flags=re.DOTALL)
+    return cleaned.strip()
+    
 # --- File Download Utility ---
 def download_attachment(url: str, temp_dir: str) -> Optional[str]:
     """
@@ -365,9 +373,9 @@ class IntelligentAgent:
                 response = self.client.chat_completion(
                     messages=messages,
                     max_tokens=max_tokens,
-                    temperature=temperature,
-                    chat_template_kwargs = {"enable_thinking": False}
+                    temperature=temperature
                 )
+                response = remove_thinking_tags(response)
                 return response.choices[0].message.content.strip()
             except Exception as chat_error:
                 if self.debug:
@@ -378,9 +386,9 @@ class IntelligentAgent:
                     prompt,
                     max_new_tokens=max_tokens,
                     temperature=temperature,
-                    chat_template_kwargs = {"enable_thinking": False},
                     do_sample=temperature > 0
                 )
+                response = remove_thinking_tags(response)
                 return response.strip()
                 
         except Exception as e: