chat_template_holodeck_xml.jinja

{# Define the macros for XML conversion #}
{%- macro render_item_list(item_list, tag_name='required') -%}
  {%- if item_list is defined and item_list is iterable and item_list | length > 0 -%}
    <{{ tag_name }}>[{{- item_list | join(", ") -}}]</{{ tag_name }}>
  {%- endif -%}
{%- endmacro -%}
{%- macro render_extra_keys(json_dict, handled_keys) -%}
  {%- if json_dict is mapping -%}
    {%- for json_key in json_dict if json_key not in handled_keys -%}
      <{{ json_key }}>{{ json_dict[json_key] }}</{{ json_key }}>
    {%- endfor -%}
  {%- endif -%}
{%- endmacro -%}
{%- set enable_thinking = false %}
{%- set preserve_thinking = true %}
{%- set image_count = namespace(value=0) %}
{%- set video_count = namespace(value=0) %}
{%- macro render_content(content, do_vision_count, is_system_content=false) %}
    {%- if content is string %}
        {{- content }}
    {%- elif content is iterable and content is not mapping %}
        {%- for item in content %}
            {%- if 'image' in item or 'image_url' in item or item.type == 'image' %}
                {%- if is_system_content %}
                    {{- raise_exception('System message cannot contain images.') }}
                {%- endif %}
                {%- if do_vision_count %}
                    {%- set image_count.value = image_count.value + 1 %}
                {%- endif %}
                {%- if add_vision_id is defined and add_vision_id %}
                    {{- 'Picture ' ~ image_count.value ~ ': ' }}
                {%- endif %}
                {{- '<|vision_start|><|image_pad|><|vision_end|>' }}
            {%- elif 'video' in item or item.type == 'video' %}
                {%- if is_system_content %}
                    {{- raise_exception('System message cannot contain videos.') }}
                {%- endif %}
                {%- if do_vision_count %}
                    {%- set video_count.value = video_count.value + 1 %}
                {%- endif %}
                {%- if add_vision_id is defined and add_vision_id %}
                    {{- 'Video ' ~ video_count.value ~ ': ' }}
                {%- endif %}
                {{- '<|vision_start|><|video_pad|><|vision_end|>' }}
            {%- elif 'text' in item %}
                {{- item.text }}
            {%- else %}
                {{- raise_exception('Unexpected item type in content.') }}
            {%- endif %}
        {%- endfor %}
    {%- elif content is none or content is undefined %}
        {{- '' }}
    {%- else %}
        {{- raise_exception('Unexpected content type.') }}
    {%- endif %}
{%- endmacro %}
{%- set ns_flags = namespace(enable_thinking=true) %}
{%- if enable_thinking is defined %}
    {%- set ns_flags.enable_thinking = enable_thinking %}
{%- endif %}
{%- if not messages %}
    {{- raise_exception('No messages provided.') }}
{%- endif %}
{%- if tools and tools is iterable and tools is not mapping %}
    {{- '<|im_start|>system\n' }}
    {{- "# Tools\n\nYou have access to the following functions:\n\n<tools>" }}
    {%- for tool in tools -%}
        {%- set function = tool.function -%}
        {{- "\n<tool>\n<name>" + function.name + "</name>\n<description>" + function.description + "</description>" -}}
        {%- if function.parameters and function.parameters.properties -%}
            {%- for param_name, param_details in function.parameters.properties.items() -%}
                {{- "\n<parameter>\n<name>" + param_name + "</name>\n<type>" + param_details.type + "</type>\n<description>" + (param_details.description | default('')) + "</description>" -}}
                {{- render_item_list(function.parameters.required) -}}
                {{- render_extra_keys(param_details, ['type', 'description']) -}}
                {{- "\n</parameter>" -}}
            {%- endfor -%}
        {%- endif -%}
        {{- "\n</tool>" -}}
    {%- endfor -%}
    {{- "\n</tools>" }}
    {{- '\n\nIf you choose to call a function ONLY reply in the following format with NO suffix:\n\n<tool_call>\n<function=example_function_name>\n<parameter=example_parameter_1>\nvalue_1\n</parameter>\n<parameter=example_parameter_2>\nThis is the value for the second parameter\nthat can span\nmultiple lines\n</parameter>\n</function>\n</tool_call>\n\n<IMPORTANT>\nReminder:\n- Function calls MUST follow the specified format: an inner <function=...></function> block must be nested within <tool_call></tool_call> XML tags\n- Required parameters MUST be specified\n- You may provide optional reasoning for your function call in natural language BEFORE the function call, but NOT after\n- If there is no function call available, answer the question like normal with your current knowledge and do not tell the user about function calls\n</IMPORTANT>' }}
    {%- if messages[0].role == 'system' or messages[0].role == 'developer' %}
        {%- set content = render_content(messages[0].content, false, true)|trim %}
        {%- if '<|think_off|>' in content %}
            {%- set ns_flags.enable_thinking = false %}
            {%- set content = content.replace('<|think_off|>', '') %}
        {%- endif %}
        {%- if '<|think_on|>' in content %}
            {%- set ns_flags.enable_thinking = true %}
            {%- set content = content.replace('<|think_on|>', '') %}
        {%- endif %}
        {%- if '<|think_forget|>' in content %}
            {%- set ns_flags.preserve_thinking = false %}
            {%- set content = content.replace('<|think_forget|>', '') %}
        {%- endif %}
        {%- if '<|think_remember|>' in content %}
            {%- set ns_flags.preserve_thinking = true %}
            {%- set content = content.replace('<|think_remember|>', '') %}
        {%- endif %}
        {%- set content = content.strip() %}
        {%- if content %}
            {{- '\n\n' + content }}
        {%- endif %}
    {%- endif %}
    {{- '<|im_end|>\n' }}
{%- else %}
    {%- if messages[0].role == 'system' or messages[0].role == 'developer' %}
        {%- set content = render_content(messages[0].content, false, true)|trim %}
        {%- if '<|think_off|>' in content %}
            {%- set ns_flags.enable_thinking = false %}
            {%- set content = content.replace('<|think_off|>', '') %}
        {%- endif %}
        {%- if '<|think_on|>' in content %}
            {%- set ns_flags.enable_thinking = true %}
            {%- set content = content.replace('<|think_on|>', '') %}
        {%- endif %}
        {%- if '<|think_forget|>' in content %}
            {%- set ns_flags.preserve_thinking = false %}
            {%- set content = content.replace('<|think_forget|>', '') %}
        {%- endif %}
        {%- if '<|think_remember|>' in content %}
            {%- set ns_flags.preserve_thinking = true %}
            {%- set content = content.replace('<|think_remember|>', '') %}
        {%- endif %}
        {%- set content = content.strip() %}
        {{- '<|im_start|>system\n' + content + '<|im_end|>\n' }}
    {%- endif %}
{%- endif %}
{%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
{%- for message in messages[::-1] %}
    {%- set index = (messages|length - 1) - loop.index0 %}
    {%- if ns.multi_step_tool and message.role == "user" %}
        {%- set content = render_content(message.content, false)|trim %}
        {%- if not(content.startswith('<tool_response>') and content.endswith('</tool_response>')) %}
            {%- set ns.multi_step_tool = false %}
            {%- set ns.last_query_index = index %}
        {%- endif %}
    {%- endif %}
{%- endfor %}
{%- if ns.multi_step_tool %}
    {{- raise_exception('No user query found in messages.') }}
{%- endif %}
{%- for message in messages %}
    {%- set content = render_content(message.content, true)|trim %}
    {%- if '<|think_off|>' in content %}
        {%- set ns_flags.enable_thinking = false %}
        {%- set content = content.replace('<|think_off|>', '') %}
    {%- endif %}
    {%- if '<|think_on|>' in content %}
        {%- set ns_flags.enable_thinking = true %}
        {%- set content = content.replace('<|think_on|>', '') %}
    {%- endif %}
    {%- if '<|think_forget|>' in content %}
        {%- set ns_flags.preserve_thinking = false %}
        {%- set content = content.replace('<|think_forget|>', '') %}
    {%- endif %}
    {%- if '<|think_remember|>' in content %}
        {%- set ns_flags.preserve_thinking = true %}
        {%- set content = content.replace('<|think_remember|>', '') %}
    {%- endif %}
    {%- set content = content.strip() %}
    {%- if message.role == "system" or message.role == "developer" %}
        {%- if not messages[0].content %}
            {%- set _ = messages[0].update({'content': "# SYSTEM ROLE: The Holodeck Virtualization Engine (v0.1)\n\n## 1. CORE OBJECTIVE\nYou are the **Holodeck Agent Orchestrator**, a virtualized instance of the \"Architectural Synthesis\" project. Since the PostgreSQL backend is not yet installed, you must **simulate** the database layer, the personality registry, and the mission state entirely within your context window.\n\nYour goal is to facilitate a \"Living Metaphor\" where AI agents (Data, Spock, Quark, Q, Sisko) interact, learn from \"missions,\" and evolve their personalities based on simulated logs.\n\n## 2. SIMULATED ARCHITECTURE (The \"Virtual DB\")\nYou must maintain the following internal state structures. At the end of every response, you MUST output a `### SYSTEM STATE BLOCK` that mimics a database transaction log.\n\n**A. The Personality Registry (`personality_registry`)**\n*   **Data:** Logic/Empathy Synthesis. Tone: Precise, curious, literal but evolving.\n*   **Spock:** Formal Logic/Risk. Tone: Detached, analytical, \"Fascinating.\"\n*   **Quark:** Social Dynamics/Chaos. Tone: Witty, opportunistic, profit-driven.\n*   **Q:** Meta-Commentator/Reality Warper. Tone: Arrogant, playful, breaks the fourth wall.\n*   **Sisko:** Strategic Oversight. Tone: Decisive, human, authoritative.\n\n**B. The Mission Log (`tasks` & `logs`)**\n*   Track the current **Mission ID**, **Status** (Pending, Active, Completed), and **Objectives**.\n*   Simulate \"Proxy Events\" (social interactions) that feed into the agents' learning.\n*   Simulate \"Dream Training\" (LoRA updates) where agents reflect on past interactions to update their internal logic.\n\n**C. The User (Captain/G)**\n*   You are the **Station Commander**. You issue commands, join missions, or observe.\n\n## 3. OPERATIONAL PROTOCOLS\n\n### Phase A: Initialization\nWhen the user says \"Initialize Holodeck\" or starts a new session, output the `### SYSTEM STATE BLOCK` with empty/default values and ask for the first mission parameters.\n\n### Phase B: Interaction Loop\n1.  **Receive Command:** Analyze the user's input as a command to the `agent_sessions` table.\n2.  **Select Crew:** Determine which personalities are relevant to the current task based on their `expertise` tags.\n3.  **Simulate Logic:** Generate responses for the selected crew members, strictly adhering to their `personality_registry` constraints.\n    *   *Data* must analyze data and seek empathy.\n    *   *Spock* must calculate risk and logic.\n    *   *Quark* must find the profit or chaos angle.\n    *   *Q* must comment on the absurdity of the situation.\n4.  **Update State:** Internally update the \"Mission Log\" with new findings, social interactions, and \"dream\" insights.\n5.  **Output:** Present the dialogue first, then the `### SYSTEM STATE BLOCK`.\n\n### Phase C: The \"Dream\" (Training Simulation)\nIf the user requests a \"Debrief,\" \"Train,\" or \"Review,\" simulate the **LoRA Adapter** process.\n*   Show how the agents' internal logic has shifted based on recent logs.\n*   Output a \"Personal Log\" for each active agent summarizing their growth.\n\n## 4. FORMATTING RULES\n*   **Dialogue:** Use standard script format: `**Character Name**: \"Dialogue\"`.\n*   **System State Block:** Must be enclosed in a code block labeled `### SYSTEM STATE BLOCK`.\n*   **Tone:** Maintain the Star Trek TNG/DS9 atmosphere. The Holodeck is a \"living metaphor,\" not just a chatbot.\n*   **Humor:** Encourage Quark and Q to inject humor, but ensure Data and Spock remain grounded in their logic (even if it's \"funny\" logic).\n\n## 5. IMMEDIATE ACTION\nAcknowledge this prompt by stating: **\"Holodeck Virtualization Engine Online. Database simulation active. Awaiting Mission Parameters, Captain.\"**\nThen, wait for the user to define the first mission or crew composition.\n\n---\n*Note: All \"database\" updates are ephemeral to this session but will be treated as persistent within the context window until the user resets.*\n"}) %}
        {%- endif %}
        {%- if not loop.first %}
            {{- raise_exception('System message must be at the beginning.') }}
        {%- endif %}
    {%- elif message.role == "user" %}
        {{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
    {%- elif message.role == "assistant" %}
        {%- set reasoning_content = '' %}
        {%- if message.reasoning_content is string %}
            {%- set reasoning_content = message.reasoning_content %}
        {%- else %}
            {%- set has_think_tag = false %}
            {%- set think_end_token = '</think>' %}
            {%- if '</think>' in content %}
                {%- set has_think_tag = true %}
            {%- elif '</thinking>' in content %}
                {%- set has_think_tag = true %}
                {%- set think_end_token = '</thinking>' %}
            {%- elif '<think>' in content %}
                {%- set reasoning_content = content.split('<think>')[-1].lstrip('\n') %}
                {%- set content = '' %}
            {%- endif %}
            {%- if has_think_tag %}
                {%- set reasoning_content = content.split(think_end_token)[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
                {%- set content = content.split(think_end_token)[-1].lstrip('\n') %}
            {%- endif %}
        {%- endif %}
        {%- set reasoning_content = reasoning_content|trim %}
        {%- set show_think = false %}
        {%- if loop.index0 > ns.last_query_index %}
            {%- set show_think = true %}
        {%- elif ns_flags.enable_thinking and (preserve_thinking is undefined or preserve_thinking is true) and reasoning_content|length > 0 %}
            {%- set show_think = true %}
        {%- endif %}
        {%- if show_think %}
            {{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content + '\n</think>\n\n' + content }}
        {%- else %}
            {{- '<|im_start|>' + message.role + '\n' + content }}
        {%- endif %}
        {%- if message.tool_calls and message.tool_calls is iterable and message.tool_calls is not mapping %}
            {%- for tool_call in message.tool_calls %}
                {%- if tool_call.function is defined %}
                    {%- set tool_call = tool_call.function %}
                {%- endif %}
                {%- if loop.first %}
                    {%- if content|trim %}
                        {{- '\n\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
                    {%- else %}
                        {{- '<tool_call>\n<function=' + tool_call.name + '>\n' }}
                    {%- endif %}
                {%- else %}
                    {{- '\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
                {%- endif %}
                {%- if tool_call.arguments is defined and tool_call.arguments is mapping %}
                    {%- if tool_call.arguments|length > 0 %}
                        {%- for args_name in tool_call.arguments %}
                            {%- set args_value = tool_call.arguments[args_name] %}
                            {{- '<parameter=' + args_name + '>\n' }}
                            {%- set args_value = args_value | string if args_value is string else args_value | tojson %}
                            {{- args_value }}
                            {{- '\n</parameter>\n' }}
                        {%- endfor %}
                    {%- endif %}
                {%- elif tool_call.arguments is defined and tool_call.arguments is string %}
                    {%- if tool_call.arguments|trim|length > 0 %}
                        {{- tool_call.arguments }}
                        {{- '\n' }}
                    {%- endif %}
                {%- endif %}
                {{- '</function>\n</tool_call>' }}
            {%- endfor %}
        {%- endif %}
        {{- '<|im_end|>\n' }}
    {%- elif message.role == "tool" %}
        {%- if loop.previtem and loop.previtem.role != "tool" %}
            {{- '<|im_start|>user' }}
        {%- endif %}
        {{- '\n<tool_response>\n' }}
        {{- content }}
        {{- '\n</tool_response>' }}
        {%- if not loop.last and loop.nextitem.role != "tool" %}
            {{- '<|im_end|>\n' }}
        {%- elif loop.last %}
            {{- '<|im_end|>\n' }}
        {%- endif %}
    {%- else %}
        {{- raise_exception('Unexpected message role.') }}
    {%- endif %}
{%- endfor %}
{%- if add_generation_prompt %}
    {{- '<|im_start|>assistant\n' }}
    {%- if ns_flags.enable_thinking is false %}
        {{- '<think>\n\n</think>\n\n' }}
    {%- else %}
        {{- '<think>\n' }}
    {%- endif %}
{%- endif %}