|
@@ -75,7 +75,6 @@ class CompleteAgentFlow:
|
|
|
# 添加节点
|
|
# 添加节点
|
|
|
workflow.add_node("planning_node", self._planning_node)
|
|
workflow.add_node("planning_node", self._planning_node)
|
|
|
workflow.add_node("outline_generator", self._outline_generator_node)
|
|
workflow.add_node("outline_generator", self._outline_generator_node)
|
|
|
- workflow.add_node("metric_calculator", self._metric_calculator_node)
|
|
|
|
|
workflow.add_node("data_classify", self._data_classify_node)
|
|
workflow.add_node("data_classify", self._data_classify_node)
|
|
|
|
|
|
|
|
# 设置入口点
|
|
# 设置入口点
|
|
@@ -96,7 +95,7 @@ class CompleteAgentFlow:
|
|
|
# 从各个节点返回规划节点重新决策
|
|
# 从各个节点返回规划节点重新决策
|
|
|
workflow.add_edge("data_classify", "planning_node")
|
|
workflow.add_edge("data_classify", "planning_node")
|
|
|
workflow.add_edge("outline_generator", "planning_node")
|
|
workflow.add_edge("outline_generator", "planning_node")
|
|
|
- workflow.add_edge("metric_calculator", END)
|
|
|
|
|
|
|
+ workflow.add_edge("metric_calculator", "planning_node")
|
|
|
|
|
|
|
|
return workflow
|
|
return workflow
|
|
|
|
|
|
|
@@ -419,7 +418,9 @@ class CompleteAgentFlow:
|
|
|
"intent_category": "指标计算"
|
|
"intent_category": "指标计算"
|
|
|
}
|
|
}
|
|
|
print(f" 使用知识ID: {config_name}")
|
|
print(f" 使用知识ID: {config_name}")
|
|
|
- results = await self.rules_engine_agent.calculate_metrics(intent_result)
|
|
|
|
|
|
|
+ # 将打好标的数据集传入指标计算函数中
|
|
|
|
|
+ data_set_classified = state.get("data_set_classified", [])
|
|
|
|
|
+ results = await self.rules_engine_agent.calculate_metrics(intent_result, data_set_classified)
|
|
|
else:
|
|
else:
|
|
|
# 使用传统指标计算（模拟）
|
|
# 使用传统指标计算（模拟）
|
|
|
# 这里简化处理，实际应该根据配置文件调用相应的API
|
|
# 这里简化处理，实际应该根据配置文件调用相应的API
|
