From 477804769cec75c9a6c6e59b1b58b281f028a451 Mon Sep 17 00:00:00 2001 From: tanwei Date: Sat, 5 Jul 2025 23:53:24 +0800 Subject: [PATCH 1/2] feat(agent): Add count field and support non-ASCII characters in chart action output (#2807) --- .../src/dbgpt/agent/expand/actions/chart_action.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/packages/dbgpt-core/src/dbgpt/agent/expand/actions/chart_action.py b/packages/dbgpt-core/src/dbgpt/agent/expand/actions/chart_action.py index 6af1092f5..16f175d62 100644 --- a/packages/dbgpt-core/src/dbgpt/agent/expand/actions/chart_action.py +++ b/packages/dbgpt-core/src/dbgpt/agent/expand/actions/chart_action.py @@ -87,9 +87,17 @@ class ChartAction(Action[SqlInput]): param_dict = model_to_dict(param) if not data_df.empty: + param_dict["count"] = len(data_df) + param_dict["data"] = json.loads( - data_df.to_json(orient="records", date_format="iso", date_unit="s") + data_df.to_json( + orient="records", + force_ascii=False, + date_format="iso", + date_unit="s", + ) ) + content = json.dumps(param_dict) return ActionOutput( From 0bac938975a069f75fc64344165316cd999eb5aa Mon Sep 17 00:00:00 2001 From: tanwei Date: Mon, 7 Jul 2025 01:27:03 +0800 Subject: [PATCH 2/2] feat(agent): Support non-ASCII characters in chart action JSON output --- .../dbgpt-core/src/dbgpt/agent/expand/actions/chart_action.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/dbgpt-core/src/dbgpt/agent/expand/actions/chart_action.py b/packages/dbgpt-core/src/dbgpt/agent/expand/actions/chart_action.py index 16f175d62..c3b50b916 100644 --- a/packages/dbgpt-core/src/dbgpt/agent/expand/actions/chart_action.py +++ b/packages/dbgpt-core/src/dbgpt/agent/expand/actions/chart_action.py @@ -98,7 +98,7 @@ class ChartAction(Action[SqlInput]): ) ) - content = json.dumps(param_dict) + content = json.dumps(param_dict, ensure_ascii=False) return ActionOutput( is_exe_success=True,