From b901cbc9a615b4188755febefc959a0f5d68c1e3 Mon Sep 17 00:00:00 2001 From: Tam Date: Tue, 29 Apr 2025 19:28:09 +0800 Subject: [PATCH] feat(agent): The SQL execution result of the DataScientist can be passed to the next agent (#2654) Co-authored-by: tam --- .../src/dbgpt/agent/expand/actions/chart_action.py | 11 +++++++++-- 1 file changed, 9 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 b2f9386a0..6af1092f5 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 @@ -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,