langchain一个agent集成sql查询和Python代码执行能力

01_langchain_study/01_langchain_task.py

@@ -579,15 +579,10 @@ if __name__ == "__main__":
     print("=" * 60)
     
     # 测试: 绘制图表
-    # print("\n【测试】绘制图表 - 各部门薪资分布柱状图:")
-    # result = run_data_analysis(
-    #     "查询各部门的平均薪资，并用 matplotlib 绘制柱状图，"
-    #     "图表标题为'各部门平均薪资分布'，x轴为部门名称，y轴为平均薪资，"
-    #     "保存图表到 d:/agentlearning/lqq-agent-study/salary_chart.png"
-    # )
-        # 测试: 绘制图表
-    print("\n【直接回答版:")
+    print("\n【测试】绘制图表 - 各部门薪资分布柱状图:")
     result = run_data_analysis(
-        "今天上海天气怎么样，会来台风吗"
+        "查询各部门的平均薪资，并用 matplotlib 绘制柱状图，"
+        "图表标题为'各部门平均薪资分布'，x轴为部门名称，y轴为平均薪资，"
+        "保存图表到 d:/agentlearning/lqq-agent-study/salary_chart.png"
     )
     print(result)