mirror of
https://github.com/woodchen-ink/Oapi-Feishu.git
synced 2025-07-18 05:42:08 +08:00
34 lines
831 B
Go
34 lines
831 B
Go
package chatgpt
|
||
|
||
import (
|
||
"errors"
|
||
"github.com/sashabaranov/go-openai"
|
||
)
|
||
|
||
const (
|
||
ChatMessageRoleSystem = "system"
|
||
ChatMessageRoleUser = "user"
|
||
ChatMessageRoleAssistant = "assistant"
|
||
)
|
||
|
||
func CheckChatCompletionMessages(messages []openai.ChatCompletionMessage) error {
|
||
hasSystemMsg := false
|
||
for _, msg := range messages {
|
||
if msg.Role != ChatMessageRoleSystem && msg.Role != ChatMessageRoleUser && msg.Role != ChatMessageRoleAssistant {
|
||
return errors.New("invalid message role")
|
||
}
|
||
if msg.Role == ChatMessageRoleSystem {
|
||
if hasSystemMsg {
|
||
return errors.New("more than one system message")
|
||
}
|
||
hasSystemMsg = true
|
||
} else {
|
||
// 对于非 system 角色的消息，Content 不能为空
|
||
if msg.Content == "" {
|
||
return errors.New("empty content in non-system message")
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
}
|