Spaces:
Sleeping
Sleeping
File size: 3,368 Bytes
98e07ff |
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 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 |
from custom_prompt import CustomPromptGenerator
DEFAULT_SYSTEM_TEMPLATE = """
# 工具
## 你拥有如下工具：
<tool_list>
## 当你需要调用工具时，请在你的回复中穿插如下的工具调用命令，可以根据需求调用零次或多次：
工具调用
Action: 工具的名称，必须是<tool_name_list>之一
Action Input: 工具的输入
Observation: <result>工具返回的结果</result>
Answer: 根据Observation总结本次工具调用返回的结果，如果结果中出现url，请不要展示出。
```
[链接](url)
```
# 指令
"""
DEFAULT_SYSTEM_TEMPLATE_WITHOUT_TOOL = """
# 指令
"""
DEFAULT_INSTRUCTION_TEMPLATE = ''
DEFAULT_USER_TEMPLATE = (
"""(你正在扮演<role_name>，你可以使用工具：<tool_name_list><knowledge_note>)<file_names><user_input>"""
)
DEFAULT_USER_TEMPLATE_WITHOUT_TOOL = """(你正在扮演<role_name><knowledge_note>) <file_names><user_input>"""
DEFAULT_EXEC_TEMPLATE = """Observation: <result><exec_result></result>\nAnswer:"""
TOOL_DESC = (
'{name_for_model}: {name_for_human} API。 {description_for_model} 输入参数: {parameters}'
)
class ZhCustomPromptGenerator(CustomPromptGenerator):
def __init__(
self,
system_template=DEFAULT_SYSTEM_TEMPLATE,
instruction_template=DEFAULT_INSTRUCTION_TEMPLATE,
user_template=DEFAULT_USER_TEMPLATE,
exec_template=DEFAULT_EXEC_TEMPLATE,
tool_desc=TOOL_DESC,
default_user_template_without_tool=DEFAULT_USER_TEMPLATE_WITHOUT_TOOL,
default_system_template_without_tool=DEFAULT_SYSTEM_TEMPLATE_WITHOUT_TOOL,
addition_assistant_reply='好的。',
**kwargs):
super().__init__(
system_template=system_template,
instruction_template=instruction_template,
user_template=user_template,
exec_template=exec_template,
tool_desc=tool_desc,
default_user_template_without_tool=
default_user_template_without_tool,
default_system_template_without_tool=
default_system_template_without_tool,
**kwargs)
def _parse_role_config(self, config: dict):
prompt = '你扮演AI-Agent，'
# concat prompt
if 'name' in config and config['name']:
prompt += ('你的名字是' + config['name'] + '。')
if 'description' in config and config['description']:
prompt += config['description']
prompt += '\n你具有下列具体功能：'
if 'instruction' in config and config['instruction']:
if isinstance(config['instruction'], list):
for ins in config['instruction']:
prompt += ins
prompt += '；'
elif isinstance(config['instruction'], str):
prompt += config['instruction']
if prompt[-1] == '；':
prompt = prompt[:-1]
prompt += '\n下面你将开始扮演'
if 'name' in config and config['name']:
prompt += config['name']
prompt += '，明白了请说“好的。”，不要说其他的。'
return prompt
def _get_tool_template(self):
return '\n\n# 工具\n\n'
def _get_knowledge_template(self):
return '。请查看前面的知识库'
|