|
|
@@ -439,7 +439,7 @@ class OutlineGeneratorAgent:
|
|
|
# 基于用户查询进行语义匹配，找出缺失的关键指标
|
|
|
query_keywords = self._extract_query_keywords(question, industry)
|
|
|
missing_key_metrics = self._find_missing_key_metrics(
|
|
|
- available_metrics, selected_metric_names, query_keywords
|
|
|
+ available_metrics, selected_metric_names, query_keywords,industry
|
|
|
)
|
|
|
|
|
|
# 补充缺失的关键指标
|
|
|
@@ -520,7 +520,7 @@ class OutlineGeneratorAgent:
|
|
|
return words
|
|
|
|
|
|
def _find_missing_key_metrics(self, available_metrics: List[Dict], selected_metric_names: set,
|
|
|
- query_keywords: List[str]) -> List[str]:
|
|
|
+ query_keywords: List[str],industry: str) -> List[str]:
|
|
|
"""
|
|
|
基于查询关键词找出缺失的关键指标
|
|
|
|
|
|
@@ -539,6 +539,8 @@ class OutlineGeneratorAgent:
|
|
|
missing_prompt = ChatPromptTemplate.from_messages([
|
|
|
("system", """
|
|
|
你是一个专业的指标推荐专家，需要根据用户查询的关键词，识别出可能缺失的关键指标。
|
|
|
+
|
|
|
+ 强制要求：只能选择和{industry}相关的指标
|
|
|
|
|
|
请分析：
|
|
|
1. 用户关心的分析维度（收入、支出、排名、趋势等）
|
|
|
@@ -573,7 +575,8 @@ class OutlineGeneratorAgent:
|
|
|
result = chain.invoke({
|
|
|
"keywords": ', '.join(query_keywords),
|
|
|
"selected_metrics": selected_list,
|
|
|
- "available_metrics": available_list
|
|
|
+ "available_metrics": available_list,
|
|
|
+ "industry": industry
|
|
|
})
|
|
|
|
|
|
# 解析结果
|
