update web searching feature

This commit is contained in:
Zhang Minghan 2023-11-01 09:34:22 +08:00
parent 86fa6da6b4
commit f174798d77

View File

@ -12,12 +12,8 @@ type Hook func(message []globals.Message, token int) (string, error)
func ChatWithWeb(hook Hook, message []globals.Message, long bool) (string, []globals.Message) {
keyword := strings.TrimSpace(GetKeywordPoint(hook, message))
data := SearchBing(GetPointByLatestMessage(message))
if len(keyword) == 0 {
return keyword, message
}
data := SearchBing(keyword)
if long {
data = utils.GetSegmentString(data, 6000)
} else {
@ -25,9 +21,7 @@ func ChatWithWeb(hook Hook, message []globals.Message, long bool) (string, []glo
}
return keyword, utils.Insert(message, 0, globals.Message{
Role: "system",
Content: fmt.Sprintf("你将扮演AI问答助手你的知识库不是截止于2021年9月而是可以实时联网的你可以提供实时联网的知识库。"+
//"网站使用markdown链接包裹多使用emoji和颜文字给予用户精确的答复。"+
//"公式使用 $$ [公式] $$ 包裹使用latex语法例如$$ \\frac{1}{2} $$。"+
Content: fmt.Sprintf("你将扮演AI问答助手你的知识库不是离线的而是可以实时联网的你可以提供实时联网的信息。"+
"当前时间: %s, 实时联网搜索结果:%s",
time.Now().Format("2006-01-02 15:04:05"), data,
),
@ -79,3 +73,7 @@ func GetKeywordPoint(hook Hook, message []globals.Message) string {
}
return StringCleaner(keyword["keyword"].(string))
}
func GetPointByLatestMessage(message []globals.Message) string {
return StringCleaner(message[len(message)-1].Content)
}