fix: convert system message to user message for zhipu

This commit is contained in:
JustSong
2023-07-27 23:32:00 +08:00
parent 9033a0dcad
commit 17b9b9efad
+15 -4
View File
@@ -111,10 +111,21 @@ func getZhipuToken(apikey string) string {
func requestOpenAI2Zhipu(request GeneralOpenAIRequest) *ZhipuRequest { func requestOpenAI2Zhipu(request GeneralOpenAIRequest) *ZhipuRequest {
messages := make([]ZhipuMessage, 0, len(request.Messages)) messages := make([]ZhipuMessage, 0, len(request.Messages))
for _, message := range request.Messages { for _, message := range request.Messages {
messages = append(messages, ZhipuMessage{ if message.Role == "system" {
Role: message.Role, messages = append(messages, ZhipuMessage{
Content: message.Content, Role: "system",
}) Content: message.Content,
})
messages = append(messages, ZhipuMessage{
Role: "user",
Content: "Okay",
})
} else {
messages = append(messages, ZhipuMessage{
Role: message.Role,
Content: message.Content,
})
}
} }
return &ZhipuRequest{ return &ZhipuRequest{
Prompt: messages, Prompt: messages,