File size: 2,619 Bytes
7feac49 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
# Copyright (c) Alibaba, Inc. and its affiliates.
from typing import List, Union
from .base import BaseAgentTemplate
class ReactEnAgentTemplate(BaseAgentTemplate):
def _format_tools(self, tools: List[Union[str, dict]], system: str, user_message=None) -> str:
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_model}: Call this tool to interact with the {tool_desc.name_for_human} API. '
f'What is the {tool_desc.name_for_human} API useful for? {tool_desc.description_for_model} '
f'Parameters: {tool_desc.parameters} {tool_desc.args_format}')
return """Answer the following questions as best you can. You have access to the following tools:
""" + '\n\n'.join(tool_descs) + f"""
Use the following format:
Question: the input question you must answer
Thought: you should always think about what to do
Action: the action to take, should be one of [{','.join(tool_names)}]
Action Input: the input to the action
Observation: the result of the action
... (this Thought/Action/Action Input/Observation can be repeated zero or more times)
Thought: I now know the final answer
Final Answer: the final answer to the original input question
Begin!
"""
class ReactZnAgentTemplate(BaseAgentTemplate):
def _format_tools(self, tools: List[Union[str, dict]], system: str, user_message=None) -> str:
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_model}: 调用此工具与 {tool_desc.name_for_human} API 进行交互。'
f'{tool_desc.name_for_human} 有什么用？{tool_desc.description_for_model} '
f'输入参数：{tool_desc.parameters} {tool_desc.args_format}')
return """尽可能地回答以下问题。你可以使用以下工具:
""" + '\n\n'.join(tool_descs) + f"""
请按照以下格式进行:
Question: 需要你回答的输入问题
Thought: 你应该总是思考该做什么
Action: 需要使用的工具，应该是[{','.join(tool_names)}]中的一个
Action Input: 传入工具的内容
Observation: 行动的结果
... (这个Thought/Action/Action Input/Observation可以重复N次)
Thought: 我现在知道最后的答案
Final Answer: 对原始输入问题的最终答案
现在开始！
"""
|