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

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

Browse Source 
Co-authored-by: tanwei <tanwei@fulan.com.cn>
This commit is contained in:
Tam 2025-07-23 10:58:45 +08:00 committed by GitHub
parent bcb4eb3983
commit fd6572322c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -87,10 +87,18 @@ 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, ensure_ascii=False)
return ActionOutput( return ActionOutput(
is_exe_success=True, is_exe_success=True,