mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2025-01-13 04:00:16 +00:00
153 lines
4.9 KiB
Django/Jinja
153 lines
4.9 KiB
Django/Jinja
{%- macro json_to_python_type(json_spec) %}
|
|
{%- set basic_type_map = {
|
|
"string": "str",
|
|
"number": "float",
|
|
"integer": "int",
|
|
"boolean": "bool"
|
|
} %}
|
|
|
|
{%- if basic_type_map[json_spec.type] is defined %}
|
|
{{- basic_type_map[json_spec.type] }}
|
|
{%- elif json_spec.type == "array" %}
|
|
{{- "list[" + json_to_python_type(json_spec|items) + "]"}}
|
|
{%- elif json_spec.type == "object" %}
|
|
{%- if json_spec.additionalProperties is defined %}
|
|
{{- "dict[str, " + json_to_python_type(json_spec.additionalProperties) + ']'}}
|
|
{%- else %}
|
|
{{- "dict" }}
|
|
{%- endif %}
|
|
{%- elif json_spec.type is iterable %}
|
|
{{- "Union[" }}
|
|
{%- for t in json_spec.type %}
|
|
{{- json_to_python_type({"type": t}) }}
|
|
{%- if not loop.last %}
|
|
{{- "," }}
|
|
{%- endif %}
|
|
{%- endfor %}
|
|
{{- "]" }}
|
|
{%- else %}
|
|
{{- "Any" }}
|
|
{%- endif %}
|
|
{%- endmacro %}
|
|
|
|
|
|
{{- bos_token }}
|
|
{{- '<|im_start|>system
|
|
' }}
|
|
{{- "You are a function calling AI model. You are provided with function signatures within <tools></tools> XML tags. You may call one or more functions to assist with the user query. Don't make assumptions about what values to plug into functions. Here are the available tools: <tools> " }}
|
|
{%- for tool in tools %}
|
|
{%- if tool.function is defined %}
|
|
{%- set tool = tool.function %}
|
|
{%- endif %}
|
|
{{- '{"type": "function", "function": ' }}
|
|
{{- '{"name": "' + tool.name + '", ' }}
|
|
{{- '"description": "' + tool.name + '(' }}
|
|
{%- for param_name, param_fields in tool.parameters.properties|items %}
|
|
{{- param_name + ": " + json_to_python_type(param_fields) }}
|
|
{%- if not loop.last %}
|
|
{{- ", " }}
|
|
{%- endif %}
|
|
{%- endfor %}
|
|
{{- ")" }}
|
|
{%- if tool.return is defined %}
|
|
{{- " -> " + json_to_python_type(tool.return) }}
|
|
{%- endif %}
|
|
{{- " - " + tool.description + "
|
|
|
|
" }}
|
|
{%- for param_name, param_fields in tool.parameters.properties|items %}
|
|
{%- if loop.first %}
|
|
{{- " Args:
|
|
" }}
|
|
{%- endif %}
|
|
{{- " " + param_name + "(" + json_to_python_type(param_fields) + "): " + param_fields.description|trim }}
|
|
{%- endfor %}
|
|
{%- if tool.return is defined and tool.return.description is defined %}
|
|
{{- "
|
|
Returns:
|
|
" + tool.return.description }}
|
|
{%- endif %}
|
|
{{- '"' }}
|
|
{{- ', "parameters": ' }}
|
|
{%- if tool.parameters.properties | length == 0 %}
|
|
{{- "{}" }}
|
|
{%- else %}
|
|
{{- tool.parameters|tojson }}
|
|
{%- endif %}
|
|
{{- "}" }}
|
|
{%- if not loop.last %}
|
|
{{- "
|
|
" }}
|
|
{%- endif %}
|
|
{%- endfor %}
|
|
{{- " </tools>" }}
|
|
{{- 'Use the following pydantic model json schema for each tool call you will make: {"properties": {"name": {"title": "Name", "type": "string"}, "arguments": {"title": "Arguments", "type": "object"}}, "required": ["name", "arguments"], "title": "FunctionCall", "type": "object"}}
|
|
' }}
|
|
{{- "For each function call return a json object with function name and arguments within <tool_call></tool_call> XML tags as follows:
|
|
" }}
|
|
{{- "<tool_call>
|
|
" }}
|
|
{{- '{"name": <function-name>, "arguments": <args-dict>}
|
|
' }}
|
|
{{- '</tool_call><|im_end|>
|
|
' }}
|
|
{%- for message in messages %}
|
|
{%- if message.role == "user" or message.role == "system" or (message.role == "assistant" and message.tool_calls is not defined) %}
|
|
{{- '<|im_start|>' + message.role + '
|
|
' + message.content + '<|im_end|>' + '
|
|
' }}
|
|
{%- elif message.role == "assistant" %}
|
|
{{- '<|im_start|>' + message.role }}
|
|
{%- for tool_call in message.tool_calls %}
|
|
{{- '
|
|
<tool_call>
|
|
' }} {%- if tool_call.function is defined %}
|
|
{%- set tool_call = tool_call.function %}
|
|
{%- endif %}
|
|
{{- '{' }}
|
|
{{- '"name": "' }}
|
|
{{- tool_call.name }}
|
|
{{- '"' }}
|
|
{{- ', '}}
|
|
{%- if tool_call.arguments is defined %}
|
|
{{- '"arguments": ' }}
|
|
{%- if tool_call.arguments is string %}
|
|
{{- tool_call.arguments }}
|
|
{%- else %}
|
|
{{- tool_call.arguments|tojson }}
|
|
{%- endif %}
|
|
{%- endif %}
|
|
{{- '}' }}
|
|
{{- '
|
|
</tool_call>' }}
|
|
{%- endfor %}
|
|
{{- '<|im_end|>
|
|
' }}
|
|
{%- elif message.role == "tool" %}
|
|
{%- if loop.previtem and loop.previtem.role != "tool" %}
|
|
{{- '<|im_start|>tool
|
|
' }}
|
|
{%- endif %}
|
|
{{- '<tool_response>
|
|
' }}
|
|
{{- message.content }}
|
|
{%- if not loop.last %}
|
|
{{- '
|
|
</tool_response>
|
|
' }}
|
|
{%- else %}
|
|
{{- '
|
|
</tool_response>' }}
|
|
{%- endif %}
|
|
{%- if not loop.last and loop.nextitem.role != "tool" %}
|
|
{{- '<|im_end|>' }}
|
|
{%- elif loop.last %}
|
|
{{- '<|im_end|>' }}
|
|
{%- endif %}
|
|
{%- endif %}
|
|
{%- endfor %}
|
|
{%- if add_generation_prompt %}
|
|
{{- '<|im_start|>assistant
|
|
' }}
|
|
{%- endif %}
|