mirror of
https://github.com/alibaba/higress.git
synced 2026-02-26 21:50:53 +08:00
47 lines
870 B
Go
47 lines
870 B
Go
package dashscope
|
|
|
|
var MessageStore ChatMessages
|
|
|
|
func init() {
|
|
MessageStore = make(ChatMessages, 0)
|
|
MessageStore.Clear() //清理和初始化
|
|
|
|
}
|
|
|
|
type ChatMessages []Message
|
|
|
|
// 枚举出角色
|
|
const (
|
|
RoleUser = "user"
|
|
RoleAssistant = "assistant"
|
|
RoleSystem = "system"
|
|
)
|
|
|
|
func (cm *ChatMessages) Clear() {
|
|
*cm = make([]Message, 0) //重新初始化
|
|
}
|
|
|
|
// 添加角色和对应的prompt
|
|
func (cm *ChatMessages) AddFor(msg string, role string) {
|
|
*cm = append(*cm, Message{
|
|
Role: role,
|
|
Content: msg,
|
|
})
|
|
}
|
|
|
|
// 添加Assistant角色的prompt
|
|
func (cm *ChatMessages) AddForAssistant(msg string) {
|
|
cm.AddFor(msg, RoleAssistant)
|
|
|
|
}
|
|
|
|
// 添加System角色的prompt
|
|
func (cm *ChatMessages) AddForSystem(msg string) {
|
|
cm.AddFor(msg, RoleSystem)
|
|
}
|
|
|
|
// 添加User角色的prompt
|
|
func (cm *ChatMessages) AddForUser(msg string) {
|
|
cm.AddFor(msg, RoleUser)
|
|
}
|