wehub-resource-sync
6188617037
Environment Corruption Check / test-python-versions (3.12.8) (push) Failing after 2s
Environment Corruption Check / test-python-versions (3.11.11) (push) Failing after 1s
Pre-commit checks / pre-commit-check (push) Failing after 1s
Environment Corruption Check / test-python-versions (3.13.2) (push) Failing after 4s
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from pydantic import Field
|
|
|
|
from app.agent.toolcall import ToolCallAgent
|
|
from app.config import config
|
|
from app.prompt.visualization import NEXT_STEP_PROMPT, SYSTEM_PROMPT
|
|
from app.tool import Terminate, ToolCollection
|
|
from app.tool.chart_visualization.chart_prepare import VisualizationPrepare
|
|
from app.tool.chart_visualization.data_visualization import DataVisualization
|
|
from app.tool.chart_visualization.python_execute import NormalPythonExecute
|
|
|
|
|
|
class DataAnalysis(ToolCallAgent):
|
|
"""
|
|
A data analysis agent that uses planning to solve various data analysis tasks.
|
|
|
|
This agent extends ToolCallAgent with a comprehensive set of tools and capabilities,
|
|
including Data Analysis, Chart Visualization, Data Report.
|
|
"""
|
|
|
|
name: str = "Data_Analysis"
|
|
description: str = "An analytical agent that utilizes python and data visualization tools to solve diverse data analysis tasks"
|
|
|
|
system_prompt: str = SYSTEM_PROMPT.format(directory=config.workspace_root)
|
|
next_step_prompt: str = NEXT_STEP_PROMPT
|
|
|
|
max_observe: int = 15000
|
|
max_steps: int = 20
|
|
|
|
# Add general-purpose tools to the tool collection
|
|
available_tools: ToolCollection = Field(
|
|
default_factory=lambda: ToolCollection(
|
|
NormalPythonExecute(),
|
|
VisualizationPrepare(),
|
|
DataVisualization(),
|
|
Terminate(),
|
|
)
|
|
)
|