ai-data-science-team 0.0.0.9008__tar.gz → 0.0.0.9009__tar.gz

{ai_data_science_team-0.0.0.9008/ai_data_science_team.egg-info → ai_data_science_team-0.0.0.9009}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ai-data-science-team
-Version: 0.0.0.9008
+Version: 0.0.0.9009
 Summary: Build and run an AI-powered data science team.
 Home-page: https://github.com/business-science/ai-data-science-team
 Author: Matt Dancho
@@ -21,17 +21,23 @@ Requires-Dist: langchain_experimental
 Requires-Dist: langgraph>=0.2.57
 Requires-Dist: openai
 Requires-Dist: pandas
+Requires-Dist: sqlalchemy
 Requires-Dist: numpy
 Requires-Dist: plotly
 Requires-Dist: streamlit
 Requires-Dist: scikit-learn
 Requires-Dist: xgboost
+Provides-Extra: machine-learning-agent
+Requires-Dist: h2o; extra == "machine-learning-agent"
+Provides-Extra: all
+Requires-Dist: h2o; extra == "all"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
 Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
+Dynamic: provides-extra
 Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
@@ -47,13 +53,13 @@ Dynamic: summary
   <em>An AI-powered data science team of agents to help you perform common data science tasks 10X faster</em>
 </div>
 <div align="center">
-  <a href="https://pypi.python.org/pypi/ai-data-science-team"><img src="https://img.shields.io/pypi/v/ai-data-science-team.svg" alt="PyPI"></a>
-  <a href="https://github.com/business-science/ai-data-science-team"><img src="https://img.shields.io/pypi/pyversions/ai-data-science-team.svg" alt="versions"></a>
-  <a href="https://github.com/business-science/ai-data-science-team/blob/main/LICENSE"><img src="https://img.shields.io/github/license/business-science/ai-data-science-team.svg?v" alt="license"></a>
+  <a href="https://pypi.python.org/pypi/ai-data-science-team"><img src="https://img.shields.io/pypi/v/ai-data-science-team.svg?style=for-the-badge" alt="PyPI"></a>
+  <a href="https://github.com/business-science/ai-data-science-team"><img src="https://img.shields.io/pypi/pyversions/ai-data-science-team.svg?style=for-the-badge" alt="versions"></a>
+  <a href="https://github.com/business-science/ai-data-science-team/blob/main/LICENSE"><img src="https://img.shields.io/github/license/business-science/ai-data-science-team.svg?style=for-the-badge" alt="license"></a>
 </div>
 
 
-# Your AI Data Science Team (An Army Of Agents)
+# Your AI Data Science Team (🪖 An Army Of Agents)
 
 **An AI-powered data science team of agents to help you perform common data science tasks 10X faster**.
 
@@ -74,14 +80,16 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
 
 ## Table of Contents
 
