refactor: standardize user role in chat body messages
This commit is contained in:
13
llm.go
13
llm.go
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user