coai/addition/web/utils.go
2023-12-08 10:15:45 +08:00

25 lines
460 B
Go

package web
import (
"chat/globals"
"chat/manager/conversation"
)
func UsingWebSegment(instance *conversation.Conversation) []globals.Message {
segment := conversation.CopyMessage(instance.GetChatMessage())
if instance.IsEnableWeb() {
segment = ChatWithWeb(segment)
}
return segment
}
func UsingWebNativeSegment(enable bool, message []globals.Message) []globals.Message {
if enable {
return ChatWithWeb(message)
} else {
return message
}
}