mirror of
https://github.com/danielmiessler/fabric
synced 2024-11-08 07:11:06 +00:00
23 lines
713 B
Go
23 lines
713 B
Go
|
package common
|
||
|
|
|
||
|
|
// NormalizeMessages remove empty messages and ensure messages order user-assist-user
|
||
|
|
func NormalizeMessages(msgs []*Message, defaultUserMessage string) (ret []*Message) {
|
||
|
|
// Iterate over messages to enforce the odd position rule for user messages
|
||
|
|
fullMessageIndex := 0
|
||
|
|
for _, message := range msgs {
|
||
|
|
if message.Content == "" {
|
||
|
|
// Skip empty messages as the anthropic API doesn't accept them
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
// Ensure, that each odd position shall be a user message
|
||
|
|
if fullMessageIndex%2 == 0 && message.Role != "user" {
|
||
|
|
ret = append(ret, &Message{Role: "user", Content: defaultUserMessage})
|
||
|
|
fullMessageIndex++
|
||
|
|
}
|
||
|
|
ret = append(ret, message)
|
||
|
|
fullMessageIndex++
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|