From 700d8a3245f1133c37e035039e511a1e5dce1a5d Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 3 Feb 2025 22:35:01 +0800
Subject: feat: strip reasoning contents (#1141)

---
 assets/arena.html | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

(limited to 'assets/arena.html')

diff --git a/assets/arena.html b/assets/arena.html
index 31973eb..2296de2 100644
--- a/assets/arena.html
+++ b/assets/arena.html
@@ -870,6 +870,7 @@
               });
             }
           }
+          sanitizeMessages(messages);
           const body = {
             model: chat.model,
             messages: messages,
@@ -991,6 +992,20 @@
       });
     }
 
+    function sanitizeMessages(messages) {
+      let messagesLen = messages.length;
+      for (let i = 0; i < messagesLen; i++) {
+        const message = messages[i];
+        if (typeof message.content === "string" && message.role === "assistant" && i !== messagesLen - 1) {
+          message.content = stripThinkTag(message.content);
+        }
+      }
+    }
+
+    function stripThinkTag(text) {
+      return text.replace(/^\s*<think>([\s\S]*?)<\/think>(\s*|$)/g, '')
+    }
+
     function setupMarked() {
       const renderer = {
         code({ text, lang }) {
@@ -1013,7 +1028,12 @@
         name: 'think',
         level: 'block',
         start(src) {
-          return src.indexOf('<think>');
+          const match = /^(\s*)<think>/.exec(src);
+          if (match) {
+            return match[1].length
+          } else {
+            return -1;
+          }
         },
         tokenizer(src, tokens) {
           const rule = /^\s*<think>([\s\S]*?)(<\/think>|$)/;
-- 
cgit v1.2.3