From 477804769cec75c9a6c6e59b1b58b281f028a451 Mon Sep 17 00:00:00 2001 From: tanwei Date: Sat, 5 Jul 2025 23:53:24 +0800 Subject: [PATCH] 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(