Просмотр исходного кода

server : filter out harmony thought messages (#15278)

Aldehir Rojas 5 месяцев назад
Родитель
Сommit
e885445bc1
2 измененных файлов с 6 добавлено и 1 удалено
  1. BIN
      tools/server/public/index.html.gz
  2. 6 1
      tools/server/webui/src/utils/misc.ts

BIN
tools/server/public/index.html.gz


+ 6 - 1
tools/server/webui/src/utils/misc.ts

@@ -130,7 +130,12 @@ export function filterThoughtFromMsgs(messages: APIMessage[]) {
       role: msg.role,
       content:
         msg.role === 'assistant'
-          ? contentStr.split('</think>').at(-1)!.trim()
+          ? contentStr
+              .split(
+                /<\/think>|<\|start\|>assistant<\|channel\|>final<\|message\|>/
+              )
+              .at(-1)!
+              .trim()
           : contentStr,
     } as APIMessage;
   });