github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-30 15:21:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(agent): The SQL execution result of the DataScientist can be passed to the next agent (#2654)

Browse Source 
Co-authored-by: tam <tanwe@fulan.com.cn>
This commit is contained in:
Tam 2025-04-29 19:28:09 +08:00 committed by GitHub
parent 2abd68d6c0
commit b901cbc9a6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -4,7 +4,7 @@ import json
import logging
from typing import List, Optional
from dbgpt._private.pydantic import BaseModel, Field, model_to_json
from dbgpt._private.pydantic import BaseModel, Field, model_to_dict, model_to_json
from dbgpt.vis.tags.vis_chart import Vis, VisChart
from ...core.action.base import Action, ActionOutput
@ -85,9 +85,16 @@ class ChartAction(Action[SqlInput]):
chart=json.loads(model_to_json(param)), data_df=data_df
)
param_dict = model_to_dict(param)
if not data_df.empty:
param_dict["data"] = json.loads(
data_df.to_json(orient="records", date_format="iso", date_unit="s")
)
content = json.dumps(param_dict)
return ActionOutput(
is_exe_success=True,
content=model_to_json(param),
content=content,
view=view,
resource_type=self.resource_need.value,
resource_value=db._db_name,