fix: add missing bos_token to example templates

.buildkite/test-pipeline.yaml

@@ -203,6 +203,14 @@ steps:
     - python3 offline_inference_scoring.py
     - python3 offline_profile.py --model facebook/opt-125m run_num_steps --num-steps 2
 
+- label: Examples Unit Test
+  working_dir: "/vllm-workspace/tests"
+  source_file_dependencies:
+  - tests/examples/
+  - examples/
+  commands:
+    - pytest -v -s examples
+
 - label: Prefix Caching Test # 9min
   mirror_hardwares: [amd]
   source_file_dependencies:

examples/template_alpaca.jinja

@@ -1,3 +1,4 @@
+{{ bos_token -}}
 {{ (messages|selectattr('role', 'equalto', 'system')|list|last).content|trim if (messages|selectattr('role', 'equalto', 'system')|list) else '' }}
 
 {% for message in messages %}

examples/template_baichuan.jinja

@@ -1,3 +1,4 @@
+{{ bos_token -}}
 {{ (messages|selectattr('role', 'equalto', 'system')|list|last).content|trim if (messages|selectattr('role', 'equalto', 'system')|list) else '' }}
 
 {%- for message in messages -%}

examples/template_blip2.jinja

@@ -1,3 +1,4 @@
+{{ bos_token }}
 {%- for message in messages -%}
     {%- if message['role'] == 'user' -%}
         {{- 'Question: ' + message['content'] + ' ' -}}

examples/template_chatglm.jinja

@@ -1,3 +1,4 @@
+{{- bos_token -}}
 {%- set counter = namespace(index=0) -%}
 {%- for message in messages -%}
     {%- if message['role'] == 'user' -%}

examples/template_chatglm2.jinja

@@ -1,3 +1,4 @@
+{{- bos_token -}}
 {%- set counter = namespace(index=1) -%}
 {%- for message in messages -%}
     {%- if message['role'] == 'user' -%}

examples/template_chatml.jinja

@@ -1,2 +1,3 @@
+{{ bos_token -}}
 {% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content']}}{% if (loop.last and add_generation_prompt) or not loop.last %}{{ '<|im_end|>' + '\n'}}{% endif %}{% endfor %}
 {% if add_generation_prompt and messages[-1]['role'] != 'assistant' %}{{ '<|im_start|>assistant\n' }}{% endif %}

examples/template_dse_qwen2_vl.jinja

@@ -1,3 +1,4 @@
+{{ bos_token -}}
 {% set image_count = namespace(value=0) %}{% set video_count = namespace(value=0) %}{% for message in messages %}{% if loop.first and message['role'] != 'system' %}{% raw %}<|im_start|>system
 You are a helpful assistant.<|im_end|>
 {% endraw %}{% endif %}<|im_start|>{{ message['role'] }}{% raw %}

examples/template_falcon.jinja

@@ -1,3 +1,4 @@
+{{- bos_token -}}
 {%- for message in messages -%}
     {%- if message['role'] == 'user' -%}
         {{- 'User: ' + message['content'] -}}

examples/template_falcon_180b.jinja

@@ -1,3 +1,4 @@
+{{- bos_token -}}
 {%- for message in messages -%}
     {%- if message['role'] == 'system' -%}
         {{- 'System: ' + message['content'] -}}

examples/template_inkbot.jinja

@@ -1,3 +1,4 @@
+{{ bos_token -}}
 <#meta#>
 - Date: {{ (messages|selectattr('role', 'equalto', 'meta-current_date')|list|last).content|trim if (messages|selectattr('role', 'equalto', 'meta-current_date')|list) else '' }}
 - Task: {{ (messages|selectattr('role', 'equalto', 'meta-task_name')|list|last).content|trim if (messages|selectattr('role', 'equalto', 'meta-task_name')|list) else '' }}

examples/template_vlm2vec.jinja

@@ -1,3 +1,4 @@
+{{- bos_token -}}
 {%- if messages | length > 1 -%}
     {{ raise_exception('Embedding models should only embed one message at a time') }}
 {%- endif -%}

examples/tool_chat_template_granite.jinja

@@ -1,3 +1,4 @@
+{{- bos_token }}
 {%- if tools %}
     {{- '<|start_of_role|>available_tools<|end_of_role|>
 ' }}

examples/tool_chat_template_granite_20b_fc.jinja

@@ -1,3 +1,4 @@
+{{- bos_token }}
 {%- macro json_to_python_type(json_spec) %}
     {%- set basic_type_map = {
     "string": "str",

tests/examples/test_jinja.py

@@ -0,0 +1,48 @@
+from pathlib import Path
+
+import pytest
+import transformers
+
+jinja_paths = [
+    pytest.param(path, id=path.stem)
+    for path in sorted((Path(__name__).parent.parent /
+                        "examples").glob("*.jinja"))
+]
+
+
+@pytest.mark.parametrize("path", jinja_paths)
+@pytest.mark.parametrize("num_messages", [1, 3])
+def test_bos(path: Path, num_messages: int) -> None:
+    with path.open("r", encoding="utf-8") as f:
+        chat_template = f.read()
+    # We might guess an appropriate tokenizer model from the file name but we
+    # don't maintain such list.
+    # Use arbitrary BOS for testing. It doesn't have to match the str in the
+    # correct tokenizer.
+    bos_token = "=BOS="
+    tokenizer = transformers.PreTrainedTokenizerBase(
+        chat_template=chat_template, bos_token=bos_token, eos_token="=EOS=")
+    conversation = [
+        {
+            "role": "user",
+            "content": "1"
+        },
+        {
+            "role": "assistant",
+            "content": "2"
+        },
+        {
+            "role": "user",
+            "content": "3"
+        },
+    ][:num_messages]
+    try:
+        prompt: str = tokenizer.apply_chat_template(conversation=conversation,
+                                                    tokenize=False)
+    except Exception as e:
+        if str(e
+               ) == "Embedding models should only embed one message at a time":
+            pytest.skip(reason=str(e))
+        raise
+    assert prompt.startswith(bos_token)
+    assert prompt.count(bos_token) == 1