From fd6572322c1712ce6808c1bc1c22a8710c68c40a Mon Sep 17 00:00:00 2001 From: Tam Date: Wed, 23 Jul 2025 10:58:45 +0800 Subject: [PATCH] feat(agent): Add count field and support non-ASCII characters in DataScientist action output (#2812) Co-authored-by: tanwei --- .../src/dbgpt/agent/expand/actions/chart_action.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) 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..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 @@ -87,10 +87,18 @@ 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) + + content = json.dumps(param_dict, ensure_ascii=False) return ActionOutput( is_exe_success=True,