github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-02 00:28:00 +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)
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,