|
|
|
|
|
from typing import List, Union |
|
|
|
|
|
from .base import AgentKeyword, BaseAgentTemplate |
|
|
|
|
|
keyword = AgentKeyword( |
|
|
action='✿FUNCTION✿:', |
|
|
action_input='✿ARGS✿:', |
|
|
observation='✿RESULT✿:', |
|
|
) |
|
|
|
|
|
|
|
|
class QwenEnAgentTemplate(BaseAgentTemplate): |
|
|
keyword = keyword |
|
|
|
|
|
def _get_tool_names_descs(self, tools): |
|
|
tool_names = [] |
|
|
tool_descs = [] |
|
|
for tool in tools: |
|
|
tool_desc = self._parse_tool(tool, 'en') |
|
|
tool_names.append(tool_desc.name_for_model) |
|
|
tool_descs.append(f'### {tool_desc.name_for_human}\n\n' |
|
|
f'{tool_desc.name_for_model}: {tool_desc.description_for_model} ' |
|
|
f'Parameters: {tool_desc.parameters} {tool_desc.args_format}') |
|
|
return tool_names, tool_descs |
|
|
|
|
|
def _format_tools(self, tools: List[Union[str, dict]], system: str, user_message=None) -> str: |
|
|
tool_names, tool_descs = self._get_tool_names_descs(tools) |
|
|
return f"""{system} |
|
|
|
|
|
# Tools |
|
|
|
|
|
## You have access to the following tools: |
|
|
|
|
|
""" + '\n\n'.join(tool_descs) + f""" |
|
|
|
|
|
## When you need to call a tool, please insert the following command in your reply, which can be called zero or multiple times according to your needs: |
|
|
|
|
|
✿FUNCTION✿: The tool to use, should be one of [{','.join(tool_names)}] |
|
|
✿ARGS✿: The input of the tool |
|
|
✿RESULT✿: Tool results |
|
|
✿RETURN✿: Reply based on tool results. Images need to be rendered as """ |
|
|
|
|
|
|
|
|
class QwenZhAgentTemplate(BaseAgentTemplate): |
|
|
keyword = keyword |
|
|
|
|
|
def _get_tool_names_descs(self, tools): |
|
|
tool_names = [] |
|
|
tool_descs = [] |
|
|
for tool in tools: |
|
|
tool_desc = self._parse_tool(tool, 'zh') |
|
|
tool_names.append(tool_desc.name_for_model) |
|
|
tool_descs.append(f'### {tool_desc.name_for_human}\n\n' |
|
|
f'{tool_desc.name_for_model}: {tool_desc.description_for_model} ' |
|
|
f'输入参数：{tool_desc.parameters} {tool_desc.args_format}') |
|
|
return tool_names, tool_descs |
|
|
|
|
|
def _format_tools(self, tools: List[Union[str, dict]], system: str, user_message=None) -> str: |
|
|
tool_names, tool_descs = self._get_tool_names_descs(tools) |
|
|
return f"""{system} |
|
|
|
|
|
# 工具 |
|
|
|
|
|
## 你拥有如下工具： |
|
|
|
|
|
""" + '\n\n'.join(tool_descs) + f""" |
|
|
|
|
|
## 你可以在回复中插入零次、一次或多次以下命令以调用工具： |
|
|
|
|
|
✿FUNCTION✿: 工具名称，必须是[{','.join(tool_names)}]之一。 |
|
|
✿ARGS✿: 工具输入 |
|
|
✿RESULT✿: 工具结果 |
|
|
✿RETURN✿: 根据工具结果进行回复，需将图片用渲染出来""" |
|
|
|
|
|
|
|
|
class QwenEnParallelAgentTemplate(QwenEnAgentTemplate): |
|
|
|
|
|
def _format_tools(self, tools: List[Union[str, dict]], system: str, user_message=None) -> str: |
|
|
tool_names, tool_descs = self._get_tool_names_descs(tools) |
|
|
return f"""{system} |
|
|
|
|
|
# Tools |
|
|
|
|
|
## You have access to the following tools: |
|
|
|
|
|
""" + '\n\n'.join(tool_descs) + f""" |
|
|
|
|
|
## Insert the following command in your reply when you need to call N tools in parallel: |
|
|
|
|
|
✿FUNCTION✿: The name of tool 1, should be one of [{','.join(tool_names)}] |
|
|
✿ARGS✿: The input of tool 1 |
|
|
✿FUNCTION✿: The name of tool 2 |
|
|
✿ARGS✿: The input of tool 2 |
|
|
... |
|
|
✿FUNCTION✿: The name of tool N |
|
|
✿ARGS✿: The input of tool N |
|
|
✿RESULT✿: The result of tool 1 |
|
|
✿RESULT✿: The result of tool 2 |
|
|
... |
|
|
✿RESULT✿: he result of tool N |
|
|
✿RETURN✿: Reply based on tool results. Images need to be rendered as """ |
|
|
|
|
|
|
|
|
class QwenZhParallelAgentTemplate(QwenZhAgentTemplate): |
|
|
|
|
|
def _format_tools(self, tools: List[Union[str, dict]], system: str, user_message=None) -> str: |
|
|
tool_names, tool_descs = self._get_tool_names_descs(tools) |
|
|
return f"""{system} |
|
|
|
|
|
# 工具 |
|
|
|
|
|
## 你拥有如下工具： |
|
|
|
|
|
""" + '\n\n'.join(tool_descs) + f""" |
|
|
|
|
|
## 你可以在回复中插入以下命令以并行调用N个工具： |
|
|
|
|
|
✿FUNCTION✿: 工具1的名称，必须是[{','.join(tool_names)}]之一 |
|
|
✿ARGS✿: 工具1的输入 |
|
|
✿FUNCTION✿: 工具2的名称 |
|
|
✿ARGS✿: 工具2的输入 |
|
|
... |
|
|
✿FUNCTION✿: 工具N的名称 |
|
|
✿ARGS✿: 工具N的输入 |
|
|
✿RESULT✿: 工具1的结果 |
|
|
✿RESULT✿: 工具2的结果 |
|
|
... |
|
|
✿RESULT✿: 工具N的结果 |
|
|
✿RETURN✿: 根据工具结果进行回复，需将图片用渲染出来""" |
|
|
|
