File size: 1,828 Bytes
ed8a2e0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
{% for message in messages if message.role == 'user' and message.content is iterable and message.content is not string %}
    {% for item in message.content if item.type == 'image' %}
        {{- '<image>' -}}
    {% endfor %}
{% endfor %}

{{- '<|begin_of_sentence|>' -}}

{%- set system_message = namespace(value=none) -%}
{%- for message in messages if message.role == 'system' -%}
    {%- set system_message.value = message.content -%}
{%- endfor -%}
{%- if system_message.value -%}
    {{- system_message.value -}}
{%- endif -%}

{%- for message in messages -%}
    {%- if message.role == "user" -%}
        {{- '<|User|>' -}}
        {%- if message.content is string -%}
            {{- message.content -}}
        {%- elif message.content is iterable and message.content is not string -%}
            {%- for item in message.content if item.type == "text" -%}
                {{- item.text -}}
            {%- endfor -%}
        {%- endif -%}

    {%- elif message.role == "assistant" -%}
        {%- set thinking_tag = "" -%}
        {%- if enable_thinking is defined -%}
            {%- set thinking_tag = "</think>" if not enable_thinking else "<think>" -%}
        {%- endif -%}
        {{- '<|Assistant|>' + thinking_tag -}}

        {%- if message.content is string -%}
            {{- message.content -}}
        {%- elif message.content is iterable and message.content is not string -%}
            {%- for item in message.content if item.type == "text" -%}
                {{- item.text -}}
            {%- endfor -%}
        {%- endif -%}

        {{- '<|end_of_sentence|>' -}}
    {%- endif -%}
{%- endfor -%}

{%- if add_generation_prompt -%}
    {{- '<|Assistant|>' -}}
    {%- if enable_thinking is defined -%}
        {{- "</think>" if not enable_thinking else "<think>" -}}
    {%- endif -%}
{%- endif -%}