Remove thinking from response

Author: mcasperson

background.js

@@ -65,9 +65,11 @@ messenger.messages.onNewMailReceived.addListener(async (folder, messages) => {
             continue
         }
 
-        console.log(summary)
+        const processedSummary = await processResponse(summary)
 
-        await sendNewEmail(message, summary)
+        console.log(processedSummary)
+
+        await sendNewEmail(message, processedSummary)
     }
 })
 
@@ -170,6 +172,16 @@ async function getPrompt(content) {
     return await getLlamaPrompt(content)
 }
 
+async function processResponse(response) {
+    const model = await getModel()
+
+    if (model.startsWith("qwen")) {
+        return response.replace(/<think>.*?<\/think>/gs, '').trim()
+    }
+
+    return response
+}
+
 async function getPhiPrompt(content) {
     return "<|im_start|>system<|im_sep|>" +
         "You are an expert in reading and summarizing emails." +

manifest.json

@@ -2,7 +2,7 @@
   "manifest_version": 2,
   "name": "Ollama Summarizer",
   "description": "Generate email summaries with Ollama",
-  "version": "1.9",
+  "version": "1.10",
   "author": "Matthew Casperson",
   "browser_specific_settings": {
     "gecko": {