Merge pull request #73 from DeepInsight-AI/new_pre

V1.2.2

Author: LiChengxin007

ai/agents/agent_instance_util.py

@@ -1,9 +1,10 @@
 from .prompt_matplotlib import MYSQL_MATPLOTLIB_TIPS_MESS
-from .prompt_echarts import CSV_ECHART_TIPS_MESS, MYSQL_ECHART_TIPS_MESS, POSTGRESQL_ECHART_TIPS_MESS
+from .prompt_echarts import CSV_ECHART_TIPS_MESS, MYSQL_ECHART_TIPS_MESS, POSTGRESQL_ECHART_TIPS_MESS, MONGODB_ECHART_TIPS_MESS
 
 __all__ = [
     "CSV_ECHART_TIPS_MESS",
     "MYSQL_ECHART_TIPS_MESS",
     "MYSQL_MATPLOTLIB_TIPS_MESS",
-    "POSTGRESQL_ECHART_TIPS_MESS"
+    "POSTGRESQL_ECHART_TIPS_MESS",
+    "MONGODB_ECHART_TIPS_MESS"
 ]

ai/agents/prompt/__init__.py

@@ -335,3 +335,67 @@
                         ].
 
 '''
+
+MONGODB_ECHART_TIPS_MESS = '''
+Here are some examples of generating mongodb and pyecharts Code based on the given question.
+Please generate new one based on the data and question human asks you, import the neccessary libraries and make sure the code is correct.
+
+IMPORTANT: You need to follow the coding style, and the type of the x, y axis. But also need to focus on the column name of the uploaded tables(if exists). Generally, PyEcharts does not accept numpy.int or numpy.float, etc. It only supports built-in data type like int, float, and str.
+Pay attention to check whether the query statement in the execution code block can correctly query the data.
+
+
+    Q: A `stacked` line chart comparing sales and profit over time would be useful. Could you help plot it?
+    Note: stacked line chart is more fancy in display, while the former is more neat.
+    <code>
+    import pymongo
+    import pandas as pd
+    from pyecharts.charts import Bar
+    from pyecharts import options as opts
+    import json
+
+
+    connectionString = "mongodb://your_host:your_port/your_dbname"
+    kwargs = {
+        'username': 'your_username',
+        'password': 'your_password',
+    }
+
+    db_connection = pymongo.MongoClient(
+        connectionString, **kwargs
+    )
+    conn = db_connection["your_dbname"]
+    res = list(conn['your_table'].find())
+    years = [str(_['year']) for _ in res]
+    sales = [str(_['sales']) for _ in res]
+    bar = Bar()
+    bar.add_xaxis(years)
+    bar.add_yaxis("Sales", sales)
+    bar.set_global_opts(
+        xaxis_opts=opts.AxisOpts(
+            type_="category",
+            name="Year",
+        ),
+        yaxis_opts=opts.AxisOpts(
+            type_="value",
+            name="Sales",
+        ),
+        title_opts=opts.TitleOpts(title="Sales over Years"),
+    )
+
+    ret_json = bar.dump_options()
+    echart_code = json.loads(ret_json)
+
+    out_put = [{"echart_name": "Sales over Years", "echart_code": echart_code}]
+    print(out_put)
+    </code>
+
+
+      The output should be formatted as a JSON instance that conforms to the JSON schema below, the JSON is a list of dict,
+                        [
+                        {"echart_name": "Sales over Years", "echart_code": ret_json}
+                        {},
+                        {},
+                        ].
+
+
+'''

ai/agents/prompt/prompt_echarts.py

@@ -2,3 +2,4 @@
 from .analysis_csv import AnalysisCsv
 from .analysis_pg import AnalysisPostgresql
 from .analysis_sr import AnalysisStarrocks
+from .analysis_mongodb import AnalysisMongoDB

ai/backend/aidb/analysis/__init__.py

@@ -1,3 +1,5 @@
 from .autopilot_mysql import AutopilotMysql
 from .autopilot import Autopilot
 from .autopilot_starrocks_api import AutopilotStarrocks
+from .autopilot_mongodb import AutopilotMongoDB
+

ai/backend/aidb/analysis/analysis_mongodb.py

@@ -4,6 +4,7 @@
 from jinja2 import Template
 from pathlib import Path
 import time
+import os
 
 class Autopilot(AIDB):
     def get_agent_user_proxy(self):
@@ -153,23 +154,25 @@ def generate_report_template(self, data):
                 data['report_thought'].remove(item)
 
         # 获取当前工作目录的路径
-        current_directory = Path.cwd()
+        current_directory = CONFIG.up_file_path 
 
-        if str(current_directory).endswith('/ai'):
-            html_template_path = str(current_directory) + '/backend/aidb/autopilot/'
-        else:
-            html_template_path = str(current_directory) + '/ai/backend/aidb/autopilot/'
-
-        print('html_template_path:', html_template_path)
+        # 构建路径时使用 os.path.join，并使用 os.path.normpath 进行规范化
+        html_template_path = os.path.join(os.path.normpath(current_directory.replace(r'\user_upload_files', '')), 'ai', 'backend', 'aidb', 'autopilot')
+        html_template_path = html_template_path.replace('\\', '/')
 
         if CONFIG.web_language == 'CN':
-            html_template_path = html_template_path + 'html_template'
+            html_template_path = os.path.join(html_template_path, 'html_template')
         else:
-            html_template_path = html_template_path + 'html_template_en'
+            html_template_path = os.path.join(html_template_path, 'html_template_en')
 
         # 读取模板文件
-        with open(html_template_path + '/report_2.html', 'r') as file:
+        template_file_path = os.path.join(html_template_path, 'report_2.html')
+
+        with open(template_file_path, 'r', encoding='utf-8') as file:
             template_str = file.read()
+
+        print('html_template_path:', html_template_path)
+        print('template_str:', template_str)
             # print('template_str :', template_str)
 
         # 使用Jinja2渲染模板