Visualization

VisualizationTools enables agents to create various types of charts and plots using matplotlib.

Example

The following agent can create various types of data visualizations:

1from kern.agent import Agent
2from kern.tools.visualization import VisualizationTools
3
4agent = Agent(
5 instructions=[
6 "You are a data visualization assistant that creates charts and plots",
7 "Generate clear, informative visualizations based on user data",
8 "Save charts to files and provide insights about the data",
9 "Choose appropriate chart types for different data patterns",
10 ],
11 tools=[VisualizationTools(output_dir="my_charts")],
12)
13
14agent.print_response("Create a bar chart showing sales by quarter: Q1=100, Q2=150, Q3=120, Q4=180", stream=True)

Toolkit Params

ParameterTypeDefaultDescription
output_dirstr"charts"Directory to save generated charts.
enable_create_bar_chartboolTrueEnable bar chart creation.
enable_create_line_chartboolTrueEnable line chart creation.
enable_create_pie_chartboolTrueEnable pie chart creation.
enable_create_scatter_plotboolTrueEnable scatter plot creation.
enable_create_histogramboolTrueEnable histogram creation.

Toolkit Functions

FunctionDescription
create_bar_chartCreate bar charts for categorical data comparison.
create_line_chartCreate line charts for time series and trend visualization.
create_pie_chartCreate pie charts for proportional data representation.
create_scatter_plotCreate scatter plots for correlation analysis.
create_histogramCreate histograms for data distribution visualization.

Developer Resources