From f94ec0434c631dfa0d0fa838dda0b30665a8e7a7 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 23 Oct 2025 10:25:18 -0700 Subject: [PATCH] include tool in default message_property_mappings --- src/axolotl/prompt_strategies/chat_template.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/axolotl/prompt_strategies/chat_template.py b/src/axolotl/prompt_strategies/chat_template.py index 57d3bfdf2..b806ecdec 100644 --- a/src/axolotl/prompt_strategies/chat_template.py +++ b/src/axolotl/prompt_strategies/chat_template.py @@ -48,9 +48,9 @@ class ChatTemplatePrompter(Prompter): ): # check if message_property_mappings is None or empty dict if message_property_mappings is None or (not message_property_mappings): + default_message_property_mappings_keys = ["role", "content", "tool"] message_property_mappings = { - "role": "role", - "content": "content", + prop: prop for prop in default_message_property_mappings_keys } if template_thinking_key and field_thinking: message_property_mappings[template_thinking_key] = field_thinking