Update chat_template.jinja

chat_template.jinja

@@ -26,12 +26,14 @@
         {{- '<|im_start|>system\n' + messages[0].content + '<|im_end|>\n' }}
     {%- endif %}
 {%- endif %}
+
 {%- set ns = namespace(last_user_index=-1) %}
 {%- for m in messages %}
     {%- if m.role == 'user' %}
         {%- set ns.last_user_index = loop.index0 %}
     {%- endif %}
 {%- endfor %}
+
 {%- for message in messages %}
     {%- set content = message.get('content', '') or '' %}
     {%- if (message.role == 'system' and not loop.first) %}
@@ -45,28 +47,17 @@
                 {%- if not loop.first %}
                     {{- '\n' }}
                 {%- endif %}
-                {%- if content['type'] == 'image_url' %}
-                    {%- set media_url = content.get('image_url', {}).get('url', '') %}
-                    {%- set url_lower = media_url.lower() %}
-                    {%- set video_extensions = ['.mp4', '.avi', '.mov', '.mkv', '.webm', '.flv', '.wmv', '.m4v'] %}
-                    {%- set ns_check = namespace(is_video=False) %}
-                    {%- for ext in video_extensions %}
-                        {%- if url_lower.endswith(ext) %}
-                            {%- set ns_check.is_video = True %}
-                        {%- endif %}
-                    {%- endfor %}
-                    {%- if ns_check.is_video %}
-                        {%- set video_id = 'video_%02d' % ns_vid.count %}
-                        {%- set ns_vid.count = ns_vid.count + 1 %}
-                        {{- '<|mime_start|>{"id": "' + video_id + '", "type": "video/mp4", "filename": "video.mp4"}<|mime_end|>\n' }}
-                        {{- '<|video_aux_start|>다음 중 video_duration은 비디오 길이 정보입니다. 참고하여 답변하세요. {"video_duration": "<|video_meta_duration|>"}<|video_aux_end|>\n'}}
-                        {{- '<|video_start|><|VIDEO_PAD|><|video_end|>\n'}}
-                    {%- else %}
-                        {%- set image_id = 'image_%02d' % ns_img.count %}
-                        {%- set ns_img.count = ns_img.count + 1 %}
-                        {{- '<|mime_start|>{"id": "' + image_id + '", "type": "image/jpeg", "filename": "image.jpg"}<|mime_end|>\n' }}
-                        {{- '<|image_start|><|IMAGE_PAD|><|image_end|>' }}
-                    {%- endif %}
+                {%- if content['type'] in ['video_url', 'video'] %}
+                    {%- set video_id = 'video_%02d' % ns_vid.count %}
+                    {%- set ns_vid.count = ns_vid.count + 1 %}
+                    {{- '<|mime_start|>{"id": "' + video_id + '", "type": "video/mp4", "filename": "video.mp4"}<|mime_end|>\n' }}
+                    {{- '<|video_aux_start|>다음 중 video_duration은 비디오 길이 정보입니다. 참고하여 답변하세요. {"video_duration": "<|video_meta_duration|>"}<|video_aux_end|>\n'}}
+                    {{- '<|video_start|><|VIDEO_PAD|><|video_end|>' }}
+                {%- elif content['type'] in ['image_url', 'image'] %}
+                    {%- set image_id = 'image_%02d' % ns_img.count %}
+                    {%- set ns_img.count = ns_img.count + 1 %}
+                    {{- '<|mime_start|>{"id": "' + image_id + '", "type": "image/jpeg", "filename": "image.jpg"}<|mime_end|>\n' }}
+                    {{- '<|image_start|><|IMAGE_PAD|><|image_end|>' }}
                 {%- elif content['type'] == 'text' %}
                     {{- content['text'] }}
                 {%- endif %}
@@ -132,4 +123,4 @@
     {%- else %}
         {{- '<|im_start|>assistant\n<think>\n\n</think>\n\n' }}
     {%- endif %}
-{%- endif %}
+{%- endif %}