|
|
{%- macro append_new_param_info(param_declaration, comment_info, examples_info, depth) -%} |
|
|
{%- set offset = "" -%} |
|
|
{%- if depth >= 1 -%} |
|
|
{%- set offset = " " * depth -%} |
|
|
{%- endif -%} |
|
|
{%- if comment_info != "<|NONE|>" -%} |
|
|
{{ "\n" + offset + comment_info }} |
|
|
{%- if examples_info | length > 0 -%} |
|
|
|
|
|
{%- for example in examples_info -%} |
|
|
{{ "\n" + offset + "// " + example|string|replace("'", '"') }} |
|
|
{%- endfor -%} |
|
|
{%- endif -%} |
|
|
{%- endif -%} |
|
|
{{ "\n" + offset + param_declaration }} |
|
|
{%- endmacro -%} |
|
|
|
|
|
{%- macro convert_data_type(param_type) -%} |
|
|
{%- if param_type == "integer" or param_type == "float" -%} |
|
|
{{ "number" }} |
|
|
{%- else -%} |
|
|
{{ param_type }} |
|
|
{%- endif -%} |
|
|
{%- endmacro -%} |
|
|
|
|
|
{%- macro get_param_type(param) -%} |
|
|
{%- set param_type = "any" -%} |
|
|
|
|
|
{%- if "type" in param -%} |
|
|
{%- set raw_param_type = param["type"] -%} |
|
|
{%- if raw_param_type is iterable and raw_param_type is not string -%} |
|
|
{%- set param_type = raw_param_type | join(" | ") -%} |
|
|
{%- else -%} |
|
|
{%- set param_type = raw_param_type -%} |
|
|
{%- endif -%} |
|
|
{{ convert_data_type(param_type) }} |
|
|
{%- elif "oneOf" in param -%} |
|
|
{%- set one_of_types = param["oneOf"]|selectattr("type", "defined")|list -%} |
|
|
{%- set one_of_types = one_of_types|map(attribute="type")|unique|list -%} |
|
|
{{ convert_data_type(one_of_types | join(" | ")) }} |
|
|
{%- endif -%} |
|
|
{%- endmacro -%} |
|
|
|
|
|
{%- macro get_format_param(param) -%} |
|
|
{%- if "format" in param -%} |
|
|
{{ param["format"] }} |
|
|
{%- elif "oneOf" in param -%} |
|
|
{%- set formats = [] -%} |
|
|
{%- for item in param["oneOf"] -%} |
|
|
{%- if "format" in item -%} |
|
|
{%- if item["format"] == param["oneOf"][-1]["format"] -%} |
|
|
{{ item["format"] }} |
|
|
{%- else -%} |
|
|
{{ item["format"] + " or "}} |
|
|
{%- endif -%} |
|
|
{%- endif -%} |
|
|
{%- endfor -%} |
|
|
{%- else -%} |
|
|
{{ "<|NONE|>" }} |
|
|
{%- endif -%} |
|
|
{%- endmacro -%} |
|
|
|
|
|
{%- macro get_param_info(param) -%} |
|
|
{%- set param_type = param.get("type", "any") -%} |
|
|
{%- set format_param = get_format_param(param) -%} |
|
|
|
|
|
{%- if "description" in param or "default" in param or format_param != "<|NONE|>" or param["maximum"] or param["minimum"] or param["maxLength"] or param["minLength"] -%} |
|
|
{{ "//" }} |
|
|
{%- if "description" in param -%} |
|
|
{%- set desc = param["description"] -%} |
|
|
{%- if not desc.endswith(".") -%} |
|
|
{%- set desc = desc + "." -%} |
|
|
{%- endif -%} |
|
|
{{ " " + desc }} |
|
|
{%- endif -%} |
|
|
|
|
|
{%- if "default" in param -%} |
|
|
{%- set default_value = param["default"] -%} |
|
|
{%- if param_type == "string" -%} |
|
|
{%- set default_value = '"' ~ default_value ~ '"' -%} |
|
|
{%- endif -%} |
|
|
{{ " Default=" ~ default_value ~ "." }} |
|
|
{%- endif -%} |
|
|
|
|
|
{%- set format_param = get_format_param(param) -%} |
|
|
{%- if format_param != "<|NONE|>" -%} |
|
|
{{ " Format=" ~ format_param }} |
|
|
{%- endif -%} |
|
|
|
|
|
{%- for field, field_name in [("maximum", "Maximum"), ("minimum", "Minimum"), ("maxLength", "Maximum length"), ("minLength", "Minimum length")] -%} |
|
|
{%- if field in param -%} |
|
|
{{ " " + field_name ~ "=" ~ param[field] }} |
|
|
{%- endif -%} |
|
|
{%- endfor -%} |
|
|
{%- else -%} |
|
|
{{ "<|NONE|>"}} |
|
|
{%- endif -%} |
|
|
{%- endmacro -%} |
|
|
|
|
|
{%- macro get_enum_option_str(enum_options) -%} |
|
|
{%- for v in enum_options -%} |
|
|
{%- if v is string -%} |
|
|
{{ '"' + v + '"' }} |
|
|
{%- else -%} |
|
|
{{ v }} |
|
|
{%- endif -%} |
|
|
{%- if enum_options|length > 0 and v != enum_options[-1] -%} |
|
|
{{ " | " }} |
|
|
{%- endif -%} |
|
|
{%- endfor -%} |
|
|
{%- endmacro -%} |
|
|
|
|
|
{%- macro get_array_typescript(param_name, param_dic, depth) -%} |
|
|
{%- set offset = '' -%} |
|
|
{%- if depth >= 1 -%} |
|
|
{%- set offset = " " * depth -%} |
|
|
{%- endif -%} |
|
|
{%- set items_info = param_dic.get('items', {}) -%} |
|
|
|
|
|
{%- if items_info|length == 0 -%} |
|
|
{%- if param_name -%} |
|
|
{{ "\n" + offset + param_name + ": []" }} |
|
|
{%- else -%} |
|
|
{{ "\n" + offset + "[]" }} |
|
|
{%- endif -%} |
|
|
{%- else -%} |
|
|
{%- set array_type = get_param_type(items_info) -%} |
|
|
{%- if array_type == 'object' -%} |
|
|
{%- if param_name -%} |
|
|
{{ "\n" + offset + param_name + ": {" }} |
|
|
{%- else -%} |
|
|
{{ "\n" + offset + "{" }} |
|
|
{%- endif -%} |
|
|
{{ get_parameter_typescript(items_info.get('properties', {}), items_info.get('required', []), depth + 1) -}} |
|
|
{{- "\n" + offset + "}[]" }} |
|
|
{%- elif array_type == 'array' -%} |
|
|
{%- set item_info = get_array_typescript(None, items_info, depth + 1) -%} |
|
|
{%- if not param_name -%} |
|
|
{{ "\n" + item_info + "[]" }} |
|
|
{%- else -%} |
|
|
{{ "\n" + offset + param_name + ": " + item_info|trim + "[]" }} |
|
|
{%- endif -%} |
|
|
{%- else -%} |
|
|
{%- if 'enum' in items_info -%} |
|
|
{%- set item_type = get_enum_option_str(items_info['enum']) -%} |
|
|
{%- if param_name is none -%} |
|
|
{{ "(" + item_type + ")[]"}} |
|
|
{%- else -%} |
|
|
{{ "\n" + offset + param_name + ": (" + item_type + ")[]" }} |
|
|
{%- endif -%} |
|
|
{%- else -%} |
|
|
{%- if param_name is none -%} |
|
|
{{ "\n" + array_type + "[]" }} |
|
|
{%- else -%} |
|
|
{{ "\n" + offset + param_name + ": " + array_type + "[]," }} |
|
|
{%- endif -%} |
|
|
{%- endif -%} |
|
|
{%- endif -%} |
|
|
{%- endif -%} |
|
|
{%- endmacro -%} |
|
|
|
|
|
{%- macro get_parameter_typescript(properties, required_params, depth=0) -%} |
|
|
{%- set res = "" -%} |
|
|
{%- for param_name, param in properties.items() -%} |
|
|
{%- if param is mapping -%} |
|
|
{%- set comment_info = get_param_info(param) -%} |
|
|
|
|
|
{%- set examples_info = [] -%} |
|
|
{%- if "examples" in param -%} |
|
|
{%- set examples_info = ["Example " + param_name + ":"] -%} |
|
|
{%- set examples_info = examples_info + param["examples"] -%} |
|
|
{%- endif -%} |
|
|
|
|
|
|
|
|
{%- set param_declaration = param_name -%} |
|
|
{%- if required_params is iterable and param_name not in required_params -%} |
|
|
{%- set param_declaration = param_declaration + "?" -%} |
|
|
{%- endif -%} |
|
|
|
|
|
{%- set param_type = get_param_type(param) -%} |
|
|
|
|
|
|
|
|
{%- set offset = "" -%} |
|
|
{%- if depth >= 1 -%} |
|
|
{%- set offset = " " * depth -%} |
|
|
{%- endif -%} |
|
|
|
|
|
{%- if param_type == "object" -%} |
|
|
{%- if comment_info != "<|NONE|>" -%} |
|
|
{{ "\n" + offset + comment_info }} |
|
|
{%- endif -%} |
|
|
{%- if examples_info|length > 0 -%} |
|
|
{%- for example in examples_info -%} |
|
|
{{ "\n" + offset + "// " + example|string|replace("'", '"') }} |
|
|
{%- endfor -%} |
|
|
{%- endif -%} |
|
|
{%- set param_declaration = param_declaration + ": {" -%} |
|
|
{{ "\n" + offset + param_declaration -}} |
|
|
{{- get_parameter_typescript(param.get("properties", {}), param.get("required", []), depth + 1) -}} |
|
|
{{- "\n" + offset + "}," }} |
|
|
{%- elif param_type == "array" -%} |
|
|
{%- set item_info = param.get("items", {}) -%} |
|
|
{%- if "type" not in item_info -%} |
|
|
{%- set param_declaration = param_declaration + ": []," -%} |
|
|
{{ append_new_param_info(param_declaration, comment_info, examples_info, depth) }} |
|
|
{%- else -%} |
|
|
{%- if comment_info != "<|NONE|>" -%} |
|
|
{{ "\n" + offset + comment_info }} |
|
|
{%- endif -%} |
|
|
{%- if examples_info|length > 0 -%} |
|
|
{%- for example in examples_info -%} |
|
|
{{ "\n" + offset + "// " + example|string|replace("'", '"') }} |
|
|
{%- endfor -%} |
|
|
{%- endif -%} |
|
|
{%- set array_declaration = get_array_typescript(param_declaration, param, depth) -%} |
|
|
{%- if not array_declaration.endswith(",") -%} |
|
|
{%- set array_declaration = array_declaration + "," -%} |
|
|
{%- endif -%} |
|
|
{{ array_declaration}} |
|
|
{%- endif -%} |
|
|
{%- else -%} |
|
|
{%- if "enum" in param -%} |
|
|
{%- set param_type = get_enum_option_str(param["enum"]) -%} |
|
|
{%- endif -%} |
|
|
{%- if "nullable" in param and param["nullable"] -%} |
|
|
{%- set param_type = param_type + " | null" -%} |
|
|
{%- endif -%} |
|
|
{%- set param_declaration = param_declaration + ": " + param_type + "," -%} |
|
|
{{ append_new_param_info(param_declaration, comment_info, examples_info, depth) }} |
|
|
{%- endif -%} |
|
|
{%- endif -%} |
|
|
{%- endfor -%} |
|
|
{%- endmacro -%} |
|
|
|
|
|
{%- macro generate_schema_from_functions(functions, namespace='functions') -%} |
|
|
{{ "// Supported function definitions that should be called when necessary.\n" -}} |
|
|
{{- "namespace " + namespace + " {\n\n" -}} |
|
|
|
|
|
{%- for function in functions -%} |
|
|
{%- if function.get("function") -%} |
|
|
{%- set function = function.get("function") -%} |
|
|
{%- endif -%} |
|
|
|
|
|
{%- set function_name = function.get("name") -%} |
|
|
{%- if function_name -%} |
|
|
{%- set description = function.get('description', '') -%} |
|
|
{%- set parameters = function.get('parameters', {}) -%} |
|
|
{{- "// " + description + "\n" -}} |
|
|
{{- "type " + function_name -}} |
|
|
{%- if parameters and parameters.get("properties") -%} |
|
|
{{- " = (_: {" -}} |
|
|
{%- set required_params = parameters.get("required", []) -%} |
|
|
{{ get_parameter_typescript(parameters.get("properties"), required_params, 0) -}} |
|
|
{{- "\n}) => any;\n\n" }} |
|
|
{%- else -%} |
|
|
{{ " = () => any;\n\n" }} |
|
|
{%- endif -%} |
|
|
{%- endif -%} |
|
|
{%- endfor -%} |
|
|
{{ "} // namespace " + namespace }} |
|
|
{%- endmacro -%} |
|
|
{%- if not tools is defined -%} |
|
|
{%- set tools = none -%} |
|
|
{%- endif -%} |
|
|
|
|
|
{%- set has_code_interpreter = tools | selectattr("type", "equalto", "code_interpreter") | list | length > 0 -%} |
|
|
{%- if has_code_interpreter -%} |
|
|
{%- set tools = tools | rejectattr("type", "equalto", "code_interpreter") | list -%} |
|
|
{%- endif -%} |
|
|
|
|
|
|
|
|
{{- bos_token + "<|start_header_id|>system<|end_header_id|>\n\n" }} |
|
|
{%- if has_code_interpreter %} |
|
|
{{- "Environment: ipython\n\n" }} |
|
|
{%- else -%} |
|
|
{{ "\n"}} |
|
|
{%- endif %} |
|
|
{%- if tools %} |
|
|
{{- "\nYou have access to the following functions:\n\n" }} |
|
|
{%- for t in tools %} |
|
|
{%- if "type" in t -%} |
|
|
{{ "Use the function '" + t["function"]["name"] + "' to '" + t["function"]["description"] + "'\n" + t["function"] | tojson() }} |
|
|
{%- else -%} |
|
|
{{ "Use the function '" + t["name"] + "' to '" + t["description"] + "'\n" + t | tojson }} |
|
|
{%- endif -%} |
|
|
{{- "\n\n" }} |
|
|
{%- endfor %} |
|
|
{{- '\nThink very carefully before calling functions.\nIf a you choose to call a function ONLY reply in the following format:\n<{start_tag}={function_name}>{parameters}{end_tag}\nwhere\n\nstart_tag => `<function`\nparameters => a JSON dict with the function argument name as key and function argument value as value.\nend_tag => `</function>`\n\nHere is an example,\n<function=example_function_name>{"example_name": "example_value"}</function>\n\nReminder:\n- If looking for real time information use relevant functions before falling back to brave_search\n- Function calls MUST follow the specified format, start with <function= and end with </function>\n- Required parameters MUST be specified\n- Only call one function at a time\n- Put the entire function call reply on one line\n\n' -}} |
|
|
{%- endif %} |
|
|
{{- "<|eot_id|>" -}} |
|
|
|
|
|
{%- for message in messages -%} |
|
|
{%- if message['role'] == 'user' or message['role'] == 'system' -%} |
|
|
{{ '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n' + message['content'] + '<|eot_id|>' }} |
|
|
{%- elif message['role'] == 'tool' -%} |
|
|
{{ '<|start_header_id|>ipython<|end_header_id|>\n\n' + message['content'] + '<|eot_id|>' }} |
|
|
{%- else -%} |
|
|
{%- if (message['content'] and message['content']|length > 0) or ('tool_calls' in message and message['tool_calls'] and message['tool_calls']|length > 0) -%} |
|
|
{{ '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'}} |
|
|
{%- endif -%} |
|
|
{%- if message['content'] and message['content']|length > 0 -%} |
|
|
{{ message['content'] }} |
|
|
{%- endif -%} |
|
|
{%- if 'tool_calls' in message and message['tool_calls'] and message['tool_calls']|length > 0 -%} |
|
|
{%- for tool_call in message['tool_calls'] -%} |
|
|
{%- if tool_call["function"]["name"] == "python" -%} |
|
|
{{ '<|python_tag|>' + tool_call['function']['arguments'] }} |
|
|
{%- else -%} |
|
|
{{ '<function=' + tool_call['function']['name'] + '>' + tool_call['function']['arguments'] + '</function>' }} |
|
|
{%- endif -%} |
|
|
{%- endfor -%} |
|
|
{{ '<|eom_id|>' }} |
|
|
{%- elif message['content'] and message['content']|length > 0 -%} |
|
|
{{ '<|eot_id|>' }} |
|
|
{%- endif -%} |
|
|
{%- endif -%} |
|
|
{%- endfor -%} |
|
|
{%- if add_generation_prompt -%} |
|
|
{{ '<|start_header_id|>assistant<|end_header_id|>\n\n' }} |
|
|
{%- endif -%} |
|
|
|
