From 1edd0961aed75139e6667b9ef4cd98d416308a45 Mon Sep 17 00:00:00 2001
From: Zhang Minghan <zmh13054618081@dingtalk.com>
Date: Thu, 10 Aug 2023 23:34:36 +0800
Subject: [PATCH] update web empty response feature

---
 api/utils.go | 15 ++-------------
 1 file changed, 2 insertions(+), 13 deletions(-)

diff --git a/api/utils.go b/api/utils.go
index 62740ca..f3d96ff 100644
--- a/api/utils.go
+++ b/api/utils.go
@@ -29,28 +29,17 @@ func GetRandomKey(apikey string) string {
 }
 
 func StringCleaner(content string) string {
-	for _, replacer := range []string{",", "、", "，", "。", "：", ":", "；", ";", "！", "!", "？", "?", "（", "）", "(", ")", "关键字"} {
+	for _, replacer := range []string{",", "、", "，", "。", "：", ":", "；", ";", "！", "!", "？", "?", "（", "）", "(", ")", "关键字", "空"} {
 		content = strings.ReplaceAll(content, replacer, " ")
 	}
 	return strings.TrimSpace(content)
 }
 
 func SearchWeb(message []types.ChatGPTMessage) string {
-	source := make([]string, 0)
-	for _, item := range message {
-		if item.Role == "user" && item.Content != "" {
-			source = append(source, item.Content)
-		}
-	}
-	if len(source) == 0 {
-		return ""
-	}
-
-	source = utils.GetLatestSegment(source, 3)
 	keyword, _ := GetChatGPTResponse([]types.ChatGPTMessage{{
 		Role: "user",
 		Content: fmt.Sprintf("你是一个AI助手，我将你用来总结用户输入的内容并输出到bing搜索引擎上，"+
-			"请总结关键字，不要输出其他内容，不能输出特殊字符：\n%s", strings.Join(source, " ")),
+			"请总结关键字，不要输出其他内容，不能输出特殊字符（如果不需要搜索，请输出空）：\n%s", message[len(message)-1].Content),
 	}}, 40)
 
 	return StringCleaner(keyword)