mirror of
https://github.com/coaidev/coai.git
synced 2025-05-20 05:20:15 +09:00
40 lines
1.2 KiB
Go
40 lines
1.2 KiB
Go
package web
|
|
|
|
import (
|
|
"chat/adapter/chatgpt"
|
|
"chat/globals"
|
|
"chat/manager/conversation"
|
|
)
|
|
|
|
func UsingWebSegment(instance *conversation.Conversation) (string, []globals.Message) {
|
|
var keyword string
|
|
var segment []globals.Message
|
|
|
|
if instance.IsEnableWeb() {
|
|
keyword, segment = ChatWithWeb(func(message []globals.Message, token int) (string, error) {
|
|
return chatgpt.NewChatInstanceFromConfig("gpt3").CreateChatRequest(&chatgpt.ChatProps{
|
|
Model: globals.GPT3TurboInstruct,
|
|
Message: message,
|
|
Token: token,
|
|
})
|
|
}, conversation.CopyMessage(instance.GetMessageSegment(12)), globals.IsLongContextModel(instance.GetModel()))
|
|
} else {
|
|
segment = conversation.CopyMessage(instance.GetMessageSegment(12))
|
|
}
|
|
return keyword, segment
|
|
}
|
|
|
|
func UsingWebNativeSegment(enable bool, message []globals.Message) (string, []globals.Message) {
|
|
if enable {
|
|
return ChatWithWeb(func(message []globals.Message, token int) (string, error) {
|
|
return chatgpt.NewChatInstanceFromConfig("gpt3").CreateChatRequest(&chatgpt.ChatProps{
|
|
Model: globals.GPT3TurboInstruct,
|
|
Message: message,
|
|
Token: token,
|
|
})
|
|
}, message, false)
|
|
} else {
|
|
return "", message
|
|
}
|
|
}
|