From 24ecf67c1c2d0ca83307b1c7ea76ac92a1490c22 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=A2=AB=E9=81=97=E5=BF=98=E7=9A=84=E8=AE=B0=E5=BF=86?=
 <32097237+YIWANG-sketch@users.noreply.github.com>
Date: Thu, 28 Nov 2024 00:51:18 +0800
Subject: [PATCH] fix Role replacement

---
 formatMessages.mjs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/formatMessages.mjs b/formatMessages.mjs
index 4f15539..96fd2b9 100644
--- a/formatMessages.mjs
+++ b/formatMessages.mjs
@@ -147,7 +147,7 @@ function getRoleFeatures(isClaudeModel, useBackspacePrefix) {
 
 // 转换角色
 function convertRoles(messages, roleFeatures) {
-    const { prefix, systemRole, userRole, assistantRole } = roleFeatures;
+    const { systemRole, userRole, assistantRole } = roleFeatures;
     const roleMap = {
         'system': systemRole,
         'user': userRole,
@@ -158,7 +158,7 @@ function convertRoles(messages, roleFeatures) {
     return messages.map(message => {
         let currentRole = message.role;
 
-        if (currentRole.startsWith(prefix)) {
+        if (currentRole.startsWith('\u0008')) {
             // 包含前缀不需要转换
             return message;
         } else {