mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-19 08:56:37 +08:00
feat: 自动整理claude不规范prompt
This commit is contained in:
parent
53e8790024
commit
1a39ef74ce
@ -70,8 +70,39 @@ func RequestOpenAI2ClaudeMessage(textRequest dto.GeneralOpenAIRequest) (*ClaudeR
|
|||||||
if claudeRequest.MaxTokens == 0 {
|
if claudeRequest.MaxTokens == 0 {
|
||||||
claudeRequest.MaxTokens = 4096
|
claudeRequest.MaxTokens = 4096
|
||||||
}
|
}
|
||||||
|
formatMessages := make([]dto.Message, 0)
|
||||||
|
var lastMessage *dto.Message
|
||||||
|
for i, message := range textRequest.Messages {
|
||||||
|
if message.Role == "system" {
|
||||||
|
if i != 0 {
|
||||||
|
message.Role = "user"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if message.Role == "" {
|
||||||
|
message.Role = "user"
|
||||||
|
}
|
||||||
|
fmtMessage := dto.Message{
|
||||||
|
Role: message.Role,
|
||||||
|
Content: message.Content,
|
||||||
|
}
|
||||||
|
if lastMessage != nil && lastMessage.Role == message.Role {
|
||||||
|
if lastMessage.IsStringContent() && message.IsStringContent() {
|
||||||
|
content, _ := json.Marshal(strings.Trim(fmt.Sprintf("%s %s", lastMessage.StringContent(), message.StringContent()), "\""))
|
||||||
|
fmtMessage.Content = content
|
||||||
|
// delete last message
|
||||||
|
formatMessages = formatMessages[:len(formatMessages)-1]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if fmtMessage.Content == nil {
|
||||||
|
content, _ := json.Marshal("...")
|
||||||
|
fmtMessage.Content = content
|
||||||
|
}
|
||||||
|
formatMessages = append(formatMessages, fmtMessage)
|
||||||
|
lastMessage = &message
|
||||||
|
}
|
||||||
|
|
||||||
claudeMessages := make([]ClaudeMessage, 0)
|
claudeMessages := make([]ClaudeMessage, 0)
|
||||||
for _, message := range textRequest.Messages {
|
for _, message := range formatMessages {
|
||||||
if message.Role == "system" {
|
if message.Role == "system" {
|
||||||
claudeRequest.System = message.StringContent()
|
claudeRequest.System = message.StringContent()
|
||||||
} else {
|
} else {
|
||||||
|
Loading…
Reference in New Issue
Block a user