{%- macro format_parameters(properties, required) -%} {%- set standard_keys = ['description', 'type', 'properties', 'required', 'nullable'] -%} {%- set ns = namespace(found_first=false) -%} {%- for key, value in properties | dictsort -%} {%- if key not in standard_keys -%} {%- if ns.found_first %},{% endif -%} {%- set ns.found_first = true -%} {{- key }}:{description:{{ value['description'] }} {%- if value['type'] | upper == 'STRING' -%} {%- if value['enum'] -%} ,enum:{{ format_argument(value['enum']) }} {%- endif -%} {%- elif value['type'] | upper == 'OBJECT' -%} ,properties:{ {%- if value['properties'] is defined and value['properties'] is mapping -%} {{- format_parameters(value['properties'], value['required'] | default([])) -}} {%- elif value is mapping -%} {{- format_parameters(value, value['required'] | default([])) -}} {%- endif -%} } {%- if value['required'] -%} ,required:[ {%- for item in value['required'] | default([]) -%} {{- item -}} {%- if not loop.last %},{% endif -%} {%- endfor -%} ] {%- endif -%} {%- elif value['type'] | upper == 'ARRAY' -%} {%- if value['items'] is mapping and value['items'] -%} ,items:{ {%- set ns_items = namespace(found_first=false) -%} {%- for item_key, item_value in value['items'] | dictsort -%} {%- if item_value is not none -%} {%- if ns_items.found_first %},{% endif -%} {%- set ns_items.found_first = true -%} {%- if item_key == 'properties' -%} properties:{ {%- if item_value is mapping -%} {{- format_parameters(item_value, value['items']['required'] | default([])) -}} {%- endif -%} } {%- elif item_key == 'required' -%} required:[ {%- for req_item in item_value -%} {{- req_item -}} {%- if not loop.last %},{% endif -%} {%- endfor -%} ] {%- elif item_key == 'type' -%} {%- if item_value is string -%} type:{{ format_argument(item_value | upper) }} {%- else -%} type:{{ format_argument(item_value | map('upper') | list) }} {%- endif -%} {%- else -%} {{ item_key }}:{{ format_argument(item_value) }} {%- endif -%} {%- endif -%} {%- endfor -%} } {%- endif -%} {%- endif -%} ,type:{{ value['type'] | upper }}} {%- endif -%} {%- endfor -%} {%- endmacro -%} {% macro format_function_declaration(tool_data) -%} declaration:{{- tool_data['function']['name'] -}} {description:{{- tool_data['function']['description'] -}} {%- set params = tool_data['function']['parameters'] -%} {%- if params -%} ,parameters:{ {%- if params['properties'] -%} properties:{ {{- format_parameters(params['properties'], params['required']) -}} }, {%- endif -%} {%- if params['required'] -%} required:[ {%- for item in params['required'] -%} {{- item -}} {{- ',' if not loop.last -}} {%- endfor -%} ], {%- endif -%} {%- if params['type'] -%} type:{{- params['type'] | upper -}}} {%- endif -%} {%- endif -%} } {%- endmacro -%} {% macro format_argument(argument, escape_keys=True) -%} {%- if argument is string -%} {{- '' + argument + '' -}} {%- elif argument is boolean -%} {%- if argument -%} {{- 'true' -}} {%- else -%} {{- 'false' -}} {%- endif -%} {%- elif argument is mapping -%} {{- '{' -}} {%- set ns = namespace(found_first=false) -%} {%- for key, value in argument | dictsort -%} {%- if ns.found_first %},{% endif -%} {%- set ns.found_first = true -%} {%- if escape_keys -%} {{- '' + key + '' -}} {%- else -%} {{- key -}} {%- endif -%} :{{- format_argument(value, escape_keys=escape_keys) -}} {%- endfor -%} {{- '}' -}} {%- elif argument is sequence -%} {{- '[' -}} {%- for item in argument -%} {{- format_argument(item, escape_keys=escape_keys) -}} {%- if not loop.last %},{% endif -%} {%- endfor -%} {{- ']' -}} {%- else -%} {{- argument -}} {%- endif -%} {%- endmacro -%} {{ bos_token }} {%- set ns = namespace(prev_message_type=None) -%} {#- Tool Declarations -#} {%- set loop_messages = messages -%} {%- if tools or messages[0]['role'] == 'system' or messages[0]['role'] == 'developer' -%} {{- 'developer\n' -}} {%- if messages[0]['role'] == 'system' or messages[0]['role'] == 'developer' -%} {%- if messages[0]['content'] is string -%} {{- messages[0]['content'] | trim -}} {%- elif messages[0]['content'] is sequence -%} {%- for item in messages[0]['content'] -%} {%- if item['type'] == 'text' -%} {{- item['text'] | trim -}} {%- endif -%} {%- endfor -%} {%- endif -%} {%- set loop_messages = messages[1:] -%} {%- endif -%} {%- if tools -%} {%- for tool in tools %} {{- '' -}} {{- format_function_declaration(tool) | trim }} {{- '' -}} {%- endfor %} {%- endif -%} {{- '\n' }} {%- endif %} {#- Loop through messages. -#} {%- for message in loop_messages -%} {%- if (message['role'] == 'assistant') -%} {#- Rename "assistant" to "model". -#} {%- set role = "model" -%} {%- else -%} {%- set role = message['role'] -%} {%- endif -%} {%- if role != 'tool' -%} {%- if ns.prev_message_type != 'tool_response' -%} {{- '' + role + '\n' }} {%- endif -%} {%- set ns.prev_message_type = None -%} {%- if 'content' in message and message['content'] is not none -%} {%- if message['content'] is string -%} {{ message['content'] | trim }} {%- elif message['content'] is sequence -%} {%- for item in message['content'] -%} {%- if item['type'] == 'image' -%} {{ '' }} {%- elif item['type'] == 'text' -%} {{ item['text'] | trim }} {%- endif -%} {%- endfor -%} {%- else -%} {{ raise_exception("Invalid content type in user/assistant message") }} {%- endif -%} {%- set ns.prev_message_type = 'content' -%} {%- endif -%} {%- if 'tool_calls' in message and message['tool_calls'] and message['tool_calls'] is iterable -%} {#- Tool Calls -#} {%- for tool_call in message['tool_calls'] -%} {% set function = tool_call['function'] %} {{- 'call:' + function['name'] + '{' -}} {%- if 'arguments' in function -%} {%- if function['arguments'] is mapping -%} {%- set ns = namespace(found_first=false) -%} {%- for key, value in function['arguments'] | dictsort -%} {%- if ns.found_first %},{% endif -%} {%- set ns.found_first = true -%} {{- key -}}:{{- format_argument(value, escape_keys=False) -}} {%- endfor -%} {%- elif function['arguments'] is string -%} {# This handles string-JSON, just in case #} {{ function['arguments'] }} {%- endif %} {%- endif -%} {{- '}' -}} {%- endfor -%} {%- if loop.last -%} {{ '' }} {%- endif -%} {%- set ns.prev_message_type = 'tool_call' -%} {%- endif -%} {%- else -%} {#- Tool Responses -#} {%- if 'content' in message and message['content'] -%} {%- if message['content'] is mapping -%} {%- if 'name' in message['content'] and 'response' in message['content'] -%} {{ 'response:' + message['content']['name'] | trim + '{' }} {%- set response_ns = namespace(found_first=false) -%} {%- for key, value in message['content']['response'] | dictsort -%} {%- if response_ns.found_first %},{% endif -%} {%- set response_ns.found_first = true -%} {{- key -}}:{{- format_argument(value, escape_keys=False) -}} {%- endfor -%} {{- '}' -}} {%- elif 'name' in message -%} {{ 'response:' + message['name'] | trim + '{' }} {%- set response_ns = namespace(found_first=false) -%} {%- for key, value in message['content'] | dictsort -%} {%- if response_ns.found_first %},{% endif -%} {%- set response_ns.found_first = true -%} {{- key -}}:{{- format_argument(value, escape_keys=False) -}} {%- endfor -%} {{- '}' -}} {%- else -%} {{ raise_exception("Invalid tool response mapping: must contain 'name' and 'response' keys, or 'name' must be in the message.") }} {%- endif -%} {%- elif message['content'] is string -%} {%- if 'name' in message -%} {{ 'response:' + message['name'] | trim + '{value:' + format_argument(message['content'], escape_keys=False) + '}' }} {%- else -%} {{ raise_exception("Invalid tool response: 'name' must be provided.") }} {%- endif -%} {%- elif message['content'] is sequence -%} {%- for item in message['content'] -%} {%- if item is mapping -%} {%- if 'name' in item and 'response' in item -%} {{ 'response:' + item['name'] | trim + '{' }} {%- set response_ns = namespace(found_first=false) -%} {%- for key, value in item['response'] | dictsort -%} {%- if response_ns.found_first %},{% endif -%} {%- set response_ns.found_first = true -%} {{- key -}}:{{- format_argument(value, escape_keys=False) -}} {%- endfor -%} {{- '}' -}} {%- elif 'name' in message -%} {{ 'response:' + message['name'] | trim + '{' }} {%- set response_ns = namespace(found_first=false) -%} {%- for key, value in item | dictsort -%} {%- if response_ns.found_first %},{% endif -%} {%- set response_ns.found_first = true -%} {{- key -}}:{{- format_argument(value, escape_keys=False) -}} {%- endfor -%} {{- '}' -}} {%- else -%} {{ raise_exception("Invalid tool response mapping: must contain 'name' and 'response' keys, or 'name' must be in the message.") }} {%- endif -%} {%- else -%} {{ raise_exception("Invalid tool response message: multiple responses must all be mappings") }} {%- endif -%} {%- endfor -%} {%- else -%} {{ raise_exception("Invalid content type in tool message: must be mapping, sequence of mappings, or string.") }} {%- endif -%} {%- endif -%} {%- set ns.prev_message_type = 'tool_response' -%} {%- endif -%} {%- if ns.prev_message_type not in ['tool_call', 'tool_response'] -%} {{ '\n' }} {%- endif -%} {%- endfor -%} {%- if add_generation_prompt -%} {%- if ns.prev_message_type != 'tool_response' -%} {{- 'model\n' -}} {%- endif -%} {%- endif -%}