优化意图识别模块中的字符串格式化,更新分类信息提示内容,增强可读性和准确性。
This commit is contained in:
@@ -101,9 +101,9 @@ class IntentRecognizer:
|
|||||||
Returns:
|
Returns:
|
||||||
分类结果
|
分类结果
|
||||||
"""
|
"""
|
||||||
formatted_prompt = classification_prompt.replace("{user_input}", query)
|
formatted_prompt = classification_prompt.format(user_input=query,
|
||||||
formatted_prompt = formatted_prompt.replace("{classification_info}", classification_info)
|
classification_info=classification_info,
|
||||||
formatted_prompt = formatted_prompt.replace("{output_format}", self.classification_parser.get_format_instructions())
|
output_format=self.classification_parser.get_format_instructions())
|
||||||
# 将关键词列表转换为JSON字符串
|
# 将关键词列表转换为JSON字符串
|
||||||
terms_dict = [term.model_dump() for term in keywords.terms]
|
terms_dict = [term.model_dump() for term in keywords.terms]
|
||||||
keywords_str = json.dumps(terms_dict, ensure_ascii=False)
|
keywords_str = json.dumps(terms_dict, ensure_ascii=False)
|
||||||
@@ -207,7 +207,9 @@ class IntentRecognizer:
|
|||||||
# 准备问题改写提示
|
# 准备问题改写提示
|
||||||
terms_dict = [term.model_dump(exclude={"description"}) for term in keywords.terms]
|
terms_dict = [term.model_dump(exclude={"description"}) for term in keywords.terms]
|
||||||
keywords_str = json.dumps(terms_dict, ensure_ascii=False)
|
keywords_str = json.dumps(terms_dict, ensure_ascii=False)
|
||||||
formatted_prompt = query_rewrite_prompt.format(query=query, output_format=self.query_rewrite_parser.get_format_instructions(),keywords=keywords_str)
|
formatted_prompt = query_rewrite_prompt.format(query=query,
|
||||||
|
output_format=self.query_rewrite_parser.get_format_instructions(),
|
||||||
|
keywords=keywords_str)
|
||||||
|
|
||||||
|
|
||||||
# 调用LLM
|
# 调用LLM
|
||||||
|
|||||||
@@ -41,7 +41,11 @@ classification_info="""【垂直领域分类】:
|
|||||||
2. 数据问题:涉及电力造价费用、造价指标等
|
2. 数据问题:涉及电力造价费用、造价指标等
|
||||||
|
|
||||||
【安装下载注册包括以下三类】:
|
【安装下载注册包括以下三类】:
|
||||||
1. 后缀名查询:询问有关软件后缀名、工程文件扩展名等问题,例如:BDY3是什么文件?、用什么软件打开.BDY3文件?
|
1. 后缀名咨询:所有涉及文件扩展名的使用场景、软件关联等问题,包括但不限于:询问文件是否由特定软件打开、扩展名与软件的匹配关系、扩展名含义及关联等
|
||||||
|
示例:
|
||||||
|
"这个文件用配网软件能打开吗?(隐含扩展名关联)",
|
||||||
|
".bphq18 是什么类型的文件?",
|
||||||
|
"用哪个软件打开.BDY3文件?"
|
||||||
2. 软件锁类:询问软件锁信息、锁注册号查询、许可证查询、锁激活问题等软件锁相关问题
|
2. 软件锁类:询问软件锁信息、锁注册号查询、许可证查询、锁激活问题等软件锁相关问题
|
||||||
3. 安装下载类:安装下载咨询、组件(插件)选择、环境配置等
|
3. 安装下载类:安装下载咨询、组件(插件)选择、环境配置等
|
||||||
4. 问题排查类:软件安装下载失败、报错,系统兼容性问题等
|
4. 问题排查类:软件安装下载失败、报错,系统兼容性问题等
|
||||||
@@ -62,10 +66,10 @@ classification_prompt="""
|
|||||||
【示例】
|
【示例】
|
||||||
用户输入1: 技改T1怎样新建工程
|
用户输入1: 技改T1怎样新建工程
|
||||||
输出1:
|
输出1:
|
||||||
{
|
{{
|
||||||
"vertical_classification":"软件咨询",
|
"vertical_classification":"软件咨询",
|
||||||
"sub_classification":"软件功能"
|
"sub_classification":"软件功能"
|
||||||
}
|
}}
|
||||||
"""
|
"""
|
||||||
|
|
||||||
query_rewrite_prompt = """
|
query_rewrite_prompt = """
|
||||||
|
|||||||
Reference in New Issue
Block a user