Merge 48c8496 into 6ad461d

Hoshinonyaruko · Apr 20, 2024 · 2a50ea3 · 2a50ea3
2 parents 6ad461d + 48c8496
commit 2a50ea3
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 4 deletions.
diff --git a/applogic/chatgpt.go b/applogic/chatgpt.go
@@ -361,9 +361,15 @@ func (app *App) ChatHandlerChatgpt(w http.ResponseWriter, r *http.Request) {
 
 					newContent := ""
 					for _, choice := range eventData.Choices {
+						// 如果新内容以旧内容开头
 						if strings.HasPrefix(choice.Delta.Content, lastResponseText) {
-							// 如果新内容以旧内容开头，剔除旧内容部分，只保留新增的部分
-							newContent += choice.Delta.Content[len(lastResponseText):]
+							// 特殊情况：当新内容和旧内容完全相同时，处理逻辑应当与新内容不以旧内容开头时相同
+							if choice.Delta.Content == lastResponseText {
+								newContent += choice.Delta.Content
+							} else {
+								// 剔除旧内容部分，只保留新增的部分
+								newContent += choice.Delta.Content[len(lastResponseText):]
+							}
 						} else {
 							// 如果新内容不以旧内容开头，可能是并发情况下的新消息，直接使用新内容
 							newContent += choice.Delta.Content

diff --git a/applogic/rwkv.go b/applogic/rwkv.go
@@ -365,9 +365,15 @@ func (app *App) ChatHandlerRwkv(w http.ResponseWriter, r *http.Request) {
 
 					newContent := ""
 					for _, choice := range eventData.Choices {
+						// 如果新内容以旧内容开头
 						if strings.HasPrefix(choice.Delta.Content, lastResponseText) {
-							// 如果新内容以旧内容开头，剔除旧内容部分，只保留新增的部分
-							newContent += choice.Delta.Content[len(lastResponseText):]
+							// 特殊情况：当新内容和旧内容完全相同时，处理逻辑应当与新内容不以旧内容开头时相同
+							if choice.Delta.Content == lastResponseText {
+								newContent += choice.Delta.Content
+							} else {
+								// 剔除旧内容部分，只保留新增的部分
+								newContent += choice.Delta.Content[len(lastResponseText):]
+							}
 						} else {
 							// 如果新内容不以旧内容开头，可能是并发情况下的新消息，直接使用新内容
 							newContent += choice.Delta.Content

diff --git a/utils/utils.go b/utils/utils.go
@@ -186,6 +186,9 @@ func SendGroupMessage(groupID int64, userID int64, message string, selfid string
 		message = acnode.CheckWordOUT(message)
 	}
 
+	// 去除末尾的换行符 不去除会导致不好看
+	message = removeTrailingCRLFs(message)
+
 	// 构造请求体
 	requestBody, err := json.Marshal(map[string]interface{}{
 		"group_id": groupID,
@@ -272,6 +275,9 @@ func SendPrivateMessage(UserID int64, message string, selfid string) error {
 		message = acnode.CheckWordOUT(message)
 	}
 
+	// 去除末尾的换行符 不去除会导致不好看
+	message = removeTrailingCRLFs(message)
+
 	// 构造请求体
 	requestBody, err := json.Marshal(map[string]interface{}{
 		"user_id": UserID,