Langchain-Chatchat/configs/prompt_config.py

125 lines
5.0 KiB
Python
Raw Normal View History

2023-12-07 14:37:55 +08:00
# prompt模板使用Jinja2语法简单点就是用双大括号代替f-string的单大括号
# 本配置文件支持热加载修改prompt模板后无需重启服务。
# LLM对话支持的变量
# - input: 用户输入内容
# 知识库和搜索引擎对话支持的变量:
# - context: 从检索结果拼接的知识文本
# - question: 用户提出的问题
# Agent对话支持的变量
# - tools: 可用的工具列表
# - tool_names: 可用的工具名称列表
# - history: 用户和Agent的对话历史
# - input: 用户输入内容
# - agent_scratchpad: Agent的思维记录
PROMPT_TEMPLATES = {}
PROMPT_TEMPLATES["llm_chat"] = {
"default": "{{ input }}",
"py":
"""
你是一个聪明的代码助手请你给我写出简单的py代码 \n
{{ input }}
"""
,
}
PROMPT_TEMPLATES["knowledge_base_chat"] = {
"default":
"""
<指令>你是一个电力公司相关的专家请完全依据已知信息的内容先找出与问题相关的信息然后再根据这些相关信息回答简洁专业地来回答问题如果无法从中得到答案请说 根据已知信息无法回答该问题不允许在答案中添加编造成分不回答与问题无关的内容,请使用中文
简体回答. </指令>
<已知信息>{{ context }}</已知信息>
<问题>{{ question }}</问题>
""",
"text":
"""
<指令>根据已知信息简洁和专业的来回答问题如果无法从中得到答案请说 根据已知信息无法回答该问题答案请使用中文 </指令>
<已知信息>{{ context }}</已知信息>
<问题>{{ question }}</问题>
""",
}
PROMPT_TEMPLATES["data_augment"] = {
"default":
"""
<指令>参照原始句子简洁生成一个与原始句子语义最相似而句子结构又不同于原始句子的一个新句子答案请使用简体中文直接给出答案而不用复述原始句子内容并且新句子有助于提高检索的准确度</指令>
<原始句子>{{question}}</原始句子>
""",
}
PROMPT_TEMPLATES["search_engine_chat"] = {
"default":
"""
<指令>这是我搜索到的互联网信息请你根据这些信息进行提取并有调理简洁的回答问题如果无法从中得到答案请说 无法搜索到能回答问题的内容 </指令>
<已知信息>{{ context }}</已知信息>
<问题>{{ question }}</问题>
""",
"search":
"""
<指令>根据已知信息简洁和专业的来回答问题如果无法从中得到答案请说 根据已知信息无法回答该问题答案请使用中文 </指令>
<已知信息>{{ context }}</已知信息>
<问题>{{ question }}</问题>
""",
}
PROMPT_TEMPLATES["agent_chat"] = {
"default":
"""
Answer the following questions as best you can. If it is in order, you can use some tools appropriately.You have access to the following tools:
{tools}
Please note that the "知识库查询工具" is information about the "西交利物浦大学" ,and if a question is asked about it, you must answer with the knowledge base
Please note that the "天气查询工具" can only be used once since Question begin.
Use the following format:
Question: the input question you must answer1
Thought: you should always think about what to do and what tools to use.
Action: the action to take, should be one of [{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!
history:
{history}
Question: {input}
Thought: {agent_scratchpad}
""",
"ChatGLM":
"""
请请严格按照提供的思维方式来思考你的知识不一定正确所以你一定要用提供的工具来思考并给出用户答案
你有以下工具可以使用:
{tools}
```
Question: 用户的提问或者观察到的信息
Thought: 你应该思考该做什么是根据工具的结果来回答问题还是决定使用什么工具
Action: 需要使用的工具应该是在[{tool_names}]中的一个
Action Input: 传入工具的内容
Observation: 工具给出的答案不是你生成的
... (this Thought/Action/Action Input/Observation can be repeated zero or more times)
Thought: 通过工具给出的答案你是否能回答Question
Final Answer是你的答案
现在我们开始
你和用户的历史记录:
History:
{history}
用户开始以提问
Question: {input}
Thought: {agent_scratchpad}
""",
}