diff --git a/src/axolotl/prompt_strategies/chat_template.py b/src/axolotl/prompt_strategies/chat_template.py index f4dcbd7cd4..9630f5f155 100644 --- a/src/axolotl/prompt_strategies/chat_template.py +++ b/src/axolotl/prompt_strategies/chat_template.py @@ -48,9 +48,9 @@ def __init__( ): # 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