fix: convert system message to user message for zhipu

This commit is contained in:
JustSong 2023-07-27 23:32:00 +08:00
parent b9f6461dd4
commit d1b6f492b6

View File

@ -111,11 +111,22 @@ 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 {
if message.Role == "system" {
messages = append(messages, ZhipuMessage{
Role: "system",
Content: message.Content,
})
messages = append(messages, ZhipuMessage{
Role: "user",
Content: "Okay",
})
} else {
messages = append(messages, ZhipuMessage{ messages = append(messages, ZhipuMessage{
Role: message.Role, Role: message.Role,
Content: message.Content, Content: message.Content,
}) })
} }
}
return &ZhipuRequest{ return &ZhipuRequest{
Prompt: messages, Prompt: messages,
Temperature: request.Temperature, Temperature: request.Temperature,