Set role and content from transformers

Signed-off-by: Ettore Di Giacinto <[email protected]>
mudler · Dec 7, 2024 · 4e8fed4 · 4e8fed4
1 parent d611d16
commit 4e8fed4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/templates/evaluator.go b/pkg/templates/evaluator.go
@@ -104,9 +104,9 @@ func (e *Evaluator) templateJinjaChat(templateName string, messageData []ChatMes
 		// TODO: this is not correct, we have to map jinja tokenizer template from transformers to our own
 
 		messages = append(messages, map[string]interface{}{
-			"Role":         message.Role,
-			"RoleName":     message.RoleName,
-			"Content":      message.Content,
+			//"role":         message.Role,
+			"role":         message.RoleName,
+			"content":      message.Content,
 			"FunctionCall": message.FunctionCall,
 			"FunctionName": message.FunctionName,
 			"LastMessage":  message.LastMessage,