tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

include tool in default message_property_mappings

Browse Source
This commit is contained in:
Wing Lian
2025-10-23 10:25:18 -07:00
parent 77828d3559
commit f94ec0434c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/prompt_strategies/chat_template.py
View File

@@ -48,9 +48,9 @@ class ChatTemplatePrompter(Prompter):
): ):
# check if message_property_mappings is None or empty dict # check if message_property_mappings is None or empty dict
if message_property_mappings is None or (not message_property_mappings): if message_property_mappings is None or (not message_property_mappings):
default_message_property_mappings_keys = ["role", "content", "tool"]
message_property_mappings = { message_property_mappings = {
"role": "role", prop: prop for prop in default_message_property_mappings_keys
"content": "content",
} }
if template_thinking_key and field_thinking: if template_thinking_key and field_thinking:
message_property_mappings[template_thinking_key] = field_thinking message_property_mappings[template_thinking_key] = field_thinking