bugfix(ChatExcel): ChatExcel Language confusion bug (#732)

1.Fix ChatExcel Language confusion bug #719
eosphoros-ai · Oct 26, 2023 · 2ba610c · 2ba610c
2 parents 02de3d4 + c0421d0
commit 2ba610c
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/pilot/scene/chat_data/chat_excel/excel_learning/prompt.py b/pilot/scene/chat_data/chat_excel/excel_learning/prompt.py
@@ -33,11 +33,27 @@
     {response}
 """
 
-RESPONSE_FORMAT_SIMPLE = {
+_RESPONSE_FORMAT_SIMPLE_ZH = {
     "DataAnalysis": "数据内容分析总结",
     "ColumnAnalysis": [{"column name1": "字段1介绍，专业术语解释(请尽量简单明了)"}],
     "AnalysisProgram": ["1.分析方案1，图表展示方式1", "2.分析方案2，图表展示方式2"],
 }
+_RESPONSE_FORMAT_SIMPLE_EN = {
+    "DataAnalysis": "Data content analysis summary",
+    "ColumnAnalysis": [
+        {
+            "column name1": "Introduction to Column 1 and explanation of professional terms (please try to be as simple and clear as possible)"
+        }
+    ],
+    "AnalysisProgram": [
+        "1. Analysis plan 1, chart display type 1",
+        "2. Analysis plan 2, chart display type 2",
+    ],
+}
+
+RESPONSE_FORMAT_SIMPLE = (
+    _RESPONSE_FORMAT_SIMPLE_EN if CFG.LANGUAGE == "en" else _RESPONSE_FORMAT_SIMPLE_ZH
+)
 
 
 _DEFAULT_TEMPLATE = (