Update marketing assistant model name to "gpt-4o

coolbeevip · Jul 10, 2024 · 8d8436a · 8d8436a
1 parent 33b3fb7
commit 8d8436a
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 3 deletions.
diff --git a/src/langchain_lab/langgraph/marketing_analysis_assistant.py b/src/langchain_lab/langgraph/marketing_analysis_assistant.py
@@ -203,7 +203,7 @@ def run(self):
         with open(f"sales_analysis_report_{self.model_name}.md", "w") as f:
             f.write("# 市场部销售智能助手（POC）\n\n")
             f.write(f"> {self.model_name}\n\n")
-            f.write("![image - 20240710141823753](assets / marketing_analysis_assistant.png)\n\n")
+            f.write("![image-20240710141823753](assets/marketing_analysis_assistant.png)\n\n")
             f.write("## 多代理协商过程\n\n")
             for s in self.graph.stream(
                 {

diff --git a/tests/langchain_lab/langgraph/sales_analysis_report_gpt-3.5-turbo.md b/tests/langchain_lab/langgraph/sales_analysis_report_gpt-3.5-turbo.md
@@ -2,7 +2,7 @@
 
 > gpt-3.5-turbo
 
-![image - 20240710141823753](assets / marketing_analysis_assistant.png)
+![image-20240710141823753](assets/marketing_analysis_assistant.png)
 
 ## 多代理协商过程
 

diff --git a/tests/langchain_lab/langgraph/sales_analysis_report_gpt-4o.md b/tests/langchain_lab/langgraph/sales_analysis_report_gpt-4o.md
@@ -0,0 +1,30 @@
+# 市场部销售智能助手（POC）
+
+> gpt-4o
+
+![image-20240710141823753](assets/marketing_analysis_assistant.png)
+
+## 多代理协商过程
+
+### 销售员
+
+{'function_call': {'arguments': '{}', 'name': 'load_sales_data_tool'}}
+
+
+### 数据分析工具
+
+load_sales_data_tool response: Sales Data:
+
+|    |   Year |   Product_A |   Product_B |   Product_C |
+|---:|-------:|------------:|------------:|------------:|
+|  0 |   2019 |        1091 |         540 |         802 |
+|  1 |   2020 |         979 |         527 |         782 |
+|  2 |   2021 |         970 |         521 |         816 |
+|  3 |   2022 |        1070 |         559 |         752 |
+|  4 |   2023 |        1059 |         533 |         730 |
+
+### 销售员
+
+{'function_call': {'arguments': 'import pandas as pd\n\n# Load the sales data from the CSV file\ndata = {\n    \'Year\': [2019, 2020, 2021, 2022, 2023],\n    \'Product_A\': [1091, 979, 970, 1070, 1059],\n    \'Product_B\': [540, 527, 521, 559, 533],\n    \'Product_C\': [802, 782, 816, 752, 730]\n}\ndf = pd.DataFrame(data)\n\n# Perform basic statistical analysis\nstatistics = df.describe()\n\n# Calculate correlation matrix\ncorrelation_matrix = df.corr()\n\n# Print the results\nprint("Basic Statistical Analysis:\\n", statistics)\nprint("\\nCorrelation Matrix:\\n", correlation_matrix)\n\n# Provide insights based on the analysis\ninsights = """\nInsights:\n1. Product_A has the highest average sales over the past 5 years.\n2. Product_B has the lowest average sales over the past 5 years.\n3. Product_A sales peaked in 2019, whereas Product_C sales peaked in 2021.\n4. There is a moderate positive correlation between Product_A and Product_B sales.\n5. Product_C sales have shown a declining trend over the past 3 years.\n"""\n\nprint(insights)\n', 'name': 'data_analysis_tool'}}
+
+
diff --git a/tests/langchain_lab/langgraph/test_marketing_analysis_assistant.py b/tests/langchain_lab/langgraph/test_marketing_analysis_assistant.py
@@ -14,7 +14,7 @@ class TestMarketingAnalysisAssistant(TestCase):
     def test_main(self):
         assistant = MarketingAnalysisAssistant(openai_api_base=os.environ["OPENAI_API_BASE"],
                                                openai_api_key=os.environ["OPENAI_API_KEY"],
-                                               model_name="gpt-3.5-turbo",
+                                               model_name="gpt-4o",
                                                recursion_limit=50)
         assistant.run()