File size: 4,693 Bytes
6f7430a |
1 2 3 4 |
{
"chat_template": "{% set image_count = namespace(value=0) %}{% set video_count = namespace(value=0) %}{{- '<|im_start|>' + messages[0].role + '\n' }}{%- if messages[0]['content'] is string %}{{ messages[0]['content'] }}{%- else %}{%- for content in messages[0]['content'] %}{%- if content.type == 'image' or ('image' in content) or ('image_url' in content) %}{% set image_count.value = image_count.value + 1 %}{%- if add_vision_id %}Picture {{ image_count.value }}:{%- endif %}<|vision_start|><|image_pad|><|vision_end|>{%- elif content.type == 'video' or ('video' in content) %}{% set video_count.value = video_count.value + 1 %}{%- if add_vision_id %}Video {{ video_count.value }}:{%- endif %}<|vision_start|><|video_pad|><|vision_end|>{%- elif 'text' in content %}{{ content.text }}{%- endif %}{%- endfor %}{%- endif %}{%- if messages[0].tool_calls %}{%- for tool_call in messages[0].tool_calls %}{%- if tool_call.function is defined %}{%- set tool_call = tool_call.function %}{%- endif %}<tool_call>{\"name\": \"{{ tool_call.name }}\", \"arguments\": {{ tool_call.arguments | tojson }}} </tool_call>{%- endfor %}{%- endif %}{{- '<|im_end|>\n' }}{%- if tools %}{{- '<|im_start|>system\n' }}{{- \"\n\n# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>\" }}{%- for tool in tools %}{{- \"\n\" }}{{- tool | tojson }}{%- endfor %}{{- \"\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n\" }}<tool_call>{\"name\": <function-name>, \"arguments\": <args-json-object>}</tool_call>{{- '<|im_end|>\n' }}{%- endif %}{%- for message in messages[1:] %}{%- if (message.role == \"user\") or (message.role == \"system\") or (message.role == \"assistant\" and not message.tool_calls) %}{{- '<|im_start|>' + message.role + '\n' }}{%- if message.content is string %}{{ message.content }}{%- else %}{%- for content in message.content %}{%- if content.type == 'image' or ('image' in content) or ('image_url' in content) %}{% set image_count.value = image_count.value + 1 %}{%- if add_vision_id %}Picture {{ image_count.value }}:{%- endif %}<|vision_start|><|image_pad|><|vision_end|>{%- elif content.type == 'video' or ('video' in content) %}{% set video_count.value = video_count.value + 1 %}{%- if add_vision_id %}Video {{ video_count.value }}:{%- endif %}<|vision_start|><|video_pad|><|vision_end|>{%- elif 'text' in content %}{{ content.text }}{%- endif %}{%- endfor %}{%- endif %}{{- '<|im_end|>\n' }}{%- elif message.role == \"assistant\" %}{{- '<|im_start|>' + message.role }}{%- if message.content %}{% if message.content is string %}{{ '\n' + message.content }}{% else %}{%- for content in message.content %}{%- if content.type == 'image' or ('image' in content) or ('image_url' in content) %}{% set image_count.value = image_count.value + 1 %}{%- if add_vision_id %}Picture {{ image_count.value }}:{%- endif %}<|vision_start|><|image_pad|><|vision_end|>{%- elif content.type == 'video' or ('video' in content) %}{% set video_count.value = video_count.value + 1 %}{%- if add_vision_id %}Video {{ video_count.value }}:{%- endif %}<|vision_start|><|video_pad|><|vision_end|>{%- elif 'text' in content %}{{ content.text }}{%- endif %}{%- endfor %}{% endif %}{%- endif %}{%- for tool_call in message.tool_calls %}{%- if tool_call.function is defined %}{%- set tool_call = tool_call.function %}{%- endif %}<tool_call>{\"name\": \"{{ tool_call.name }}\", \"arguments\": {{ tool_call.arguments | tojson }}} </tool_call>{%- endfor %}{{- '<|im_end|>\n' }}{%- elif message.role == \"tool\" %}{%- if (loop.index0 == 0) or (messages[loop.index0].role != \"tool\") %}{{- '<|im_start|>user' }}{%- endif %}{{- '\n<tool_response>\n' }}{%- if message.content is string %}{{ message.content }}{%- else %}{%- for content in message.content %}{%- if content.type == 'image' or ('image' in content) or ('image_url' in content) %}{% set image_count.value = image_count.value + 1 %}{%- if add_vision_id %}Picture {{ image_count.value }}:{%- endif %}<|vision_start|><|image_pad|><|vision_end|>{%- elif content.type == 'video' or ('video' in content) %}{% set video_count.value = video_count.value + 1 %}{%- if add_vision_id %}Video {{ video_count.value }}:{%- endif %}<|vision_start|><|video_pad|><|vision_end|>{%- elif 'text' in content %}{{ content.text }}{%- endif %}{%- endfor %}{%- endif %}{{- '\n</tool_response>' }}{%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}{{- '<|im_end|>\n' }}{%- endif %}{%- endif %}{%- endfor %}{%- if add_generation_prompt %}{{- '<|im_start|>assistant\n' }}{%- endif %}"
}
|