mirror of
https://github.com/coaidev/coai.git
synced 2025-05-20 05:20:15 +09:00
60 lines
1.7 KiB
Go
60 lines
1.7 KiB
Go
package api
|
||
|
||
import (
|
||
"chat/types"
|
||
"chat/utils"
|
||
"chat/web"
|
||
"fmt"
|
||
"math/rand"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
func ChatWithWeb(message []types.ChatGPTMessage) (string, []types.ChatGPTMessage) {
|
||
keyword := SearchWeb(message)
|
||
return keyword, utils.Insert(message, 0, types.ChatGPTMessage{
|
||
Role: "system",
|
||
Content: fmt.Sprintf("你将扮演AI问答助手，你的知识库不是截止于2021年9月，而是可以实时联网的，你可以提供实时联网的知识库。"+
|
||
"官网网站使用链接包裹，给予用户精确的答复。"+
|
||
"当前时间: %s, 你的知识库：%s",
|
||
time.Now().Format("2006-01-02 15:04:05"), web.SearchBing(keyword),
|
||
),
|
||
})
|
||
}
|
||
|
||
func GetRandomKey(apikey string) string {
|
||
arr := strings.Split(apikey, "|")
|
||
idx := rand.Intn(len(arr))
|
||
return arr[idx]
|
||
}
|
||
|
||
func StringCleaner(content string) string {
|
||
for _, replacer := range []string{",", "、", "，", "。", "：", ":", "；", ";", "！", "!", "？", "?", "（", "）", "(", ")"} {
|
||
content = strings.ReplaceAll(content, replacer, " ")
|
||
}
|
||
return 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 ""
|
||
}
|
||
|
||
keyword, _ := GetChatGPTResponse([]types.ChatGPTMessage{{
|
||
Role: "system",
|
||
Content: "在接下来的对话中，不要回答问题，你需要总结接下来的对话中出现的内容，" +
|
||
"然后用关键字和空格分词，输出精简，请不要输出冗余内容，" +
|
||
"不能出现逗号顿号等特殊字符，仅回答出现的关键词。",
|
||
}, {
|
||
Role: "user",
|
||
Content: strings.Join(source, " "),
|
||
}}, 40)
|
||
return StringCleaner(keyword)
|
||
}
|