From 4abfa3c7e9979391df99d26b3c723a1a042eec85 Mon Sep 17 00:00:00 2001
From: "tuyang.yhj" <tuyang.yhj@alibaba-inc.com>
Date: Thu, 29 Jun 2023 18:53:52 +0800
Subject: [PATCH] WEB API independent

---
 pilot/connections/rdbms/py_study/study_data.py         | 10 ++++++++++
 pilot/scene/chat_dashboard/chat.py                     |  6 ++++++
 .../chat_dashboard/data_preparation/report_schma.py    |  2 ++
 3 files changed, 18 insertions(+)
 create mode 100644 pilot/connections/rdbms/py_study/study_data.py

diff --git a/pilot/connections/rdbms/py_study/study_data.py b/pilot/connections/rdbms/py_study/study_data.py
new file mode 100644
index 000000000..c19c39f33
--- /dev/null
+++ b/pilot/connections/rdbms/py_study/study_data.py
@@ -0,0 +1,10 @@
+from pilot.common.sql_database import Database
+from pilot.configs.config import Config
+
+CFG = Config()
+
+if __name__ == "__main__":
+    connect = CFG.local_db.get_session("gpt-user")
+    datas = CFG.local_db.run(connect, "SELECT * FROM users; ")
+
+    print(datas)
\ No newline at end of file
diff --git a/pilot/scene/chat_dashboard/chat.py b/pilot/scene/chat_dashboard/chat.py
index a9f84e431..dd614399c 100644
--- a/pilot/scene/chat_dashboard/chat.py
+++ b/pilot/scene/chat_dashboard/chat.py
@@ -65,6 +65,12 @@ class ChatDashboard(BaseChat):
             try:
                 datas = self.database.run(self.db_connect, chart_item.sql)
                 chart_data: ChartData = ChartData()
+                chart_data.chart_sql = chart_item['sql']
+                chart_data.chart_type = chart_item['showcase']
+                chart_data.chart_name = chart_item['title']
+                chart_data.chart_desc = chart_item['thoughts']
+                chart_data.column_name = datas[0]
+                chart_data.values =datas
             except Exception as e:
                 # TODO 修复流程
                 print(str(e))
diff --git a/pilot/scene/chat_dashboard/data_preparation/report_schma.py b/pilot/scene/chat_dashboard/data_preparation/report_schma.py
index 84468f02e..9327fc4d5 100644
--- a/pilot/scene/chat_dashboard/data_preparation/report_schma.py
+++ b/pilot/scene/chat_dashboard/data_preparation/report_schma.py
@@ -4,7 +4,9 @@ from typing import TypeVar, Union, List, Generic, Any
 
 class ChartData(BaseModel):
     chart_uid: str
+    chart_name: str
     chart_type: str
+    chart_desc: str
     chart_sql: str
     column_name: List
     values: List