refactor: standardize user role in chat body messages

This commit is contained in:
Grail Finder (aider)
2025-03-10 22:05:48 +03:00
parent 686bb1fedd
commit 53e9dff387

13
llm.go
View File

@@ -246,8 +246,17 @@ func (ds DeepSeekerChat) FormMsg(msg, role string, resume bool) (io.Reader, erro
chatBody.Messages = append(chatBody.Messages, ragMsg) chatBody.Messages = append(chatBody.Messages, ragMsg)
} }
} }
// copy chat body and replace config.UserRole with "user"; ai! // Create copy of chat body with standardized user role
models.NewDSCharReq(chatBody) modifiedBody := *chatBody
modifiedBody.Messages = make([]models.RoleMsg, len(chatBody.Messages))
for i, msg := range chatBody.Messages {
if msg.Role == cfg.UserRole {
modifiedBody.Messages[i].Role = "user"
} else {
modifiedBody.Messages[i] = msg
}
}
models.NewDSCharReq(&modifiedBody)
data, err := json.Marshal(chatBody) data, err := json.Marshal(chatBody)
if err != nil { if err != nil {
logger.Error("failed to form a msg", "error", err) logger.Error("failed to form a msg", "error", err)