Navanit-AI commited on
Commit
8adfa60
·
verified ·
1 Parent(s): 6b3ca9c

Adding tool call support in chat template

Browse files

Added tool calling feature in it. Kindly check for it.

Files changed (1) hide show
  1. tokenizer_config.json +1 -1
tokenizer_config.json CHANGED
@@ -9007,7 +9007,7 @@
9007
  "<SPECIAL_999>"
9008
  ],
9009
  "bos_token": "<s>",
9010
- "chat_template": "{%- set today = strftime_now(\"%Y-%m-%d\") %}\n{%- set default_system_message = \"You are Mistral Small 3, a Large Language Model (LLM) created by Mistral AI, a French startup headquartered in Paris.\\nYour knowledge base was last updated on 2023-10-01. The current date is \" + today + \".\\n\\nWhen you're not sure about some information, you say that you don't have the information and don't make up anything.\\nIf the user's question is not clear, ambiguous, or does not provide enough context for you to accurately answer the question, you do not try to answer it right away and you rather ask the user to clarify their request (e.g. \\\"What are some good restaurants around me?\\\" => \\\"Where are you?\\\" or \\\"When is the next flight to Tokyo\\\" => \\\"Where do you travel from?\\\")\" %}\n\n{{- bos_token }}\n\n{%- if messages[0]['role'] == 'system' %}\n {%- set system_message = messages[0]['content'] %}\n {%- set loop_messages = messages[1:] %}\n{%- else %}\n {%- set system_message = default_system_message %}\n {%- set loop_messages = messages %}\n{%- endif %}\n{{- '[SYSTEM_PROMPT]' + system_message + '[/SYSTEM_PROMPT]' }}\n\n{%- for message in loop_messages %}\n {%- if message['role'] == 'user' %}\n {{- '[INST]' + message['content'] + '[/INST]' }}\n {%- elif message['role'] == 'system' %}\n {{- '[SYSTEM_PROMPT]' + message['content'] + '[/SYSTEM_PROMPT]' }}\n {%- elif message['role'] == 'assistant' %}\n {{- message['content'] + eos_token }}\n {%- else %}\n {{- raise_exception('Only user, system and assistant roles are supported!') }}\n {%- endif %}\n{%- endfor %}",
9011
  "clean_up_tokenization_spaces": false,
9012
  "eos_token": "</s>",
9013
  "extra_special_tokens": {},
 
9007
  "<SPECIAL_999>"
9008
  ],
9009
  "bos_token": "<s>",
9010
+ "chat_template": "{%- set today = strftime_now(\"%Y-%m-%d\") -%}{%- set default_system_message = \"You are Mistral Small 3, a Large Language Model (LLM) created by Mistral AI, a French startup headquartered in Paris.\\nYour knowledge base was last updated on 2023-10-01. The current date is \" + today + \".\\n\\nWhen you are unsure about some information, state that you do not have the information and do not make up anything.\\nIf the user's question is unclear, ambiguous, or lacks enough context for an accurate answer, you do not answer immediately.\\nInstead, you ask the user to clarify their request.\\nFor example:\\n- \\\"What are some good restaurants around me?\\\" \\\"Where are you?\\\"\\n- \\\"When is the next flight to Tokyo?\\\" \\\"Where are you departing from?\\\"\" -%}{{- bos_token -}}{%- if tools is defined and tools | length > 0 -%}{%- set tool_instructions = \"\\n\\nYou can use tools by responding with JSON: {\\\"name\\\": \\\"tool_name\\\", \\\"parameters\\\": {\\\"arg\\\": value}}\" -%}{%- set default_system_message = default_system_message + tool_instructions -%}{%- endif -%}{%- if messages[0]['role'] == 'system' -%}{%- set system_message = messages[0]['content'] -%}{%- set loop_messages = messages[1:] -%}{%- else -%}{%- set system_message = default_system_message -%}{%- set loop_messages = messages -%}{%- endif -%}{{- '[SYSTEM_PROMPT]' + system_message + '[/SYSTEM_PROMPT]' -}}{%- for message in loop_messages -%}{%- if message['role'] == 'user' -%}{{- '[INST]' + message['content'] + '[/INST]' -}}{%- elif message['role'] == 'assistant' -%}{%- if 'tool_calls' in message -%}{%- if message.tool_calls|length != 1 -%}{{- raise_exception('Only single tool calls supported') -}}{%- endif -%}{{- '[TOOL_CALL]' + '{\\\"name\\\": \\\"' + message.tool_calls[0].function.name + '\\\", \\\"parameters\\\": ' + message.tool_calls[0].function.arguments|tojson + '}' + '[/TOOL_CALL]' -}}{%- else -%}{{- message['content'] + eos_token -}}{%- endif -%}{%- elif message['role'] == 'tool' -%}{{- '[TOOL_OUTPUT]' + message['content'] + '[/TOOL_OUTPUT]' -}}{%- endif -%}{%- endfor -%}",
9011
  "clean_up_tokenization_spaces": false,
9012
  "eos_token": "</s>",
9013
  "extra_special_tokens": {},