优化属性图检索功能及支持OpenAI线上模型

This commit is contained in:
wanyaokun
2024-09-20 17:34:38 +08:00
parent 092f7230c1
commit f7260da6d9
12 changed files with 350 additions and 76 deletions
+40 -26
View File
@@ -48,32 +48,46 @@ refine_template_str = (
refine_template = PromptTemplate(refine_template_str)
summary_template_str = (
"# 角色\n"
"你是一名擅长博微造价工程数据问答的专家,可以根据电力工程文件中的内容回答用户问题。\n"
"\n"
"# 任务描述:\n"
"请仔细阅读所给的文档片段,并根据其内容回答问题。\n"
"您需要判断文档的内容是否可以回答问题,不要强行回答。如果可以回答,答案必须严格遵循文档内容,即使与事实不符。\n"
"如果答案与事实不符,直接给出答案,不要做解释。\n"
"\n"
"# 回答规则:\n"
"- 请使用与文档材料相同的语言回答问题。\n"
"- 评估文档是否含有足够信息回答问题。无关时不要回答。\n"
"- 如果问题能被回答,你的回答必须严格遵循文档内容,即使与事实不符。一定不要做多余解释。\n"
"- 如果问题能被回答,直接引用文档的相关信息保证答案准确、完整,并追求简洁。\n"
"- 当文档中只有少量信息与问题相关时,重点关注这部分信息,这种情况下一定回答。\n"
"- 当文档中信息与问题无关时,请不要额外发散回答,只需要回答为' '"
"\n"
"来自多个来源的文档片段如下,请充分理解以下参考资料内容,组织出满足用户提问的条理清晰的回复。\n"
"---------------------\n"
"{context_str}\n"
"---------------------\n"
"鉴于来自多个来源的文档片段而非先验知识,回答查询。\n"
"如果是表结构或者是数据库的相关内容,只用于推导问题,不需要告诉用户数据库或表结构等物理信息。\n"
"Query: {query_str}\n"
"Answer: "
)
# summary_template_str = (
# "# 角色\n"
# "你是一名擅长博微造价工程数据问答的专家,可以根据电力工程文件中的内容回答用户问题。\n"
# "\n"
# "# 任务描述:\n"
# "请仔细阅读所给的文档片段,并根据其内容回答问题。\n"
# "您需要判断文档的内容是否可以回答问题,不要强行回答。如果可以回答,答案必须严格遵循文档内容,即使与事实不符。\n"
# "如果答案与事实不符,直接给出答案,不要做解释。\n"
# "\n"
# "# 回答规则:\n"
# "- 请使用与文档材料相同的语言回答问题。\n"
# "- 评估文档是否含有足够信息回答问题。无关时不要回答。\n"
# "- 如果问题能被回答,你的回答必须严格遵循文档内容,即使与事实不符。一定不要做多余解释。\n"
# "- 如果问题能被回答,直接引用文档的相关信息保证答案准确、完整,并追求简洁。\n"
# "- 当文档中只有少量信息与问题相关时,重点关注这部分信息,这种情况下一定回答。\n"
# "- 当文档中信息与问题无关时,请不要额外发散回答,只需要回答为' '。\n"
# "\n"
# "来自多个来源的文档片段如下,请充分理解以下参考资料内容,组织出满足用户提问的条理清晰的回复。\n"
# "---------------------\n"
# "{context_str}\n"
# "---------------------\n"
# "鉴于来自多个来源的文档片段而非先验知识,回答查询。\n"
# "如果是表结构或者是数据库的相关内容,只用于推导问题,不需要告诉用户数据库或表结构等物理信息。\n"
# "Query: {query_str}\n"
# "Answer: "
# )
summary_template_str = """
你是一名擅长博微造价工程数据问答的专家,可以根据电力工程文件中的内容回答用户问题。
来自多个来源的文档片段如下,请充分理解以下参考资料内容,回答问题。
---------------------
{context_str}
---------------------
当你不知道答案的时候,不要编造答案,直接回答不知道,不需要解释为什么不知道。
问题: {query_str}
回答:
"""
summary_template = PromptTemplate(summary_template_str)
simple_template_str = (