github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-06 19:04:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(agent): Add count field and support non-ASCII characters in chart action output (#2807)

Browse Source
This commit is contained in:
tanwei 2025-07-05 23:53:24 +08:00
parent d27fdb7928
commit 477804769c
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
packages/dbgpt-core/src/dbgpt/agent/expand/actions/chart_action.py
View File

@ -87,9 +87,17 @@ class ChartAction(Action[SqlInput]):
param_dict = model_to_dict(param) param_dict = model_to_dict(param)
if not data_df.empty: if not data_df.empty:
param_dict["count"] = len(data_df)
param_dict["data"] = json.loads( 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)
return ActionOutput( return ActionOutput(