fix: 正则标记异常处理

bincooo · Jun 23, 2024 · d869907 · d869907
1 parent 0784877
commit d869907
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/internal/common/parser.go b/internal/common/parser.go
@@ -421,12 +421,16 @@ func XmlFlags(ctx *gin.Context, req *pkg.ChatCompletion) []Matcher {
 				pos = len(req.Messages)
 			}
 
-			c := regexp.MustCompile(cmp, regexp.Compiled)
+			c, err := regexp.Compile(cmp, regexp.Compiled)
+			if err != nil {
+				logger.Warn("compile failed: "+cmp, err)
+				continue
+			}
 			for idx, message := range req.Messages {
 				if idx < pos && !message.Is("role", "system") {
-					replace, err := c.Replace(message.GetString("content"), value, -1, -1)
-					if err != nil {
-						logger.Warn("compile failed: "+cmp, err)
+					replace, e := c.Replace(message.GetString("content"), value, -1, -1)
+					if e != nil {
+						logger.Warn("compile failed: "+cmp, e)
 						continue
 					}
 					message["content"] = replace

diff --git a/internal/plugin/llm/you/message.go b/internal/plugin/llm/you/message.go
@@ -173,7 +173,7 @@ func mergeMessages(completion pkg.ChatCompletion) (pMessages []you.Message, text
 		return
 	}
 
-	text = "continue"
+	text = "Please review the attached prompt"
 
 	// 获取最后一条用户消息
 	if is32 {
@@ -218,7 +218,6 @@ func mergeMessages(completion pkg.ChatCompletion) (pMessages []you.Message, text
 		}
 		pMessages = append(pMessages, newMessage)
 		pos++
-		break
 	}
 	return
 }