github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-31 15:47:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: dashboard editor run error (#1380)

Browse Source
This commit is contained in:
Aries-ckt 2024-04-07 15:04:28 +08:00 committed by GitHub
parent f2a6284c0e
commit d4da50330f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 23 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dbgpt/app/openapi/api_v1/editor/api_editor_v1.py
View File

@ -24,6 +24,7 @@ from dbgpt.app.openapi.editor_view_model import (
from dbgpt.app.scene import ChatFactory
from dbgpt.app.scene.chat_dashboard.data_loader import DashboardDataLoader
from dbgpt.serve.conversation.serve import Serve as ConversationServe
from dbgpt.util.date_utils import convert_datetime_in_row
from ._chat_history.chat_hisotry_factory import ChatHistory
@ -173,9 +174,8 @@ async def editor_chart_run(run_param: dict = Body()):
field_names, chart_values = dashboard_data_loader.get_chart_values_by_data(
colunms, sql_result, sql
)
start_time = time.time() * 1000
# 计算执行耗时
sql_result = [convert_datetime_in_row(row) for row in sql_result]
end_time = time.time() * 1000
sql_run_data: SqlRunData = SqlRunData(
result_info="",

20
dbgpt/app/scene/chat_dashboard/data_loader.py
View File

@ -40,16 +40,16 @@ class DashboardDataLoader:
)
for field_name in field_names[1:]:
if not field_map[field_name]:
logger.info("More than 2 non-numeric column:" + field_name)
else:
for data in datas:
value_item = ValueItem(
name=data[0],
type=field_name,
value=data[field_names.index(field_name)],
)
values.append(value_item)
# if not field_map[field_name]:
# logger.info("More than 2 non-numeric column:" + field_name)
# else:
for data in datas:
value_item = ValueItem(
name=data[0],
type=field_name,
value=str(data[field_names.index(field_name)]),
)
values.append(value_item)
return field_names, values
except Exception as e:
logger.debug("Prepare Chart Data Failed!" + str(e))

11
dbgpt/util/date_utils.py Normal file
View File

@ -0,0 +1,11 @@
import datetime
def is_datetime(value):
return isinstance(value, datetime.datetime)
def convert_datetime_in_row(row):
return [value.strftime('%Y-%m-%d %H:%M:%S') if is_datetime(value)
else value for value in row]