From 53e9dff38725b20d74339368de807f69fde45728 Mon Sep 17 00:00:00 2001 From: "Grail Finder (aider)" Date: Mon, 10 Mar 2025 22:05:48 +0300 Subject: refactor: standardize user role in chat body messages --- llm.go | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/llm.go b/llm.go index 2af1d4b..af8412d 100644 --- a/llm.go +++ b/llm.go @@ -246,8 +246,17 @@ func (ds DeepSeekerChat) FormMsg(msg, role string, resume bool) (io.Reader, erro chatBody.Messages = append(chatBody.Messages, ragMsg) } } - // copy chat body and replace config.UserRole with "user"; ai! - models.NewDSCharReq(chatBody) + // Create copy of chat body with standardized user role + 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) if err != nil { logger.Error("failed to form a msg", "error", err) -- cgit v1.2.3