-- [Your AI Data Science Team (An Army Of Agents)](#your-ai-data-science-team-an-army-of-agents)
+- [Your AI Data Science Team (🪖 An Army Of Agents)](#your-ai-data-science-team--an-army-of-agents)
   - [Table of Contents](#table-of-contents)
   - [Companies That Want A Custom AI Data Science Team (And AI Apps)](#companies-that-want-a-custom-ai-data-science-team-and-ai-apps)
-  - [Free How To Build AI Agents for Data Scientists Workshop](#free-how-to-build-ai-agents-for-data-scientists-workshop)
+  - [Generative AI for Data Scientists Workshop](#generative-ai-for-data-scientists-workshop)
   - [Data Science Agents](#data-science-agents)
-    - [Coming Soon: Multi-Agents](#coming-soon-multi-agents)
-    - [...And after that, the Multi-Agent Data Science Apps](#and-after-that-the-multi-agent-data-science-apps)
+    - [NEW: Multi-Agents](#new-multi-agents)
+    - [Coming Soon: Data Science Apps](#coming-soon-data-science-apps)
     - [Agents Available Now](#agents-available-now)
+      - [Data Science Agents](#data-science-agents-1)
+      - [Multi-Agents](#multi-agents)
     - [Agents Coming Soon](#agents-coming-soon)
   - [Disclaimer](#disclaimer)
   - [Installation](#installation)
@@ -94,11 +102,11 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
 
 ## Companies That Want A Custom AI Data Science Team (And AI Apps)
 
-Want to have your own _customized_ enterprise-grade AI Data Science Team and domain-specifici AI-powered Apps? 
+Want to have your own _customized_ enterprise-grade AI Data Science Team and *domain-specific* AI-powered Apps? 
 
 **Send inquiries here:** [https://www.business-science.io/contact.html](https://www.business-science.io/contact.html)
 
-## Free How To Build AI Agents for Data Scientists Workshop
+## Generative AI for Data Scientists Workshop
 
 If you're an aspiring data scientist who wants to learn how to build AI Agents and AI Apps for your company that performs Data Science, Business Intelligence, Churn Modeling, Time Series Forecasting, and more, then I'd love to help you. 
 
@@ -110,13 +118,13 @@ This project is a work in progress. New data science agents will be released soo
 
 ![Data Science Team](/img/ai_data_science_team.jpg)
 
-### Coming Soon: Multi-Agents
+### NEW: Multi-Agents
 
-This is the internals of the Business Intelligence SQL Agent I'm working on:
+This is the internals of the SQL Data Analyst Agent that connects to SQL databases to pull data into the data science environment. It creates pipelines to automate data extraction, performs Joins, Aggregations, and other SQL Query operations. And it includes a Data Visualization Agent that creates visualizations to help you understand your data.:
 
 ![Business Intelligence SQL Agent](/img/multi_agent_sql_data_visualization.jpg)
 
-### ...And after that, the Multi-Agent Data Science Apps
+### Coming Soon: Data Science Apps
 
 This is a top secret project I'm working on. It's a multi-agent data science app that performs time series forecasting.
 
@@ -124,12 +132,18 @@ This is a top secret project I'm working on. It's a multi-agent data science app
 
 ### Agents Available Now
 
+#### Data Science Agents
+
 1. **Data Wrangling Agent:** Merges, Joins, Preps and Wrangles data into a format that is ready for data analysis.
 2. **Data Visualization Agent:** Creates visualizations to help you understand your data. Returns JSON serializable plotly visualizations.
 3. **Data Cleaning Agent:** Performs Data Preparation steps including handling missing values, outliers, and data type conversions.
 4. **Feature Engineering Agent:** Converts the prepared data into ML-ready data. Adds features to increase predictive accuracy of ML models.
 5. **SQL Database Agent:** Connects to SQL databases to pull data into the data science environment. Creates pipelines to automate data extraction. Performs Joins, Aggregations, and other SQL Query operations.
 
+#### Multi-Agents
+
+1. **SQL Data Analyst Agent:** Connects to SQL databases to pull data into the data science environment. Creates pipelines to automate data extraction. Performs Joins, Aggregations, and other SQL Query operations. Includes a Data Visualization Agent that creates visualizations to help you understand your data.
+
 ### Agents Coming Soon
 
 1. **Data Analyst:** Analyzes data structure, creates exploratory visualizations, and performs correlation analysis to identify relationships.

{ai_data_science_team-0.0.0.9008 → ai_data_science_team-0.0.0.9009}/README.md

@@ -9,13 +9,13 @@
   <em>An AI-powered data science team of agents to help you perform common data science tasks 10X faster</em>
 </div>
 <div align="center">
-  <a href="https://pypi.python.org/pypi/ai-data-science-team"><img src="https://img.shields.io/pypi/v/ai-data-science-team.svg" alt="PyPI"></a>
-  <a href="https://github.com/business-science/ai-data-science-team"><img src="https://img.shields.io/pypi/pyversions/ai-data-science-team.svg" alt="versions"></a>
-  <a href="https://github.com/business-science/ai-data-science-team/blob/main/LICENSE"><img src="https://img.shields.io/github/license/business-science/ai-data-science-team.svg?v" alt="license"></a>
+  <a href="https://pypi.python.org/pypi/ai-data-science-team"><img src="https://img.shields.io/pypi/v/ai-data-science-team.svg?style=for-the-badge" alt="PyPI"></a>
+  <a href="https://github.com/business-science/ai-data-science-team"><img src="https://img.shields.io/pypi/pyversions/ai-data-science-team.svg?style=for-the-badge" alt="versions"></a>
+  <a href="https://github.com/business-science/ai-data-science-team/blob/main/LICENSE"><img src="https://img.shields.io/github/license/business-science/ai-data-science-team.svg?style=for-the-badge" alt="license"></a>
 </div>
 
 
-# Your AI Data Science Team (An Army Of Agents)
+# Your AI Data Science Team (🪖 An Army Of Agents)
 
 **An AI-powered data science team of agents to help you perform common data science tasks 10X faster**.
 
@@ -36,14 +36,16 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
 
 ## Table of Contents
 
-- [Your AI Data Science Team (An Army Of Agents)](#your-ai-data-science-team-an-army-of-agents)
+- [Your AI Data Science Team (🪖 An Army Of Agents)](#your-ai-data-science-team--an-army-of-agents)
   - [Table of Contents](#table-of-contents)
   - [Companies That Want A Custom AI Data Science Team (And AI Apps)](#companies-that-want-a-custom-ai-data-science-team-and-ai-apps)
-  - [Free How To Build AI Agents for Data Scientists Workshop](#free-how-to-build-ai-agents-for-data-scientists-workshop)
+  - [Generative AI for Data Scientists Workshop](#generative-ai-for-data-scientists-workshop)
   - [Data Science Agents](#data-science-agents)
-    - [Coming Soon: Multi-Agents](#coming-soon-multi-agents)
-    - [...And after that, the Multi-Agent Data Science Apps](#and-after-that-the-multi-agent-data-science-apps)
+    - [NEW: Multi-Agents](#new-multi-agents)
+    - [Coming Soon: Data Science Apps](#coming-soon-data-science-apps)
     - [Agents Available Now](#agents-available-now)
+      - [Data Science Agents](#data-science-agents-1)
+      - [Multi-Agents](#multi-agents)
     - [Agents Coming Soon](#agents-coming-soon)
   - [Disclaimer](#disclaimer)
   - [Installation](#installation)
@@ -56,11 +58,11 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
 
 ## Companies That Want A Custom AI Data Science Team (And AI Apps)
 
-Want to have your own _customized_ enterprise-grade AI Data Science Team and domain-specifici AI-powered Apps? 
+Want to have your own _customized_ enterprise-grade AI Data Science Team and *domain-specific* AI-powered Apps? 
 
 **Send inquiries here:** [https://www.business-science.io/contact.html](https://www.business-science.io/contact.html)
 
-## Free How To Build AI Agents for Data Scientists Workshop
+## Generative AI for Data Scientists Workshop
 
 If you're an aspiring data scientist who wants to learn how to build AI Agents and AI Apps for your company that performs Data Science, Business Intelligence, Churn Modeling, Time Series Forecasting, and more, then I'd love to help you. 
 
@@ -72,13 +74,13 @@ This project is a work in progress. New data science agents will be released soo
 
 ![Data Science Team](/img/ai_data_science_team.jpg)
 
-### Coming Soon: Multi-Agents
+### NEW: Multi-Agents
 
-This is the internals of the Business Intelligence SQL Agent I'm working on:
+This is the internals of the SQL Data Analyst Agent that connects to SQL databases to pull data into the data science environment. It creates pipelines to automate data extraction, performs Joins, Aggregations, and other SQL Query operations. And it includes a Data Visualization Agent that creates visualizations to help you understand your data.:
 
 ![Business Intelligence SQL Agent](/img/multi_agent_sql_data_visualization.jpg)
 
-### ...And after that, the Multi-Agent Data Science Apps
+### Coming Soon: Data Science Apps
 
 This is a top secret project I'm working on. It's a multi-agent data science app that performs time series forecasting.
 
@@ -86,12 +88,18 @@ This is a top secret project I'm working on. It's a multi-agent data science app
 
 ### Agents Available Now
 
+#### Data Science Agents
+
 1. **Data Wrangling Agent:** Merges, Joins, Preps and Wrangles data into a format that is ready for data analysis.
 2. **Data Visualization Agent:** Creates visualizations to help you understand your data. Returns JSON serializable plotly visualizations.
 3. **Data Cleaning Agent:** Performs Data Preparation steps including handling missing values, outliers, and data type conversions.
 4. **Feature Engineering Agent:** Converts the prepared data into ML-ready data. Adds features to increase predictive accuracy of ML models.
 5. **SQL Database Agent:** Connects to SQL databases to pull data into the data science environment. Creates pipelines to automate data extraction. Performs Joins, Aggregations, and other SQL Query operations.
 
+#### Multi-Agents
+
+1. **SQL Data Analyst Agent:** Connects to SQL databases to pull data into the data science environment. Creates pipelines to automate data extraction. Performs Joins, Aggregations, and other SQL Query operations. Includes a Data Visualization Agent that creates visualizations to help you understand your data.
+
 ### Agents Coming Soon
 
 1. **Data Analyst:** Analyzes data structure, creates exploratory visualizations, and performs correlation analysis to identify relationships.

ai_data_science_team-0.0.0.9009/ai_data_science_team/_version.py

@@ -0,0 +1 @@
+__version__ = "0.0.0.9009"

{ai_data_science_team-0.0.0.9008 → ai_data_science_team-0.0.0.9009}/ai_data_science_team/agents/__init__.py

@@ -3,4 +3,3 @@ from ai_data_science_team.agents.feature_engineering_agent import make_feature_e
 from ai_data_science_team.agents.data_wrangling_agent import make_data_wrangling_agent, DataWranglingAgent
 from ai_data_science_team.agents.sql_database_agent import make_sql_database_agent, SQLDatabaseAgent
 from ai_data_science_team.agents.data_visualization_agent import make_data_visualization_agent, DataVisualizationAgent
-

{ai_data_science_team-0.0.0.9008 → ai_data_science_team-0.0.0.9009}/ai_data_science_team/agents/data_cleaning_agent.py

@@ -14,7 +14,7 @@ from langgraph.types import Command
 from langgraph.checkpoint.memory import MemorySaver
 
 import os
-import io
+import json
 import pandas as pd
 
 from IPython.display import Markdown
@@ -23,12 +23,18 @@ from ai_data_science_team.templates import(
     node_func_execute_agent_code_on_data, 
     node_func_human_review,
     node_func_fix_agent_code, 
-    node_func_explain_agent_code, 
+    node_func_report_agent_outputs,
     create_coding_agent_graph,
     BaseAgent,
 )
 from ai_data_science_team.tools.parsers import PythonOutputParser
-from ai_data_science_team.tools.regex import relocate_imports_inside_function, add_comments_to_top, format_agent_name, format_recommended_steps
+from ai_data_science_team.tools.regex import (
+    relocate_imports_inside_function, 
+    add_comments_to_top, 
+    format_agent_name, 
+    format_recommended_steps, 
+    get_generic_summary,
+)
 from ai_data_science_team.tools.metadata import get_dataframe_summary
 from ai_data_science_team.tools.logging import log_ai_function
 
@@ -37,7 +43,6 @@ AGENT_NAME = "data_cleaning_agent"
 LOG_PATH = os.path.join(os.getcwd(), "logs/")
 
 
-
 # Class
 class DataCleaningAgent(BaseAgent):
     """
@@ -89,8 +94,8 @@ class DataCleaningAgent(BaseAgent):
         Cleans the provided dataset asynchronously based on user instructions.
     invoke_agent(user_instructions: str, data_raw: pd.DataFrame, max_retries=3, retry_count=0)
         Cleans the provided dataset synchronously based on user instructions.
-    explain_cleaning_steps()
-        Returns an explanation of the cleaning steps performed by the agent.
+    get_workflow_summary()
+        Retrieves a summary of the agent's workflow.
     get_log_summary()
         Retrieves a summary of logged operations if logging is enabled.
     get_state_keys()
@@ -178,7 +183,6 @@ class DataCleaningAgent(BaseAgent):
         self.response=None
         return make_data_cleaning_agent(**self._params)
 
-    
     def ainvoke_agent(self, data_raw: pd.DataFrame, user_instructions: str=None, max_retries:int=3, retry_count:int=0, **kwargs):
         """
         Asynchronously invokes the agent. The response is stored in the response attribute.
@@ -239,15 +243,16 @@ class DataCleaningAgent(BaseAgent):
         self.response = response
         return None
 
-    def explain_cleaning_steps(self):
+    def get_workflow_summary(self, markdown=False):
         """
-        Provides an explanation of the cleaning steps performed by the agent.
-
-        Returns:
-            str: Explanation of the cleaning steps.
+        Retrieves the agent's workflow summary, if logging is enabled.
         """
-        messages = self.response.get("messages", [])
-        return messages
+        if self.response and self.response.get("messages"):
+            summary = get_generic_summary(json.loads(self.response.get("messages")[-1].content))
+            if markdown:
+                return Markdown(summary)
+            else:
+                return summary
 
     def get_log_summary(self, markdown=False):
         """
@@ -255,7 +260,13 @@ class DataCleaningAgent(BaseAgent):
         """
         if self.response:
             if self.response.get('data_cleaner_function_path'):
-                log_details = f"Log Path: {self.response.get('data_cleaner_function_path')}"
+                log_details = f"""
+## Data Cleaning Agent Log Summary:
+
+Function Path: {self.response.get('data_cleaner_function_path')}
+
+Function Name: {self.response.get('data_cleaner_function_name')}
+                """
                 if markdown:
                     return Markdown(log_details) 
                 else:
@@ -462,7 +473,7 @@ def make_data_cleaning_agent(
             Below are summaries of all datasets provided:
             {all_datasets_summary}
 
-            Return the steps as a bullet point list (no code, just the steps).
+            Return steps as a numbered list. You can return short code snippets to demonstrate actions. But do not return a fully coded solution. The code will be generated separately by a Coding Agent.
             
             Avoid these:
             1. Do not include steps to save files.
@@ -633,32 +644,31 @@ def make_data_cleaning_agent(
             function_name=state.get("data_cleaner_function_name"),
         )
     
-    def explain_data_cleaner_code(state: GraphState):        
-        return node_func_explain_agent_code(
+    # Final reporting node
+    def report_agent_outputs(state: GraphState):
+        return node_func_report_agent_outputs(
             state=state,
-            code_snippet_key="data_cleaner_function",
+            keys_to_include=[
+                "recommended_steps",
+                "data_cleaner_function",
+                "data_cleaner_function_path",
+                "data_cleaner_function_name",
+                "data_cleaner_error",
+            ],
             result_key="messages",
-            error_key="data_cleaner_error",
-            llm=llm,  
             role=AGENT_NAME,
-            explanation_prompt_template="""
-            Explain the data cleaning steps that the data cleaning agent performed in this function. 
-            Keep the summary succinct and to the point.\n\n# Data Cleaning Agent:\n\n{code}
-            """,
-            success_prefix="# Data Cleaning Agent:\n\n ",
-            error_message="The Data Cleaning Agent encountered an error during data cleaning. Data could not be explained."
+            custom_title="Data Cleaning Agent Outputs"
         )
-        
-    # Define the graph
+
     node_functions = {
         "recommend_cleaning_steps": recommend_cleaning_steps,
         "human_review": human_review,
         "create_data_cleaner_code": create_data_cleaner_code,
         "execute_data_cleaner_code": execute_data_cleaner_code,
         "fix_data_cleaner_code": fix_data_cleaner_code,
-        "explain_data_cleaner_code": explain_data_cleaner_code
+        "report_agent_outputs": report_agent_outputs, 
     }
-    
+
     app = create_coding_agent_graph(
         GraphState=GraphState,
         node_functions=node_functions,
@@ -666,16 +676,17 @@ def make_data_cleaning_agent(
         create_code_node_name="create_data_cleaner_code",
         execute_code_node_name="execute_data_cleaner_code",
         fix_code_node_name="fix_data_cleaner_code",
-        explain_code_node_name="explain_data_cleaner_code",
+        explain_code_node_name="report_agent_outputs", 
         error_key="data_cleaner_error",
-        human_in_the_loop=human_in_the_loop,  # or False
+        human_in_the_loop=human_in_the_loop,
         human_review_node_name="human_review",
         checkpointer=MemorySaver() if human_in_the_loop else None,
         bypass_recommended_steps=bypass_recommended_steps,
         bypass_explain_code=bypass_explain_code,
     )
-        
+
     return app
+     
 
 
 

{ai_data_science_team-0.0.0.9008 → ai_data_science_team-0.0.0.9009}/ai_data_science_team/agents/data_visualization_agent.py

@@ -17,6 +17,7 @@ from langgraph.types import Command
 from langgraph.checkpoint.memory import MemorySaver
 
 import os
+import json 
 import pandas as pd
 
 from IPython.display import Markdown
@@ -25,7 +26,7 @@ from ai_data_science_team.templates import(
     node_func_execute_agent_code_on_data, 
     node_func_human_review,
     node_func_fix_agent_code, 
-    node_func_explain_agent_code, 
+    node_func_report_agent_outputs,
     create_coding_agent_graph,
     BaseAgent,
 )
@@ -34,7 +35,8 @@ from ai_data_science_team.tools.regex import (
     relocate_imports_inside_function, 
     add_comments_to_top, 
     format_agent_name, 
-    format_recommended_steps
+    format_recommended_steps, 
+    get_generic_summary,
 )
 from ai_data_science_team.tools.metadata import get_dataframe_summary
 from ai_data_science_team.tools.logging import log_ai_function
@@ -93,8 +95,8 @@ class DataVisualizationAgent(BaseAgent):
         Asynchronously generates a visualization based on user instructions.
     invoke_agent(user_instructions: str, data_raw: pd.DataFrame, max_retries=3, retry_count=0)
         Synchronously generates a visualization based on user instructions.
-    explain_visualization_steps()
-        Returns an explanation of the visualization steps performed by the agent.
+    get_workflow_summary()
+        Retrieves a summary of the agent's workflow.
     get_log_summary()
         Retrieves a summary of logged operations if logging is enabled.
     get_plotly_graph()
@@ -257,40 +259,34 @@ class DataVisualizationAgent(BaseAgent):
         self.response = response
         return None
 
-    def explain_visualization_steps(self):
+    def get_workflow_summary(self, markdown=False):
         """
-        Provides an explanation of the visualization steps performed by the agent.
-
-        Returns
-        -------
-        str
-            Explanation of the visualization steps, if any are available.
+        Retrieves the agent's workflow summary, if logging is enabled.
         """
-        if self.response:
-            return self.response.get("messages", [])
-        return []
+        if self.response and self.response.get("messages"):
+            summary = get_generic_summary(json.loads(self.response.get("messages")[-1].content))
+            if markdown:
+                return Markdown(summary)
+            else:
+                return summary
 
     def get_log_summary(self, markdown=False):
         """
         Logs a summary of the agent's operations, if logging is enabled.
+        """
+        if self.response:
+            if self.response.get('data_visualization_function_path'):
+                log_details = f"""
+## Data Visualization Agent Log Summary:
 
-        Parameters
-        ----------
-        markdown : bool, optional
-            If True, returns Markdown-formatted output.
+Function Path: {self.response.get('data_visualization_function_path')}
 
-        Returns
-        -------
-        str or None
-            Summary of logs or None if no logs are available.
-        """
-        if self.response and self.response.get('data_visualization_function_path'):
-            log_details = f"Log Path: {self.response.get('data_visualization_function_path')}"
-            if markdown:
-                return Markdown(log_details)
-            else:
-                return log_details
-        return None
+Function Name: {self.response.get('data_visualization_function_name')}
+                """
+                if markdown:
+                    return Markdown(log_details) 
+                else:
+                    return log_details
 
     def get_plotly_graph(self):
         """
@@ -719,20 +715,20 @@ def make_data_visualization_agent(
             function_name=state.get("data_visualization_function_name"),
         )
     
-    def explain_data_visualization_code(state: GraphState):        
-        return node_func_explain_agent_code(
+    # Final reporting node
+    def report_agent_outputs(state: GraphState):
+        return node_func_report_agent_outputs(
             state=state,
-            code_snippet_key="data_visualization_function",
+            keys_to_include=[
+                "recommended_steps",
+                "data_visualization_function",
+                "data_visualization_function_path",
+                "data_visualization_function_name",
+                "data_visualization_error",
+            ],
             result_key="messages",
-            error_key="data_visualization_error",
-            llm=llm,  
             role=AGENT_NAME,
-            explanation_prompt_template="""
-            Explain the data visualization steps that the data visualization agent performed in this function. 
-            Keep the summary succinct and to the point.\n\n# Data Visualization Agent:\n\n{code}
-            """,
-            success_prefix="# Data Visualization Agent:\n\n ",
-            error_message="The Data Visualization Agent encountered an error during data visualization. No explanation could be provided."
+            custom_title="Data Visualization Agent Outputs"
         )
         
     # Define the graph
@@ -742,7 +738,7 @@ def make_data_visualization_agent(
         "chart_generator": chart_generator,
         "execute_data_visualization_code": execute_data_visualization_code,
         "fix_data_visualization_code": fix_data_visualization_code,
-        "explain_data_visualization_code": explain_data_visualization_code
+        "report_agent_outputs": report_agent_outputs,
     }
     
     app = create_coding_agent_graph(
@@ -752,7 +748,7 @@ def make_data_visualization_agent(
         create_code_node_name="chart_generator",
         execute_code_node_name="execute_data_visualization_code",
         fix_code_node_name="fix_data_visualization_code",
-        explain_code_node_name="explain_data_visualization_code",
+        explain_code_node_name="report_agent_outputs",
         error_key="data_visualization_error",
         human_in_the_loop=human_in_the_loop,  # or False
         human_review_node_name="human_review",