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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ai-data-science-team
-Version: 0.0.0.9008
+Version: 0.0.0.9010
 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,26 @@ 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
+Requires-Dist: psutil
+Provides-Extra: machine-learning
+Requires-Dist: h2o; extra == "machine-learning"
+Requires-Dist: mlflow; extra == "machine-learning"
+Provides-Extra: all
+Requires-Dist: h2o; extra == "all"
+Requires-Dist: mlflow; 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
@@ -39,7 +48,7 @@ Dynamic: summary
 <div align="center">
   <a href="https://github.com/business-science/ai-data-science-team">
     <picture>
-      <img src="/img/ai_data_science_team_logo.jpg" alt="AI Data Science Team" width="400">
+      <img src="/img/ai_data_science_team_logo_small.jpg" alt="AI Data Science Team" width="400">
   </picture>
   </a>
 </div>
@@ -47,13 +56,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 +83,19 @@ 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)
+    - [Data Science Apps](#data-science-apps)
+    - [Apps Available Now](#apps-available-now)
+      - [🔥 Agentic Applications](#-agentic-applications)
     - [Agents Available Now](#agents-available-now)
+      - [🔥🔥 NEW! Machine Learning Agents](#-new-machine-learning-agents)
+      - [Data Science Agents](#data-science-agents-1)
+      - [Multi-Agents](#multi-agents)
     - [Agents Coming Soon](#agents-coming-soon)
   - [Disclaimer](#disclaimer)
   - [Installation](#installation)
@@ -94,11 +108,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,32 +124,50 @@ 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
+### 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.
 
-![Multi-Agent Data Science App](/img/ai_powered_apps.jpg)
+![Multi-Agent Data Science App](/img/ai_powered_apps.jpg) 
+
+### Apps Available Now
+
+[See all available apps here](/apps)
+
+#### 🔥 Agentic Applications
+
+1. **SQL Database Agent App:** Connects any SQL Database, generates SQL queries from natural language, and returns data as a downloadable table. [See Application](/apps/sql-database-agent-app/)
 
 ### Agents Available Now
 
-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.
+#### 🔥🔥 NEW! Machine Learning Agents
+
+1. **🔥 H2O Machine Learning Agent:** Builds and logs 100's of high-performance machine learning models. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/ml_agents/h2o_machine_learning_agent.ipynb)
+2. **🔥 MLflow Tools Agent (MLOps):** This agent has 11+ tools for managing models, ML projects, and making production ML predictions with MLflow. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/ml_agents/mlflow_tools_agent.ipynb)
+
+#### Data Science Agents
+
+1. **Data Wrangling Agent:** Merges, Joins, Preps and Wrangles data into a format that is ready for data analysis. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/data_wrangling_agent.ipynb)
+2. **Data Visualization Agent:** Creates visualizations to help you understand your data. Returns JSON serializable plotly visualizations. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/data_visualization_agent.ipynb)
+3. **Data Cleaning Agent:** Performs Data Preparation steps including handling missing values, outliers, and data type conversions. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/data_cleaning_agent.ipynb)
+4. **Feature Engineering Agent:** Converts the prepared data into ML-ready data. Adds features to increase predictive accuracy of ML models. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/feature_engineering_agent.ipynb)
+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. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/sql_database_agent.ipynb)
+
+#### 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. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/multiagents/sql_data_analyst.ipynb)
 
 ### Agents Coming Soon
 
 1. **Data Analyst:** Analyzes data structure, creates exploratory visualizations, and performs correlation analysis to identify relationships.
-2. **Machine Learning Agent:** Builds and logs the machine learning models.
-3. **Interpretability Agent:** Performs Interpretable ML to explain why the model returned predictions including which features were the most important to the model.
-4. **Supervisor:** Forms task list. Moderates sub-agents. Returns completed assignment. 
+2. **Interpretability Agent:** Performs Interpretable ML to explain why the model returned predictions including which features were the most important to the model.
+3. **Supervisor:** Forms task list. Moderates sub-agents. Returns completed assignment. 
 
 ## Disclaimer
 

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

@@ -1,7 +1,7 @@
 <div align="center">
   <a href="https://github.com/business-science/ai-data-science-team">
     <picture>
-      <img src="/img/ai_data_science_team_logo.jpg" alt="AI Data Science Team" width="400">
+      <img src="/img/ai_data_science_team_logo_small.jpg" alt="AI Data Science Team" width="400">
   </picture>
   </a>
 </div>
@@ -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,19 @@ 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)
+    - [Data Science Apps](#data-science-apps)
+    - [Apps Available Now](#apps-available-now)
+      - [🔥 Agentic Applications](#-agentic-applications)
     - [Agents Available Now](#agents-available-now)
+      - [🔥🔥 NEW! Machine Learning Agents](#-new-machine-learning-agents)
+      - [Data Science Agents](#data-science-agents-1)
+      - [Multi-Agents](#multi-agents)
     - [Agents Coming Soon](#agents-coming-soon)
   - [Disclaimer](#disclaimer)
   - [Installation](#installation)
@@ -56,11 +61,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,32 +77,50 @@ 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
+### 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.
 
-![Multi-Agent Data Science App](/img/ai_powered_apps.jpg)
+![Multi-Agent Data Science App](/img/ai_powered_apps.jpg) 
+
+### Apps Available Now
+
+[See all available apps here](/apps)
+
+#### 🔥 Agentic Applications
+
+1. **SQL Database Agent App:** Connects any SQL Database, generates SQL queries from natural language, and returns data as a downloadable table. [See Application](/apps/sql-database-agent-app/)
 
 ### Agents Available Now
 
-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.
+#### 🔥🔥 NEW! Machine Learning Agents
+
+1. **🔥 H2O Machine Learning Agent:** Builds and logs 100's of high-performance machine learning models. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/ml_agents/h2o_machine_learning_agent.ipynb)
+2. **🔥 MLflow Tools Agent (MLOps):** This agent has 11+ tools for managing models, ML projects, and making production ML predictions with MLflow. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/ml_agents/mlflow_tools_agent.ipynb)
+
+#### Data Science Agents
+
+1. **Data Wrangling Agent:** Merges, Joins, Preps and Wrangles data into a format that is ready for data analysis. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/data_wrangling_agent.ipynb)
+2. **Data Visualization Agent:** Creates visualizations to help you understand your data. Returns JSON serializable plotly visualizations. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/data_visualization_agent.ipynb)
+3. **Data Cleaning Agent:** Performs Data Preparation steps including handling missing values, outliers, and data type conversions. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/data_cleaning_agent.ipynb)
+4. **Feature Engineering Agent:** Converts the prepared data into ML-ready data. Adds features to increase predictive accuracy of ML models. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/feature_engineering_agent.ipynb)
+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. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/sql_database_agent.ipynb)
+
+#### 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. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/multiagents/sql_data_analyst.ipynb)
 
 ### Agents Coming Soon
 
 1. **Data Analyst:** Analyzes data structure, creates exploratory visualizations, and performs correlation analysis to identify relationships.
-2. **Machine Learning Agent:** Builds and logs the machine learning models.
-3. **Interpretability Agent:** Performs Interpretable ML to explain why the model returned predictions including which features were the most important to the model.
-4. **Supervisor:** Forms task list. Moderates sub-agents. Returns completed assignment. 
+2. **Interpretability Agent:** Performs Interpretable ML to explain why the model returned predictions including which features were the most important to the model.
+3. **Supervisor:** Forms task list. Moderates sub-agents. Returns completed assignment. 
 
 ## Disclaimer
 

ai_data_science_team-0.0.0.9010/ai_data_science_team/_version.py

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

{ai_data_science_team-0.0.0.9008 → ai_data_science_team-0.0.0.9010}/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.9010}/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,21 +23,26 @@ 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.metadata import get_dataframe_summary
-from ai_data_science_team.tools.logging import log_ai_function
+from ai_data_science_team.parsers.parsers import PythonOutputParser
+from ai_data_science_team.utils.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.dataframe import get_dataframe_summary
+from ai_data_science_team.utils.logging import log_ai_function
 
 # Setup
 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,8 +183,7 @@ 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):
+    async 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.
 
@@ -200,7 +204,7 @@ class DataCleaningAgent(BaseAgent):
         --------
             None. The response is stored in the response attribute.
         """
-        response = self._compiled_graph.ainvoke({
+        response = await self._compiled_graph.ainvoke({
             "user_instructions": user_instructions,
             "data_raw": data_raw.to_dict(),
             "max_retries": max_retries,
@@ -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.9010/ai_data_science_team/agents/data_loader_tools_agent.py

@@ -0,0 +1,69 @@
+
+
+
+from typing import Any, Optional, Annotated, Sequence, List, Dict
+import operator
+
+import pandas as pd
+import os
+
+from IPython.display import Markdown
+
+from langchain_core.messages import BaseMessage, AIMessage
+
+from langgraph.prebuilt import create_react_agent, ToolNode
+from langgraph.prebuilt.chat_agent_executor import AgentState
+from langgraph.graph import START, END, StateGraph
+
+from ai_data_science_team.templates import BaseAgent
+from ai_data_science_team.utils.regex import format_agent_name
+from ai_data_science_team.tools.data_loader import (
+    load_directory,
+    load_file,
+    list_directory_contents,
+    list_directory_recursive,
+    get_file_info,
+    search_files_by_pattern,
+)
+
+AGENT_NAME = "data_loader_tools_agent"
+
+tools = [
+    load_directory,
+    load_file,
+    list_directory_contents,
+    list_directory_recursive,
+    get_file_info,
+    search_files_by_pattern,
+]
+
+
+
+def make_data_loader_tools_agent(
+    model: Any,
+    directory: Optional[str] = os.getcwd(),
+):
+    """
+    Creates a Data Loader Agent that can interact with data loading tools.
+    
+    Parameters:
+    ----------
+    model : langchain.llms.base.LLM
+        The language model used to generate the tool calling agent.
+    directory : str, optional
+        The directory to search for files. Defaults to the current working directory.
+    
+    Returns:
+    --------
+    Data Loader Agent
+        An agent that can interact with data loading tools.
+    """
+    
+    class GraphState(AgentState):
+        internal_messages: Annotated[Sequence[BaseMessage], operator.add]
+        directory: str
+        user_instructions: str
+        data_artifacts: dict
+    
+    pass
+